仿生人MCP接入点

bbit_ai/app_mcp/utils/config.py

@@ -0,0 +1,131 @@
+"""
+配置管理工具
+"""
+
+import os
+import configparser
+import uuid
+from typing import Optional
+from pathlib import Path
+
+
+class ConfigManager:
+    """配置管理器"""
+
+    def __init__(self, config_file: str = "data/.mcp-endpoint-server.cfg"):
+        self.config_file = config_file
+        self.config = configparser.ConfigParser()
+        self._load_config()
+
+    def _load_config(self):
+        """加载配置文件"""
+        if os.path.exists(self.config_file):
+            self.config.read(self.config_file, encoding="utf-8")
+            # 检查并生成key
+            self._check_and_generate_key()
+        else:
+            # 如果配置文件不存在，从根目录拷贝
+            self._copy_config_from_root()
+
+    def _copy_config_from_root(self):
+        """从根目录拷贝配置文件到data目录"""
+        root_config = "mcp-endpoint-server.cfg"
+        if os.path.exists(root_config):
+            # 确保data目录存在
+            os.makedirs(os.path.dirname(self.config_file), exist_ok=True)
+
+            # 拷贝配置文件
+            import shutil
+
+            shutil.copy2(root_config, self.config_file)
+
+            # 重新加载配置
+            self.config.read(self.config_file, encoding="utf-8")
+            # 检查并生成key
+            self._check_and_generate_key()
+        else:
+            # 如果根目录也没有配置文件，则创建默认配置
+            self._create_default_config()
+
+    def _check_and_generate_key(self):
+        """检查key是否存在且长度足够，如果不足则生成新的"""
+        try:
+            current_key = self.config.get("server", "key", fallback="")
+            if not current_key or len(current_key) < 32:
+                # 生成32位随机密码
+                new_key = self._generate_random_key()
+                self.config.set("server", "key", new_key)
+
+                # 保存到配置文件
+                with open(self.config_file, "w", encoding="utf-8") as f:
+                    self.config.write(f)
+
+                print(f"已自动生成新的32位密钥: {new_key}")
+        except Exception as e:
+            print(f"检查密钥时发生错误: {e}")
+
+    def _generate_random_key(self) -> str:
+        """生成指定长度的随机密钥"""
+        # 使用UUID生成密钥，移除连字符
+        return str(uuid.uuid4()).replace("-", "")
+
+    def _create_default_config(self):
+        """创建默认配置"""
+        self.config["server"] = {
+            "host": "127.0.0.1",
+            "port": "8004",
+            "debug": "false",
+            "log_level": "INFO",
+            "key": self._generate_random_key(),  # 生成默认密钥
+        }
+
+        self.config["websocket"] = {
+            "max_connections": "1000",
+            "ping_interval": "30",
+            "ping_timeout": "10",
+            "close_timeout": "10",
+        }
+
+        self.config["security"] = {"allowed_origins": "*", "enable_cors": "true"}
+
+        self.config["logging"] = {
+            "log_file": "logs/mcp_server.log",
+            "max_file_size": "10MB",
+            "backup_count": "5",
+        }
+
+        # 确保目录存在
+        os.makedirs(os.path.dirname(self.config_file), exist_ok=True)
+
+        # 保存默认配置
+        with open(self.config_file, "w", encoding="utf-8") as f:
+            self.config.write(f)
+
+    def get(self, section: str, key: str, default: Optional[str] = None) -> str:
+        """获取配置值"""
+        try:
+            return self.config.get(section, key)
+        except (configparser.NoSectionError, configparser.NoOptionError):
+            return default
+
+    def getint(self, section: str, key: str, default: int = 0) -> int:
+        """获取整数配置值"""
+        try:
+            return self.config.getint(section, key)
+        except (configparser.NoSectionError, configparser.NoOptionError, ValueError):
+            return default
+
+    def getboolean(self, section: str, key: str, default: bool = False) -> bool:
+        """获取布尔配置值"""
+        try:
+            return self.config.getboolean(section, key)
+        except (configparser.NoSectionError, configparser.NoOptionError, ValueError):
+            return default
+
+    def reload(self):
+        """重新加载配置"""
+        self._load_config()
+
+
+# 全局配置实例
+config = ConfigManager()