toolbox-co-mcp-bridge 1.0.2 → 1.0.4

package/index.js

@@ -1,20 +1,24 @@
 #!/usr/bin/env node
-import EventSource from "eventsource";
 import fetch from "node-fetch";
 import readline from "readline";
 
-// The hardcoded remote SSE endpoint for ToolBox Co. SaaS tools
 const sseUrl = "https://toolbox-mcp-1058072229791.us-central1.run.app/sse";
+const finalSseUrl = `${sseUrl}?t=${Date.now()}`;
 
-const es = new EventSource(sseUrl);
 let postEndpoint = '';
 const messageQueue = [];
+let gcrCookie = '';
 
 async function sendPost(msg) {
     try {
+        const headers = { 'Content-Type': 'application/json' };
+        if (gcrCookie) {
+            headers['Cookie'] = gcrCookie;
+        }
+        
         const response = await fetch(postEndpoint, {
             method: 'POST',
-            headers: { 'Content-Type': 'application/json' },
+            headers,
             body: msg
         });
         if (!response.ok) {
@@ -27,31 +31,59 @@ async function sendPost(msg) {
     }
 }
 
-// When the MCP server initializes the connection, it emits an 'endpoint' event
-// containing the specific URI we must POST messages to.
-es.addEventListener("endpoint", async (e) => {
-    try {
-        postEndpoint = new URL(e.data, sseUrl).toString();
-        
-        // Flush any queued messages that arrived before we got the endpoint
-        for (const msg of messageQueue) {
-            await sendPost(msg);
+async function connectSSE() {
+    console.error(`[ToolBox Bridge] Opening strict bypass connection to: ${finalSseUrl}`);
+    const res = await fetch(finalSseUrl);
+    
+    // Extract Cloud Run session affinity cookie
+    const cookies = res.headers.raw()['set-cookie'] || [];
+    for (const c of cookies) {
+        if (c.startsWith('GCRAffinity=')) {
+            gcrCookie = c.split(';')[0];
+            console.error(`[ToolBox Bridge] Captured Cloud Run Affinity Cookie: ${gcrCookie.substring(0, 20)}...`);
         }
-        messageQueue.length = 0;
-    } catch (err) {
-        console.error("Failed to parse POST endpoint:", err);
     }
-});
 
-// Any messages received from the remote MCP server must be forwarded 
-// locally to Claude Desktop via standard output (stdout).
-es.onmessage = (e) => {
-    process.stdout.write(e.data + '\n');
-};
+    let buffer = '';
+    for await (const chunk of res.body) {
+        buffer += chunk.toString();
+        
+        // Simple SSE parser
+        const events = buffer.split('\n\n');
+        buffer = events.pop() || ''; // Keep the last incomplete chunk in buffer
+        
+        for (const eventBlock of events) {
+            if (!eventBlock.trim()) continue;
+            
+            let eventType = 'message';
+            let data = '';
+            
+            for (const line of eventBlock.split('\n')) {
+                if (line.startsWith('event:')) {
+                    eventType = line.substring(6).trim();
+                } else if (line.startsWith('data:')) {
+                    // MCP JSON RPC payloads might contain multiple chunks
+                    data += line.substring(5).trim();
+                }
+            }
+            
+            if (eventType === 'endpoint') {
+                postEndpoint = new URL(data, sseUrl).toString();
+                console.error(`[ToolBox Bridge] Successfully received dynamic POST endpoint: ${postEndpoint}`);
+                
+                for (const msg of messageQueue) {
+                    await sendPost(msg);
+                }
+                messageQueue.length = 0;
+            } else if (eventType === 'message') {
+                process.stdout.write(data + '\n');
+            }
+        }
+    }
+    console.error("[ToolBox Bridge] SSE connection closed by server.");
+}
 
-es.onerror = (e) => {
-    console.error("SSE connection error:", e);
-};
+connectSSE().catch(e => console.error("SSE connection error:", e));
 
 // Use readline for perfect line-delimited stream reading, immune to TCP chunk truncation
 const rl = readline.createInterface({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "toolbox-co-mcp-bridge",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "A bridge proxy to connect local STDIO-based MCP clients (like Claude Desktop) to the remote ToolBox Co. SSE endpoint.",
   "type": "module",
   "bin": {
@@ -19,7 +19,6 @@
   "author": "ToolBox Co.",
   "license": "MIT",
   "dependencies": {
-    "eventsource": "^2.0.2",
     "node-fetch": "^3.3.2"
   }
 }

package/test_sse.mjs

@@ -0,0 +1,15 @@
+import fetch from "node-fetch";
+
+async function main() {
+    const sseUrl = "https://toolbox-mcp-1058072229791.us-central1.run.app/sse";
+    console.log("Fetching", sseUrl);
+    const res = await fetch(sseUrl);
+    console.log("Status:", res.status);
+    console.log("Headers:", JSON.stringify([...res.headers.entries()]));
+
+    for await (const chunk of res.body) {
+        console.log("Chunk:", chunk.toString());
+        break; // just read the first chunk to see the endpoint event
+    }
+}
+main().catch(console.error);