aipex-mcp-bridge 1.0.2 → 1.0.3

package/dist/bridge.js

@@ -51,6 +51,30 @@ var cachedToolsTimestamp = 0;
 var pingInterval = null;
 var lastConnectionTime = null;
 var lastDisconnectReason = null;
+var connectionWaiters = [];
+var WAIT_FOR_CONNECTION_MS = 15e3;
+function waitForConnection() {
+  return new Promise((resolve, reject) => {
+    if (aipexReady && aipexSocket?.readyState === WebSocket.OPEN) {
+      resolve();
+      return;
+    }
+    const timer = setTimeout(() => {
+      const idx = connectionWaiters.indexOf(wrapped);
+      if (idx !== -1) connectionWaiters.splice(idx, 1);
+      reject(new Error("Timed out waiting for AIPex to connect"));
+    }, WAIT_FOR_CONNECTION_MS);
+    const wrapped = () => {
+      clearTimeout(timer);
+      resolve();
+    };
+    connectionWaiters.push(wrapped);
+  });
+}
+function notifyConnectionWaiters() {
+  const waiters = connectionWaiters.splice(0);
+  for (const w of waiters) w();
+}
 var nextAipexId = 1;
 var aipexPending = /* @__PURE__ */ new Map();
 function respond(id, result) {
@@ -137,6 +161,7 @@ async function doAipexHandshake(socket) {
   aipexReady = true;
   lastConnectionTime = Date.now();
   log(`Handshake complete. ${cachedTools.length} tools available.`);
+  notifyConnectionWaiters();
 }
 async function doAipexHandshakeWithRetry(socket) {
   for (let attempt = 1; attempt <= HANDSHAKE_RETRY_COUNT; attempt++) {
@@ -239,14 +264,19 @@ async function handleAgentRequest(req) {
   }
   if (method === "tools/call") {
     if (!aipexReady || !aipexSocket || aipexSocket.readyState !== WebSocket.OPEN) {
-      const ctx = lastConnectionTime ? ` Last connected: ${new Date(lastConnectionTime).toISOString()}.` : "";
-      const reason = lastDisconnectReason ? ` Disconnect reason: ${lastDisconnectReason}.` : "";
-      respondError(
-        id,
-        -32e3,
-        `AIPex extension not connected.${ctx}${reason} Open AIPex Options and connect to ws://localhost:${WS_PORT}`
-      );
-      return;
+      log(`AIPex not ready, waiting up to ${WAIT_FOR_CONNECTION_MS / 1e3}s for reconnect...`);
+      try {
+        await waitForConnection();
+      } catch {
+        const ctx = lastConnectionTime ? ` Last connected: ${new Date(lastConnectionTime).toISOString()}.` : "";
+        const reason = lastDisconnectReason ? ` Disconnect reason: ${lastDisconnectReason}.` : "";
+        respondError(
+          id,
+          -32e3,
+          `AIPex extension not connected.${ctx}${reason} Open AIPex Options and connect to ws://localhost:${WS_PORT}`
+        );
+        return;
+      }
     }
     try {
       const result = await sendToAipex(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aipex-mcp-bridge",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "MCP bridge that connects AI agents (Cursor, Claude, VS Code Copilot, etc.) to the AIPex browser extension via WebSocket",
   "type": "module",
   "bin": {