@nestjs-mcp/server 0.1.1 → 0.2.0-alpha.1

package/README.md

@@ -8,6 +8,7 @@
 [![Known Vulnerabilities](https://snyk.io/test/github/adrian-d-hidalgo/nestjs-mcp-server/badge.svg)](https://snyk.io/test/github/adrian-d-hidalgo/nestjs-mcp-server)
 [![MIT License](https://img.shields.io/badge/license-MIT-green.svg)](./LICENSE)
 [![PRs Welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg)](./CONTRIBUTING.md)
+[![Contributor Covenant](https://img.shields.io/badge/Contributor%20Covenant-2.1-4baaaa.svg)](CODE_OF_CONDUCT.md)
 
 ---
 
@@ -40,6 +41,7 @@
   - [Prompt Decorator](#prompt-decorator)
   - [Resource Decorator](#resource-decorator)
   - [Tool Decorator](#tool-decorator)
+  - [RequestHandlerExtra Parameter](#requesthandlerextra-argument)
 - [Guards](#guards)
   - [Global-level guards](#global-level-guards)
   - [Resolver-level guards](#resolver-level-guards)
@@ -496,6 +498,68 @@ export class MyTools {
 }
 ```
 
+### RequestHandlerExtra Argument
+
+All MCP capability methods (`@Prompt`, `@Resource`, `@Tool`) always receive a `RequestHandlerExtra` object as their last parameter. This object extends the original type from `@modelcontextprotocol/sdk` and provides essential context about the current MCP request.
+
+**Properties from SDK:**
+
+- `signal`: An `AbortSignal` used to communicate if the request was cancelled
+- `authInfo`: Optional information about a validated access token
+- `sessionId`: The session ID from the transport, if available
+- `sendNotification`: Function to send a notification related to the current request
+- `sendRequest`: Function to send a request related to the current request
+
+**Extended Properties:**
+
+- `headers`: HTTP headers from the original request (added by @nestjs-mcp/server)
+
+**Usage Example:**
+
+```ts
+import { Tool, Resolver, SessionManager } from '@nestjs-mcp/server';
+import { RequestHandlerExtra } from '@nestjs-mcp/server';
+import { CallToolResult } from '@modelcontextprotocol/sdk/types';
+
+@Resolver('auth')
+export class AuthResolver {
+  @Tool({
+    name: 'authenticate_user',
+    description: 'Authenticates a user with credentials',
+    // ...other options
+  })
+  authenticateUser(
+    params: { username: string; password: string },
+    extra: RequestHandlerExtra, // Always the last parameter
+  ): CallToolResult {
+    // Access the session ID
+    console.log(`Request received in session: ${extra.sessionId}`);
+
+    // Access request headers (extended property)
+    const authHeader = extra.headers.authorization;
+    const userAgent = extra.headers['user-agent'];
+    console.log(`Request from: ${userAgent}`);
+
+    // Check if request was cancelled
+    if (extra.signal.aborted) {
+      return {
+        content: [{ type: 'text', text: 'Request was cancelled' }],
+      };
+    }
+
+    // Implement authentication logic
+    return {
+      content: [{ type: 'text', text: 'Authentication successful' }],
+    };
+  }
+}
+```
+
+**Important Notes:**
+
+- `extra` is always the last parameter in any method decorated with `@Resource`, `@Prompt`, or `@Tool`
+- The `headers` property is an extension added by @nestjs-mcp/server to access HTTP headers directly
+
 ---
 
 ## Guards
@@ -856,3 +920,5 @@ MIT — see [LICENSE](./LICENSE) for details.
 ## Contributions
 
 Contributions are welcome! Please see [CONTRIBUTING.md](./CONTRIBUTING.md) for guidelines, reporting issues, and pull request rules.
+
+Before contributing, please read our [Code of Conduct](./CODE_OF_CONDUCT.md) to understand the expectations for behavior in our community.

package/dist/controllers/sse/sse.controller.d.ts

@@ -1,8 +1,10 @@
 import { Request, Response } from 'express';
+import { McpLoggerService } from '../../registry/logger.service';
 import { SseService } from './sse.service';
 export declare class SseController {
+    private readonly logger;
     private readonly service;
-    constructor(service: SseService);
+    constructor(logger: McpLoggerService, service: SseService);
     handleSse(req: Request, res: Response): Promise<void>;
     handleMessages(req: Request, res: Response): Promise<void>;
 }

package/dist/controllers/sse/sse.controller.js

@@ -14,16 +14,21 @@ var __param = (this && this.__param) || function (paramIndex, decorator) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SseController = void 0;
 const common_1 = require("@nestjs/common");
+const logger_service_1 = require("../../registry/logger.service");
 const sse_service_1 = require("./sse.service");
 let SseController = class SseController {
+    logger;
     service;
-    constructor(service) {
+    constructor(logger, service) {
+        this.logger = logger;
         this.service = service;
     }
     async handleSse(req, res) {
+        this.logger.log('[SSE] Connection established');
         await this.service.handleSse(req, res);
     }
     async handleMessages(req, res) {
+        this.logger.log('[SSE] Message received');
         await this.service.handleMessage(req, res);
     }
 };
@@ -46,6 +51,7 @@ __decorate([
 ], SseController.prototype, "handleMessages", null);
 exports.SseController = SseController = __decorate([
     (0, common_1.Controller)(),
-    __metadata("design:paramtypes", [sse_service_1.SseService])
+    __metadata("design:paramtypes", [logger_service_1.McpLoggerService,
+        sse_service_1.SseService])
 ], SseController);
 //# sourceMappingURL=sse.controller.js.map

package/dist/controllers/sse/sse.controller.js.map

@@ -1 +1 @@
-{"version":3,"file":"sse.controller.js","sourceRoot":"","sources":["../../../src/controllers/sse/sse.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAiE;AAGjE,+CAA2C;AAGpC,IAAM,aAAa,GAAnB,MAAM,aAAa;IACK;IAA7B,YAA6B,OAAmB;QAAnB,YAAO,GAAP,OAAO,CAAY;IAAG,CAAC;IAG9C,AAAN,KAAK,CAAC,SAAS,CAAQ,GAAY,EAAS,GAAa;QACvD,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACzC,CAAC;IAGK,AAAN,KAAK,CAAC,cAAc,CAAQ,GAAY,EAAS,GAAa;QAC5D,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC7C,CAAC;CACF,CAAA;AAZY,sCAAa;AAIlB;IADL,IAAA,YAAG,EAAC,KAAK,CAAC;IACM,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;8CAE1C;AAGK;IADL,IAAA,aAAI,EAAC,UAAU,CAAC;IACK,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;mDAE/C;wBAXU,aAAa;IADzB,IAAA,mBAAU,GAAE;qCAE2B,wBAAU;GADrC,aAAa,CAYzB"}
+{"version":3,"file":"sse.controller.js","sourceRoot":"","sources":["../../../src/controllers/sse/sse.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAiE;AAGjE,kEAAiE;AACjE,+CAA2C;AAGpC,IAAM,aAAa,GAAnB,MAAM,aAAa;IAEL;IACA;IAFnB,YACmB,MAAwB,EACxB,OAAmB;QADnB,WAAM,GAAN,MAAM,CAAkB;QACxB,YAAO,GAAP,OAAO,CAAY;IACnC,CAAC;IAGE,AAAN,KAAK,CAAC,SAAS,CAAQ,GAAY,EAAS,GAAa;QACvD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;QAChD,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACzC,CAAC;IAGK,AAAN,KAAK,CAAC,cAAc,CAAQ,GAAY,EAAS,GAAa;QAC5D,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC7C,CAAC;CACF,CAAA;AAjBY,sCAAa;AAOlB;IADL,IAAA,YAAG,EAAC,KAAK,CAAC;IACM,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;8CAG1C;AAGK;IADL,IAAA,aAAI,EAAC,UAAU,CAAC;IACK,WAAA,IAAA,YAAG,GAAE,CAAA;IAAgB,WAAA,IAAA,YAAG,GAAE,CAAA;;;;mDAG/C;wBAhBU,aAAa;IADzB,IAAA,mBAAU,GAAE;qCAGgB,iCAAgB;QACf,wBAAU;GAH3B,aAAa,CAiBzB"}

package/dist/controllers/sse/sse.service.d.ts

@@ -3,13 +3,14 @@ import { Request, Response } from 'express';
 import { McpServerOptions } from '../../interfaces/mcp-server-options.interface';
 import { McpLoggerService } from '../../registry/logger.service';
 import { RegistryService } from '../../registry/registry.service';
+import { SessionManager } from '../../services/session.manager';
 export declare class SseService implements OnModuleInit {
     private readonly options;
     private readonly registry;
     private readonly logger;
+    private readonly sessionManager;
     private server;
-    private transports;
-    constructor(options: McpServerOptions, registry: RegistryService, logger: McpLoggerService);
+    constructor(options: McpServerOptions, registry: RegistryService, logger: McpLoggerService, sessionManager: SessionManager);
     onModuleInit(): Promise<void>;
     handleSse(req: Request, res: Response): Promise<void>;
     handleMessage(req: Request, res: Response): Promise<void>;

package/dist/controllers/sse/sse.service.js

@@ -18,16 +18,18 @@ const sse_js_1 = require("@modelcontextprotocol/sdk/server/sse.js");
 const common_1 = require("@nestjs/common");
 const logger_service_1 = require("../../registry/logger.service");
 const registry_service_1 = require("../../registry/registry.service");
+const session_manager_1 = require("../../services/session.manager");
 let SseService = class SseService {
     options;
     registry;
     logger;
+    sessionManager;
     server;
-    transports = {};
-    constructor(options, registry, logger) {
+    constructor(options, registry, logger, sessionManager) {
         this.options = options;
         this.registry = registry;
         this.logger = logger;
+        this.sessionManager = sessionManager;
         this.server = new mcp_js_1.McpServer(this.options.serverInfo, this.options.options);
     }
     async onModuleInit() {
@@ -37,18 +39,30 @@ let SseService = class SseService {
     }
     async handleSse(req, res) {
         const transport = new sse_js_1.SSEServerTransport('/messages', res);
-        this.transports[transport.sessionId] = transport;
+        this.sessionManager.setSession(transport.sessionId, {
+            transport,
+            request: req,
+        });
         this.logger.debug(`Starting SSE for sessionId: ${transport.sessionId}`, 'api');
         res.on('close', () => {
-            delete this.transports[transport.sessionId];
+            this.sessionManager.deleteSession(transport.sessionId);
         });
         await this.server.connect(transport);
     }
     async handleMessage(req, res) {
         const sessionId = req.query.sessionId;
-        const transport = this.transports[sessionId];
+        const session = this.sessionManager.getSession(sessionId);
+        if (!session) {
+            res.status(400).send('Invalid or missing sessionId');
+            return;
+        }
         this.logger.debug(`Receiving SSE message for sessionId: ${sessionId}`, 'api');
         this.logger.debug(`SSE message: ${JSON.stringify(req.body)}`, 'MCP_SERVER');
+        const transport = session.transport;
+        if (!(transport instanceof sse_js_1.SSEServerTransport)) {
+            res.status(400).send('Invalid transport');
+            return;
+        }
         try {
             if (transport) {
                 await transport.handlePostMessage(req, res, req.body);
@@ -73,6 +87,7 @@ exports.SseService = SseService = __decorate([
     (0, common_1.Injectable)(),
     __param(0, (0, common_1.Inject)('MCP_SERVER_OPTIONS')),
     __metadata("design:paramtypes", [Object, registry_service_1.RegistryService,
-        logger_service_1.McpLoggerService])
+        logger_service_1.McpLoggerService,
+        session_manager_1.SessionManager])
 ], SseService);
 //# sourceMappingURL=sse.service.js.map

package/dist/controllers/sse/sse.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"sse.service.js","sourceRoot":"","sources":["../../../src/controllers/sse/sse.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oEAAoE;AACpE,oEAA6E;AAC7E,2CAAkE;AAIlE,kEAAiE;AACjE,sEAAkE;AAG3D,IAAM,UAAU,GAAhB,MAAM,UAAU;IAOF;IACA;IACA;IARX,MAAM,CAAY;IAElB,UAAU,GAAG,EAAwC,CAAC;IAE9D,YAEmB,OAAyB,EACzB,QAAyB,EACzB,MAAwB;QAFxB,YAAO,GAAP,OAAO,CAAkB;QACzB,aAAQ,GAAR,QAAQ,CAAiB;QACzB,WAAM,GAAN,MAAM,CAAkB;QAEzC,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,sCAAsC,EAAE,YAAY,CAAC,CAAC;QACtE,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8BAA8B,EAAE,YAAY,CAAC,CAAC;IAChE,CAAC;IAOD,KAAK,CAAC,SAAS,CAAC,GAAY,EAAE,GAAa;QAEzC,MAAM,SAAS,GAAG,IAAI,2BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3D,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,SAAS,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+BAA+B,SAAS,CAAC,SAAS,EAAE,EACpD,KAAK,CACN,CAAC;QAEF,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACnB,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAKD,KAAK,CAAC,aAAa,CAAC,GAAY,EAAE,GAAa;QAC7C,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,SAAmB,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,wCAAwC,SAAS,EAAE,EACnD,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;QAE5E,IAAI,CAAC;YACH,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,SAAS,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YAE3D,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,8BAA8B,EAC9B,YAAY,EACZ,YAAY,CACb,CAAC;YAEF,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,UAAU,EAAE,GAAG;gBACf,KAAK,EAAE,uBAAuB;gBAC9B,OAAO,EAAE,YAAY;aACtB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF,CAAA;AA/EY,gCAAU;qBAAV,UAAU;IADtB,IAAA,mBAAU,GAAE;IAOR,WAAA,IAAA,eAAM,EAAC,oBAAoB,CAAC,CAAA;6CAEF,kCAAe;QACjB,iCAAgB;GAThC,UAAU,CA+EtB"}
+{"version":3,"file":"sse.service.js","sourceRoot":"","sources":["../../../src/controllers/sse/sse.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oEAAoE;AACpE,oEAA6E;AAC7E,2CAAkE;AAIlE,kEAAiE;AACjE,sEAAkE;AAClE,oEAAgE;AAGzD,IAAM,UAAU,GAAhB,MAAM,UAAU;IAKF;IACA;IACA;IACA;IAPX,MAAM,CAAY;IAE1B,YAEmB,OAAyB,EACzB,QAAyB,EACzB,MAAwB,EACxB,cAA8B;QAH9B,YAAO,GAAP,OAAO,CAAkB;QACzB,aAAQ,GAAR,QAAQ,CAAiB;QACzB,WAAM,GAAN,MAAM,CAAkB;QACxB,mBAAc,GAAd,cAAc,CAAgB;QAE/C,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,sCAAsC,EAAE,YAAY,CAAC,CAAC;QACtE,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8BAA8B,EAAE,YAAY,CAAC,CAAC;IAChE,CAAC;IAOD,KAAK,CAAC,SAAS,CAAC,GAAY,EAAE,GAAa;QAEzC,MAAM,SAAS,GAAG,IAAI,2BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAE3D,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,EAAE;YAClD,SAAS;YACT,OAAO,EAAE,GAAG;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+BAA+B,SAAS,CAAC,SAAS,EAAE,EACpD,KAAK,CACN,CAAC;QAEF,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACnB,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAKD,KAAK,CAAC,aAAa,CAAC,GAAY,EAAE,GAAa;QAC7C,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,SAAmB,CAAC;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE1D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,wCAAwC,SAAS,EAAE,EACnD,KAAK,CACN,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;QAE5E,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QAEpC,IAAI,CAAC,CAAC,SAAS,YAAY,2BAAkB,CAAC,EAAE,CAAC;YAC/C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,SAAS,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YAE3D,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,8BAA8B,EAC9B,YAAY,EACZ,YAAY,CACb,CAAC;YAEF,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,UAAU,EAAE,GAAG;gBACf,KAAK,EAAE,uBAAuB;gBAC9B,OAAO,EAAE,YAAY;aACtB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF,CAAA;AA9FY,gCAAU;qBAAV,UAAU;IADtB,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,eAAM,EAAC,oBAAoB,CAAC,CAAA;6CAEF,kCAAe;QACjB,iCAAgB;QACR,gCAAc;GARtC,UAAU,CA8FtB"}

package/dist/controllers/streamable/streamable.service.d.ts

@@ -3,22 +3,17 @@ import { Request, Response } from 'express';
 import { McpModuleTransportOptions, McpServerOptions } from '../../interfaces/mcp-server-options.interface';
 import { McpLoggerService } from '../../registry/logger.service';
 import { RegistryService } from '../../registry/registry.service';
+import { SessionManager } from '../../services/session.manager';
 export declare class StreamableService implements OnModuleInit {
     private readonly options;
     private readonly transportOptions;
     private readonly registry;
     private readonly logger;
+    private readonly sessionManager;
     private server;
-    private transports;
-    constructor(options: McpServerOptions, transportOptions: McpModuleTransportOptions, registry: RegistryService, logger: McpLoggerService);
+    constructor(options: McpServerOptions, transportOptions: McpModuleTransportOptions, registry: RegistryService, logger: McpLoggerService, sessionManager: SessionManager);
     onModuleInit(): Promise<void>;
-    handlePostRequest(req: Request<any, any, any, {
-        sessionId?: string;
-    }>, res: Response): Promise<void>;
-    handleGetRequest(req: Request<any, any, any, {
-        sessionId?: string;
-    }>, res: Response): Promise<void>;
-    handleDeleteRequest(req: Request<any, any, any, {
-        sessionId?: string;
-    }>, res: Response): Promise<void>;
+    handlePostRequest(req: Request, res: Response): Promise<void>;
+    handleGetRequest(req: Request, res: Response): Promise<void>;
+    handleDeleteRequest(req: Request, res: Response): Promise<void>;
 }

package/dist/controllers/streamable/streamable.service.js

@@ -20,18 +20,20 @@ const common_1 = require("@nestjs/common");
 const crypto_1 = require("crypto");
 const logger_service_1 = require("../../registry/logger.service");
 const registry_service_1 = require("../../registry/registry.service");
+const session_manager_1 = require("../../services/session.manager");
 let StreamableService = class StreamableService {
     options;
     transportOptions;
     registry;
     logger;
+    sessionManager;
     server;
-    transports = {};
-    constructor(options, transportOptions, registry, logger) {
+    constructor(options, transportOptions, registry, logger, sessionManager) {
         this.options = options;
         this.transportOptions = transportOptions;
         this.registry = registry;
         this.logger = logger;
+        this.sessionManager = sessionManager;
         this.server = new mcp_js_1.McpServer(this.options.serverInfo, this.options.options);
     }
     async onModuleInit() {
@@ -42,23 +44,34 @@ let StreamableService = class StreamableService {
         const sessionId = req.headers['mcp-session-id'];
         let transport;
         const { options } = this.transportOptions?.streamable || {};
-        if (sessionId && this.transports[sessionId]) {
-            transport = this.transports[sessionId];
+        if (sessionId && this.sessionManager.getSession(sessionId)) {
+            const session = this.sessionManager.getSession(sessionId);
+            if (!session) {
+                throw new Error('Session not found');
+            }
+            if (!(session.transport instanceof streamableHttp_js_1.StreamableHTTPServerTransport)) {
+                throw new Error('Invalid transport');
+            }
+            transport = session.transport;
         }
         else if (!sessionId && (0, types_js_1.isInitializeRequest)(req.body)) {
             transport = new streamableHttp_js_1.StreamableHTTPServerTransport({
                 sessionIdGenerator: () => options?.sessionIdGenerator?.() || (0, crypto_1.randomUUID)(),
                 onsessioninitialized: (sessionId) => {
-                    this.transports[sessionId] = transport;
+                    this.sessionManager.setSession(sessionId, {
+                        transport,
+                        request: req,
+                    });
                 },
                 enableJsonResponse: options?.enableJsonResponse,
                 eventStore: options?.eventStore,
             });
             transport.onclose = () => {
                 if (transport.sessionId) {
-                    delete this.transports[transport.sessionId];
+                    this.sessionManager.deleteSession(transport.sessionId);
                 }
             };
+            await this.server.connect(transport);
         }
         else {
             res.status(400).json({
@@ -75,11 +88,18 @@ let StreamableService = class StreamableService {
     }
     async handleGetRequest(req, res) {
         const sessionId = req.headers['mcp-session-id'];
-        if (!sessionId || !this.transports[sessionId]) {
+        if (!sessionId || !this.sessionManager.getSession(sessionId)) {
             res.status(400).send('Invalid or missing session ID');
             return;
         }
-        const transport = this.transports[sessionId];
+        const session = this.sessionManager.getSession(sessionId);
+        if (!session) {
+            throw new Error('Session not found');
+        }
+        const { transport } = session;
+        if (!(transport instanceof streamableHttp_js_1.StreamableHTTPServerTransport)) {
+            throw new Error('Invalid transport');
+        }
         await transport.handleRequest(req, res);
     }
     async handleDeleteRequest(req, res) {
@@ -88,7 +108,11 @@ let StreamableService = class StreamableService {
             res.status(400).json({ error: 'Missing sessionId' });
             return;
         }
-        const transport = this.transports[sessionId];
+        const session = this.sessionManager.getSession(sessionId);
+        if (!session) {
+            throw new Error('Session not found');
+        }
+        const { transport } = session;
         if (transport) {
             this.logger.debug(`Closing streamable transport for sessionId: ${sessionId}`, 'STREAMABLE');
             await transport.close();
@@ -97,7 +121,7 @@ let StreamableService = class StreamableService {
                 res.status(400).json({ error: 'Invalid sessionId format' });
                 return;
             }
-            delete this.transports[sessionId];
+            this.sessionManager.deleteSession(sessionId);
             res.status(200).json({ success: true, sessionId });
         }
         else {
@@ -112,6 +136,7 @@ exports.StreamableService = StreamableService = __decorate([
     __param(0, (0, common_1.Inject)('MCP_SERVER_OPTIONS')),
     __param(1, (0, common_1.Inject)('MCP_TRANSPORT_OPTIONS')),
     __metadata("design:paramtypes", [Object, Object, registry_service_1.RegistryService,
-        logger_service_1.McpLoggerService])
+        logger_service_1.McpLoggerService,
+        session_manager_1.SessionManager])
 ], StreamableService);
 //# sourceMappingURL=streamable.service.js.map

package/dist/controllers/streamable/streamable.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"streamable.service.js","sourceRoot":"","sources":["../../../src/controllers/streamable/streamable.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oEAAoE;AACpE,0FAAmG;AACnG,iEAAyE;AACzE,2CAAkE;AAClE,mCAAoC;AAOpC,kEAAiE;AACjE,sEAAkE;AAK3D,IAAM,iBAAiB,GAAvB,MAAM,iBAAiB;IAOT;IAEA;IACA;IACA;IAVX,MAAM,CAAY;IAElB,UAAU,GAAG,EAAmD,CAAC;IAEzE,YAEmB,OAAyB,EAEzB,gBAA2C,EAC3C,QAAyB,EACzB,MAAwB;QAJxB,YAAO,GAAP,OAAO,CAAkB;QAEzB,qBAAgB,GAAhB,gBAAgB,CAA2B;QAC3C,aAAQ,GAAR,QAAQ,CAAiB;QACzB,WAAM,GAAN,MAAM,CAAkB;QAEzC,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IAYD,KAAK,CAAC,iBAAiB,CACrB,GAAmD,EACnD,GAAa;QAEb,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,IAAI,SAAwC,CAAC;QAE7C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,gBAAgB,EAAE,UAAU,IAAI,EAAE,CAAC;QAE5D,IAAI,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5C,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,CAAC,SAAS,IAAI,IAAA,8BAAmB,EAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACvD,SAAS,GAAG,IAAI,iDAA6B,CAAC;gBAC5C,kBAAkB,EAAE,GAAG,EAAE,CACvB,OAAO,EAAE,kBAAkB,EAAE,EAAE,IAAI,IAAA,mBAAU,GAAE;gBACjD,oBAAoB,EAAE,CAAC,SAAS,EAAE,EAAE;oBAClC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,SAAS,CAAC;gBACzC,CAAC;gBACD,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;gBAC/C,UAAU,EAAE,OAAO,EAAE,UAAU;aAChC,CAAC,CAAC;YAEH,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE;gBACvB,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;oBACxB,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YAEN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE;oBACL,IAAI,EAAE,CAAC,KAAK;oBACZ,OAAO,EAAE,2CAA2C;iBACrD;gBACD,EAAE,EAAE,IAAI;aACT,CAAC,CAAC;YAEH,OAAO;QACT,CAAC;QAED,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;IAUD,KAAK,CAAC,gBAAgB,CACpB,GAAmD,EACnD,GAAa;QAEb,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QAEtE,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE7C,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC1C,CAAC;IAYD,KAAK,CAAC,mBAAmB,CACvB,GAAmD,EACnD,GAAa;QAEb,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QAEtE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+CAA+C,SAAS,EAAE,EAC1D,YAAY,CACb,CAAC;YAEF,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;YAExB,MAAM,WAAW,GACf,wEAAwE,CAAC;YAE3E,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC;gBAC5D,OAAO;YACT,CAAC;YAED,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAElC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,gDAAgD,SAAS,EAAE,EAC3D,YAAY,CACb,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,qBAAqB,EAAE,SAAS,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;CACF,CAAA;AAtJY,8CAAiB;4BAAjB,iBAAiB;IAD7B,IAAA,mBAAU,GAAE;IAOR,WAAA,IAAA,eAAM,EAAC,oBAAoB,CAAC,CAAA;IAE5B,WAAA,IAAA,eAAM,EAAC,uBAAuB,CAAC,CAAA;qDAEL,kCAAe;QACjB,iCAAgB;GAXhC,iBAAiB,CAsJ7B"}
+{"version":3,"file":"streamable.service.js","sourceRoot":"","sources":["../../../src/controllers/streamable/streamable.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oEAAoE;AACpE,0FAAmG;AACnG,iEAAyE;AACzE,2CAAkE;AAClE,mCAAoC;AAOpC,kEAAiE;AACjE,sEAAkE;AAClE,oEAAgE;AAIzD,IAAM,iBAAiB,GAAvB,MAAM,iBAAiB;IAKT;IAEA;IACA;IACA;IACA;IATX,MAAM,CAAY;IAE1B,YAEmB,OAAyB,EAEzB,gBAA2C,EAC3C,QAAyB,EACzB,MAAwB,EACxB,cAA8B;QAL9B,YAAO,GAAP,OAAO,CAAkB;QAEzB,qBAAgB,GAAhB,gBAAgB,CAA2B;QAC3C,aAAQ,GAAR,QAAQ,CAAiB;QACzB,WAAM,GAAN,MAAM,CAAkB;QACxB,mBAAc,GAAd,cAAc,CAAgB;QAE/C,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IAYD,KAAK,CAAC,iBAAiB,CAAC,GAAY,EAAE,GAAa;QACjD,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QACtE,IAAI,SAAwC,CAAC;QAE7C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,gBAAgB,EAAE,UAAU,IAAI,EAAE,CAAC;QAE5D,IAAI,SAAS,IAAI,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAE1D,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;YAED,IAAI,CAAC,CAAC,OAAO,CAAC,SAAS,YAAY,iDAA6B,CAAC,EAAE,CAAC;gBAClE,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;YAED,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QAChC,CAAC;aAAM,IAAI,CAAC,SAAS,IAAI,IAAA,8BAAmB,EAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAEvD,SAAS,GAAG,IAAI,iDAA6B,CAAC;gBAC5C,kBAAkB,EAAE,GAAG,EAAE,CACvB,OAAO,EAAE,kBAAkB,EAAE,EAAE,IAAI,IAAA,mBAAU,GAAE;gBACjD,oBAAoB,EAAE,CAAC,SAAS,EAAE,EAAE;oBAClC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,EAAE;wBACxC,SAAS;wBACT,OAAO,EAAE,GAAG;qBACb,CAAC,CAAC;gBACL,CAAC;gBACD,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;gBAC/C,UAAU,EAAE,OAAO,EAAE,UAAU;aAChC,CAAC,CAAC;YAEH,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE;gBACvB,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;oBACxB,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YAEN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE;oBACL,IAAI,EAAE,CAAC,KAAK;oBACZ,OAAO,EAAE,2CAA2C;iBACrD;gBACD,EAAE,EAAE,IAAI;aACT,CAAC,CAAC;YAEH,OAAO;QACT,CAAC;QAED,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;IAUD,KAAK,CAAC,gBAAgB,CAAC,GAAY,EAAE,GAAa;QAChD,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QAEtE,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7D,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE1D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC;QAE9B,IAAI,CAAC,CAAC,SAAS,YAAY,iDAA6B,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC1C,CAAC;IAYD,KAAK,CAAC,mBAAmB,CAAC,GAAY,EAAE,GAAa;QACnD,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAuB,CAAC;QAEtE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE1D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC;QAE9B,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+CAA+C,SAAS,EAAE,EAC1D,YAAY,CACb,CAAC;YAEF,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;YAExB,MAAM,WAAW,GACf,wEAAwE,CAAC;YAE3E,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC;gBAC5D,OAAO;YACT,CAAC;YAED,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAE7C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,gDAAgD,SAAS,EAAE,EAC3D,YAAY,CACb,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,qBAAqB,EAAE,SAAS,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;CACF,CAAA;AA5KY,8CAAiB;4BAAjB,iBAAiB;IAD7B,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,eAAM,EAAC,oBAAoB,CAAC,CAAA;IAE5B,WAAA,IAAA,eAAM,EAAC,uBAAuB,CAAC,CAAA;qDAEL,kCAAe;QACjB,iCAAgB;QACR,gCAAc;GAVtC,iBAAiB,CA4K7B"}

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export * from './mcp.module';
 export * from './interfaces';
 export * from './registry';
+export * from './services';
 export * from './decorators';

package/dist/index.js

@@ -17,5 +17,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./mcp.module"), exports);
 __exportStar(require("./interfaces"), exports);
 __exportStar(require("./registry"), exports);
+__exportStar(require("./services"), exports);
 __exportStar(require("./decorators"), exports);
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AACA,+CAA6B;AAG7B,+CAA6B;AAG7B,6CAA2B;AAG3B,+CAA6B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AACA,+CAA6B;AAG7B,+CAA6B;AAG7B,6CAA2B;AAG3B,6CAA2B;AAG3B,+CAA6B"}

package/dist/interfaces/capabilities.interface.d.ts

@@ -1,5 +1,7 @@
-import { CompleteResourceTemplateCallback, ListResourcesCallback } from '@modelcontextprotocol/sdk/server/mcp';
-import { ZodOptional, ZodRawShape, ZodType, ZodTypeDef } from 'zod';
+import { CompleteResourceTemplateCallback, ListResourcesCallback } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { RequestHandlerExtra as SdkRequestHandlerExtra } from '@modelcontextprotocol/sdk/shared/protocol.js';
+import { ServerNotification, ServerRequest } from '@modelcontextprotocol/sdk/types.js';
+import { z, ZodOptional, ZodRawShape, ZodType, ZodTypeAny, ZodTypeDef } from 'zod';
 export interface ResourceBaseOptions {
     name: string;
 }
@@ -49,4 +51,32 @@ export interface TemplateCallbacks {
         [variable: string]: CompleteResourceTemplateCallback;
     };
 }
+export type RequestHandlerExtra = SdkRequestHandlerExtra<ServerRequest, ServerNotification> & {
+    headers: Record<string, string>;
+};
+export declare class ResourceUriHandlerParams {
+    readonly uri: URL;
+    readonly extra: RequestHandlerExtra;
+    private constructor();
+    static from(uri: URL, extra: RequestHandlerExtra): ResourceUriHandlerParams;
+}
+export declare class ResourceTemplateHandlerParams {
+    readonly uri: URL;
+    readonly variables?: Record<string, string>;
+    readonly extra: RequestHandlerExtra;
+    private constructor();
+    static from(uri: URL, extra: RequestHandlerExtra, variables?: Record<string, string>): ResourceTemplateHandlerParams;
+}
+export declare class PromptHandlerParams<Args extends PromptArgsRawShape | undefined = undefined> {
+    readonly args?: Args extends PromptArgsRawShape ? z.objectOutputType<Args, ZodTypeAny> : undefined;
+    readonly extra: RequestHandlerExtra;
+    private constructor();
+    static from<Args extends PromptArgsRawShape | undefined>(extra: RequestHandlerExtra, args?: Args extends PromptArgsRawShape ? z.objectOutputType<Args, ZodTypeAny> : undefined): PromptHandlerParams<Args>;
+}
+export declare class ToolHandlerParams<Args extends ZodRawShape | undefined = undefined> {
+    readonly args?: Args extends ZodRawShape ? z.objectOutputType<Args, ZodTypeAny> : undefined;
+    readonly extra: RequestHandlerExtra;
+    private constructor();
+    static from<Args extends ZodRawShape | undefined>(extra: RequestHandlerExtra, args?: Args extends ZodRawShape ? z.objectOutputType<Args, ZodTypeAny> : undefined): ToolHandlerParams<Args>;
+}
 export {};

package/dist/interfaces/capabilities.interface.js

@@ -1,3 +1,54 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolHandlerParams = exports.PromptHandlerParams = exports.ResourceTemplateHandlerParams = exports.ResourceUriHandlerParams = void 0;
+class ResourceUriHandlerParams {
+    uri;
+    extra;
+    constructor(uri, extra) {
+        this.uri = uri;
+        this.extra = extra;
+    }
+    static from(uri, extra) {
+        return new ResourceUriHandlerParams(uri, extra);
+    }
+}
+exports.ResourceUriHandlerParams = ResourceUriHandlerParams;
+class ResourceTemplateHandlerParams {
+    uri;
+    variables;
+    extra;
+    constructor(uri, extra, variables) {
+        this.uri = uri;
+        this.extra = extra;
+        this.variables = variables;
+    }
+    static from(uri, extra, variables) {
+        return new ResourceTemplateHandlerParams(uri, extra, variables);
+    }
+}
+exports.ResourceTemplateHandlerParams = ResourceTemplateHandlerParams;
+class PromptHandlerParams {
+    args;
+    extra;
+    constructor(extra, args) {
+        this.extra = extra;
+        this.args = args;
+    }
+    static from(extra, args) {
+        return new PromptHandlerParams(extra, args);
+    }
+}
+exports.PromptHandlerParams = PromptHandlerParams;
+class ToolHandlerParams {
+    args;
+    extra;
+    constructor(extra, args) {
+        this.extra = extra;
+        this.args = args;
+    }
+    static from(extra, args) {
+        return new ToolHandlerParams(extra, args);
+    }
+}
+exports.ToolHandlerParams = ToolHandlerParams;
 //# sourceMappingURL=capabilities.interface.js.map

package/dist/interfaces/capabilities.interface.js.map

@@ -1 +1 @@
-{"version":3,"file":"capabilities.interface.js","sourceRoot":"","sources":["../../src/interfaces/capabilities.interface.ts"],"names":[],"mappings":""}
+{"version":3,"file":"capabilities.interface.js","sourceRoot":"","sources":["../../src/interfaces/capabilities.interface.ts"],"names":[],"mappings":";;;AAmHA,MAAa,wBAAwB;IACnB,GAAG,CAAM;IACT,KAAK,CAAsB;IAE3C,YAAoB,GAAQ,EAAE,KAA0B;QACtD,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,GAAQ,EAAE,KAA0B;QAC9C,OAAO,IAAI,wBAAwB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;CACF;AAZD,4DAYC;AAED,MAAa,6BAA6B;IACxB,GAAG,CAAM;IACT,SAAS,CAA0B;IACnC,KAAK,CAAsB;IAE3C,YACE,GAAQ,EACR,KAA0B,EAC1B,SAAkC;QAElC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,IAAI,CACT,GAAQ,EACR,KAA0B,EAC1B,SAAkC;QAElC,OAAO,IAAI,6BAA6B,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IAClE,CAAC;CACF;AAtBD,sEAsBC;AAED,MAAa,mBAAmB;IAGd,IAAI,CAEN;IACE,KAAK,CAAsB;IAE3C,YACE,KAA0B,EAC1B,IAEa;QAEb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,IAAI,CACT,KAA0B,EAC1B,IAEa;QAEb,OAAO,IAAI,mBAAmB,CAAO,KAAK,EAAE,IAAI,CAAC,CAAC;IACpD,CAAC;CACF;AA1BD,kDA0BC;AAED,MAAa,iBAAiB;IAGZ,IAAI,CAEN;IACE,KAAK,CAAsB;IAE3C,YACE,KAA0B,EAC1B,IAEa;QAEb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,IAAI,CACT,KAA0B,EAC1B,IAEa;QAEb,OAAO,IAAI,iBAAiB,CAAO,KAAK,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;CACF;AA1BD,8CA0BC"}

package/dist/interfaces/context.interface.d.ts

@@ -0,0 +1,6 @@
+import { ExecutionContext } from '@nestjs/common';
+import { PromptHandlerParams, ResourceTemplateHandlerParams, ResourceUriHandlerParams, ToolHandlerParams } from './capabilities.interface';
+export interface McpExecutionContext extends ExecutionContext {
+    args: ResourceUriHandlerParams | ResourceTemplateHandlerParams | PromptHandlerParams | ToolHandlerParams;
+    getSessionId: () => string;
+}

package/dist/interfaces/context.interface.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=context.interface.js.map

package/dist/interfaces/context.interface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.interface.js","sourceRoot":"","sources":["../../src/interfaces/context.interface.ts"],"names":[],"mappings":""}

package/dist/interfaces/index.d.ts

@@ -1,2 +1,3 @@
-export * from './guards.interface';
+export * from './capabilities.interface';
+export * from './context.interface';
 export * from './mcp-server-options.interface';

package/dist/interfaces/index.js

@@ -14,6 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./guards.interface"), exports);
+__exportStar(require("./capabilities.interface"), exports);
+__exportStar(require("./context.interface"), exports);
 __exportStar(require("./mcp-server-options.interface"), exports);
 //# sourceMappingURL=index.js.map

package/dist/interfaces/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/interfaces/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,iEAA+C"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/interfaces/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2DAAyC;AACzC,sDAAoC;AACpC,iEAA+C"}

package/dist/interfaces/mcp-server-options.interface.d.ts

@@ -1,6 +1,6 @@
 import { StreamableHTTPServerTransportOptions } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
-import { ProtocolOptions } from '@modelcontextprotocol/sdk/shared/protocol';
-import { Implementation, ServerCapabilities } from '@modelcontextprotocol/sdk/types';
+import { ProtocolOptions } from '@modelcontextprotocol/sdk/shared/protocol.js';
+import { Implementation, ServerCapabilities } from '@modelcontextprotocol/sdk/types.js';
 import { Provider, Type } from '@nestjs/common';
 export type ServerOptions = {
     instructions?: string;

package/dist/interfaces/message.types.d.ts

@@ -0,0 +1,8 @@
+import { Request as ExpressRequest, Response as ExpressResponse } from 'express';
+export type Request = ExpressRequest & {
+    auth?: any;
+};
+export type McpMessage = {
+    req: Request;
+    res: ExpressResponse;
+};

package/dist/interfaces/{guards.interface.js → message.types.js}

@@ -1,3 +1,3 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=guards.interface.js.map
+//# sourceMappingURL=message.types.js.map

package/dist/interfaces/message.types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message.types.js","sourceRoot":"","sources":["../../src/interfaces/message.types.ts"],"names":[],"mappings":""}

package/dist/mcp.module.js

@@ -10,11 +10,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.McpModule = void 0;
 const common_1 = require("@nestjs/common");
 const core_1 = require("@nestjs/core");
+const async_hooks_1 = require("async_hooks");
 const sse_1 = require("./controllers/sse");
 const streamable_1 = require("./controllers/streamable");
 const discovery_service_1 = require("./registry/discovery.service");
 const logger_service_1 = require("./registry/logger.service");
 const registry_service_1 = require("./registry/registry.service");
+const session_manager_1 = require("./services/session.manager");
 let McpModule = McpModule_1 = class McpModule {
     static getActiveTransportControllersAndProviders(transports) {
         const controllers = new Set();
@@ -170,7 +172,17 @@ exports.McpModule = McpModule;
 exports.McpModule = McpModule = McpModule_1 = __decorate([
     (0, common_1.Module)({
         imports: [core_1.DiscoveryModule],
-        providers: [registry_service_1.RegistryService, discovery_service_1.DiscoveryService, logger_service_1.McpLoggerService],
+        providers: [
+            registry_service_1.RegistryService,
+            discovery_service_1.DiscoveryService,
+            {
+                provide: async_hooks_1.AsyncLocalStorage,
+                useValue: new async_hooks_1.AsyncLocalStorage(),
+            },
+            logger_service_1.McpLoggerService,
+            session_manager_1.SessionManager,
+        ],
+        exports: [session_manager_1.SessionManager],
     })
 ], McpModule);
 //# sourceMappingURL=mcp.module.js.map