aicodeswitch 1.10.1 → 2.0.1

package/dist/server/database.js

@@ -39,12 +39,6 @@ class DatabaseManager {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "accessLogDb", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "errorLogDb", {
             enumerable: true,
             configurable: true,
@@ -64,12 +58,6 @@ class DatabaseManager {
             writable: true,
             value: null
         });
-        Object.defineProperty(this, "accessLogsCountCache", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
         Object.defineProperty(this, "errorLogsCountCache", {
             enumerable: true,
             configurable: true,
@@ -94,7 +82,6 @@ class DatabaseManager {
         // 设置 read_uncommitted = 0 确保读取最新提交的数据
         this.db.pragma('read_uncommitted = 0');
         this.logDb = new level_1.Level(path_1.default.join(dataPath, 'logs'), { valueEncoding: 'json' });
-        this.accessLogDb = new level_1.Level(path_1.default.join(dataPath, 'access-logs'), { valueEncoding: 'json' });
         this.errorLogDb = new level_1.Level(path_1.default.join(dataPath, 'error-logs'), { valueEncoding: 'json' });
         this.blacklistDb = new level_1.Level(path_1.default.join(dataPath, 'service-blacklist'), { valueEncoding: 'json' });
     }
@@ -180,6 +167,103 @@ class DatabaseManager {
                 this.db.exec('ALTER TABLE api_services ADD COLUMN enable_proxy INTEGER DEFAULT 0;');
                 console.log('[DB] Migration completed: enable_proxy column added');
             }
+            // 检查rules表是否有请求次数相关字段
+            const hasRequestCountLimit = rulesColumns.some((col) => col.name === 'request_count_limit');
+            const hasTotalRequestsUsed = rulesColumns.some((col) => col.name === 'total_requests_used');
+            const hasRequestResetInterval = rulesColumns.some((col) => col.name === 'request_reset_interval');
+            const hasRequestLastResetAt = rulesColumns.some((col) => col.name === 'request_last_reset_at');
+            if (!hasRequestCountLimit) {
+                console.log('[DB] Running migration: Adding request_count_limit column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN request_count_limit INTEGER;');
+                console.log('[DB] Migration completed: request_count_limit column added');
+            }
+            if (!hasTotalRequestsUsed) {
+                console.log('[DB] Running migration: Adding total_requests_used column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN total_requests_used INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: total_requests_used column added');
+            }
+            if (!hasRequestResetInterval) {
+                console.log('[DB] Running migration: Adding request_reset_interval column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN request_reset_interval INTEGER;');
+                console.log('[DB] Migration completed: request_reset_interval column added');
+            }
+            if (!hasRequestLastResetAt) {
+                console.log('[DB] Running migration: Adding request_last_reset_at column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN request_last_reset_at INTEGER;');
+                console.log('[DB] Migration completed: request_last_reset_at column added');
+            }
+            // 检查rules表是否有request_reset_base_time字段
+            const hasRequestResetBaseTime = rulesColumns.some((col) => col.name === 'request_reset_base_time');
+            if (!hasRequestResetBaseTime) {
+                console.log('[DB] Running migration: Adding request_reset_base_time column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN request_reset_base_time INTEGER;');
+                console.log('[DB] Migration completed: request_reset_base_time column added');
+            }
+            // 检查rules表是否有token_reset_base_time字段
+            const hasTokenResetBaseTime = rulesColumns.some((col) => col.name === 'token_reset_base_time');
+            if (!hasTokenResetBaseTime) {
+                console.log('[DB] Running migration: Adding token_reset_base_time column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN token_reset_base_time INTEGER;');
+                console.log('[DB] Migration completed: token_reset_base_time column added');
+            }
+            // 检查api_services表是否有超量配置相关字段
+            // Token超量配置
+            const hasEnableTokenLimit = columns.some((col) => col.name === 'enable_token_limit');
+            if (!hasEnableTokenLimit) {
+                console.log('[DB] Running migration: Adding enable_token_limit column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN enable_token_limit INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: enable_token_limit column added');
+            }
+            const hasServiceTokenLimit = columns.some((col) => col.name === 'token_limit');
+            if (!hasServiceTokenLimit) {
+                console.log('[DB] Running migration: Adding token_limit column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN token_limit INTEGER;');
+                console.log('[DB] Migration completed: token_limit column added');
+            }
+            const hasTokenResetInterval = columns.some((col) => col.name === 'token_reset_interval');
+            if (!hasTokenResetInterval) {
+                console.log('[DB] Running migration: Adding token_reset_interval column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN token_reset_interval INTEGER;');
+                console.log('[DB] Migration completed: token_reset_interval column added');
+            }
+            const hasServiceTokenResetBaseTime = columns.some((col) => col.name === 'token_reset_base_time');
+            if (!hasServiceTokenResetBaseTime) {
+                console.log('[DB] Running migration: Adding token_reset_base_time column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN token_reset_base_time INTEGER;');
+                console.log('[DB] Migration completed: token_reset_base_time column added');
+            }
+            // 请求次数超量配置
+            const hasEnableRequestLimit = columns.some((col) => col.name === 'enable_request_limit');
+            if (!hasEnableRequestLimit) {
+                console.log('[DB] Running migration: Adding enable_request_limit column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN enable_request_limit INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: enable_request_limit column added');
+            }
+            const hasServiceRequestCountLimit = columns.some((col) => col.name === 'request_count_limit');
+            if (!hasServiceRequestCountLimit) {
+                console.log('[DB] Running migration: Adding request_count_limit column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN request_count_limit INTEGER;');
+                console.log('[DB] Migration completed: request_count_limit column added');
+            }
+            const hasServiceRequestResetInterval = columns.some((col) => col.name === 'request_reset_interval');
+            if (!hasServiceRequestResetInterval) {
+                console.log('[DB] Running migration: Adding request_reset_interval column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN request_reset_interval INTEGER;');
+                console.log('[DB] Migration completed: request_reset_interval column added');
+            }
+            const hasServiceRequestResetBaseTime = columns.some((col) => col.name === 'request_reset_base_time');
+            if (!hasServiceRequestResetBaseTime) {
+                console.log('[DB] Running migration: Adding request_reset_base_time column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN request_reset_base_time INTEGER;');
+                console.log('[DB] Migration completed: request_reset_base_time column added');
+            }
+            // 检查api_services表是否有auth_type字段
+            const hasAuthType = columns.some((col) => col.name === 'auth_type');
+            if (!hasAuthType) {
+                console.log('[DB] Running migration: Adding auth_type column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN auth_type TEXT DEFAULT NULL;');
+                console.log('[DB] Migration completed: auth_type column added');
+            }
         });
     }
     migrateMaxOutputTokensToModelLimits() {
@@ -294,6 +378,10 @@ class DatabaseManager {
         total_tokens_used INTEGER DEFAULT 0,
         reset_interval INTEGER,
         last_reset_at INTEGER,
+        request_count_limit INTEGER,
+        total_requests_used INTEGER DEFAULT 0,
+        request_reset_interval INTEGER,
+        request_last_reset_at INTEGER,
         created_at INTEGER NOT NULL,
         updated_at INTEGER NOT NULL,
         FOREIGN KEY (route_id) REFERENCES routes(id) ON DELETE CASCADE,
@@ -304,6 +392,21 @@ class DatabaseManager {
         key TEXT PRIMARY KEY,
         value TEXT NOT NULL
       );
+
+      CREATE TABLE IF NOT EXISTS sessions (
+        id TEXT PRIMARY KEY,
+        target_type TEXT NOT NULL CHECK(target_type IN ('claude-code', 'codex')),
+        title TEXT,
+        first_request_at INTEGER NOT NULL,
+        last_request_at INTEGER NOT NULL,
+        request_count INTEGER DEFAULT 1,
+        total_tokens INTEGER DEFAULT 0,
+        vendor_id TEXT,
+        vendor_name TEXT,
+        service_id TEXT,
+        service_name TEXT,
+        model TEXT
+      );
     `);
     }
     ensureDefaultConfig() {
@@ -372,9 +475,20 @@ class DatabaseManager {
             apiUrl: row.api_url,
             apiKey: row.api_key,
             sourceType: row.source_type,
+            authType: row.auth_type,
             supportedModels: row.supported_models ? row.supported_models.split(',').map((model) => model.trim()).filter((model) => model.length > 0) : undefined,
             modelLimits: row.model_limits ? JSON.parse(row.model_limits) : undefined,
             enableProxy: row.enable_proxy === 1,
+            // Token超量配置
+            enableTokenLimit: row.enable_token_limit === 1,
+            tokenLimit: row.token_limit,
+            tokenResetInterval: row.token_reset_interval,
+            tokenResetBaseTime: row.token_reset_base_time,
+            // 请求次数超量配置
+            enableRequestLimit: row.enable_request_limit === 1,
+            requestCountLimit: row.request_count_limit,
+            requestResetInterval: row.request_reset_interval,
+            requestResetBaseTime: row.request_reset_base_time,
             createdAt: row.created_at,
             updatedAt: row.updated_at,
         }));
@@ -388,15 +502,15 @@ class DatabaseManager {
         const id = crypto_1.default.randomUUID();
         const now = Date.now();
         this.db
-            .prepare('INSERT INTO api_services (id, vendor_id, name, api_url, api_key, source_type, supported_models, model_limits, enable_proxy, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-            .run(id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.enableProxy ? 1 : 0, now, now);
+            .prepare('INSERT INTO api_services (id, vendor_id, name, api_url, api_key, source_type, auth_type, supported_models, model_limits, enable_proxy, enable_token_limit, token_limit, token_reset_interval, token_reset_base_time, enable_request_limit, request_count_limit, request_reset_interval, request_reset_base_time, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
+            .run(id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.authType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.enableProxy ? 1 : 0, service.enableTokenLimit ? 1 : 0, service.tokenLimit || null, service.tokenResetInterval || null, service.tokenResetBaseTime || null, service.enableRequestLimit ? 1 : 0, service.requestCountLimit || null, service.requestResetInterval || null, service.requestResetBaseTime || null, now, now);
         return Object.assign(Object.assign({}, service), { id, createdAt: now, updatedAt: now });
     }
     updateAPIService(id, service) {
         const now = Date.now();
         const result = this.db
-            .prepare('UPDATE api_services SET name = ?, api_url = ?, api_key = ?, source_type = ?, supported_models = ?, model_limits = ?, enable_proxy = ?, updated_at = ? WHERE id = ?')
-            .run(service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.enableProxy !== undefined ? (service.enableProxy ? 1 : 0) : null, now, id);
+            .prepare('UPDATE api_services SET vendor_id = ?, name = ?, api_url = ?, api_key = ?, source_type = ?, auth_type = ?, supported_models = ?, model_limits = ?, enable_proxy = ?, enable_token_limit = ?, token_limit = ?, token_reset_interval = ?, token_reset_base_time = ?, enable_request_limit = ?, request_count_limit = ?, request_reset_interval = ?, request_reset_base_time = ?, updated_at = ? WHERE id = ?')
+            .run(service.vendorId, service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.authType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.enableProxy !== undefined ? (service.enableProxy ? 1 : 0) : null, service.enableTokenLimit !== undefined ? (service.enableTokenLimit ? 1 : 0) : null, service.tokenLimit !== undefined ? service.tokenLimit : null, service.tokenResetInterval !== undefined ? service.tokenResetInterval : null, service.tokenResetBaseTime !== undefined ? service.tokenResetBaseTime : null, service.enableRequestLimit !== undefined ? (service.enableRequestLimit ? 1 : 0) : null, service.requestCountLimit !== undefined ? service.requestCountLimit : null, service.requestResetInterval !== undefined ? service.requestResetInterval : null, service.requestResetBaseTime !== undefined ? service.requestResetBaseTime : null, now, id);
         // 调试日志: 记录更新操作
         if (result.changes > 0 && process.env.NODE_ENV === 'development') {
             console.log(`[DB] Updated service ${id}: ${service.name} -> ${service.apiUrl}`);
@@ -472,23 +586,56 @@ class DatabaseManager {
             totalTokensUsed: row.total_tokens_used,
             resetInterval: row.reset_interval,
             lastResetAt: row.last_reset_at,
+            tokenResetBaseTime: row.token_reset_base_time,
+            requestCountLimit: row.request_count_limit,
+            totalRequestsUsed: row.total_requests_used,
+            requestResetInterval: row.request_reset_interval,
+            requestLastResetAt: row.request_last_reset_at,
+            requestResetBaseTime: row.request_reset_base_time,
             createdAt: row.created_at,
             updatedAt: row.updated_at,
         }));
     }
+    getRule(id) {
+        const row = this.db.prepare('SELECT * FROM rules WHERE id = ?').get(id);
+        if (!row)
+            return undefined;
+        return {
+            id: row.id,
+            routeId: row.route_id,
+            contentType: row.content_type,
+            targetServiceId: row.target_service_id,
+            targetModel: row.target_model,
+            replacedModel: row.replaced_model,
+            sortOrder: row.sort_order,
+            timeout: row.timeout,
+            tokenLimit: row.token_limit,
+            totalTokensUsed: row.total_tokens_used,
+            resetInterval: row.reset_interval,
+            lastResetAt: row.last_reset_at,
+            tokenResetBaseTime: row.token_reset_base_time,
+            requestCountLimit: row.request_count_limit,
+            totalRequestsUsed: row.total_requests_used,
+            requestResetInterval: row.request_reset_interval,
+            requestLastResetAt: row.request_last_reset_at,
+            requestResetBaseTime: row.request_reset_base_time,
+            createdAt: row.created_at,
+            updatedAt: row.updated_at,
+        };
+    }
     createRule(route) {
         const id = crypto_1.default.randomUUID();
         const now = Date.now();
         this.db
-            .prepare('INSERT INTO rules (id, route_id, content_type, target_service_id, target_model, replaced_model, sort_order, timeout, token_limit, total_tokens_used, reset_interval, last_reset_at, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-            .run(id, route.routeId, route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, route.timeout || null, route.tokenLimit || null, route.totalTokensUsed || 0, route.resetInterval || null, route.lastResetAt || null, now, now);
+            .prepare('INSERT INTO rules (id, route_id, content_type, target_service_id, target_model, replaced_model, sort_order, timeout, token_limit, total_tokens_used, reset_interval, last_reset_at, token_reset_base_time, request_count_limit, total_requests_used, request_reset_interval, request_last_reset_at, request_reset_base_time, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
+            .run(id, route.routeId, route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, route.timeout || null, route.tokenLimit || null, route.totalTokensUsed || 0, route.resetInterval || null, route.lastResetAt || null, route.tokenResetBaseTime || null, route.requestCountLimit || null, route.totalRequestsUsed || 0, route.requestResetInterval || null, route.requestLastResetAt || null, route.requestResetBaseTime || null, now, now);
         return Object.assign(Object.assign({}, route), { id, createdAt: now, updatedAt: now });
     }
     updateRule(id, route) {
         const now = Date.now();
         const result = this.db
-            .prepare('UPDATE rules SET content_type = ?, target_service_id = ?, target_model = ?, replaced_model = ?, sort_order = ?, timeout = ?, token_limit = ?, reset_interval = ?, updated_at = ? WHERE id = ?')
-            .run(route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, route.timeout !== undefined ? route.timeout : null, route.tokenLimit !== undefined ? route.tokenLimit : null, route.resetInterval !== undefined ? route.resetInterval : null, now, id);
+            .prepare('UPDATE rules SET content_type = ?, target_service_id = ?, target_model = ?, replaced_model = ?, sort_order = ?, timeout = ?, token_limit = ?, reset_interval = ?, token_reset_base_time = ?, request_count_limit = ?, request_reset_interval = ?, request_reset_base_time = ?, updated_at = ? WHERE id = ?')
+            .run(route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, route.timeout !== undefined ? route.timeout : null, route.tokenLimit !== undefined ? route.tokenLimit : null, route.resetInterval !== undefined ? route.resetInterval : null, route.tokenResetBaseTime !== undefined ? route.tokenResetBaseTime : null, route.requestCountLimit !== undefined ? route.requestCountLimit : null, route.requestResetInterval !== undefined ? route.requestResetInterval : null, route.requestResetBaseTime !== undefined ? route.requestResetBaseTime : null, now, id);
         return result.changes > 0;
     }
     deleteRule(id) {
@@ -527,28 +674,133 @@ class DatabaseManager {
      */
     checkAndResetRuleIfNeeded(ruleId) {
         const rule = this.db
-            .prepare('SELECT reset_interval, last_reset_at FROM rules WHERE id = ?')
+            .prepare('SELECT reset_interval, last_reset_at, token_reset_base_time FROM rules WHERE id = ?')
             .get(ruleId);
         if (!rule || !rule.reset_interval) {
             return false; // 没有设置重置间隔
         }
         const now = Date.now();
         const resetIntervalMs = rule.reset_interval * 60 * 60 * 1000; // 小时转毫秒
+        const baseTime = rule.token_reset_base_time;
         const lastResetAt = rule.last_reset_at || 0;
-        // 检查是否已经到了重置时间
+        // 场景1: 设置了时间基点
+        if (baseTime) {
+            if (now >= baseTime) {
+                this.resetRuleTokenUsageWithBaseTime(ruleId, baseTime);
+                return true;
+            }
+            return false;
+        }
+        // 场景2: 未设置时间基点，使用原始逻辑（向后兼容）
         if (now - lastResetAt >= resetIntervalMs) {
             this.resetRuleTokenUsage(ruleId);
             return true;
         }
         return false;
     }
+    /**
+     * 重置规则的Token使用量（带时间基点更新）
+     */
+    resetRuleTokenUsageWithBaseTime(ruleId, currentBaseTime) {
+        const now = Date.now();
+        const rule = this.db
+            .prepare('SELECT reset_interval FROM rules WHERE id = ?')
+            .get(ruleId);
+        if (!rule || !rule.reset_interval) {
+            return false;
+        }
+        const resetIntervalMs = rule.reset_interval * 60 * 60 * 1000;
+        // 计算下一个时间基点
+        let nextBaseTime = currentBaseTime;
+        while (nextBaseTime <= now) {
+            nextBaseTime += resetIntervalMs;
+        }
+        const result = this.db
+            .prepare('UPDATE rules SET total_tokens_used = 0, last_reset_at = ?, token_reset_base_time = ? WHERE id = ?')
+            .run(now, nextBaseTime, ruleId);
+        return result.changes > 0;
+    }
+    /**
+     * 增加规则的请求次数
+     * @param ruleId 规则ID
+     * @param count 增加的次数
+     * @returns 是否成功
+     */
+    incrementRuleRequestCount(ruleId, count) {
+        const result = this.db
+            .prepare('UPDATE rules SET total_requests_used = total_requests_used + ? WHERE id = ?')
+            .run(count, ruleId);
+        return result.changes > 0;
+    }
+    /**
+     * 重置规则的请求次数
+     * @param ruleId 规则ID
+     * @returns 是否成功
+     */
+    resetRuleRequestCount(ruleId) {
+        const now = Date.now();
+        const result = this.db
+            .prepare('UPDATE rules SET total_requests_used = 0, request_last_reset_at = ? WHERE id = ?')
+            .run(now, ruleId);
+        return result.changes > 0;
+    }
+    /**
+     * 检查并重置到期的规则（请求次数）
+     * 如果规则设置了request_reset_interval且已经到了重置时间，则自动重置请求次数
+     * @param ruleId 规则ID
+     * @returns 是否进行了重置
+     */
+    checkAndResetRequestCountIfNeeded(ruleId) {
+        const rule = this.db
+            .prepare('SELECT request_reset_interval, request_last_reset_at, request_reset_base_time FROM rules WHERE id = ?')
+            .get(ruleId);
+        if (!rule || !rule.request_reset_interval) {
+            return false; // 没有设置重置间隔
+        }
+        const now = Date.now();
+        const resetIntervalMs = rule.request_reset_interval * 60 * 60 * 1000; // 小时转毫秒
+        const baseTime = rule.request_reset_base_time;
+        const lastResetAt = rule.request_last_reset_at || 0;
+        // 场景1: 设置了时间基点
+        if (baseTime) {
+            if (now >= baseTime) {
+                this.resetRuleRequestCountWithBaseTime(ruleId, baseTime);
+                return true;
+            }
+            return false;
+        }
+        // 场景2: 未设置时间基点，使用原始逻辑（向后兼容）
+        if (now - lastResetAt >= resetIntervalMs) {
+            this.resetRuleRequestCount(ruleId);
+            return true;
+        }
+        return false;
+    }
+    /**
+     * 重置规则的请求次数（带时间基点更新）
+     */
+    resetRuleRequestCountWithBaseTime(ruleId, currentBaseTime) {
+        const now = Date.now();
+        const rule = this.db
+            .prepare('SELECT request_reset_interval FROM rules WHERE id = ?')
+            .get(ruleId);
+        if (!rule || !rule.request_reset_interval) {
+            return false;
+        }
+        const resetIntervalMs = rule.request_reset_interval * 60 * 60 * 1000;
+        // 计算下一个时间基点
+        let nextBaseTime = currentBaseTime;
+        while (nextBaseTime <= now) {
+            nextBaseTime += resetIntervalMs;
+        }
+        const result = this.db
+            .prepare('UPDATE rules SET total_requests_used = 0, request_last_reset_at = ?, request_reset_base_time = ? WHERE id = ?')
+            .run(now, nextBaseTime, ruleId);
+        return result.changes > 0;
+    }
     // Log operations
     addLog(log) {
         return __awaiter(this, void 0, void 0, function* () {
-            const { path } = log;
-            if (!path.startsWith('/v1/')) {
-                return;
-            }
             const id = crypto_1.default.randomUUID();
             yield this.logDb.put(id, JSON.stringify(Object.assign(Object.assign({}, log), { id })));
             // 清除缓存
@@ -587,55 +839,6 @@ class DatabaseManager {
             this.logsCountCache = null;
         });
     }
-    // Access log operations
-    addAccessLog(log) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const id = crypto_1.default.randomUUID();
-            yield this.accessLogDb.put(id, JSON.stringify(Object.assign(Object.assign({}, log), { id })));
-            // 清除缓存
-            this.accessLogsCountCache = null;
-            return id;
-        });
-    }
-    updateAccessLog(id, data) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const log = yield this.accessLogDb.get(id);
-            const updatedLog = Object.assign(Object.assign({}, JSON.parse(log)), data);
-            yield this.accessLogDb.put(id, JSON.stringify(updatedLog));
-        });
-    }
-    getAccessLogs() {
-        return __awaiter(this, arguments, void 0, function* (limit = 100, offset = 0) {
-            var _a, e_2, _b, _c;
-            const allLogs = [];
-            try {
-                for (var _d = true, _e = __asyncValues(this.accessLogDb.iterator()), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
-                    _c = _f.value;
-                    _d = false;
-                    const [, value] = _c;
-                    allLogs.push(JSON.parse(value));
-                }
-            }
-            catch (e_2_1) { e_2 = { error: e_2_1 }; }
-            finally {
-                try {
-                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
-                }
-                finally { if (e_2) throw e_2.error; }
-            }
-            // Sort by timestamp in descending order (newest first)
-            allLogs.sort((a, b) => b.timestamp - a.timestamp);
-            // Apply offset and limit
-            return allLogs.slice(offset, offset + limit);
-        });
-    }
-    clearAccessLogs() {
-        return __awaiter(this, void 0, void 0, function* () {
-            yield this.accessLogDb.clear();
-            // 清除缓存
-            this.accessLogsCountCache = null;
-        });
-    }
     // Error log operations
     addErrorLog(log) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -647,7 +850,7 @@ class DatabaseManager {
     }
     getErrorLogs() {
         return __awaiter(this, arguments, void 0, function* (limit = 100, offset = 0) {
-            var _a, e_3, _b, _c;
+            var _a, e_2, _b, _c;
             const allLogs = [];
             try {
                 for (var _d = true, _e = __asyncValues(this.errorLogDb.iterator()), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
@@ -657,12 +860,12 @@ class DatabaseManager {
                     allLogs.push(JSON.parse(value));
                 }
             }
-            catch (e_3_1) { e_3 = { error: e_3_1 }; }
+            catch (e_2_1) { e_2 = { error: e_2_1 }; }
             finally {
                 try {
                     if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
                 }
-                finally { if (e_3) throw e_3.error; }
+                finally { if (e_2) throw e_2.error; }
             }
             // Sort by timestamp in descending order (newest first)
             allLogs.sort((a, b) => b.timestamp - a.timestamp);
@@ -682,7 +885,7 @@ class DatabaseManager {
      */
     getLogsCount() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, e_4, _b, _c;
+            var _a, e_3, _b, _c;
             const now = Date.now();
             if (this.logsCountCache && now - this.logsCountCache.timestamp < this.CACHE_TTL) {
                 return this.logsCountCache.count;
@@ -696,53 +899,23 @@ class DatabaseManager {
                     count++;
                 }
             }
-            catch (e_4_1) { e_4 = { error: e_4_1 }; }
+            catch (e_3_1) { e_3 = { error: e_3_1 }; }
             finally {
                 try {
                     if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
                 }
-                finally { if (e_4) throw e_4.error; }
+                finally { if (e_3) throw e_3.error; }
             }
             this.logsCountCache = { count, timestamp: now };
             return count;
         });
     }
-    /**
-     * 获取访问日志总数（带缓存）
-     */
-    getAccessLogsCount() {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, e_5, _b, _c;
-            const now = Date.now();
-            if (this.accessLogsCountCache && now - this.accessLogsCountCache.timestamp < this.CACHE_TTL) {
-                return this.accessLogsCountCache.count;
-            }
-            let count = 0;
-            try {
-                for (var _d = true, _e = __asyncValues(this.accessLogDb.iterator()), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
-                    _c = _f.value;
-                    _d = false;
-                    const _ = _c;
-                    count++;
-                }
-            }
-            catch (e_5_1) { e_5 = { error: e_5_1 }; }
-            finally {
-                try {
-                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
-                }
-                finally { if (e_5) throw e_5.error; }
-            }
-            this.accessLogsCountCache = { count, timestamp: now };
-            return count;
-        });
-    }
     /**
      * 获取错误日志总数（带缓存）
      */
     getErrorLogsCount() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, e_6, _b, _c;
+            var _a, e_4, _b, _c;
             const now = Date.now();
             if (this.errorLogsCountCache && now - this.errorLogsCountCache.timestamp < this.CACHE_TTL) {
                 return this.errorLogsCountCache.count;
@@ -756,12 +929,12 @@ class DatabaseManager {
                     count++;
                 }
             }
-            catch (e_6_1) { e_6 = { error: e_6_1 }; }
+            catch (e_4_1) { e_4 = { error: e_4_1 }; }
             finally {
                 try {
                     if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
                 }
-                finally { if (e_6) throw e_6.error; }
+                finally { if (e_4) throw e_4.error; }
             }
             this.errorLogsCountCache = { count, timestamp: now };
             return count;
@@ -790,7 +963,7 @@ class DatabaseManager {
             }
         });
     }
-    addToBlacklist(serviceId, routeId, contentType, errorMessage, statusCode) {
+    addToBlacklist(serviceId, routeId, contentType, errorMessage, statusCode, errorType) {
         return __awaiter(this, void 0, void 0, function* () {
             const key = `${routeId}:${contentType}:${serviceId}`;
             const now = Date.now();
@@ -804,6 +977,7 @@ class DatabaseManager {
                 entry.errorCount++;
                 entry.lastError = errorMessage;
                 entry.lastStatusCode = statusCode;
+                entry.errorType = errorType;
                 yield this.blacklistDb.put(key, JSON.stringify(entry));
             }
             catch (error) {
@@ -818,6 +992,7 @@ class DatabaseManager {
                         errorCount: 1,
                         lastError: errorMessage,
                         lastStatusCode: statusCode,
+                        errorType,
                     };
                     yield this.blacklistDb.put(key, JSON.stringify(entry));
                 }
@@ -827,9 +1002,15 @@ class DatabaseManager {
             }
         });
     }
+    removeFromBlacklist(serviceId, routeId, contentType) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const key = `${routeId}:${contentType}:${serviceId}`;
+            yield this.blacklistDb.del(key);
+        });
+    }
     cleanupExpiredBlacklist() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, e_7, _b, _c;
+            var _a, e_5, _b, _c;
             const now = Date.now();
             let count = 0;
             try {
@@ -844,12 +1025,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_7_1) { e_7 = { error: e_7_1 }; }
+            catch (e_5_1) { e_5 = { error: e_5_1 }; }
             finally {
                 try {
                     if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
                 }
-                finally { if (e_7) throw e_7.error; }
+                finally { if (e_5) throw e_5.error; }
             }
             return count;
         });
@@ -902,8 +1083,8 @@ class DatabaseManager {
                 // Import API services
                 for (const service of importData.apiServices) {
                     this.db
-                        .prepare('INSERT INTO api_services (id, vendor_id, name, api_url, api_key, source_type, supported_models, model_limits, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-                        .run(service.id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.createdAt, service.updatedAt);
+                        .prepare('INSERT INTO api_services (id, vendor_id, name, api_url, api_key, source_type, auth_type, supported_models, model_limits, enable_proxy, enable_token_limit, token_limit, token_reset_interval, token_reset_base_time, enable_request_limit, request_count_limit, request_reset_interval, request_reset_base_time, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
+                        .run(service.id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.sourceType || null, service.authType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, service.enableProxy ? 1 : 0, service.enableTokenLimit ? 1 : 0, service.tokenLimit || null, service.tokenResetInterval || null, service.tokenResetBaseTime || null, service.enableRequestLimit ? 1 : 0, service.requestCountLimit || null, service.requestResetInterval || null, service.requestResetBaseTime || null, service.createdAt, service.updatedAt);
                 }
                 // Import routes
                 for (const route of importData.routes) {
@@ -914,8 +1095,8 @@ class DatabaseManager {
                 // Import rules
                 for (const rule of importData.rules) {
                     this.db
-                        .prepare('INSERT INTO rules (id, route_id, content_type, target_service_id, target_model, replaced_model, sort_order, timeout, token_limit, total_tokens_used, reset_interval, last_reset_at, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-                        .run(rule.id, rule.routeId, rule.contentType || 'default', rule.targetServiceId, rule.targetModel || null, rule.replacedModel || null, rule.sortOrder || 0, rule.timeout || null, rule.tokenLimit || null, rule.totalTokensUsed || 0, rule.resetInterval || null, rule.lastResetAt || null, rule.createdAt, rule.updatedAt);
+                        .prepare('INSERT INTO rules (id, route_id, content_type, target_service_id, target_model, replaced_model, sort_order, timeout, token_limit, total_tokens_used, reset_interval, last_reset_at, token_reset_base_time, request_count_limit, total_requests_used, request_reset_interval, request_last_reset_at, request_reset_base_time, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
+                        .run(rule.id, rule.routeId, rule.contentType || 'default', rule.targetServiceId, rule.targetModel || null, rule.replacedModel || null, rule.sortOrder || 0, rule.timeout || null, rule.tokenLimit || null, rule.totalTokensUsed || 0, rule.resetInterval || null, rule.lastResetAt || null, rule.tokenResetBaseTime || null, rule.requestCountLimit || null, rule.totalRequestsUsed || 0, rule.requestResetInterval || null, rule.requestLastResetAt || null, rule.requestResetBaseTime || null, rule.createdAt, rule.updatedAt);
                 }
                 // Update config
                 this.updateConfig(importData.config);
@@ -930,7 +1111,7 @@ class DatabaseManager {
     // Statistics operations
     getStatistics() {
         return __awaiter(this, arguments, void 0, function* (days = 30) {
-            var _a, e_8, _b, _c, _d, e_9, _e, _f;
+            var _a, e_6, _b, _c, _d, e_7, _e, _f;
             var _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y;
             const now = Date.now();
             const startTime = now - days * 24 * 60 * 60 * 1000;
@@ -947,12 +1128,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_8_1) { e_8 = { error: e_8_1 }; }
+            catch (e_6_1) { e_6 = { error: e_6_1 }; }
             finally {
                 try {
                     if (!_z && !_a && (_b = _0.return)) yield _b.call(_0);
                 }
-                finally { if (e_8) throw e_8.error; }
+                finally { if (e_6) throw e_6.error; }
             }
             // Get all error logs
             const errorLogs = [];
@@ -970,12 +1151,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_9_1) { e_9 = { error: e_9_1 }; }
+            catch (e_7_1) { e_7 = { error: e_7_1 }; }
             finally {
                 try {
                     if (!_2 && !_d && (_e = _3.return)) yield _e.call(_3);
                 }
-                finally { if (e_9) throw e_9.error; }
+                finally { if (e_7) throw e_7.error; }
             }
             // Get vendors and services for mapping
             const vendors = this.getVendors();
@@ -1150,10 +1331,180 @@ class DatabaseManager {
             };
         });
     }
+    getRuleBlacklistStatus(serviceId, routeId, contentType) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const key = `${routeId}:${contentType}:${serviceId}`;
+            try {
+                const value = yield this.blacklistDb.get(key);
+                const entry = JSON.parse(value);
+                // 检查是否过期
+                if (Date.now() > entry.expiresAt) {
+                    yield this.blacklistDb.del(key);
+                    return null;
+                }
+                return entry;
+            }
+            catch (error) {
+                if (error.code === 'LEVEL_NOT_FOUND') {
+                    return null;
+                }
+                throw error;
+            }
+        });
+    }
+    // Session operations
+    /**
+     * 创建或更新 session
+     * 当有新的请求日志时调用此方法来更新 session 信息
+     */
+    upsertSession(session) {
+        const now = Date.now();
+        const existing = this.db.prepare('SELECT * FROM sessions WHERE id = ?').get(session.id);
+        if (existing) {
+            // 更新现有 session
+            this.db.prepare(`
+        UPDATE sessions SET
+          last_request_at = ?,
+          request_count = request_count + 1,
+          total_tokens = total_tokens + ?,
+          vendor_id = ?,
+          vendor_name = ?,
+          service_id = ?,
+          service_name = ?,
+          model = ?
+        WHERE id = ?
+      `).run(now, session.totalTokens || 0, session.vendorId || null, session.vendorName || null, session.serviceId || null, session.serviceName || null, session.model || null, session.id);
+        }
+        else {
+            // 创建新 session
+            this.db.prepare(`
+        INSERT INTO sessions (id, target_type, title, first_request_at, last_request_at, request_count, total_tokens, vendor_id, vendor_name, service_id, service_name, model)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      `).run(session.id, session.targetType, session.title || null, session.firstRequestAt, now, 1, session.totalTokens || 0, session.vendorId || null, session.vendorName || null, session.serviceId || null, session.serviceName || null, session.model || null);
+        }
+    }
+    /**
+     * 获取所有 sessions
+     */
+    getSessions(limit = 100, offset = 0) {
+        const rows = this.db.prepare('SELECT * FROM sessions ORDER BY last_request_at DESC LIMIT ? OFFSET ?').all(limit, offset);
+        return rows.map((row) => ({
+            id: row.id,
+            targetType: row.target_type,
+            title: row.title,
+            firstRequestAt: row.first_request_at,
+            lastRequestAt: row.last_request_at,
+            requestCount: row.request_count,
+            totalTokens: row.total_tokens,
+            vendorId: row.vendor_id,
+            vendorName: row.vendor_name,
+            serviceId: row.service_id,
+            serviceName: row.service_name,
+            model: row.model,
+        }));
+    }
+    /**
+     * 根据 session ID 获取 session
+     */
+    getSession(id) {
+        const row = this.db.prepare('SELECT * FROM sessions WHERE id = ?').get(id);
+        if (!row)
+            return null;
+        return {
+            id: row.id,
+            targetType: row.target_type,
+            title: row.title,
+            firstRequestAt: row.first_request_at,
+            lastRequestAt: row.last_request_at,
+            requestCount: row.request_count,
+            totalTokens: row.total_tokens,
+            vendorId: row.vendor_id,
+            vendorName: row.vendor_name,
+            serviceId: row.service_id,
+            serviceName: row.service_name,
+            model: row.model,
+        };
+    }
+    /**
+     * 获取 sessions 总数
+     */
+    getSessionsCount() {
+        const result = this.db.prepare('SELECT COUNT(*) as count FROM sessions').get();
+        return result.count;
+    }
+    /**
+     * 删除指定 session
+     */
+    deleteSession(id) {
+        const result = this.db.prepare('DELETE FROM sessions WHERE id = ?').run(id);
+        return result.changes > 0;
+    }
+    /**
+     * 清空所有 sessions
+     */
+    clearSessions() {
+        this.db.prepare('DELETE FROM sessions').run();
+    }
+    /**
+     * 获取指定 session 的请求日志
+     * @param sessionId session ID
+     * @param limit 限制数量
+     */
+    getLogsBySessionId(sessionId_1) {
+        return __awaiter(this, arguments, void 0, function* (sessionId, limit = 100) {
+            var _a, e_8, _b, _c;
+            // 从 LevelDB 中读取所有日志
+            const allLogs = [];
+            try {
+                for (var _d = true, _e = __asyncValues(this.logDb.iterator()), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
+                    _c = _f.value;
+                    _d = false;
+                    const [, value] = _c;
+                    const log = JSON.parse(value);
+                    // 检查日志是否属于该 session（通过 headers 中的 session_id 或 body 中的 metadata.user_id）
+                    if (this.isLogBelongsToSession(log, sessionId)) {
+                        allLogs.push(log);
+                    }
+                }
+            }
+            catch (e_8_1) { e_8 = { error: e_8_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
+                }
+                finally { if (e_8) throw e_8.error; }
+            }
+            // 按时间倒序排序并限制数量
+            allLogs.sort((a, b) => b.timestamp - a.timestamp);
+            return allLogs.slice(0, limit);
+        });
+    }
+    /**
+     * 检查日志是否属于指定 session
+     */
+    isLogBelongsToSession(log, sessionId) {
+        var _a, _b;
+        // 检查 headers 中的 session_id（Codex）
+        if (((_a = log.headers) === null || _a === void 0 ? void 0 : _a['session_id']) === sessionId) {
+            return true;
+        }
+        // 检查 body 中的 metadata.user_id（Claude Code）
+        if (log.body) {
+            try {
+                const body = JSON.parse(log.body);
+                if (((_b = body.metadata) === null || _b === void 0 ? void 0 : _b.user_id) === sessionId) {
+                    return true;
+                }
+            }
+            catch (_c) {
+                // 忽略解析错误
+            }
+        }
+        return false;
+    }
     close() {
         this.db.close();
         this.logDb.close();
-        this.accessLogDb.close();
         this.errorLogDb.close();
         this.blacklistDb.close();
     }