up-mcp-bridge 1.0.0 → 1.0.2

package/README.md

@@ -209,9 +209,7 @@ You can specify multiple `--ignore-tool` flags to ignore different patterns. Exa
       ]
 ```
 
-### Auto-Reconnect (Experimental)
-
-⚠️ **This feature is experimental and has known issues being investigated.**
+### Auto-Reconnect
 
 To enable automatic reconnection when the remote server restarts, add the `--auto-reconnect` flag:
 
@@ -231,11 +229,20 @@ To enable automatic reconnection when the remote server restarts, add the `--aut
 ```
 
 This enables:
+- **Session-aware ping**: Detects stale SSE sessions via `/ping?sessionId=XXX` endpoint
 - Automatic reconnection with exponential backoff when the server becomes unavailable
 - Message queuing during reconnection attempts
 - MCP session re-initialization after successful reconnection
 - Transparent handling for the MCP client
 
+**How session-aware ping works:**
+1. When connected, the bridge extracts the `sessionId` from the SSE endpoint URL
+2. Periodic health checks include the `sessionId` parameter
+3. If the server returns `410 Gone`, the session has expired (e.g., server restarted)
+4. The bridge immediately triggers reconnection instead of waiting for SSE errors
+
+This ensures fast recovery when the remote server restarts, avoiding tool call timeouts.
+
 **Default parameters:**
 - Max reconnection attempts: 20
 - Base delay: 1000ms
@@ -254,18 +261,7 @@ This enables:
 ]
 ```
 
-#### Known Issues Under Investigation
-
-1. **MCP calls may hang after server restart**: In some scenarios, MCP tool calls hang indefinitely after the remote server restarts, even though the proxy reports a successful reconnection. A full restart cycle (stop client → restart server → start client) resolves the issue, but this defeats the purpose of auto-reconnect.
-
-2. **npx execution issues**: When using `npx up-mcp-bridge`, environment variables (like `NODE_TLS_REJECT_UNAUTHORIZED`) may not propagate correctly to the spawned process, causing SSL issues with self-signed certificates.
-
-3. **HTTP vs HTTPS behavior**: The proxy behaves differently when connecting via HTTP vs HTTPS. HTTP connections (even on localhost) may exhibit different hanging behavior.
-
-**Status**: These issues are being actively investigated. If you encounter problems, please report them with:
-- Your configuration (sanitized)
-- Steps to reproduce
-- Logs from the proxy (`--debug` flag)
+**Note:** Session-aware ping requires server support (e.g., ultraPRO Desktop v1.0+). Servers without this feature will still work with standard ping-based health checks.
 
 ### Transport Strategies
 

package/dist/{chunk-EOYXIWZ7.js → chunk-3OI3JDYP.js}

@@ -18043,7 +18043,7 @@ var Client = class extends Protocol {
 };
 
 // package.json
-var version2 = "1.0.0";
+var version2 = "1.0.1";
 
 // node_modules/pkce-challenge/dist/index.node.js
 var crypto;
@@ -20026,10 +20026,13 @@ var PING_TIMEOUT_MS = 2e3;
 var MAX_PING_FAILURES = 3;
 var MAX_SSE_ERRORS_BEFORE_RECONNECT = 2;
 var SSE_ERROR_PATTERNS = ["timeout", "terminated", "aborted", "network", "ECONNRESET", "ECONNREFUSED"];
-async function pingServer(serverUrl) {
+async function pingServer(serverUrl, sessionId) {
   try {
     const url2 = new URL(serverUrl);
-    const pingUrl = `${url2.protocol}//${url2.host}/ping`;
+    let pingUrl = `${url2.protocol}//${url2.host}/ping`;
+    if (sessionId) {
+      pingUrl += `?sessionId=${encodeURIComponent(sessionId)}`;
+    }
     const controller = new AbortController();
     const timeout = setTimeout(() => controller.abort(), PING_TIMEOUT_MS);
     try {
@@ -20038,13 +20041,25 @@ async function pingServer(serverUrl) {
         method: "GET"
       });
       clearTimeout(timeout);
-      return response.ok;
+      if (response.status === 410) {
+        return { alive: true, sessionExpired: true };
+      }
+      return { alive: response.ok, sessionExpired: false };
     } catch (e) {
       clearTimeout(timeout);
-      return false;
+      return { alive: false, sessionExpired: false };
     }
   } catch (e) {
-    return false;
+    return { alive: false, sessionExpired: false };
+  }
+}
+function extractSessionId(transport) {
+  try {
+    const endpoint = transport._endpoint;
+    if (!endpoint) return void 0;
+    return endpoint.searchParams?.get("sessionId") || void 0;
+  } catch {
+    return void 0;
   }
 }
 var pid = process.pid;
@@ -20154,6 +20169,17 @@ function mcpProxy({
   let consecutiveTimeouts = 0;
   let consecutiveSseErrors = 0;
   const pendingRequests = /* @__PURE__ */ new Map();
+  let currentSessionId;
+  let sessionIdExtracted = false;
+  function tryExtractSessionId() {
+    if (sessionIdExtracted) return;
+    const sessionId = extractSessionId(currentTransportToServer);
+    if (sessionId) {
+      currentSessionId = sessionId;
+      sessionIdExtracted = true;
+      log(`[Session] Extracted sessionId from transport: ${sessionId}`);
+    }
+  }
   let savedInitializeMessage = null;
   let initializeIdCounter = 1e6;
   async function reinitializeMcpSession(transport) {
@@ -20287,8 +20313,20 @@ function mcpProxy({
         return;
       }
       debugLog(`[Ping] Checking server health for request ${message.id}...`);
-      const isAlive = await pingServer(serverUrl);
-      if (isAlive) {
+      const pingResult = await pingServer(serverUrl, currentSessionId);
+      if (pingResult.sessionExpired) {
+        log(`[Ping] Session expired for request ${message.id}, triggering immediate reconnection`);
+        clearInterval(pingInterval);
+        connectionHealthy = false;
+        const pending = pendingRequests.get(message.id);
+        if (pending) {
+          pendingRequests.delete(message.id);
+          queueMessageForRetry(pending.message);
+        }
+        currentTransportToServer.close().catch(onServerError);
+        return;
+      }
+      if (pingResult.alive) {
         if (pingFailures > 0) {
           debugLog(`[Ping] Server recovered, resetting failure counter (was ${pingFailures})`);
           pingFailures = 0;
@@ -20386,6 +20424,7 @@ function mcpProxy({
     serverTransport.onmessage = (_message) => {
       const message = messageTransformer.interceptResponse(_message);
       log("[Remote\u2192Local]", message.method || message.id);
+      tryExtractSessionId();
       if (message.id !== void 0) {
         clearRequestTracking(message.id);
       }
@@ -20437,6 +20476,9 @@ function mcpProxy({
             connectionHealthy = true;
             consecutiveTimeouts = 0;
             consecutiveSseErrors = 0;
+            sessionIdExtracted = false;
+            currentSessionId = void 0;
+            tryExtractSessionId();
             log(`Flushing ${pendingMessages.length} queued messages...`);
             while (pendingMessages.length > 0) {
               const pending = pendingMessages.shift();
@@ -20991,17 +21033,28 @@ async function parseCommandLineArgs(args, usage) {
   };
 }
 function setupSignalHandlers(cleanup) {
-  process.on("SIGINT", async () => {
-    log("\nShutting down...");
-    await cleanup();
-    process.exit(0);
-  });
+  let isShuttingDown = false;
+  const gracefulShutdown = async (signal) => {
+    if (isShuttingDown) {
+      log(`[Shutdown] Already shutting down, ignoring ${signal}`);
+      return;
+    }
+    isShuttingDown = true;
+    log(`[Shutdown] Received ${signal}, cleaning up...`);
+    try {
+      await cleanup();
+      log("[Shutdown] Cleanup complete, exiting with code 0");
+      process.exit(0);
+    } catch (error2) {
+      log("[Shutdown] Error during cleanup:", error2);
+      process.exit(1);
+    }
+  };
+  process.on("SIGINT", () => gracefulShutdown("SIGINT"));
+  process.on("SIGTERM", () => gracefulShutdown("SIGTERM"));
   process.stdin.resume();
-  process.stdin.on("end", async () => {
-    log("\nShutting down...");
-    await cleanup();
-    process.exit(0);
-  });
+  process.stdin.on("end", () => gracefulShutdown("stdin-end"));
+  process.stdin.on("close", () => gracefulShutdown("stdin-close"));
 }
 function getServerUrlHash(serverUrl, authorizeResource, headers) {
   const parts = [serverUrl];

package/dist/client.js

@@ -12,7 +12,7 @@ import {
   parseCommandLineArgs,
   setupSignalHandlers,
   version
-} from "./chunk-EOYXIWZ7.js";
+} from "./chunk-3OI3JDYP.js";
 
 // src/client.ts
 import { EventEmitter } from "events";

package/dist/proxy.js

@@ -10,7 +10,7 @@ import {
   mcpProxy,
   parseCommandLineArgs,
   setupSignalHandlers
-} from "./chunk-EOYXIWZ7.js";
+} from "./chunk-3OI3JDYP.js";
 
 // src/proxy.ts
 import { EventEmitter } from "events";
@@ -172,8 +172,8 @@ async function runProxy(serverUrl, callbackPort, headers, transportStrategy = "h
     }
     log("Press Ctrl+C to exit");
     const cleanup = async () => {
-      await remoteTransport.close();
       await localTransport.close();
+      await remoteTransport.close();
       if (server) {
         server.close();
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "up-mcp-bridge",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Remote proxy for MCP with auto-reconnect support. Fork of mcp-remote with transparent server restart handling.",
   "keywords": [
     "mcp",