aicodeswitch 1.10.1 → 2.0.1

package/dist/server/main.js

@@ -17,11 +17,13 @@ const cors_1 = __importDefault(require("cors"));
 const dotenv_1 = __importDefault(require("dotenv"));
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
+const crypto_1 = require("crypto");
 const database_1 = require("./database");
 const proxy_server_1 = require("./proxy-server");
 const os_1 = __importDefault(require("os"));
 const auth_1 = require("./auth");
 const version_check_1 = require("./version-check");
+const utils_1 = require("./utils");
 const dotenvPath = path_1.default.resolve(os_1.default.homedir(), '.aicodeswitch/aicodeswitch.conf');
 if (fs_1.default.existsSync(dotenvPath)) {
     dotenv_1.default.config({ path: dotenvPath });
@@ -283,6 +285,54 @@ const registerRoutes = (dbManager, proxyServer) => {
     app.put('/api/rules/:id', (req, res) => res.json(dbManager.updateRule(req.params.id, req.body)));
     app.delete('/api/rules/:id', (req, res) => res.json(dbManager.deleteRule(req.params.id)));
     app.put('/api/rules/:id/reset-tokens', (req, res) => res.json(dbManager.resetRuleTokenUsage(req.params.id)));
+    app.put('/api/rules/:id/reset-requests', (req, res) => res.json(dbManager.resetRuleRequestCount(req.params.id)));
+    // 解除规则的黑名单状态
+    app.put('/api/rules/:id/clear-blacklist', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const { id } = req.params;
+        const rule = dbManager.getRule(id);
+        if (!rule) {
+            res.status(404).json({ error: 'Rule not found' });
+            return;
+        }
+        // 找到该规则所属的路由
+        const routes = dbManager.getRoutes();
+        const route = routes.find(r => {
+            const rules = dbManager.getRules(r.id);
+            return rules.some(r => r.id === id);
+        });
+        if (!route) {
+            res.status(404).json({ error: 'Route not found' });
+            return;
+        }
+        try {
+            yield dbManager.removeFromBlacklist(rule.targetServiceId, route.id, rule.contentType);
+            res.json({ success: true });
+        }
+        catch (error) {
+            console.error('Error clearing blacklist:', error);
+            res.status(500).json({ error: 'Failed to clear blacklist' });
+        }
+    })));
+    // 获取规则的黑名单状态
+    app.get('/api/rules/:routeId/blacklist-status', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const { routeId } = req.params;
+        const rules = dbManager.getRules(routeId);
+        try {
+            const results = yield Promise.all(rules.map((rule) => __awaiter(void 0, void 0, void 0, function* () {
+                const blacklistStatus = yield dbManager.getRuleBlacklistStatus(rule.targetServiceId, routeId, rule.contentType);
+                return {
+                    ruleId: rule.id,
+                    isBlacklisted: blacklistStatus !== null,
+                    blacklistEntry: blacklistStatus,
+                };
+            })));
+            res.json(results);
+        }
+        catch (error) {
+            console.error('Error getting blacklist status:', error);
+            res.status(500).json({ error: 'Failed to get blacklist status' });
+        }
+    })));
     app.get('/api/logs', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
         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;
@@ -295,18 +345,6 @@ const registerRoutes = (dbManager, proxyServer) => {
         yield dbManager.clearLogs();
         res.json(true);
     })));
-    app.get('/api/access-logs', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
-        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.getAccessLogs(limit, offset);
-        res.json(logs);
-    })));
-    app.delete('/api/access-logs', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
-        yield dbManager.clearAccessLogs();
-        res.json(true);
-    })));
     app.get('/api/error-logs', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
         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;
@@ -323,10 +361,6 @@ const registerRoutes = (dbManager, proxyServer) => {
         const count = yield dbManager.getLogsCount();
         res.json({ count });
     })));
-    app.get('/api/access-logs/count', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
-        const count = yield dbManager.getAccessLogsCount();
-        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 });
@@ -384,6 +418,41 @@ const registerRoutes = (dbManager, proxyServer) => {
         const stats = yield dbManager.getStatistics(days);
         res.json(stats);
     })));
+    // Sessions 相关端点
+    app.get('/api/sessions', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        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 sessions = dbManager.getSessions(limit, offset);
+        res.json(sessions);
+    })));
+    app.get('/api/sessions/count', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const count = dbManager.getSessionsCount();
+        res.json({ count });
+    })));
+    app.get('/api/sessions/:id', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const session = dbManager.getSession(req.params.id);
+        if (!session) {
+            res.status(404).json({ error: 'Session not found' });
+            return;
+        }
+        res.json(session);
+    })));
+    app.get('/api/sessions/:id/logs', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const rawLimit = typeof req.query.limit === 'string' ? parseInt(req.query.limit, 10) : NaN;
+        const limit = Number.isFinite(rawLimit) ? rawLimit : 100;
+        const logs = yield dbManager.getLogsBySessionId(req.params.id, limit);
+        res.json(logs);
+    })));
+    app.delete('/api/sessions/:id', asyncHandler((req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        const result = dbManager.deleteSession(req.params.id);
+        res.json(result);
+    })));
+    app.delete('/api/sessions', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        dbManager.clearSessions();
+        res.json(true);
+    })));
     app.get('/api/docs/recommend-vendors', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
         const resp = yield fetch('https://unpkg.com/aicodeswitch/docs/vendors-recommand.md');
         if (!resp.ok) {
@@ -402,24 +471,97 @@ const registerRoutes = (dbManager, proxyServer) => {
         const text = yield resp.text();
         res.type('text/plain').send(text);
     })));
-    app.use(express_1.default.static(path_1.default.resolve(__dirname, '../ui')));
+    // Migration 相关端点
+    const getMigrationHashPath = () => path_1.default.join(dataDir, '.migration-hash');
+    // 查找 migration.md 文件的路径
+    const findMigrationPath = () => {
+        // 可能的路径列表
+        const possiblePaths = [
+            // 开发环境：src/server/main.ts -> public/migration.md
+            path_1.default.resolve(__dirname, '../../public/migration.md'),
+            // 生产环境：dist/server/main.js -> dist/ui/migration.md
+            path_1.default.resolve(__dirname, '../ui/migration.md'),
+        ];
+        for (const possiblePath of possiblePaths) {
+            if (fs_1.default.existsSync(possiblePath)) {
+                return possiblePath;
+            }
+        }
+        return null;
+    };
+    app.get('/api/migration', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        try {
+            // 读取 migration.md 文件
+            const migrationPath = findMigrationPath();
+            if (!migrationPath) {
+                res.json({ shouldShow: false, content: '' });
+                return;
+            }
+            const content = fs_1.default.readFileSync(migrationPath, 'utf-8');
+            // 计算当前内容的 hash
+            const currentHash = (0, crypto_1.createHash)('sha256').update(content).digest('hex');
+            // 读取之前保存的 hash
+            const hashPath = getMigrationHashPath();
+            let savedHash = '';
+            if (fs_1.default.existsSync(hashPath)) {
+                savedHash = fs_1.default.readFileSync(hashPath, 'utf-8').trim();
+            }
+            // 如果 hash 不同，需要显示弹窗
+            const shouldShow = savedHash !== currentHash;
+            res.json({ shouldShow, content: shouldShow ? content : '' });
+        }
+        catch (error) {
+            console.error('Failed to read migration file:', error);
+            res.json({ shouldShow: false, content: '' });
+        }
+    })));
+    app.post('/api/migration/ack', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        try {
+            // 读取 migration.md 文件并计算 hash
+            const migrationPath = findMigrationPath();
+            if (!migrationPath) {
+                res.json({ success: false });
+                return;
+            }
+            const content = fs_1.default.readFileSync(migrationPath, 'utf-8');
+            const hash = (0, crypto_1.createHash)('sha256').update(content).digest('hex');
+            // 保存 hash 到文件
+            const hashPath = getMigrationHashPath();
+            fs_1.default.writeFileSync(hashPath, hash, 'utf-8');
+            res.json({ success: true });
+        }
+        catch (error) {
+            console.error('Failed to acknowledge migration:', error);
+            res.json({ success: false });
+        }
+    })));
 };
 const start = () => __awaiter(void 0, void 0, void 0, function* () {
     fs_1.default.mkdirSync(dataDir, { recursive: true });
     const dbManager = new database_1.DatabaseManager(dataDir);
-    yield dbManager.initialize();
     const proxyServer = new proxy_server_1.ProxyServer(dbManager, app);
+    yield dbManager.initialize();
+    // Initialize proxy server and register proxy routes last
+    proxyServer.initialize();
     // Register admin routes first
     registerRoutes(dbManager, proxyServer);
-    // Initialize proxy server and register proxy routes last
-    yield proxyServer.initialize();
-    const adminServer = app.listen(port, host, () => {
+    yield proxyServer.registerProxyRoutes();
+    app.use(express_1.default.static(path_1.default.resolve(__dirname, '../ui')));
+    const isPortUsable = yield (0, utils_1.checkPortUsable)(port);
+    if (!isPortUsable) {
+        console.error(`端口 ${port} 已被占用，无法启动服务。请执行 aicos stop 后重启。`);
+        process.exit(1);
+    }
+    const server = app.listen(port, host, () => {
         console.log(`Admin server running on http://${host}:${port}`);
     });
     const shutdown = () => __awaiter(void 0, void 0, void 0, function* () {
         console.log('Shutting down server...');
         dbManager.close();
-        adminServer.close(() => process.exit(0));
+        server.close(() => {
+            console.log('Server stopped.');
+            process.exit(0);
+        });
     });
     process.on('SIGINT', shutdown);
     process.on('SIGTERM', shutdown);