update:优化文档

Author: openrz

README.md

@@ -19,122 +19,11 @@
      └─── 消息转发 ───────┴─── 消息转发 ───────┘
 ```
 
-## 安装依赖
+## 运行
 
 ```bash
-conda remove -n mcp-endpoint-server --all -y
-conda create -n mcp-endpoint-server python=3.10 -y
-conda activate mcp-endpoint-server
 
-pip install -r requirements.txt
-
-cp mcp-endpoint-server.cfg data/.mcp-endpoint-server.cfg
-
-```
-
-## 配置说明
-
-### 配置文件位置
-- 示例配置: `mcp-endpoint-server.cfg`
-- 实际配置: `data/.mcp-endpoint-server.cfg`
-
-### 配置项说明
-
-```ini
-[server]
-host = 127.0.0.1          # 服务器监听地址
-port = 8004               # 服务器监听端口
-debug = false             # 调试模式
-log_level = INFO          # 日志级别
-
-[websocket]
-max_connections = 1000    # 最大连接数
-ping_interval = 30        # 心跳间隔(秒)
-ping_timeout = 10         # 心跳超时(秒)
-close_timeout = 10        # 关闭超时(秒)
-
-[security]
-allowed_origins = *       # 允许的源
-enable_cors = true        # 启用CORS
-
-[logging]
-log_file = logs/mcp_server.log  # 日志文件路径
-max_file_size = 10MB            # 日志文件最大大小
-backup_count = 5                # 日志备份数量
-```
-
-## 启动服务器
-
-```bash
-python main.py
 ```
 
-### 测试数据
-```json
-{"id":0,"jsonrpc":"2.0","method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{"sampling":{},"roots":{"listChanged":false}},"clientInfo":{"name":"xz-mcp-broker","version":"0.0.1"}}}
-
-
-{"jsonrpc":"2.0","method":"notifications/initialized"}
-
-
-{"id":1,"jsonrpc":"2.0","method":"tools/list","params":{}}
-
-
-{"id":2,"jsonrpc":"2.0","method":"ping","params":{}}
-
-
-{"id":10,"jsonrpc":"2.0","method":"tools/call","params":{"name":"calculator","arguments":{"python_expression":"130000 * 130000"},"serialNumber":null}}
-```
-
-### Docker部署
-
-```dockerfile
-FROM python:3.9-slim
-
-WORKDIR /app
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-
-COPY . .
-EXPOSE 8004
-
-CMD ["python", "main.py"]
-```
-
-## 监控和维护
-
-### 日志查看
-```bash
-tail -f logs/mcp_server.log
-```
-
-### 连接状态检查
-```bash
-curl http://127.0.0.1:8004/stats
-```
-
-### 健康检查
-```bash
-curl http://127.0.0.1:8004/health
-```
-
-## 开发说明
-
-### 项目结构
-```
-mcp-endpoint-server/
-├── src/
-│   ├── core/                 # 核心模块
-│   │   └── connection_manager.py
-│   ├── handlers/             # 处理器
-│   │   └── websocket_handler.py
-│   ├── utils/                # 工具模块
-│   │   ├── config.py
-│   │   └── logger.py
-│   └── server.py             # 主服务器
-├── data/                     # 配置文件目录
-├── logs/                     # 日志目录
-├── main.py                   # 主入口
-├── requirements.txt          # 依赖文件
-└── README.md                 # 说明文档
-```
+## 技术细节
+[技术细节](./README_dev.md)

README_dev.md

@@ -0,0 +1,51 @@
+# MCP Endpoint Server
+
+## 安装依赖
+
+```bash
+conda remove -n mcp-endpoint-server --all -y
+conda create -n mcp-endpoint-server python=3.10 -y
+conda activate mcp-endpoint-server
+
+pip install -r requirements.txt
+
+python main.py
+```
+
+### 测试数据
+```json
+{"id":0,"jsonrpc":"2.0","method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{"sampling":{},"roots":{"listChanged":false}},"clientInfo":{"name":"xz-mcp-broker","version":"0.0.1"}}}
+
+
+{"jsonrpc":"2.0","method":"notifications/initialized"}
+
+
+{"id":1,"jsonrpc":"2.0","method":"tools/list","params":{}}
+
+
+{"id":2,"jsonrpc":"2.0","method":"ping","params":{}}
+
+
+{"id":10,"jsonrpc":"2.0","method":"tools/call","params":{"name":"calculator","arguments":{"python_expression":"130000 * 130000"},"serialNumber":null}}
+```
+
+## 开发说明
+
+### 项目结构
+```
+mcp-endpoint-server/
+├── src/
+│   ├── core/                 # 核心模块
+│   │   └── connection_manager.py
+│   ├── handlers/             # 处理器
+│   │   └── websocket_handler.py
+│   ├── utils/                # 工具模块
+│   │   ├── config.py
+│   │   └── logger.py
+│   └── server.py             # 主服务器
+├── data/                     # 配置文件目录
+├── logs/                     # 日志目录
+├── main.py                   # 主入口
+├── requirements.txt          # 依赖文件
+└── README.md                 # 说明文档
+```

src/utils/config.py

@@ -4,6 +4,7 @@
 
 import os
 import configparser
+import uuid
 from typing import Optional
 from pathlib import Path
 
@@ -20,17 +21,62 @@ 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"] = {