@agimon-ai/mcp-proxy 0.10.0 → 0.10.2

package/dist/cli.cjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-const require_src = require("./src-DIQjnEBk.cjs");
+const require_src = require("./src-DMlOlvPm.cjs");
 let node_fs = require("node:fs");
 let node_fs_promises = require("node:fs/promises");
 let js_yaml = require("js-yaml");
@@ -7,18 +7,18 @@ js_yaml = require_src.__toESM(js_yaml);
 let node_crypto = require("node:crypto");
 let node_path = require("node:path");
 node_path = require_src.__toESM(node_path);
+let _agimon_ai_foundation_process_registry = require("@agimon-ai/foundation-process-registry");
 let node_child_process = require("node:child_process");
 let liquidjs = require("liquidjs");
 let commander = require("commander");
 let node_url = require("node:url");
-let _agimon_ai_foundation_process_registry = require("@agimon-ai/foundation-process-registry");
 let _agimon_ai_foundation_port_registry = require("@agimon-ai/foundation-port-registry");
-//#region src/commands/prestart-http.ts
+//#region src/commands/http-runtime.ts
 /**
-* Prestart HTTP Command
+* HTTP Runtime Helper
 *
 * Starts an mcp-proxy HTTP runtime in the background, waits until it is healthy,
-* and then exits so the runtime can be reused by other commands.
+* and returns the runtime details for callers that need to reuse it.
 */
 const WORKSPACE_MARKERS = [
 	"pnpm-workspace.yaml",
@@ -26,7 +26,7 @@ const WORKSPACE_MARKERS = [
 	".git"
 ];
 const DEFAULT_HOST$1 = "localhost";
-const DEFAULT_TIMEOUT_MS = 12e4;
+const DEFAULT_HTTP_RUNTIME_TIMEOUT_MS = 12e4;
 const POLL_INTERVAL_MS = 250;
 function resolveWorkspaceRoot(startPath = process.env.PROJECT_PATH || process.cwd()) {
 	let current = node_path.default.resolve(startPath);
@@ -82,7 +82,7 @@ function buildCliCandidates() {
 	throw new Error("Unable to locate mcp-proxy CLI entrypoint");
 }
 function parseTimeoutMs(value) {
-	if (!value) return DEFAULT_TIMEOUT_MS;
+	if (!value) return DEFAULT_HTTP_RUNTIME_TIMEOUT_MS;
 	const parsed = Number.parseInt(value, 10);
 	if (!Number.isInteger(parsed) || parsed <= 0) throw new Error(`Invalid timeout value: ${value}`);
 	return parsed;
@@ -208,17 +208,12 @@ async function prestartHttpRuntime(options) {
 		throw new Error(`Failed to prestart HTTP runtime '${serverId}': ${error instanceof Error ? error.message : String(error)}`, { cause: error });
 	}
 }
-const prestartHttpCommand = new commander.Command("prestart-http").description("Start an mcp-proxy HTTP runtime in the background and wait until it is healthy").option("--id <id>", "Server identifier to assign to the runtime").option("--host <host>", "Host to bind to", DEFAULT_HOST$1).option("-p, --port <port>", "Preferred HTTP port for the runtime", (value) => Number.parseInt(value, 10)).option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--registry-path <path>", "Custom registry path or directory for service discovery").option("--registry-dir <path>", "Custom registry directory for service discovery").option("--timeout-ms <ms>", "How long to wait for the runtime to become healthy", String(DEFAULT_TIMEOUT_MS)).action(async (options) => {
-	try {
-		const { host, port, serverId, workspaceRoot } = await prestartHttpRuntime(options);
-		process.stdout.write(`mcp-proxy HTTP runtime ready at http://${host}:${port} (${serverId})\n`);
-		process.stdout.write(`runtimeId=${serverId}\n`);
-		process.stdout.write(`runtimeUrl=http://${host}:${port}\n`);
-		process.stdout.write(`workspaceRoot=${workspaceRoot}\n`);
-	} catch (error) {
-		throw new Error(`Failed to prestart HTTP runtime '${options.id || "generated-server-id"}': ${error instanceof Error ? error.message : String(error)}`, { cause: error });
-	}
-});
+function writePrestartHttpResult(result) {
+	process.stdout.write(`mcp-proxy HTTP runtime ready at http://${result.host}:${result.port} (${result.serverId})\n`);
+	process.stdout.write(`runtimeId=${result.serverId}\n`);
+	process.stdout.write(`runtimeUrl=http://${result.host}:${result.port}\n`);
+	process.stdout.write(`workspaceRoot=${result.workspaceRoot}\n`);
+}
 //#endregion
 //#region src/commands/bootstrap.ts
 function toErrorMessage$9(error) {
@@ -1156,9 +1151,14 @@ async function createAndStartHttpRuntime(serverOptions, config, resolvedConfigPa
 	console.error(`Runtime state: http://${runtimeRecord.host}:${runtimeRecord.port} (${runtimeRecord.serverId})`);
 }
 async function startStdioTransport(serverOptions) {
+	let server;
 	try {
-		await startServer(new require_src.StdioTransportHandler(await require_src.createServer(serverOptions), createStdioSafeLogger()));
+		server = await require_src.createServer(serverOptions);
+		await startServer(new require_src.StdioTransportHandler(server, createStdioSafeLogger()), async () => {
+			await server?.dispose?.();
+		});
 	} catch (error) {
+		await server?.dispose?.();
 		throw new Error(`Failed to start stdio transport: ${toErrorMessage$3(error)}`);
 	}
 }
@@ -1272,11 +1272,33 @@ async function startTransport(transportType, options, resolvedConfigPath, server
 		throw new Error(`Failed to start transport '${transportType}': ${toErrorMessage$3(error)}`);
 	}
 }
+async function prestartHttpRuntimeCommand(options, resolvedConfigPath) {
+	try {
+		writePrestartHttpResult(await prestartHttpRuntime({
+			id: options.id,
+			host: options.host ?? DEFAULT_HOST,
+			port: options.port,
+			config: options.config || resolvedConfigPath,
+			cache: options.cache,
+			definitionsCache: options.definitionsCache,
+			clearDefinitionsCache: options.clearDefinitionsCache,
+			proxyMode: options.proxyMode,
+			timeoutMs: options.timeoutMs
+		}));
+	} catch (error) {
+		console.error(`Failed to prestart HTTP runtime '${options.id || "generated-server-id"}': ${toErrorMessage$3(error)}`);
+		process.exit(1);
+	}
+}
 /**
 * MCP Serve command
 */
-const mcpServeCommand = new commander.Command("mcp-serve").description("Start MCP server with specified transport").option("-t, --type <type>", `Transport type: ${TRANSPORT_TYPE_STDIO}, ${TRANSPORT_TYPE_HTTP}, ${TRANSPORT_TYPE_SSE}, or ${TRANSPORT_TYPE_STDIO_HTTP}`).option("-p, --port <port>", "Port to listen on (http/sse) or backend port for stdio-http", (val) => Number.parseInt(val, 10)).option("--host <host>", "Host to bind to (http/sse) or backend host for stdio-http").option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--id <id>", "Unique server identifier (overrides config file id, auto-generated if not provided)").action(async (options) => {
+const mcpServeCommand = new commander.Command("mcp-serve").description("Start MCP server with specified transport or prestart the HTTP runtime in the background").option("-t, --type <type>", `Transport type: ${TRANSPORT_TYPE_STDIO}, ${TRANSPORT_TYPE_HTTP}, ${TRANSPORT_TYPE_SSE}, or ${TRANSPORT_TYPE_STDIO_HTTP}`).option("-p, --port <port>", "Port to listen on (http/sse) or backend port for stdio-http", (val) => Number.parseInt(val, 10)).option("--host <host>", "Host to bind to (http/sse) or backend host for stdio-http").option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--id <id>", "Unique server identifier (overrides config file id, auto-generated if not provided)").option("--prestart-http", "Prestart the HTTP runtime in the background and exit after it becomes healthy", false).option("--timeout-ms <ms>", "How long to wait for the HTTP runtime to become healthy", String(DEFAULT_HTTP_RUNTIME_TIMEOUT_MS)).action(async (options) => {
 	try {
+		if (options.prestartHttp) {
+			await prestartHttpRuntimeCommand(options, options.config || await findConfigFileAsync() || void 0);
+			return;
+		}
 		const resolvedConfigPath = options.config || await findConfigFileAsync() || void 0;
 		const proxyDefaults = resolvedConfigPath ? loadProxyDefaults(resolvedConfigPath) : {};
 		const transportType = validateTransportType((options.type ?? proxyDefaults.type ?? TRANSPORT_TYPE_STDIO).toLowerCase());
@@ -1703,7 +1725,6 @@ async function main() {
 		program.name("mcp-proxy").description("MCP proxy server package").version(require_src.version);
 		program.addCommand(initCommand);
 		program.addCommand(mcpServeCommand);
-		program.addCommand(prestartHttpCommand);
 		program.addCommand(searchToolsCommand);
 		program.addCommand(describeToolsCommand);
 		program.addCommand(useToolCommand);

package/dist/cli.mjs

@@ -1,22 +1,22 @@
 #!/usr/bin/env node
-import { C as DefinitionsCacheService, D as version, T as findConfigFile, b as RuntimeStateService, d as StdioHttpTransportHandler, f as StdioTransportHandler, m as HttpTransportHandler, n as createServer, o as createProxyIoCContainer, p as SseTransportHandler, r as createSessionServer, t as TRANSPORT_MODE, u as initializeSharedServices, v as StopServerService, w as generateServerId } from "./src-9L_ccN5I.mjs";
+import { C as DefinitionsCacheService, D as version, T as findConfigFile, b as RuntimeStateService, d as StdioHttpTransportHandler, f as StdioTransportHandler, m as HttpTransportHandler, n as createServer, o as createProxyIoCContainer, p as SseTransportHandler, r as createSessionServer, t as TRANSPORT_MODE, u as initializeSharedServices, v as StopServerService, w as generateServerId } from "./src-pCtALj-B.mjs";
 import { constants, existsSync, readFileSync } from "node:fs";
 import { access, writeFile } from "node:fs/promises";
 import yaml from "js-yaml";
 import { randomUUID } from "node:crypto";
 import path, { dirname, join, resolve } from "node:path";
+import { ProcessRegistryService, createProcessLease, resolveSiblingRegistryPath } from "@agimon-ai/foundation-process-registry";
 import { spawn } from "node:child_process";
 import { Liquid } from "liquidjs";
 import { Command } from "commander";
 import { fileURLToPath } from "node:url";
-import { ProcessRegistryService, createProcessLease, resolveSiblingRegistryPath } from "@agimon-ai/foundation-process-registry";
 import { DEFAULT_PORT_RANGE, PortRegistryService } from "@agimon-ai/foundation-port-registry";
-//#region src/commands/prestart-http.ts
+//#region src/commands/http-runtime.ts
 /**
-* Prestart HTTP Command
+* HTTP Runtime Helper
 *
 * Starts an mcp-proxy HTTP runtime in the background, waits until it is healthy,
-* and then exits so the runtime can be reused by other commands.
+* and returns the runtime details for callers that need to reuse it.
 */
 const WORKSPACE_MARKERS = [
 	"pnpm-workspace.yaml",
@@ -24,7 +24,7 @@ const WORKSPACE_MARKERS = [
 	".git"
 ];
 const DEFAULT_HOST$1 = "localhost";
-const DEFAULT_TIMEOUT_MS = 12e4;
+const DEFAULT_HTTP_RUNTIME_TIMEOUT_MS = 12e4;
 const POLL_INTERVAL_MS = 250;
 function resolveWorkspaceRoot(startPath = process.env.PROJECT_PATH || process.cwd()) {
 	let current = path.resolve(startPath);
@@ -80,7 +80,7 @@ function buildCliCandidates() {
 	throw new Error("Unable to locate mcp-proxy CLI entrypoint");
 }
 function parseTimeoutMs(value) {
-	if (!value) return DEFAULT_TIMEOUT_MS;
+	if (!value) return DEFAULT_HTTP_RUNTIME_TIMEOUT_MS;
 	const parsed = Number.parseInt(value, 10);
 	if (!Number.isInteger(parsed) || parsed <= 0) throw new Error(`Invalid timeout value: ${value}`);
 	return parsed;
@@ -206,17 +206,12 @@ async function prestartHttpRuntime(options) {
 		throw new Error(`Failed to prestart HTTP runtime '${serverId}': ${error instanceof Error ? error.message : String(error)}`, { cause: error });
 	}
 }
-const prestartHttpCommand = new Command("prestart-http").description("Start an mcp-proxy HTTP runtime in the background and wait until it is healthy").option("--id <id>", "Server identifier to assign to the runtime").option("--host <host>", "Host to bind to", DEFAULT_HOST$1).option("-p, --port <port>", "Preferred HTTP port for the runtime", (value) => Number.parseInt(value, 10)).option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--registry-path <path>", "Custom registry path or directory for service discovery").option("--registry-dir <path>", "Custom registry directory for service discovery").option("--timeout-ms <ms>", "How long to wait for the runtime to become healthy", String(DEFAULT_TIMEOUT_MS)).action(async (options) => {
-	try {
-		const { host, port, serverId, workspaceRoot } = await prestartHttpRuntime(options);
-		process.stdout.write(`mcp-proxy HTTP runtime ready at http://${host}:${port} (${serverId})\n`);
-		process.stdout.write(`runtimeId=${serverId}\n`);
-		process.stdout.write(`runtimeUrl=http://${host}:${port}\n`);
-		process.stdout.write(`workspaceRoot=${workspaceRoot}\n`);
-	} catch (error) {
-		throw new Error(`Failed to prestart HTTP runtime '${options.id || "generated-server-id"}': ${error instanceof Error ? error.message : String(error)}`, { cause: error });
-	}
-});
+function writePrestartHttpResult(result) {
+	process.stdout.write(`mcp-proxy HTTP runtime ready at http://${result.host}:${result.port} (${result.serverId})\n`);
+	process.stdout.write(`runtimeId=${result.serverId}\n`);
+	process.stdout.write(`runtimeUrl=http://${result.host}:${result.port}\n`);
+	process.stdout.write(`workspaceRoot=${result.workspaceRoot}\n`);
+}
 //#endregion
 //#region src/commands/bootstrap.ts
 function toErrorMessage$9(error) {
@@ -1154,9 +1149,14 @@ async function createAndStartHttpRuntime(serverOptions, config, resolvedConfigPa
 	console.error(`Runtime state: http://${runtimeRecord.host}:${runtimeRecord.port} (${runtimeRecord.serverId})`);
 }
 async function startStdioTransport(serverOptions) {
+	let server;
 	try {
-		await startServer(new StdioTransportHandler(await createServer(serverOptions), createStdioSafeLogger()));
+		server = await createServer(serverOptions);
+		await startServer(new StdioTransportHandler(server, createStdioSafeLogger()), async () => {
+			await server?.dispose?.();
+		});
 	} catch (error) {
+		await server?.dispose?.();
 		throw new Error(`Failed to start stdio transport: ${toErrorMessage$3(error)}`);
 	}
 }
@@ -1270,11 +1270,33 @@ async function startTransport(transportType, options, resolvedConfigPath, server
 		throw new Error(`Failed to start transport '${transportType}': ${toErrorMessage$3(error)}`);
 	}
 }
+async function prestartHttpRuntimeCommand(options, resolvedConfigPath) {
+	try {
+		writePrestartHttpResult(await prestartHttpRuntime({
+			id: options.id,
+			host: options.host ?? DEFAULT_HOST,
+			port: options.port,
+			config: options.config || resolvedConfigPath,
+			cache: options.cache,
+			definitionsCache: options.definitionsCache,
+			clearDefinitionsCache: options.clearDefinitionsCache,
+			proxyMode: options.proxyMode,
+			timeoutMs: options.timeoutMs
+		}));
+	} catch (error) {
+		console.error(`Failed to prestart HTTP runtime '${options.id || "generated-server-id"}': ${toErrorMessage$3(error)}`);
+		process.exit(1);
+	}
+}
 /**
 * MCP Serve command
 */
-const mcpServeCommand = new Command("mcp-serve").description("Start MCP server with specified transport").option("-t, --type <type>", `Transport type: ${TRANSPORT_TYPE_STDIO}, ${TRANSPORT_TYPE_HTTP}, ${TRANSPORT_TYPE_SSE}, or ${TRANSPORT_TYPE_STDIO_HTTP}`).option("-p, --port <port>", "Port to listen on (http/sse) or backend port for stdio-http", (val) => Number.parseInt(val, 10)).option("--host <host>", "Host to bind to (http/sse) or backend host for stdio-http").option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--id <id>", "Unique server identifier (overrides config file id, auto-generated if not provided)").action(async (options) => {
+const mcpServeCommand = new Command("mcp-serve").description("Start MCP server with specified transport or prestart the HTTP runtime in the background").option("-t, --type <type>", `Transport type: ${TRANSPORT_TYPE_STDIO}, ${TRANSPORT_TYPE_HTTP}, ${TRANSPORT_TYPE_SSE}, or ${TRANSPORT_TYPE_STDIO_HTTP}`).option("-p, --port <port>", "Port to listen on (http/sse) or backend port for stdio-http", (val) => Number.parseInt(val, 10)).option("--host <host>", "Host to bind to (http/sse) or backend host for stdio-http").option("-c, --config <path>", "Path to MCP server configuration file").option("--no-cache", "Disable configuration caching, always reload from config file").option("--definitions-cache <path>", "Path to prefetched tool/prompt/skill definitions cache file").option("--clear-definitions-cache", "Delete definitions cache before startup", false).option("--proxy-mode <mode>", "How mcp-proxy exposes downstream tools: meta, flat, or search", "meta").option("--id <id>", "Unique server identifier (overrides config file id, auto-generated if not provided)").option("--prestart-http", "Prestart the HTTP runtime in the background and exit after it becomes healthy", false).option("--timeout-ms <ms>", "How long to wait for the HTTP runtime to become healthy", String(DEFAULT_HTTP_RUNTIME_TIMEOUT_MS)).action(async (options) => {
 	try {
+		if (options.prestartHttp) {
+			await prestartHttpRuntimeCommand(options, options.config || await findConfigFileAsync() || void 0);
+			return;
+		}
 		const resolvedConfigPath = options.config || await findConfigFileAsync() || void 0;
 		const proxyDefaults = resolvedConfigPath ? loadProxyDefaults(resolvedConfigPath) : {};
 		const transportType = validateTransportType((options.type ?? proxyDefaults.type ?? TRANSPORT_TYPE_STDIO).toLowerCase());
@@ -1701,7 +1723,6 @@ async function main() {
 		program.name("mcp-proxy").description("MCP proxy server package").version(version);
 		program.addCommand(initCommand);
 		program.addCommand(mcpServeCommand);
-		program.addCommand(prestartHttpCommand);
 		program.addCommand(searchToolsCommand);
 		program.addCommand(describeToolsCommand);
 		program.addCommand(useToolCommand);

package/dist/index.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_src = require("./src-DIQjnEBk.cjs");
+const require_src = require("./src-DMlOlvPm.cjs");
 exports.ConfigFetcherService = require_src.ConfigFetcherService;
 exports.DefinitionsCacheService = require_src.DefinitionsCacheService;
 exports.DescribeToolsTool = require_src.DescribeToolsTool;

package/dist/index.d.cts

@@ -525,12 +525,12 @@ declare class McpClientManagerService {
   private serverConfigs;
   private connectionPromises;
   private logger;
-  constructor(logger?: LoggerLike);
+  private readonly repositoryPath;
+  constructor(logger?: LoggerLike, repositoryPath?: string);
   /**
    * Kill all stdio MCP server child processes.
-   * Sends SIGTERM first, then SIGKILL after 1s if the process hasn't exited.
-   * Must be called by the owner (e.g. transport/command layer) during shutdown.
-   * Awaiting the returned promise ensures force-kill timers complete before process.exit().
+   * Delegates to disconnectAll() so each client closes its child process and
+   * releases the corresponding process-registry lease before shutdown exits.
    */
   cleanupChildProcesses(): Promise<void>;
   /**

package/dist/index.d.mts

@@ -525,12 +525,12 @@ declare class McpClientManagerService {
   private serverConfigs;
   private connectionPromises;
   private logger;
-  constructor(logger?: LoggerLike);
+  private readonly repositoryPath;
+  constructor(logger?: LoggerLike, repositoryPath?: string);
   /**
    * Kill all stdio MCP server child processes.
-   * Sends SIGTERM first, then SIGKILL after 1s if the process hasn't exited.
-   * Must be called by the owner (e.g. transport/command layer) during shutdown.
-   * Awaiting the returned promise ensures force-kill timers complete before process.exit().
+   * Delegates to disconnectAll() so each client closes its child process and
+   * releases the corresponding process-registry lease before shutdown exits.
    */
   cleanupChildProcesses(): Promise<void>;
   /**

package/dist/index.mjs

@@ -1,2 +1,2 @@
-import { C as DefinitionsCacheService, E as ConfigFetcherService, S as createProxyLogger, T as findConfigFile, _ as DescribeToolsTool, a as createProxyContainer, b as RuntimeStateService, c as createStdioHttpTransportHandler, d as StdioHttpTransportHandler, f as StdioTransportHandler, g as SearchListToolsTool, h as UseToolTool, i as createHttpTransportHandler, l as createStdioTransportHandler, m as HttpTransportHandler, n as createServer, p as SseTransportHandler, r as createSessionServer, s as createSseTransportHandler, t as TRANSPORT_MODE, u as initializeSharedServices, v as StopServerService, w as generateServerId, x as McpClientManagerService, y as SkillService } from "./src-9L_ccN5I.mjs";
+import { C as DefinitionsCacheService, E as ConfigFetcherService, S as createProxyLogger, T as findConfigFile, _ as DescribeToolsTool, a as createProxyContainer, b as RuntimeStateService, c as createStdioHttpTransportHandler, d as StdioHttpTransportHandler, f as StdioTransportHandler, g as SearchListToolsTool, h as UseToolTool, i as createHttpTransportHandler, l as createStdioTransportHandler, m as HttpTransportHandler, n as createServer, p as SseTransportHandler, r as createSessionServer, s as createSseTransportHandler, t as TRANSPORT_MODE, u as initializeSharedServices, v as StopServerService, w as generateServerId, x as McpClientManagerService, y as SkillService } from "./src-pCtALj-B.mjs";
 export { ConfigFetcherService, DefinitionsCacheService, DescribeToolsTool, HttpTransportHandler, McpClientManagerService, RuntimeStateService, SearchListToolsTool, SkillService, SseTransportHandler, StdioHttpTransportHandler, StdioTransportHandler, StopServerService, TRANSPORT_MODE, UseToolTool, createHttpTransportHandler, createProxyContainer, createProxyLogger, createServer, createSessionServer, createSseTransportHandler, createStdioHttpTransportHandler, createStdioTransportHandler, findConfigFile, generateServerId, initializeSharedServices };

package/dist/{src-DIQjnEBk.cjs → src-DMlOlvPm.cjs}

@@ -33,6 +33,7 @@ let _modelcontextprotocol_sdk_client_index_js = require("@modelcontextprotocol/s
 let _modelcontextprotocol_sdk_client_sse_js = require("@modelcontextprotocol/sdk/client/sse.js");
 let _modelcontextprotocol_sdk_client_stdio_js = require("@modelcontextprotocol/sdk/client/stdio.js");
 let _modelcontextprotocol_sdk_client_streamableHttp_js = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
+let _agimon_ai_foundation_process_registry = require("@agimon-ai/foundation-process-registry");
 let node_child_process = require("node:child_process");
 let liquidjs = require("liquidjs");
 let _agimon_ai_foundation_validator = require("@agimon-ai/foundation-validator");
@@ -47,7 +48,7 @@ let _modelcontextprotocol_sdk_server_sse_js = require("@modelcontextprotocol/sdk
 let _modelcontextprotocol_sdk_server_stdio_js = require("@modelcontextprotocol/sdk/server/stdio.js");
 let _modelcontextprotocol_sdk_server_index_js = require("@modelcontextprotocol/sdk/server/index.js");
 //#region package.json
-var version = "0.9.1";
+var version = "0.10.1";
 //#endregion
 //#region src/utils/mcpConfigSchema.ts
 /**
@@ -1538,6 +1539,7 @@ async function createProxyLogger(options = {}) {
 //#region src/services/McpClientManagerService.ts
 /** Default connection timeout in milliseconds (30 seconds) */
 const DEFAULT_CONNECTION_TIMEOUT_MS = 3e4;
+const PROCESS_REGISTRY_SERVICE_TYPE = "tool";
 /**
 * Checks if an error is a session-related error from an HTTP backend
 * (e.g., downstream server restarted and no longer recognizes the session ID).
@@ -1592,6 +1594,7 @@ var McpClient = class {
 	logger;
 	client;
 	childProcess;
+	processLease;
 	connected = false;
 	reconnectFn;
 	reconnectPromise;
@@ -1608,6 +1611,9 @@ var McpClient = class {
 	setChildProcess(process) {
 		this.childProcess = process;
 	}
+	setProcessLease(processLease) {
+		this.processLease = processLease;
+	}
 	setConnected(connected) {
 		this.connected = connected;
 	}
@@ -1706,9 +1712,20 @@ var McpClient = class {
 		});
 	}
 	async close() {
-		if (this.childProcess) this.childProcess.kill();
-		await this.client.close();
+		const childProcess = this.childProcess;
+		const processLease = this.processLease;
+		this.childProcess = void 0;
+		this.processLease = void 0;
 		this.connected = false;
+		try {
+			if (childProcess) childProcess.kill();
+			await this.client.close();
+		} finally {
+			await processLease?.release({
+				kill: false,
+				releasePort: false
+			});
+		}
 	}
 };
 /**
@@ -1719,38 +1736,18 @@ var McpClientManagerService = class {
 	serverConfigs = /* @__PURE__ */ new Map();
 	connectionPromises = /* @__PURE__ */ new Map();
 	logger;
-	constructor(logger = console) {
+	repositoryPath;
+	constructor(logger = console, repositoryPath = process.cwd()) {
 		this.logger = logger;
+		this.repositoryPath = repositoryPath;
 	}
 	/**
 	* Kill all stdio MCP server child processes.
-	* Sends SIGTERM first, then SIGKILL after 1s if the process hasn't exited.
-	* Must be called by the owner (e.g. transport/command layer) during shutdown.
-	* Awaiting the returned promise ensures force-kill timers complete before process.exit().
+	* Delegates to disconnectAll() so each client closes its child process and
+	* releases the corresponding process-registry lease before shutdown exits.
 	*/
 	async cleanupChildProcesses() {
-		const killPromises = [];
-		for (const [serverName, client] of this.clients) try {
-			const childProcess = client["childProcess"];
-			if (childProcess && !childProcess.killed) {
-				this.logger.info(`Killing stdio MCP server: ${serverName} (PID: ${childProcess.pid})`);
-				childProcess.kill("SIGTERM");
-				killPromises.push(new Promise((resolve) => {
-					setTimeout(() => {
-						try {
-							if (!childProcess.killed) {
-								this.logger.warn(`Force killing stdio MCP server: ${serverName} (PID: ${childProcess.pid})`);
-								childProcess.kill("SIGKILL");
-							}
-						} catch {}
-						resolve();
-					}, 1e3);
-				}));
-			}
-		} catch (error) {
-			this.logger.warn(`Failed to kill child process for ${serverName}`, error);
-		}
-		await Promise.all(killPromises);
+		await this.disconnectAll();
 	}
 	/**
 	* Connect to an MCP server based on its configuration with timeout
@@ -1846,7 +1843,22 @@ var McpClientManagerService = class {
 		});
 		await mcpClient["client"].connect(transport);
 		const childProcess = transport["_process"];
-		if (childProcess) mcpClient.setChildProcess(childProcess);
+		if (childProcess) {
+			if (typeof childProcess.pid !== "number" || childProcess.pid <= 0) throw new Error(`Failed to capture pid for stdio child process on server "${mcpClient.serverName}"`);
+			mcpClient.setChildProcess(childProcess);
+			const processLease = await (0, _agimon_ai_foundation_process_registry.createProcessLease)({
+				repositoryPath: this.repositoryPath,
+				serviceName: mcpClient.serverName,
+				serviceType: PROCESS_REGISTRY_SERVICE_TYPE,
+				pid: childProcess.pid,
+				command: config.command,
+				args: config.args,
+				metadata: { transport: "stdio" }
+			});
+			mcpClient.setProcessLease(processLease);
+			return;
+		}
+		throw new Error(`Failed to capture stdio child process for server "${mcpClient.serverName}"`);
 	}
 	async connectHttpClient(mcpClient, config) {
 		const transport = new _modelcontextprotocol_sdk_client_streamableHttp_js.StreamableHTTPClientTransport(new URL(config.url), { requestInit: config.headers ? { headers: config.headers } : void 0 });
@@ -4938,7 +4950,10 @@ async function createSessionServer(shared) {
 * For multi-session HTTP transport, use createProxyContainer() + createSessionServer() instead.
 */
 async function createServer(options) {
-	return createSessionServer(await createProxyContainer(options));
+	const shared = await createProxyContainer(options);
+	const server = await createSessionServer(shared);
+	Object.assign(server, { dispose: shared.dispose });
+	return server;
 }
 //#endregion
 //#region src/types/index.ts

package/dist/{src-9L_ccN5I.mjs → src-pCtALj-B.mjs}

@@ -10,6 +10,7 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
 import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { createProcessLease } from "@agimon-ai/foundation-process-registry";
 import { spawn } from "node:child_process";
 import { Liquid } from "liquidjs";
 import { coerceArgs, formatZodError, jsonSchemaToZod } from "@agimon-ai/foundation-validator";
@@ -24,7 +25,7 @@ import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 //#region package.json
-var version = "0.9.1";
+var version = "0.10.1";
 //#endregion
 //#region src/utils/mcpConfigSchema.ts
 /**
@@ -1515,6 +1516,7 @@ async function createProxyLogger(options = {}) {
 //#region src/services/McpClientManagerService.ts
 /** Default connection timeout in milliseconds (30 seconds) */
 const DEFAULT_CONNECTION_TIMEOUT_MS = 3e4;
+const PROCESS_REGISTRY_SERVICE_TYPE = "tool";
 /**
 * Checks if an error is a session-related error from an HTTP backend
 * (e.g., downstream server restarted and no longer recognizes the session ID).
@@ -1569,6 +1571,7 @@ var McpClient = class {
 	logger;
 	client;
 	childProcess;
+	processLease;
 	connected = false;
 	reconnectFn;
 	reconnectPromise;
@@ -1585,6 +1588,9 @@ var McpClient = class {
 	setChildProcess(process) {
 		this.childProcess = process;
 	}
+	setProcessLease(processLease) {
+		this.processLease = processLease;
+	}
 	setConnected(connected) {
 		this.connected = connected;
 	}
@@ -1683,9 +1689,20 @@ var McpClient = class {
 		});
 	}
 	async close() {
-		if (this.childProcess) this.childProcess.kill();
-		await this.client.close();
+		const childProcess = this.childProcess;
+		const processLease = this.processLease;
+		this.childProcess = void 0;
+		this.processLease = void 0;
 		this.connected = false;
+		try {
+			if (childProcess) childProcess.kill();
+			await this.client.close();
+		} finally {
+			await processLease?.release({
+				kill: false,
+				releasePort: false
+			});
+		}
 	}
 };
 /**
@@ -1696,38 +1713,18 @@ var McpClientManagerService = class {
 	serverConfigs = /* @__PURE__ */ new Map();
 	connectionPromises = /* @__PURE__ */ new Map();
 	logger;
-	constructor(logger = console) {
+	repositoryPath;
+	constructor(logger = console, repositoryPath = process.cwd()) {
 		this.logger = logger;
+		this.repositoryPath = repositoryPath;
 	}
 	/**
 	* Kill all stdio MCP server child processes.
-	* Sends SIGTERM first, then SIGKILL after 1s if the process hasn't exited.
-	* Must be called by the owner (e.g. transport/command layer) during shutdown.
-	* Awaiting the returned promise ensures force-kill timers complete before process.exit().
+	* Delegates to disconnectAll() so each client closes its child process and
+	* releases the corresponding process-registry lease before shutdown exits.
 	*/
 	async cleanupChildProcesses() {
-		const killPromises = [];
-		for (const [serverName, client] of this.clients) try {
-			const childProcess = client["childProcess"];
-			if (childProcess && !childProcess.killed) {
-				this.logger.info(`Killing stdio MCP server: ${serverName} (PID: ${childProcess.pid})`);
-				childProcess.kill("SIGTERM");
-				killPromises.push(new Promise((resolve) => {
-					setTimeout(() => {
-						try {
-							if (!childProcess.killed) {
-								this.logger.warn(`Force killing stdio MCP server: ${serverName} (PID: ${childProcess.pid})`);
-								childProcess.kill("SIGKILL");
-							}
-						} catch {}
-						resolve();
-					}, 1e3);
-				}));
-			}
-		} catch (error) {
-			this.logger.warn(`Failed to kill child process for ${serverName}`, error);
-		}
-		await Promise.all(killPromises);
+		await this.disconnectAll();
 	}
 	/**
 	* Connect to an MCP server based on its configuration with timeout
@@ -1823,7 +1820,22 @@ var McpClientManagerService = class {
 		});
 		await mcpClient["client"].connect(transport);
 		const childProcess = transport["_process"];
-		if (childProcess) mcpClient.setChildProcess(childProcess);
+		if (childProcess) {
+			if (typeof childProcess.pid !== "number" || childProcess.pid <= 0) throw new Error(`Failed to capture pid for stdio child process on server "${mcpClient.serverName}"`);
+			mcpClient.setChildProcess(childProcess);
+			const processLease = await createProcessLease({
+				repositoryPath: this.repositoryPath,
+				serviceName: mcpClient.serverName,
+				serviceType: PROCESS_REGISTRY_SERVICE_TYPE,
+				pid: childProcess.pid,
+				command: config.command,
+				args: config.args,
+				metadata: { transport: "stdio" }
+			});
+			mcpClient.setProcessLease(processLease);
+			return;
+		}
+		throw new Error(`Failed to capture stdio child process for server "${mcpClient.serverName}"`);
 	}
 	async connectHttpClient(mcpClient, config) {
 		const transport = new StreamableHTTPClientTransport(new URL(config.url), { requestInit: config.headers ? { headers: config.headers } : void 0 });
@@ -4915,7 +4927,10 @@ async function createSessionServer(shared) {
 * For multi-session HTTP transport, use createProxyContainer() + createSessionServer() instead.
 */
 async function createServer$1(options) {
-	return createSessionServer(await createProxyContainer(options));
+	const shared = await createProxyContainer(options);
+	const server = await createSessionServer(shared);
+	Object.assign(server, { dispose: shared.dispose });
+	return server;
 }
 //#endregion
 //#region src/types/index.ts

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@agimon-ai/mcp-proxy",
   "description": "MCP proxy server package",
-  "version": "0.10.0",
+  "version": "0.10.2",
   "license": "AGPL-3.0",
   "keywords": [
     "mcp",