mcp-proxy 2.12.2 → 2.13.1

package/dist/index.js

@@ -1,7 +1,9 @@
 import {
+  InMemoryEventStore,
   proxyServer,
+  startHTTPStreamServer,
   startSSEServer
-} from "./chunk-HTMOSQ6H.js";
+} from "./chunk-F2LFKNGG.js";
 
 // src/tapTransport.ts
 var tapTransport = (transport, eventHandler) => {
@@ -25,22 +27,22 @@ var tapTransport = (transport, eventHandler) => {
   };
   transport.onerror = async (error) => {
     eventHandler({
-      type: "onerror",
-      error
+      error,
+      type: "onerror"
     });
     return originalOnError?.(error);
   };
   transport.onmessage = async (message) => {
     eventHandler({
-      type: "onmessage",
-      message
+      message,
+      type: "onmessage"
     });
     return originalOnMessage?.(message);
   };
   transport.send = async (message) => {
     eventHandler({
-      type: "send",
-      message
+      message,
+      type: "send"
     });
     return originalSend?.(message);
   };
@@ -53,7 +55,9 @@ var tapTransport = (transport, eventHandler) => {
   return transport;
 };
 export {
+  InMemoryEventStore,
   proxyServer,
+  startHTTPStreamServer,
   startSSEServer,
   tapTransport
 };

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/tapTransport.ts"],"sourcesContent":["import { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport { JSONRPCMessage } from \"@modelcontextprotocol/sdk/types.js\";\n\ntype TransportEvent =\n  | {\n      type: \"close\";\n    }\n  | {\n      type: \"onclose\";\n    }\n  | {\n      type: \"onerror\";\n      error: Error;\n    }\n  | {\n      type: \"onmessage\";\n      message: JSONRPCMessage;\n    }\n  | {\n      type: \"send\";\n      message: JSONRPCMessage;\n    }\n  | {\n      type: \"start\";\n    };\n\nexport const tapTransport = (\n  transport: Transport,\n  eventHandler: (event: TransportEvent) => void,\n) => {\n  const originalClose = transport.close.bind(transport);\n  const originalOnClose = transport.onclose?.bind(transport);\n  const originalOnError = transport.onerror?.bind(transport);\n  const originalOnMessage = transport.onmessage?.bind(transport);\n  const originalSend = transport.send.bind(transport);\n  const originalStart = transport.start.bind(transport);\n\n  transport.close = async () => {\n    eventHandler({\n      type: \"close\",\n    });\n\n    return originalClose?.();\n  };\n\n  transport.onclose = async () => {\n    eventHandler({\n      type: \"onclose\",\n    });\n\n    return originalOnClose?.();\n  };\n\n  transport.onerror = async (error: Error) => {\n    eventHandler({\n      type: \"onerror\",\n      error,\n    });\n\n    return originalOnError?.(error);\n  };\n\n  transport.onmessage = async (message: JSONRPCMessage) => {\n    eventHandler({\n      type: \"onmessage\",\n      message,\n    });\n\n    return originalOnMessage?.(message);\n  };\n\n  transport.send = async (message: JSONRPCMessage) => {\n    eventHandler({\n      type: \"send\",\n      message,\n    });\n\n    return originalSend?.(message);\n  };\n\n  transport.start = async () => {\n    eventHandler({\n      type: \"start\",\n    });\n\n    return originalStart?.();\n  };\n\n  return transport;\n};\n"],"mappings":";;;;;;AA0BO,IAAM,eAAe,CAC1B,WACA,iBACG;AACH,QAAM,gBAAgB,UAAU,MAAM,KAAK,SAAS;AACpD,QAAM,kBAAkB,UAAU,SAAS,KAAK,SAAS;AACzD,QAAM,kBAAkB,UAAU,SAAS,KAAK,SAAS;AACzD,QAAM,oBAAoB,UAAU,WAAW,KAAK,SAAS;AAC7D,QAAM,eAAe,UAAU,KAAK,KAAK,SAAS;AAClD,QAAM,gBAAgB,UAAU,MAAM,KAAK,SAAS;AAEpD,YAAU,QAAQ,YAAY;AAC5B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,gBAAgB;AAAA,EACzB;AAEA,YAAU,UAAU,YAAY;AAC9B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,kBAAkB;AAAA,EAC3B;AAEA,YAAU,UAAU,OAAO,UAAiB;AAC1C,iBAAa;AAAA,MACX,MAAM;AAAA,MACN;AAAA,IACF,CAAC;AAED,WAAO,kBAAkB,KAAK;AAAA,EAChC;AAEA,YAAU,YAAY,OAAO,YAA4B;AACvD,iBAAa;AAAA,MACX,MAAM;AAAA,MACN;AAAA,IACF,CAAC;AAED,WAAO,oBAAoB,OAAO;AAAA,EACpC;AAEA,YAAU,OAAO,OAAO,YAA4B;AAClD,iBAAa;AAAA,MACX,MAAM;AAAA,MACN;AAAA,IACF,CAAC;AAED,WAAO,eAAe,OAAO;AAAA,EAC/B;AAEA,YAAU,QAAQ,YAAY;AAC5B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,gBAAgB;AAAA,EACzB;AAEA,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../src/tapTransport.ts"],"sourcesContent":["import { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport { JSONRPCMessage } from \"@modelcontextprotocol/sdk/types.js\";\n\ntype TransportEvent =\n  | {\n      error: Error;\n      type: \"onerror\";\n    }\n  | {\n      message: JSONRPCMessage;\n      type: \"onmessage\";\n    }\n  | {\n      message: JSONRPCMessage;\n      type: \"send\";\n    }\n  | {\n      type: \"close\";\n    }\n  | {\n      type: \"onclose\";\n    }\n  | {\n      type: \"start\";\n    };\n\nexport const tapTransport = (\n  transport: Transport,\n  eventHandler: (event: TransportEvent) => void,\n) => {\n  const originalClose = transport.close.bind(transport);\n  const originalOnClose = transport.onclose?.bind(transport);\n  const originalOnError = transport.onerror?.bind(transport);\n  const originalOnMessage = transport.onmessage?.bind(transport);\n  const originalSend = transport.send.bind(transport);\n  const originalStart = transport.start.bind(transport);\n\n  transport.close = async () => {\n    eventHandler({\n      type: \"close\",\n    });\n\n    return originalClose?.();\n  };\n\n  transport.onclose = async () => {\n    eventHandler({\n      type: \"onclose\",\n    });\n\n    return originalOnClose?.();\n  };\n\n  transport.onerror = async (error: Error) => {\n    eventHandler({\n      error,\n      type: \"onerror\",\n    });\n\n    return originalOnError?.(error);\n  };\n\n  transport.onmessage = async (message: JSONRPCMessage) => {\n    eventHandler({\n      message,\n      type: \"onmessage\",\n    });\n\n    return originalOnMessage?.(message);\n  };\n\n  transport.send = async (message: JSONRPCMessage) => {\n    eventHandler({\n      message,\n      type: \"send\",\n    });\n\n    return originalSend?.(message);\n  };\n\n  transport.start = async () => {\n    eventHandler({\n      type: \"start\",\n    });\n\n    return originalStart?.();\n  };\n\n  return transport;\n};\n"],"mappings":";;;;;;;;AA0BO,IAAM,eAAe,CAC1B,WACA,iBACG;AACH,QAAM,gBAAgB,UAAU,MAAM,KAAK,SAAS;AACpD,QAAM,kBAAkB,UAAU,SAAS,KAAK,SAAS;AACzD,QAAM,kBAAkB,UAAU,SAAS,KAAK,SAAS;AACzD,QAAM,oBAAoB,UAAU,WAAW,KAAK,SAAS;AAC7D,QAAM,eAAe,UAAU,KAAK,KAAK,SAAS;AAClD,QAAM,gBAAgB,UAAU,MAAM,KAAK,SAAS;AAEpD,YAAU,QAAQ,YAAY;AAC5B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,gBAAgB;AAAA,EACzB;AAEA,YAAU,UAAU,YAAY;AAC9B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,kBAAkB;AAAA,EAC3B;AAEA,YAAU,UAAU,OAAO,UAAiB;AAC1C,iBAAa;AAAA,MACX;AAAA,MACA,MAAM;AAAA,IACR,CAAC;AAED,WAAO,kBAAkB,KAAK;AAAA,EAChC;AAEA,YAAU,YAAY,OAAO,YAA4B;AACvD,iBAAa;AAAA,MACX;AAAA,MACA,MAAM;AAAA,IACR,CAAC;AAED,WAAO,oBAAoB,OAAO;AAAA,EACpC;AAEA,YAAU,OAAO,OAAO,YAA4B;AAClD,iBAAa;AAAA,MACX;AAAA,MACA,MAAM;AAAA,IACR,CAAC;AAED,WAAO,eAAe,OAAO;AAAA,EAC/B;AAEA,YAAU,QAAQ,YAAY;AAC5B,iBAAa;AAAA,MACX,MAAM;AAAA,IACR,CAAC;AAED,WAAO,gBAAgB;AAAA,EACzB;AAEA,SAAO;AACT;","names":[]}

package/eslint.config.ts

@@ -0,0 +1,14 @@
+import eslint from "@eslint/js";
+import eslintConfigPrettier from "eslint-config-prettier/flat";
+import perfectionist from "eslint-plugin-perfectionist";
+import tseslint from "typescript-eslint";
+
+export default tseslint.config(
+  eslint.configs.recommended,
+  tseslint.configs.recommended,
+  perfectionist.configs["recommended-alphabetical"],
+  eslintConfigPrettier,
+  {
+    ignores: ["**/*.js"],
+  },
+);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-proxy",
-  "version": "2.12.2",
+  "version": "2.13.1",
   "main": "dist/index.js",
   "scripts": {
     "build": "tsup",
@@ -42,18 +42,22 @@
     ]
   },
   "devDependencies": {
+    "@eslint/js": "^9.25.1",
     "@sebbo2002/semantic-release-jsr": "^2.0.5",
     "@tsconfig/node22": "^22.0.1",
     "@types/node": "^22.14.1",
     "@types/yargs": "^17.0.33",
     "eslint": "^9.25.0",
+    "eslint-config-prettier": "^10.1.2",
     "eslint-plugin-perfectionist": "^4.11.0",
     "get-port-please": "^3.1.2",
+    "jiti": "^2.4.2",
     "prettier": "^3.5.3",
     "semantic-release": "^24.2.3",
     "tsup": "^8.4.0",
     "tsx": "^4.19.3",
     "typescript": "^5.8.3",
+    "typescript-eslint": "^8.31.1",
     "vitest": "^3.1.1"
   },
   "tsup": {

package/src/InMemoryEventStore.ts

@@ -0,0 +1,91 @@
+/**
+ * This is a copy of the InMemoryEventStore from the typescript-sdk
+ * https://github.com/modelcontextprotocol/typescript-sdk/blob/main/src/inMemoryEventStore.ts
+ */
+
+import type { EventStore } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import type { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";
+
+/**
+ * Simple in-memory implementation of the EventStore interface for resumability
+ * This is primarily intended for examples and testing, not for production use
+ * where a persistent storage solution would be more appropriate.
+ */
+export class InMemoryEventStore implements EventStore {
+  private events: Map<string, { message: JSONRPCMessage; streamId: string }> =
+    new Map();
+
+  /**
+   * Replays events that occurred after a specific event ID
+   * Implements EventStore.replayEventsAfter
+   */
+  async replayEventsAfter(
+    lastEventId: string,
+    {
+      send,
+    }: { send: (eventId: string, message: JSONRPCMessage) => Promise<void> },
+  ): Promise<string> {
+    if (!lastEventId || !this.events.has(lastEventId)) {
+      return "";
+    }
+
+    // Extract the stream ID from the event ID
+    const streamId = this.getStreamIdFromEventId(lastEventId);
+    if (!streamId) {
+      return "";
+    }
+
+    let foundLastEvent = false;
+
+    // Sort events by eventId for chronological ordering
+    const sortedEvents = [...this.events.entries()].sort((a, b) =>
+      a[0].localeCompare(b[0]),
+    );
+
+    for (const [
+      eventId,
+      { message, streamId: eventStreamId },
+    ] of sortedEvents) {
+      // Only include events from the same stream
+      if (eventStreamId !== streamId) {
+        continue;
+      }
+
+      // Start sending events after we find the lastEventId
+      if (eventId === lastEventId) {
+        foundLastEvent = true;
+        continue;
+      }
+
+      if (foundLastEvent) {
+        await send(eventId, message);
+      }
+    }
+    return streamId;
+  }
+
+  /**
+   * Stores an event with a generated event ID
+   * Implements EventStore.storeEvent
+   */
+  async storeEvent(streamId: string, message: JSONRPCMessage): Promise<string> {
+    const eventId = this.generateEventId(streamId);
+    this.events.set(eventId, { message, streamId });
+    return eventId;
+  }
+
+  /**
+   * Generates a unique event ID for a given stream ID
+   */
+  private generateEventId(streamId: string): string {
+    return `${streamId}_${Date.now()}_${Math.random().toString(36).substring(2, 10)}`;
+  }
+
+  /**
+   * Extracts the stream ID from an event ID
+   */
+  private getStreamIdFromEventId(eventId: string): string {
+    const parts = eventId.split("_");
+    return parts.length > 0 ? parts[0] : "";
+  }
+}

package/src/StdioClientTransport.ts

@@ -11,33 +11,23 @@ import { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";
 import { ChildProcess, IOType, spawn } from "node:child_process";
 import { Stream } from "node:stream";
 
-type TransportEvent =
-  | {
-      type: "close";
-    }
-  | {
-      type: "error";
-      error: Error;
-    }
-  | {
-      type: "data";
-      chunk: string;
-    }
-  | {
-      type: "message";
-      message: JSONRPCMessage;
-    };
-
 export type StdioServerParameters = {
+  /**
+   * Command line arguments to pass to the executable.
+   */
+  args?: string[];
+
   /**
    * The executable to run to start the server.
    */
   command: string;
 
   /**
-   * Command line arguments to pass to the executable.
+   * The working directory to use when spawning the process.
+   *
+   * If not specified, the current working directory will be inherited.
    */
-  args?: string[];
+  cwd?: string;
 
   /**
    * The environment to use when spawning the process.
@@ -47,46 +37,90 @@ export type StdioServerParameters = {
   env: Record<string, string>;
 
   /**
-   * How to handle stderr of the child process. This matches the semantics of Node's `child_process.spawn`.
-   *
-   * The default is "inherit", meaning messages to stderr will be printed to the parent process's stderr.
+   * A function to call when an event occurs.
    */
-  stderr?: IOType | Stream | number;
+  onEvent?: (event: TransportEvent) => void;
 
   /**
-   * The working directory to use when spawning the process.
+   * How to handle stderr of the child process. This matches the semantics of Node's `child_process.spawn`.
    *
-   * If not specified, the current working directory will be inherited.
-   */
-  cwd?: string;
-
-  /**
-   * A function to call when an event occurs.
+   * The default is "inherit", meaning messages to stderr will be printed to the parent process's stderr.
    */
-  onEvent?: (event: TransportEvent) => void;
+  stderr?: IOType | number | Stream;
 };
 
+type TransportEvent =
+  | {
+      chunk: string;
+      type: "data";
+    }
+  | {
+      error: Error;
+      type: "error";
+    }
+  | {
+      message: JSONRPCMessage;
+      type: "message";
+    }
+  | {
+      type: "close";
+    };
+
 /**
  * Client transport for stdio: this will connect to a server by spawning a process and communicating with it over stdin/stdout.
  *
  * This transport is only available in Node.js environments.
  */
 export class StdioClientTransport implements Transport {
-  private process?: ChildProcess;
-  private abortController: AbortController = new AbortController();
-  private readBuffer: ReadBuffer = new ReadBuffer();
-  private serverParams: StdioServerParameters;
-  private onEvent?: (event: TransportEvent) => void;
-
   onclose?: () => void;
   onerror?: (error: Error) => void;
   onmessage?: (message: JSONRPCMessage) => void;
+  /**
+   * The stderr stream of the child process, if `StdioServerParameters.stderr` was set to "pipe" or "overlapped".
+   *
+   * This is only available after the process has been started.
+   */
+  get stderr(): null | Stream {
+    return this.process?.stderr ?? null;
+  }
+  private abortController: AbortController = new AbortController();
+
+  private onEvent?: (event: TransportEvent) => void;
+  private process?: ChildProcess;
+  private readBuffer: ReadBuffer = new ReadBuffer();
+
+  private serverParams: StdioServerParameters;
 
   constructor(server: StdioServerParameters) {
     this.serverParams = server;
     this.onEvent = server.onEvent;
   }
 
+  async close(): Promise<void> {
+    this.onEvent?.({
+      type: "close",
+    });
+
+    this.abortController.abort();
+    this.process = undefined;
+    this.readBuffer.clear();
+  }
+
+  send(message: JSONRPCMessage): Promise<void> {
+    return new Promise((resolve) => {
+      if (!this.process?.stdin) {
+        throw new Error("Not connected");
+      }
+
+      const json = serializeMessage(message);
+      if (this.process.stdin.write(json)) {
+        resolve();
+      } else {
+        this.process.stdin.once("drain", resolve);
+      }
+    });
+  }
+
   /**
    * Starts the server process and prepares to communicate with it.
    */
@@ -102,11 +136,11 @@ export class StdioClientTransport implements Transport {
         this.serverParams.command,
         this.serverParams.args ?? [],
         {
+          cwd: this.serverParams.cwd,
           env: this.serverParams.env,
-          stdio: ["pipe", "pipe", this.serverParams.stderr ?? "inherit"],
           shell: false,
           signal: this.abortController.signal,
-          cwd: this.serverParams.cwd,
+          stdio: ["pipe", "pipe", this.serverParams.stderr ?? "inherit"],
         },
       );
 
@@ -125,6 +159,7 @@ export class StdioClientTransport implements Transport {
         resolve();
       });
 
+      // eslint-disable-next-line @typescript-eslint/no-unused-vars
       this.process.on("close", (_code) => {
         this.onEvent?.({
           type: "close",
@@ -136,8 +171,8 @@ export class StdioClientTransport implements Transport {
 
       this.process.stdin?.on("error", (error) => {
         this.onEvent?.({
-          type: "error",
           error,
+          type: "error",
         });
 
         this.onerror?.(error);
@@ -145,8 +180,8 @@ export class StdioClientTransport implements Transport {
 
       this.process.stdout?.on("data", (chunk) => {
         this.onEvent?.({
-          type: "data",
           chunk: chunk.toString(),
+          type: "data",
         });
 
         this.readBuffer.append(chunk);
@@ -155,8 +190,8 @@ export class StdioClientTransport implements Transport {
 
       this.process.stdout?.on("error", (error) => {
         this.onEvent?.({
-          type: "error",
           error,
+          type: "error",
         });
 
         this.onerror?.(error);
@@ -164,15 +199,6 @@ export class StdioClientTransport implements Transport {
     });
   }
 
-  /**
-   * The stderr stream of the child process, if `StdioServerParameters.stderr` was set to "pipe" or "overlapped".
-   *
-   * This is only available after the process has been started.
-   */
-  get stderr(): Stream | null {
-    return this.process?.stderr ?? null;
-  }
-
   private processReadBuffer() {
     while (true) {
       try {
@@ -183,44 +209,19 @@ export class StdioClientTransport implements Transport {
         }
 
         this.onEvent?.({
-          type: "message",
           message,
+          type: "message",
         });
 
         this.onmessage?.(message);
       } catch (error) {
         this.onEvent?.({
-          type: "error",
           error: error as Error,
+          type: "error",
         });
 
         this.onerror?.(error as Error);
       }
     }
   }
-
-  async close(): Promise<void> {
-    this.onEvent?.({
-      type: "close",
-    });
-
-    this.abortController.abort();
-    this.process = undefined;
-    this.readBuffer.clear();
-  }
-
-  send(message: JSONRPCMessage): Promise<void> {
-    return new Promise((resolve) => {
-      if (!this.process?.stdin) {
-        throw new Error("Not connected");
-      }
-
-      const json = serializeMessage(message);
-      if (this.process.stdin.write(json)) {
-        resolve();
-      } else {
-        this.process.stdin.once("drain", resolve);
-      }
-    });
-  }
 }

package/src/bin/mcp-proxy.ts

@@ -1,15 +1,18 @@
 #!/usr/bin/env node
 
-import yargs from "yargs";
-import { hideBin } from "yargs/helpers";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { EventSource } from "eventsource";
 import { setTimeout } from "node:timers";
-import { StdioClientTransport } from "../StdioClientTransport.js";
 import util from "node:util";
-import { startSSEServer } from "../startSSEServer.js";
+import yargs from "yargs";
+import { hideBin } from "yargs/helpers";
+
+import { InMemoryEventStore } from "../InMemoryEventStore.js";
 import { proxyServer } from "../proxyServer.js";
+import { startHTTPStreamServer } from "../startHTTPStreamServer.js";
+import { startSSEServer } from "../startSSEServer.js";
+import { StdioClientTransport } from "../StdioClientTransport.js";
 
 util.inspect.defaultOptions.depth = 8;
 
@@ -22,31 +25,36 @@ const argv = await yargs(hideBin(process.argv))
   .scriptName("mcp-proxy")
   .command("$0 <command> [args...]", "Run a command with MCP arguments")
   .positional("command", {
-    type: "string",
-    describe: "The command to run",
     demandOption: true,
+    describe: "The command to run",
+    type: "string",
   })
   .positional("args", {
-    type: "string",
     array: true,
     describe: "The arguments to pass to the command",
+    type: "string",
   })
   .env("MCP_PROXY")
   .options({
     debug: {
-      type: "boolean",
-      describe: "Enable debug logging",
       default: false,
+      describe: "Enable debug logging",
+      type: "boolean",
     },
     endpoint: {
+      describe: "The endpoint to listen on",
       type: "string",
-      describe: "The endpoint to listen on for SSE",
-      default: "/sse",
     },
     port: {
-      type: "number",
-      describe: "The port to listen on for SSE",
       default: 8080,
+      describe: "The port to listen on",
+      type: "number",
+    },
+    server: {
+      choices: ["sse", "stream"],
+      default: "sse",
+      describe: "The server type to use (sse or stream)",
+      type: "string",
     },
   })
   .help()
@@ -54,15 +62,15 @@ const argv = await yargs(hideBin(process.argv))
 
 const connect = async (client: Client) => {
   const transport = new StdioClientTransport({
-    command: argv.command,
     args: argv.args,
+    command: argv.command,
     env: process.env as Record<string, string>,
-    stderr: "pipe",
     onEvent: (event) => {
       if (argv.debug) {
         console.debug("transport event", event);
       }
     },
+    stderr: "pipe",
   });
 
   await client.connect(transport);
@@ -86,27 +94,40 @@ const proxy = async () => {
     version: string;
   };
 
-  const serverCapabilities = client.getServerCapabilities() as {};
+  const serverCapabilities = client.getServerCapabilities() as {
+    capabilities: Record<string, unknown>;
+  };
 
-  console.info("starting the SSE server on port %d", argv.port);
+  console.info("starting the %s server on port %d", argv.server, argv.port);
 
-  await startSSEServer({
-    createServer: async () => {
-      const server = new Server(serverVersion, {
-        capabilities: serverCapabilities,
-      });
+  const createServer = async () => {
+    const server = new Server(serverVersion, {
+      capabilities: serverCapabilities,
+    });
 
-      proxyServer({
-        server,
-        client,
-        serverCapabilities,
-      });
+    proxyServer({
+      client,
+      server,
+      serverCapabilities,
+    });
 
-      return server;
-    },
-    port: argv.port,
-    endpoint: argv.endpoint as `/${string}`,
-  });
+    return server;
+  };
+
+  if (argv.server === "sse") {
+    await startSSEServer({
+      createServer,
+      endpoint: argv.endpoint || ("/sse" as `/${string}`),
+      port: argv.port,
+    });
+  } else {
+    await startHTTPStreamServer({
+      createServer,
+      endpoint: argv.endpoint || ("/stream" as `/${string}`),
+      eventStore: new InMemoryEventStore(),
+      port: argv.port,
+    });
+  }
 };
 
 const main = async () => {

package/src/index.ts

@@ -1,3 +1,5 @@
-export { tapTransport } from "./tapTransport.js";
+export { InMemoryEventStore } from "./InMemoryEventStore.js";
 export { proxyServer } from "./proxyServer.js";
+export { startHTTPStreamServer } from "./startHTTPStreamServer.js";
 export { startSSEServer } from "./startSSEServer.js";
+export { tapTransport } from "./tapTransport.js";

package/src/proxyServer.ts

@@ -1,3 +1,4 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import {
   CallToolRequestSchema,
@@ -9,20 +10,19 @@ import {
   ListToolsRequestSchema,
   LoggingMessageNotificationSchema,
   ReadResourceRequestSchema,
-  SubscribeRequestSchema,
-  UnsubscribeRequestSchema,
   ResourceUpdatedNotificationSchema,
   ServerCapabilities,
+  SubscribeRequestSchema,
+  UnsubscribeRequestSchema,
 } from "@modelcontextprotocol/sdk/types.js";
-import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 
 export const proxyServer = async ({
-  server,
   client,
+  server,
   serverCapabilities,
 }: {
-  server: Server;
   client: Client;
+  server: Server;
   serverCapabilities: ServerCapabilities;
 }) => {
   if (serverCapabilities?.logging) {
@@ -62,10 +62,10 @@ export const proxyServer = async ({
 
     if (serverCapabilities?.resources.subscribe) {
       server.setNotificationHandler(
-          ResourceUpdatedNotificationSchema,
-          async (args) => {
-            return client.notification(args);
-          },
+        ResourceUpdatedNotificationSchema,
+        async (args) => {
+          return client.notification(args);
+        },
       );
 
       server.setRequestHandler(SubscribeRequestSchema, async (args) => {