abu-browser-bridge 0.1.0 → 0.2.0

package/dist/wsServer.d.ts

@@ -1,11 +1,15 @@
 /**
  * WebSocket server that accepts connections from the Chrome Extension.
  * Routes requests from MCP tools to the extension and returns responses.
+ *
+ * Also exposes a lightweight HTTP discovery endpoint on a fixed port
+ * so the Chrome Extension can reliably find the WS port.
  */
 import type { BridgeResponse } from './types.js';
 /**
- * Start the WebSocket server. If the default port is occupied (by a stale bridge),
- * try a few alternative ports. The Extension will discover the port via a well-known range.
+ * Start the WebSocket server. If the default port is occupied,
+ * try a few alternative ports. The Extension discovers the port
+ * via the HTTP discovery endpoint.
  */
 export declare function startWSServer(port?: number): Promise<number>;
 /**

package/dist/wsServer.js

@@ -1,14 +1,22 @@
 /**
  * WebSocket server that accepts connections from the Chrome Extension.
  * Routes requests from MCP tools to the extension and returns responses.
+ *
+ * Also exposes a lightweight HTTP discovery endpoint on a fixed port
+ * so the Chrome Extension can reliably find the WS port.
  */
 import { WebSocketServer, WebSocket } from 'ws';
+import { createServer } from 'http';
 import { createConnection } from 'net';
-const DEFAULT_PORT = 9876;
-const HEARTBEAT_INTERVAL = 30_000;
+const DEFAULT_WS_PORT = 9876;
+const DISCOVERY_PORT = 9875;
+const HEARTBEAT_INTERVAL = 15_000; // 15s — more frequent for faster disconnect detection
+const PONG_TIMEOUT = 5_000; // If no pong within 5s, consider dead
 let wss = null;
+let discoveryServer = null;
 let extensionSocket = null;
 let heartbeatTimer = null;
+let pongReceived = true;
 const pendingRequests = new Map();
 let requestCounter = 0;
 let activePort = null;
@@ -17,7 +25,6 @@ function generateId() {
 }
 /**
  * Check if a port is in use by trying to connect to it.
- * Cross-platform — works on macOS, Windows, Linux.
  */
 function isPortInUse(port, host) {
     return new Promise((resolve) => {
@@ -36,11 +43,65 @@ function isPortInUse(port, host) {
         });
     });
 }
+// --- HTTP Discovery Endpoint ---
 /**
- * Start the WebSocket server. If the default port is occupied (by a stale bridge),
- * try a few alternative ports. The Extension will discover the port via a well-known range.
+ * Start the HTTP discovery server on a fixed well-known port.
+ * Chrome Extension queries this to find the actual WS port.
+ *
+ * GET /status → { wsPort, pid, extensionConnected, uptime }
  */
-export async function startWSServer(port = DEFAULT_PORT) {
+function startDiscoveryServer() {
+    return new Promise((resolve, reject) => {
+        const startTime = Date.now();
+        discoveryServer = createServer((req, res) => {
+            // CORS headers for Chrome Extension fetch
+            res.setHeader('Access-Control-Allow-Origin', '*');
+            res.setHeader('Access-Control-Allow-Methods', 'GET, OPTIONS');
+            res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+            if (req.method === 'OPTIONS') {
+                res.writeHead(204);
+                res.end();
+                return;
+            }
+            if (req.url === '/status' || req.url === '/') {
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({
+                    wsPort: activePort,
+                    pid: process.pid,
+                    extensionConnected: isExtensionConnected(),
+                    uptime: Math.round((Date.now() - startTime) / 1000),
+                    version: '0.1.1',
+                }));
+                return;
+            }
+            res.writeHead(404);
+            res.end('Not found');
+        });
+        discoveryServer.on('error', (err) => {
+            if (err.code === 'EADDRINUSE') {
+                console.error(`[abu-bridge] Discovery port ${DISCOVERY_PORT} in use — old bridge still running?`);
+                // Not fatal — extension can still fall back to port scanning
+                resolve();
+            }
+            else {
+                reject(err);
+            }
+        });
+        discoveryServer.listen(DISCOVERY_PORT, '127.0.0.1', () => {
+            console.error(`[abu-bridge] Discovery endpoint: http://127.0.0.1:${DISCOVERY_PORT}/status`);
+            resolve();
+        });
+    });
+}
+// --- WebSocket Server ---
+/**
+ * Start the WebSocket server. If the default port is occupied,
+ * try a few alternative ports. The Extension discovers the port
+ * via the HTTP discovery endpoint.
+ */
+export async function startWSServer(port = DEFAULT_WS_PORT) {
+    // Start discovery endpoint first (non-blocking if port taken)
+    await startDiscoveryServer();
     const MAX_RETRIES = 5;
     for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {
         const tryPort = port + attempt;
@@ -56,7 +117,6 @@ export async function startWSServer(port = DEFAULT_PORT) {
         }
         catch (err) {
             const msg = err instanceof Error ? err.message : String(err);
-            // EADDRINUSE can still happen in a race condition
             if (msg.includes('EADDRINUSE') && attempt < MAX_RETRIES - 1) {
                 console.error(`[abu-bridge] Port ${tryPort} race condition, trying next...`);
                 continue;
@@ -87,6 +147,11 @@ function listenOnPort(port) {
                 extensionSocket.close(1000, 'Replaced by new connection');
             }
             extensionSocket = ws;
+            pongReceived = true;
+            // Handle pong responses for heartbeat
+            ws.on('pong', () => {
+                pongReceived = true;
+            });
             ws.on('message', (data) => {
                 try {
                     const msg = JSON.parse(data.toString());
@@ -117,6 +182,14 @@ function listenOnPort(port) {
 function startHeartbeat() {
     heartbeatTimer = setInterval(() => {
         if (extensionSocket && extensionSocket.readyState === WebSocket.OPEN) {
+            if (!pongReceived) {
+                // No pong since last ping — connection is dead
+                console.error('[abu-bridge] Extension not responding to heartbeat, closing connection');
+                extensionSocket.terminate();
+                extensionSocket = null;
+                return;
+            }
+            pongReceived = false;
             extensionSocket.ping();
         }
     }, HEARTBEAT_INTERVAL);
@@ -180,5 +253,9 @@ export function stopWSServer() {
         wss.close();
         wss = null;
     }
+    if (discoveryServer) {
+        discoveryServer.close();
+        discoveryServer = null;
+    }
     activePort = null;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "abu-browser-bridge",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "MCP Server that bridges Abu AI assistant with Chrome Extension for browser automation",
   "type": "module",
   "license": "MIT",
@@ -11,7 +11,7 @@
   },
   "keywords": ["mcp", "browser", "automation", "chrome-extension", "abu"],
   "bin": {
-    "abu-browser-bridge": "./dist/index.js"
+    "abu-browser-bridge": "dist/index.js"
   },
   "files": [
     "dist"