@orkestrel/mcp 0.0.13 → 0.0.15

package/dist/src/server/index.cjs

@@ -1602,13 +1602,21 @@ function createHTTPClientTransport(options) {
 *   as a frame — a NOTIFICATION sends nothing, and a non-request message (a stray response) is
 *   ignored. A `dispatch` / `send` fault surfaces on `mcp.emitter`'s `error` event rather than
 *   escaping the (async) message pump.
+* - **Closes on the spine's `stop`.** It holds every socket it claimed and, on `options.emitter`'s
+*   `stop` event, closes each one with the RFC 6455 close handshake, so the spine's drain settles
+*   at once and each client reads a clean goodbye. Node detaches an upgraded socket from the
+*   connection set the spine's own close walks, so the claimant is the only thing that can end
+*   it: an ingress that held its sockets open would cost `stop()` the whole `drain` budget and
+*   then have the connection cut mid-protocol. A socket the peer already dropped is gone from
+*   the set (its transport's `close` removes it), and closing a dead one is a no-op either way.
 *
 * It is MECHANISM, not policy: compose an auth guard IN FRONT by registering an upgrade
 * handler BEFORE this one — that handler can claim (decline + destroy) an unauthenticated
 * upgrade so it never reaches this pump.
 *
 * @param mcp - The transport-agnostic {@link MCPDispatcherInterface} to expose over WebSocket
-* @param options - Optional `path` (default {@link DEFAULT_MCP_PATH}) and `subprotocol`
+* @param options - The spine's `emitter` (REQUIRED — the `stop` event this ingress closes its
+*   sockets on), plus optional `path` (default {@link DEFAULT_MCP_PATH}) and `subprotocol`
 *   (default {@link MCP_WEBSOCKET_SUBPROTOCOL}); see {@link WebSocketServerOptions}
 * @returns An {@link UpgradeHandler} to register with the spine's `upgrade` seam
 *
@@ -1618,12 +1626,16 @@ function createHTTPClientTransport(options) {
 * import { createWebSocketServer } from '@src/server'
 *
 * const mcp = createMCPServer({ identity: { name: 'docs', version: '1.0.0' }, tools: createToolManager() })
-* server.upgrade(createWebSocketServer(mcp)) // an MCP client now connects over ws://…/mcp
+* server.upgrade(createWebSocketServer(mcp, { emitter: server.emitter })) // ws://…/mcp
 * ```
 */
 function createWebSocketServer(mcp, options) {
-	const path = options?.path ?? "/mcp";
-	const subprotocol = options?.subprotocol ?? "mcp";
+	const path = options.path ?? "/mcp";
+	const subprotocol = options.subprotocol ?? "mcp";
+	const live = /* @__PURE__ */ new Set();
+	options.emitter.on("stop", () => {
+		for (const transport of live) transport.close();
+	});
 	return (request, socket, head) => {
 		const upgrade = request.headers["upgrade"];
 		if (!(0, _orkestrel_contract.isString)(upgrade) || upgrade.toLowerCase() !== "websocket") return false;
@@ -1638,6 +1650,8 @@ function createWebSocketServer(mcp, options) {
 			head,
 			protocol: subprotocol
 		}));
+		live.add(transport);
+		transport.emitter.on("close", () => live.delete(transport));
 		(0, _src_core.bindServer)(mcp, bridgeMessageTransport(transport));
 		transport.start();
 		return true;