mcp-proxy 2.8.2 → 2.9.0

package/dist/mcp-proxy.js

@@ -0,0 +1,405 @@
+#!/usr/bin/env node
+
+// src/bin/mcp-proxy.ts
+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/promises";
+
+// src/StdioClientTransport.ts
+import {
+  ReadBuffer,
+  serializeMessage
+} from "@modelcontextprotocol/sdk/shared/stdio.js";
+import { spawn } from "node:child_process";
+var StdioClientTransport = class {
+  process;
+  abortController = new AbortController();
+  readBuffer = new ReadBuffer();
+  serverParams;
+  onEvent;
+  onclose;
+  onerror;
+  onmessage;
+  constructor(server) {
+    this.serverParams = server;
+    this.onEvent = server.onEvent;
+  }
+  /**
+   * Starts the server process and prepares to communicate with it.
+   */
+  async start() {
+    if (this.process) {
+      throw new Error(
+        "StdioClientTransport already started! If using Client class, note that connect() calls start() automatically."
+      );
+    }
+    return new Promise((resolve, reject) => {
+      this.process = spawn(
+        this.serverParams.command,
+        this.serverParams.args ?? [],
+        {
+          env: this.serverParams.env,
+          stdio: ["pipe", "pipe", this.serverParams.stderr ?? "inherit"],
+          shell: false,
+          signal: this.abortController.signal,
+          cwd: this.serverParams.cwd
+        }
+      );
+      this.process.on("error", (error) => {
+        if (error.name === "AbortError") {
+          this.onclose?.();
+          return;
+        }
+        reject(error);
+        this.onerror?.(error);
+      });
+      this.process.on("spawn", () => {
+        resolve();
+      });
+      this.process.on("close", (_code) => {
+        this.onEvent?.({
+          type: "close"
+        });
+        this.process = void 0;
+        this.onclose?.();
+      });
+      this.process.stdin?.on("error", (error) => {
+        this.onEvent?.({
+          type: "error",
+          error
+        });
+        this.onerror?.(error);
+      });
+      this.process.stdout?.on("data", (chunk) => {
+        this.onEvent?.({
+          type: "data",
+          chunk: chunk.toString()
+        });
+        this.readBuffer.append(chunk);
+        this.processReadBuffer();
+      });
+      this.process.stdout?.on("error", (error) => {
+        this.onEvent?.({
+          type: "error",
+          error
+        });
+        this.onerror?.(error);
+      });
+    });
+  }
+  /**
+   * 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() {
+    return this.process?.stderr ?? null;
+  }
+  processReadBuffer() {
+    while (true) {
+      try {
+        const message = this.readBuffer.readMessage();
+        if (message === null) {
+          break;
+        }
+        this.onEvent?.({
+          type: "message",
+          message
+        });
+        this.onmessage?.(message);
+      } catch (error) {
+        this.onEvent?.({
+          type: "error",
+          error
+        });
+        this.onerror?.(error);
+      }
+    }
+  }
+  async close() {
+    this.onEvent?.({
+      type: "close"
+    });
+    this.abortController.abort();
+    this.process = void 0;
+    this.readBuffer.clear();
+  }
+  send(message) {
+    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);
+      }
+    });
+  }
+};
+
+// src/bin/mcp-proxy.ts
+import util from "node:util";
+
+// src/startSSEServer.ts
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import http from "http";
+var startSSEServer = async ({
+  port,
+  createServer,
+  endpoint,
+  onConnect,
+  onClose
+}) => {
+  const activeTransports = {};
+  const httpServer = http.createServer(async (req, res) => {
+    if (req.headers.origin) {
+      try {
+        const origin = new URL(req.headers.origin);
+        res.setHeader("Access-Control-Allow-Origin", origin.origin);
+        res.setHeader("Access-Control-Allow-Credentials", "true");
+        res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
+        res.setHeader("Access-Control-Allow-Headers", "*");
+      } catch (error) {
+        console.error("Error parsing origin:", error);
+      }
+    }
+    if (req.method === "OPTIONS") {
+      res.writeHead(204);
+      res.end();
+      return;
+    }
+    if (req.method === "GET" && req.url === `/ping`) {
+      res.writeHead(200).end("pong");
+      return;
+    }
+    if (req.method === "GET" && req.url === endpoint) {
+      const transport = new SSEServerTransport("/messages", res);
+      let server;
+      try {
+        server = await createServer(req);
+      } catch (error) {
+        if (error instanceof Response) {
+          res.writeHead(error.status).end(error.statusText);
+          return;
+        }
+        res.writeHead(500).end("Error creating server");
+        return;
+      }
+      activeTransports[transport.sessionId] = transport;
+      let closed = false;
+      res.on("close", async () => {
+        closed = true;
+        try {
+          await server.close();
+        } catch (error) {
+          console.error("Error closing server:", error);
+        }
+        delete activeTransports[transport.sessionId];
+        onClose?.(server);
+      });
+      try {
+        await server.connect(transport);
+        await transport.send({
+          jsonrpc: "2.0",
+          method: "sse/connection",
+          params: { message: "SSE Connection established" }
+        });
+        onConnect?.(server);
+      } catch (error) {
+        if (!closed) {
+          console.error("Error connecting to server:", error);
+          res.writeHead(500).end("Error connecting to server");
+        }
+      }
+      return;
+    }
+    if (req.method === "POST" && req.url?.startsWith("/messages")) {
+      const sessionId = new URL(
+        req.url,
+        "https://example.com"
+      ).searchParams.get("sessionId");
+      if (!sessionId) {
+        res.writeHead(400).end("No sessionId");
+        return;
+      }
+      const activeTransport = activeTransports[sessionId];
+      if (!activeTransport) {
+        res.writeHead(400).end("No active transport");
+        return;
+      }
+      await activeTransport.handlePostMessage(req, res);
+      return;
+    }
+    res.writeHead(404).end();
+  });
+  await new Promise((resolve) => {
+    httpServer.listen(port, "::", () => {
+      resolve(void 0);
+    });
+  });
+  return {
+    close: async () => {
+      for (const transport of Object.values(activeTransports)) {
+        await transport.close();
+      }
+      return new Promise((resolve, reject) => {
+        httpServer.close((error) => {
+          if (error) {
+            reject(error);
+            return;
+          }
+          resolve();
+        });
+      });
+    }
+  };
+};
+
+// src/proxyServer.ts
+import {
+  CallToolRequestSchema,
+  CompleteRequestSchema,
+  GetPromptRequestSchema,
+  ListPromptsRequestSchema,
+  ListResourcesRequestSchema,
+  ListResourceTemplatesRequestSchema,
+  ListToolsRequestSchema,
+  LoggingMessageNotificationSchema,
+  ReadResourceRequestSchema
+} from "@modelcontextprotocol/sdk/types.js";
+var proxyServer = async ({
+  server,
+  client,
+  serverCapabilities
+}) => {
+  if (serverCapabilities?.logging) {
+    server.setNotificationHandler(
+      LoggingMessageNotificationSchema,
+      async (args) => {
+        return client.notification(args);
+      }
+    );
+  }
+  if (serverCapabilities?.prompts) {
+    server.setRequestHandler(GetPromptRequestSchema, async (args) => {
+      return client.getPrompt(args.params);
+    });
+    server.setRequestHandler(ListPromptsRequestSchema, async (args) => {
+      return client.listPrompts(args.params);
+    });
+  }
+  if (serverCapabilities?.resources) {
+    server.setRequestHandler(ListResourcesRequestSchema, async (args) => {
+      return client.listResources(args.params);
+    });
+    server.setRequestHandler(
+      ListResourceTemplatesRequestSchema,
+      async (args) => {
+        return client.listResourceTemplates(args.params);
+      }
+    );
+    server.setRequestHandler(ReadResourceRequestSchema, async (args) => {
+      return client.readResource(args.params);
+    });
+  }
+  if (serverCapabilities?.tools) {
+    server.setRequestHandler(CallToolRequestSchema, async (args) => {
+      return client.callTool(args.params);
+    });
+    server.setRequestHandler(ListToolsRequestSchema, async (args) => {
+      return client.listTools(args.params);
+    });
+  }
+  server.setRequestHandler(CompleteRequestSchema, async (args) => {
+    return client.complete(args.params);
+  });
+};
+
+// src/bin/mcp-proxy.ts
+util.inspect.defaultOptions.depth = 8;
+if (!("EventSource" in global)) {
+  global.EventSource = EventSource;
+}
+var 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
+}).positional("args", {
+  type: "string",
+  array: true,
+  describe: "The arguments to pass to the command"
+}).options({
+  debug: {
+    type: "boolean",
+    describe: "Enable debug logging",
+    default: false
+  },
+  endpoint: {
+    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
+  }
+}).help().parseAsync();
+var connect = async (client) => {
+  const transport = new StdioClientTransport({
+    command: argv.command,
+    args: argv.args,
+    env: process.env,
+    stderr: "pipe",
+    onEvent: (event) => {
+      console.debug("transport event", event);
+    }
+  });
+  await client.connect(transport);
+};
+var proxy = async () => {
+  const client = new Client(
+    {
+      name: "mcp-proxy",
+      version: "1.0.0"
+    },
+    {
+      capabilities: {}
+    }
+  );
+  await connect(client);
+  const serverVersion = client.getServerVersion();
+  const serverCapabilities = client.getServerCapabilities();
+  console.info("starting the SSE server on port %d", argv.port);
+  await startSSEServer({
+    createServer: async () => {
+      const server = new Server(serverVersion, {
+        capabilities: serverCapabilities
+      });
+      proxyServer({
+        server,
+        client,
+        serverCapabilities
+      });
+      return server;
+    },
+    port: argv.port,
+    endpoint: argv.endpoint
+  });
+};
+var main = async () => {
+  try {
+    await proxy();
+  } catch (error) {
+    console.error("could not start the proxy", error);
+    await setTimeout(1e3);
+    process.exit(1);
+  }
+};
+await main();
+//# sourceMappingURL=mcp-proxy.js.map

package/dist/mcp-proxy.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/bin/mcp-proxy.ts","../src/StdioClientTransport.ts","../src/startSSEServer.ts","../src/proxyServer.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport yargs from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport { EventSource } from \"eventsource\";\nimport { setTimeout } from \"node:timers/promises\";\nimport { StdioClientTransport } from \"../StdioClientTransport.js\";\nimport util from \"node:util\";\nimport { startSSEServer } from \"../startSSEServer.js\";\nimport { proxyServer } from \"../proxyServer.js\";\n\nutil.inspect.defaultOptions.depth = 8;\n\nif (!(\"EventSource\" in global)) {\n  // @ts-expect-error - figure out how to use --experimental-eventsource with vitest\n  global.EventSource = EventSource;\n}\n\nconst argv = await yargs(hideBin(process.argv))\n  .scriptName(\"mcp-proxy\")\n  .command(\"$0 <command> [args...]\", \"Run a command with MCP arguments\")\n  .positional(\"command\", {\n    type: \"string\",\n    describe: \"The command to run\",\n    demandOption: true,\n  })\n  .positional(\"args\", {\n    type: \"string\",\n    array: true,\n    describe: \"The arguments to pass to the command\",\n  })\n  .options({\n    debug: {\n      type: \"boolean\",\n      describe: \"Enable debug logging\",\n      default: false,\n    },\n    endpoint: {\n      type: \"string\",\n      describe: \"The endpoint to listen on for SSE\",\n      default: \"/sse\",\n    },\n    port: {\n      type: \"number\",\n      describe: \"The port to listen on for SSE\",\n      default: 8080,\n    },\n  })\n  .help()\n  .parseAsync();\n\nconst connect = async (client: Client) => {\n  const transport = new StdioClientTransport({\n    command: argv.command,\n    args: argv.args,\n    env: process.env as Record<string, string>,\n    stderr: \"pipe\",\n    onEvent: (event) => {\n      console.debug(\"transport event\", event);\n    },\n  });\n\n  await client.connect(transport);\n};\n\nconst proxy = async () => {\n  const client = new Client(\n    {\n      name: \"mcp-proxy\",\n      version: \"1.0.0\",\n    },\n    {\n      capabilities: {},\n    },\n  );\n\n  await connect(client);\n\n  const serverVersion = client.getServerVersion() as {\n    name: string;\n    version: string;\n  };\n\n  const serverCapabilities = client.getServerCapabilities() as {};\n\n  console.info(\"starting the SSE server on port %d\", argv.port);\n\n  await startSSEServer({\n    createServer: async () => {\n      const server = new Server(serverVersion, {\n        capabilities: serverCapabilities,\n      });\n\n      proxyServer({\n        server,\n        client,\n        serverCapabilities,\n      });\n\n      return server;\n    },\n    port: argv.port,\n    endpoint: argv.endpoint as `/${string}`,\n  });\n};\n\nconst main = async () => {\n  try {\n    await proxy();\n  } catch (error) {\n    console.error(\"could not start the proxy\", error);\n\n    await setTimeout(1000);\n\n    process.exit(1);\n  }\n};\n\nawait main();\n","/**\n * Forked from https://github.com/modelcontextprotocol/typescript-sdk/blob/66e1508162d37c0b83b0637ebcd7f07946e3d210/src/client/stdio.ts#L90\n */\n\nimport {\n  ReadBuffer,\n  serializeMessage,\n} from \"@modelcontextprotocol/sdk/shared/stdio.js\";\nimport { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport { JSONRPCMessage } from \"@modelcontextprotocol/sdk/types.js\";\nimport { ChildProcess, IOType, spawn } from \"node:child_process\";\nimport { Stream } from \"node:stream\";\n\ntype TransportEvent =\n  | {\n      type: \"close\";\n    }\n  | {\n      type: \"error\";\n      error: Error;\n    }\n  | {\n      type: \"data\";\n      chunk: string;\n    }\n  | {\n      type: \"message\";\n      message: JSONRPCMessage;\n    };\n\nexport type StdioServerParameters = {\n  /**\n   * The executable to run to start the server.\n   */\n  command: string;\n\n  /**\n   * Command line arguments to pass to the executable.\n   */\n  args?: string[];\n\n  /**\n   * The environment to use when spawning the process.\n   *\n   * If not specified, the result of getDefaultEnvironment() will be used.\n   */\n  env: Record<string, string>;\n\n  /**\n   * How to handle stderr of the child process. This matches the semantics of Node's `child_process.spawn`.\n   *\n   * The default is \"inherit\", meaning messages to stderr will be printed to the parent process's stderr.\n   */\n  stderr?: IOType | Stream | number;\n\n  /**\n   * The working directory to use when spawning the process.\n   *\n   * If not specified, the current working directory will be inherited.\n   */\n  cwd?: string;\n\n  /**\n   * A function to call when an event occurs.\n   */\n  onEvent?: (event: TransportEvent) => void;\n};\n\n/**\n * Client transport for stdio: this will connect to a server by spawning a process and communicating with it over stdin/stdout.\n *\n * This transport is only available in Node.js environments.\n */\nexport class StdioClientTransport implements Transport {\n  private process?: ChildProcess;\n  private abortController: AbortController = new AbortController();\n  private readBuffer: ReadBuffer = new ReadBuffer();\n  private serverParams: StdioServerParameters;\n  private onEvent?: (event: TransportEvent) => void;\n\n  onclose?: () => void;\n  onerror?: (error: Error) => void;\n  onmessage?: (message: JSONRPCMessage) => void;\n\n  constructor(server: StdioServerParameters) {\n    this.serverParams = server;\n    this.onEvent = server.onEvent;\n  }\n\n  /**\n   * Starts the server process and prepares to communicate with it.\n   */\n  async start(): Promise<void> {\n    if (this.process) {\n      throw new Error(\n        \"StdioClientTransport already started! If using Client class, note that connect() calls start() automatically.\",\n      );\n    }\n\n    return new Promise((resolve, reject) => {\n      this.process = spawn(\n        this.serverParams.command,\n        this.serverParams.args ?? [],\n        {\n          env: this.serverParams.env,\n          stdio: [\"pipe\", \"pipe\", this.serverParams.stderr ?? \"inherit\"],\n          shell: false,\n          signal: this.abortController.signal,\n          cwd: this.serverParams.cwd,\n        },\n      );\n\n      this.process.on(\"error\", (error) => {\n        if (error.name === \"AbortError\") {\n          // Expected when close() is called.\n          this.onclose?.();\n          return;\n        }\n\n        reject(error);\n        this.onerror?.(error);\n      });\n\n      this.process.on(\"spawn\", () => {\n        resolve();\n      });\n\n      this.process.on(\"close\", (_code) => {\n        this.onEvent?.({\n          type: \"close\",\n        });\n\n        this.process = undefined;\n        this.onclose?.();\n      });\n\n      this.process.stdin?.on(\"error\", (error) => {\n        this.onEvent?.({\n          type: \"error\",\n          error,\n        });\n\n        this.onerror?.(error);\n      });\n\n      this.process.stdout?.on(\"data\", (chunk) => {\n        this.onEvent?.({\n          type: \"data\",\n          chunk: chunk.toString(),\n        });\n\n        this.readBuffer.append(chunk);\n        this.processReadBuffer();\n      });\n\n      this.process.stdout?.on(\"error\", (error) => {\n        this.onEvent?.({\n          type: \"error\",\n          error,\n        });\n\n        this.onerror?.(error);\n      });\n    });\n  }\n\n  /**\n   * The stderr stream of the child process, if `StdioServerParameters.stderr` was set to \"pipe\" or \"overlapped\".\n   *\n   * This is only available after the process has been started.\n   */\n  get stderr(): Stream | null {\n    return this.process?.stderr ?? null;\n  }\n\n  private processReadBuffer() {\n    while (true) {\n      try {\n        const message = this.readBuffer.readMessage();\n\n        if (message === null) {\n          break;\n        }\n\n        this.onEvent?.({\n          type: \"message\",\n          message,\n        });\n\n        this.onmessage?.(message);\n      } catch (error) {\n        this.onEvent?.({\n          type: \"error\",\n          error: error as Error,\n        });\n\n        this.onerror?.(error as Error);\n      }\n    }\n  }\n\n  async close(): Promise<void> {\n    this.onEvent?.({\n      type: \"close\",\n    });\n\n    this.abortController.abort();\n    this.process = undefined;\n    this.readBuffer.clear();\n  }\n\n  send(message: JSONRPCMessage): Promise<void> {\n    return new Promise((resolve) => {\n      if (!this.process?.stdin) {\n        throw new Error(\"Not connected\");\n      }\n\n      const json = serializeMessage(message);\n      if (this.process.stdin.write(json)) {\n        resolve();\n      } else {\n        this.process.stdin.once(\"drain\", resolve);\n      }\n    });\n  }\n}\n","import { SSEServerTransport } from \"@modelcontextprotocol/sdk/server/sse.js\";\nimport http from \"http\";\nimport { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\n\nexport type SSEServer = {\n  close: () => Promise<void>;\n};\n\ntype ServerLike = {\n  connect: Server[\"connect\"];\n  close: Server[\"close\"];\n};\n\nexport const startSSEServer = async <T extends ServerLike>({\n  port,\n  createServer,\n  endpoint,\n  onConnect,\n  onClose,\n}: {\n  port: number;\n  endpoint: string;\n  createServer: (request: http.IncomingMessage) => Promise<T>;\n  onConnect?: (server: T) => void;\n  onClose?: (server: T) => void;\n}): Promise<SSEServer> => {\n  const activeTransports: Record<string, SSEServerTransport> = {};\n\n  /**\n   * @author https://dev.classmethod.jp/articles/mcp-sse/\n   */\n  const httpServer = http.createServer(async (req, res) => {\n    if (req.headers.origin) {\n      try {\n        const origin = new URL(req.headers.origin);\n\n        res.setHeader(\"Access-Control-Allow-Origin\", origin.origin);\n        res.setHeader(\"Access-Control-Allow-Credentials\", \"true\");\n        res.setHeader(\"Access-Control-Allow-Methods\", \"GET, POST, OPTIONS\");\n        res.setHeader(\"Access-Control-Allow-Headers\", \"*\");\n      } catch (error) {\n        console.error(\"Error parsing origin:\", error);\n      }\n    }\n\n    if (req.method === \"OPTIONS\") {\n      res.writeHead(204);\n      res.end();\n      return;\n    }\n\n    if (req.method === \"GET\" && req.url === `/ping`) {\n      res.writeHead(200).end(\"pong\");\n\n      return;\n    }\n\n    if (req.method === \"GET\" && req.url === endpoint) {\n      const transport = new SSEServerTransport(\"/messages\", res);\n\n      let server: T;\n\n      try {\n        server = await createServer(req);\n      } catch (error) {\n        if (error instanceof Response) {\n          res.writeHead(error.status).end(error.statusText);\n\n          return;\n        }\n\n        res.writeHead(500).end(\"Error creating server\");\n\n        return;\n      }\n\n      activeTransports[transport.sessionId] = transport;\n\n      let closed = false;\n\n      res.on(\"close\", async () => {\n        closed = true;\n\n        try {\n          await server.close();\n        } catch (error) {\n          console.error(\"Error closing server:\", error);\n        }\n\n        delete activeTransports[transport.sessionId];\n\n        onClose?.(server);\n      });\n\n      try {\n        await server.connect(transport);\n\n        await transport.send({\n          jsonrpc: \"2.0\",\n          method: \"sse/connection\",\n          params: { message: \"SSE Connection established\" },\n        });\n\n        onConnect?.(server);\n      } catch (error) {\n        if (!closed) {\n          console.error(\"Error connecting to server:\", error);\n\n          res.writeHead(500).end(\"Error connecting to server\");\n        }\n      }\n\n      return;\n    }\n\n    if (req.method === \"POST\" && req.url?.startsWith(\"/messages\")) {\n      const sessionId = new URL(\n        req.url,\n        \"https://example.com\",\n      ).searchParams.get(\"sessionId\");\n\n      if (!sessionId) {\n        res.writeHead(400).end(\"No sessionId\");\n\n        return;\n      }\n\n      const activeTransport: SSEServerTransport | undefined =\n        activeTransports[sessionId];\n\n      if (!activeTransport) {\n        res.writeHead(400).end(\"No active transport\");\n\n        return;\n      }\n\n      await activeTransport.handlePostMessage(req, res);\n\n      return;\n    }\n\n    res.writeHead(404).end();\n  });\n\n  await new Promise((resolve) => {\n    httpServer.listen(port, \"::\", () => {\n      resolve(undefined);\n    });\n  });\n\n  return {\n    close: async () => {\n      for (const transport of Object.values(activeTransports)) {\n        await transport.close();\n      }\n\n      return new Promise((resolve, reject) => {\n        httpServer.close((error) => {\n          if (error) {\n            reject(error);\n\n            return;\n          }\n\n          resolve();\n        });\n      });\n    },\n  };\n};\n","import { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport {\n  CallToolRequestSchema,\n  CompleteRequestSchema,\n  GetPromptRequestSchema,\n  ListPromptsRequestSchema,\n  ListResourcesRequestSchema,\n  ListResourceTemplatesRequestSchema,\n  ListToolsRequestSchema,\n  LoggingMessageNotificationSchema,\n  ReadResourceRequestSchema,\n  ServerCapabilities,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\n\nexport const proxyServer = async ({\n  server,\n  client,\n  serverCapabilities,\n}: {\n  server: Server;\n  client: Client;\n  serverCapabilities: ServerCapabilities;\n}) => {\n  if (serverCapabilities?.logging) {\n    server.setNotificationHandler(\n      LoggingMessageNotificationSchema,\n      async (args) => {\n        return client.notification(args);\n      },\n    );\n  }\n\n  if (serverCapabilities?.prompts) {\n    server.setRequestHandler(GetPromptRequestSchema, async (args) => {\n      return client.getPrompt(args.params);\n    });\n\n    server.setRequestHandler(ListPromptsRequestSchema, async (args) => {\n      return client.listPrompts(args.params);\n    });\n  }\n\n  if (serverCapabilities?.resources) {\n    server.setRequestHandler(ListResourcesRequestSchema, async (args) => {\n      return client.listResources(args.params);\n    });\n\n    server.setRequestHandler(\n      ListResourceTemplatesRequestSchema,\n      async (args) => {\n        return client.listResourceTemplates(args.params);\n      },\n    );\n\n    server.setRequestHandler(ReadResourceRequestSchema, async (args) => {\n      return client.readResource(args.params);\n    });\n  }\n\n  if (serverCapabilities?.tools) {\n    server.setRequestHandler(CallToolRequestSchema, async (args) => {\n      return client.callTool(args.params);\n    });\n\n    server.setRequestHandler(ListToolsRequestSchema, async (args) => {\n      return client.listTools(args.params);\n    });\n  }\n\n  server.setRequestHandler(CompleteRequestSchema, async (args) => {\n    return client.complete(args.params);\n  });\n};\n"],"mappings":";;;AAEA,OAAO,WAAW;AAClB,SAAS,eAAe;AACxB,SAAS,cAAc;AACvB,SAAS,cAAc;AACvB,SAAS,mBAAmB;AAC5B,SAAS,kBAAkB;;;ACH3B;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAGP,SAA+B,aAAa;AA+DrC,IAAM,uBAAN,MAAgD;AAAA,EAC7C;AAAA,EACA,kBAAmC,IAAI,gBAAgB;AAAA,EACvD,aAAyB,IAAI,WAAW;AAAA,EACxC;AAAA,EACA;AAAA,EAER;AAAA,EACA;AAAA,EACA;AAAA,EAEA,YAAY,QAA+B;AACzC,SAAK,eAAe;AACpB,SAAK,UAAU,OAAO;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAuB;AAC3B,QAAI,KAAK,SAAS;AAChB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,WAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAK,UAAU;AAAA,QACb,KAAK,aAAa;AAAA,QAClB,KAAK,aAAa,QAAQ,CAAC;AAAA,QAC3B;AAAA,UACE,KAAK,KAAK,aAAa;AAAA,UACvB,OAAO,CAAC,QAAQ,QAAQ,KAAK,aAAa,UAAU,SAAS;AAAA,UAC7D,OAAO;AAAA,UACP,QAAQ,KAAK,gBAAgB;AAAA,UAC7B,KAAK,KAAK,aAAa;AAAA,QACzB;AAAA,MACF;AAEA,WAAK,QAAQ,GAAG,SAAS,CAAC,UAAU;AAClC,YAAI,MAAM,SAAS,cAAc;AAE/B,eAAK,UAAU;AACf;AAAA,QACF;AAEA,eAAO,KAAK;AACZ,aAAK,UAAU,KAAK;AAAA,MACtB,CAAC;AAED,WAAK,QAAQ,GAAG,SAAS,MAAM;AAC7B,gBAAQ;AAAA,MACV,CAAC;AAED,WAAK,QAAQ,GAAG,SAAS,CAAC,UAAU;AAClC,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,QACR,CAAC;AAED,aAAK,UAAU;AACf,aAAK,UAAU;AAAA,MACjB,CAAC;AAED,WAAK,QAAQ,OAAO,GAAG,SAAS,CAAC,UAAU;AACzC,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAED,aAAK,UAAU,KAAK;AAAA,MACtB,CAAC;AAED,WAAK,QAAQ,QAAQ,GAAG,QAAQ,CAAC,UAAU;AACzC,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN,OAAO,MAAM,SAAS;AAAA,QACxB,CAAC;AAED,aAAK,WAAW,OAAO,KAAK;AAC5B,aAAK,kBAAkB;AAAA,MACzB,CAAC;AAED,WAAK,QAAQ,QAAQ,GAAG,SAAS,CAAC,UAAU;AAC1C,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAED,aAAK,UAAU,KAAK;AAAA,MACtB,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,IAAI,SAAwB;AAC1B,WAAO,KAAK,SAAS,UAAU;AAAA,EACjC;AAAA,EAEQ,oBAAoB;AAC1B,WAAO,MAAM;AACX,UAAI;AACF,cAAM,UAAU,KAAK,WAAW,YAAY;AAE5C,YAAI,YAAY,MAAM;AACpB;AAAA,QACF;AAEA,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAED,aAAK,YAAY,OAAO;AAAA,MAC1B,SAAS,OAAO;AACd,aAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAED,aAAK,UAAU,KAAc;AAAA,MAC/B;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,QAAuB;AAC3B,SAAK,UAAU;AAAA,MACb,MAAM;AAAA,IACR,CAAC;AAED,SAAK,gBAAgB,MAAM;AAC3B,SAAK,UAAU;AACf,SAAK,WAAW,MAAM;AAAA,EACxB;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO,IAAI,QAAQ,CAAC,YAAY;AAC9B,UAAI,CAAC,KAAK,SAAS,OAAO;AACxB,cAAM,IAAI,MAAM,eAAe;AAAA,MACjC;AAEA,YAAM,OAAO,iBAAiB,OAAO;AACrC,UAAI,KAAK,QAAQ,MAAM,MAAM,IAAI,GAAG;AAClC,gBAAQ;AAAA,MACV,OAAO;AACL,aAAK,QAAQ,MAAM,KAAK,SAAS,OAAO;AAAA,MAC1C;AAAA,IACF,CAAC;AAAA,EACH;AACF;;;ADxNA,OAAO,UAAU;;;AETjB,SAAS,0BAA0B;AACnC,OAAO,UAAU;AAYV,IAAM,iBAAiB,OAA6B;AAAA,EACzD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAM0B;AACxB,QAAM,mBAAuD,CAAC;AAK9D,QAAM,aAAa,KAAK,aAAa,OAAO,KAAK,QAAQ;AACvD,QAAI,IAAI,QAAQ,QAAQ;AACtB,UAAI;AACF,cAAM,SAAS,IAAI,IAAI,IAAI,QAAQ,MAAM;AAEzC,YAAI,UAAU,+BAA+B,OAAO,MAAM;AAC1D,YAAI,UAAU,oCAAoC,MAAM;AACxD,YAAI,UAAU,gCAAgC,oBAAoB;AAClE,YAAI,UAAU,gCAAgC,GAAG;AAAA,MACnD,SAAS,OAAO;AACd,gBAAQ,MAAM,yBAAyB,KAAK;AAAA,MAC9C;AAAA,IACF;AAEA,QAAI,IAAI,WAAW,WAAW;AAC5B,UAAI,UAAU,GAAG;AACjB,UAAI,IAAI;AACR;AAAA,IACF;AAEA,QAAI,IAAI,WAAW,SAAS,IAAI,QAAQ,SAAS;AAC/C,UAAI,UAAU,GAAG,EAAE,IAAI,MAAM;AAE7B;AAAA,IACF;AAEA,QAAI,IAAI,WAAW,SAAS,IAAI,QAAQ,UAAU;AAChD,YAAM,YAAY,IAAI,mBAAmB,aAAa,GAAG;AAEzD,UAAI;AAEJ,UAAI;AACF,iBAAS,MAAM,aAAa,GAAG;AAAA,MACjC,SAAS,OAAO;AACd,YAAI,iBAAiB,UAAU;AAC7B,cAAI,UAAU,MAAM,MAAM,EAAE,IAAI,MAAM,UAAU;AAEhD;AAAA,QACF;AAEA,YAAI,UAAU,GAAG,EAAE,IAAI,uBAAuB;AAE9C;AAAA,MACF;AAEA,uBAAiB,UAAU,SAAS,IAAI;AAExC,UAAI,SAAS;AAEb,UAAI,GAAG,SAAS,YAAY;AAC1B,iBAAS;AAET,YAAI;AACF,gBAAM,OAAO,MAAM;AAAA,QACrB,SAAS,OAAO;AACd,kBAAQ,MAAM,yBAAyB,KAAK;AAAA,QAC9C;AAEA,eAAO,iBAAiB,UAAU,SAAS;AAE3C,kBAAU,MAAM;AAAA,MAClB,CAAC;AAED,UAAI;AACF,cAAM,OAAO,QAAQ,SAAS;AAE9B,cAAM,UAAU,KAAK;AAAA,UACnB,SAAS;AAAA,UACT,QAAQ;AAAA,UACR,QAAQ,EAAE,SAAS,6BAA6B;AAAA,QAClD,CAAC;AAED,oBAAY,MAAM;AAAA,MACpB,SAAS,OAAO;AACd,YAAI,CAAC,QAAQ;AACX,kBAAQ,MAAM,+BAA+B,KAAK;AAElD,cAAI,UAAU,GAAG,EAAE,IAAI,4BAA4B;AAAA,QACrD;AAAA,MACF;AAEA;AAAA,IACF;AAEA,QAAI,IAAI,WAAW,UAAU,IAAI,KAAK,WAAW,WAAW,GAAG;AAC7D,YAAM,YAAY,IAAI;AAAA,QACpB,IAAI;AAAA,QACJ;AAAA,MACF,EAAE,aAAa,IAAI,WAAW;AAE9B,UAAI,CAAC,WAAW;AACd,YAAI,UAAU,GAAG,EAAE,IAAI,cAAc;AAErC;AAAA,MACF;AAEA,YAAM,kBACJ,iBAAiB,SAAS;AAE5B,UAAI,CAAC,iBAAiB;AACpB,YAAI,UAAU,GAAG,EAAE,IAAI,qBAAqB;AAE5C;AAAA,MACF;AAEA,YAAM,gBAAgB,kBAAkB,KAAK,GAAG;AAEhD;AAAA,IACF;AAEA,QAAI,UAAU,GAAG,EAAE,IAAI;AAAA,EACzB,CAAC;AAED,QAAM,IAAI,QAAQ,CAAC,YAAY;AAC7B,eAAW,OAAO,MAAM,MAAM,MAAM;AAClC,cAAQ,MAAS;AAAA,IACnB,CAAC;AAAA,EACH,CAAC;AAED,SAAO;AAAA,IACL,OAAO,YAAY;AACjB,iBAAW,aAAa,OAAO,OAAO,gBAAgB,GAAG;AACvD,cAAM,UAAU,MAAM;AAAA,MACxB;AAEA,aAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,mBAAW,MAAM,CAAC,UAAU;AAC1B,cAAI,OAAO;AACT,mBAAO,KAAK;AAEZ;AAAA,UACF;AAEA,kBAAQ;AAAA,QACV,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAAA,EACF;AACF;;;ACxKA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAGA,IAAM,cAAc,OAAO;AAAA,EAChC;AAAA,EACA;AAAA,EACA;AACF,MAIM;AACJ,MAAI,oBAAoB,SAAS;AAC/B,WAAO;AAAA,MACL;AAAA,MACA,OAAO,SAAS;AACd,eAAO,OAAO,aAAa,IAAI;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAEA,MAAI,oBAAoB,SAAS;AAC/B,WAAO,kBAAkB,wBAAwB,OAAO,SAAS;AAC/D,aAAO,OAAO,UAAU,KAAK,MAAM;AAAA,IACrC,CAAC;AAED,WAAO,kBAAkB,0BAA0B,OAAO,SAAS;AACjE,aAAO,OAAO,YAAY,KAAK,MAAM;AAAA,IACvC,CAAC;AAAA,EACH;AAEA,MAAI,oBAAoB,WAAW;AACjC,WAAO,kBAAkB,4BAA4B,OAAO,SAAS;AACnE,aAAO,OAAO,cAAc,KAAK,MAAM;AAAA,IACzC,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA,OAAO,SAAS;AACd,eAAO,OAAO,sBAAsB,KAAK,MAAM;AAAA,MACjD;AAAA,IACF;AAEA,WAAO,kBAAkB,2BAA2B,OAAO,SAAS;AAClE,aAAO,OAAO,aAAa,KAAK,MAAM;AAAA,IACxC,CAAC;AAAA,EACH;AAEA,MAAI,oBAAoB,OAAO;AAC7B,WAAO,kBAAkB,uBAAuB,OAAO,SAAS;AAC9D,aAAO,OAAO,SAAS,KAAK,MAAM;AAAA,IACpC,CAAC;AAED,WAAO,kBAAkB,wBAAwB,OAAO,SAAS;AAC/D,aAAO,OAAO,UAAU,KAAK,MAAM;AAAA,IACrC,CAAC;AAAA,EACH;AAEA,SAAO,kBAAkB,uBAAuB,OAAO,SAAS;AAC9D,WAAO,OAAO,SAAS,KAAK,MAAM;AAAA,EACpC,CAAC;AACH;;;AH5DA,KAAK,QAAQ,eAAe,QAAQ;AAEpC,IAAI,EAAE,iBAAiB,SAAS;AAE9B,SAAO,cAAc;AACvB;AAEA,IAAM,OAAO,MAAM,MAAM,QAAQ,QAAQ,IAAI,CAAC,EAC3C,WAAW,WAAW,EACtB,QAAQ,0BAA0B,kCAAkC,EACpE,WAAW,WAAW;AAAA,EACrB,MAAM;AAAA,EACN,UAAU;AAAA,EACV,cAAc;AAChB,CAAC,EACA,WAAW,QAAQ;AAAA,EAClB,MAAM;AAAA,EACN,OAAO;AAAA,EACP,UAAU;AACZ,CAAC,EACA,QAAQ;AAAA,EACP,OAAO;AAAA,IACL,MAAM;AAAA,IACN,UAAU;AAAA,IACV,SAAS;AAAA,EACX;AAAA,EACA,UAAU;AAAA,IACR,MAAM;AAAA,IACN,UAAU;AAAA,IACV,SAAS;AAAA,EACX;AAAA,EACA,MAAM;AAAA,IACJ,MAAM;AAAA,IACN,UAAU;AAAA,IACV,SAAS;AAAA,EACX;AACF,CAAC,EACA,KAAK,EACL,WAAW;AAEd,IAAM,UAAU,OAAO,WAAmB;AACxC,QAAM,YAAY,IAAI,qBAAqB;AAAA,IACzC,SAAS,KAAK;AAAA,IACd,MAAM,KAAK;AAAA,IACX,KAAK,QAAQ;AAAA,IACb,QAAQ;AAAA,IACR,SAAS,CAAC,UAAU;AAClB,cAAQ,MAAM,mBAAmB,KAAK;AAAA,IACxC;AAAA,EACF,CAAC;AAED,QAAM,OAAO,QAAQ,SAAS;AAChC;AAEA,IAAM,QAAQ,YAAY;AACxB,QAAM,SAAS,IAAI;AAAA,IACjB;AAAA,MACE,MAAM;AAAA,MACN,SAAS;AAAA,IACX;AAAA,IACA;AAAA,MACE,cAAc,CAAC;AAAA,IACjB;AAAA,EACF;AAEA,QAAM,QAAQ,MAAM;AAEpB,QAAM,gBAAgB,OAAO,iBAAiB;AAK9C,QAAM,qBAAqB,OAAO,sBAAsB;AAExD,UAAQ,KAAK,sCAAsC,KAAK,IAAI;AAE5D,QAAM,eAAe;AAAA,IACnB,cAAc,YAAY;AACxB,YAAM,SAAS,IAAI,OAAO,eAAe;AAAA,QACvC,cAAc;AAAA,MAChB,CAAC;AAED,kBAAY;AAAA,QACV;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAED,aAAO;AAAA,IACT;AAAA,IACA,MAAM,KAAK;AAAA,IACX,UAAU,KAAK;AAAA,EACjB,CAAC;AACH;AAEA,IAAM,OAAO,YAAY;AACvB,MAAI;AACF,UAAM,MAAM;AAAA,EACd,SAAS,OAAO;AACd,YAAQ,MAAM,6BAA6B,KAAK;AAEhD,UAAM,WAAW,GAAI;AAErB,YAAQ,KAAK,CAAC;AAAA,EAChB;AACF;AAEA,MAAM,KAAK;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-proxy",
-  "version": "2.8.2",
+  "version": "2.9.0",
   "main": "dist/MCPProxy.js",
   "scripts": {
     "build": "tsup",
@@ -19,8 +19,8 @@
   "author": "Frank Fiegel <frank@glama.ai>",
   "license": "MIT",
   "description": "A TypeScript SSE proxy for MCP servers that use stdio transport.",
-  "module": "dist/MCPProxy.js",
-  "types": "dist/MCPProxy.d.ts",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.6.0",
     "eventsource": "^3.0.5",

package/src/StdioClientTransport.ts

@@ -0,0 +1,226 @@
+/**
+ * Forked from https://github.com/modelcontextprotocol/typescript-sdk/blob/66e1508162d37c0b83b0637ebcd7f07946e3d210/src/client/stdio.ts#L90
+ */
+
+import {
+  ReadBuffer,
+  serializeMessage,
+} from "@modelcontextprotocol/sdk/shared/stdio.js";
+import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
+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 = {
+  /**
+   * The executable to run to start the server.
+   */
+  command: string;
+
+  /**
+   * Command line arguments to pass to the executable.
+   */
+  args?: string[];
+
+  /**
+   * The environment to use when spawning the process.
+   *
+   * If not specified, the result of getDefaultEnvironment() will be used.
+   */
+  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.
+   */
+  stderr?: IOType | Stream | number;
+
+  /**
+   * The working directory to use when spawning the process.
+   *
+   * If not specified, the current working directory will be inherited.
+   */
+  cwd?: string;
+
+  /**
+   * A function to call when an event occurs.
+   */
+  onEvent?: (event: TransportEvent) => void;
+};
+
+/**
+ * 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;
+
+  constructor(server: StdioServerParameters) {
+    this.serverParams = server;
+    this.onEvent = server.onEvent;
+  }
+
+  /**
+   * Starts the server process and prepares to communicate with it.
+   */
+  async start(): Promise<void> {
+    if (this.process) {
+      throw new Error(
+        "StdioClientTransport already started! If using Client class, note that connect() calls start() automatically.",
+      );
+    }
+
+    return new Promise((resolve, reject) => {
+      this.process = spawn(
+        this.serverParams.command,
+        this.serverParams.args ?? [],
+        {
+          env: this.serverParams.env,
+          stdio: ["pipe", "pipe", this.serverParams.stderr ?? "inherit"],
+          shell: false,
+          signal: this.abortController.signal,
+          cwd: this.serverParams.cwd,
+        },
+      );
+
+      this.process.on("error", (error) => {
+        if (error.name === "AbortError") {
+          // Expected when close() is called.
+          this.onclose?.();
+          return;
+        }
+
+        reject(error);
+        this.onerror?.(error);
+      });
+
+      this.process.on("spawn", () => {
+        resolve();
+      });
+
+      this.process.on("close", (_code) => {
+        this.onEvent?.({
+          type: "close",
+        });
+
+        this.process = undefined;
+        this.onclose?.();
+      });
+
+      this.process.stdin?.on("error", (error) => {
+        this.onEvent?.({
+          type: "error",
+          error,
+        });
+
+        this.onerror?.(error);
+      });
+
+      this.process.stdout?.on("data", (chunk) => {
+        this.onEvent?.({
+          type: "data",
+          chunk: chunk.toString(),
+        });
+
+        this.readBuffer.append(chunk);
+        this.processReadBuffer();
+      });
+
+      this.process.stdout?.on("error", (error) => {
+        this.onEvent?.({
+          type: "error",
+          error,
+        });
+
+        this.onerror?.(error);
+      });
+    });
+  }
+
+  /**
+   * 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 {
+        const message = this.readBuffer.readMessage();
+
+        if (message === null) {
+          break;
+        }
+
+        this.onEvent?.({
+          type: "message",
+          message,
+        });
+
+        this.onmessage?.(message);
+      } catch (error) {
+        this.onEvent?.({
+          type: "error",
+          error: error as 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);
+      }
+    });
+  }
+}