aicodeswitch 3.0.1 → 3.0.3

package/dist/server/fs-database.js

@@ -8,6 +8,17 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
@@ -26,7 +37,7 @@ class FileSystemDatabaseManager {
     get vendorsFile() { return path_1.default.join(this.dataPath, 'vendors.json'); }
     get servicesFile() { return path_1.default.join(this.dataPath, 'services.json'); }
     get routesFile() { return path_1.default.join(this.dataPath, 'routes.json'); }
-    get rulesFile() { return path_1.default.join(this.dataPath, 'rules.json'); }
+    get rulesFile() { return path_1.default.join(this.dataPath, 'rules.json'); } // legacy
     get configFile() { return path_1.default.join(this.dataPath, 'config.json'); }
     get sessionsFile() { return path_1.default.join(this.dataPath, 'sessions.json'); }
     get logsDir() { return path_1.default.join(this.dataPath, 'logs'); }
@@ -76,12 +87,7 @@ class FileSystemDatabaseManager {
             writable: true,
             value: []
         });
-        Object.defineProperty(this, "apiServices", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: []
-        });
+        // 移除独立的 apiServices 存储，现在作为 vendor 的属性
         Object.defineProperty(this, "routes", {
             enumerable: true,
             configurable: true,
@@ -131,6 +137,18 @@ class FileSystemDatabaseManager {
             writable: true,
             value: this.createEmptyStatistics()
         });
+        Object.defineProperty(this, "contentTypeDistributionInitialized", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "contentTypeDistributionInitializing", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
         // 缓存机制
         Object.defineProperty(this, "logsCountCache", {
             enumerable: true,
@@ -178,10 +196,9 @@ class FileSystemDatabaseManager {
     loadAllData() {
         return __awaiter(this, void 0, void 0, function* () {
             yield Promise.all([
-                this.loadVendors(),
-                this.loadServices(),
+                this.loadVendors(), // loadVendors 内部会处理旧 services.json 的迁移
+                // 删除: this.loadServices(),
                 this.loadRoutes(),
-                this.loadRules(),
                 this.loadConfig(),
                 this.loadSessions(),
                 this.loadLogsIndex(),
@@ -200,59 +217,172 @@ class FileSystemDatabaseManager {
             catch (_a) {
                 this.vendors = [];
             }
+            // 兼容性检查：如果存在旧的 services.json，自动迁移
+            yield this.migrateServicesIfNeeded();
         });
     }
-    saveVendors() {
-        return __awaiter(this, void 0, void 0, function* () {
-            yield promises_1.default.writeFile(this.vendorsFile, JSON.stringify(this.vendors, null, 2));
-        });
-    }
-    loadServices() {
+    /**
+     * 检测并迁移旧的 services.json 到新结构
+     * 旧格式：vendors.json 和 services.json 分离
+     * 新格式：vendors.json 包含嵌套的 services 数组
+     */
+    migrateServicesIfNeeded() {
         return __awaiter(this, void 0, void 0, function* () {
+            const oldServicesFile = this.servicesFile;
             try {
-                const data = yield promises_1.default.readFile(this.servicesFile, 'utf-8');
-                this.apiServices = JSON.parse(data);
+                yield promises_1.default.access(oldServicesFile);
+                console.log('[Database] 发现旧的 services.json 文件，开始迁移到新结构...');
+                // 读取旧服务数据
+                const servicesData = yield promises_1.default.readFile(oldServicesFile, 'utf-8');
+                const oldServices = JSON.parse(servicesData);
+                console.log(`[Database] 准备迁移 ${oldServices.length} 个服务...`);
+                // 按 vendorId 分组
+                const servicesByVendor = new Map();
+                for (const service of oldServices) {
+                    if (!service.vendorId) {
+                        console.warn(`[Database] 跳过没有 vendorId 的服务: ${service.id}`);
+                        continue;
+                    }
+                    if (!servicesByVendor.has(service.vendorId)) {
+                        servicesByVendor.set(service.vendorId, []);
+                    }
+                    // 移除 vendorId 字段，因为现在通过父级关系隐式关联
+                    const { vendorId } = service, serviceWithoutVendorId = __rest(service, ["vendorId"]);
+                    servicesByVendor.get(service.vendorId).push(serviceWithoutVendorId);
+                }
+                // 合并到 vendors 数组
+                let migratedCount = 0;
+                for (const vendor of this.vendors) {
+                    const services = servicesByVendor.get(vendor.id);
+                    if (services) {
+                        vendor.services = services;
+                        migratedCount += services.length;
+                    }
+                    else {
+                        vendor.services = [];
+                    }
+                }
+                // 保存新的 vendors.json
+                yield this.saveVendors();
+                console.log(`[Database] 迁移完成：${migratedCount} 个服务已迁移`);
             }
-            catch (_a) {
-                this.apiServices = [];
+            catch (err) {
+                if (err.code === 'ENOENT') {
+                    // 旧文件不存在，这是正常的（新安装或已迁移）
+                    return;
+                }
+                console.error('[Database] 迁移 services 时出错:', err);
             }
         });
     }
-    saveServices() {
+    saveVendors() {
         return __awaiter(this, void 0, void 0, function* () {
-            yield promises_1.default.writeFile(this.servicesFile, JSON.stringify(this.apiServices, null, 2));
+            // 确保每个供应商都有 services 数组
+            const normalizedVendors = this.vendors.map(v => (Object.assign(Object.assign({}, v), { services: v.services || [] })));
+            yield promises_1.default.writeFile(this.vendorsFile, JSON.stringify(normalizedVendors, null, 2));
         });
     }
+    // loadServices 和 saveServices 已移除
+    // 服务现在作为供应商的属性存储在 vendors.json 中
+    // 迁移逻辑见 migrateServicesIfNeeded() 方法
     loadRoutes() {
         return __awaiter(this, void 0, void 0, function* () {
+            let routesFileFormat = 'missing';
+            let routesFromFile = [];
+            let rulesFromFile = [];
+            let hasRulesInRoutesFile = false;
             try {
                 const data = yield promises_1.default.readFile(this.routesFile, 'utf-8');
-                this.routes = JSON.parse(data);
+                const parsed = JSON.parse(data);
+                if (Array.isArray(parsed)) {
+                    routesFileFormat = 'array';
+                    routesFromFile = parsed;
+                }
+                else if (parsed && typeof parsed === 'object') {
+                    routesFileFormat = 'combined';
+                    routesFromFile = Array.isArray(parsed.routes) ? parsed.routes : [];
+                    if (Array.isArray(parsed.rules)) {
+                        rulesFromFile = parsed.rules;
+                        hasRulesInRoutesFile = true;
+                    }
+                }
+                else {
+                    routesFileFormat = 'unknown';
+                }
             }
             catch (_a) {
-                this.routes = [];
+                routesFileFormat = 'missing';
             }
+            this.routes = routesFromFile;
+            this.rules = rulesFromFile;
+            // 兼容旧的 rules.json 文件（迁移到 routes.json 的 rules 属性）
+            yield this.migrateRulesIfNeeded(routesFileFormat, hasRulesInRoutesFile);
         });
     }
-    saveRoutes() {
+    saveRoutesData() {
         return __awaiter(this, void 0, void 0, function* () {
-            yield promises_1.default.writeFile(this.routesFile, JSON.stringify(this.routes, null, 2));
+            const payload = {
+                routes: this.routes,
+                rules: this.rules,
+            };
+            yield promises_1.default.writeFile(this.routesFile, JSON.stringify(payload, null, 2));
         });
     }
-    loadRules() {
+    saveRoutes() {
         return __awaiter(this, void 0, void 0, function* () {
-            try {
-                const data = yield promises_1.default.readFile(this.rulesFile, 'utf-8');
-                this.rules = JSON.parse(data);
-            }
-            catch (_a) {
-                this.rules = [];
-            }
+            yield this.saveRoutesData();
         });
     }
     saveRules() {
         return __awaiter(this, void 0, void 0, function* () {
-            yield promises_1.default.writeFile(this.rulesFile, JSON.stringify(this.rules, null, 2));
+            yield this.saveRoutesData();
+        });
+    }
+    /**
+     * 检测并迁移旧的 rules.json 到 routes.json 的 rules 属性
+     * 旧格式：routes.json + rules.json 分离
+     * 新格式：routes.json 内包含 { routes, rules }
+     */
+    migrateRulesIfNeeded(routesFileFormat, hasRulesInRoutesFile) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const oldRulesFile = this.rulesFile;
+            const oldRulesExists = yield promises_1.default.access(oldRulesFile)
+                .then(() => true)
+                .catch(() => false);
+            let merged = false;
+            if (oldRulesExists) {
+                try {
+                    const data = yield promises_1.default.readFile(oldRulesFile, 'utf-8');
+                    const oldRules = JSON.parse(data);
+                    if (Array.isArray(oldRules)) {
+                        if (this.rules.length > 0) {
+                            const mergedMap = new Map();
+                            oldRules.forEach((rule, index) => {
+                                const key = (rule === null || rule === void 0 ? void 0 : rule.id) || `legacy-${index}`;
+                                if (!mergedMap.has(key)) {
+                                    mergedMap.set(key, rule);
+                                }
+                            });
+                            this.rules.forEach((rule, index) => {
+                                const key = (rule === null || rule === void 0 ? void 0 : rule.id) || `current-${index}`;
+                                mergedMap.set(key, rule);
+                            });
+                            this.rules = Array.from(mergedMap.values());
+                        }
+                        else {
+                            this.rules = oldRules;
+                        }
+                        merged = true;
+                    }
+                }
+                catch (error) {
+                    console.error('[Database] 迁移 rules.json 时出错:', error);
+                }
+            }
+            // 如果 routes.json 还是旧格式/缺失，或从旧 rules.json 合并过数据，或缺少 rules 字段，则写入新格式
+            if (routesFileFormat !== 'combined' || merged || (routesFileFormat === 'combined' && !hasRulesInRoutesFile)) {
+                yield this.saveRoutesData();
+            }
         });
     }
     loadConfig() {
@@ -387,10 +517,6 @@ class FileSystemDatabaseManager {
                 // 保存索引
                 yield this.saveLogsIndex();
                 console.log(`[Database] Successfully migrated ${migratedCount} log entries to ${this.logShardsIndex.length} shard(s)`);
-                // 备份旧文件
-                const backupFile = path_1.default.join(this.dataPath, 'logs.json.backup');
-                yield promises_1.default.rename(oldLogsFile, backupFile);
-                console.log(`[Database] Old logs.json backed up to ${backupFile}`);
             }
             catch (err) {
                 if (err.code === 'ENOENT') {
@@ -533,9 +659,11 @@ class FileSystemDatabaseManager {
             try {
                 const data = yield promises_1.default.readFile(this.statisticsFile, 'utf-8');
                 this.statistics = JSON.parse(data);
+                this.contentTypeDistributionInitialized = this.statistics.contentTypeDistribution.length > 0;
             }
             catch (_a) {
                 this.statistics = this.createEmptyStatistics();
+                this.contentTypeDistributionInitialized = false;
                 // 创建空文件
                 yield this.saveStatistics();
             }
@@ -573,11 +701,17 @@ class FileSystemDatabaseManager {
             return b.createdAt - a.createdAt;
         });
     }
+    // 新增：获取单个供应商（带服务）
+    getVendor(id) {
+        return this.vendors.find(v => v.id === id);
+    }
     createVendor(vendor) {
         return __awaiter(this, void 0, void 0, function* () {
+            console.log('[数据库] 创建供应商，输入数据:', JSON.stringify(vendor, null, 2));
             const id = crypto_1.default.randomUUID();
             const now = Date.now();
-            const newVendor = Object.assign(Object.assign({}, vendor), { id, createdAt: now, updatedAt: now });
+            const newVendor = Object.assign(Object.assign({}, vendor), { id, services: vendor.services || [], createdAt: now, updatedAt: now });
+            console.log('[数据库] 创建供应商，返回数据:', JSON.stringify(newVendor, null, 2));
             this.vendors.push(newVendor);
             yield this.saveVendors();
             return newVendor;
@@ -589,7 +723,7 @@ class FileSystemDatabaseManager {
             if (index === -1)
                 return false;
             const now = Date.now();
-            this.vendors[index] = Object.assign(Object.assign(Object.assign({}, this.vendors[index]), vendor), { id, updatedAt: now });
+            this.vendors[index] = Object.assign(Object.assign(Object.assign({}, this.vendors[index]), vendor), { id, services: vendor.services !== undefined ? vendor.services : this.vendors[index].services, updatedAt: now });
             yield this.saveVendors();
             return true;
         });
@@ -599,9 +733,13 @@ class FileSystemDatabaseManager {
             const index = this.vendors.findIndex(v => v.id === id);
             if (index === -1)
                 return false;
-            // 删除关联的服务
-            this.apiServices = this.apiServices.filter(s => s.vendorId !== id);
-            yield this.saveServices();
+            // 检查是否有服务被规则使用
+            const vendor = this.vendors[index];
+            const serviceIds = (vendor.services || []).map(s => s.id);
+            const rulesUsingServices = this.rules.filter(r => serviceIds.includes(r.targetServiceId));
+            if (rulesUsingServices.length > 0) {
+                throw new Error(`无法删除供应商：有 ${rulesUsingServices.length} 个规则正在使用该供应商的服务`);
+            }
             this.vendors.splice(index, 1);
             yield this.saveVendors();
             return true;
@@ -609,29 +747,89 @@ class FileSystemDatabaseManager {
     }
     // API Service operations
     getAPIServices(vendorId) {
-        const services = vendorId
-            ? this.apiServices.filter(s => s.vendorId === vendorId)
-            : this.apiServices;
-        return services.sort((a, b) => b.createdAt - a.createdAt);
+        if (vendorId) {
+            const vendor = this.vendors.find(v => v.id === vendorId);
+            if (!vendor)
+                return [];
+            // 返回指定供应商的服务，并添加 vendorId
+            return (vendor.services || []).map(service => (Object.assign(Object.assign({}, service), { vendorId: vendor.id // 添加 vendorId 以便前端使用
+             })));
+        }
+        // 返回所有供应商的所有服务（扁平化），并添加 vendorId
+        const allServices = [];
+        for (const vendor of this.vendors) {
+            if (vendor.services) {
+                const servicesWithVendorId = vendor.services.map(service => (Object.assign(Object.assign({}, service), { vendorId: vendor.id // 添加 vendorId 以便前端使用
+                 })));
+                allServices.push(...servicesWithVendorId);
+            }
+        }
+        return allServices.sort((a, b) => b.createdAt - a.createdAt);
+    }
+    // 新增：通过 ID 获取服务
+    getAPIService(id) {
+        var _a;
+        for (const vendor of this.vendors) {
+            const service = (_a = vendor.services) === null || _a === void 0 ? void 0 : _a.find(s => s.id === id);
+            if (service) {
+                return service;
+            }
+        }
+        return undefined;
+    }
+    // 新增：获取服务所属的供应商
+    getVendorByServiceId(serviceId) {
+        var _a;
+        for (const vendor of this.vendors) {
+            if ((_a = vendor.services) === null || _a === void 0 ? void 0 : _a.some(s => s.id === serviceId)) {
+                return vendor;
+            }
+        }
+        return undefined;
     }
     createAPIService(service) {
         return __awaiter(this, void 0, void 0, function* () {
+            console.log('[数据库] 创建服务，输入数据:', JSON.stringify(service, null, 2));
+            // 从 vendorId 找到供应商
+            const vendorId = service.vendorId;
+            if (!vendorId) {
+                throw new Error('创建服务时必须提供 vendorId');
+            }
+            const vendor = this.vendors.find(v => v.id === vendorId);
+            if (!vendor) {
+                throw new Error(`供应商不存在: ${vendorId}`);
+            }
+            // 移除 vendorId 字段（数据存储时不需要）
+            const _a = service, { vendorId: _ } = _a, serviceData = __rest(_a, ["vendorId"]);
             const id = crypto_1.default.randomUUID();
             const now = Date.now();
-            const newService = Object.assign(Object.assign({}, service), { id, createdAt: now, updatedAt: now });
-            this.apiServices.push(newService);
-            yield this.saveServices();
-            return newService;
+            const newService = Object.assign(Object.assign({}, serviceData), { id, createdAt: now, updatedAt: now });
+            console.log('[数据库] 创建服务，最终数据:', JSON.stringify(newService, null, 2));
+            if (!vendor.services) {
+                vendor.services = [];
+            }
+            vendor.services.push(newService);
+            // 更新供应商的 updatedAt 时间
+            vendor.updatedAt = now;
+            yield this.saveVendors();
+            console.log('[数据库] 服务已保存，当前总数:', vendor.services.length);
+            return Object.assign(Object.assign({}, newService), { vendorId });
         });
     }
     updateAPIService(id, service) {
         return __awaiter(this, void 0, void 0, function* () {
-            const index = this.apiServices.findIndex(s => s.id === id);
+            // 查找服务所属的供应商
+            const vendor = this.getVendorByServiceId(id);
+            if (!vendor)
+                return false;
+            const index = vendor.services.findIndex(s => s.id === id);
             if (index === -1)
                 return false;
             const now = Date.now();
-            this.apiServices[index] = Object.assign(Object.assign(Object.assign({}, this.apiServices[index]), service), { id, updatedAt: now });
-            yield this.saveServices();
+            vendor.services[index] = Object.assign(Object.assign(Object.assign({}, vendor.services[index]), service), { id, updatedAt: now });
+            // 更新供应商的 updatedAt 时间
+            vendor.updatedAt = now;
+            yield this.saveVendors();
             // 同步规则的超量限制
             yield this.syncRulesWithServiceLimits(id, service);
             return true;
@@ -639,14 +837,22 @@ class FileSystemDatabaseManager {
     }
     deleteAPIService(id) {
         return __awaiter(this, void 0, void 0, function* () {
-            const index = this.apiServices.findIndex(s => s.id === id);
+            // 查找服务所属的供应商
+            const vendor = this.getVendorByServiceId(id);
+            if (!vendor)
+                return false;
+            const index = vendor.services.findIndex(s => s.id === id);
             if (index === -1)
                 return false;
-            // 删除关联的规则
-            this.rules = this.rules.filter(r => r.targetServiceId !== id);
-            yield this.saveRules();
-            this.apiServices.splice(index, 1);
-            yield this.saveServices();
+            // 检查是否有规则正在使用此服务
+            const rulesUsingService = this.rules.filter(r => r.targetServiceId === id);
+            if (rulesUsingService.length > 0) {
+                throw new Error(`无法删除服务：有 ${rulesUsingService.length} 个规则正在使用此服务`);
+            }
+            vendor.services.splice(index, 1);
+            // 更新供应商的 updatedAt 时间
+            vendor.updatedAt = Date.now();
+            yield this.saveVendors();
             return true;
         });
     }
@@ -656,7 +862,7 @@ class FileSystemDatabaseManager {
             if (relatedRules.length === 0)
                 return;
             const now = Date.now();
-            const currentService = this.apiServices.find(s => s.id === serviceId);
+            const currentService = this.getAPIService(serviceId);
             if (!currentService)
                 return;
             let updated = false;
@@ -952,7 +1158,8 @@ class FileSystemDatabaseManager {
     addLog(log) {
         return __awaiter(this, void 0, void 0, function* () {
             const id = crypto_1.default.randomUUID();
-            const logWithId = Object.assign(Object.assign({}, log), { id });
+            const contentType = this.resolveLogContentType(log);
+            const logWithId = Object.assign(Object.assign({}, log), { contentType, id });
             // 获取目标分片文件名
             const filename = yield this.getLogShardFilename(logWithId.timestamp);
             // 加载现有分片数据
@@ -1158,11 +1365,20 @@ class FileSystemDatabaseManager {
     // Export/Import operations
     exportData(password) {
         return __awaiter(this, void 0, void 0, function* () {
+            // 扁平化所有服务（兼容旧格式）
+            const allServices = [];
+            for (const vendor of this.vendors) {
+                if (vendor.services) {
+                    // 为每个服务添加 vendorId（兼容旧格式）
+                    const servicesWithVendorId = vendor.services.map(s => (Object.assign(Object.assign({}, s), { vendorId: vendor.id })));
+                    allServices.push(...servicesWithVendorId);
+                }
+            }
             const exportData = {
-                version: '1.0.0',
+                version: '2.0.0', // 更新版本号
                 exportDate: Date.now(),
                 vendors: this.vendors,
-                apiServices: this.apiServices,
+                apiServices: allServices, // 兼容旧格式
                 routes: this.routes,
                 rules: this.rules,
                 config: this.config,
@@ -1178,16 +1394,36 @@ class FileSystemDatabaseManager {
                 const decrypted = crypto_js_1.default.AES.decrypt(encryptedData, password);
                 const jsonData = decrypted.toString(crypto_js_1.default.enc.Utf8);
                 const importData = JSON.parse(jsonData);
-                this.vendors = importData.vendors;
-                this.apiServices = importData.apiServices;
+                // 检测数据版本
+                const isNewFormat = importData.vendors.some((v) => v.services);
+                if (isNewFormat || importData.version >= '2.0.0') {
+                    // 新格式：直接使用 vendors（已包含 services）
+                    this.vendors = importData.vendors;
+                }
+                else {
+                    // 旧格式：需要重建 services 关系
+                    const { vendors, apiServices } = importData;
+                    // 构建 vendorId -> services 映射
+                    const servicesByVendor = new Map();
+                    for (const service of apiServices) {
+                        if (!service.vendorId)
+                            continue;
+                        if (!servicesByVendor.has(service.vendorId)) {
+                            servicesByVendor.set(service.vendorId, []);
+                        }
+                        const { vendorId } = service, serviceWithoutVendorId = __rest(service, ["vendorId"]);
+                        servicesByVendor.get(service.vendorId).push(serviceWithoutVendorId);
+                    }
+                    // 合并到 vendors
+                    this.vendors = vendors.map((vendor) => (Object.assign(Object.assign({}, vendor), { services: servicesByVendor.get(vendor.id) || [] })));
+                }
                 this.routes = importData.routes;
                 this.rules = importData.rules;
                 this.config = importData.config;
                 yield Promise.all([
                     this.saveVendors(),
-                    this.saveServices(),
+                    // 删除: this.saveServices(),
                     this.saveRoutes(),
-                    this.saveRules(),
                     this.saveConfig(),
                 ]);
                 return true;
@@ -1198,6 +1434,79 @@ class FileSystemDatabaseManager {
             }
         });
     }
+    inferContentTypeFromLog(log) {
+        var _a, _b;
+        const requestModel = ((_a = log.requestModel) === null || _a === void 0 ? void 0 : _a.toLowerCase()) || '';
+        let bodyText = '';
+        if (typeof log.body === 'string') {
+            bodyText = log.body;
+        }
+        else if (log.body !== undefined) {
+            try {
+                bodyText = JSON.stringify(log.body);
+            }
+            catch (_c) {
+                bodyText = '';
+            }
+        }
+        const lowerBody = bodyText.toLowerCase();
+        if (lowerBody.includes('image') || lowerBody.includes('base64')) {
+            return 'image-understanding';
+        }
+        if (requestModel.includes('think')) {
+            return 'thinking';
+        }
+        if ((((_b = log.usage) === null || _b === void 0 ? void 0 : _b.inputTokens) || 0) > 12000) {
+            return 'long-context';
+        }
+        return 'default';
+    }
+    resolveLogContentType(log) {
+        if (log.contentType) {
+            return log.contentType;
+        }
+        if (log.ruleId) {
+            const rule = this.getRule(log.ruleId);
+            if (rule === null || rule === void 0 ? void 0 : rule.contentType) {
+                return rule.contentType;
+            }
+        }
+        return this.inferContentTypeFromLog(log);
+    }
+    ensureContentTypeDistribution() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.contentTypeDistributionInitialized || this.contentTypeDistributionInitializing) {
+                return;
+            }
+            this.contentTypeDistributionInitializing = true;
+            try {
+                if (this.logShardsIndex.length === 0) {
+                    this.contentTypeDistributionInitialized = true;
+                    return;
+                }
+                const counts = new Map();
+                let totalRequests = 0;
+                for (const shard of this.logShardsIndex) {
+                    const shardLogs = yield this.loadLogShard(shard.filename);
+                    for (const log of shardLogs) {
+                        totalRequests++;
+                        const contentType = this.resolveLogContentType(log);
+                        counts.set(contentType, (counts.get(contentType) || 0) + 1);
+                    }
+                }
+                this.statistics.contentTypeDistribution = Array.from(counts.entries()).map(([contentType, count]) => ({
+                    contentType,
+                    count,
+                    percentage: totalRequests > 0 ? Math.round((count / totalRequests) * 100) : 0,
+                }));
+                this.contentTypeDistributionInitialized = true;
+                yield this.saveStatistics();
+            }
+            finally {
+                this.contentTypeDistributionInitializing = false;
+            }
+        });
+    }
     // Statistics operations
     /**
      * 更新统计数据 - 在每次添加日志时调用
@@ -1297,6 +1606,19 @@ class FileSystemDatabaseManager {
                 modelStats.avgResponseTime =
                     (modelStats.avgResponseTime * (modelStats.totalRequests - 1) + responseTime) / modelStats.totalRequests;
             }
+            // 更新 contentTypeDistribution
+            const resolvedContentType = this.resolveLogContentType(log);
+            let contentTypeStats = this.statistics.contentTypeDistribution.find(s => s.contentType === resolvedContentType);
+            if (!contentTypeStats) {
+                contentTypeStats = { contentType: resolvedContentType, count: 0, percentage: 0 };
+                this.statistics.contentTypeDistribution.push(contentTypeStats);
+            }
+            contentTypeStats.count++;
+            const totalRequests = this.statistics.overview.totalRequests;
+            for (const entry of this.statistics.contentTypeDistribution) {
+                entry.percentage = totalRequests > 0 ? Math.round((entry.count / totalRequests) * 100) : 0;
+            }
+            this.contentTypeDistributionInitialized = true;
             // 更新 timeline
             const date = new Date(log.timestamp).toISOString().split('T')[0];
             let timelineStats = this.statistics.timeline.find(t => t.date === date);
@@ -1324,6 +1646,7 @@ class FileSystemDatabaseManager {
      */
     getStatistics() {
         return __awaiter(this, arguments, void 0, function* (days = 30) {
+            yield this.ensureContentTypeDistribution();
             const now = Date.now();
             const startTime = now - days * 24 * 60 * 60 * 1000;
             // 过滤 timeline 数据
@@ -1420,7 +1743,8 @@ class FileSystemDatabaseManager {
         // 检查 body 中的 metadata.user_id（Claude Code）
         if (log.body) {
             try {
-                const body = JSON.parse(log.body);
+                // body 可能是对象（已解析）或字符串（未解析）
+                const body = typeof log.body === 'string' ? JSON.parse(log.body) : log.body;
                 if (((_b = body.metadata) === null || _b === void 0 ? void 0 : _b.user_id) === sessionId) {
                     return true;
                 }