@frontmcp/sdk 0.2.5 → 0.3.0

package/src/transport/adapters/transport.local.adapter.d.ts

@@ -0,0 +1,41 @@
+import { AuthenticatedServerRequest } from '../../server/server.types';
+import { TransportKey, TypedElicitResult } from '../transport.types';
+import { Server as McpServer } from '@modelcontextprotocol/sdk/server/index.js';
+import { ElicitResult, RequestId } from '@modelcontextprotocol/sdk/types.js';
+import { InMemoryEventStore } from '../transport.event-store';
+import { AuthInfo } from '@modelcontextprotocol/sdk/server/auth/types.js';
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import { SSEServerTransport } from '../legacy/legacy.sse.tranporter';
+import { ZodObject } from 'zod';
+import { FrontMcpLogger, ServerResponse } from '../../common';
+import { Scope } from '../../scope';
+export declare abstract class LocalTransportAdapter<T extends StreamableHTTPServerTransport | SSEServerTransport> {
+    #private;
+    protected readonly scope: Scope;
+    protected readonly key: TransportKey;
+    protected readonly onDispose: () => void;
+    protected logger: FrontMcpLogger;
+    protected transport: T;
+    protected eventStore: InMemoryEventStore;
+    protected elicitHandler: {
+        resolve: (result: ElicitResult) => void;
+        reject: (err: unknown) => void;
+    } | undefined;
+    ready: Promise<void>;
+    server: McpServer;
+    constructor(scope: Scope, key: TransportKey, onDispose: () => void, res: ServerResponse);
+    abstract createTransport(sessionId: string, response: ServerResponse): T;
+    abstract initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    abstract sendElicitRequest<T extends ZodObject<any>>(relatedRequestId: RequestId, message: string, requestedSchema: T): Promise<TypedElicitResult<T>>;
+    abstract handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    connectServer(): Promise<void>;
+    get newRequestId(): RequestId;
+    destroy(reason?: string): Promise<void>;
+    /**
+     * Ping the connected client for this transport.
+     * Returns true on success, false on timeout/error.
+     */
+    ping(timeoutMs?: number): Promise<boolean>;
+    protected ensureAuthInfo(req: AuthenticatedServerRequest, transport: LocalTransportAdapter<T>): AuthInfo;
+    handleIfElicitResult(req: AuthenticatedServerRequest): boolean;
+}

package/src/transport/adapters/transport.local.adapter.js

@@ -0,0 +1,127 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LocalTransportAdapter = void 0;
+const index_js_1 = require("@modelcontextprotocol/sdk/server/index.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const transport_event_store_1 = require("../transport.event-store");
+const common_1 = require("../../common");
+const mcp_handlers_1 = require("../mcp-handlers");
+class LocalTransportAdapter {
+    scope;
+    key;
+    onDispose;
+    logger;
+    transport;
+    eventStore = new transport_event_store_1.InMemoryEventStore();
+    elicitHandler = undefined;
+    #requestId = 1;
+    ready;
+    server;
+    constructor(scope, key, onDispose, res) {
+        this.scope = scope;
+        this.key = key;
+        this.onDispose = onDispose;
+        this.logger = scope.logger.child('LocalTransportAdapter');
+        this.transport = this.createTransport(key.sessionId, res);
+        this.ready = this.connectServer();
+    }
+    connectServer() {
+        const { info } = this.scope.metadata;
+        // TODO: collect server options from scope
+        const serverOptions = {
+            instructions: '',
+            capabilities: {
+                tools: {
+                    subscribe: true,
+                    listChanged: true,
+                },
+            },
+            serverInfo: info,
+        };
+        this.server = new index_js_1.Server(info, serverOptions);
+        const handlers = (0, mcp_handlers_1.createMcpHandlers)({
+            scope: this.scope,
+            serverOptions,
+        });
+        for (const handler of handlers) {
+            this.server.setRequestHandler(handler.requestSchema, handler.handler);
+        }
+        return this.server.connect(this.transport);
+    }
+    get newRequestId() {
+        return this.#requestId++;
+    }
+    async destroy(reason) {
+        console.log('destroying transporter, reason:', reason);
+        try {
+            // if(!this.transport.closed){
+            //   this.transport.close();
+            // }
+        }
+        catch {
+            /* empty */
+        }
+        if (reason) {
+            console.warn(`Destroying transporter: ${reason}`);
+        }
+        else {
+            try {
+                this.onDispose?.();
+            }
+            catch {
+                /* empty */
+            }
+        }
+    }
+    /**
+     * Ping the connected client for this transport.
+     * Returns true on success, false on timeout/error.
+     */
+    async ping(timeoutMs = 10_000) {
+        try {
+            await this.ready; // ensure server.connect(...) finished
+            // Preferred if your SDK exposes it:
+            // await this.server.ping({ timeout: timeoutMs });
+            // Works on all versions (low-level request):
+            await this.server.request({ method: 'ping' }, types_js_1.EmptyResultSchema, { timeout: timeoutMs });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    ensureAuthInfo(req, transport) {
+        const { token, user, session } = req[common_1.ServerRequestTokens.auth];
+        req.auth = {
+            token,
+            user,
+            sessionId: session.id,
+            sessionIdPayload: session.payload,
+            scopes: [],
+            clientId: user.sub ?? '',
+            transport,
+        };
+        return req.auth;
+    }
+    handleIfElicitResult(req) {
+        if (!this.elicitHandler) {
+            return false;
+        }
+        if (req.body.error) {
+            const { code, message } = req.body.error;
+            if (code === -32600 && message === 'Elicitation not supported') {
+                this.elicitHandler.reject(req.body.error);
+                return true;
+            }
+            return false;
+        }
+        const parsed = types_js_1.ElicitResultSchema.safeParse(req.body?.result);
+        if (parsed.success) {
+            this.elicitHandler.resolve(parsed.data);
+            return true;
+        }
+        return false;
+    }
+}
+exports.LocalTransportAdapter = LocalTransportAdapter;
+//# sourceMappingURL=transport.local.adapter.js.map

package/src/transport/adapters/transport.local.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.local.adapter.js","sourceRoot":"","sources":["../../../../src/transport/adapters/transport.local.adapter.ts"],"names":[],"mappings":";;;AAEA,wEAA8E;AAC9E,iEAAkH;AAClH,oEAA4D;AAK5D,yCAAiF;AAEjF,kDAAkD;AAElD,MAAsB,qBAAqB;IAWpB;IACA;IACA;IAZX,MAAM,CAAiB;IACvB,SAAS,CAAI;IACb,UAAU,GAAG,IAAI,0CAAkB,EAAE,CAAC;IACtC,aAAa,GACrB,SAAS,CAAC;IACZ,UAAU,GAAG,CAAC,CAAC;IACf,KAAK,CAAgB;IACrB,MAAM,CAAY;IAElB,YACqB,KAAY,EACZ,GAAiB,EACjB,SAAqB,EACxC,GAAmB;QAHA,UAAK,GAAL,KAAK,CAAO;QACZ,QAAG,GAAH,GAAG,CAAc;QACjB,cAAS,GAAT,SAAS,CAAY;QAGxC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;IACpC,CAAC;IAcD,aAAa;QACX,MAAM,EAAC,IAAI,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAEnC,0CAA0C;QAC1C,MAAM,aAAa,GAAG;YACpB,YAAY,EAAE,EAAE;YAChB,YAAY,EAAE;gBACZ,KAAK,EAAE;oBACL,SAAS,EAAE,IAAI;oBACf,WAAW,EAAE,IAAI;iBAClB;aACF;YACD,UAAU,EAAE,IAAI;SACjB,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAS,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,IAAA,gCAAiB,EAAC;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa;SACd,CAAC,CAAC;QAEH,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,OAAc,CAAC,CAAC;QAC/E,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAe;QAC3B,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,MAAM,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,8BAA8B;YAC9B,4BAA4B;YAC5B,IAAI;QACN,CAAC;QAAC,MAAM,CAAC;YACP,WAAW;QACb,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,2BAA2B,MAAM,EAAE,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,WAAW;YACb,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,MAAM;QAC3B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,sCAAsC;YACxD,oCAAoC;YACpC,kDAAkD;YAElD,6CAA6C;YAC7C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAC,MAAM,EAAE,MAAM,EAAC,EAAE,4BAAiB,EAAE,EAAC,OAAO,EAAE,SAAS,EAAC,CAAC,CAAC;YACrF,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAES,cAAc,CAAC,GAA+B,EAAE,SAAmC;QAC3F,MAAM,EAAC,KAAK,EAAE,IAAI,EAAE,OAAO,EAAC,GAAG,GAAG,CAAC,4BAAmB,CAAC,IAAI,CAAC,CAAC;QAC7D,GAAG,CAAC,IAAI,GAAG;YACT,KAAK;YACL,IAAI;YACJ,SAAS,EAAE,OAAQ,CAAC,EAAE;YACtB,gBAAgB,EAAE,OAAQ,CAAC,OAAO;YAClC,MAAM,EAAE,EAAE;YACV,QAAQ,EAAE,IAAI,CAAC,GAAG,IAAI,EAAE;YACxB,SAAS;SACS,CAAC;QACrB,OAAO,GAAG,CAAC,IAAI,CAAC;IAClB,CAAC;IAED,oBAAoB,CAAC,GAA+B;QAElD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,EAAC,IAAI,EAAE,OAAO,EAAC,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;YACvC,IAAI,IAAI,KAAK,CAAC,KAAK,IAAI,OAAO,KAAK,2BAA2B,EAAE,CAAC;gBAC/D,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC1C,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,6BAAkB,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACxC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAzID,sDAyIC","sourcesContent":["import {AuthenticatedServerRequest} from '../../server/server.types';\nimport {TransportKey, TypedElicitResult} from '../transport.types';\nimport {Server as McpServer} from '@modelcontextprotocol/sdk/server/index.js';\nimport {EmptyResultSchema, ElicitResult, RequestId, ElicitResultSchema} from '@modelcontextprotocol/sdk/types.js';\nimport {InMemoryEventStore} from '../transport.event-store';\nimport {AuthInfo} from '@modelcontextprotocol/sdk/server/auth/types.js';\nimport {StreamableHTTPServerTransport} from '@modelcontextprotocol/sdk/server/streamableHttp.js';\nimport {SSEServerTransport} from '../legacy/legacy.sse.tranporter';\nimport {ZodObject} from 'zod';\nimport {FrontMcpLogger, ServerRequestTokens, ServerResponse} from '../../common';\nimport {Scope} from '../../scope';\nimport {createMcpHandlers} from '../mcp-handlers';\n\nexport abstract class LocalTransportAdapter<T extends StreamableHTTPServerTransport | SSEServerTransport> {\n  protected logger: FrontMcpLogger;\n  protected transport: T;\n  protected eventStore = new InMemoryEventStore();\n  protected elicitHandler: { resolve: (result: ElicitResult) => void; reject: (err: unknown) => void } | undefined =\n    undefined;\n  #requestId = 1;\n  ready: Promise<void>;\n  server: McpServer;\n\n  constructor(\n    protected readonly scope: Scope,\n    protected readonly key: TransportKey,\n    protected readonly onDispose: () => void,\n    res: ServerResponse,\n  ) {\n    this.logger = scope.logger.child('LocalTransportAdapter');\n    this.transport = this.createTransport(key.sessionId, res);\n    this.ready = this.connectServer();\n  }\n\n  abstract createTransport(sessionId: string, response: ServerResponse): T;\n\n  abstract initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;\n\n  abstract sendElicitRequest<T extends ZodObject<any>>(\n    relatedRequestId: RequestId,\n    message: string,\n    requestedSchema: T,\n  ): Promise<TypedElicitResult<T>>;\n\n  abstract handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;\n\n  connectServer() {\n    const {info} = this.scope.metadata;\n\n    // TODO: collect server options from scope\n    const serverOptions = {\n      instructions: '',\n      capabilities: {\n        tools: {\n          subscribe: true,\n          listChanged: true,\n        },\n      },\n      serverInfo: info,\n    };\n    this.server = new McpServer(info, serverOptions);\n    const handlers = createMcpHandlers({\n      scope: this.scope,\n      serverOptions,\n    });\n\n    for (const handler of handlers) {\n      this.server.setRequestHandler(handler.requestSchema, handler.handler as any);\n    }\n    return this.server.connect(this.transport);\n  }\n\n  get newRequestId(): RequestId {\n    return this.#requestId++;\n  }\n\n  async destroy(reason?: string): Promise<void> {\n    console.log('destroying transporter, reason:', reason);\n\n    try {\n      // if(!this.transport.closed){\n      //   this.transport.close();\n      // }\n    } catch {\n      /* empty */\n    }\n    if (reason) {\n      console.warn(`Destroying transporter: ${reason}`);\n    } else {\n      try {\n        this.onDispose?.();\n      } catch {\n        /* empty */\n      }\n    }\n  }\n\n  /**\n   * Ping the connected client for this transport.\n   * Returns true on success, false on timeout/error.\n   */\n  async ping(timeoutMs = 10_000): Promise<boolean> {\n    try {\n      await this.ready; // ensure server.connect(...) finished\n      // Preferred if your SDK exposes it:\n      // await this.server.ping({ timeout: timeoutMs });\n\n      // Works on all versions (low-level request):\n      await this.server.request({method: 'ping'}, EmptyResultSchema, {timeout: timeoutMs});\n      return true;\n    } catch {\n      return false;\n    }\n  }\n\n  protected ensureAuthInfo(req: AuthenticatedServerRequest, transport: LocalTransportAdapter<T>) {\n    const {token, user, session} = req[ServerRequestTokens.auth];\n    req.auth = {\n      token,\n      user,\n      sessionId: session!.id,\n      sessionIdPayload: session!.payload,\n      scopes: [],\n      clientId: user.sub ?? '',\n      transport,\n    } satisfies AuthInfo;\n    return req.auth;\n  }\n\n  handleIfElicitResult(req: AuthenticatedServerRequest): boolean {\n\n    if (!this.elicitHandler) {\n      return false;\n    }\n    if (req.body.error) {\n      const {code, message} = req.body.error;\n      if (code === -32600 && message === 'Elicitation not supported') {\n        this.elicitHandler.reject(req.body.error);\n        return true;\n      }\n      return false;\n    }\n\n    const parsed = ElicitResultSchema.safeParse(req.body?.result);\n    if (parsed.success) {\n      this.elicitHandler.resolve(parsed.data);\n      return true;\n    }\n    return false;\n  }\n}\n"]}

package/src/transport/adapters/transport.sse.adapter.d.ts

@@ -0,0 +1,14 @@
+import { AuthenticatedServerRequest } from '../../server/server.types';
+import { TypedElicitResult } from '../transport.types';
+import { SSEServerTransport } from '../legacy/legacy.sse.tranporter';
+import { LocalTransportAdapter } from './transport.local.adapter';
+import { RequestId } from '@modelcontextprotocol/sdk/types.js';
+import { ZodObject } from 'node_modules/zod/v3/types.cjs';
+import { ServerResponse } from '../../common';
+export declare class TransportSSEAdapter extends LocalTransportAdapter<SSEServerTransport> {
+    sessionId: string;
+    createTransport(sessionId: string, res: ServerResponse): SSEServerTransport;
+    initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    sendElicitRequest<T extends ZodObject<any>>(relatedRequestId: RequestId, message: string, requestedSchema: T): Promise<TypedElicitResult<T>>;
+}

package/src/transport/adapters/transport.sse.adapter.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TransportSSEAdapter = void 0;
+const legacy_sse_tranporter_1 = require("../legacy/legacy.sse.tranporter");
+const transport_local_adapter_1 = require("./transport.local.adapter");
+const transport_error_1 = require("../transport.error");
+const zod_to_json_schema_1 = require("zod-to-json-schema");
+class TransportSSEAdapter extends transport_local_adapter_1.LocalTransportAdapter {
+    sessionId;
+    createTransport(sessionId, res) {
+        this.sessionId = sessionId;
+        this.logger.info(`new transport session: ${sessionId.slice(0, 40)}`);
+        const transport = new legacy_sse_tranporter_1.SSEServerTransport('/message', res, {
+            sessionId: sessionId,
+        });
+        transport.onerror = (error) => {
+            console.error('SSE error:', error);
+        };
+        transport.onclose = this.destroy.bind(this);
+        return transport;
+    }
+    initialize(req, res) {
+        this.logger.verbose(`[${this.sessionId}] handle initialize request`);
+        this.ensureAuthInfo(req, this);
+        return Promise.resolve();
+    }
+    async handleRequest(req, res) {
+        const authInfo = this.ensureAuthInfo(req, this);
+        if (this.handleIfElicitResult(req)) {
+            this.logger.verbose(`[${this.sessionId}] handle get request`);
+            return;
+        }
+        if (req.method === 'GET') {
+            this.logger.verbose(`[${this.sessionId}] handle get request`);
+            return this.transport.handleMessage(req.body, { requestInfo: req, authInfo });
+        }
+        else {
+            this.logger.verbose(`[${this.sessionId}] handle post request`);
+            return this.transport.handlePostMessage(req, res, req.body);
+        }
+    }
+    async sendElicitRequest(relatedRequestId, message, requestedSchema) {
+        console.log('sendElicitRequest', { relatedRequestId });
+        await this.transport.send((0, transport_error_1.rpcRequest)(this.newRequestId, 'elicitation/create', {
+            message,
+            requestedSchema: (0, zod_to_json_schema_1.zodToJsonSchema)(requestedSchema),
+        }));
+        return new Promise((resolve, reject) => {
+            this.elicitHandler = {
+                resolve: (result) => {
+                    resolve(result);
+                    this.elicitHandler = undefined;
+                },
+                reject: (err) => {
+                    reject(err);
+                    this.elicitHandler = undefined;
+                },
+            };
+        });
+    }
+}
+exports.TransportSSEAdapter = TransportSSEAdapter;
+//# sourceMappingURL=transport.sse.adapter.js.map

package/src/transport/adapters/transport.sse.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.sse.adapter.js","sourceRoot":"","sources":["../../../../src/transport/adapters/transport.sse.adapter.ts"],"names":[],"mappings":";;;AAEA,2EAAqE;AACrE,uEAAkE;AAGlE,wDAAgD;AAChD,2DAAqD;AAGrD,MAAa,mBAAoB,SAAQ,+CAAyC;IAChF,SAAS,CAAS;IAET,eAAe,CAAC,SAAiB,EAAE,GAAmB;QAC7D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACrE,MAAM,SAAS,GAAG,IAAI,0CAAkB,CAAC,UAAU,EAAE,GAAG,EAAE;YACxD,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QACH,SAAS,CAAC,OAAO,GAAG,CAAC,KAAK,EAAE,EAAE;YAC5B,OAAO,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC,CAAC;QACF,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,UAAU,CAAC,GAA+B,EAAE,GAAmB;QAC7D,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,SAAS,6BAA6B,CAAC,CAAC;QACrE,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC/B,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAA+B,EAAE,GAAmB;QAEtE,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAEhD,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,SAAS,sBAAsB,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,SAAS,sBAAsB,CAAC,CAAC;YAC9D,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;QAChF,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,SAAS,uBAAuB,CAAC,CAAC;YAC/D,OAAO,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,gBAA2B,EAC3B,OAAe,EACf,eAAkB;QAGlB,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QACvD,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CACvB,IAAA,4BAAU,EAAC,IAAI,CAAC,YAAY,EAAE,oBAAoB,EAAE;YAClD,OAAO;YACP,eAAe,EAAE,IAAA,oCAAe,EAAC,eAAe,CAAC;SAClD,CAAC,CACH,CAAC;QAEF,OAAO,IAAI,OAAO,CAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3D,IAAI,CAAC,aAAa,GAAG;gBACnB,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE;oBAClB,OAAO,CAAC,MAA8B,CAAC,CAAC;oBACxC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBACjC,CAAC;gBACD,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;oBACd,MAAM,CAAC,GAAG,CAAC,CAAC;oBACZ,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBACjC,CAAC;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAlED,kDAkEC","sourcesContent":["import { AuthenticatedServerRequest } from '../../server/server.types';\nimport { TypedElicitResult } from '../transport.types';\nimport { SSEServerTransport } from '../legacy/legacy.sse.tranporter';\nimport { LocalTransportAdapter } from './transport.local.adapter';\nimport { RequestId } from '@modelcontextprotocol/sdk/types.js';\nimport { ZodObject } from 'node_modules/zod/v3/types.cjs';\nimport { rpcRequest } from '../transport.error';\nimport { zodToJsonSchema } from 'zod-to-json-schema';\nimport { ServerResponse } from '../../common';\n\nexport class TransportSSEAdapter extends LocalTransportAdapter<SSEServerTransport> {\n  sessionId: string;\n\n  override createTransport(sessionId: string, res: ServerResponse): SSEServerTransport {\n    this.sessionId = sessionId;\n    this.logger.info(`new transport session: ${sessionId.slice(0, 40)}`);\n    const transport = new SSEServerTransport('/message', res, {\n      sessionId: sessionId,\n    });\n    transport.onerror = (error) => {\n      console.error('SSE error:', error);\n    };\n    transport.onclose = this.destroy.bind(this);\n    return transport;\n  }\n\n  initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void> {\n    this.logger.verbose(`[${this.sessionId}] handle initialize request`);\n    this.ensureAuthInfo(req, this);\n    return Promise.resolve();\n  }\n\n  async handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void> {\n\n    const authInfo = this.ensureAuthInfo(req, this);\n\n    if (this.handleIfElicitResult(req)) {\n      this.logger.verbose(`[${this.sessionId}] handle get request`);\n      return;\n    }\n    if (req.method === 'GET') {\n      this.logger.verbose(`[${this.sessionId}] handle get request`);\n      return this.transport.handleMessage(req.body, { requestInfo: req, authInfo });\n    } else {\n      this.logger.verbose(`[${this.sessionId}] handle post request`);\n      return this.transport.handlePostMessage(req, res, req.body);\n    }\n  }\n\n  async sendElicitRequest<T extends ZodObject<any>>(\n    relatedRequestId: RequestId,\n    message: string,\n    requestedSchema: T,\n  ): Promise<TypedElicitResult<T>> {\n\n    console.log('sendElicitRequest', { relatedRequestId });\n    await this.transport.send(\n      rpcRequest(this.newRequestId, 'elicitation/create', {\n        message,\n        requestedSchema: zodToJsonSchema(requestedSchema),\n      }),\n    );\n\n    return new Promise<TypedElicitResult<T>>((resolve, reject) => {\n      this.elicitHandler = {\n        resolve: (result) => {\n          resolve(result as TypedElicitResult<T>);\n          this.elicitHandler = undefined;\n        },\n        reject: (err) => {\n          reject(err);\n          this.elicitHandler = undefined;\n        },\n      };\n    });\n  }\n}\n"]}

package/src/transport/adapters/transport.streamable-http.adapter.d.ts

@@ -0,0 +1,13 @@
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import { TypedElicitResult } from '../transport.types';
+import { AuthenticatedServerRequest } from '../../server/server.types';
+import { LocalTransportAdapter } from './transport.local.adapter';
+import { RequestId } from '@modelcontextprotocol/sdk/types.js';
+import { ZodObject } from 'zod';
+import { ServerResponse } from '../../common';
+export declare class TransportStreamableHttpAdapter extends LocalTransportAdapter<StreamableHTTPServerTransport> {
+    createTransport(sessionId: string, response: ServerResponse): StreamableHTTPServerTransport;
+    initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void>;
+    sendElicitRequest<T extends ZodObject<any>>(relatedRequestId: RequestId, message: string, requestedSchema: T): Promise<TypedElicitResult<T>>;
+}

package/src/transport/adapters/transport.streamable-http.adapter.js

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TransportStreamableHttpAdapter = void 0;
+const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+const transport_local_adapter_1 = require("./transport.local.adapter");
+const zod_to_json_schema_1 = require("zod-to-json-schema");
+const transport_error_1 = require("../transport.error");
+class TransportStreamableHttpAdapter extends transport_local_adapter_1.LocalTransportAdapter {
+    createTransport(sessionId, response) {
+        return new streamableHttp_js_1.StreamableHTTPServerTransport({
+            sessionIdGenerator: () => {
+                return sessionId;
+            },
+            onsessionclosed: () => {
+                // this.destroy();
+            },
+            onsessioninitialized: (sessionId) => {
+                console.log(`session initialized: ${sessionId.slice(0, 40)}`);
+            },
+            eventStore: this.eventStore,
+        });
+    }
+    initialize(req, res) {
+        this.ensureAuthInfo(req, this);
+        return this.transport.handleRequest(req, res, req.body);
+    }
+    async handleRequest(req, res) {
+        this.ensureAuthInfo(req, this);
+        //
+        // if ((await this.isInitialized) && isInitializeRequest(req.body)) {
+        //   await this.transport.close();
+        //   this.isInitialized = this.connectStreamableTransport(req.authSession.session!.id);
+        //   await this.ready;
+        // }
+        if (req.method === 'GET') {
+            return this.transport.handleRequest(req, res);
+        }
+        else {
+            if (this.handleIfElicitResult(req)) {
+                return;
+            }
+            return this.transport.handleRequest(req, res, req.body);
+        }
+    }
+    async sendElicitRequest(relatedRequestId, message, requestedSchema) {
+        await this.transport.send((0, transport_error_1.rpcRequest)(this.newRequestId, 'elicitation/create', {
+            message,
+            requestedSchema: (0, zod_to_json_schema_1.zodToJsonSchema)(requestedSchema),
+        }), { relatedRequestId });
+        return new Promise((resolve, reject) => {
+            this.elicitHandler = {
+                resolve: (result) => {
+                    resolve(result);
+                    this.elicitHandler = undefined;
+                },
+                reject: (err) => {
+                    reject(err);
+                    this.elicitHandler = undefined;
+                },
+            };
+        });
+    }
+}
+exports.TransportStreamableHttpAdapter = TransportStreamableHttpAdapter;
+//# sourceMappingURL=transport.streamable-http.adapter.js.map

package/src/transport/adapters/transport.streamable-http.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.streamable-http.adapter.js","sourceRoot":"","sources":["../../../../src/transport/adapters/transport.streamable-http.adapter.ts"],"names":[],"mappings":";;;AAAA,0FAAmG;AAGnG,uEAAkE;AAElE,2DAAqD;AAErD,wDAAgD;AAGhD,MAAa,8BAA+B,SAAQ,+CAAoD;IAE7F,eAAe,CAAC,SAAiB,EAAE,QAAwB;QAChE,OAAO,IAAI,iDAA6B,CAAC;YACvC,kBAAkB,EAAE,GAAG,EAAE;gBACvB,OAAO,SAAS,CAAA;YAClB,CAAC;YACD,eAAe,EAAE,GAAG,EAAE;gBACpB,kBAAkB;YACpB,CAAC;YACD,oBAAoB,EAAE,CAAC,SAAS,EAAE,EAAE;gBAClC,OAAO,CAAC,GAAG,CAAC,wBAAwB,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC,CAAA;IACN,CAAC;IAED,UAAU,CAAC,GAA+B,EAAE,GAAmB;QAC7D,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC/B,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAA+B,EAAE,GAAmB;QACtE,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC/B,EAAE;QACF,qEAAqE;QACrE,kCAAkC;QAClC,uFAAuF;QACvF,sBAAsB;QACtB,IAAI;QAEJ,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC;aAAM,CAAC;YACN,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,OAAO,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,gBAA2B,EAC3B,OAAe,EACf,eAAkB;QAElB,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CACvB,IAAA,4BAAU,EAAC,IAAI,CAAC,YAAY,EAAE,oBAAoB,EAAE;YAClD,OAAO;YACP,eAAe,EAAE,IAAA,oCAAe,EAAC,eAAe,CAAC;SAClD,CAAC,EACF,EAAE,gBAAgB,EAAE,CACrB,CAAC;QAEF,OAAO,IAAI,OAAO,CAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3D,IAAI,CAAC,aAAa,GAAG;gBACnB,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE;oBAClB,OAAO,CAAC,MAA8B,CAAC,CAAC;oBACxC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBACjC,CAAC;gBACD,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;oBACd,MAAM,CAAC,GAAG,CAAC,CAAC;oBACZ,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;gBACjC,CAAC;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CAEF;AApED,wEAoEC","sourcesContent":["import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';\nimport { TypedElicitResult } from '../transport.types';\nimport { AuthenticatedServerRequest } from '../../server/server.types';\nimport { LocalTransportAdapter } from './transport.local.adapter';\nimport { RequestId } from '@modelcontextprotocol/sdk/types.js';\nimport { zodToJsonSchema } from 'zod-to-json-schema';\nimport { ZodObject } from 'zod';\nimport { rpcRequest } from '../transport.error';\nimport { ServerResponse } from '../../common';\n\nexport class TransportStreamableHttpAdapter extends LocalTransportAdapter<StreamableHTTPServerTransport> {\n\n  override createTransport(sessionId: string, response: ServerResponse): StreamableHTTPServerTransport {\n      return new StreamableHTTPServerTransport({\n        sessionIdGenerator: () => {\n          return sessionId\n        },\n        onsessionclosed: () => {\n          // this.destroy();\n        },\n        onsessioninitialized: (sessionId) => {\n          console.log(`session initialized: ${sessionId.slice(0, 40)}`);\n        },\n        eventStore: this.eventStore,\n      })\n  }\n\n  initialize(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void> {\n    this.ensureAuthInfo(req, this);\n    return this.transport.handleRequest(req, res, req.body);\n  }\n\n  async handleRequest(req: AuthenticatedServerRequest, res: ServerResponse): Promise<void> {\n    this.ensureAuthInfo(req, this);\n    //\n    // if ((await this.isInitialized) && isInitializeRequest(req.body)) {\n    //   await this.transport.close();\n    //   this.isInitialized = this.connectStreamableTransport(req.authSession.session!.id);\n    //   await this.ready;\n    // }\n\n    if (req.method === 'GET') {\n      return this.transport.handleRequest(req, res);\n    } else {\n      if (this.handleIfElicitResult(req)) {\n        return;\n      }\n      return this.transport.handleRequest(req, res, req.body);\n    }\n  }\n\n  async sendElicitRequest<T extends ZodObject<any>>(\n    relatedRequestId: RequestId,\n    message: string,\n    requestedSchema: T,\n  ): Promise<TypedElicitResult<T>> {\n    await this.transport.send(\n      rpcRequest(this.newRequestId, 'elicitation/create', {\n        message,\n        requestedSchema: zodToJsonSchema(requestedSchema),\n      }),\n      { relatedRequestId },\n    );\n\n    return new Promise<TypedElicitResult<T>>((resolve, reject) => {\n      this.elicitHandler = {\n        resolve: (result) => {\n          resolve(result as TypedElicitResult<T>);\n          this.elicitHandler = undefined;\n        },\n        reject: (err) => {\n          reject(err);\n          this.elicitHandler = undefined;\n        },\n      };\n    });\n  }\n\n}\n"]}

package/src/transport/flows/handle.sse.flow.d.ts

@@ -0,0 +1,92 @@
+import { httpInputSchema, FlowRunOptions, httpOutputSchema, FlowBase } from '../../common';
+import { z } from 'zod';
+export declare const plan: {
+    readonly pre: ["parseInput", "router"];
+    readonly execute: ["onInitialize", "onMessage", "onElicitResult"];
+    readonly post: [];
+    readonly finalize: ["cleanup"];
+};
+export declare const stateSchema: z.ZodObject<{
+    token: z.ZodString;
+    session: z.ZodObject<{
+        id: z.ZodString;
+        payload: z.ZodObject<{
+            nodeId: z.ZodString;
+            authSig: z.ZodString;
+            uuid: z.ZodString;
+            iat: z.ZodNumber;
+            protocol: z.ZodEnum<["legacy-sse", "sse", "streamable-http", "stateful-http", "stateless-http"]>;
+        }, "strip", z.ZodTypeAny, {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        }, {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        }>;
+    }, "strip", z.ZodTypeAny, {
+        id: string;
+        payload: {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        };
+    }, {
+        id: string;
+        payload: {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        };
+    }>;
+    requestType: z.ZodOptional<z.ZodEnum<["initialize", "message", "elicitResult"]>>;
+}, "strip", z.ZodTypeAny, {
+    session: {
+        id: string;
+        payload: {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        };
+    };
+    token: string;
+    requestType?: "message" | "initialize" | "elicitResult" | undefined;
+}, {
+    session: {
+        id: string;
+        payload: {
+            uuid: string;
+            nodeId: string;
+            authSig: string;
+            iat: number;
+            protocol: "sse" | "legacy-sse" | "streamable-http" | "stateful-http" | "stateless-http";
+        };
+    };
+    token: string;
+    requestType?: "message" | "initialize" | "elicitResult" | undefined;
+}>;
+declare const name: "handle:legacy-sse";
+declare global {
+    interface ExtendFlows {
+        'handle:legacy-sse': FlowRunOptions<HandleSseFlow, typeof plan, typeof httpInputSchema, typeof httpOutputSchema, typeof stateSchema>;
+    }
+}
+export default class HandleSseFlow extends FlowBase<typeof name> {
+    paseInput(): Promise<void>;
+    router(): Promise<void>;
+    onInitialize(): Promise<void>;
+    onElicitResult(): Promise<void>;
+    onMessage(): Promise<void>;
+}
+export {};

package/src/transport/flows/handle.sse.flow.js

@@ -0,0 +1,129 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.stateSchema = exports.plan = void 0;
+const tslib_1 = require("tslib");
+const common_1 = require("../../common");
+const zod_1 = require("zod");
+const session_id_utils_1 = require("../../auth/session/utils/session-id.utils");
+exports.plan = {
+    pre: [
+        'parseInput',
+        'router',
+    ],
+    execute: [
+        'onInitialize',
+        'onMessage',
+        'onElicitResult',
+    ],
+    post: [],
+    finalize: [
+        'cleanup',
+    ],
+};
+exports.stateSchema = zod_1.z.object({
+    token: zod_1.z.string(),
+    session: common_1.sessionIdSchema,
+    requestType: zod_1.z.enum(['initialize', 'message', 'elicitResult']).optional(),
+});
+const name = 'handle:legacy-sse';
+const { Stage } = (0, common_1.FlowHooksOf)(name);
+let HandleSseFlow = class HandleSseFlow extends common_1.FlowBase {
+    async paseInput() {
+        const { request } = this.rawInput;
+        let { token, session } = request[common_1.ServerRequestTokens.auth];
+        if (!session) {
+            session = (0, session_id_utils_1.createSessionId)('legacy-sse', token);
+            request[common_1.ServerRequestTokens.auth].session = session;
+        }
+        this.state.set(exports.stateSchema.parse({ token, session }));
+    }
+    async router() {
+        const { request } = this.rawInput;
+        const scope = this.scope;
+        const requestPath = (0, common_1.normalizeEntryPrefix)(request.path);
+        const prefix = (0, common_1.normalizeEntryPrefix)(scope.entryPath);
+        const scopePath = (0, common_1.normalizeScopeBase)(scope.routeBase);
+        if (requestPath === `${prefix}${scopePath}`) {
+            this.state.set('requestType', 'initialize');
+        }
+        else if (requestPath === `${prefix}${scopePath}/message`) {
+            this.state.set('requestType', 'message');
+        }
+    }
+    async onInitialize() {
+        const transportService = this.scope.transportService;
+        const { request, response } = this.rawInput;
+        const { token, session } = this.state.required;
+        const transport = await transportService.createTransporter('sse', token, session.id, response);
+        await transport.initialize(request, response);
+        this.handled();
+    }
+    async onElicitResult() {
+        // const transport = await transportService.getTransporter('sse', token, session.id);
+        // if (!transport) {
+        //   this.respond(httpRespond.rpcError('session not initialized'));
+        //   return;
+        // }
+        // await transport.handleRequest(request, response);
+        this.fail(new Error('Not implemented'));
+    }
+    async onMessage() {
+        const transportService = this.scope.transportService;
+        const { request, response } = this.rawInput;
+        const { token, session } = this.state.required;
+        const transport = await transportService.getTransporter('sse', token, session.id);
+        if (!transport) {
+            this.respond(common_1.httpRespond.rpcError('session not initialized'));
+            return;
+        }
+        await transport.handleRequest(request, response);
+        this.handled();
+    }
+};
+tslib_1.__decorate([
+    Stage('parseInput'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], HandleSseFlow.prototype, "paseInput", null);
+tslib_1.__decorate([
+    Stage('router'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], HandleSseFlow.prototype, "router", null);
+tslib_1.__decorate([
+    Stage('onInitialize', {
+        filter: ({ state: { requestType } }) => requestType === 'initialize',
+    }),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], HandleSseFlow.prototype, "onInitialize", null);
+tslib_1.__decorate([
+    Stage('onElicitResult', {
+        filter: ({ state: { requestType } }) => requestType === 'elicitResult',
+    }),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], HandleSseFlow.prototype, "onElicitResult", null);
+tslib_1.__decorate([
+    Stage('onMessage', {
+        filter: ({ state: { requestType } }) => requestType === 'message',
+    }),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], HandleSseFlow.prototype, "onMessage", null);
+HandleSseFlow = tslib_1.__decorate([
+    (0, common_1.Flow)({
+        name,
+        access: 'authorized',
+        inputSchema: common_1.httpInputSchema,
+        outputSchema: common_1.httpOutputSchema,
+        plan: exports.plan,
+    })
+], HandleSseFlow);
+exports.default = HandleSseFlow;
+//# sourceMappingURL=handle.sse.flow.js.map

package/src/transport/flows/handle.sse.flow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handle.sse.flow.js","sourceRoot":"","sources":["../../../../src/transport/flows/handle.sse.flow.ts"],"names":[],"mappings":";;;;AAAA,yCAIsB;AACtB,6BAAsB;AAEtB,gFAA0E;AAE7D,QAAA,IAAI,GAAG;IAClB,GAAG,EAAE;QACH,YAAY;QACZ,QAAQ;KACT;IACD,OAAO,EAAE;QACP,cAAc;QACd,WAAW;QACX,gBAAgB;KACjB;IACD,IAAI,EAAE,EAAE;IACR,QAAQ,EAAE;QACR,SAAS;KACV;CACkC,CAAC;AAGzB,QAAA,WAAW,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;IACjB,OAAO,EAAE,wBAAe;IACxB,WAAW,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,YAAY,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,EAAE;CAC1E,CAAC,CAAC;AAEH,MAAM,IAAI,GAAG,mBAA4B,CAAC;AAC1C,MAAM,EAAC,KAAK,EAAC,GAAG,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC;AAsBnB,IAAM,aAAa,GAAnB,MAAM,aAAc,SAAQ,iBAAqB;IAGxD,AAAN,KAAK,CAAC,SAAS;QACb,MAAM,EAAC,OAAO,EAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QAEhC,IAAI,EAAC,KAAK,EAAE,OAAO,EAAC,GAAG,OAAO,CAAC,4BAAmB,CAAC,IAAI,CAAkB,CAAC;QAE1E,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAA,kCAAe,EAAC,YAAY,EAAE,KAAK,CAAC,CAAC;YAC/C,OAAO,CAAC,4BAAmB,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,OAAO,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,mBAAW,CAAC,KAAK,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,CAAC,CAAC,CAAC;IACtD,CAAC;IAGK,AAAN,KAAK,CAAC,MAAM;QACV,MAAM,EAAC,OAAO,EAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAc,CAAC;QAClC,MAAM,WAAW,GAAG,IAAA,6BAAoB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,IAAA,6BAAoB,EAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,IAAA,2BAAkB,EAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,WAAW,KAAK,GAAG,MAAM,GAAG,SAAS,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAC9C,CAAC;aAAM,IAAI,WAAW,KAAK,GAAG,MAAM,GAAG,SAAS,UAAU,EAAE,CAAC;YAC3D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAKK,AAAN,KAAK,CAAC,YAAY;QAChB,MAAM,gBAAgB,GAAI,IAAI,CAAC,KAAe,CAAC,gBAAgB,CAAC;QAEhE,MAAM,EAAC,OAAO,EAAE,QAAQ,EAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC1C,MAAM,EAAC,KAAK,EAAE,OAAO,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,gBAAgB,CAAC,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC/F,MAAM,SAAS,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAKK,AAAN,KAAK,CAAC,cAAc;QAClB,qFAAqF;QACrF,oBAAoB;QACpB,mEAAmE;QACnE,YAAY;QACZ,IAAI;QACJ,oDAAoD;QACpD,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAC1C,CAAC;IAKK,AAAN,KAAK,CAAC,SAAS;QACb,MAAM,gBAAgB,GAAI,IAAI,CAAC,KAAe,CAAC,gBAAgB,CAAC;QAEhE,MAAM,EAAC,OAAO,EAAE,QAAQ,EAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC1C,MAAM,EAAC,KAAK,EAAE,OAAO,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,gBAAgB,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QAClF,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,OAAO,CAAC,oBAAW,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;QACD,MAAM,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;CAGF,CAAA;AAtEO;IADL,KAAK,CAAC,YAAY,CAAC;;;;8CAWnB;AAGK;IADL,KAAK,CAAC,QAAQ,CAAC;;;;2CAYf;AAKK;IAHL,KAAK,CAAC,cAAc,EAAE;QACrB,MAAM,EAAE,CAAC,EAAC,KAAK,EAAE,EAAC,WAAW,EAAC,EAAC,EAAE,EAAE,CAAC,WAAW,KAAK,YAAY;KACjE,CAAC;;;;iDASD;AAKK;IAHL,KAAK,CAAC,gBAAgB,EAAE;QACvB,MAAM,EAAE,CAAC,EAAC,KAAK,EAAE,EAAC,WAAW,EAAC,EAAC,EAAE,EAAE,CAAC,WAAW,KAAK,cAAc;KACnE,CAAC;;;;mDASD;AAKK;IAHL,KAAK,CAAC,WAAW,EAAE;QAClB,MAAM,EAAE,CAAC,EAAC,KAAK,EAAE,EAAC,WAAW,EAAC,EAAC,EAAE,EAAE,CAAC,WAAW,KAAK,SAAS;KAC9D,CAAC;;;;8CAaD;AAtEkB,aAAa;IAPjC,IAAA,aAAI,EAAC;QACJ,IAAI;QACJ,MAAM,EAAE,YAAY;QACpB,WAAW,EAAE,wBAAe;QAC5B,YAAY,EAAE,yBAAgB;QAC9B,IAAI,EAAJ,YAAI;KACL,CAAC;GACmB,aAAa,CAyEjC;kBAzEoB,aAAa","sourcesContent":["import {\n  Flow, httpInputSchema, FlowRunOptions, httpOutputSchema, FlowPlan,\n  FlowBase, FlowHooksOf,\n  sessionIdSchema, httpRespond, ServerRequestTokens, Authorization, normalizeEntryPrefix, normalizeScopeBase,\n} from '../../common';\nimport {z} from 'zod';\nimport {Scope} from '../../scope';\nimport {createSessionId} from '../../auth/session/utils/session-id.utils';\n\nexport const plan = {\n  pre: [\n    'parseInput',\n    'router',\n  ],\n  execute: [\n    'onInitialize',\n    'onMessage',\n    'onElicitResult',\n  ],\n  post: [],\n  finalize: [\n    'cleanup',\n  ],\n} as const satisfies FlowPlan<string>;\n\n\nexport const stateSchema = z.object({\n  token: z.string(),\n  session: sessionIdSchema,\n  requestType: z.enum(['initialize', 'message', 'elicitResult']).optional(),\n});\n\nconst name = 'handle:legacy-sse' as const;\nconst {Stage} = FlowHooksOf(name);\n\n\ndeclare global {\n  interface ExtendFlows {\n    'handle:legacy-sse': FlowRunOptions<\n      HandleSseFlow,\n      typeof plan,\n      typeof httpInputSchema,\n      typeof httpOutputSchema,\n      typeof stateSchema\n    >;\n  }\n}\n\n@Flow({\n  name,\n  access: 'authorized',\n  inputSchema: httpInputSchema,\n  outputSchema: httpOutputSchema,\n  plan,\n})\nexport default class HandleSseFlow extends FlowBase<typeof name> {\n\n  @Stage('parseInput')\n  async paseInput() {\n    const {request} = this.rawInput;\n\n    let {token, session} = request[ServerRequestTokens.auth] as Authorization;\n\n    if (!session) {\n      session = createSessionId('legacy-sse', token);\n      request[ServerRequestTokens.auth].session = session;\n    }\n    this.state.set(stateSchema.parse({token, session}));\n  }\n\n  @Stage('router')\n  async router() {\n    const {request} = this.rawInput;\n    const scope = this.scope as Scope;\n    const requestPath = normalizeEntryPrefix(request.path);\n    const prefix = normalizeEntryPrefix(scope.entryPath);\n    const scopePath = normalizeScopeBase(scope.routeBase);\n    if (requestPath === `${prefix}${scopePath}`) {\n      this.state.set('requestType', 'initialize');\n    } else if (requestPath === `${prefix}${scopePath}/message`) {\n      this.state.set('requestType', 'message');\n    }\n  }\n\n  @Stage('onInitialize', {\n    filter: ({state: {requestType}}) => requestType === 'initialize',\n  })\n  async onInitialize() {\n    const transportService = (this.scope as Scope).transportService;\n\n    const {request, response} = this.rawInput;\n    const {token, session} = this.state.required;\n    const transport = await transportService.createTransporter('sse', token, session.id, response);\n    await transport.initialize(request, response);\n    this.handled();\n  }\n\n  @Stage('onElicitResult', {\n    filter: ({state: {requestType}}) => requestType === 'elicitResult',\n  })\n  async onElicitResult() {\n    // const transport = await transportService.getTransporter('sse', token, session.id);\n    // if (!transport) {\n    //   this.respond(httpRespond.rpcError('session not initialized'));\n    //   return;\n    // }\n    // await transport.handleRequest(request, response);\n    this.fail(new Error('Not implemented'));\n  }\n\n  @Stage('onMessage', {\n    filter: ({state: {requestType}}) => requestType === 'message',\n  })\n  async onMessage() {\n    const transportService = (this.scope as Scope).transportService;\n\n    const {request, response} = this.rawInput;\n    const {token, session} = this.state.required;\n    const transport = await transportService.getTransporter('sse', token, session.id);\n    if (!transport) {\n      this.respond(httpRespond.rpcError('session not initialized'));\n      return;\n    }\n    await transport.handleRequest(request, response);\n    this.handled();\n  }\n\n\n}\n"]}