axon-code 2.4.0

package/dist/chrome-mcp/socket-client.js

@@ -0,0 +1,325 @@
+/**
+ * Socket Client - 运行在 MCP 服务器进程中
+ *
+ * 架构：
+ * MCP Server (包含此 Socket Client) → Socket → Native Host → Native Messaging → Chrome 扩展
+ */
+import { connect } from 'net';
+import { EventEmitter } from 'events';
+import { getSocketPath } from './native-host.js';
+const MAX_MESSAGE_SIZE = 1048576; // 1MB
+const CONNECT_TIMEOUT = 5000; // 5 秒
+const RECONNECT_DELAY = 1000; // 1 秒初始延迟
+const MAX_RECONNECT_ATTEMPTS = 10; // 最大重连次数
+const MAX_RECONNECT_DELAY = 30000; // 最大重连延迟 30 秒
+/**
+ * Socket 连接错误
+ */
+export class SocketConnectionError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'SocketConnectionError';
+    }
+}
+/**
+ * Socket Client - 连接到 Native Host Socket Server
+ */
+export class SocketClient extends EventEmitter {
+    socket = null;
+    buffer = Buffer.alloc(0);
+    connected = false;
+    connecting = false;
+    pendingToolCalls = new Map();
+    callId = 0;
+    notificationHandler = null;
+    reconnectAttempts = 0;
+    reconnectTimer = null;
+    logger;
+    constructor(options = {}) {
+        super();
+        this.logger = options.logger || {
+            info: console.log,
+            warn: console.warn,
+            error: console.error
+        };
+    }
+    /**
+     * 设置通知处理器
+     */
+    setNotificationHandler(handler) {
+        this.notificationHandler = handler;
+    }
+    /**
+     * 确保已连接
+     */
+    async ensureConnected() {
+        if (this.connected)
+            return true;
+        if (this.connecting) {
+            // 等待连接完成
+            return new Promise((resolve) => {
+                const checkConnection = () => {
+                    if (this.connected) {
+                        resolve(true);
+                    }
+                    else if (!this.connecting) {
+                        resolve(false);
+                    }
+                    else {
+                        setTimeout(checkConnection, 100);
+                    }
+                };
+                checkConnection();
+            });
+        }
+        try {
+            await this.connect();
+            return this.connected;
+        }
+        catch (err) {
+            this.logger.warn('Failed to connect to socket:', err);
+            return false;
+        }
+    }
+    /**
+     * 连接到 Socket Server
+     */
+    async connect() {
+        if (this.connected || this.connecting)
+            return;
+        this.connecting = true;
+        const socketPath = getSocketPath();
+        return new Promise((resolve, reject) => {
+            const timeoutId = setTimeout(() => {
+                this.connecting = false;
+                reject(new SocketConnectionError('Connection timeout'));
+            }, CONNECT_TIMEOUT);
+            this.socket = connect(socketPath, () => {
+                clearTimeout(timeoutId);
+                this.connected = true;
+                this.connecting = false;
+                this.reconnectAttempts = 0; // 重置重连计数
+                this.logger.info('Connected to socket server');
+                resolve();
+            });
+            this.socket.on('data', (data) => {
+                this.handleData(data);
+            });
+            this.socket.on('close', () => {
+                this.connected = false;
+                this.connecting = false;
+                this.logger.info('Disconnected from socket server');
+                // 拒绝所有等待中的调用
+                for (const [, pending] of this.pendingToolCalls) {
+                    pending.reject(new SocketConnectionError('Connection closed'));
+                }
+                this.pendingToolCalls.clear();
+                // 自动重连
+                this.scheduleReconnect();
+            });
+            this.socket.on('error', (err) => {
+                clearTimeout(timeoutId);
+                this.connected = false;
+                this.connecting = false;
+                this.logger.error('Socket error:', err);
+                // 对于连接被拒绝/重置/管道断开的错误，尝试重连
+                if (err.code && ['ECONNREFUSED', 'ECONNRESET', 'EPIPE', 'ENOENT'].includes(err.code)) {
+                    this.scheduleReconnect();
+                }
+                reject(new SocketConnectionError(err.message));
+            });
+        });
+    }
+    /**
+     * 调度自动重连
+     */
+    scheduleReconnect() {
+        // 如果已有重连计划，跳过
+        if (this.reconnectTimer) {
+            this.logger.info('Reconnect already scheduled, skipping');
+            return;
+        }
+        // 如果已达到最大重连次数，放弃
+        if (this.reconnectAttempts >= MAX_RECONNECT_ATTEMPTS) {
+            this.logger.info('Max reconnection attempts reached');
+            this.cleanup();
+            return;
+        }
+        this.reconnectAttempts++;
+        // 指数退避延迟，最大 30 秒
+        const delay = Math.min(RECONNECT_DELAY * Math.pow(1.5, this.reconnectAttempts - 1), MAX_RECONNECT_DELAY);
+        this.logger.info(`Reconnecting in ${Math.round(delay)}ms (attempt ${this.reconnectAttempts}/${MAX_RECONNECT_ATTEMPTS})`);
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectTimer = null;
+            this.connect().catch((err) => {
+                this.logger.error('Reconnect failed:', err);
+            });
+        }, delay);
+    }
+    /**
+     * 清理资源
+     */
+    cleanup() {
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+        this.closeSocket();
+    }
+    /**
+     * 关闭 socket 连接
+     */
+    closeSocket() {
+        if (this.socket) {
+            this.socket.removeAllListeners();
+            this.socket.end();
+            this.socket.destroy();
+            this.socket = null;
+        }
+        this.connected = false;
+        this.connecting = false;
+    }
+    /**
+     * 处理接收到的数据
+     */
+    handleData(data) {
+        this.buffer = Buffer.concat([this.buffer, data]);
+        while (this.buffer.length >= 4) {
+            const messageLength = this.buffer.readUInt32LE(0);
+            if (messageLength === 0 || messageLength > MAX_MESSAGE_SIZE) {
+                this.logger.error(`Invalid message length: ${messageLength}`);
+                this.buffer = Buffer.alloc(0);
+                return;
+            }
+            if (this.buffer.length < 4 + messageLength) {
+                // 消息不完整
+                return;
+            }
+            const messageData = this.buffer.subarray(4, 4 + messageLength);
+            this.buffer = this.buffer.subarray(4 + messageLength);
+            try {
+                const message = JSON.parse(messageData.toString('utf-8'));
+                this.handleMessage(message);
+            }
+            catch (err) {
+                this.logger.error('Failed to parse message:', err);
+            }
+        }
+    }
+    /**
+     * 处理消息
+     */
+    handleMessage(message) {
+        if (!message || typeof message !== 'object')
+            return;
+        const msg = message;
+        this.logger.info(`Received message: ${JSON.stringify(msg).substring(0, 300)}`);
+        // 检查是否是工具调用响应（包含 result 或 error 字段）
+        // 官方格式不包含 callId，所以我们只检查 result/error
+        if ('result' in msg || 'error' in msg) {
+            this.logger.info('Received tool response');
+            // 由于官方代码不使用 callId，我们处理第一个等待中的请求
+            const firstPending = this.pendingToolCalls.entries().next().value;
+            if (firstPending) {
+                const [callId, pending] = firstPending;
+                this.pendingToolCalls.delete(callId);
+                pending.resolve(msg);
+            }
+            return;
+        }
+        // 兼容旧格式：检查是否有 callId
+        if ('callId' in msg && typeof msg.callId === 'string') {
+            const pending = this.pendingToolCalls.get(msg.callId);
+            if (pending) {
+                this.pendingToolCalls.delete(msg.callId);
+                pending.resolve(msg);
+            }
+            return;
+        }
+        // 检查是否是通知
+        if ('method' in msg && this.notificationHandler) {
+            this.notificationHandler(msg);
+            return;
+        }
+        this.logger.info('Received unknown message type');
+    }
+    /**
+     * 发送消息到 Socket Server
+     */
+    send(message) {
+        if (!this.socket || !this.connected) {
+            throw new SocketConnectionError('Not connected');
+        }
+        const json = Buffer.from(JSON.stringify(message), 'utf-8');
+        const header = Buffer.alloc(4);
+        header.writeUInt32LE(json.length, 0);
+        this.socket.write(header);
+        this.socket.write(json);
+    }
+    /**
+     * 调用工具
+     */
+    async callTool(toolName, args) {
+        if (!this.connected) {
+            throw new SocketConnectionError('Not connected');
+        }
+        const callId = `call_${++this.callId}_${Date.now()}`;
+        return new Promise((resolve, reject) => {
+            // 设置超时
+            const timeoutId = setTimeout(() => {
+                this.pendingToolCalls.delete(callId);
+                reject(new SocketConnectionError('Tool call timeout'));
+            }, 60000); // 60 秒超时
+            this.pendingToolCalls.set(callId, {
+                resolve: (result) => {
+                    clearTimeout(timeoutId);
+                    resolve(result);
+                },
+                reject: (error) => {
+                    clearTimeout(timeoutId);
+                    reject(error);
+                }
+            });
+            try {
+                // Chrome 扩展期望的消息格式:
+                // { type: "tool_request", method: "execute_tool", params: { tool, client_id, args } }
+                const message = {
+                    type: 'tool_request',
+                    method: 'execute_tool',
+                    params: {
+                        tool: toolName,
+                        client_id: 'axon',
+                        args: args
+                    }
+                };
+                this.logger.info(`Sending tool call: ${JSON.stringify(message).substring(0, 200)}`);
+                this.send(message);
+                this.logger.info('Tool call sent successfully');
+            }
+            catch (err) {
+                this.pendingToolCalls.delete(callId);
+                clearTimeout(timeoutId);
+                reject(err);
+            }
+        });
+    }
+    /**
+     * 断开连接
+     */
+    disconnect() {
+        this.cleanup();
+    }
+    /**
+     * 检查是否已连接
+     */
+    isConnected() {
+        return this.connected;
+    }
+}
+/**
+ * 创建 Socket Client 实例
+ */
+export function createSocketClient(options) {
+    return new SocketClient(options);
+}
+//# sourceMappingURL=socket-client.js.map

package/dist/chrome-mcp/socket-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-client.js","sourceRoot":"","sources":["../../src/chrome-mcp/socket-client.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAU,OAAO,EAAE,MAAM,KAAK,CAAC;AACtC,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,MAAM,gBAAgB,GAAG,OAAO,CAAC,CAAC,MAAM;AACxC,MAAM,eAAe,GAAG,IAAI,CAAC,CAAC,MAAM;AACpC,MAAM,eAAe,GAAG,IAAI,CAAC,CAAC,UAAU;AACxC,MAAM,sBAAsB,GAAG,EAAE,CAAC,CAAC,SAAS;AAC5C,MAAM,mBAAmB,GAAG,KAAK,CAAC,CAAC,cAAc;AAEjD;;GAEG;AACH,MAAM,OAAO,qBAAsB,SAAQ,KAAK;IAC9C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,uBAAuB,CAAC;IACtC,CAAC;CACF;AAcD;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,YAAY;IACpC,MAAM,GAAkB,IAAI,CAAC;IAC7B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzB,SAAS,GAAG,KAAK,CAAC;IAClB,UAAU,GAAG,KAAK,CAAC;IACnB,gBAAgB,GAGnB,IAAI,GAAG,EAAE,CAAC;IACP,MAAM,GAAG,CAAC,CAAC;IACX,mBAAmB,GAA6C,IAAI,CAAC;IACrE,iBAAiB,GAAG,CAAC,CAAC;IACtB,cAAc,GAAyC,IAAI,CAAC;IAC5D,MAAM,CAIZ;IAEF,YAAY,UAMR,EAAE;QACJ,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI;YAC9B,IAAI,EAAE,OAAO,CAAC,GAAG;YACjB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,sBAAsB,CAAC,OAAwC;QAC7D,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QACnB,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC;QAChC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,SAAS;YACT,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC7B,MAAM,eAAe,GAAG,GAAG,EAAE;oBAC3B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;wBACnB,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChB,CAAC;yBAAM,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;wBAC5B,OAAO,CAAC,KAAK,CAAC,CAAC;oBACjB,CAAC;yBAAM,CAAC;wBACN,UAAU,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;oBACnC,CAAC;gBACH,CAAC,CAAC;gBACF,eAAe,EAAE,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC,SAAS,CAAC;QACxB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO;QACnB,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAE9C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;QAEnC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,MAAM,CAAC,IAAI,qBAAqB,CAAC,oBAAoB,CAAC,CAAC,CAAC;YAC1D,CAAC,EAAE,eAAe,CAAC,CAAC;YAEpB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,GAAG,EAAE;gBACrC,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;gBACtB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC,CAAC,SAAS;gBACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;gBAC/C,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC9B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAC3B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;gBAEpD,aAAa;gBACb,KAAK,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBAChD,OAAO,CAAC,MAAM,CAAC,IAAI,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,CAAC;gBACjE,CAAC;gBACD,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAE9B,OAAO;gBACP,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;gBACrD,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;gBAExC,0BAA0B;gBAC1B,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBACrF,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,CAAC;gBAED,MAAM,CAAC,IAAI,qBAAqB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACjD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,cAAc;QACd,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,iBAAiB;QACjB,IAAI,IAAI,CAAC,iBAAiB,IAAI,sBAAsB,EAAE,CAAC;YACrD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;YACtD,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,iBAAiB;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CACpB,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC,EAC3D,mBAAmB,CACpB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,iBAAiB,IAAI,sBAAsB,GAAG,CAAC,CAAC;QAEzH,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC;IAED;;OAEG;IACK,OAAO;QACb,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED;;OAEG;IACK,WAAW;QACjB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,IAAY;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;QAEjD,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAElD,IAAI,aAAa,KAAK,CAAC,IAAI,aAAa,GAAG,gBAAgB,EAAE,CAAC;gBAC5D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,aAAa,EAAE,CAAC,CAAC;gBAC9D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC9B,OAAO;YACT,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,aAAa,EAAE,CAAC;gBAC3C,QAAQ;gBACR,OAAO;YACT,CAAC;YAED,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC;YAEtD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC1D,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,OAAgB;QACpC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO;QAEpD,MAAM,GAAG,GAAG,OAAkC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAE/E,oCAAoC;QACpC,sCAAsC;QACtC,IAAI,QAAQ,IAAI,GAAG,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC3C,iCAAiC;YACjC,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YAClE,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,YAAY,CAAC;gBACvC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrC,OAAO,CAAC,OAAO,CAAC,GAAqB,CAAC,CAAC;YACzC,CAAC;YACD,OAAO;QACT,CAAC;QAED,qBAAqB;QACrB,IAAI,QAAQ,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACzC,OAAO,CAAC,OAAO,CAAC,GAAqB,CAAC,CAAC;YACzC,CAAC;YACD,OAAO;QACT,CAAC;QAED,UAAU;QACV,IAAI,QAAQ,IAAI,GAAG,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAChD,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,OAAe;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,IAAI,qBAAqB,CAAC,eAAe,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAErC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAA6B;QAC5D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,qBAAqB,CAAC,eAAe,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QAErD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,OAAO;YACP,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrC,MAAM,CAAC,IAAI,qBAAqB,CAAC,mBAAmB,CAAC,CAAC,CAAC;YACzD,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,SAAS;YAEpB,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,EAAE;gBAChC,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE;oBAClB,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC;gBACD,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;oBAChB,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,oBAAoB;gBACpB,sFAAsF;gBACtF,MAAM,OAAO,GAAG;oBACd,IAAI,EAAE,cAAc;oBACpB,MAAM,EAAE,cAAc;oBACtB,MAAM,EAAE;wBACN,IAAI,EAAE,QAAQ;wBACd,SAAS,EAAE,MAAM;wBACjB,IAAI,EAAE,IAAI;qBACX;iBACF,CAAC;gBACF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpF,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAClD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrC,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAMlC;IACC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC"}

package/dist/chrome-mcp/socket-server.d.ts

@@ -0,0 +1,65 @@
+/**
+ * Socket Server - 运行在 Native Host 进程中
+ *
+ * 架构：
+ * Chrome 扩展 → Native Messaging → Native Host (包含此 Socket Server) ← Socket ← MCP Client
+ */
+/**
+ * Socket Server - 管理与 MCP 客户端的连接
+ */
+export declare class SocketServer {
+    private mcpClients;
+    private nextClientId;
+    private server;
+    private running;
+    /**
+     * 启动 Socket 服务器
+     */
+    start(): Promise<void>;
+    /**
+     * 停止 Socket 服务器
+     */
+    stop(): Promise<void>;
+    /**
+     * 处理来自 Chrome 扩展的消息
+     */
+    handleChromeMessage(message: string): Promise<void>;
+    /**
+     * 转发消息到所有 MCP 客户端
+     */
+    private forwardToMcpClients;
+    /**
+     * 处理 MCP 客户端连接
+     */
+    private handleMcpClient;
+    /**
+     * 处理 MCP 客户端缓冲区中的消息
+     */
+    private processMcpClientBuffer;
+    /**
+     * 处理来自 MCP 客户端的消息
+     */
+    private handleMcpClientMessage;
+}
+/**
+ * Native Message Reader - 从 stdin 读取 Native Messaging 消息
+ */
+export declare class NativeMessageReader {
+    private buffer;
+    private pendingResolve;
+    private closed;
+    constructor();
+    /**
+     * 读取下一条消息
+     */
+    read(): Promise<string | null>;
+    /**
+     * 尝试处理缓冲区中的消息
+     */
+    private tryProcessMessage;
+}
+/**
+ * 运行 Native Host 主循环
+ */
+export declare function runNativeHost(): Promise<void>;
+//# sourceMappingURL=socket-server.d.ts.map

package/dist/chrome-mcp/socket-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-server.d.ts","sourceRoot":"","sources":["../../src/chrome-mcp/socket-server.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAkDH;;GAEG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,UAAU,CAAyC;IAC3D,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,OAAO,CAAS;IAExB;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0C5B;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA6B3B;;OAEG;IACG,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkDzD;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAoB3B;;OAEG;IACH,OAAO,CAAC,eAAe;IA8BvB;;OAEG;IACH,OAAO,CAAC,sBAAsB;IA6B9B;;OAEG;IACH,OAAO,CAAC,sBAAsB;CAO/B;AAED;;GAEG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,cAAc,CAAiD;IACvE,OAAO,CAAC,MAAM,CAAS;;IAyBvB;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IASpC;;OAEG;IACH,OAAO,CAAC,iBAAiB;CAuB1B;AAED;;GAEG;AACH,wBAAsB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CA2CnD"}