aicodeswitch 2.0.10 → 2.0.11

package/dist/server/websocket-service.js

@@ -0,0 +1,148 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolInstallationWS = void 0;
+exports.createToolInstallationWSServer = createToolInstallationWSServer;
+// @ts-ignore - ws 类型声明可能需要手动安装 @types/ws
+const ws_1 = require("ws");
+const tools_service_1 = require("./tools-service");
+const os_1 = __importDefault(require("os"));
+/**
+ * WebSocket 连接管理
+ */
+class ToolInstallationWS {
+    constructor(ws, req) {
+        Object.defineProperty(this, "ws", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "childProcess", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        this.ws = ws;
+        console.log(`[WS] 新的 WebSocket 连接: ${req.socket.remoteAddress}`);
+        this.ws.on('message', (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                console.log(`[WS] 收到消息:`, message.type);
+                if (message.type === 'input') {
+                    // 用户输入（如密码），发送到子进程的 stdin
+                    if (this.childProcess && this.childProcess.stdin) {
+                        console.log(`[WS] 发送用户输入到子进程:`, message.data.slice(0, 10));
+                        this.childProcess.stdin.write(message.data);
+                    }
+                    else {
+                        console.warn(`[WS] 子进程不存在或无 stdin，无法发送输入`);
+                    }
+                }
+            }
+            catch (err) {
+                console.error(`[WS] 解析消息失败:`, err);
+            }
+        });
+        this.ws.on('close', () => {
+            console.log(`[WS] WebSocket 连接关闭`);
+            this.cleanup();
+        });
+        this.ws.on('error', (err) => {
+            console.error(`[WS] WebSocket 错误:`, err);
+            this.cleanup();
+        });
+    }
+    /**
+     * 开始安装工具
+     */
+    startInstallation(toolName) {
+        console.log(`[WS] 开始安装 ${toolName}`);
+        const callbacks = {
+            onData: (data) => {
+                this.sendMessage({ type: 'stdout', data });
+            },
+            onError: (data) => {
+                this.sendMessage({ type: 'stderr', data });
+            },
+            onClose: (code) => {
+                const success = code === 0;
+                this.sendMessage({ type: 'close', data: { code, success } });
+                // 延迟关闭连接，让客户端收到最终消息
+                setTimeout(() => {
+                    this.ws.close();
+                }, 1000);
+            },
+        };
+        // 启动安装进程
+        this.childProcess = (0, tools_service_1.installTool)(toolName, callbacks);
+        // 发送启动信息
+        const platform = os_1.default.platform();
+        const command = platform === 'win32'
+            ? `npm install -g ${toolName === 'claude-code' ? '@anthropic-ai/claude-code' : '@openai/codex'}`
+            : `sudo npm install -g ${toolName === 'claude-code' ? '@anthropic-ai/claude-code' : '@openai/codex'}`;
+        this.sendMessage({
+            type: 'start',
+            data: {
+                tool: toolName,
+                os: platform,
+                command,
+                pid: this.childProcess.pid || 0,
+            },
+        });
+    }
+    /**
+     * 发送消息到客户端
+     */
+    sendMessage(message) {
+        if (this.ws.readyState === ws_1.WebSocket.OPEN) {
+            this.ws.send(JSON.stringify(message));
+        }
+    }
+    /**
+     * 清理资源
+     */
+    cleanup() {
+        if (this.childProcess) {
+            console.log(`[WS] 清理资源，终止子进程`);
+            // 给进程5秒时间正常退出
+            setTimeout(() => {
+                if (this.childProcess && !this.childProcess.killed) {
+                    this.childProcess.kill('SIGTERM');
+                }
+            }, 5000);
+        }
+    }
+}
+exports.ToolInstallationWS = ToolInstallationWS;
+/**
+ * 创建 WebSocket 服务器
+ */
+function createToolInstallationWSServer() {
+    const wss = new ws_1.WebSocketServer({ noServer: true });
+    wss.on('connection', (ws, req) => {
+        const wsHandler = new ToolInstallationWS(ws, req);
+        // 监听第一次消息，获取要安装的工具名称
+        ws.once('message', (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                if (message.type === 'install' && message.tool) {
+                    wsHandler.startInstallation(message.tool);
+                }
+                else {
+                    ws.send(JSON.stringify({ type: 'error', data: '无效的请求' }));
+                    ws.close();
+                }
+            }
+            catch (error) {
+                console.error(`[WS] 解析安装请求失败:`, error);
+                ws.send(JSON.stringify({ type: 'error', data: '解析请求失败' }));
+                ws.close();
+            }
+        });
+    });
+    return wss;
+}