@pyrokine/mcp-chrome 1.0.0 → 1.1.0

package/dist/extension/socket-server.js

@@ -0,0 +1,177 @@
+/**
+ * Unix Socket 服务器
+ *
+ * 负责与 Native Host 通信，Native Host 再通过 Native Messaging 与 Extension 通信
+ */
+import { EventEmitter } from 'events';
+import { createServer } from 'net';
+import { unlinkSync, existsSync } from 'fs';
+import { tmpdir } from 'os';
+import { join } from 'path';
+export class ExtensionSocketServer extends EventEmitter {
+    server = null;
+    clientSocket = null;
+    socketPath;
+    messageHandlers = new Map();
+    buffer = '';
+    constructor(socketName = 'mcp-chrome.sock') {
+        super();
+        // 使用 /tmp 目录，跨平台兼容
+        this.socketPath = join(tmpdir(), socketName);
+    }
+    getSocketPath() {
+        return this.socketPath;
+    }
+    async start() {
+        // 清理可能存在的旧 socket 文件
+        if (existsSync(this.socketPath)) {
+            unlinkSync(this.socketPath);
+        }
+        return new Promise((resolve, reject) => {
+            this.server = createServer((socket) => {
+                console.error('[Socket] Native Host connected');
+                this.clientSocket = socket;
+                this.emit('connected');
+                socket.on('data', (data) => {
+                    this.handleData(data.toString());
+                });
+                socket.on('close', () => {
+                    console.error('[Socket] Native Host disconnected');
+                    this.clientSocket = null;
+                    this.emit('disconnected');
+                });
+                socket.on('error', (error) => {
+                    console.error('[Socket] Error:', error.message);
+                });
+            });
+            this.server.listen(this.socketPath, () => {
+                console.error(`[Socket] Server listening on ${this.socketPath}`);
+                resolve();
+            });
+            this.server.on('error', (error) => {
+                reject(error);
+            });
+        });
+    }
+    async stop() {
+        return new Promise((resolve) => {
+            if (this.clientSocket) {
+                this.clientSocket.destroy();
+                this.clientSocket = null;
+            }
+            if (this.server) {
+                this.server.close(() => {
+                    // 清理 socket 文件
+                    if (existsSync(this.socketPath)) {
+                        try {
+                            unlinkSync(this.socketPath);
+                        }
+                        catch {
+                            // 忽略清理错误
+                        }
+                    }
+                    resolve();
+                });
+                this.server = null;
+            }
+            else {
+                resolve();
+            }
+        });
+    }
+    isConnected() {
+        return this.clientSocket !== null && !this.clientSocket.destroyed;
+    }
+    /**
+     * 发送命令到 Extension（通过 Native Host）
+     * 如果未连接，会等待连接建立
+     */
+    async sendCommand(action, params = {}) {
+        if (!this.isConnected()) {
+            // 等待连接，最多 30 秒（与命令超时一致）
+            // 如果需要更长时间，用户应该先确保 Extension 已连接
+            console.error('[MCP] Extension disconnected, waiting for reconnection (30s timeout)...');
+            const connected = await this.waitForConnection(30000);
+            if (!connected) {
+                throw new Error('Extension 连接超时。请检查：1) Chrome 是否运行 2) Extension 是否已安装 3) 在 Extension popup 中点击连接');
+            }
+            console.error('[MCP] Extension reconnected successfully');
+        }
+        return new Promise((resolve, reject) => {
+            const messageId = this.generateId();
+            const message = {
+                id: messageId,
+                action,
+                params,
+            };
+            const timeout = setTimeout(() => {
+                this.messageHandlers.delete(messageId);
+                reject(new Error(`Request timeout for action: ${action}`));
+            }, 30000);
+            this.messageHandlers.set(messageId, { resolve, reject, timeout });
+            // 发送消息，每条消息以换行符分隔
+            this.clientSocket.write(JSON.stringify(message) + '\n');
+        });
+    }
+    handleData(data) {
+        this.buffer += data;
+        // 按换行符分割消息
+        const lines = this.buffer.split('\n');
+        this.buffer = lines.pop() || ''; // 保留不完整的最后一行
+        for (const line of lines) {
+            if (!line.trim())
+                continue;
+            try {
+                const message = JSON.parse(line);
+                this.handleMessage(message);
+            }
+            catch (error) {
+                console.error('[Socket] Error parsing message:', error);
+            }
+        }
+    }
+    handleMessage(message) {
+        if (message.id && this.messageHandlers.has(message.id)) {
+            const handler = this.messageHandlers.get(message.id);
+            clearTimeout(handler.timeout);
+            this.messageHandlers.delete(message.id);
+            if (message.success) {
+                handler.resolve(message.data);
+            }
+            else {
+                handler.reject(new Error(message.error || 'Unknown error'));
+            }
+        }
+        else {
+            this.emit('message', message);
+        }
+    }
+    /**
+     * 等待连接建立
+     * @param timeout 超时时间（毫秒），0 表示无限等待
+     */
+    waitForConnection(timeout) {
+        return new Promise((resolve) => {
+            if (this.isConnected()) {
+                resolve(true);
+                return;
+            }
+            const startTime = Date.now();
+            const checkInterval = setInterval(() => {
+                if (this.isConnected()) {
+                    clearInterval(checkInterval);
+                    resolve(true);
+                }
+                else if (timeout > 0 && Date.now() - startTime > timeout) {
+                    // 只有 timeout > 0 时才检查超时
+                    clearInterval(checkInterval);
+                    resolve(false);
+                }
+            }, 100);
+        });
+    }
+    generateId() {
+        return Date.now().toString(36) + Math.random().toString(36).substring(2);
+    }
+}
+//# sourceMappingURL=socket-server.js.map

package/dist/extension/socket-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-server.js","sourceRoot":"","sources":["../../src/extension/socket-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAC,YAAY,EAAC,MAAM,QAAQ,CAAA;AACnC,OAAO,EAAC,YAAY,EAAiB,MAAM,KAAK,CAAA;AAChD,OAAO,EAAC,UAAU,EAAE,UAAU,EAAC,MAAM,IAAI,CAAA;AACzC,OAAO,EAAC,MAAM,EAAC,MAAM,IAAI,CAAA;AACzB,OAAO,EAAC,IAAI,EAAC,MAAM,MAAM,CAAA;AAezB,MAAM,OAAO,qBAAsB,SAAQ,YAAY;IAC3C,MAAM,GAAkB,IAAI,CAAA;IAC5B,YAAY,GAAkB,IAAI,CAAA;IAClC,UAAU,CAAQ;IAClB,eAAe,GAAG,IAAI,GAAG,EAI7B,CAAA;IACI,MAAM,GAAG,EAAE,CAAA;IAEnB,YAAY,UAAU,GAAG,iBAAiB;QACtC,KAAK,EAAE,CAAA;QACP,mBAAmB;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAA;IAChD,CAAC;IAED,aAAa;QACT,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK;QACP,qBAAqB;QACrB,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC/B,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,MAAM,EAAE,EAAE;gBAClC,OAAO,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAA;gBAE/C,IAAI,CAAC,YAAY,GAAG,MAAM,CAAA;gBAC1B,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;gBAEtB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;oBACvB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAA;gBACpC,CAAC,CAAC,CAAA;gBAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;oBACpB,OAAO,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAA;oBAClD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;oBACxB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAA;gBAC7B,CAAC,CAAC,CAAA;gBAEF,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACzB,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAA;gBACnD,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE;gBACrC,OAAO,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAA;gBAChE,OAAO,EAAE,CAAA;YACb,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC9B,MAAM,CAAC,KAAK,CAAC,CAAA;YACjB,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,IAAI;QACN,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAA;gBAC3B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;YAC5B,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACd,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;oBACnB,eAAe;oBACf,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC9B,IAAI,CAAC;4BACD,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;wBAC/B,CAAC;wBAAC,MAAM,CAAC;4BACL,SAAS;wBACb,CAAC;oBACL,CAAC;oBACD,OAAO,EAAE,CAAA;gBACb,CAAC,CAAC,CAAA;gBACF,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;YACtB,CAAC;iBAAM,CAAC;gBACJ,OAAO,EAAE,CAAA;YACb,CAAC;QACL,CAAC,CAAC,CAAA;IACN,CAAC;IAED,WAAW;QACP,OAAO,IAAI,CAAC,YAAY,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAA;IACrE,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,SAAkB,EAAE;QAClD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YACtB,wBAAwB;YACxB,iCAAiC;YACjC,OAAO,CAAC,KAAK,CAAC,yEAAyE,CAAC,CAAA;YACxF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAA;YACrD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,iFAAiF,CAAC,CAAA;YACtG,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAA;QAC7D,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAA;YACnC,MAAM,OAAO,GAAkB;gBAC3B,EAAE,EAAE,SAAS;gBACb,MAAM;gBACN,MAAM;aACT,CAAA;YAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;gBACtC,MAAM,CAAC,IAAI,KAAK,CAAC,+BAA+B,MAAM,EAAE,CAAC,CAAC,CAAA;YAC9D,CAAC,EAAE,KAAK,CAAC,CAAA;YAET,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,EAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC,CAAA;YAE/D,kBAAkB;YAClB,IAAI,CAAC,YAAa,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CAAA;QAC5D,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,UAAU,CAAC,IAAY;QAC3B,IAAI,CAAC,MAAM,IAAI,IAAI,CAAA;QAEnB,WAAW;QACX,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACrC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA,CAAC,aAAa;QAE7C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAQ;YAE1B,IAAI,CAAC;gBACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAmB,CAAA;gBAClD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAA;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAA;YAC3D,CAAC;QACL,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,OAAuB;QACzC,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAA;YACrD,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;YAC7B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAEvC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACjC,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,eAAe,CAAC,CAAC,CAAA;YAC/D,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QACjC,CAAC;IACL,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,OAAe;QACrC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACrB,OAAO,CAAC,IAAI,CAAC,CAAA;gBACb,OAAM;YACV,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC5B,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;gBACnC,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;oBACrB,aAAa,CAAC,aAAa,CAAC,CAAA;oBAC5B,OAAO,CAAC,IAAI,CAAC,CAAA;gBACjB,CAAC;qBAAM,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CAAC;oBACzD,wBAAwB;oBACxB,aAAa,CAAC,aAAa,CAAC,CAAA;oBAC5B,OAAO,CAAC,KAAK,CAAC,CAAA;gBAClB,CAAC;YACL,CAAC,EAAE,GAAG,CAAC,CAAA;QACX,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,UAAU;QACd,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;IAC5E,CAAC;CACJ"}

package/dist/extension/ws-server.d.ts

@@ -0,0 +1,40 @@
+/**
+ * WebSocket 服务器
+ *
+ * 负责与 Chrome Extension 通信
+ */
+import { EventEmitter } from 'events';
+interface WebSocketMessage {
+    id: string;
+    action: string;
+    params?: unknown;
+}
+type MessageHandler = (message: WebSocketMessage) => Promise<unknown>;
+export declare class ExtensionWebSocketServer extends EventEmitter {
+    private app;
+    private server;
+    private wss;
+    private extensionSocket;
+    private messageHandlers;
+    private port;
+    private host;
+    private heartbeatInterval;
+    private heartbeatFrequency;
+    private commandHandler;
+    constructor(port?: number, host?: string);
+    onCommand(handler: MessageHandler): void;
+    private setupExpress;
+    private setupWebSocket;
+    private handleMessage;
+    sendCommand(action: string, params?: unknown): Promise<unknown>;
+    private waitForConnection;
+    private generateId;
+    private getLocalIP;
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    private startHeartbeat;
+    private stopHeartbeat;
+    isExtensionConnected(): boolean;
+}
+export {};
+//# sourceMappingURL=ws-server.d.ts.map

package/dist/extension/ws-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-server.d.ts","sourceRoot":"","sources":["../../src/extension/ws-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAC,YAAY,EAAC,MAAM,QAAQ,CAAA;AAMnC,UAAU,gBAAgB;IACtB,EAAE,EAAE,MAAM,CAAA;IACV,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,OAAO,CAAA;CACnB;AASD,KAAK,cAAc,GAAG,CAAC,OAAO,EAAE,gBAAgB,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;AAErE,qBAAa,wBAAyB,SAAQ,YAAY;IACtD,OAAO,CAAC,GAAG,CAAqB;IAChC,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,GAAG,CAAkB;IAC7B,OAAO,CAAC,eAAe,CAAyB;IAChD,OAAO,CAAC,eAAe,CAInB;IACJ,OAAO,CAAC,IAAI,CAAQ;IACpB,OAAO,CAAC,IAAI,CAAQ;IACpB,OAAO,CAAC,iBAAiB,CAA8B;IACvD,OAAO,CAAC,kBAAkB,CAAQ;IAClC,OAAO,CAAC,cAAc,CAA8B;gBAExC,IAAI,SAAO,EAAE,IAAI,SAAY;IAWzC,SAAS,CAAC,OAAO,EAAE,cAAc;IAIjC,OAAO,CAAC,YAAY;IA2CpB,OAAO,CAAC,cAAc;IAgDtB,OAAO,CAAC,aAAa;IAgBR,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAE,OAAY,GAAG,OAAO,CAAC,OAAO,CAAC;IA+BhF,OAAO,CAAC,iBAAiB;IAoBzB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,UAAU;IAmBX,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBtB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5B,OAAO,CAAC,cAAc;IAetB,OAAO,CAAC,aAAa;IAOd,oBAAoB,IAAI,OAAO;CAGzC"}

package/dist/extension/ws-server.js

@@ -0,0 +1,246 @@
+/**
+ * WebSocket 服务器
+ *
+ * 负责与 Chrome Extension 通信
+ */
+import { WebSocketServer, WebSocket } from 'ws';
+import { EventEmitter } from 'events';
+import express from 'express';
+import cors from 'cors';
+import { createServer } from 'http';
+import { networkInterfaces } from 'os';
+export class ExtensionWebSocketServer extends EventEmitter {
+    app;
+    server;
+    wss;
+    extensionSocket = null;
+    messageHandlers = new Map();
+    port;
+    host;
+    heartbeatInterval = null;
+    heartbeatFrequency = 20000;
+    commandHandler = null;
+    constructor(port = 9222, host = '0.0.0.0') {
+        super();
+        this.port = port;
+        this.host = host;
+        this.app = express();
+        this.setupExpress();
+        this.server = createServer(this.app);
+        this.setupWebSocket();
+    }
+    onCommand(handler) {
+        this.commandHandler = handler;
+    }
+    setupExpress() {
+        this.app.use(cors({
+            origin: '*',
+            methods: ['GET', 'POST', 'OPTIONS'],
+            allowedHeaders: ['Content-Type', 'Authorization'],
+        }));
+        this.app.use(express.json());
+        // Health check
+        this.app.get('/health', (_req, res) => {
+            res.json({
+                status: 'ok',
+                extensionConnected: this.extensionSocket !== null,
+                timestamp: new Date().toISOString(),
+            });
+        });
+        // Server info
+        this.app.get('/info', (_req, res) => {
+            res.json({
+                host: this.host,
+                port: this.port,
+                extensionConnected: this.extensionSocket !== null,
+                wsUrl: `ws://${this.getLocalIP()}:${this.port}/ws`,
+            });
+        });
+        // Manual command (for testing)
+        this.app.post('/command', async (req, res) => {
+            try {
+                const { action, params } = req.body;
+                const result = await this.sendCommand(action, params);
+                res.json({ success: true, data: result });
+            }
+            catch (error) {
+                res.status(500).json({
+                    success: false,
+                    error: error instanceof Error ? error.message : 'Unknown error',
+                });
+            }
+        });
+    }
+    setupWebSocket() {
+        this.wss = new WebSocketServer({
+            server: this.server,
+            path: '/ws',
+        });
+        this.wss.on('connection', (ws, req) => {
+            console.error('[WS] Extension connected from:', req.socket.remoteAddress);
+            this.extensionSocket = ws;
+            this.emit('extensionConnected');
+            this.startHeartbeat();
+            ws.on('message', (data) => {
+                try {
+                    const message = JSON.parse(data.toString());
+                    // 心跳响应
+                    if (message.type === 'pong' || message.type === 'ping') {
+                        return;
+                    }
+                    this.handleMessage(message);
+                }
+                catch (error) {
+                    console.error('[WS] Error parsing message:', error);
+                }
+            });
+            ws.on('close', () => {
+                console.error('[WS] Extension disconnected');
+                this.extensionSocket = null;
+                this.stopHeartbeat();
+                this.emit('extensionDisconnected');
+            });
+            ws.on('error', (error) => {
+                console.error('[WS] Error:', error);
+            });
+            // 发送连接确认
+            ws.send(JSON.stringify({
+                type: 'connection',
+                status: 'connected',
+                timestamp: Date.now(),
+            }));
+        });
+    }
+    handleMessage(message) {
+        if (message.id && this.messageHandlers.has(message.id)) {
+            const handler = this.messageHandlers.get(message.id);
+            clearTimeout(handler.timeout);
+            this.messageHandlers.delete(message.id);
+            if (message.success) {
+                handler.resolve(message.data);
+            }
+            else {
+                handler.reject(new Error(message.error || 'Unknown error'));
+            }
+        }
+        else {
+            this.emit('message', message);
+        }
+    }
+    async sendCommand(action, params = {}) {
+        // 如果 Extension 没连接，等待最多 10 秒让它连接
+        if (!this.extensionSocket || this.extensionSocket.readyState !== WebSocket.OPEN) {
+            const connected = await this.waitForConnection(10000);
+            if (!connected) {
+                throw new Error('Extension not connected. Please click the MCP Chrome extension icon in Chrome toolbar to activate it, ' +
+                    'or refresh the extension at chrome://extensions page.');
+            }
+        }
+        return new Promise((resolve, reject) => {
+            const messageId = this.generateId();
+            const message = {
+                id: messageId,
+                action,
+                params,
+            };
+            const timeout = setTimeout(() => {
+                this.messageHandlers.delete(messageId);
+                reject(new Error(`Request timeout for action: ${action}`));
+            }, 30000);
+            this.messageHandlers.set(messageId, { resolve, reject, timeout });
+            this.extensionSocket.send(JSON.stringify(message));
+        });
+    }
+    waitForConnection(timeout) {
+        return new Promise((resolve) => {
+            if (this.extensionSocket && this.extensionSocket.readyState === WebSocket.OPEN) {
+                resolve(true);
+                return;
+            }
+            const startTime = Date.now();
+            const checkInterval = setInterval(() => {
+                if (this.extensionSocket && this.extensionSocket.readyState === WebSocket.OPEN) {
+                    clearInterval(checkInterval);
+                    resolve(true);
+                }
+                else if (Date.now() - startTime > timeout) {
+                    clearInterval(checkInterval);
+                    resolve(false);
+                }
+            }, 500);
+        });
+    }
+    generateId() {
+        return Date.now().toString(36) + Math.random().toString(36).substring(2);
+    }
+    getLocalIP() {
+        const nets = networkInterfaces();
+        for (const name of Object.keys(nets)) {
+            const networkList = nets[name];
+            if (networkList) {
+                for (const net of networkList) {
+                    if (net.family === 'IPv4' && !net.internal) {
+                        if (name === 'eth0' || name.includes('eth')) {
+                            return net.address;
+                        }
+                    }
+                }
+            }
+        }
+        return '127.0.0.1';
+    }
+    start() {
+        return new Promise((resolve, reject) => {
+            this.server.listen(this.port, this.host, () => {
+                const actualIP = this.getLocalIP();
+                console.error(`[WS] Server running on:`);
+                console.error(`  Local: http://localhost:${this.port}`);
+                console.error(`  Network: http://${actualIP}:${this.port}`);
+                console.error(`  WebSocket: ws://${actualIP}:${this.port}/ws`);
+                resolve();
+            });
+            this.server.on('error', (error) => {
+                if (error.code === 'EADDRINUSE') {
+                    reject(new Error(`Port ${this.port} is already in use`));
+                }
+                else {
+                    reject(error);
+                }
+            });
+        });
+    }
+    stop() {
+        return new Promise((resolve) => {
+            this.stopHeartbeat();
+            this.wss.close(() => {
+                this.server.close(() => {
+                    resolve();
+                });
+            });
+        });
+    }
+    startHeartbeat() {
+        this.stopHeartbeat();
+        this.heartbeatInterval = setInterval(() => {
+            if (this.extensionSocket && this.extensionSocket.readyState === WebSocket.OPEN) {
+                this.extensionSocket.send(JSON.stringify({
+                    type: 'ping',
+                    timestamp: Date.now(),
+                }));
+            }
+            else {
+                this.stopHeartbeat();
+            }
+        }, this.heartbeatFrequency);
+    }
+    stopHeartbeat() {
+        if (this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+            this.heartbeatInterval = null;
+        }
+    }
+    isExtensionConnected() {
+        return this.extensionSocket !== null && this.extensionSocket.readyState === WebSocket.OPEN;
+    }
+}
+//# sourceMappingURL=ws-server.js.map

package/dist/extension/ws-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-server.js","sourceRoot":"","sources":["../../src/extension/ws-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAC,eAAe,EAAE,SAAS,EAAC,MAAM,IAAI,CAAA;AAC7C,OAAO,EAAC,YAAY,EAAC,MAAM,QAAQ,CAAA;AACnC,OAAO,OAAO,MAAM,SAAS,CAAA;AAC7B,OAAO,IAAI,MAAM,MAAM,CAAA;AACvB,OAAO,EAAC,YAAY,EAAuB,MAAM,MAAM,CAAA;AACvD,OAAO,EAAC,iBAAiB,EAAC,MAAM,IAAI,CAAA;AAiBpC,MAAM,OAAO,wBAAyB,SAAQ,YAAY;IAC9C,GAAG,CAAqB;IACxB,MAAM,CAAY;IAClB,GAAG,CAAkB;IACrB,eAAe,GAAqB,IAAI,CAAA;IACxC,eAAe,GAAG,IAAI,GAAG,EAI7B,CAAA;IACI,IAAI,CAAQ;IACZ,IAAI,CAAQ;IACZ,iBAAiB,GAA0B,IAAI,CAAA;IAC/C,kBAAkB,GAAG,KAAK,CAAA;IAC1B,cAAc,GAA0B,IAAI,CAAA;IAEpD,YAAY,IAAI,GAAG,IAAI,EAAE,IAAI,GAAG,SAAS;QACrC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAEhB,IAAI,CAAC,GAAG,GAAG,OAAO,EAAE,CAAA;QACpB,IAAI,CAAC,YAAY,EAAE,CAAA;QACnB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACpC,IAAI,CAAC,cAAc,EAAE,CAAA;IACzB,CAAC;IAED,SAAS,CAAC,OAAuB;QAC7B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAA;IACjC,CAAC;IAEO,YAAY;QAChB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;YACd,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC;YACnC,cAAc,EAAE,CAAC,cAAc,EAAE,eAAe,CAAC;SACpD,CAAC,CAAC,CAAA;QAEH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAA;QAE5B,eAAe;QACf,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAClC,GAAG,CAAC,IAAI,CAAC;gBACL,MAAM,EAAE,IAAI;gBACZ,kBAAkB,EAAE,IAAI,CAAC,eAAe,KAAK,IAAI;gBACjD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACtC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,cAAc;QACd,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAChC,GAAG,CAAC,IAAI,CAAC;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,kBAAkB,EAAE,IAAI,CAAC,eAAe,KAAK,IAAI;gBACjD,KAAK,EAAE,QAAQ,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,CAAC,IAAI,KAAK;aACrD,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,+BAA+B;QAC/B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACzC,IAAI,CAAC;gBACD,MAAM,EAAC,MAAM,EAAE,MAAM,EAAC,GAAG,GAAG,CAAC,IAAI,CAAA;gBACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;gBACrD,GAAG,CAAC,IAAI,CAAC,EAAC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAC,CAAC,CAAA;YAC3C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACjB,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;iBAClE,CAAC,CAAA;YACN,CAAC;QACL,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,cAAc;QAClB,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,KAAK;SACd,CAAC,CAAA;QAEF,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE;YAClC,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;YAEzE,IAAI,CAAC,eAAe,GAAG,EAAE,CAAA;YACzB,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAA;YAC/B,IAAI,CAAC,cAAc,EAAE,CAAA;YAErB,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBACtB,IAAI,CAAC;oBACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAA;oBAE3C,OAAO;oBACP,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACrD,OAAM;oBACV,CAAC;oBAED,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAA;gBAC/B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAA;gBACvD,CAAC;YACL,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAChB,OAAO,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAA;gBAC5C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAA;gBAC3B,IAAI,CAAC,aAAa,EAAE,CAAA;gBACpB,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAA;YACtC,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACrB,OAAO,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,CAAA;YACvC,CAAC,CAAC,CAAA;YAEF,SAAS;YACT,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;gBACnB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACxB,CAAC,CAAC,CAAA;QACP,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,aAAa,CAAC,OAA0B;QAC5C,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAA;YACrD,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;YAC7B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAEvC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACjC,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,eAAe,CAAC,CAAC,CAAA;YAC/D,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QACjC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,SAAkB,EAAE;QACzD,iCAAiC;QACjC,IAAI,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC9E,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAA;YACrD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CACX,wGAAwG;oBACxG,uDAAuD,CAC1D,CAAA;YACL,CAAC;QACL,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAA;YACnC,MAAM,OAAO,GAAqB;gBAC9B,EAAE,EAAE,SAAS;gBACb,MAAM;gBACN,MAAM;aACT,CAAA;YAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;gBACtC,MAAM,CAAC,IAAI,KAAK,CAAC,+BAA+B,MAAM,EAAE,CAAC,CAAC,CAAA;YAC9D,CAAC,EAAE,KAAK,CAAC,CAAA;YAET,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,EAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC,CAAA;YAE/D,IAAI,CAAC,eAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QACvD,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,iBAAiB,CAAC,OAAe;QACrC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC7E,OAAO,CAAC,IAAI,CAAC,CAAA;gBACb,OAAM;YACV,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC5B,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE;gBACnC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;oBAC7E,aAAa,CAAC,aAAa,CAAC,CAAA;oBAC5B,OAAO,CAAC,IAAI,CAAC,CAAA;gBACjB,CAAC;qBAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CAAC;oBAC1C,aAAa,CAAC,aAAa,CAAC,CAAA;oBAC5B,OAAO,CAAC,KAAK,CAAC,CAAA;gBAClB,CAAC;YACL,CAAC,EAAE,GAAG,CAAC,CAAA;QACX,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,UAAU;QACd,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;IAC5E,CAAC;IAEO,UAAU;QACd,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAA;QAEhC,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAA;YAC9B,IAAI,WAAW,EAAE,CAAC;gBACd,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;oBAC5B,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;wBACzC,IAAI,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;4BAC1C,OAAO,GAAG,CAAC,OAAO,CAAA;wBACtB,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,WAAW,CAAA;IACtB,CAAC;IAEM,KAAK;QACR,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAA;gBAClC,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAA;gBACxC,OAAO,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;gBACvD,OAAO,CAAC,KAAK,CAAC,qBAAqB,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;gBAC3D,OAAO,CAAC,KAAK,CAAC,qBAAqB,QAAQ,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,CAAA;gBAC9D,OAAO,EAAE,CAAA;YACb,CAAC,CAAC,CAAA;YAEF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAA4B,EAAE,EAAE;gBACrD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,oBAAoB,CAAC,CAAC,CAAA;gBAC5D,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,KAAK,CAAC,CAAA;gBACjB,CAAC;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;IAEM,IAAI;QACP,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3B,IAAI,CAAC,aAAa,EAAE,CAAA;YACpB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE;gBAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;oBACnB,OAAO,EAAE,CAAA;gBACb,CAAC,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,cAAc;QAClB,IAAI,CAAC,aAAa,EAAE,CAAA;QAEpB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACtC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC7E,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;oBACrC,IAAI,EAAE,MAAM;oBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACxB,CAAC,CAAC,CAAA;YACP,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,aAAa,EAAE,CAAA;YACxB,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAA;IAC/B,CAAC;IAEO,aAAa;QACjB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;YACrC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAA;QACjC,CAAC;IACL,CAAC;IAEM,oBAAoB;QACvB,OAAO,IAAI,CAAC,eAAe,KAAK,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,CAAA;IAC9F,CAAC;CACJ"}

package/dist/index.js

@@ -13,7 +13,7 @@
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
-import { getSession } from './core/index.js';
+import { getUnifiedSession } from './core/index.js';
 import { browseToolDefinition, cookiesToolDefinition, evaluateToolDefinition, extractToolDefinition, handleBrowse, handleCookies, handleEvaluate, handleExtract, handleInput, handleLogs, handleManage, handleWait, inputToolDefinition, logsToolDefinition, manageToolDefinition, waitToolDefinition, } from './tools/index.js';
 /**
  * 所有工具定义
@@ -47,7 +47,7 @@ const handlers = {
 function createServer() {
     const server = new Server({
         name: 'mcp-chrome',
-        version: '1.0.0',
+        version: '1.1.0',
     }, {
         capabilities: {
             tools: {},
@@ -87,8 +87,8 @@ function createServer() {
  */
 async function cleanup() {
     try {
-        // 先关闭浏览器会话
-        await getSession().close();
+        // 关闭统一会话（包括 Extension 和 CDP）
+        await getUnifiedSession().close();
     }
     catch {
         // 忽略清理错误
@@ -98,6 +98,8 @@ async function cleanup() {
  * 主函数
  */
 async function main() {
+    // 启动 Extension HTTP/WebSocket 服务器
+    await getUnifiedSession().startExtensionServer();
     const server = createServer();
     const transport = new StdioServerTransport();
     await server.connect(transport);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA;;;;;;;;;;GAUG;AAEH,OAAO,EAAC,MAAM,EAAC,MAAM,2CAA2C,CAAA;AAChE,OAAO,EAAC,oBAAoB,EAAC,MAAM,2CAA2C,CAAA;AAC9E,OAAO,EAAC,qBAAqB,EAAE,sBAAsB,EAAC,MAAM,oCAAoC,CAAA;AAEhG,OAAO,EAAC,UAAU,EAAC,MAAM,iBAAiB,CAAA;AAC1C,OAAO,EACH,oBAAoB,EACpB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,YAAY,EACZ,aAAa,EACb,cAAc,EACd,aAAa,EACb,WAAW,EACX,UAAU,EACV,YAAY,EACZ,UAAU,EACV,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAA;AAEzB;;GAEG;AACH,MAAM,KAAK,GAAG;IACV,oBAAoB;IACpB,mBAAmB;IACnB,qBAAqB;IACrB,kBAAkB;IAClB,qBAAqB;IACrB,kBAAkB;IAClB,sBAAsB;IACtB,oBAAoB;CACvB,CAAA;AAED;;GAEG;AACH,MAAM,QAAQ,GAMV;IACA,MAAM,EAAE,YAAY;IACpB,KAAK,EAAE,WAAW;IAClB,OAAO,EAAE,aAAa;IACtB,IAAI,EAAE,UAAU;IAChB,OAAO,EAAE,aAAa;IACtB,IAAI,EAAE,UAAU;IAChB,QAAQ,EAAE,cAAc;IACxB,MAAM,EAAE,YAAY;CACvB,CAAA;AAED;;GAEG;AACH,SAAS,YAAY;IACjB,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB;QACI,IAAI,EAAE,YAAY;QAClB,OAAO,EAAE,OAAO;KACnB,EACD;QACI,YAAY,EAAE;YACV,KAAK,EAAE,EAAE;SACZ;KACJ,CACJ,CAAA;IAED,OAAO;IACP,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACxD,OAAO,EAAC,KAAK,EAAC,CAAA;IAClB,CAAC,CAAC,CAAA;IAEF,OAAO;IACP,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAC9D,MAAM,EAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,GAAG,OAAO,CAAC,MAAM,CAAA;QAE9C,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAA;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACI,KAAK,EAAE;gCACH,IAAI,EAAE,cAAc;gCACpB,OAAO,EAAE,SAAS,IAAI,EAAE;gCACxB,UAAU,EAAE,SAAS,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;6BAC1D;yBACJ,CAAC;qBAC1B;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAA;QACL,CAAC;QAED,OAAO,OAAO,CAAC,IAAI,CAAC,CAAA;IACxB,CAAC,CAAC,CAAA;IAEF,OAAO,MAAM,CAAA;AACjB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,OAAO;IAClB,IAAI,CAAC;QACD,WAAW;QACX,MAAM,UAAU,EAAE,CAAC,KAAK,EAAE,CAAA;IAC9B,CAAC;IAAC,MAAM,CAAC;QACL,SAAS;IACb,CAAC;AACL,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI;IACf,MAAM,MAAM,GAAM,YAAY,EAAE,CAAA;IAChC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAA;IAE5C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAE/B,OAAO;IACP,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC5B,MAAM,OAAO,EAAE,CAAA;QACf,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACpB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,OAAO,EAAE,CAAA;QACf,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACpB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnB,CAAC,CAAC,CAAA;AACN,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;IACzB,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,MAAM,OAAO,EAAE,CAAA;IACf,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;AACnB,CAAC,CAAC,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA;;;;;;;;;;GAUG;AAEH,OAAO,EAAC,MAAM,EAAC,MAAM,2CAA2C,CAAA;AAChE,OAAO,EAAC,oBAAoB,EAAC,MAAM,2CAA2C,CAAA;AAC9E,OAAO,EAAC,qBAAqB,EAAE,sBAAsB,EAAC,MAAM,oCAAoC,CAAA;AAEhG,OAAO,EAAa,iBAAiB,EAAC,MAAM,iBAAiB,CAAA;AAC7D,OAAO,EACH,oBAAoB,EACpB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,YAAY,EACZ,aAAa,EACb,cAAc,EACd,aAAa,EACb,WAAW,EACX,UAAU,EACV,YAAY,EACZ,UAAU,EACV,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAA;AAEzB;;GAEG;AACH,MAAM,KAAK,GAAG;IACV,oBAAoB;IACpB,mBAAmB;IACnB,qBAAqB;IACrB,kBAAkB;IAClB,qBAAqB;IACrB,kBAAkB;IAClB,sBAAsB;IACtB,oBAAoB;CACvB,CAAA;AAED;;GAEG;AACH,MAAM,QAAQ,GAMV;IACA,MAAM,EAAE,YAAY;IACpB,KAAK,EAAE,WAAW;IAClB,OAAO,EAAE,aAAa;IACtB,IAAI,EAAE,UAAU;IAChB,OAAO,EAAE,aAAa;IACtB,IAAI,EAAE,UAAU;IAChB,QAAQ,EAAE,cAAc;IACxB,MAAM,EAAE,YAAY;CACvB,CAAA;AAED;;GAEG;AACH,SAAS,YAAY;IACjB,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB;QACI,IAAI,EAAE,YAAY;QAClB,OAAO,EAAE,OAAO;KACnB,EACD;QACI,YAAY,EAAE;YACV,KAAK,EAAE,EAAE;SACZ;KACJ,CACJ,CAAA;IAED,OAAO;IACP,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACxD,OAAO,EAAC,KAAK,EAAC,CAAA;IAClB,CAAC,CAAC,CAAA;IAEF,OAAO;IACP,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAC9D,MAAM,EAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAC,GAAG,OAAO,CAAC,MAAM,CAAA;QAE9C,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAA;QAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACI,KAAK,EAAE;gCACH,IAAI,EAAE,cAAc;gCACpB,OAAO,EAAE,SAAS,IAAI,EAAE;gCACxB,UAAU,EAAE,SAAS,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;6BAC1D;yBACJ,CAAC;qBAC1B;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAA;QACL,CAAC;QAED,OAAO,OAAO,CAAC,IAAI,CAAC,CAAA;IACxB,CAAC,CAAC,CAAA;IAEF,OAAO,MAAM,CAAA;AACjB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,OAAO;IAClB,IAAI,CAAC;QACD,6BAA6B;QAC7B,MAAM,iBAAiB,EAAE,CAAC,KAAK,EAAE,CAAA;IACrC,CAAC;IAAC,MAAM,CAAC;QACL,SAAS;IACb,CAAC;AACL,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI;IACf,kCAAkC;IAClC,MAAM,iBAAiB,EAAE,CAAC,oBAAoB,EAAE,CAAA;IAEhD,MAAM,MAAM,GAAM,YAAY,EAAE,CAAA;IAChC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAA;IAE5C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAE/B,OAAO;IACP,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC5B,MAAM,OAAO,EAAE,CAAA;QACf,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACpB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnB,CAAC,CAAC,CAAA;IAEF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,OAAO,EAAE,CAAA;QACf,MAAM,MAAM,CAAC,KAAK,EAAE,CAAA;QACpB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnB,CAAC,CAAC,CAAA;AACN,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;IACzB,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,MAAM,OAAO,EAAE,CAAA;IACf,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;AACnB,CAAC,CAAC,CAAA"}