aicodeswitch 5.1.2 → 5.2.0

package/dist/server/rules-status-service.js

@@ -10,12 +10,14 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.rulesStatusBroadcaster = exports.RulesStatusBroadcaster = void 0;
+const events_1 = require("events");
 /**
  * 规则状态管理服务
  * 负责管理所有规则的状态（使用中、空闲、错误、挂起）
  */
-class RulesStatusBroadcaster {
+class RulesStatusBroadcaster extends events_1.EventEmitter {
     constructor() {
+        super();
         Object.defineProperty(this, "ruleStates", {
             enumerable: true,
             configurable: true,
@@ -70,6 +72,8 @@ class RulesStatusBroadcaster {
             writable: true,
             value: 30000
         }); // error 状态30秒后自动恢复
+        // 允许多标签页场景下的多连接监听
+        this.setMaxListeners(50);
         // 启动定期状态检查定时器
         this.startSyncInterval();
     }
@@ -99,7 +103,7 @@ class RulesStatusBroadcaster {
                     data.timestamp &&
                     now - data.timestamp > this.ERROR_RECOVERY_TIMEOUT) {
                     console.log(`[RulesStatusBroadcaster] 规则 ${ruleId} 错误状态已超时，自动恢复为 idle 状态`);
-                    this.ruleStates.set(ruleId, {
+                    this.updateRuleStatus({
                         ruleId,
                         status: 'idle',
                         routeId: data.routeId,
@@ -130,7 +134,7 @@ class RulesStatusBroadcaster {
                         if (!isBlacklisted) {
                             // 黑名单已过期，恢复为 idle 状态
                             console.log(`[RulesStatusBroadcaster] 规则 ${ruleId} 黑名单已过期，恢复为 idle 状态`);
-                            this.ruleStates.set(ruleId, {
+                            this.updateRuleStatus({
                                 ruleId,
                                 status: 'idle',
                                 routeId,
@@ -172,6 +176,8 @@ class RulesStatusBroadcaster {
     updateRuleStatus(data) {
         // 更新本地状态
         this.ruleStates.set(data.ruleId, data);
+        // 通知 SSE 客户端状态已变更
+        this.emit('statusChanged', data);
     }
     /**
      * 标记规则正在使用
@@ -251,6 +257,22 @@ class RulesStatusBroadcaster {
             timestamp: Date.now(),
         });
     }
+    /**
+     * 刷新规则使用中的不活动定时器（轻量级，仅重置定时器，不修改状态）
+     * 用于 streaming 过程中持续保持 in_use 状态
+     */
+    refreshRuleInUse(routeId, ruleId) {
+        const currentStatus = this.ruleStates.get(ruleId);
+        // 仅当状态已经是 in_use 时才刷新定时器
+        if ((currentStatus === null || currentStatus === void 0 ? void 0 : currentStatus.status) !== 'in_use')
+            return;
+        const timeoutKey = `${routeId}:${ruleId}`;
+        this.clearRuleTimeout(timeoutKey);
+        const timeout = setTimeout(() => {
+            this.markRuleIdle(routeId, ruleId);
+        }, this.INACTIVITY_TIMEOUT);
+        this.ruleTimeouts.set(timeoutKey, timeout);
+    }
     /**
      * 更新规则使用量
      */
@@ -272,6 +294,12 @@ class RulesStatusBroadcaster {
      */
     clearRuleStatus(ruleId) {
         this.ruleStates.delete(ruleId);
+        // 通知 SSE 客户端状态已清除（恢复为 idle）
+        this.emit('statusChanged', {
+            ruleId,
+            status: 'idle',
+            timestamp: Date.now(),
+        });
     }
     /**
      * 获取当前活动的规则列表
@@ -304,6 +332,7 @@ class RulesStatusBroadcaster {
         this.idleDebounceTimeouts.forEach((timeout) => clearTimeout(timeout));
         this.idleDebounceTimeouts.clear();
         this.ruleStates.clear();
+        this.removeAllListeners();
     }
 }
 exports.RulesStatusBroadcaster = RulesStatusBroadcaster;

package/dist/server/session-launcher.js

@@ -0,0 +1,282 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveProjectDir = resolveProjectDir;
+exports.writePromptToTempFile = writePromptToTempFile;
+exports.cleanupTempFile = cleanupTempFile;
+exports.cleanupOldTempFiles = cleanupOldTempFiles;
+exports.launchTargetTool = launchTargetTool;
+exports.launchTargetWithFallback = launchTargetWithFallback;
+const child_process_1 = require("child_process");
+const fs_1 = require("fs");
+const path_1 = require("path");
+const os_1 = require("os");
+function which(cmd) {
+    try {
+        const command = process.platform === 'win32' ? `where ${cmd}` : `which ${cmd}`;
+        (0, child_process_1.execSync)(command, { stdio: 'ignore' });
+        return true;
+    }
+    catch (_a) {
+        return false;
+    }
+}
+function isToolInstalled(tool) {
+    if (tool === 'claude-code')
+        return which('claude');
+    if (tool === 'codex')
+        return which('codex');
+    return false;
+}
+function getToolCli(tool) {
+    return tool === 'claude-code' ? 'claude' : 'codex';
+}
+// ─── 项目目录自动推断 ───
+/**
+ * 从 Claude Code session 文件中查找 cwd
+ * Claude Code 在 ~/.claude/sessions/<PID>.json 中记录 sessionId 和 cwd
+ */
+function resolveProjectDirFromClaudeSessions(sessionId) {
+    try {
+        const sessionsDir = (0, path_1.join)((0, os_1.homedir)(), '.claude', 'sessions');
+        if (!(0, fs_1.existsSync)(sessionsDir))
+            return null;
+        const files = (0, fs_1.readdirSync)(sessionsDir).filter(f => f.endsWith('.json'));
+        for (const file of files) {
+            try {
+                const content = (0, fs_1.readFileSync)((0, path_1.join)(sessionsDir, file), 'utf-8');
+                const meta = JSON.parse(content);
+                if (meta.sessionId === sessionId && meta.cwd) {
+                    return meta.cwd;
+                }
+            }
+            catch ( /* skip unreadable files */_a) { /* skip unreadable files */ }
+        }
+    }
+    catch ( /* ignore */_b) { /* ignore */ }
+    return null;
+}
+/**
+ * 从 Codex session 文件中查找 cwd
+ * Codex 在 ~/.codex/sessions/ 目录下存储 JSONL 格式的会话文件
+ */
+function resolveProjectDirFromCodexSessions(sessionId) {
+    try {
+        const sessionsDir = (0, path_1.join)((0, os_1.homedir)(), '.codex', 'sessions');
+        if (!(0, fs_1.existsSync)(sessionsDir))
+            return null;
+        const files = (0, fs_1.readdirSync)(sessionsDir);
+        for (const file of files) {
+            // Codex session 文件名格式: <uuid>.jsonl 或在子目录中
+            const filePath = (0, path_1.join)(sessionsDir, file);
+            try {
+                if ((0, fs_1.existsSync)(filePath) && !filePath.endsWith('.jsonl'))
+                    continue;
+                const content = (0, fs_1.readFileSync)(filePath, 'utf-8');
+                const lines = content.trim().split('\n');
+                for (const line of lines) {
+                    try {
+                        const entry = JSON.parse(line);
+                        // Codex session 文件中可能包含 cwd 字段
+                        if (entry.sessionId === sessionId && entry.cwd)
+                            return entry.cwd;
+                        if (entry.cwd && line.includes(sessionId))
+                            return entry.cwd;
+                    }
+                    catch ( /* skip unparseable lines */_a) { /* skip unparseable lines */ }
+                }
+            }
+            catch ( /* skip unreadable files */_b) { /* skip unreadable files */ }
+        }
+    }
+    catch ( /* ignore */_c) { /* ignore */ }
+    return null;
+}
+/**
+ * 自动推断会话对应的项目目录
+ */
+function resolveProjectDir(sessionId, _sourceTool) {
+    // 优先从 Claude Code sessions 查找
+    let cwd = resolveProjectDirFromClaudeSessions(sessionId);
+    if (cwd)
+        return cwd;
+    // 然后从 Codex sessions 查找
+    cwd = resolveProjectDirFromCodexSessions(sessionId);
+    if (cwd)
+        return cwd;
+    return null;
+}
+// ─── 临时文件管理 ───
+function writePromptToTempFile(prompt, sessionId) {
+    const fileName = `aicodeswitch-migration-${sessionId}.txt`;
+    const filePath = (0, path_1.join)((0, os_1.tmpdir)(), fileName);
+    (0, fs_1.writeFileSync)(filePath, prompt, 'utf-8');
+    return filePath;
+}
+function cleanupTempFile(filePath) {
+    setTimeout(() => {
+        try {
+            (0, fs_1.unlinkSync)(filePath);
+        }
+        catch ( /* already cleaned up */_a) { /* already cleaned up */ }
+    }, 30000);
+}
+function cleanupOldTempFiles() {
+    try {
+        const dir = (0, os_1.tmpdir)();
+        const files = (0, fs_1.readdirSync)(dir);
+        const now = Date.now();
+        for (const file of files) {
+            if (file.startsWith('aicodeswitch-migration-') && file.endsWith('.txt')) {
+                const fullPath = (0, path_1.join)(dir, file);
+                try {
+                    const { statSync } = require('fs');
+                    const stat = statSync(fullPath);
+                    if (now - stat.mtimeMs > 3600000) {
+                        (0, fs_1.unlinkSync)(fullPath);
+                    }
+                }
+                catch ( /* ignore per-file errors */_a) { /* ignore per-file errors */ }
+            }
+        }
+    }
+    catch ( /* ignore */_b) { /* ignore */ }
+}
+// ─── 命令构建 ───
+function buildCommand(tool, promptFilePath, projectDir) {
+    const cli = getToolCli(tool);
+    const cdPrefix = projectDir ? `cd "${projectDir}" && ` : '';
+    const readCmd = process.platform === 'win32' ? `type "${promptFilePath}"` : `cat "${promptFilePath}"`;
+    return `${cdPrefix}${readCmd} | ${cli}`;
+}
+// ─── 终端启动 ───
+function launchViaOSAScript(command) {
+    return __awaiter(this, void 0, void 0, function* () {
+        return new Promise((resolve) => {
+            const escaped = command.replace(/"/g, '\\"');
+            const script = `tell app "Terminal" to do script "${escaped}"`;
+            const child = (0, child_process_1.spawn)('osascript', ['-e', script], {
+                stdio: 'ignore',
+                detached: true,
+            });
+            child.on('error', () => {
+                // Try iTerm2 as fallback
+                const itermScript = `tell app "iTerm" to tell current window to set newTab to (create tab with default profile) then write session 1 of newTab text "${escaped}"`;
+                const child2 = (0, child_process_1.spawn)('osascript', ['-e', itermScript], {
+                    stdio: 'ignore',
+                    detached: true,
+                });
+                child2.on('error', () => resolve(null));
+                child2.on('spawn', () => resolve(child2.pid || 0));
+            });
+            child.on('spawn', () => resolve(child.pid || 0));
+        });
+    });
+}
+function launchViaTerminal(command) {
+    return __awaiter(this, void 0, void 0, function* () {
+        return new Promise((resolve) => {
+            const child = (0, child_process_1.spawn)('gnome-terminal', ['--', 'bash', '-c', `${command}; exec bash`], {
+                stdio: 'ignore',
+                detached: true,
+            });
+            child.on('error', () => {
+                const child2 = (0, child_process_1.spawn)('xterm', ['-e', `${command}; exec bash`], {
+                    stdio: 'ignore',
+                    detached: true,
+                });
+                child2.on('error', () => resolve(null));
+                child2.on('spawn', () => resolve(child2.pid || 0));
+            });
+            child.on('spawn', () => resolve(child.pid || 0));
+        });
+    });
+}
+// ─── 回退结果 ───
+function createFallbackResult(tool, command, promptFilePath) {
+    const toolName = tool === 'claude-code' ? 'Claude Code' : 'Codex';
+    return {
+        success: false,
+        method: 'fallback',
+        reason: `无法自动启动 ${toolName}`,
+        command,
+        promptFilePath,
+        fallbackSuggestions: [
+            `请在终端中执行: ${command}`,
+            '或复制下方 Prompt 内容，在新会话中粘贴',
+        ],
+    };
+}
+// ─── 主启动逻辑 ───
+function launchTargetTool(tool, promptFilePath, projectDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const command = buildCommand(tool, promptFilePath, projectDir);
+        try {
+            let pid = null;
+            if (process.platform === 'darwin') {
+                pid = yield launchViaOSAScript(command);
+            }
+            else if (process.platform === 'linux') {
+                pid = yield launchViaTerminal(command);
+            }
+            else {
+                // Windows: open new cmd window
+                const cdPrefix = projectDir ? `cd /d "${projectDir}" && ` : '';
+                const cli = getToolCli(tool);
+                const winCommand = `start cmd /k "${cdPrefix}type "${promptFilePath}" | ${cli}"`;
+                yield new Promise((resolve, reject) => {
+                    const child = (0, child_process_1.spawn)('cmd', ['/c', winCommand], {
+                        stdio: 'ignore',
+                        detached: true,
+                    });
+                    child.on('error', reject);
+                    child.on('spawn', () => resolve());
+                });
+                pid = 0;
+            }
+            if (pid !== null) {
+                return {
+                    success: true,
+                    method: 'cli-launch',
+                    pid: pid || undefined,
+                    command,
+                    promptFilePath,
+                };
+            }
+        }
+        catch (_a) {
+            // Fall through to fallback
+        }
+        return createFallbackResult(tool, command, promptFilePath);
+    });
+}
+function launchTargetWithFallback(tool, promptFilePath, prompt, projectDir) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const installed = isToolInstalled(tool);
+        if (!installed) {
+            const toolName = tool === 'claude-code' ? 'Claude Code' : 'Codex';
+            const command = buildCommand(tool, promptFilePath, projectDir);
+            return {
+                success: false,
+                method: 'fallback',
+                reason: `${toolName} CLI (${getToolCli(tool)}) not found in PATH`,
+                command,
+                prompt,
+                promptFilePath,
+                fallbackSuggestions: [
+                    `请在终端中执行: ${command}`,
+                    '或复制下方 Prompt 内容，在新会话中粘贴',
+                ],
+            };
+        }
+        return launchTargetTool(tool, promptFilePath, projectDir);
+    });
+}