@mysten-incubation/memwal-mcp 0.0.13 → 0.0.14-dev.1

package/dist/streamable.js

@@ -0,0 +1,136 @@
+/**
+ * Streamable HTTP transport for the stdio bridge.
+ *
+ * The legacy transport splits a call in two: POST to `/api/mcp/messages`, then
+ * wait for the reply to arrive on a separate `/api/mcp/sse` stream. Everything
+ * expensive in `bridge.ts` follows from that split — the `inFlight` map, the
+ * `sent` flag, the 404-means-never-ran reset, the idle watchdog, and the
+ * replay-on-reconnect path — because a POST can succeed while its reply is
+ * lost, and the bridge cannot tell that from a call still running.
+ *
+ * Streamable HTTP (MCP 2025-06) collapses that: one endpoint, and the reply
+ * comes back on the same request. The relayer has served it since
+ * `mcp_proxy.rs:751` ("Single endpoint that supersedes the SSE+POST split");
+ * only the bridge was still on the old transport.
+ *
+ * This module wraps the MCP SDK's own client transport rather than hand-rolling
+ * the protocol: session-id round-tripping, the optional SSE upgrade on a POST
+ * response, and resumption tokens are all spec details that are easy to get
+ * subtly wrong and that the SDK already implements.
+ */
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { log } from "./logger.js";
+/**
+ * Resolve the transport from `MEMWAL_MCP_TRANSPORT`.
+ *
+ * Defaults to `sse`, the transport every released bridge has used. Streamable
+ * HTTP is opt-in until it has production mileage: this runs on users' machines
+ * against their real memories, so the new path proves itself before it becomes
+ * the one that runs by default.
+ *
+ * An unrecognised value falls back rather than throwing — a typo in a user's
+ * MCP config must not stop their memory from working.
+ */
+export function resolveTransport(raw) {
+    const value = raw?.trim().toLowerCase();
+    if (!value)
+        return "sse";
+    if (value === "http" || value === "streamable" || value === "streamable-http") {
+        return "http";
+    }
+    if (value === "sse")
+        return "sse";
+    log.warn("bridge.transport_unrecognized", { value, using: "sse" });
+    return "sse";
+}
+/**
+ * The Streamable HTTP endpoint for a relayer base URL.
+ *
+ * `/api/mcp` — the same base the SSE transport hangs `/api/mcp/sse` and
+ * `/api/mcp/messages` off, minus the split.
+ */
+export function streamableUrl(relayerUrl) {
+    return `${relayerUrl.replace(/\/+$/, "")}/api/mcp`;
+}
+/** HTTP status carried on the SDK's transport error, when it has one. */
+function statusOf(err) {
+    const code = err?.code;
+    return typeof code === "number" ? code : 0;
+}
+export async function openStreamableSession(relayerUrl, creds, extraHeaders = {}) {
+    const url = streamableUrl(relayerUrl);
+    // Queue + waiter rather than an event emitter, so a message that arrives
+    // before `runBridge` pulls from the iterator is buffered instead of
+    // dropped. The SSE path does the same thing for the same reason.
+    const queue = [];
+    let wake = null;
+    let closed = false;
+    const push = (msg) => {
+        queue.push(msg);
+        const resume = wake;
+        wake = null;
+        resume?.();
+    };
+    const finish = () => {
+        closed = true;
+        const resume = wake;
+        wake = null;
+        resume?.();
+    };
+    const transport = new StreamableHTTPClientTransport(new URL(url), {
+        requestInit: {
+            headers: {
+                authorization: `Bearer ${creds.delegatePrivateKey}`,
+                "x-memwal-account-id": creds.accountId,
+                ...extraHeaders,
+            },
+        },
+    });
+    transport.onmessage = push;
+    transport.onclose = finish;
+    transport.onerror = (err) => {
+        log.warn("bridge.streamable_error", { err: String(err) });
+        // Not `finish()` — the SDK transport reconnects its own stream, and
+        // tearing the session down on a transient read error is what the SSE
+        // watchdog did wrong.
+    };
+    await transport.start();
+    const iter = {
+        async next() {
+            while (queue.length === 0) {
+                if (closed)
+                    return { value: undefined, done: true };
+                await new Promise((resolve) => (wake = resolve));
+            }
+            return { value: queue.shift(), done: false };
+        },
+    };
+    return {
+        postUrl: url,
+        async send(msg) {
+            try {
+                await transport.send(msg);
+                return 200;
+            }
+            catch (err) {
+                const status = statusOf(err);
+                log.warn("bridge.streamable_send_failed", {
+                    status,
+                    err: String(err),
+                });
+                // Surface the status rather than throwing: `postIfCurrent`
+                // routes on it, and a 404 specifically means the message was
+                // discarded rather than run.
+                return status;
+            }
+        },
+        iter,
+        abort: () => {
+            void transport.close().catch(() => {
+                /* already gone */
+            });
+            finish();
+        },
+    };
+}
+//# sourceMappingURL=streamable.js.map

package/dist/streamable.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamable.js","sourceRoot":"","sources":["../src/streamable.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG;AACH,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAInG,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAKlC;;;;;;;;;;GAUG;AACH,MAAM,UAAU,gBAAgB,CAAC,GAAuB;IACpD,MAAM,KAAK,GAAG,GAAG,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACxC,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,YAAY,IAAI,KAAK,KAAK,iBAAiB,EAAE,CAAC;QAC5E,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,IAAI,KAAK,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC;IAClC,GAAG,CAAC,IAAI,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,OAAO,KAAK,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,aAAa,CAAC,UAAkB;IAC5C,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC;AACvD,CAAC;AA6BD,yEAAyE;AACzE,SAAS,QAAQ,CAAC,GAAY;IAC1B,MAAM,IAAI,GAAI,GAAiC,EAAE,IAAI,CAAC;IACtD,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACvC,UAAkB,EAClB,KAAwB,EACxB,eAAuC,EAAE;IAEzC,MAAM,GAAG,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;IAEtC,yEAAyE;IACzE,oEAAoE;IACpE,iEAAiE;IACjE,MAAM,KAAK,GAAqB,EAAE,CAAC;IACnC,IAAI,IAAI,GAAwB,IAAI,CAAC;IACrC,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,MAAM,IAAI,GAAG,CAAC,GAAmB,EAAE,EAAE;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC;QACpB,IAAI,GAAG,IAAI,CAAC;QACZ,MAAM,EAAE,EAAE,CAAC;IACf,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,GAAG,EAAE;QAChB,MAAM,GAAG,IAAI,CAAC;QACd,MAAM,MAAM,GAAG,IAAI,CAAC;QACpB,IAAI,GAAG,IAAI,CAAC;QACZ,MAAM,EAAE,EAAE,CAAC;IACf,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE;QAC9D,WAAW,EAAE;YACT,OAAO,EAAE;gBACL,aAAa,EAAE,UAAU,KAAK,CAAC,kBAAkB,EAAE;gBACnD,qBAAqB,EAAE,KAAK,CAAC,SAAS;gBACtC,GAAG,YAAY;aAClB;SACJ;KACJ,CAAC,CAAC;IAEH,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC;IAC3B,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;IAC3B,SAAS,CAAC,OAAO,GAAG,CAAC,GAAG,EAAE,EAAE;QACxB,GAAG,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1D,oEAAoE;QACpE,qEAAqE;QACrE,sBAAsB;IAC1B,CAAC,CAAC;IAEF,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;IAExB,MAAM,IAAI,GAAkC;QACxC,KAAK,CAAC,IAAI;YACN,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,IAAI,MAAM;oBAAE,OAAO,EAAE,KAAK,EAAE,SAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAC7D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC;YAC3D,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QAClD,CAAC;KACJ,CAAC;IAEF,OAAO;QACH,OAAO,EAAE,GAAG;QACZ,KAAK,CAAC,IAAI,CAAC,GAAG;YACV,IAAI,CAAC;gBACD,MAAM,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC1B,OAAO,GAAG,CAAC;YACf,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACX,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;gBAC7B,GAAG,CAAC,IAAI,CAAC,+BAA+B,EAAE;oBACtC,MAAM;oBACN,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC;iBACnB,CAAC,CAAC;gBACH,2DAA2D;gBAC3D,6DAA6D;gBAC7D,6BAA6B;gBAC7B,OAAO,MAAM,CAAC;YAClB,CAAC;QACL,CAAC;QACD,IAAI;QACJ,KAAK,EAAE,GAAG,EAAE;YACR,KAAK,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;gBAC9B,kBAAkB;YACtB,CAAC,CAAC,CAAC;YACH,MAAM,EAAE,CAAC;QACb,CAAC;KACJ,CAAC;AACN,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mysten-incubation/memwal-mcp",
-  "version": "0.0.13",
+  "version": "0.0.14-dev.1",
   "description": "Walrus Memory MCP client — single-binary stdio MCP server that bridges Cursor / Claude Desktop / Antigravity / Claude Code to the Walrus Memory relayer. Handles browser-based wallet login on first run.",
   "type": "module",
   "engines": {