@tmcp/transport-http 0.2.1 → 0.4.0

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @tmcp/transport-http
 
+## 0.4.0
+
+### Minor Changes
+
+- 8f09cd2: fix: send requests back on the same stream if available
+
+## 0.3.0
+
+### Minor Changes
+
+- 3847654: fix: answer with conflict for second SSE connection
+
+### Patch Changes
+
+- Updated dependencies [1b50780]
+    - tmcp@1.7.1
+
 ## 0.2.1
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmcp/transport-http",
-  "version": "0.2.1",
+  "version": "0.4.0",
   "description": "Transport for TMCP using HTTP",
   "type": "module",
   "main": "src/index.js",
@@ -13,7 +13,7 @@
     "./package.json": "./package.json"
   },
   "peerDependencies": {
-    "tmcp": "^1.7.0"
+    "tmcp": "^1.7.1"
   },
   "keywords": [
     "tmcp",
@@ -26,8 +26,9 @@
     "directory": "packages/transport-http"
   },
   "devDependencies": {
+    "@types/node": "^24.0.15",
     "publint": "^0.3.12",
-    "tmcp": "^1.7.0"
+    "tmcp": "^1.7.1"
   },
   "publishConfig": {
     "access": "public"

package/src/index.d.ts

@@ -1,12 +1,3 @@
-/**
- * @import { McpServer, ClientCapabilities } from "tmcp";
- */
-/**
- * @typedef {{
- * 	getSessionId?: () => string
- * 	path?: string
- * }} HttpTransportOptions
- */
 export class HttpTransport {
     /**
      *

package/src/index.js

@@ -9,6 +9,8 @@
  * }} HttpTransportOptions
  */
 
+import { AsyncLocalStorage } from 'node:async_hooks';
+
 export class HttpTransport {
 	/**
 	 * @type {McpServer<any>}
@@ -34,6 +36,11 @@ export class HttpTransport {
 	 */
 	#streams = new Map();
 
+	/**
+	 * @type {AsyncLocalStorage<ReadableStreamDefaultController | undefined>}
+	 */
+	#controller_storage = new AsyncLocalStorage();
+
 	/**
 	 *
 	 * @param {McpServer<any>} server
@@ -48,6 +55,15 @@ export class HttpTransport {
 		this.#options = { getSessionId, path };
 		this.#path = path;
 		this.#server.on('send', ({ request, context: { sessions } }) => {
+			// use the current controller if the request has an id (it means it's a request and not a notification)
+			if (request.id != null) {
+				const controller = this.#controller_storage.getStore();
+				if (!controller)
+					throw new Error('Controller not found in storage');
+
+				controller.enqueue('data: ' + JSON.stringify(request) + '\n\n');
+				return;
+			}
 			for (let [session_id, controller] of this.#session.entries()) {
 				if (sessions === undefined || sessions.includes(session_id)) {
 					controller.enqueue(
@@ -87,15 +103,24 @@ export class HttpTransport {
 		// If session already exists, return existing stream
 		const existing_stream = this.#streams.get(session_id);
 		if (existing_stream) {
-			return new Response(existing_stream, {
-				headers: {
-					'Content-Type': 'text/event-stream',
-					'Cache-Control': 'no-cache',
-					Connection: 'keep-alive',
-					'mcp-session-id': session_id,
+			return new Response(
+				JSON.stringify({
+					jsonrpc: '2.0',
+					error: {
+						code: -32000,
+						message:
+							'Conflict: Only one SSE stream is allowed per session',
+					},
+					id: null,
+				}),
+				{
+					headers: {
+						'Content-Type': 'application/json',
+						'mcp-session-id': session_id,
+					},
+					status: 409,
 				},
-				status: 200,
-			});
+			);
 		}
 
 		// Create new long-lived stream for notifications
@@ -151,22 +176,31 @@ export class HttpTransport {
 
 		try {
 			const body = await request.clone().json();
-			const response = await this.#server.receive(body, session_id);
 
-			// Determine status code based on response type
-			// 202 Accepted for notifications/responses, 200 OK for standard requests
-			const status = response == null || response.id == null ? 202 : 200;
+			/**
+			 * @type {ReadableStreamDefaultController | undefined}
+			 */
+			let controller;
 
 			// Create a short-lived stream that closes after sending the response
 			const stream = new ReadableStream({
-				start(controller) {
-					controller.enqueue(
-						'data: ' + JSON.stringify(response) + '\n\n',
-					);
-					controller.close();
+				start(_controller) {
+					controller = _controller;
 				},
 			});
 
+			const response = await this.#controller_storage.run(
+				controller,
+				() => this.#server.receive(body, session_id),
+			);
+
+			controller?.enqueue('data: ' + JSON.stringify(response) + '\n\n');
+			controller?.close();
+
+			// Determine status code based on response type
+			// 202 Accepted for notifications/responses, 200 OK for standard requests
+			const status = response == null || response.id == null ? 202 : 200;
+
 			return new Response(stream, {
 				headers: {
 					'Content-Type': 'text/event-stream',