@agimon-ai/mcp-proxy 0.10.0 → 0.10.1

package/dist/cli.cjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-const require_src = require("./src-DIQjnEBk.cjs");
+const require_src = require("./src-CRhtaDaz.cjs");
 let node_fs = require("node:fs");
 let node_fs_promises = require("node:fs/promises");
 let js_yaml = require("js-yaml");
@@ -7,11 +7,11 @@ 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
 /**
@@ -1156,9 +1156,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)}`);
 	}
 }

package/dist/cli.mjs

@@ -1,15 +1,15 @@
 #!/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-C48MCBti.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
 /**
@@ -1154,9 +1154,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)}`);
 	}
 }

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-CRhtaDaz.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-C48MCBti.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-9L_ccN5I.mjs → src-C48MCBti.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.0";
 //#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/dist/{src-DIQjnEBk.cjs → src-CRhtaDaz.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.0";
 //#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/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@agimon-ai/mcp-proxy",
   "description": "MCP proxy server package",
-  "version": "0.10.0",
+  "version": "0.10.1",
   "license": "AGPL-3.0",
   "keywords": [
     "mcp",
@@ -30,8 +30,8 @@
     "zod": "4.3.6",
     "@agimon-ai/foundation-process-registry": "0.8.0",
     "@agimon-ai/log-sink-mcp": "0.8.0",
-    "@agimon-ai/foundation-validator": "0.5.0",
-    "@agimon-ai/foundation-port-registry": "0.8.0"
+    "@agimon-ai/foundation-port-registry": "0.8.0",
+    "@agimon-ai/foundation-validator": "0.5.0"
   },
   "devDependencies": {
     "@types/js-yaml": "4.0.9",