aicodeswitch 3.0.6 → 3.0.8

package/README.md

@@ -153,6 +153,11 @@ aicodeswitch内部，会根据“源类型”来转换数据。例如，你的
 你可以在 aicodeswitch 中集中统一管理 skills，把skills分发给claude code和codex，随时启用和停用skills。
 另外，你可以基于自然语言搜索skills，找到skill之后，支持一键安装。
 
+## MCP管理
+
+你可以在 aicodeswitch 中集中统一管理 MCP，把MCP分发给claude code和codex，随时启用和停用MCP。
+并且，你还可以通过MCP来实现图像理解。
+
 ## 日志
 
 在**日志**页面，您可以查看：

package/bin/cli.js

@@ -7,7 +7,7 @@ const commands = {
   start: require('./start'),
   stop: require('./stop'),
   restart: require('./restart'),
-  update: require('./update'),
+  upgrade: require('./upgrade'),
   restore: require('./restore'),
   version: require('./version'),
   ui: require('./ui'),
@@ -22,7 +22,7 @@ Commands:
   stop       Stop the AI Code Switch server
   restart    Restart the AI Code Switch server
   ui         Open the web UI in browser (starts server if needed)
-  update     Update to the latest version and restart
+  upgrade    Upgrade to the latest version and restart
   restore    Restore original configuration files
   version    Show current version information
 
@@ -31,7 +31,7 @@ Example:
   aicos stop
   aicos restart
   aicos ui
-  aicos update
+  aicos upgrade
   aicos restore
   aicos restore claude-code
   aicos restore codex

package/bin/{update.js → upgrade.js}

@@ -29,7 +29,7 @@ const getLatestVersion = () => {
       path: `/${PACKAGE_NAME}`,
       method: 'GET',
       headers: {
-        'User-Agent': 'aicodeswitch-update'
+        'User-Agent': 'aicodeswitch-upgrade'
       }
     };
 
@@ -142,8 +142,8 @@ const compareVersions = (v1, v2) => {
   return 0;
 };
 
-// 主更新逻辑
-const update = async () => {
+// 主升级逻辑
+const upgrade = async () => {
   console.log('\n');
 
   const currentVersion = getCurrentVersion();
@@ -180,8 +180,8 @@ const update = async () => {
   } catch (err) {
     checkSpinner.fail(chalk.red('Failed to check for updates'));
     console.log(chalk.yellow(`\nError: ${err.message}\n`));
-    console.log(chalk.white('You can manually update by running:\n'));
-    console.log(chalk.cyan('  npm update -g aicodeswitch\n'));
+    console.log(chalk.white('You can manually upgrade by running:\n'));
+    console.log(chalk.cyan('  npm install -g aicodeswitch\n'));
     process.exit(1);
   }
 
@@ -212,7 +212,7 @@ const update = async () => {
     chalk.yellow.bold('⬆️  New version available!\n\n') +
     chalk.white('Current: ') + chalk.gray(currentVersion) + '\n' +
     chalk.white('Latest:  ') + chalk.green.bold(latestVersion) + '\n\n' +
-    chalk.gray('Preparing to update...'),
+    chalk.gray('Preparing to upgrade...'),
     {
       padding: 1,
       margin: 1,
@@ -230,9 +230,9 @@ const update = async () => {
     console.log(boxen(
       chalk.yellow.bold('⚠️  Sudo privileges required\n\n') +
       chalk.white('This operation requires ') + chalk.yellow.bold('sudo') + chalk.white(' privileges.\n\n') +
-      chalk.white('Please run the following command to update:\n\n') +
+      chalk.white('Please run the following command to upgrade:\n\n') +
       chalk.cyan.bold('  sudo npm install -g ' + PACKAGE_NAME + '@latest\n\n') +
-      chalk.gray('After updating, run ') + chalk.cyan('aicos restart') + chalk.gray(' to restart the server.'),
+      chalk.gray('After upgrading, run ') + chalk.cyan('aicos restart') + chalk.gray(' to restart the server.'),
       {
         padding: 1,
         margin: 1,
@@ -244,26 +244,26 @@ const update = async () => {
     process.exit(0);
   }
 
-  // 执行更新
-  const updateSpinner = ora({
-    text: chalk.cyan('Updating to latest version...'),
+  // 执行升级
+  const upgradeSpinner = ora({
+    text: chalk.cyan('Upgrading to latest version...'),
     color: 'cyan'
   }).start();
 
   try {
     await execCommand('npm', ['install', '-g', `${PACKAGE_NAME}@latest`]);
-    updateSpinner.succeed(chalk.green('Update successful!'));
+    upgradeSpinner.succeed(chalk.green('Upgrade successful!'));
   } catch (err) {
-    updateSpinner.fail(chalk.red('Update failed'));
-    console.log(chalk.yellow(`\nUpdate failed with error code ${err.code || 'unknown'}\n`));
-    console.log(chalk.white('You can try manually updating:\n'));
+    upgradeSpinner.fail(chalk.red('Upgrade failed'));
+    console.log(chalk.yellow(`\nUpgrade failed with error code ${err.code || 'unknown'}\n`));
+    console.log(chalk.white('You can try manually upgrading:\n'));
     console.log(chalk.cyan(`  npm install -g ${PACKAGE_NAME}@latest\n`));
     process.exit(1);
   }
 
   console.log('');
   console.log(boxen(
-    chalk.green.bold('✓ Successfully updated!\n\n') +
+    chalk.green.bold('✓ Successfully upgraded!\n\n') +
     chalk.white('Previous version: ') + chalk.gray(currentVersion) + '\n' +
     chalk.white('New version:     ') + chalk.green.bold(latestVersion),
     {
@@ -280,4 +280,4 @@ const update = async () => {
   console.log('\n');
 };
 
-module.exports = update;
+module.exports = upgrade;

package/bin/version.js

@@ -38,7 +38,7 @@ const version = () => {
   ));
 
   console.log(chalk.cyan('💡 Tips:\n'));
-  console.log(chalk.white('  • Check for updates: ') + chalk.yellow('aicos update'));
+  console.log(chalk.white('  • Check for updates: ') + chalk.yellow('aicos upgrade'));
   console.log(chalk.white('  • Restart server:     ') + chalk.yellow('aicos restart\n'));
 
   process.exit(0);

package/dist/server/fs-database.js

@@ -45,6 +45,7 @@ class FileSystemDatabaseManager {
     get errorLogsFile() { return path_1.default.join(this.dataPath, 'error-logs.json'); }
     get blacklistFile() { return path_1.default.join(this.dataPath, 'blacklist.json'); }
     get statisticsFile() { return path_1.default.join(this.dataPath, 'statistics.json'); }
+    get mcpFile() { return path_1.default.join(this.dataPath, 'mcps.json'); }
     // 创建空的统计数据结构
     createEmptyStatistics() {
         return {
@@ -130,6 +131,12 @@ class FileSystemDatabaseManager {
             writable: true,
             value: new Map()
         });
+        Object.defineProperty(this, "mcps", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
         // 持久化统计数据
         Object.defineProperty(this, "statistics", {
             enumerable: true,
@@ -215,6 +222,7 @@ class FileSystemDatabaseManager {
                 this.loadErrorLogs(),
                 this.loadBlacklist(),
                 this.loadStatistics(),
+                this.loadMCPs(),
             ]);
         });
     }
@@ -684,6 +692,22 @@ class FileSystemDatabaseManager {
             yield promises_1.default.writeFile(this.statisticsFile, JSON.stringify(this.statistics, null, 2));
         });
     }
+    loadMCPs() {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const data = yield promises_1.default.readFile(this.mcpFile, 'utf-8');
+                this.mcps = JSON.parse(data);
+            }
+            catch (_a) {
+                this.mcps = [];
+            }
+        });
+    }
+    saveMCPs() {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield promises_1.default.writeFile(this.mcpFile, JSON.stringify(this.mcps, null, 2));
+        });
+    }
     ensureDefaultConfig() {
         return __awaiter(this, void 0, void 0, function* () {
             if (!this.config) {
@@ -1263,6 +1287,101 @@ class FileSystemDatabaseManager {
             return count;
         });
     }
+    /**
+     * 搜索请求日志内容
+     * @param query 搜索关键词
+     * @param limit 返回��量限制
+     * @param offset 偏移量
+     * @returns 匹配的日志列表
+     */
+    searchLogs(query_1) {
+        return __awaiter(this, arguments, void 0, function* (query, limit = 100, offset = 0) {
+            const searchQuery = query.toLowerCase().trim();
+            if (!searchQuery) {
+                return this.getLogs(limit, offset);
+            }
+            const allMatches = [];
+            const sortedShards = [...this.logShardsIndex].sort((a, b) => b.endTime - a.endTime);
+            // 遍历所有分片进行搜索
+            for (const shard of sortedShards) {
+                const shardLogs = yield this.loadLogShard(shard.filename);
+                for (const log of shardLogs) {
+                    if (this.logMatchesQuery(log, searchQuery)) {
+                        allMatches.push(log);
+                    }
+                }
+            }
+            // 按时间倒序排列并分页
+            return allMatches
+                .sort((a, b) => b.timestamp - a.timestamp)
+                .slice(offset, offset + limit);
+        });
+    }
+    /**
+     * 搜索请求日志内容数量
+     * @param query 搜索关键词
+     * @returns 匹配的日志数量
+     */
+    searchLogsCount(query) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const searchQuery = query.toLowerCase().trim();
+            if (!searchQuery) {
+                return this.getLogsCount();
+            }
+            let count = 0;
+            for (const shard of this.logShardsIndex) {
+                const shardLogs = yield this.loadLogShard(shard.filename);
+                for (const log of shardLogs) {
+                    if (this.logMatchesQuery(log, searchQuery)) {
+                        count++;
+                    }
+                }
+            }
+            return count;
+        });
+    }
+    /**
+     * 检查日志是否匹配搜索查询
+     */
+    logMatchesQuery(log, query) {
+        // 搜索请求体
+        if (log.body) {
+            const bodyStr = typeof log.body === 'string' ? log.body : JSON.stringify(log.body);
+            if (bodyStr.toLowerCase().includes(query)) {
+                return true;
+            }
+        }
+        // 搜索响应体
+        if (log.responseBody && typeof log.responseBody === 'string') {
+            if (log.responseBody.toLowerCase().includes(query)) {
+                return true;
+            }
+        }
+        // 搜索流式响应块
+        if (log.streamChunks && log.streamChunks.length > 0) {
+            for (const chunk of log.streamChunks) {
+                if (chunk.toLowerCase().includes(query)) {
+                    return true;
+                }
+            }
+        }
+        // 搜索错误信息
+        if (log.error && log.error.toLowerCase().includes(query)) {
+            return true;
+        }
+        // 搜索路径
+        if (log.path && log.path.toLowerCase().includes(query)) {
+            return true;
+        }
+        // 搜索模型名称
+        if (log.requestModel && log.requestModel.toLowerCase().includes(query)) {
+            return true;
+        }
+        if (log.targetModel && log.targetModel.toLowerCase().includes(query)) {
+            return true;
+        }
+        return false;
+    }
     // Error log operations
     addErrorLog(log) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -1294,6 +1413,78 @@ class FileSystemDatabaseManager {
             return count;
         });
     }
+    /**
+     * 搜索错误日志内容
+     * @param query 搜索关键词
+     * @param limit 返回数量限制
+     * @param offset 偏移量
+     * @returns 匹配的错误日志列表
+     */
+    searchErrorLogs(query_1) {
+        return __awaiter(this, arguments, void 0, function* (query, limit = 100, offset = 0) {
+            const searchQuery = query.toLowerCase().trim();
+            if (!searchQuery) {
+                return this.getErrorLogs(limit, offset);
+            }
+            const matches = this.errorLogs.filter(log => this.errorLogMatchesQuery(log, searchQuery));
+            return matches
+                .sort((a, b) => b.timestamp - a.timestamp)
+                .slice(offset, offset + limit);
+        });
+    }
+    /**
+     * 搜索错误日志内容数量
+     * @param query 搜索关键词
+     * @returns 匹配的错误日志数量
+     */
+    searchErrorLogsCount(query) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const searchQuery = query.toLowerCase().trim();
+            if (!searchQuery) {
+                return this.getErrorLogsCount();
+            }
+            return this.errorLogs.filter(log => this.errorLogMatchesQuery(log, searchQuery)).length;
+        });
+    }
+    /**
+     * 检查错误日志是否匹配搜索查询
+     */
+    errorLogMatchesQuery(log, query) {
+        // 搜索错误信息
+        if (log.errorMessage && log.errorMessage.toLowerCase().includes(query)) {
+            return true;
+        }
+        // 搜索错误堆栈
+        if (log.errorStack && log.errorStack.toLowerCase().includes(query)) {
+            return true;
+        }
+        // 搜索请求体
+        if (log.requestBody) {
+            const bodyStr = typeof log.requestBody === 'string' ? log.requestBody : JSON.stringify(log.requestBody);
+            if (bodyStr.toLowerCase().includes(query)) {
+                return true;
+            }
+        }
+        // 搜索响应体
+        if (log.responseBody) {
+            const bodyStr = typeof log.responseBody === 'string' ? log.responseBody : JSON.stringify(log.responseBody);
+            if (bodyStr.toLowerCase().includes(query)) {
+                return true;
+            }
+        }
+        // 搜索路径
+        if (log.path && log.path.toLowerCase().includes(query)) {
+            return true;
+        }
+        // 搜索模型名称
+        if (log.requestModel && log.requestModel.toLowerCase().includes(query)) {
+            return true;
+        }
+        if (log.targetModel && log.targetModel.toLowerCase().includes(query)) {
+            return true;
+        }
+        return false;
+    }
     // Service blacklist operations
     isServiceBlacklisted(serviceId, routeId, contentType) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -1442,13 +1633,22 @@ class FileSystemDatabaseManager {
         if (!rule.routeId || typeof rule.routeId !== 'string') {
             return { valid: false, error: `规则[${index}](${rule.id}) 缺少有效的 routeId 字段` };
         }
-        if (!rule.targetServiceId || typeof rule.targetServiceId !== 'string') {
-            return { valid: false, error: `规则[${index}](${rule.id}) 缺少有效的 targetServiceId 字段` };
-        }
         const validContentTypes = ['default', 'background', 'thinking', 'long-context', 'image-understanding', 'model-mapping'];
         if (!rule.contentType || !validContentTypes.includes(rule.contentType)) {
             return { valid: false, error: `规则[${index}](${rule.id}) 的 contentType 无效` };
         }
+        // 如果使用MCP（仅对图像理解类型有效），则不需要验证targetServiceId
+        if (rule.useMCP === true && rule.contentType === 'image-understanding') {
+            if (!rule.mcpId || typeof rule.mcpId !== 'string') {
+                return { valid: false, error: `规则[${index}](${rule.id}) 使用MCP时缺少有效的 mcpId 字段` };
+            }
+        }
+        else {
+            // 不使用MCP时，必须验证targetServiceId
+            if (!rule.targetServiceId || typeof rule.targetServiceId !== 'string') {
+                return { valid: false, error: `规则[${index}](${rule.id}) 缺少有效的 targetServiceId 字段` };
+            }
+        }
         return { valid: true };
     }
     /**
@@ -2027,6 +2227,47 @@ class FileSystemDatabaseManager {
             return entry;
         });
     }
+    // MCP 工具相关操作
+    getMCPs() {
+        return this.mcps.sort((a, b) => b.createdAt - a.createdAt);
+    }
+    getMCP(id) {
+        return this.mcps.find(m => m.id === id);
+    }
+    getMCPsByTarget(targetType) {
+        return this.mcps.filter(m => { var _a; return (_a = m.targets) === null || _a === void 0 ? void 0 : _a.includes(targetType); });
+    }
+    createMCP(mcp) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const id = crypto_1.default.randomUUID();
+            const now = Date.now();
+            const newMCP = Object.assign(Object.assign({}, mcp), { id, createdAt: now, updatedAt: now });
+            this.mcps.push(newMCP);
+            yield this.saveMCPs();
+            return newMCP;
+        });
+    }
+    updateMCP(id, mcp) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const index = this.mcps.findIndex(m => m.id === id);
+            if (index === -1)
+                return false;
+            const now = Date.now();
+            this.mcps[index] = Object.assign(Object.assign(Object.assign({}, this.mcps[index]), mcp), { id, updatedAt: now });
+            yield this.saveMCPs();
+            return true;
+        });
+    }
+    deleteMCP(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const index = this.mcps.findIndex(m => m.id === id);
+            if (index === -1)
+                return false;
+            this.mcps.splice(index, 1);
+            yield this.saveMCPs();
+            return true;
+        });
+    }
     // Close method for compatibility (no-op for filesystem database)
     close() {
         // 文件系统数据库不需要关闭连接

package/dist/server/main.js

@@ -120,7 +120,8 @@ const writeClaudeConfig = (dbManager) => __awaiter(void 0, void 0, void 0, funct
                 ANTHROPIC_AUTH_TOKEN: config.apiKey || "api_key",
                 ANTHROPIC_API_KEY: "",
                 ANTHROPIC_BASE_URL: `http://${host}:${port}/claude-code`,
-                API_TIMEOUT_MS: "3000000"
+                API_TIMEOUT_MS: "3000000",
+                CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC: 1
             }
         };
         fs_1.default.writeFileSync(claudeSettingsPath, JSON.stringify(claudeSettings, null, 2));
@@ -777,6 +778,16 @@ const registerRoutes = (dbManager, proxyServer) => {
         const result = yield dbManager.activateRoute(req.params.id);
         if (result) {
             yield proxyServer.reloadRoutes();
+            // 激活路由后，同步MCP配置
+            const routes = dbManager.getRoutes();
+            const route = routes.find(r => r.id === req.params.id);
+            if (route) {
+                const mcps = dbManager.getMCPs();
+                const hasMCPForTarget = mcps.some(m => { var _a; return (_a = m.targets) === null || _a === void 0 ? void 0 : _a.includes(route.targetType); });
+                if (hasMCPForTarget) {
+                    yield writeMCPConfig(route.targetType);
+                }
+            }
         }
         res.json(result);
     })));
@@ -914,10 +925,38 @@ const registerRoutes = (dbManager, proxyServer) => {
         const count = yield dbManager.getLogsCount();
         res.json({ count });
     })));
+    app.get('/api/logs/search', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const query = typeof req.query.query === 'string' ? req.query.query : '';
+        const rawLimit = typeof req.query.limit === 'string' ? parseInt(req.query.limit, 10) : NaN;
+        const rawOffset = typeof req.query.offset === 'string' ? parseInt(req.query.offset, 10) : NaN;
+        const limit = Number.isFinite(rawLimit) ? rawLimit : 100;
+        const offset = Number.isFinite(rawOffset) ? rawOffset : 0;
+        const logs = yield dbManager.searchLogs(query, limit, offset);
+        res.json(logs);
+    })));
+    app.get('/api/logs/search/count', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const query = typeof req.query.query === 'string' ? req.query.query : '';
+        const count = yield dbManager.searchLogsCount(query);
+        res.json({ count });
+    })));
     app.get('/api/error-logs/count', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
         const count = yield dbManager.getErrorLogsCount();
         res.json({ count });
     })));
+    app.get('/api/error-logs/search', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const query = typeof req.query.query === 'string' ? req.query.query : '';
+        const rawLimit = typeof req.query.limit === 'string' ? parseInt(req.query.limit, 10) : NaN;
+        const rawOffset = typeof req.query.offset === 'string' ? parseInt(req.query.offset, 10) : NaN;
+        const limit = Number.isFinite(rawLimit) ? rawLimit : 100;
+        const offset = Number.isFinite(rawOffset) ? rawOffset : 0;
+        const logs = yield dbManager.searchErrorLogs(query, limit, offset);
+        res.json(logs);
+    })));
+    app.get('/api/error-logs/search/count', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const query = typeof req.query.query === 'string' ? req.query.query : '';
+        const count = yield dbManager.searchErrorLogsCount(query);
+        res.json({ count });
+    })));
     app.get('/api/config', (_req, res) => res.json(dbManager.getConfig()));
     app.put('/api/config', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
         const config = req.body;
@@ -1446,6 +1485,132 @@ ${instruction}
             res.status(500).json({ error: '获取工具状态失败' });
         }
     })));
+    // MCP 工具管理相关路由
+    app.get('/api/mcps', (_req, res) => {
+        res.json(dbManager.getMCPs());
+    });
+    app.get('/api/mcps/:id', (req, res) => {
+        const mcp = dbManager.getMCP(req.params.id);
+        if (!mcp) {
+            res.status(404).json({ error: 'MCP工具不存在' });
+            return;
+        }
+        res.json(mcp);
+    });
+    app.post('/api/mcps', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const mcpData = req.body;
+        const result = yield dbManager.createMCP(Object.assign(Object.assign({}, mcpData), { targets: mcpData.targets || [] }));
+        // 如果有激活的路由，立即写入MCP配置
+        if (mcpData.targets) {
+            const routes = dbManager.getRoutes();
+            for (const target of mcpData.targets) {
+                const activeRoute = routes.find(r => r.targetType === target && r.isActive);
+                if (activeRoute) {
+                    yield writeMCPConfig(target);
+                }
+            }
+        }
+        res.json(result);
+    })));
+    app.put('/api/mcps/:id', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const result = yield dbManager.updateMCP(req.params.id, req.body);
+        res.json(result);
+    })));
+    app.delete('/api/mcps/:id', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const mcp = dbManager.getMCP(req.params.id);
+        if (!mcp) {
+            res.status(404).json({ error: 'MCP工具不存在' });
+            return;
+        }
+        const result = yield dbManager.deleteMCP(req.params.id);
+        // 从Claude Code和Codex配置中移除该MCP
+        if (mcp.targets) {
+            for (const target of mcp.targets) {
+                yield removeMCPFromConfig(target, mcp.id);
+            }
+        }
+        res.json(result);
+    })));
+    // 写入MCP配置到Claude Code或Codex的全局配置文件
+    const writeMCPConfig = (targetType) => __awaiter(void 0, void 0, void 0, function* () {
+        try {
+            const homeDir = os_1.default.homedir();
+            const mcps = dbManager.getMCPsByTarget(targetType);
+            if (targetType === 'claude-code') {
+                // Claude Code配置文件路径
+                const claudeJsonPath = path_1.default.join(homeDir, '.claude.json');
+                let claudeJson = {};
+                // 读取现有配置
+                if (fs_1.default.existsSync(claudeJsonPath)) {
+                    claudeJson = JSON.parse(fs_1.default.readFileSync(claudeJsonPath, 'utf8'));
+                }
+                // 确保mcpServers存在
+                if (!claudeJson.mcpServers) {
+                    claudeJson.mcpServers = {};
+                }
+                // 写入所有启用的MCP
+                for (const mcp of mcps) {
+                    const mcpConfig = {
+                        type: mcp.type,
+                    };
+                    if (mcp.type === 'stdio') {
+                        mcpConfig.command = mcp.command;
+                        mcpConfig.args = mcp.args;
+                    }
+                    else if (mcp.type === 'http') {
+                        mcpConfig.url = mcp.url;
+                    }
+                    else if (mcp.type === 'sse') {
+                        mcpConfig.url = mcp.url;
+                    }
+                    if (mcp.headers && Object.keys(mcp.headers).length > 0) {
+                        mcpConfig.headers = mcp.headers;
+                    }
+                    if (mcp.env && Object.keys(mcp.env).length > 0) {
+                        mcpConfig.env = mcp.env;
+                    }
+                    claudeJson.mcpServers[mcp.id] = mcpConfig;
+                }
+                fs_1.default.writeFileSync(claudeJsonPath, JSON.stringify(claudeJson, null, 2));
+                return true;
+            }
+            else if (targetType === 'codex') {
+                // Codex使用TOML格式，我们暂时不直接写入
+                // 需要后续处理Codex的MCP配置格式
+                // TODO: 实现 Codex MCP 配置写入
+                console.log('[MCP] Codex MCP配置写入暂未实现');
+                return true;
+            }
+            return false;
+        }
+        catch (error) {
+            console.error('Failed to write MCP config:', error);
+            return false;
+        }
+    });
+    // 从配置中移除MCP
+    const removeMCPFromConfig = (targetType, mcpId) => __awaiter(void 0, void 0, void 0, function* () {
+        try {
+            if (targetType === 'claude-code') {
+                const homeDir = os_1.default.homedir();
+                const claudeJsonPath = path_1.default.join(homeDir, '.claude.json');
+                if (!fs_1.default.existsSync(claudeJsonPath)) {
+                    return true;
+                }
+                const claudeJson = JSON.parse(fs_1.default.readFileSync(claudeJsonPath, 'utf8'));
+                if (claudeJson.mcpServers && claudeJson.mcpServers[mcpId]) {
+                    delete claudeJson.mcpServers[mcpId];
+                    fs_1.default.writeFileSync(claudeJsonPath, JSON.stringify(claudeJson, null, 2));
+                }
+                return true;
+            }
+            return false;
+        }
+        catch (error) {
+            console.error('Failed to remove MCP from config:', error);
+            return false;
+        }
+    });
 };
 const start = () => __awaiter(void 0, void 0, void 0, function* () {
     fs_1.default.mkdirSync(dataDir, { recursive: true });