@eaperezc/mcpgen 0.1.0 → 0.1.1

package/dist/index.d.ts

@@ -1,8 +1,7 @@
-import { I as InlineToolDefinition, c as InlineResourceDefinition, d as InlinePromptDefinition, T as TransportConfig, e as AuthConfig, L as LogLevel, S as ServerHooks, M as McpServer, f as Logger, R as RequestContextData, A as AuthContext, g as Transport, H as HttpTransportConfig } from './types-B9yzEar_.js';
-export { k as ApiKeyConfig, l as ApiKeyValidationResult, b as BasePrompt, a as BaseResource, B as BaseTool, U as ContextKey, Y as CorsConfig, K as LogEntry, Q as LogTransport, N as LoggerConfig, O as OAuthConfig, F as PromptArgumentMetadata, C as PromptDefinition, E as PromptHandler, z as PromptMessage, G as PromptMetadata, P as PromptRegistry, D as PromptResult, y as PromptRole, s as ResourceContent, t as ResourceDefinition, v as ResourceHandler, w as ResourceMetadata, r as ResourceRegistry, u as ResourceResult, x as ResourceTemplate, j as ServerCapabilities, h as ServerConfig, i as ServerInfo, X as SseTransportConfig, W as StdioTransportConfig, Z as TlsConfig, n as ToolDefinition, p as ToolHandler, q as ToolMetadata, m as ToolRegistry, o as ToolResult, V as TransportType, J as logger } from './types-B9yzEar_.js';
+import { I as InlineToolDefinition, c as InlineResourceDefinition, d as InlinePromptDefinition, T as TransportConfig, e as AuthConfig, L as LogLevel, S as ServerHooks, M as McpServer, f as Logger, R as RequestContextData, A as AuthContext, g as Transport, H as HttpTransportConfig } from './types-Bu1xVhfb.js';
+export { k as ApiKeyConfig, l as ApiKeyValidationResult, b as BasePrompt, a as BaseResource, B as BaseTool, U as ContextKey, Y as CorsConfig, K as LogEntry, Q as LogTransport, N as LoggerConfig, O as OAuthConfig, F as PromptArgumentMetadata, C as PromptDefinition, E as PromptHandler, z as PromptMessage, G as PromptMetadata, P as PromptRegistry, D as PromptResult, y as PromptRole, s as ResourceContent, t as ResourceDefinition, v as ResourceHandler, w as ResourceMetadata, r as ResourceRegistry, u as ResourceResult, x as ResourceTemplate, j as ServerCapabilities, h as ServerConfig, i as ServerInfo, X as SseTransportConfig, W as StdioTransportConfig, Z as TlsConfig, n as ToolDefinition, p as ToolHandler, q as ToolMetadata, m as ToolRegistry, o as ToolResult, V as TransportType, J as logger } from './types-Bu1xVhfb.js';
 import { z } from 'zod';
 export { z } from 'zod';
-import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 
 /**
@@ -241,33 +240,22 @@ declare class McpError extends Error {
 }
 
 /**
- * HTTP Transport for MCP servers
- *
- * Implements the MCP Streamable HTTP transport specification,
- * supporting both SSE streaming and direct HTTP responses.
- *
- * @example
- * ```typescript
- * const transport = new HttpTransport({
- *   type: 'http',
- *   port: 3000,
- *   cors: { origin: ['https://myapp.com'] },
- * });
- *
- * await transport.start(mcpServer);
- * ```
+ * Factory function to create a configured MCP Server instance
  */
+type ServerFactory = () => Server;
 declare class HttpTransport implements Transport {
     private config;
     private logger;
     private httpServer;
-    private mcpTransport;
     private running;
+    private serverFactory;
+    private sessions;
     constructor(config: HttpTransportConfig, logger?: Logger);
     /**
      * Start the HTTP transport
+     * @param serverFactory - Factory function to create new Server instances per session
      */
-    start(mcpServer?: Server): Promise<void>;
+    start(serverFactory?: ServerFactory): Promise<void>;
     /**
      * Stop the HTTP transport
      */
@@ -277,9 +265,13 @@ declare class HttpTransport implements Transport {
      */
     isRunning(): boolean;
     /**
-     * Get the underlying MCP transport (for advanced use cases)
+     * Get all active sessions (for debugging/monitoring)
+     */
+    getActiveSessions(): string[];
+    /**
+     * Get or create a session for the given session ID
      */
-    getMcpTransport(): StreamableHTTPServerTransport | null;
+    private getOrCreateSession;
     /**
      * Handle incoming HTTP request
      */

package/dist/index.js

@@ -873,8 +873,9 @@ var HttpTransport = class {
   config;
   logger;
   httpServer = null;
-  mcpTransport = null;
   running = false;
+  serverFactory = null;
+  sessions = /* @__PURE__ */ new Map();
   constructor(config, logger2) {
     this.config = {
       ...config,
@@ -886,17 +887,13 @@ var HttpTransport = class {
   }
   /**
    * Start the HTTP transport
+   * @param serverFactory - Factory function to create new Server instances per session
    */
-  async start(mcpServer) {
+  async start(serverFactory) {
     if (this.running) {
       throw new Error("Transport is already running");
     }
-    this.mcpTransport = new StreamableHTTPServerTransport({
-      sessionIdGenerator: () => randomUUID()
-    });
-    if (mcpServer) {
-      await mcpServer.connect(this.mcpTransport);
-    }
+    this.serverFactory = serverFactory ?? null;
     this.httpServer = createServer((req, res) => {
       this.handleRequest(req, res);
     });
@@ -924,10 +921,15 @@ var HttpTransport = class {
     if (!this.running) {
       return;
     }
-    if (this.mcpTransport) {
-      await this.mcpTransport.close();
-      this.mcpTransport = null;
+    for (const [sessionId, session] of this.sessions) {
+      try {
+        await session.transport.close();
+        await session.server.close();
+      } catch (error) {
+        this.logger.error(`Error closing session ${sessionId}`, error instanceof Error ? error : void 0);
+      }
     }
+    this.sessions.clear();
     if (this.httpServer) {
       return new Promise((resolve, reject) => {
         this.httpServer.close((error) => {
@@ -951,10 +953,34 @@ var HttpTransport = class {
     return this.running;
   }
   /**
-   * Get the underlying MCP transport (for advanced use cases)
+   * Get all active sessions (for debugging/monitoring)
+   */
+  getActiveSessions() {
+    return Array.from(this.sessions.keys());
+  }
+  /**
+   * Get or create a session for the given session ID
    */
-  getMcpTransport() {
-    return this.mcpTransport;
+  async getOrCreateSession(sessionId) {
+    let session = this.sessions.get(sessionId);
+    if (!session) {
+      if (!this.serverFactory) {
+        throw new Error("No server factory configured");
+      }
+      const transport = new StreamableHTTPServerTransport({
+        sessionIdGenerator: () => sessionId
+      });
+      const server = this.serverFactory();
+      await server.connect(transport);
+      session = {
+        transport,
+        server,
+        createdAt: Date.now()
+      };
+      this.sessions.set(sessionId, session);
+      this.logger.debug("Created new session", { sessionId });
+    }
+    return session;
   }
   /**
    * Handle incoming HTTP request
@@ -975,9 +1001,9 @@ var HttpTransport = class {
       return;
     }
     if (path === this.config.basePath || path === `${this.config.basePath}/`) {
-      if (!this.mcpTransport) {
+      if (!this.serverFactory) {
         res.writeHead(503, { "Content-Type": "application/json" });
-        res.end(JSON.stringify({ error: "Transport not initialized" }));
+        res.end(JSON.stringify({ error: "Server factory not configured" }));
         return;
       }
       try {
@@ -985,7 +1011,12 @@ var HttpTransport = class {
         if (req.method === "POST") {
           body = await this.parseBody(req);
         }
-        await this.mcpTransport.handleRequest(req, res, body);
+        let sessionId = req.headers["mcp-session-id"];
+        if (!sessionId) {
+          sessionId = randomUUID();
+        }
+        const session = await this.getOrCreateSession(sessionId);
+        await session.transport.handleRequest(req, res, body);
       } catch (error) {
         this.logger.error("Error handling MCP request", error instanceof Error ? error : void 0);
         if (!res.headersSent) {
@@ -1102,10 +1133,17 @@ var McpServer = class {
     };
   }
   /**
-   * Set up MCP request handlers
+   * Set up MCP request handlers on a server instance
    */
   setupHandlers() {
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+    this.setupHandlersOnServer(this.server);
+  }
+  /**
+   * Configure handlers on a given Server instance
+   * This is used both for the main server (stdio) and for per-session servers (http)
+   */
+  setupHandlersOnServer(server) {
+    server.setRequestHandler(ListToolsRequestSchema, async () => {
       return runInRequestContextAsync(async () => {
         const tools = this.toolRegistry.getAll();
         return {
@@ -1117,7 +1155,7 @@ var McpServer = class {
         };
       });
     });
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
       return runInRequestContextAsync(async () => {
         const { name, arguments: args } = request.params;
         const tool = this.toolRegistry.get(name);
@@ -1136,7 +1174,7 @@ var McpServer = class {
         };
       });
     });
-    this.server.setRequestHandler(ListResourcesRequestSchema, async () => {
+    server.setRequestHandler(ListResourcesRequestSchema, async () => {
       return runInRequestContextAsync(async () => {
         const resources = this.resourceRegistry.getAll();
         return {
@@ -1149,7 +1187,7 @@ var McpServer = class {
         };
       });
     });
-    this.server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
+    server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
       return runInRequestContextAsync(async () => {
         const { uri } = request.params;
         const resource = this.resourceRegistry.get(uri);
@@ -1165,7 +1203,7 @@ var McpServer = class {
         };
       });
     });
-    this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
+    server.setRequestHandler(ListPromptsRequestSchema, async () => {
       return runInRequestContextAsync(async () => {
         const prompts = this.promptRegistry.getAll();
         return {
@@ -1184,7 +1222,7 @@ var McpServer = class {
         };
       });
     });
-    this.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+    server.setRequestHandler(GetPromptRequestSchema, async (request) => {
       return runInRequestContextAsync(async () => {
         const { name, arguments: args } = request.params;
         const prompt = this.promptRegistry.get(name);
@@ -1202,6 +1240,24 @@ var McpServer = class {
       });
     });
   }
+  /**
+   * Create a server factory for HTTP transport (creates per-session servers)
+   */
+  createServerFactory() {
+    return () => {
+      const server = new Server(
+        {
+          name: this.config.name,
+          version: this.config.version
+        },
+        {
+          capabilities: this.getCapabilities()
+        }
+      );
+      this.setupHandlersOnServer(server);
+      return server;
+    };
+  }
   tool(toolOrName, definition) {
     if (typeof toolOrName === "string" && definition) {
       this.toolRegistry.registerInline(toolOrName, definition);
@@ -1245,14 +1301,6 @@ var McpServer = class {
     }
     const transportConfig = this.config.transport ?? "stdio";
     const transportType = typeof transportConfig === "string" ? transportConfig : transportConfig.type;
-    this.logger.info("Starting MCP server", {
-      name: this.config.name,
-      version: this.config.version,
-      transport: transportType,
-      tools: this.toolRegistry.size,
-      resources: this.resourceRegistry.size,
-      prompts: this.promptRegistry.size
-    });
     try {
       await this.hooks.onBeforeStart?.();
       if (transportType === "stdio") {
@@ -1261,7 +1309,7 @@ var McpServer = class {
       } else if (transportType === "http") {
         const httpConfig = transportConfig;
         this.httpTransport = new HttpTransport(httpConfig, this.logger);
-        await this.httpTransport.start(this.server);
+        await this.httpTransport.start(this.createServerFactory());
       } else if (transportType === "sse") {
         throw new McpError({
           code: "NOT_IMPLEMENTED" /* NOT_IMPLEMENTED */,
@@ -1270,8 +1318,12 @@ var McpServer = class {
       }
       this.running = true;
       await this.hooks.onAfterStart?.();
-      this.logger.info("MCP server started successfully", {
+      this.logger.info("MCP server started", {
+        version: this.config.version,
         transport: transportType,
+        tools: this.toolRegistry.size,
+        resources: this.resourceRegistry.size,
+        prompts: this.promptRegistry.size,
         ...transportType === "http" && {
           port: transportConfig.port,
           host: transportConfig.host ?? "0.0.0.0"