axhub-mcp-bridge 1.0.34 → 1.0.36

package/dist/server/websocket-bridge.js

@@ -0,0 +1,311 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketBridge = void 0;
+const ws_1 = require("ws");
+const logger_1 = __importDefault(require("../utils/logger"));
+class WebSocketBridge {
+    constructor() {
+        this.wss = null;
+        this.clients = new Map();
+        this.heartbeatInterval = null;
+        this.onFigmaMessageCallback = null;
+        this.onClientConnectCallback = null;
+        this.onClientDisconnectCallback = null;
+        logger_1.default.log('[WS Bridge] Initializing WebSocket Bridge');
+    }
+    /**
+     * Initialize WebSocket server on existing HTTP server
+     */
+    initialize(server) {
+        this.wss = new ws_1.WebSocketServer({
+            server,
+            path: '/ws'
+        });
+        this.wss.on('connection', (ws, req) => {
+            this.handleConnection(ws, req);
+        });
+        // Start heartbeat
+        this.startHeartbeat();
+        logger_1.default.log('[WS Bridge] WebSocket server initialized on /ws');
+    }
+    /**
+     * Handle new WebSocket connection
+     * All WebSocket connections are from Figma plugin
+     */
+    handleConnection(ws, req) {
+        const clientId = this.generateClientId();
+        // All WebSocket connections are from Figma plugin
+        const clientType = 'figma';
+        logger_1.default.log('[WS Bridge] New connection from Figma plugin', { clientId, url: req.url, host: req.headers.host });
+        const clientInfo = {
+            ws,
+            id: clientId,
+            type: clientType,
+            alive: true,
+        };
+        this.clients.set(clientId, clientInfo);
+        // Send connection confirmation
+        this.sendToClient(clientId, {
+            type: 'ws:connected',
+            payload: { clientId, clientType: clientInfo.type }
+        });
+        logger_1.default.log('[WS Bridge] Client registered', {
+            clientId,
+            type: clientInfo.type,
+            totalClients: this.clients.size,
+            url: req.url,
+            host: req.headers.host,
+            allClientTypes: Array.from(this.clients.values()).map(c => ({ id: c.id, type: c.type }))
+        });
+        // Notify extension about client connection
+        if (this.onClientConnectCallback) {
+            const stats = this.getStats();
+            this.onClientConnectCallback(clientId, { total: stats.total, figma: stats.figma });
+        }
+        // Handle messages
+        ws.on('message', (data) => {
+            this.handleMessage(clientId, data);
+        });
+        // Handle pong (heartbeat response)
+        ws.on('pong', () => {
+            const client = this.clients.get(clientId);
+            if (client) {
+                client.alive = true;
+            }
+        });
+        // Handle close
+        ws.on('close', () => {
+            logger_1.default.log('[WS Bridge] Client disconnected', { clientId });
+            const stats = this.getStats();
+            this.clients.delete(clientId);
+            // Notify extension about client disconnection
+            if (this.onClientDisconnectCallback) {
+                const newStats = this.getStats();
+                this.onClientDisconnectCallback(clientId, { total: newStats.total, figma: newStats.figma });
+            }
+        });
+        // Handle errors
+        ws.on('error', (error) => {
+            logger_1.default.error('[WS Bridge] Client error', { clientId, error: error.message });
+        });
+    }
+    /**
+     * Handle incoming message from client
+     */
+    handleMessage(fromClientId, data) {
+        var _a;
+        try {
+            const message = JSON.parse(data.toString());
+            const fromClient = this.clients.get(fromClientId);
+            if (!fromClient) {
+                logger_1.default.warn('[WS Bridge] Message from unknown client', { fromClientId });
+                return;
+            }
+            logger_1.default.log('[WS Bridge] Received message', {
+                from: fromClient.type,
+                clientId: fromClientId,
+                type: message.type,
+                target: message.target,
+                hasRequestId: !!message.___requestId
+            });
+            // Handle special message types
+            if (message.type === 'ws:register') {
+                // Update client type
+                const newType = (_a = message.payload) === null || _a === void 0 ? void 0 : _a.clientType;
+                if (newType === 'figma' || newType === 'extension') {
+                    fromClient.type = newType;
+                    logger_1.default.log('[WS Bridge] Client type updated', { clientId: fromClientId, type: newType });
+                    this.sendToClient(fromClientId, {
+                        type: 'ws:registered',
+                        payload: { clientId: fromClientId, clientType: newType }
+                    });
+                }
+                return;
+            }
+            // Route message based on target
+            this.routeMessage(fromClient, message);
+        }
+        catch (error) {
+            logger_1.default.error('[WS Bridge] Failed to parse message', {
+                fromClientId,
+                error: error.message
+            });
+        }
+    }
+    /**
+     * Register callback for Figma messages
+     * This allows native-messaging-host to receive messages from Figma plugin
+     */
+    onFigmaMessage(callback) {
+        this.onFigmaMessageCallback = callback;
+    }
+    /**
+     * Register callback for client connection events
+     */
+    onClientConnect(callback) {
+        this.onClientConnectCallback = callback;
+    }
+    /**
+     * Register callback for client disconnection events
+     */
+    onClientDisconnect(callback) {
+        this.onClientDisconnectCallback = callback;
+    }
+    /**
+     * Route message from Figma plugin
+     * Messages from Figma should be forwarded to Extension via Native Messaging
+     */
+    routeMessage(fromClient, message) {
+        // All WebSocket clients are Figma plugins
+        // Messages from Figma should be forwarded to Extension via Native Messaging
+        logger_1.default.log('[WS Bridge] Message from Figma plugin', {
+            clientId: fromClient.id,
+            messageType: message.type
+        });
+        // If target is 'broadcast', send to all other Figma clients
+        if (message.target === 'broadcast') {
+            this.broadcastMessage(fromClient.id, message);
+            return;
+        }
+        // Forward to Extension via Native Messaging callback
+        if (this.onFigmaMessageCallback) {
+            this.onFigmaMessageCallback(message);
+            logger_1.default.log('[WS Bridge] Message forwarded to Extension via Native Messaging');
+        }
+        else {
+            logger_1.default.warn('[WS Bridge] No callback registered for Figma messages');
+        }
+    }
+    /**
+     * Broadcast message to all clients except sender
+     */
+    broadcastMessage(exceptClientId, message) {
+        let sent = 0;
+        for (const [clientId, client] of this.clients.entries()) {
+            if (clientId !== exceptClientId) {
+                this.sendToClient(clientId, message);
+                sent++;
+            }
+        }
+        logger_1.default.log('[WS Bridge] Broadcast message', { type: message.type, recipients: sent });
+    }
+    /**
+     * Send message to specific client
+     */
+    sendToClient(clientId, message) {
+        const client = this.clients.get(clientId);
+        if (!client) {
+            logger_1.default.warn('[WS Bridge] Cannot send to unknown client', { clientId });
+            return;
+        }
+        if (client.ws.readyState === ws_1.WebSocket.OPEN) {
+            try {
+                client.ws.send(JSON.stringify(message));
+            }
+            catch (error) {
+                logger_1.default.error('[WS Bridge] Failed to send message', {
+                    clientId,
+                    error: error.message
+                });
+            }
+        }
+    }
+    /**
+     * Start heartbeat to detect dead connections
+     */
+    startHeartbeat() {
+        this.heartbeatInterval = setInterval(() => {
+            for (const [clientId, client] of this.clients.entries()) {
+                if (!client.alive) {
+                    logger_1.default.log('[WS Bridge] Terminating dead connection', { clientId });
+                    client.ws.terminate();
+                    this.clients.delete(clientId);
+                    continue;
+                }
+                client.alive = false;
+                client.ws.ping();
+            }
+        }, 30000); // 30 seconds
+    }
+    /**
+     * Stop heartbeat
+     */
+    stopHeartbeat() {
+        if (this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+            this.heartbeatInterval = null;
+        }
+    }
+    /**
+     * Generate unique client ID
+     */
+    generateClientId() {
+        return `client_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+    }
+    /**
+     * Get connection stats
+     * All WebSocket clients are Figma plugins
+     */
+    getStats() {
+        const details = [];
+        for (const [clientId, client] of this.clients.entries()) {
+            details.push({
+                id: clientId,
+                type: client.type,
+                alive: client.alive,
+                readyState: client.ws.readyState
+            });
+        }
+        logger_1.default.log('[WS Bridge] Stats calculated', {
+            total: this.clients.size,
+            figma: this.clients.size,
+            details
+        });
+        return {
+            total: this.clients.size,
+            figma: this.clients.size,
+            details
+        };
+    }
+    /**
+     * Send message to all Figma clients
+     * All WebSocket clients are Figma plugins
+     */
+    sendToClientType(type, message) {
+        let sent = 0;
+        logger_1.default.log('[WS Bridge] Sending message to Figma clients', { messageType: message.type, clientCount: this.clients.size });
+        for (const [clientId, client] of this.clients.entries()) {
+            // All clients are figma type
+            logger_1.default.log('[WS Bridge] Sending to client', { clientId, clientType: client.type, messageType: message.type });
+            this.sendToClient(clientId, message);
+            sent++;
+        }
+        if (sent === 0) {
+            logger_1.default.warn('[WS Bridge] No Figma clients connected', { totalClients: this.clients.size });
+        }
+        return sent;
+    }
+    /**
+     * Cleanup
+     */
+    destroy() {
+        this.stopHeartbeat();
+        // Close all client connections
+        for (const client of this.clients.values()) {
+            client.ws.close();
+        }
+        this.clients.clear();
+        if (this.wss) {
+            this.wss.close();
+            this.wss = null;
+        }
+        logger_1.default.log('[WS Bridge] WebSocket bridge destroyed');
+    }
+}
+exports.WebSocketBridge = WebSocketBridge;
+const websocketBridge = new WebSocketBridge();
+exports.default = websocketBridge;
+//# sourceMappingURL=websocket-bridge.js.map

package/dist/server/websocket-bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-bridge.js","sourceRoot":"","sources":["../../src/server/websocket-bridge.ts"],"names":[],"mappings":";;;;;;AAAA,2BAAgD;AAEhD,6DAAqC;AAgBrC,MAAa,eAAe;IAQ1B;QAPQ,QAAG,GAA2B,IAAI,CAAC;QACnC,YAAO,GAA4B,IAAI,GAAG,EAAE,CAAC;QAC7C,sBAAiB,GAA0B,IAAI,CAAC;QAChD,2BAAsB,GAAgD,IAAI,CAAC;QAC3E,4BAAuB,GAAiF,IAAI,CAAC;QAC7G,+BAA0B,GAAiF,IAAI,CAAC;QAGtH,gBAAM,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACI,UAAU,CAAC,MAAW;QAC3B,IAAI,CAAC,GAAG,GAAG,IAAI,oBAAe,CAAC;YAC7B,MAAM;YACN,IAAI,EAAE,KAAK;SACZ,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,GAAoB,EAAE,EAAE;YAChE,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,kBAAkB;QAClB,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,gBAAM,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;IAChE,CAAC;IAED;;;OAGG;IACK,gBAAgB,CAAC,EAAa,EAAE,GAAoB;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACzC,kDAAkD;QAClD,MAAM,UAAU,GAAY,OAAO,CAAC;QAEpC,gBAAM,CAAC,GAAG,CAAC,8CAA8C,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAE/G,MAAM,UAAU,GAAe;YAC7B,EAAE;YACF,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEvC,+BAA+B;QAC/B,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YAC1B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,CAAC,IAAI,EAAE;SACnD,CAAC,CAAC;QAEH,gBAAM,CAAC,GAAG,CAAC,+BAA+B,EAAE;YAC1C,QAAQ;YACR,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YAC/B,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI;YACtB,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;SACzF,CAAC,CAAC;QAEH,2CAA2C;QAC3C,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACrF,CAAC;QAED,kBAAkB;QAClB,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAY,EAAE,EAAE;YAChC,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,mCAAmC;QACnC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC;YACtB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,eAAe;QACf,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,gBAAM,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAE9B,8CAA8C;YAC9C,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;YAC9F,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,gBAAgB;QAChB,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;YAC9B,gBAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,YAAoB,EAAE,IAAY;;QACtD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAoB,CAAC;YAC/D,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAElD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,gBAAM,CAAC,IAAI,CAAC,yCAAyC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBACzE,OAAO;YACT,CAAC;YAED,gBAAM,CAAC,GAAG,CAAC,8BAA8B,EAAE;gBACzC,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,YAAY,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY;aACrC,CAAC,CAAC;YAEH,+BAA+B;YAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBACnC,qBAAqB;gBACrB,MAAM,OAAO,GAAG,MAAC,OAAO,CAAC,OAAe,0CAAE,UAAU,CAAC;gBACrD,IAAI,OAAO,KAAK,OAAO,IAAI,OAAO,KAAK,WAAW,EAAE,CAAC;oBACnD,UAAU,CAAC,IAAI,GAAG,OAAO,CAAC;oBAC1B,gBAAM,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;oBAEzF,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE;wBAC9B,IAAI,EAAE,eAAe;wBACrB,OAAO,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,EAAE;qBACzD,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO;YACT,CAAC;YAED,gCAAgC;YAChC,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAEzC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,gBAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE;gBAClD,YAAY;gBACZ,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,cAAc,CAAC,QAA4C;QAChE,IAAI,CAAC,sBAAsB,GAAG,QAAQ,CAAC;IACzC,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,QAA6E;QAClG,IAAI,CAAC,uBAAuB,GAAG,QAAQ,CAAC;IAC1C,CAAC;IAED;;OAEG;IACI,kBAAkB,CAAC,QAA6E;QACrG,IAAI,CAAC,0BAA0B,GAAG,QAAQ,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACK,YAAY,CAAC,UAAsB,EAAE,OAAwB;QACnE,0CAA0C;QAC1C,4EAA4E;QAE5E,gBAAM,CAAC,GAAG,CAAC,uCAAuC,EAAE;YAClD,QAAQ,EAAE,UAAU,CAAC,EAAE;YACvB,WAAW,EAAE,OAAO,CAAC,IAAI;SAC1B,CAAC,CAAC;QAEH,4DAA4D;QAC5D,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACnC,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,qDAAqD;QACrD,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAChC,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YACrC,gBAAM,CAAC,GAAG,CAAC,iEAAiE,CAAC,CAAC;QAChF,CAAC;aAAM,CAAC;YACN,gBAAM,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,cAAsB,EAAE,OAAwB;QACvE,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACxD,IAAI,QAAQ,KAAK,cAAc,EAAE,CAAC;gBAChC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBACrC,IAAI,EAAE,CAAC;YACT,CAAC;QACH,CAAC;QACD,gBAAM,CAAC,GAAG,CAAC,+BAA+B,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;IACxF,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,QAAgB,EAAE,OAAwB;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,gBAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;YACvE,OAAO;QACT,CAAC;QAED,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,cAAS,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,gBAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;oBACjD,QAAQ;oBACR,KAAK,EAAE,KAAK,CAAC,OAAO;iBACrB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;oBAClB,gBAAM,CAAC,GAAG,CAAC,yCAAyC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;oBACpE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;oBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC9B,SAAS;gBACX,CAAC;gBAED,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;gBACrB,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACnB,CAAC;QACH,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,aAAa;IAC1B,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,OAAO,UAAU,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC3E,CAAC;IAED;;;OAGG;IACI,QAAQ;QACb,MAAM,OAAO,GAA4E,EAAE,CAAC;QAE5F,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC,UAAU;aACjC,CAAC,CAAC;QACL,CAAC;QAED,gBAAM,CAAC,GAAG,CAAC,8BAA8B,EAAE;YACzC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YACxB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YACxB,OAAO;SACR,CAAC,CAAC;QAEH,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YACxB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YACxB,OAAO;SACR,CAAC;IACJ,CAAC;IAED;;;OAGG;IACI,gBAAgB,CAAC,IAAa,EAAE,OAAwB;QAC7D,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,gBAAM,CAAC,GAAG,CAAC,8CAA8C,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAE1H,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACxD,6BAA6B;YAC7B,gBAAM,CAAC,GAAG,CAAC,+BAA+B,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9G,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACrC,IAAI,EAAE,CAAC;QACT,CAAC;QAED,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACf,gBAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACI,OAAO;QACZ,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,+BAA+B;QAC/B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;YACjB,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;QAClB,CAAC;QAED,gBAAM,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACvD,CAAC;CACF;AAxVD,0CAwVC;AAED,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;AAC9C,kBAAe,eAAe,CAAC"}

package/dist/shared.d.ts

@@ -15,7 +15,17 @@ export declare enum NativeMessageType {
     ERROR_FROM_NATIVE_HOST = "error_from_native_host",
     CONNECT_NATIVE = "connectNative",
     PING_NATIVE = "ping_native",
-    DISCONNECT_NATIVE = "disconnect_native"
+    DISCONNECT_NATIVE = "disconnect_native",
+    START_WEBSOCKET = "start_websocket",
+    STOP_WEBSOCKET = "stop_websocket",
+    WEBSOCKET_STARTED = "websocket_started",
+    WEBSOCKET_STOPPED = "websocket_stopped",
+    WEBSOCKET_STATUS = "websocket_status",
+    FIGMA_MESSAGE = "figma_message",
+    FIGMA_REQUEST = "figma_request",
+    FIGMA_RESPONSE = "figma_response",
+    WEBSOCKET_CLIENT_CONNECTED = "websocket_client_connected",
+    WEBSOCKET_CLIENT_DISCONNECTED = "websocket_client_disconnected"
 }
 export interface NativeMessage<P = any, E = any> {
     type?: NativeMessageType;
@@ -25,13 +35,15 @@ export interface NativeMessage<P = any, E = any> {
 }
 export declare const TOOL_NAMES: {
     BROWSER: {
-        NAVIGATE: string;
         SEARCH_AXURE_PROTOTYPE: string;
         GET_AXURE_SITEMAP: string;
         GET_AXURE_PAGE_SCREENSHOT: string;
         GET_AXURE_HTML: string;
         GET_AXURE_INTERACTIONS: string;
         GET_AXURE_PAGE_TEXT: string;
+        GET_AXURE_IMAGE: string;
+        GET_PAGE_THEME: string;
+        DOWNLOAD_PAGE_DATA: string;
     };
 };
 import { type Tool } from '@modelcontextprotocol/sdk/types.js';

package/dist/shared.js

@@ -23,42 +23,36 @@ var NativeMessageType;
     NativeMessageType["CONNECT_NATIVE"] = "connectNative";
     NativeMessageType["PING_NATIVE"] = "ping_native";
     NativeMessageType["DISCONNECT_NATIVE"] = "disconnect_native";
+    // WebSocket server control
+    NativeMessageType["START_WEBSOCKET"] = "start_websocket";
+    NativeMessageType["STOP_WEBSOCKET"] = "stop_websocket";
+    NativeMessageType["WEBSOCKET_STARTED"] = "websocket_started";
+    NativeMessageType["WEBSOCKET_STOPPED"] = "websocket_stopped";
+    NativeMessageType["WEBSOCKET_STATUS"] = "websocket_status";
+    // Figma bridge messages (relayed through native host)
+    NativeMessageType["FIGMA_MESSAGE"] = "figma_message";
+    NativeMessageType["FIGMA_REQUEST"] = "figma_request";
+    NativeMessageType["FIGMA_RESPONSE"] = "figma_response";
+    // WebSocket client connection events
+    NativeMessageType["WEBSOCKET_CLIENT_CONNECTED"] = "websocket_client_connected";
+    NativeMessageType["WEBSOCKET_CLIENT_DISCONNECTED"] = "websocket_client_disconnected";
 })(NativeMessageType || (exports.NativeMessageType = NativeMessageType = {}));
 // export const DEFAULT_SERVER_PORT = 12308;
 // export const HOST_NAME = 'com.axhub.nativehost';
 exports.TOOL_NAMES = {
     BROWSER: {
-        NAVIGATE: 'chrome_navigate',
         SEARCH_AXURE_PROTOTYPE: 'search_axure',
         GET_AXURE_SITEMAP: 'get_axure_sitemap',
         GET_AXURE_PAGE_SCREENSHOT: 'get_axure_screenshot',
         GET_AXURE_HTML: 'get_axure_html',
         GET_AXURE_INTERACTIONS: 'get_axure_interactions',
         GET_AXURE_PAGE_TEXT: 'get_axure_text',
+        GET_AXURE_IMAGE: 'get_axure_image',
+        GET_PAGE_THEME: 'get_page_theme',
+        DOWNLOAD_PAGE_DATA: 'download_page_data',
     },
 };
 exports.TOOL_SCHEMAS = [
-    {
-        name: exports.TOOL_NAMES.BROWSER.NAVIGATE,
-        description: 'Navigate to a URL or refresh the current tab',
-        inputSchema: {
-            type: 'object',
-            properties: {
-                url: { type: 'string', description: 'URL to navigate to the website specified' },
-                newWindow: {
-                    type: 'boolean',
-                    description: 'Create a new window to navigate to the URL or not. Defaults to false',
-                },
-                width: { type: 'number', description: 'Viewport width in pixels (default: 1280)' },
-                height: { type: 'number', description: 'Viewport height in pixels (default: 720)' },
-                refresh: {
-                    type: 'boolean',
-                    description: 'Refresh the current active tab instead of navigating to a URL. When true, the url parameter is ignored. Defaults to false',
-                },
-            },
-            required: [],
-        },
-    },
     {
         name: exports.TOOL_NAMES.BROWSER.SEARCH_AXURE_PROTOTYPE,
         description: 'Search and find content within Axure prototype projects using semantic search',
@@ -183,5 +177,59 @@ exports.TOOL_SCHEMAS = [
             required: ['pageUrl'],
         },
     },
+    {
+        name: exports.TOOL_NAMES.BROWSER.GET_AXURE_IMAGE,
+        description: 'Download images from Axure prototype projects, supporting both absolute and relative image URLs',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                projectUrl: {
+                    type: 'string',
+                    description: 'URL of the Axure project (required when imageUrl is a relative path)',
+                },
+                imageUrl: {
+                    type: 'string',
+                    description: 'URL of the image to download. Can be absolute or relative path (e.g., "images/logo.png")',
+                },
+                storeBase64: {
+                    type: 'boolean',
+                    description: 'Return image in base64 format instead of downloading (default: false)',
+                },
+            },
+            required: ['imageUrl'],
+        },
+    },
+    {
+        name: exports.TOOL_NAMES.BROWSER.GET_PAGE_THEME,
+        description: 'Extract design theme tokens from any web page including colors, typography, spacing, radius, line widths, and shadows. Works with any webpage, not limited to Axure prototypes.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                url: {
+                    type: 'string',
+                    description: 'URL of the page to extract theme from (optional, uses current active tab if not provided)',
+                },
+            },
+            required: [],
+        },
+    },
+    {
+        name: exports.TOOL_NAMES.BROWSER.DOWNLOAD_PAGE_DATA,
+        description: 'Download complete page data package (including theme, DOM structure, styles, screenshot, and images) as a zip file and return the local file path. This tool is useful for AI to analyze page structure and design.',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                url: {
+                    type: 'string',
+                    description: 'URL of the page to download data from (optional, uses current active tab if not provided)',
+                },
+                filename: {
+                    type: 'string',
+                    description: 'Custom filename for the downloaded zip file (optional, default: cursor-export-{timestamp}.zip)',
+                },
+            },
+            required: [],
+        },
+    },
 ];
 //# sourceMappingURL=shared.js.map

package/dist/shared.js.map

@@ -1 +1 @@
-{"version":3,"file":"shared.js","sourceRoot":"","sources":["../src/shared.ts"],"names":[],"mappings":";AAAA,qDAAqD;AACrD,sEAAsE;;;AAEtE,IAAY,iBAmBX;AAnBD,WAAY,iBAAiB;IAC3B,oCAAe,CAAA;IACf,wCAAmB,CAAA;IACnB,kCAAa,CAAA;IACb,wCAAmB,CAAA;IACnB,kCAAa,CAAA;IACb,kCAAa,CAAA;IACb,oCAAe,CAAA;IACf,kDAA6B,CAAA;IAC7B,oEAA+C,CAAA;IAC/C,4CAAuB,CAAA;IACvB,8DAAyC,CAAA;IACzC,oDAAoD;IACpD,sDAAiC,CAAA;IACjC,sDAAiC,CAAA;IACjC,sEAAiD,CAAA;IACjD,qDAAgC,CAAA;IAChC,gDAA2B,CAAA;IAC3B,4DAAuC,CAAA;AACzC,CAAC,EAnBW,iBAAiB,iCAAjB,iBAAiB,QAmB5B;AASD,4CAA4C;AAC5C,mDAAmD;AAEtC,QAAA,UAAU,GAAG;IACxB,OAAO,EAAE;QACP,QAAQ,EAAE,iBAAiB;QAC3B,sBAAsB,EAAE,cAAc;QACtC,iBAAiB,EAAE,mBAAmB;QACtC,yBAAyB,EAAE,sBAAsB;QACjD,cAAc,EAAE,gBAAgB;QAChC,sBAAsB,EAAE,wBAAwB;QAChD,mBAAmB,EAAE,gBAAgB;KACtC;CACF,CAAC;AAKW,QAAA,YAAY,GAAW;IAClC;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,QAAQ;QACjC,WAAW,EAAE,8CAA8C;QAC3D,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0CAA0C,EAAE;gBAChF,SAAS,EAAE;oBACT,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,sEAAsE;iBACpF;gBACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0CAA0C,EAAE;gBAClF,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0CAA0C,EAAE;gBACnF,OAAO,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,WAAW,EACT,2HAA2H;iBAC9H;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,sBAAsB;QAC/C,WAAW,EAAE,+EAA+E;QAC5F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,yDAAyD;iBACvE;aACF;YACD,QAAQ,EAAE,CAAC,YAAY,EAAE,OAAO,CAAC;SAClC;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,iBAAiB;QAC1C,WAAW,EAAE,2EAA2E;QACxF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qGAAqG;iBACnH;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,yBAAyB;QAClD,WAAW,EAAE,gFAAgF;QAC7F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,8CAA8C;iBAC5D;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2FAA2F;iBACzG;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2EAA2E;iBACzF;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4EAA4E;iBAC1F;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,qDAAqD;iBACnE;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,cAAc;QACvC,WAAW,EAAE,4FAA4F;QACzG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uDAAuD;iBACrE;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iHAAiH;iBAC/H;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,sBAAsB;QAC/C,WAAW,EAAE,yEAAyE;QACtF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gDAAgD;iBAC9D;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,mBAAmB;QAC5C,WAAW,EAAE,2DAA2D;QACxE,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qDAAqD;iBACnE;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;CACF,CAAC"}
+{"version":3,"file":"shared.js","sourceRoot":"","sources":["../src/shared.ts"],"names":[],"mappings":";AAAA,qDAAqD;AACrD,sEAAsE;;;AAEtE,IAAY,iBAgCX;AAhCD,WAAY,iBAAiB;IAC3B,oCAAe,CAAA;IACf,wCAAmB,CAAA;IACnB,kCAAa,CAAA;IACb,wCAAmB,CAAA;IACnB,kCAAa,CAAA;IACb,kCAAa,CAAA;IACb,oCAAe,CAAA;IACf,kDAA6B,CAAA;IAC7B,oEAA+C,CAAA;IAC/C,4CAAuB,CAAA;IACvB,8DAAyC,CAAA;IACzC,oDAAoD;IACpD,sDAAiC,CAAA;IACjC,sDAAiC,CAAA;IACjC,sEAAiD,CAAA;IACjD,qDAAgC,CAAA;IAChC,gDAA2B,CAAA;IAC3B,4DAAuC,CAAA;IACvC,2BAA2B;IAC3B,wDAAmC,CAAA;IACnC,sDAAiC,CAAA;IACjC,4DAAuC,CAAA;IACvC,4DAAuC,CAAA;IACvC,0DAAqC,CAAA;IACrC,sDAAsD;IACtD,oDAA+B,CAAA;IAC/B,oDAA+B,CAAA;IAC/B,sDAAiC,CAAA;IACjC,qCAAqC;IACrC,8EAAyD,CAAA;IACzD,oFAA+D,CAAA;AACjE,CAAC,EAhCW,iBAAiB,iCAAjB,iBAAiB,QAgC5B;AASD,4CAA4C;AAC5C,mDAAmD;AAEtC,QAAA,UAAU,GAAG;IACxB,OAAO,EAAE;QACP,sBAAsB,EAAE,cAAc;QACtC,iBAAiB,EAAE,mBAAmB;QACtC,yBAAyB,EAAE,sBAAsB;QACjD,cAAc,EAAE,gBAAgB;QAChC,sBAAsB,EAAE,wBAAwB;QAChD,mBAAmB,EAAE,gBAAgB;QACrC,eAAe,EAAE,iBAAiB;QAClC,cAAc,EAAE,gBAAgB;QAChC,kBAAkB,EAAE,oBAAoB;KACzC;CACF,CAAC;AAKW,QAAA,YAAY,GAAW;IAClC;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,sBAAsB;QAC/C,WAAW,EAAE,+EAA+E;QAC5F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iDAAiD;iBAC/D;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,yDAAyD;iBACvE;aACF;YACD,QAAQ,EAAE,CAAC,YAAY,EAAE,OAAO,CAAC;SAClC;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,iBAAiB;QAC1C,WAAW,EAAE,2EAA2E;QACxF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qGAAqG;iBACnH;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,yBAAyB;QAClD,WAAW,EAAE,gFAAgF;QAC7F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,8CAA8C;iBAC5D;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2FAA2F;iBACzG;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2EAA2E;iBACzF;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4EAA4E;iBAC1F;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,qDAAqD;iBACnE;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,cAAc;QACvC,WAAW,EAAE,4FAA4F;QACzG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,uDAAuD;iBACrE;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iHAAiH;iBAC/H;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,sBAAsB;QAC/C,WAAW,EAAE,yEAAyE;QACtF,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gDAAgD;iBAC9D;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,mBAAmB;QAC5C,WAAW,EAAE,2DAA2D;QACxE,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qCAAqC;iBACnD;gBACD,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,qDAAqD;iBACnE;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,eAAe;QACxC,WAAW,EAAE,iGAAiG;QAC9G,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,sEAAsE;iBACpF;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0FAA0F;iBACxG;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,uEAAuE;iBACrF;aACF;YACD,QAAQ,EAAE,CAAC,UAAU,CAAC;SACvB;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,cAAc;QACvC,WAAW,EAAE,iLAAiL;QAC9L,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2FAA2F;iBACzG;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;IACD;QACE,IAAI,EAAE,kBAAU,CAAC,OAAO,CAAC,kBAAkB;QAC3C,WAAW,EAAE,qNAAqN;QAClO,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,GAAG,EAAE;oBACH,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,2FAA2F;iBACzG;gBACD,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gGAAgG;iBAC9G;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;CACF,CAAC"}

package/dist/utils/logger.d.ts

@@ -0,0 +1,12 @@
+declare class Logger {
+    private logFilePath;
+    constructor();
+    private formatMessage;
+    log(message: string, data?: any): void;
+    error(message: string, data?: any): void;
+    warn(message: string, data?: any): void;
+    debug(message: string, data?: any): void;
+    close(): void;
+}
+export declare const logger: Logger;
+export default logger;

package/dist/utils/logger.js

@@ -0,0 +1,85 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.logger = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+class Logger {
+    constructor() {
+        // Determine log directory - use dist/logs in production, src/logs in development
+        const isDist = __dirname.includes('/dist/');
+        const baseDir = isDist
+            ? path_1.default.join(__dirname, '..', '..')
+            : path_1.default.join(__dirname, '..');
+        const logDir = path_1.default.join(baseDir, 'dist', 'logs');
+        // Create log directory if it doesn't exist
+        if (!fs_1.default.existsSync(logDir)) {
+            fs_1.default.mkdirSync(logDir, { recursive: true });
+        }
+        // Generate log file name with timestamp
+        const timestamp = new Date()
+            .toISOString()
+            .replace(/:/g, '')
+            .replace(/\./g, '_')
+            .replace(/T/, '_')
+            .slice(0, 15); // Format: YYYYMMDD_HHMMSS
+        const logFileName = `native_host_${timestamp}.log`;
+        this.logFilePath = path_1.default.join(logDir, logFileName);
+        this.log('Logger initialized', { logFile: this.logFilePath });
+    }
+    formatMessage(level, message, data) {
+        const timestamp = new Date().toISOString();
+        let formattedMessage = `[${timestamp}] [${level}] ${message}`;
+        if (data !== undefined) {
+            try {
+                formattedMessage += ' ' + JSON.stringify(data, null, 2);
+            }
+            catch (error) {
+                formattedMessage += ' [Unable to stringify data]';
+            }
+        }
+        return formattedMessage + '\n';
+    }
+    log(message, data) {
+        try {
+            // Use synchronous write to ensure logs are immediately written
+            fs_1.default.appendFileSync(this.logFilePath, this.formatMessage('INFO', message, data));
+        }
+        catch (error) {
+            // Silently fail to avoid breaking the application
+        }
+    }
+    error(message, data) {
+        try {
+            fs_1.default.appendFileSync(this.logFilePath, this.formatMessage('ERROR', message, data));
+        }
+        catch (error) {
+            // Silently fail
+        }
+    }
+    warn(message, data) {
+        try {
+            fs_1.default.appendFileSync(this.logFilePath, this.formatMessage('WARN', message, data));
+        }
+        catch (error) {
+            // Silently fail
+        }
+    }
+    debug(message, data) {
+        try {
+            fs_1.default.appendFileSync(this.logFilePath, this.formatMessage('DEBUG', message, data));
+        }
+        catch (error) {
+            // Silently fail
+        }
+    }
+    close() {
+        // No-op for sync logging
+    }
+}
+// Export singleton instance
+exports.logger = new Logger();
+exports.default = exports.logger;
+//# sourceMappingURL=logger.js.map

package/dist/utils/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AAExB,MAAM,MAAM;IAGR;QACI,iFAAiF;QACjF,MAAM,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,MAAM;YAClB,CAAC,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC;YAClC,CAAC,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAElD,2CAA2C;QAC3C,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,YAAE,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,wCAAwC;QACxC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE;aACvB,WAAW,EAAE;aACb,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;aACjB,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;aACnB,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;aACjB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,0BAA0B;QAE7C,MAAM,WAAW,GAAG,eAAe,SAAS,MAAM,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAElD,IAAI,CAAC,GAAG,CAAC,oBAAoB,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAClE,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,OAAe,EAAE,IAAU;QAC5D,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,IAAI,gBAAgB,GAAG,IAAI,SAAS,MAAM,KAAK,KAAK,OAAO,EAAE,CAAC;QAE9D,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACrB,IAAI,CAAC;gBACD,gBAAgB,IAAI,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC5D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,gBAAgB,IAAI,6BAA6B,CAAC;YACtD,CAAC;QACL,CAAC;QAED,OAAO,gBAAgB,GAAG,IAAI,CAAC;IACnC,CAAC;IAEM,GAAG,CAAC,OAAe,EAAE,IAAU;QAClC,IAAI,CAAC;YACD,+DAA+D;YAC/D,YAAE,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACnF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,kDAAkD;QACtD,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,OAAe,EAAE,IAAU;QACpC,IAAI,CAAC;YACD,YAAE,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,gBAAgB;QACpB,CAAC;IACL,CAAC;IAEM,IAAI,CAAC,OAAe,EAAE,IAAU;QACnC,IAAI,CAAC;YACD,YAAE,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACnF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,gBAAgB;QACpB,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,OAAe,EAAE,IAAU;QACpC,IAAI,CAAC;YACD,YAAE,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,gBAAgB;QACpB,CAAC;IACL,CAAC;IAEM,KAAK;QACR,yBAAyB;IAC7B,CAAC;CACJ;AAED,4BAA4B;AACf,QAAA,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;AACnC,kBAAe,cAAM,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "axhub-mcp-bridge",
-  "version": "1.0.34",
+  "version": "1.0.36",
   "description": "Chrome Native-Messaging host (Node)",
   "main": "dist/index.js",
   "bin": {
@@ -16,7 +16,10 @@
     "lint:fix": "eslint 'src/**/*.{js,ts}' --fix",
     "format": "prettier --write 'src/**/*.{js,ts,json}'",
     "register:dev": "node dist/scripts/register-dev.js",
-    "postinstall": "node dist/scripts/postinstall.js"
+    "postinstall": "node dist/scripts/postinstall.js",
+    "inspector": "npx -y @modelcontextprotocol/inspector@latest node dist/mcp/mcp-server-stdio.js",
+    "inspector:cli": "npx -y @modelcontextprotocol/inspector@latest --cli node dist/mcp/mcp-server-stdio.js",
+    "inspector:http": "npx -y @modelcontextprotocol/inspector@latest http://127.0.0.1:12307/mcp"
   },
   "files": [
     "dist"
@@ -33,7 +36,8 @@
     "fastify": "^5.3.2",
     "is-admin": "^4.0.0",
     "pino": "^9.6.0",
-    "uuid": "^11.1.0"
+    "uuid": "^11.1.0",
+    "ws": "^8.18.3"
   },
   "devDependencies": {
     "@jest/globals": "^29.7.0",
@@ -41,6 +45,7 @@
     "@types/jest": "^29.5.14",
     "@types/node": "^22.15.3",
     "@types/supertest": "^6.0.3",
+    "@types/ws": "^8.18.1",
     "@typescript-eslint/parser": "^8.31.1",
     "cross-env": "^7.0.3",
     "husky": "^9.1.7",
@@ -59,7 +64,7 @@
     }
   },
   "lint-staged": {
-    "*.{js,ts}": [ 
+    "*.{js,ts}": [
       "eslint --fix",
       "prettier --write"
     ],
@@ -67,4 +72,4 @@
       "prettier --write"
     ]
   }
-}
+}