@lightharu/krouter 1.8.0

package/out-server/main/proxy/logger.js

@@ -0,0 +1,407 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.proxyLogger = exports.proxyLogStore = void 0;
+exports.interceptConsole = interceptConsole;
+// 代理服务器日志模块
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const redact_1 = require("../utils/redact");
+const runtimePaths_1 = require("../runtimePaths");
+const DEFAULT_CONFIG = {
+    enabled: false,
+    maxFileSize: 10 * 1024 * 1024, // 10MB
+    maxFiles: 5,
+    logToConsole: true
+};
+class ProxyLogger {
+    config;
+    logStream = null;
+    currentLogFile = '';
+    currentFileSize = 0;
+    constructor() {
+        this.config = { ...DEFAULT_CONFIG };
+    }
+    configure(config) {
+        this.config = { ...this.config, ...config };
+        if (this.config.enabled && !this.config.logDir) {
+            // 默认日志目录
+            this.config.logDir = path.join((0, runtimePaths_1.getRuntimeUserDataPath)(), 'logs', 'proxy');
+        }
+        if (this.config.enabled) {
+            this.initLogFile();
+        }
+        else {
+            this.close();
+        }
+    }
+    initLogFile() {
+        if (!this.config.logDir)
+            return;
+        try {
+            // 确保目录存在
+            fs.mkdirSync(this.config.logDir, { recursive: true });
+            // 创建新的日志文件
+            const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+            this.currentLogFile = path.join(this.config.logDir, `proxy-${timestamp}.log`);
+            this.logStream = fs.createWriteStream(this.currentLogFile, { flags: 'a' });
+            this.currentFileSize = 0;
+            this.info('Logger', 'Log file initialized', { file: this.currentLogFile });
+        }
+        catch (error) {
+            console.error('[ProxyLogger] Failed to init log file:', error);
+        }
+    }
+    rotateIfNeeded() {
+        if (!this.config.maxFileSize || this.currentFileSize < this.config.maxFileSize) {
+            return;
+        }
+        this.close();
+        this.cleanOldLogs();
+        this.initLogFile();
+    }
+    cleanOldLogs() {
+        if (!this.config.logDir || !this.config.maxFiles)
+            return;
+        try {
+            const files = fs.readdirSync(this.config.logDir)
+                .filter(f => f.startsWith('proxy-') && f.endsWith('.log'))
+                .map(f => ({
+                name: f,
+                path: path.join(this.config.logDir, f),
+                time: fs.statSync(path.join(this.config.logDir, f)).mtime.getTime()
+            }))
+                .sort((a, b) => b.time - a.time);
+            // 删除超出数量限制的旧文件
+            while (files.length >= this.config.maxFiles) {
+                const oldest = files.pop();
+                if (oldest) {
+                    fs.unlinkSync(oldest.path);
+                }
+            }
+        }
+        catch (error) {
+            console.error('[ProxyLogger] Failed to clean old logs:', error);
+        }
+    }
+    isWriting = false;
+    write(rawEntry) {
+        // 统一脱敏：message + data 里的代理账密 / token / password 等，避免明文落盘或上屏
+        const entry = {
+            ...rawEntry,
+            message: (0, redact_1.redactString)(rawEntry.message),
+            data: rawEntry.data === undefined ? undefined : (0, redact_1.redactValue)(rawEntry.data)
+        };
+        const line = JSON.stringify(entry) + '\n';
+        if (this.config.logToConsole) {
+            const prefix = `[${entry.level}][${entry.category}]`;
+            // 设置 flag 防止 console 拦截器重复写入 proxyLogStore
+            this.isWriting = true;
+            if (entry.level === 'ERROR') {
+                console.error(prefix, entry.message, entry.data || '');
+            }
+            else if (entry.level === 'WARN') {
+                console.warn(prefix, entry.message, entry.data || '');
+            }
+            else {
+                console.log(prefix, entry.message, entry.data || '');
+            }
+            this.isWriting = false;
+        }
+        if (this.config.enabled && this.logStream) {
+            this.logStream.write(line);
+            this.currentFileSize += Buffer.byteLength(line);
+            this.rotateIfNeeded();
+        }
+        // 同时添加到内存存储（用于 UI 显示）
+        exports.proxyLogStore.add(entry);
+    }
+    get _isWriting() { return this.isWriting; }
+    debug(category, message, data) {
+        this.write({
+            timestamp: new Date().toISOString(),
+            level: 'DEBUG',
+            category,
+            message,
+            data
+        });
+    }
+    info(category, message, data) {
+        this.write({
+            timestamp: new Date().toISOString(),
+            level: 'INFO',
+            category,
+            message,
+            data
+        });
+    }
+    warn(category, message, data) {
+        this.write({
+            timestamp: new Date().toISOString(),
+            level: 'WARN',
+            category,
+            message,
+            data
+        });
+    }
+    error(category, message, data) {
+        this.write({
+            timestamp: new Date().toISOString(),
+            level: 'ERROR',
+            category,
+            message,
+            data
+        });
+    }
+    // 记录请求
+    request(info) {
+        this.info('Request', `${info.method} ${info.path}`, info);
+    }
+    // 记录响应
+    response(info) {
+        if (info.error) {
+            this.error('Response', `${info.path} -> ${info.status}`, info);
+        }
+        else {
+            this.info('Response', `${info.path} -> ${info.status}`, info);
+        }
+    }
+    // 记录 Token 刷新
+    tokenRefresh(accountId, success, error) {
+        if (success) {
+            this.info('TokenRefresh', `Account ${accountId} refreshed successfully`);
+        }
+        else {
+            this.error('TokenRefresh', `Account ${accountId} refresh failed`, { error });
+        }
+    }
+    close() {
+        if (this.logStream) {
+            this.logStream.end();
+            this.logStream = null;
+        }
+    }
+    getLogDir() {
+        return this.config.logDir;
+    }
+}
+// 内存日志存储（用于 UI 显示）
+//
+// 性能修复要点：
+// 1. maxLogs 从 100万 降至 5万 — 避免序列化数百 MB 大对象阻塞主进程
+// 2. save() 改为异步 fs.promises.writeFile — 不再 freeze 主进程事件循环
+// 3. 单次写盘原子化（in-flight guard）防止并发 writeFile 导致竞态
+// 4. 写盘节流间隔从 5s 提至 30s — 大幅降低高频日志场景下的 IO 频率
+// 5. 应用退出时通过 flushSaveNow() 强制写盘，防止数据丢失
+class ProxyLogStore {
+    logs = [];
+    // 5 万条 × 平均 200 字节 ≈ 10 MB；既能覆盖常规调试需求，又把单次写盘成本控制在可接受范围内
+    maxLogs = 50000;
+    listeners = [];
+    storePath = '';
+    initialized = false;
+    initialize(userDataPath) {
+        if (this.initialized)
+            return;
+        this.initialized = true;
+        this.storePath = path.join(userDataPath, 'proxy-logs.json');
+        this.load();
+    }
+    load() {
+        try {
+            if (fs.existsSync(this.storePath)) {
+                const data = fs.readFileSync(this.storePath, 'utf-8');
+                const parsed = JSON.parse(data);
+                // 验证并过滤有效的日志条目
+                const filtered = Array.isArray(parsed) ? parsed.filter((log) => {
+                    if (!log.timestamp || isNaN(new Date(log.timestamp).getTime()))
+                        return false;
+                    if (!log.level || !log.category)
+                        return false;
+                    return true;
+                }) : [];
+                // 加载时也施加上限，避免旧版本遗留的超大日志文件导致首次启动卡顿
+                this.logs = filtered.length > this.maxLogs ? filtered.slice(-this.maxLogs) : filtered;
+                console.log(`[ProxyLogStore] Loaded ${this.logs.length} valid logs`);
+            }
+        }
+        catch (error) {
+            console.error('[ProxyLogStore] Failed to load logs:', error);
+            this.logs = [];
+        }
+    }
+    /** 异步保存日志（不阻塞主进程事件循环）。并发调用通过 in-flight 标志合并。 */
+    writeInFlight = false;
+    writePending = false;
+    async save() {
+        if (this.writeInFlight) {
+            // 已有写盘进行中：标记 pending，让其完成后立即重写最新数据
+            this.writePending = true;
+            return;
+        }
+        this.writeInFlight = true;
+        try {
+            // 拷贝引用快照（不复制数组）以保证 JSON.stringify 期间数据稳定
+            const snapshot = this.logs;
+            await fs.promises.writeFile(this.storePath, JSON.stringify(snapshot), 'utf-8');
+        }
+        catch (error) {
+            console.error('[ProxyLogStore] Failed to save logs:', error);
+        }
+        finally {
+            this.writeInFlight = false;
+            if (this.writePending) {
+                this.writePending = false;
+                // 用 microtask 而不是立即递归，避免栈深问题
+                queueMicrotask(() => { void this.save(); });
+            }
+        }
+    }
+    saveTimer = null;
+    add(entry) {
+        this.logs.push(entry);
+        // 超过最大数量时删除最旧的
+        if (this.logs.length > this.maxLogs) {
+            this.logs = this.logs.slice(-this.maxLogs);
+        }
+        // 通知监听器（异常隔离）
+        for (const listener of this.listeners) {
+            try {
+                listener(entry);
+            }
+            catch { /* ignore */ }
+        }
+        // 节流调度异步保存
+        this.scheduleSave();
+    }
+    scheduleSave() {
+        if (this.saveTimer)
+            return; // 已调度，等待 flush
+        this.saveTimer = setTimeout(() => {
+            this.saveTimer = null;
+            void this.save();
+        }, 30_000); // 30 秒批量写盘一次（异步、不阻塞）
+    }
+    /** 强制立即写盘（用于退出场景），保证最新数据落盘 */
+    async flushSaveNow() {
+        if (this.saveTimer) {
+            clearTimeout(this.saveTimer);
+            this.saveTimer = null;
+        }
+        await this.save();
+    }
+    getAll() {
+        return [...this.logs];
+    }
+    getLast(count) {
+        return this.logs.slice(-count);
+    }
+    clear() {
+        this.logs = [];
+        void this.save();
+    }
+    count() {
+        return this.logs.length;
+    }
+    onLog(listener) {
+        this.listeners.push(listener);
+        return () => {
+            const index = this.listeners.indexOf(listener);
+            if (index >= 0) {
+                this.listeners.splice(index, 1);
+            }
+        };
+    }
+}
+exports.proxyLogStore = new ProxyLogStore();
+// 单例导出
+exports.proxyLogger = new ProxyLogger();
+// 拦截主进程 console 输出，自动转发到 proxyLogStore
+// 这样所有 console.log/warn/error 都能在日志页面显示
+let consoleIntercepted = false;
+function interceptConsole() {
+    if (consoleIntercepted)
+        return;
+    consoleIntercepted = true;
+    const originalLog = console.log;
+    const originalWarn = console.warn;
+    const originalError = console.error;
+    const parseConsoleCategory = (args) => {
+        const first = String(args[0] || '');
+        // 匹配 [Category] 或 [INFO][Category] 格式
+        const bracketMatch = first.match(/^\[(?:DEBUG|INFO|WARN|ERROR)\]?\[?([^\]]*)\]?\s*(.*)/);
+        if (bracketMatch) {
+            return { category: bracketMatch[1] || 'App', message: bracketMatch[2] || '' };
+        }
+        const simpleMatch = first.match(/^\[([^\]]+)\]\s*(.*)/);
+        if (simpleMatch) {
+            return { category: simpleMatch[1], message: simpleMatch[2] || '' };
+        }
+        return { category: 'App', message: first };
+    };
+    const buildEntry = (args, level) => {
+        const { category, message } = parseConsoleCategory(args);
+        const rest = args.slice(1);
+        // data: 后续参数（对象/数组保留结构，字符串拼接）
+        let data = undefined;
+        if (rest.length === 1) {
+            data = rest[0];
+        }
+        else if (rest.length > 1) {
+            // 如果全是字符串，拼成一个；否则保持数组
+            const allStrings = rest.every(r => typeof r === 'string');
+            data = allStrings ? rest.join(' ') : rest;
+        }
+        return { timestamp: new Date().toISOString(), level, category, message, data };
+    };
+    console.log = (...args) => {
+        originalLog.apply(console, args);
+        if (exports.proxyLogger._isWriting)
+            return;
+        exports.proxyLogStore.add(buildEntry(args, 'INFO'));
+    };
+    console.warn = (...args) => {
+        originalWarn.apply(console, args);
+        if (exports.proxyLogger._isWriting)
+            return;
+        exports.proxyLogStore.add(buildEntry(args, 'WARN'));
+    };
+    console.error = (...args) => {
+        originalError.apply(console, args);
+        if (exports.proxyLogger._isWriting)
+            return;
+        exports.proxyLogStore.add(buildEntry(args, 'ERROR'));
+    };
+}

package/out-server/main/proxy/modelCatalog.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KIRO_PROXY_MODEL_PRESETS = exports.KIRO_PROXY_PREFERRED_MODEL_IDS = void 0;
+exports.normalizeKiroModelIdForCompare = normalizeKiroModelIdForCompare;
+exports.isAutoKiroModelId = isAutoKiroModelId;
+exports.KIRO_PROXY_PREFERRED_MODEL_IDS = [
+    'claude-sonnet-4.5',
+    'claude-sonnet-4',
+    'claude-opus-4.8',
+    'claude-opus-4.7',
+    'claude-opus-4.5',
+    'claude-haiku-4.5'
+];
+exports.KIRO_PROXY_MODEL_PRESETS = [
+    {
+        id: 'claude-sonnet-4.5',
+        name: 'Claude Sonnet 4.5',
+        description: 'Claude Sonnet 4.5 for Kiro coding and agent tasks',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    },
+    {
+        id: 'claude-sonnet-4',
+        name: 'Claude Sonnet 4',
+        description: 'Claude Sonnet 4 for Kiro coding and reasoning',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    },
+    {
+        id: 'claude-opus-4.8',
+        name: 'Claude Opus 4.8',
+        description: 'Claude Opus 4.8 for Kiro Power accounts',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    },
+    {
+        id: 'claude-opus-4.7',
+        name: 'Claude Opus 4.7',
+        description: 'Claude Opus 4.7 for Kiro Power accounts',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    },
+    {
+        id: 'claude-opus-4.5',
+        name: 'Claude Opus 4.5',
+        description: 'Claude Opus 4.5 for Kiro Power accounts',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    },
+    {
+        id: 'claude-haiku-4.5',
+        name: 'Claude Haiku 4.5',
+        description: 'Claude Haiku 4.5 for fast Kiro tasks',
+        inputTypes: ['TEXT', 'IMAGE'],
+        maxInputTokens: 200000,
+        maxOutputTokens: 64000,
+        modelProvider: 'anthropic'
+    }
+];
+function normalizeKiroModelIdForCompare(id) {
+    return id.trim().toLowerCase().replace(/^(claude-(?:sonnet|haiku|opus)-\d+)-(\d+)(.*)$/u, '$1.$2$3');
+}
+function isAutoKiroModelId(id) {
+    return normalizeKiroModelIdForCompare(id) === 'auto';
+}