aicodeswitch 1.8.0 → 1.10.1

package/.github/workflows/publish-to-npm.yaml

@@ -0,0 +1,34 @@
+name: Publish To NPM
+on:
+  pull_request:
+    types:
+      - closed
+    branches:
+      - main
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+      id-token: write
+    if: (github.event_name == 'push' && github.ref == 'refs/heads/main') || (github.event_name == 'pull_request' && github.event.pull_request.merged == true)
+    steps:
+      - uses: actions/checkout@v5
+      - uses: actions/setup-node@v4
+        with:
+          node-version: '24.x'
+          registry-url: 'https://registry.npmjs.org'
+      - run: npm install --verbose
+      - run: |
+          git config --global user.name 'github-actions[bot]'
+          git config --global user.email 'github-actions[bot]@users.noreply.github.com'
+      - run: npm run release
+      - run: npx can-npm-publish
+      - run: npm publish --provenance
+        env:
+          NODE_AUTH_TOKEN: ${{secrets.NPM_TOKEN}}
+      - uses: ad-m/github-push-action@master
+        with:
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          branch: main
+          tags: true

package/CHANGELOG.md

@@ -0,0 +1,15 @@
+# Changelog
+
+All notable changes to this project will be documented in this file. See [standard-version](https://github.com/conventional-changelog/standard-version) for commit guidelines.
+
+### 1.10.1 (2026-01-25)
+
+## [1.10.0](https://github.com/tangshuang/aicodeswitch/compare/v1.9.0...v1.10.0) (2026-01-25)
+
+
+### Features
+
+* 日志分页 ([0e68786](https://github.com/tangshuang/aicodeswitch/commit/0e68786b4e5bdce9bf7fc5ce85a1d4bdaf5a710c))
+* 新增了代理能力 ([d6254ae](https://github.com/tangshuang/aicodeswitch/commit/d6254ae463f01f583601ed71f64f346b63718853))
+* 优化细节 ([e4bdaef](https://github.com/tangshuang/aicodeswitch/commit/e4bdaef14bb0088e7feb5371490124c8dd169fdd))
+* github workflows ([37dd371](https://github.com/tangshuang/aicodeswitch/commit/37dd3717e79dab1c4aec99b762774cc549f1efc8))

package/dist/server/database.js

@@ -57,6 +57,31 @@ class DatabaseManager {
             writable: true,
             value: void 0
         });
+        // 缓存机制：总数查询缓存
+        Object.defineProperty(this, "logsCountCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "accessLogsCountCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "errorLogsCountCache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "CACHE_TTL", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 1000
+        }); // 1秒缓存TTL
         this.db = new better_sqlite3_1.default(path_1.default.join(dataPath, 'app.db'));
         // 启用外键约束（SQLite 默认禁用，必须手动启用才能使 ON DELETE CASCADE 生效）
         this.db.pragma('foreign_keys = ON');
@@ -100,6 +125,61 @@ class DatabaseManager {
                     console.log('[DB] Migration completed: model_limits column added');
                 }
             }
+            // 检查vendors表是否有sort_order字段
+            const vendorsColumns = this.db.pragma('table_info(vendors)');
+            const hasSortOrder = vendorsColumns.some((col) => col.name === 'sort_order');
+            if (!hasSortOrder) {
+                console.log('[DB] Running migration: Adding sort_order column to vendors table');
+                this.db.exec('ALTER TABLE vendors ADD COLUMN sort_order INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: sort_order column added to vendors');
+            }
+            // 检查rules表是否有token相关字段
+            const rulesColumns = this.db.pragma('table_info(rules)');
+            const hasTokenLimit = rulesColumns.some((col) => col.name === 'token_limit');
+            const hasTotalTokensUsed = rulesColumns.some((col) => col.name === 'total_tokens_used');
+            const hasResetInterval = rulesColumns.some((col) => col.name === 'reset_interval');
+            const hasLastResetAt = rulesColumns.some((col) => col.name === 'last_reset_at');
+            if (!hasTokenLimit) {
+                console.log('[DB] Running migration: Adding token_limit column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN token_limit INTEGER;');
+                console.log('[DB] Migration completed: token_limit column added');
+            }
+            if (!hasTotalTokensUsed) {
+                console.log('[DB] Running migration: Adding total_tokens_used column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN total_tokens_used INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: total_tokens_used column added');
+            }
+            if (!hasResetInterval) {
+                console.log('[DB] Running migration: Adding reset_interval column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN reset_interval INTEGER;');
+                console.log('[DB] Migration completed: reset_interval column added');
+            }
+            if (!hasLastResetAt) {
+                console.log('[DB] Running migration: Adding last_reset_at column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN last_reset_at INTEGER;');
+                console.log('[DB] Migration completed: last_reset_at column added');
+            }
+            // 检查rules表是否有timeout字段
+            const hasRuleTimeout = rulesColumns.some((col) => col.name === 'timeout');
+            if (!hasRuleTimeout) {
+                console.log('[DB] Running migration: Adding timeout column to rules table');
+                this.db.exec('ALTER TABLE rules ADD COLUMN timeout INTEGER;');
+                console.log('[DB] Migration completed: timeout column added to rules');
+            }
+            // 检查api_services表是否有timeout字段，如果有则移除
+            const hasServiceTimeout = columns.some((col) => col.name === 'timeout');
+            if (hasServiceTimeout) {
+                console.log('[DB] Running migration: Removing timeout column from api_services table');
+                yield this.migrateRemoveServiceTimeout();
+                console.log('[DB] Migration completed: timeout column removed from api_services');
+            }
+            // 检查api_services表是否有enable_proxy字段
+            const hasEnableProxy = columns.some((col) => col.name === 'enable_proxy');
+            if (!hasEnableProxy) {
+                console.log('[DB] Running migration: Adding enable_proxy column to api_services table');
+                this.db.exec('ALTER TABLE api_services ADD COLUMN enable_proxy INTEGER DEFAULT 0;');
+                console.log('[DB] Migration completed: enable_proxy column added');
+            }
         });
     }
     migrateMaxOutputTokensToModelLimits() {
@@ -138,12 +218,42 @@ class DatabaseManager {
             console.log('[DB] Migration completed: Replaced max_output_tokens with model_limits');
         });
     }
+    migrateRemoveServiceTimeout() {
+        return __awaiter(this, void 0, void 0, function* () {
+            // SQLite 不支持直接删除列，需要重建表
+            this.db.pragma('foreign_keys = OFF');
+            this.db.exec(`
+      CREATE TABLE api_services_new (
+        id TEXT PRIMARY KEY,
+        vendor_id TEXT NOT NULL,
+        name TEXT NOT NULL,
+        api_url TEXT NOT NULL,
+        api_key TEXT NOT NULL,
+        source_type TEXT,
+        supported_models TEXT,
+        model_limits TEXT,
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL,
+        FOREIGN KEY (vendor_id) REFERENCES vendors(id) ON DELETE CASCADE
+      );
+
+      INSERT INTO api_services_new (id, vendor_id, name, api_url, api_key, source_type, supported_models, model_limits, created_at, updated_at)
+      SELECT id, vendor_id, name, api_url, api_key, source_type, supported_models, model_limits, created_at, updated_at
+      FROM api_services;
+
+      DROP TABLE api_services;
+      ALTER TABLE api_services_new RENAME TO api_services;
+    `);
+            this.db.pragma('foreign_keys = ON');
+        });
+    }
     createTables() {
         this.db.exec(`
       CREATE TABLE IF NOT EXISTS vendors (
         id TEXT PRIMARY KEY,
         name TEXT NOT NULL,
         description TEXT,
+        sort_order INTEGER DEFAULT 0,
         created_at INTEGER NOT NULL,
         updated_at INTEGER NOT NULL
       );
@@ -154,7 +264,6 @@ class DatabaseManager {
          name TEXT NOT NULL,
          api_url TEXT NOT NULL,
          api_key TEXT NOT NULL,
-         timeout INTEGER,
          source_type TEXT,
          supported_models TEXT,
          created_at INTEGER NOT NULL,
@@ -180,6 +289,11 @@ class DatabaseManager {
         target_model TEXT,
         replaced_model TEXT,
         sort_order INTEGER DEFAULT 0,
+        timeout INTEGER,
+        token_limit INTEGER,
+        total_tokens_used INTEGER DEFAULT 0,
+        reset_interval INTEGER,
+        last_reset_at INTEGER,
         created_at INTEGER NOT NULL,
         updated_at INTEGER NOT NULL,
         FOREIGN KEY (route_id) REFERENCES routes(id) ON DELETE CASCADE,
@@ -199,9 +313,13 @@ class DatabaseManager {
                 const defaultConfig = {
                     enableLogging: true,
                     logRetentionDays: 30,
-                    maxLogSize: 1000,
+                    maxLogSize: 100000,
                     apiKey: '',
                     enableFailover: true, // 默认启用智能故障切换
+                    proxyEnabled: false, // 默认不启用代理
+                    proxyUrl: '',
+                    proxyUsername: '',
+                    proxyPassword: '',
                 };
                 this.db.prepare('INSERT INTO config (key, value) VALUES (?, ?)').run('app_config', JSON.stringify(defaultConfig));
             }
@@ -209,11 +327,12 @@ class DatabaseManager {
     }
     // Vendor operations
     getVendors() {
-        const rows = this.db.prepare('SELECT * FROM vendors ORDER BY created_at DESC').all();
+        const rows = this.db.prepare('SELECT * FROM vendors ORDER BY sort_order DESC, created_at DESC').all();
         return rows.map((row) => ({
             id: row.id,
             name: row.name,
             description: row.description,
+            sortOrder: row.sort_order,
             createdAt: row.created_at,
             updatedAt: row.updated_at,
         }));
@@ -222,15 +341,15 @@ class DatabaseManager {
         const id = crypto_1.default.randomUUID();
         const now = Date.now();
         this.db
-            .prepare('INSERT INTO vendors (id, name, description, created_at, updated_at) VALUES (?, ?, ?, ?, ?)')
-            .run(id, vendor.name, vendor.description || null, now, now);
+            .prepare('INSERT INTO vendors (id, name, description, sort_order, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)')
+            .run(id, vendor.name, vendor.description || null, vendor.sortOrder || 0, now, now);
         return Object.assign(Object.assign({}, vendor), { id, createdAt: now, updatedAt: now });
     }
     updateVendor(id, vendor) {
         const now = Date.now();
         const result = this.db
-            .prepare('UPDATE vendors SET name = ?, description = ?, updated_at = ? WHERE id = ?')
-            .run(vendor.name, vendor.description || null, now, id);
+            .prepare('UPDATE vendors SET name = ?, description = ?, sort_order = ?, updated_at = ? WHERE id = ?')
+            .run(vendor.name, vendor.description || null, vendor.sortOrder !== undefined ? vendor.sortOrder : 0, now, id);
         return result.changes > 0;
     }
     deleteVendor(id) {
@@ -252,10 +371,10 @@ class DatabaseManager {
             name: row.name,
             apiUrl: row.api_url,
             apiKey: row.api_key,
-            timeout: row.timeout,
             sourceType: row.source_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,
             createdAt: row.created_at,
             updatedAt: row.updated_at,
         }));
@@ -269,18 +388,18 @@ 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, timeout, source_type, supported_models, model_limits, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-            .run(id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.timeout || null, service.sourceType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, now, now);
+            .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);
         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 = ?, timeout = ?, source_type = ?, supported_models = ?, model_limits = ?, updated_at = ? WHERE id = ?')
-            .run(service.name, service.apiUrl, service.apiKey, service.timeout || null, service.sourceType || null, service.supportedModels ? service.supportedModels.join(',') : null, service.modelLimits ? JSON.stringify(service.modelLimits) : null, now, id);
+            .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);
         // 调试日志: 记录更新操作
         if (result.changes > 0 && process.env.NODE_ENV === 'development') {
-            console.log(`[DB] Updated service ${id}: ${service.name} -> ${service.apiUrl} (timeout: ${service.timeout})`);
+            console.log(`[DB] Updated service ${id}: ${service.name} -> ${service.apiUrl}`);
         }
         return result.changes > 0;
     }
@@ -348,6 +467,11 @@ class DatabaseManager {
             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,
             createdAt: row.created_at,
             updatedAt: row.updated_at,
         }));
@@ -356,26 +480,79 @@ class DatabaseManager {
         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, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)')
-            .run(id, route.routeId, route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, 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, 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);
         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 = ?, updated_at = ? WHERE id = ?')
-            .run(route.contentType, route.targetServiceId, route.targetModel || null, route.replacedModel || null, route.sortOrder || 0, now, id);
+            .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);
         return result.changes > 0;
     }
     deleteRule(id) {
         const result = this.db.prepare('DELETE FROM rules WHERE id = ?').run(id);
         return result.changes > 0;
     }
+    /**
+     * 增加规则的token使用量
+     * @param ruleId 规则ID
+     * @param tokensUsed 使用的token数量
+     * @returns 是否成功
+     */
+    incrementRuleTokenUsage(ruleId, tokensUsed) {
+        const result = this.db
+            .prepare('UPDATE rules SET total_tokens_used = total_tokens_used + ? WHERE id = ?')
+            .run(tokensUsed, ruleId);
+        return result.changes > 0;
+    }
+    /**
+     * 重置规则的token使用量
+     * @param ruleId 规则ID
+     * @returns 是否成功
+     */
+    resetRuleTokenUsage(ruleId) {
+        const now = Date.now();
+        const result = this.db
+            .prepare('UPDATE rules SET total_tokens_used = 0, last_reset_at = ? WHERE id = ?')
+            .run(now, ruleId);
+        return result.changes > 0;
+    }
+    /**
+     * 检查并重置到期的规则
+     * 如果规则设置了reset_interval且已经到了重置时间，则自动重置token使用量
+     * @param ruleId 规则ID
+     * @returns 是否进行了重置
+     */
+    checkAndResetRuleIfNeeded(ruleId) {
+        const rule = this.db
+            .prepare('SELECT reset_interval, last_reset_at 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 lastResetAt = rule.last_reset_at || 0;
+        // 检查是否已经到了重置时间
+        if (now - lastResetAt >= resetIntervalMs) {
+            this.resetRuleTokenUsage(ruleId);
+            return true;
+        }
+        return false;
+    }
     // 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 })));
+            // 清除缓存
+            this.logsCountCache = null;
         });
     }
     getLogs() {
@@ -406,6 +583,8 @@ class DatabaseManager {
     clearLogs() {
         return __awaiter(this, void 0, void 0, function* () {
             yield this.logDb.clear();
+            // 清除缓存
+            this.logsCountCache = null;
         });
     }
     // Access log operations
@@ -413,6 +592,8 @@ class DatabaseManager {
         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;
         });
     }
@@ -451,6 +632,8 @@ class DatabaseManager {
     clearAccessLogs() {
         return __awaiter(this, void 0, void 0, function* () {
             yield this.accessLogDb.clear();
+            // 清除缓存
+            this.accessLogsCountCache = null;
         });
     }
     // Error log operations
@@ -458,6 +641,8 @@ class DatabaseManager {
         return __awaiter(this, void 0, void 0, function* () {
             const id = crypto_1.default.randomUUID();
             yield this.errorLogDb.put(id, JSON.stringify(Object.assign(Object.assign({}, log), { id })));
+            // 清除缓存
+            this.errorLogsCountCache = null;
         });
     }
     getErrorLogs() {
@@ -488,6 +673,98 @@ class DatabaseManager {
     clearErrorLogs() {
         return __awaiter(this, void 0, void 0, function* () {
             yield this.errorLogDb.clear();
+            // 清除缓存
+            this.errorLogsCountCache = null;
+        });
+    }
+    /**
+     * 获取请求日志总数（带缓存）
+     */
+    getLogsCount() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, e_4, _b, _c;
+            const now = Date.now();
+            if (this.logsCountCache && now - this.logsCountCache.timestamp < this.CACHE_TTL) {
+                return this.logsCountCache.count;
+            }
+            let count = 0;
+            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 _ = _c;
+                    count++;
+                }
+            }
+            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_4) throw e_4.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;
+            const now = Date.now();
+            if (this.errorLogsCountCache && now - this.errorLogsCountCache.timestamp < this.CACHE_TTL) {
+                return this.errorLogsCountCache.count;
+            }
+            let count = 0;
+            try {
+                for (var _d = true, _e = __asyncValues(this.errorLogDb.iterator()), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
+                    _c = _f.value;
+                    _d = false;
+                    const _ = _c;
+                    count++;
+                }
+            }
+            catch (e_6_1) { e_6 = { error: e_6_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
+                }
+                finally { if (e_6) throw e_6.error; }
+            }
+            this.errorLogsCountCache = { count, timestamp: now };
+            return count;
         });
     }
     // Service blacklist operations
@@ -552,7 +829,7 @@ class DatabaseManager {
     }
     cleanupExpiredBlacklist() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, e_4, _b, _c;
+            var _a, e_7, _b, _c;
             const now = Date.now();
             let count = 0;
             try {
@@ -567,12 +844,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_4_1) { e_4 = { error: e_4_1 }; }
+            catch (e_7_1) { e_7 = { error: e_7_1 }; }
             finally {
                 try {
                     if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
                 }
-                finally { if (e_4) throw e_4.error; }
+                finally { if (e_7) throw e_7.error; }
             }
             return count;
         });
@@ -619,14 +896,14 @@ class DatabaseManager {
                 // Import vendors
                 for (const vendor of importData.vendors) {
                     this.db
-                        .prepare('INSERT INTO vendors (id, name, description, created_at, updated_at) VALUES (?, ?, ?, ?, ?)')
-                        .run(vendor.id, vendor.name, vendor.description || null, vendor.createdAt, vendor.updatedAt);
+                        .prepare('INSERT INTO vendors (id, name, description, sort_order, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)')
+                        .run(vendor.id, vendor.name, vendor.description || null, vendor.sortOrder || 0, vendor.createdAt, vendor.updatedAt);
                 }
                 // Import API services
                 for (const service of importData.apiServices) {
                     this.db
-                        .prepare('INSERT INTO api_services (id, vendor_id, name, api_url, api_key, timeout, source_type, supported_models, model_limits, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)')
-                        .run(service.id, service.vendorId, service.name, service.apiUrl, service.apiKey, service.timeout || null, 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, 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);
                 }
                 // Import routes
                 for (const route of importData.routes) {
@@ -637,8 +914,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, 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.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, 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);
                 }
                 // Update config
                 this.updateConfig(importData.config);
@@ -653,7 +930,7 @@ class DatabaseManager {
     // Statistics operations
     getStatistics() {
         return __awaiter(this, arguments, void 0, function* (days = 30) {
-            var _a, e_5, _b, _c, _d, e_6, _e, _f;
+            var _a, e_8, _b, _c, _d, e_9, _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;
@@ -670,12 +947,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_5_1) { e_5 = { error: e_5_1 }; }
+            catch (e_8_1) { e_8 = { error: e_8_1 }; }
             finally {
                 try {
                     if (!_z && !_a && (_b = _0.return)) yield _b.call(_0);
                 }
-                finally { if (e_5) throw e_5.error; }
+                finally { if (e_8) throw e_8.error; }
             }
             // Get all error logs
             const errorLogs = [];
@@ -693,12 +970,12 @@ class DatabaseManager {
                     }
                 }
             }
-            catch (e_6_1) { e_6 = { error: e_6_1 }; }
+            catch (e_9_1) { e_9 = { error: e_9_1 }; }
             finally {
                 try {
                     if (!_2 && !_d && (_e = _3.return)) yield _e.call(_3);
                 }
-                finally { if (e_6) throw e_6.error; }
+                finally { if (e_9) throw e_9.error; }
             }
             // Get vendors and services for mapping
             const vendors = this.getVendors();

package/dist/server/main.js

@@ -282,6 +282,7 @@ const registerRoutes = (dbManager, proxyServer) => {
     app.post('/api/rules', (req, res) => res.json(dbManager.createRule(req.body)));
     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.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;
@@ -318,6 +319,18 @@ const registerRoutes = (dbManager, proxyServer) => {
         yield dbManager.clearErrorLogs();
         res.json(true);
     })));
+    app.get('/api/logs/count', asyncHandler((_req, res) => __awaiter(void 0, void 0, void 0, function* () {
+        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 });
+    })));
     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;