aicodeswitch 3.0.7 → 3.0.9

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() {
         // 文件系统数据库不需要关闭连接