@tmcp/transport-http 0.3.0 → 0.4.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @tmcp/transport-http
 
+## 0.4.0
+
+### Minor Changes
+
+- 8f09cd2: fix: send requests back on the same stream if available
+
 ## 0.3.0
 
 ### Minor Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmcp/transport-http",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Transport for TMCP using HTTP",
   "type": "module",
   "main": "src/index.js",
@@ -26,6 +26,7 @@
     "directory": "packages/transport-http"
   },
   "devDependencies": {
+    "@types/node": "^24.0.15",
     "publint": "^0.3.12",
     "tmcp": "^1.7.1"
   },

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(
@@ -160,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',