@nodetool-ai/websocket 0.8.1 → 0.8.3

package/dist/mcp-client-config.js

@@ -0,0 +1,353 @@
+/**
+ * Register the NodeTool MCP server in the config files of local agent
+ * harnesses: Claude Code, Codex, and OpenCode.
+ *
+ * Both `nodetool mcp install` and the Settings → MCP buttons write through this
+ * module, so the two surfaces produce the same entry in the same place.
+ *
+ * An entry launches NodeTool one of two ways:
+ * - `stdio`: the harness starts `nodetool mcp serve` itself. This needs no
+ *   running app. When the app is running, the stdio process forwards to it.
+ * - `http`: the harness connects to a running NodeTool server's `/mcp` mount.
+ */
+import { existsSync, mkdirSync, readFileSync, realpathSync, writeFileSync } from "node:fs";
+import { delimiter, join } from "node:path";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+export const MCP_CLIENT_IDS = [
+    "claude",
+    "codex",
+    "opencode"
+];
+export const MCP_CLIENT_LABELS = {
+    claude: "Claude Code",
+    codex: "Codex",
+    opencode: "OpenCode"
+};
+/** The name the server has in every client config. */
+const SERVER_NAME = "nodetool";
+/** The published CLI package. It has two bins, so `npx` needs `--package`. */
+const CLI_PACKAGE = "@nodetool-ai/cli";
+const CODEX_BEGIN = "# BEGIN NODETOOL MCP";
+const CODEX_END = "# END NODETOOL MCP";
+const CODEX_BLOCK = /# BEGIN NODETOOL MCP[\s\S]*?# END NODETOOL MCP\n?/;
+/**
+ * Codex and OpenCode give a stdio server a short start window by default. The
+ * first `npx` launch downloads the CLI, and a video generation can run for
+ * minutes, so both limits are raised.
+ */
+const STARTUP_TIMEOUT_SEC = 120;
+const TOOL_TIMEOUT_SEC = 900;
+/** The stdio launch for `nodetool mcp serve`. */
+export function stdioLaunch(options) {
+    if (options.npx) {
+        return {
+            transport: "stdio",
+            command: "npx",
+            args: ["-y", `--package=${CLI_PACKAGE}`, "nodetool", "mcp", "serve"]
+        };
+    }
+    return { transport: "stdio", command: "nodetool", args: ["mcp", "serve"] };
+}
+/** The first executable with this name on `PATH`, or null. */
+function findOnPath(name, env) {
+    const dirs = (env["PATH"] ?? "").split(delimiter).filter(Boolean);
+    const exts = process.platform === "win32"
+        ? (env["PATHEXT"] ?? ".EXE;.CMD;.BAT").split(";")
+        : [""];
+    for (const dir of dirs) {
+        for (const ext of exts) {
+            const candidate = join(dir, name + ext);
+            if (existsSync(candidate))
+                return candidate;
+        }
+    }
+    return null;
+}
+/** True when an executable with this name is on `PATH`. */
+export function isOnPath(name, env = process.env) {
+    return findOnPath(name, env) !== null;
+}
+/**
+ * True when the `nodetool` a client would run is this TypeScript CLI. The
+ * legacy Python installer puts a bash `nodetool` on `PATH` that has no
+ * `mcp serve`, so a match by name alone is not enough.
+ */
+export function hasNodetoolCli(env = process.env) {
+    const found = findOnPath("nodetool", env);
+    if (!found)
+        return false;
+    try {
+        const head = readFileSync(realpathSync(found), "utf8").slice(0, 512);
+        return process.platform === "win32"
+            ? head.includes("nodetool.js")
+            : /^#!.*\bnode\b/.test(head);
+    }
+    catch {
+        return false;
+    }
+}
+/** One line a person can read: the command, or the URL. */
+export function describeLaunch(launch) {
+    return launch.transport === "stdio"
+        ? [launch.command, ...launch.args].join(" ")
+        : launch.url;
+}
+/** The `mcpServers` JSON block most MCP clients accept. */
+export function mcpServersSnippet(launch) {
+    const server = launch.transport === "stdio"
+        ? { command: launch.command, args: launch.args }
+        : { type: "http", url: launch.url };
+    return JSON.stringify({ mcpServers: { [SERVER_NAME]: server } }, null, 2);
+}
+export function mcpClientConfigPath(client, home) {
+    switch (client) {
+        case "claude":
+            return join(home, ".claude.json");
+        case "codex":
+            return join(home, ".codex", "config.toml");
+        case "opencode":
+            return join(home, ".config", "opencode", "opencode.json");
+    }
+}
+/** True when the client looks installed: its config dir or its binary. */
+export function isMcpClientPresent(client, home) {
+    switch (client) {
+        case "claude":
+            return (existsSync(join(home, ".claude.json")) ||
+                existsSync(join(home, ".claude")) ||
+                isOnPath("claude"));
+        case "codex":
+            return existsSync(join(home, ".codex")) || isOnPath("codex");
+        case "opencode":
+            return (existsSync(join(home, ".config", "opencode")) || isOnPath("opencode"));
+    }
+}
+function readJson(path) {
+    if (!existsSync(path))
+        return {};
+    return JSON.parse(readFileSync(path, "utf8"));
+}
+function writeJson(path, value) {
+    writeFileSync(path, JSON.stringify(value, null, 2) + "\n");
+}
+function asObject(value) {
+    return value && typeof value === "object" && !Array.isArray(value)
+        ? value
+        : {};
+}
+function stringArray(value) {
+    return Array.isArray(value)
+        ? value.filter((item) => typeof item === "string")
+        : [];
+}
+/**
+ * Earlier installers wrote the Claude Code entry under `projects[<home>]`. That
+ * is the local scope of the home directory, so the server appeared only in
+ * sessions started in `~`. Remove that entry when it is the loopback HTTP
+ * shape those installers wrote. Anything else there belongs to the user.
+ */
+function removeLegacyClaudeEntry(config, home) {
+    const project = asObject(asObject(config["projects"])[home]);
+    const servers = asObject(project["mcpServers"]);
+    const entry = asObject(servers[SERVER_NAME]);
+    const url = typeof entry["url"] === "string" ? entry["url"] : "";
+    if (entry["type"] !== "http" || !/^https?:\/\/127\.0\.0\.1:\d+\/mcp$/.test(url)) {
+        return false;
+    }
+    delete servers[SERVER_NAME];
+    return true;
+}
+function claudeEntry(launch) {
+    return launch.transport === "stdio"
+        ? { type: "stdio", command: launch.command, args: launch.args, env: {} }
+        : { type: "http", url: launch.url };
+}
+function opencodeEntry(launch) {
+    const timeout = STARTUP_TIMEOUT_SEC * 1000;
+    return launch.transport === "stdio"
+        ? {
+            type: "local",
+            command: [launch.command, ...launch.args],
+            enabled: true,
+            timeout
+        }
+        : { type: "remote", url: launch.url, enabled: true, timeout };
+}
+/** TOML basic strings accept JSON string escaping, so quotes cannot break out. */
+function codexBlock(launch) {
+    const location = launch.transport === "stdio"
+        ? [
+            `command = ${JSON.stringify(launch.command)}`,
+            `args = ${JSON.stringify(launch.args)}`
+        ]
+        : [`url = ${JSON.stringify(launch.url)}`];
+    return [
+        CODEX_BEGIN,
+        `[mcp_servers.${SERVER_NAME}]`,
+        ...location,
+        `startup_timeout_sec = ${STARTUP_TIMEOUT_SEC}`,
+        `tool_timeout_sec = ${TOOL_TIMEOUT_SEC}`,
+        CODEX_END
+    ].join("\n");
+}
+function launchFromEntry(entry) {
+    if (typeof entry["url"] === "string") {
+        return { transport: "http", url: entry["url"] };
+    }
+    const command = entry["command"];
+    if (typeof command === "string") {
+        return { transport: "stdio", command, args: stringArray(entry["args"]) };
+    }
+    // OpenCode stores the command and its arguments as one array.
+    const [head, ...rest] = stringArray(command);
+    return head ? { transport: "stdio", command: head, args: rest } : null;
+}
+function parseCodexLaunch(block) {
+    const url = /^url\s*=\s*(".*")\s*$/m.exec(block)?.[1];
+    if (url)
+        return { transport: "http", url: JSON.parse(url) };
+    const command = /^command\s*=\s*(".*")\s*$/m.exec(block)?.[1];
+    if (!command)
+        return null;
+    const args = /^args\s*=\s*(\[.*\])\s*$/m.exec(block)?.[1];
+    return {
+        transport: "stdio",
+        command: JSON.parse(command),
+        args: args ? stringArray(JSON.parse(args)) : []
+    };
+}
+/** The NodeTool entry the client has now, or null. Throws on unreadable config. */
+export function readMcpClientEntry(client, home) {
+    const path = mcpClientConfigPath(client, home);
+    if (!existsSync(path))
+        return null;
+    switch (client) {
+        case "claude": {
+            const servers = asObject(readJson(path)["mcpServers"]);
+            return launchFromEntry(asObject(servers[SERVER_NAME]));
+        }
+        case "codex": {
+            const block = CODEX_BLOCK.exec(readFileSync(path, "utf8"))?.[0];
+            return block ? parseCodexLaunch(block) : null;
+        }
+        case "opencode": {
+            const servers = asObject(readJson(path)["mcp"]);
+            return launchFromEntry(asObject(servers[SERVER_NAME]));
+        }
+    }
+}
+/** Write the NodeTool entry. Returns the config file path. */
+export function writeMcpClientEntry(client, launch, home) {
+    const path = mcpClientConfigPath(client, home);
+    switch (client) {
+        case "claude": {
+            // Top-level `mcpServers` is Claude Code's user scope: every project.
+            const config = readJson(path);
+            removeLegacyClaudeEntry(config, home);
+            const servers = asObject(config["mcpServers"]);
+            servers[SERVER_NAME] = claudeEntry(launch);
+            config["mcpServers"] = servers;
+            writeJson(path, config);
+            return path;
+        }
+        case "codex": {
+            mkdirSync(join(home, ".codex"), { recursive: true });
+            const block = codexBlock(launch);
+            const content = existsSync(path) ? readFileSync(path, "utf8") : "";
+            const next = CODEX_BLOCK.test(content)
+                ? content.replace(CODEX_BLOCK, block + "\n")
+                : (content.trim() ? content.trimEnd() + "\n\n" : "") + block + "\n";
+            writeFileSync(path, next);
+            return path;
+        }
+        case "opencode": {
+            mkdirSync(join(home, ".config", "opencode"), { recursive: true });
+            const config = readJson(path);
+            const servers = asObject(config["mcp"]);
+            servers[SERVER_NAME] = opencodeEntry(launch);
+            config["mcp"] = servers;
+            writeJson(path, config);
+            return path;
+        }
+    }
+}
+/** Remove the NodeTool entry. Returns false when there was none. */
+export function removeMcpClientEntry(client, home) {
+    const path = mcpClientConfigPath(client, home);
+    if (!existsSync(path))
+        return false;
+    switch (client) {
+        case "claude": {
+            const config = readJson(path);
+            const servers = asObject(config["mcpServers"]);
+            const hadEntry = SERVER_NAME in servers;
+            delete servers[SERVER_NAME];
+            const hadLegacy = removeLegacyClaudeEntry(config, home);
+            if (!hadEntry && !hadLegacy)
+                return false;
+            writeJson(path, config);
+            return true;
+        }
+        case "codex": {
+            const content = readFileSync(path, "utf8");
+            if (!CODEX_BLOCK.test(content))
+                return false;
+            writeFileSync(path, content.replace(CODEX_BLOCK, "").trimEnd() + "\n");
+            return true;
+        }
+        case "opencode": {
+            const config = readJson(path);
+            const servers = asObject(config["mcp"]);
+            if (!(SERVER_NAME in servers))
+                return false;
+            delete servers[SERVER_NAME];
+            writeJson(path, config);
+            return true;
+        }
+    }
+}
+/**
+ * Start the server the way a harness would, complete the MCP handshake, and
+ * list its tools. A stdio probe also warms the `npx` cache, so the harness's
+ * own first start stays inside its startup timeout.
+ */
+export async function probeMcpLaunch(launch, timeoutMs = STARTUP_TIMEOUT_SEC * 1000) {
+    const started = Date.now();
+    let stderr = "";
+    let transport;
+    if (launch.transport === "stdio") {
+        const stdio = new StdioClientTransport({
+            command: launch.command,
+            args: launch.args,
+            stderr: "pipe"
+        });
+        stdio.stderr?.on("data", (chunk) => {
+            stderr = (stderr + chunk.toString()).slice(-2000);
+        });
+        transport = stdio;
+    }
+    else {
+        transport = new StreamableHTTPClientTransport(new URL(launch.url));
+    }
+    const client = new Client({ name: "nodetool-mcp-probe", version: "1.0.0" });
+    try {
+        await client.connect(transport, { timeout: timeoutMs });
+        const { tools } = await client.listTools(undefined, { timeout: timeoutMs });
+        return {
+            serverName: client.getServerVersion()?.name ?? "",
+            tools: tools.map((tool) => tool.name),
+            elapsedMs: Date.now() - started
+        };
+    }
+    catch (error) {
+        const reason = error instanceof Error ? error.message : String(error);
+        const tail = stderr.trim().split("\n").slice(-5).join("\n");
+        throw new Error(tail ? `${reason}\n${tail}` : reason);
+    }
+    finally {
+        await client.close();
+    }
+}
+//# sourceMappingURL=mcp-client-config.js.map

package/dist/mcp-client-config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-client-config.js","sourceRoot":"","sources":["../src/mcp-client-config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,UAAU,EACV,SAAS,EACT,YAAY,EACZ,YAAY,EACZ,aAAa,EACd,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AASnG,MAAM,CAAC,MAAM,cAAc,GAA2B;IACpD,QAAQ;IACR,OAAO;IACP,UAAU;CACX,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,MAAM,EAAE,aAAa;IACrB,KAAK,EAAE,OAAO;IACd,QAAQ,EAAE,UAAU;CACiB,CAAC;AAExC,sDAAsD;AACtD,MAAM,WAAW,GAAG,UAAU,CAAC;AAE/B,8EAA8E;AAC9E,MAAM,WAAW,GAAG,kBAAkB,CAAC;AAEvC,MAAM,WAAW,GAAG,sBAAsB,CAAC;AAC3C,MAAM,SAAS,GAAG,oBAAoB,CAAC;AACvC,MAAM,WAAW,GAAG,mDAAmD,CAAC;AAExE;;;;GAIG;AACH,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAChC,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAE7B,iDAAiD;AACjD,MAAM,UAAU,WAAW,CAAC,OAAyB;IACnD,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAChB,OAAO;YACL,SAAS,EAAE,OAAO;YAClB,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,IAAI,EAAE,aAAa,WAAW,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,OAAO,CAAC;SACrE,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC;AAC7E,CAAC;AAED,8DAA8D;AAC9D,SAAS,UAAU,CAAC,IAAY,EAAE,GAAsB;IACtD,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,MAAM,IAAI,GACR,OAAO,CAAC,QAAQ,KAAK,OAAO;QAC1B,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,gBAAgB,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC;QACjD,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACX,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,GAAG,GAAG,CAAC,CAAC;YACxC,IAAI,UAAU,CAAC,SAAS,CAAC;gBAAE,OAAO,SAAS,CAAC;QAC9C,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,2DAA2D;AAC3D,MAAM,UAAU,QAAQ,CAAC,IAAY,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG;IACtD,OAAO,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;AACxC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,cAAc,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG;IAC9C,MAAM,KAAK,GAAG,UAAU,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;IAC1C,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,YAAY,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QACrE,OAAO,OAAO,CAAC,QAAQ,KAAK,OAAO;YACjC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC;YAC9B,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,2DAA2D;AAC3D,MAAM,UAAU,cAAc,CAAC,MAAiB;IAC9C,OAAO,MAAM,CAAC,SAAS,KAAK,OAAO;QACjC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;QAC5C,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;AACjB,CAAC;AAED,2DAA2D;AAC3D,MAAM,UAAU,iBAAiB,CAAC,MAAiB;IACjD,MAAM,MAAM,GACV,MAAM,CAAC,SAAS,KAAK,OAAO;QAC1B,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;QAChD,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;IACxC,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAmB,EAAE,IAAY;IACnE,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ;YACX,OAAO,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QACpC,KAAK,OAAO;YACV,OAAO,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,aAAa,CAAC,CAAC;QAC7C,KAAK,UAAU;YACb,OAAO,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;IAC9D,CAAC;AACH,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,kBAAkB,CAAC,MAAmB,EAAE,IAAY;IAClE,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ;YACX,OAAO,CACL,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;gBACtC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACjC,QAAQ,CAAC,QAAQ,CAAC,CACnB,CAAC;QACJ,KAAK,OAAO;YACV,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC/D,KAAK,UAAU;YACb,OAAO,CACL,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,CACtE,CAAC;IACN,CAAC;AACH,CAAC;AAID,SAAS,QAAQ,CAAC,IAAY;IAC5B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IACjC,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAe,CAAC;AAC9D,CAAC;AAED,SAAS,SAAS,CAAC,IAAY,EAAE,KAAiB;IAChD,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAChE,CAAC,CAAE,KAAoB;QACvB,CAAC,CAAC,EAAE,CAAC;AACT,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACzB,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC;QAClE,CAAC,CAAC,EAAE,CAAC;AACT,CAAC;AAED;;;;;GAKG;AACH,SAAS,uBAAuB,CAAC,MAAkB,EAAE,IAAY;IAC/D,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7D,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7C,MAAM,GAAG,GAAG,OAAO,KAAK,CAAC,KAAK,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACjE,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,MAAM,IAAI,CAAC,oCAAoC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAChF,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC;IAC5B,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,WAAW,CAAC,MAAiB;IACpC,OAAO,MAAM,CAAC,SAAS,KAAK,OAAO;QACjC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE;QACxE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,aAAa,CAAC,MAAiB;IACtC,MAAM,OAAO,GAAG,mBAAmB,GAAG,IAAI,CAAC;IAC3C,OAAO,MAAM,CAAC,SAAS,KAAK,OAAO;QACjC,CAAC,CAAC;YACE,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC;YACzC,OAAO,EAAE,IAAI;YACb,OAAO;SACR;QACH,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAClE,CAAC;AAED,kFAAkF;AAClF,SAAS,UAAU,CAAC,MAAiB;IACnC,MAAM,QAAQ,GACZ,MAAM,CAAC,SAAS,KAAK,OAAO;QAC1B,CAAC,CAAC;YACE,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;YAC7C,UAAU,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;SACxC;QACH,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC9C,OAAO;QACL,WAAW;QACX,gBAAgB,WAAW,GAAG;QAC9B,GAAG,QAAQ;QACX,yBAAyB,mBAAmB,EAAE;QAC9C,sBAAsB,gBAAgB,EAAE;QACxC,SAAS;KACV,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,KAAiB;IACxC,IAAI,OAAO,KAAK,CAAC,KAAK,CAAC,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;IAClD,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;IACjC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;IAC3E,CAAC;IACD,8DAA8D;IAC9D,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IAC7C,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACzE,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACrC,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACtD,IAAI,GAAG;QAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAW,EAAE,CAAC;IACtE,MAAM,OAAO,GAAG,4BAA4B,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9D,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAC1B,MAAM,IAAI,GAAG,2BAA2B,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1D,OAAO;QACL,SAAS,EAAE,OAAO;QAClB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAW;QACtC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;KAChD,CAAC;AACJ,CAAC;AAED,mFAAmF;AACnF,MAAM,UAAU,kBAAkB,CAChC,MAAmB,EACnB,IAAY;IAEZ,MAAM,IAAI,GAAG,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACnC,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YACvD,OAAO,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;QACD,KAAK,OAAO,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAChE,OAAO,KAAK,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAChD,CAAC;QACD,KAAK,UAAU,EAAE,CAAC;YAChB,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAChD,OAAO,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;AACH,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,mBAAmB,CACjC,MAAmB,EACnB,MAAiB,EACjB,IAAY;IAEZ,MAAM,IAAI,GAAG,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/C,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ,EAAE,CAAC;YACd,qEAAqE;YACrE,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,uBAAuB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;YAC/C,OAAO,CAAC,WAAW,CAAC,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC;YAC/B,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,OAAO,EAAE,CAAC;YACb,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACnE,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC;gBACpC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,GAAG,IAAI,CAAC;gBAC5C,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,KAAK,GAAG,IAAI,CAAC;YACtE,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,UAAU,EAAE,CAAC;YAChB,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAClE,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACxC,OAAO,CAAC,WAAW,CAAC,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;YACxB,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;AACH,CAAC;AAED,oEAAoE;AACpE,MAAM,UAAU,oBAAoB,CAAC,MAAmB,EAAE,IAAY;IACpE,MAAM,IAAI,GAAG,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IACpC,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ,EAAE,CAAC;YACd,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,WAAW,IAAI,OAAO,CAAC;YACxC,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC;YAC5B,MAAM,SAAS,GAAG,uBAAuB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACxD,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS;gBAAE,OAAO,KAAK,CAAC;YAC1C,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,OAAO,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC7C,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;YACvE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,UAAU,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACxC,IAAI,CAAC,CAAC,WAAW,IAAI,OAAO,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC5C,OAAO,OAAO,CAAC,WAAW,CAAC,CAAC;YAC5B,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;AACH,CAAC;AAQD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAiB,EACjB,SAAS,GAAG,mBAAmB,GAAG,IAAI;IAEtC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,SAAoB,CAAC;IACzB,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,IAAI,oBAAoB,CAAC;YACrC,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YACzC,MAAM,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QACH,SAAS,GAAG,KAAK,CAAC;IACpB,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IAC5E,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5E,OAAO;YACL,UAAU,EAAE,MAAM,CAAC,gBAAgB,EAAE,EAAE,IAAI,IAAI,EAAE;YACjD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;YACrC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO;SAChC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5D,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;YAAS,CAAC;QACT,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/mcp-server.d.ts

@@ -57,6 +57,13 @@ export declare function createMcpServer(options?: McpServerOptions): McpServer;
 export declare function createMcpStdioTransport(): StdioServerTransport;
 /** Connect stdio to the local API process when its MCP mount is available. */
 export declare function forwardMcpStdioToLocalServer(url?: string, stdio?: StdioServerTransport): Promise<boolean>;
+/** Time a restarted local server gets to accept connections again. */
+export declare const MCP_FORWARDER_RECONNECT_WINDOW_MS = 30000;
+/**
+ * "session-lost" and "unreachable" mean the server never ran the request;
+ * "other" covers everything else, including a connection dropped mid-call.
+ */
+export declare function classifyForwardFailure(error: unknown): "session-lost" | "unreachable" | "other";
 /**
  * How long an HTTP session may sit idle before it is closed. Every session
  * holds a transport, an `McpServer`, and that session's whole tool registry,

package/dist/mcp-server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../src/mcp-server.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAIjF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAG/E,MAAM,WAAW,gBAAgB;IAC/B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,wBAAwB,CAAC,EAAE,uBAAuB,CAAC;IACnD,+EAA+E;IAC/E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,aAAa,GAAG,gBAAgB,GAAG,cAAc,CAAC;KAC3D,CAAC;IACF,yEAAyE;IACzE,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B;;;;;OAKG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC;AAED;;;GAGG;AACH,eAAO,MAAM,0BAA0B,QAGuB,CAAC;AAI/D;;;;;;GAMG;AACH,wBAAgB,eAAe,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAgBrE;AAED;;GAEG;AACH,wBAAgB,uBAAuB,IAAI,oBAAoB,CAE9D;AAED,8EAA8E;AAC9E,wBAAsB,4BAA4B,CAChD,GAAG,GAAE,MAA+B,EACpC,KAAK,GAAE,oBAAgD,GACtD,OAAO,CAAC,OAAO,CAAC,CA2DlB;AAED;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,QAAiB,CAAC;AACtD,kFAAkF;AAClF,eAAO,MAAM,yBAAyB,IAAI,CAAC;AAC3C,6EAA6E;AAC7E,eAAO,MAAM,gBAAgB,MAAM,CAAC;AA0EpC;uCACuC;AACvC,wBAAgB,uBAAuB,IAAI,IAAI,CAI9C;AAgDD,wBAAgB,oBAAoB,IAAI,MAAM,CAK7C;AAgBD;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,gBAAgB,GACzB,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,CA+F1B"}
+{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../src/mcp-server.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAMpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAejF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAG/E,MAAM,WAAW,gBAAgB;IAC/B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,wBAAwB,CAAC,EAAE,uBAAuB,CAAC;IACnD,+EAA+E;IAC/E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,aAAa,GAAG,gBAAgB,GAAG,cAAc,CAAC;KAC3D,CAAC;IACF,yEAAyE;IACzE,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B;;;;;OAKG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAChC;AAED;;;GAGG;AACH,eAAO,MAAM,0BAA0B,QAGuB,CAAC;AAI/D;;;;;;GAMG;AACH,wBAAgB,eAAe,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAgBrE;AAED;;GAEG;AACH,wBAAgB,uBAAuB,IAAI,oBAAoB,CAE9D;AAED,8EAA8E;AAC9E,wBAAsB,4BAA4B,CAChD,GAAG,GAAE,MAA+B,EACpC,KAAK,GAAE,oBAAgD,GACtD,OAAO,CAAC,OAAO,CAAC,CAkMlB;AAED,sEAAsE;AACtE,eAAO,MAAM,iCAAiC,QAAS,CAAC;AAiBxD;;;GAGG;AACH,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,OAAO,GACb,cAAc,GAAG,aAAa,GAAG,OAAO,CAc1C;AAED;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,QAAiB,CAAC;AACtD,kFAAkF;AAClF,eAAO,MAAM,yBAAyB,IAAI,CAAC;AAC3C,6EAA6E;AAC7E,eAAO,MAAM,gBAAgB,MAAM,CAAC;AA0EpC;uCACuC;AACvC,wBAAgB,uBAAuB,IAAI,IAAI,CAI9C;AAgDD,wBAAgB,oBAAoB,IAAI,MAAM,CAK7C;AAgBD;;;GAGG;AACH,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,gBAAgB,GACzB,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,CA+F1B"}

package/dist/mcp-server.js

@@ -7,9 +7,11 @@
  */
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { StreamableHTTPClientTransport, StreamableHTTPError } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { WebStandardStreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/webStandardStreamableHttp.js";
+import { ErrorCode, isInitializedNotification, isInitializeRequest, isJSONRPCRequest, isJSONRPCResultResponse } from "@modelcontextprotocol/sdk/types.js";
+import { Agent as UndiciAgent } from "undici";
 import { MCP_GUEST_CONTRACT } from "@nodetool-ai/agents";
 import { createLogger } from "@nodetool-ai/config";
 import { registerAgentMcpTools } from "./mcp-agent-tools.js";
@@ -70,33 +72,152 @@ export async function forwardMcpStdioToLocalServer(url = getLocalMcpServerUrl(),
         }
         await probe.close();
     }
-    const http = new StreamableHTTPClientTransport(endpoint);
-    http.onmessage = (message) => {
-        if ("result" in message && "protocolVersion" in message.result) {
-            const version = message.result.protocolVersion;
-            if (typeof version === "string")
-                http.setProtocolVersion(version);
+    // The client's own handshake, kept so a replacement session can replay it.
+    let initializeRequest;
+    let initializedNotification;
+    // While a replacement session replays `initialize`, its response belongs to
+    // the forwarder, not to the client, which already received one.
+    let replayResponse;
+    const openTransport = () => {
+        const transport = new StreamableHTTPClientTransport(endpoint, {
+            fetch: fetchWithoutTimeouts
+        });
+        transport.onmessage = (message) => {
+            if (isJSONRPCResultResponse(message) && "protocolVersion" in message.result) {
+                const version = message.result.protocolVersion;
+                if (typeof version === "string")
+                    transport.setProtocolVersion(version);
+            }
+            if (replayResponse?.transport === transport &&
+                "id" in message &&
+                message.id === replayResponse.id) {
+                replayResponse.resolve();
+                return;
+            }
+            void stdio.send(message);
+        };
+        transport.onerror = (error) => {
+            log.warn("Local MCP HTTP transport failed", { error: error.message });
+        };
+        return transport;
+    };
+    const replayHandshake = async (transport) => {
+        if (!initializeRequest) {
+            throw new Error("The client never initialized the MCP session");
+        }
+        const request = initializeRequest;
+        let timer;
+        try {
+            await Promise.race([
+                new Promise((resolve, reject) => {
+                    replayResponse = { transport, id: request.id, resolve };
+                    transport.send(request).catch(reject);
+                }),
+                new Promise((_, reject) => {
+                    timer = setTimeout(() => reject(new Error("Replaying the MCP handshake timed out")), MCP_FORWARDER_HANDSHAKE_TIMEOUT_MS);
+                })
+            ]);
+        }
+        finally {
+            clearTimeout(timer);
+            replayResponse = undefined;
+        }
+        if (initializedNotification) {
+            await transport.send(initializedNotification);
         }
-        void stdio.send(message);
     };
-    http.onerror = (error) => {
-        log.warn("Local MCP HTTP transport failed", { error: error.message });
+    let http = openTransport();
+    let reconnection;
+    // One replacement session for every request that failed on the same one.
+    // A restarting server refuses connections for a few seconds, so retry those.
+    const reconnect = (failed) => {
+        if (http !== failed)
+            return reconnection ?? Promise.resolve();
+        reconnection ??= (async () => {
+            void failed.close().catch(() => undefined);
+            const deadline = Date.now() + MCP_FORWARDER_RECONNECT_WINDOW_MS;
+            for (;;) {
+                const next = openTransport();
+                try {
+                    await next.start();
+                    await replayHandshake(next);
+                    http = next;
+                    log.info("Reopened the local MCP session");
+                    return;
+                }
+                catch (error) {
+                    await next.close().catch(() => undefined);
+                    if (classifyForwardFailure(error) !== "unreachable" ||
+                        Date.now() >= deadline) {
+                        throw error;
+                    }
+                    await new Promise((resolve) => setTimeout(resolve, MCP_FORWARDER_RECONNECT_DELAY_MS));
+                }
+            }
+        })().finally(() => {
+            reconnection = undefined;
+        });
+        return reconnection;
+    };
+    // A request is resent only when the server provably never ran it: its
+    // session was gone (404) or nothing accepted the connection. A connection
+    // that drops mid-call may have run the tool already, so that request fails.
+    const forward = async (message) => {
+        await reconnection?.catch(() => undefined);
+        const transport = http;
+        try {
+            await transport.send(message);
+        }
+        catch (error) {
+            if (!initializeRequest ||
+                isInitializeRequest(message) ||
+                classifyForwardFailure(error) === "other") {
+                throw error;
+            }
+            log.info("Local MCP session lost; reconnecting", {
+                error: error instanceof Error ? error.message : String(error)
+            });
+            await reconnect(transport);
+            await http.send(message);
+        }
     };
     let initialization = Promise.resolve();
     stdio.onmessage = (message) => {
-        const send = initialization.then(() => http.send(message));
-        if ("method" in message && message.method === "initialize") {
+        if (isInitializeRequest(message)) {
+            initializeRequest = message;
+        }
+        else if (isInitializedNotification(message)) {
+            initializedNotification = message;
+        }
+        const send = initialization.then(() => forward(message));
+        if (isInitializeRequest(message)) {
             initialization = send;
         }
         void send.catch((error) => {
-            log.warn("Forwarding MCP request failed", {
-                error: error instanceof Error ? error.message : String(error)
-            });
-            void stdio.close();
+            const reason = error instanceof Error ? error.message : String(error);
+            log.warn("Forwarding MCP request failed", { error: reason });
+            // Without a session there is nothing to keep open.
+            if (isInitializeRequest(message)) {
+                void stdio.close();
+                return;
+            }
+            // Fail this one request and keep stdio open: the next request
+            // reconnects once the local server is back.
+            if (isJSONRPCRequest(message)) {
+                void stdio.send({
+                    jsonrpc: "2.0",
+                    id: message.id,
+                    error: {
+                        code: ErrorCode.ConnectionClosed,
+                        message: `NodeTool server request failed: ${reason}`
+                    }
+                });
+            }
         });
     };
     stdio.onclose = () => {
         void (async () => {
+            await reconnection?.catch(() => undefined);
             if (http.sessionId) {
                 await http.terminateSession().catch(() => undefined);
             }
@@ -107,6 +228,32 @@ export async function forwardMcpStdioToLocalServer(url = getLocalMcpServerUrl(),
     await stdio.start();
     return true;
 }
+/** Time a restarted local server gets to accept connections again. */
+export const MCP_FORWARDER_RECONNECT_WINDOW_MS = 30_000;
+const MCP_FORWARDER_RECONNECT_DELAY_MS = 500;
+const MCP_FORWARDER_HANDSHAKE_TIMEOUT_MS = 10_000;
+/**
+ * The mount answers with JSON (`enableJsonResponse`), so response headers
+ * arrive only when the tool finishes. undici's default 300 s headers timeout
+ * would fail every call longer than five minutes.
+ */
+const noTimeoutDispatcher = new UndiciAgent({ headersTimeout: 0, bodyTimeout: 0 });
+const fetchWithoutTimeouts = (url, init) => fetch(url, { ...init, dispatcher: noTimeoutDispatcher });
+/**
+ * "session-lost" and "unreachable" mean the server never ran the request;
+ * "other" covers everything else, including a connection dropped mid-call.
+ */
+export function classifyForwardFailure(error) {
+    if (error instanceof StreamableHTTPError && error.code === 404) {
+        return "session-lost";
+    }
+    for (let current = error; current instanceof Error; current = current.cause) {
+        if (current.code === "ECONNREFUSED") {
+            return "unreachable";
+        }
+    }
+    return "other";
+}
 /**
  * How long an HTTP session may sit idle before it is closed. Every session
  * holds a transport, an `McpServer`, and that session's whole tool registry,