1 files changed, 96 insertions, 1 deletions

diff --git a/HydroRoll/config.py b/HydroRoll/config.py
index 1e132e4..2029575 100644
--- a/HydroRoll/config.py
+++ b/HydroRoll/config.py
@@ -3,6 +3,11 @@ import sys
 import platform
 from importlib.metadata import version
 from iamai import Plugin
+import os
+import pickle
+import threading
+
+
 # 创建全局 ArgumentParser 对象
 global_parser = argparse.ArgumentParser(description='HydroRoll[水系] 全局命令参数')
 
@@ -15,6 +20,8 @@ class GlobalConfig:
     _iamai_version = version('iamai')
     _python_ver = sys.version
     _python_ver_raw= '.'.join(map(str, platform.python_version_tuple()[:3]))
+    current_path = os.path.dirname(os.path.abspath('__file__'))
+
     # 定义系统组件
     class HydroSystem:
         def __init__(self):
@@ -26,4 +33,92 @@ class GlobalConfig:
             self.help = '\n'.join(self.parser.format_help().replace('\r\n', '\n').replace('\r', '').split('\n')[2:-3])
     class HydroBot:
         def __init__(self) -> None:
-            self.parser = argparse.ArgumentParser(description="Bot命令")
\ No newline at end of file
+            self.parser = argparse.ArgumentParser(description="Bot命令")
+        
+
+class ConfigManager:
+    def __init__(self, directory):
+        self.directory = directory
+        self.configs = {}
+        self.locks = {}
+    
+    def get(self, filename, section, key, default=None):
+        if not self._check_file_exists(filename):
+            return default
+        config = self._get_config(filename)
+        if section not in config:
+            return default
+        return config[section].get(key, default)
+    
+    def set(self, filename, section, key, value):
+        config = self._get_config(filename)
+        if section not in config:
+            config[section] = {}
+        config[section][key] = value
+        self._save_config(filename, config)
+    
+    def delete(self, filename, section, key):
+        config = self._get_config(filename)
+        if section not in config:
+            return
+        if key not in config[section]:
+            return
+        del config[section][key]
+        self._save_config(filename, config)
+    
+    def sections(self, filename):
+        if not self._check_file_exists(filename):
+            return []
+        config = self._get_config(filename)
+        return list(config.keys())
+    
+    def section_items(self, filename, section):
+        if not self._check_file_exists(filename):
+            return []
+        config = self._get_config(filename)
+        if section not in config:
+            return []
+        return list(config[section].items())
+    
+    def files(self):
+        return [filename for filename in os.listdir(self.directory) if filename.endswith('.dat')]
+
+    def _get_lock(self, filename):
+        if filename not in self.locks:
+            self.locks[filename] = threading.Lock()
+        return self.locks[filename]
+    
+    def _get_config(self, filename):
+        with self._get_lock(filename):
+            if filename not in self.configs:
+                filepath = os.path.join(self.directory, filename)
+                if os.path.exists(filepath):
+                    try:
+                        with open(filepath, 'rb') as f:
+                            self.configs[filename] = pickle.load(f)
+                    except:
+                        pass
+                if filename not in self.configs:
+                    self.configs[filename] = {}
+        return self.configs[filename]
+    
+    def _save_config(self, filename, config):
+        with self._get_lock(filename):
+            try:
+                filepath = os.path.join(self.directory, filename)
+                backuppath = os.path.join(self.directory, filename + '.bak')
+                lockpath = os.path.join(self.directory, filename + '.lock')
+                with open(lockpath, 'wb') as f:
+                    pass
+                if os.path.exists(filepath):
+                    os.replace(filepath, backuppath)
+                with open(filepath, 'wb') as f:
+                    pickle.dump(config, f)
+                os.remove(backuppath)
+                os.remove(lockpath)
+            except:
+                pass
+    
+    def _check_file_exists(self, filename):
+        filepath = os.path.join(self.directory, filename)
+        return os.path.exists(filepath) and filename.endswith('.dat')