@leanmcp/core 0.4.0 → 0.4.1

package/dist/index.js

@@ -985,6 +985,62 @@ async function createHTTPServer(serverInput, options) {
       if (sessionId && transports[sessionId]) {
         transport = transports[sessionId];
         logger.debug(`Reusing session: ${sessionId}`);
+      } else if (sessionId && isInitializeRequest(req.body)) {
+        logger.info(`Initialize request with session ${sessionId} - checking for session restoration...`);
+        if (httpOptions.sessionStore) {
+          const exists = await httpOptions.sessionStore.sessionExists(sessionId);
+          if (exists) {
+            logger.info(`Restoring session: ${sessionId}`);
+            transport = new StreamableHTTPServerTransport2({
+              sessionIdGenerator: /* @__PURE__ */ __name(() => sessionId, "sessionIdGenerator"),
+              onsessioninitialized: /* @__PURE__ */ __name((sid) => {
+                transports[sid] = transport;
+                logger.info(`Session restored: ${sid}`);
+              }, "onsessioninitialized")
+            });
+            transport.onclose = async () => {
+              if (transport.sessionId) {
+                delete transports[transport.sessionId];
+                logger.debug(`Session cleaned up: ${transport.sessionId}`);
+                if (httpOptions.sessionStore) {
+                  await httpOptions.sessionStore.deleteSession(transport.sessionId);
+                }
+              }
+            };
+            const freshServer = await serverFactory();
+            if (freshServer && typeof freshServer.waitForInit === "function") {
+              await freshServer.waitForInit();
+            }
+            await freshServer.connect(transport);
+          } else {
+            logger.info(`Session ${sessionId} not found in store, creating new session`);
+          }
+        }
+        if (!transport) {
+          transport = new StreamableHTTPServerTransport2({
+            sessionIdGenerator: /* @__PURE__ */ __name(() => (0, import_node_crypto.randomUUID)(), "sessionIdGenerator"),
+            onsessioninitialized: /* @__PURE__ */ __name(async (newSessionId) => {
+              transports[newSessionId] = transport;
+              logger.info(`Session initialized: ${newSessionId}`);
+              if (httpOptions.sessionStore) {
+                await httpOptions.sessionStore.createSession(newSessionId);
+              }
+            }, "onsessioninitialized")
+          });
+          transport.onclose = async () => {
+            if (transport.sessionId) {
+              delete transports[transport.sessionId];
+              logger.debug(`Session cleaned up: ${transport.sessionId}`);
+              if (httpOptions.sessionStore) {
+                await httpOptions.sessionStore.deleteSession(transport.sessionId);
+              }
+            }
+          };
+          if (!mcpServer) {
+            throw new Error("MCP server not initialized");
+          }
+          await mcpServer.connect(transport);
+        }
       } else if (sessionId && !isInitializeRequest(req.body)) {
         logger.info(`Transport missing for session ${sessionId}, checking session store...`);
         if (httpOptions.sessionStore) {
@@ -1000,11 +1056,12 @@ async function createHTTPServer(serverInput, options) {
             });
             return;
           }
+          logger.info(`Auto-restoring session: ${sessionId}`);
           transport = new StreamableHTTPServerTransport2({
             sessionIdGenerator: /* @__PURE__ */ __name(() => sessionId, "sessionIdGenerator"),
             onsessioninitialized: /* @__PURE__ */ __name((sid) => {
               transports[sid] = transport;
-              logger.info(`Transport recreated for session: ${sid}`);
+              logger.info(`Session auto-restored: ${sid}`);
             }, "onsessioninitialized")
           });
           transport.onclose = async () => {
@@ -1822,14 +1879,14 @@ var init_index = __esm({
       /**
       * Auto-register all services from the mcp directory
       * Scans the directory recursively and registers all exported classes
-      * 
+      *
       * @param mcpDir - Path to the mcp directory containing service files
       * @param serviceFactories - Optional map of service class names to factory functions for dependency injection
-      * 
+      *
       * @example
       * // Auto-register services with no dependencies
       * await server.autoRegisterServices('./mcp');
-      * 
+      *
       * @example
       * // Auto-register with dependency injection
       * await server.autoRegisterServices('./mcp', {
@@ -1982,7 +2039,7 @@ var init_index = __esm({
       }
       /**
       * Watch UI manifest for changes and reload resources dynamically
-      * 
+      *
       * CRITICAL: Only for stateful mode. In stateless mode, each request
       * creates a fresh server that reads the manifest directly, making
       * watchers both unnecessary and a memory leak source.

package/dist/index.mjs

@@ -683,6 +683,62 @@ async function createHTTPServer(serverInput, options) {
       if (sessionId && transports[sessionId]) {
         transport = transports[sessionId];
         logger.debug(`Reusing session: ${sessionId}`);
+      } else if (sessionId && isInitializeRequest(req.body)) {
+        logger.info(`Initialize request with session ${sessionId} - checking for session restoration...`);
+        if (httpOptions.sessionStore) {
+          const exists = await httpOptions.sessionStore.sessionExists(sessionId);
+          if (exists) {
+            logger.info(`Restoring session: ${sessionId}`);
+            transport = new StreamableHTTPServerTransport2({
+              sessionIdGenerator: /* @__PURE__ */ __name(() => sessionId, "sessionIdGenerator"),
+              onsessioninitialized: /* @__PURE__ */ __name((sid) => {
+                transports[sid] = transport;
+                logger.info(`Session restored: ${sid}`);
+              }, "onsessioninitialized")
+            });
+            transport.onclose = async () => {
+              if (transport.sessionId) {
+                delete transports[transport.sessionId];
+                logger.debug(`Session cleaned up: ${transport.sessionId}`);
+                if (httpOptions.sessionStore) {
+                  await httpOptions.sessionStore.deleteSession(transport.sessionId);
+                }
+              }
+            };
+            const freshServer = await serverFactory();
+            if (freshServer && typeof freshServer.waitForInit === "function") {
+              await freshServer.waitForInit();
+            }
+            await freshServer.connect(transport);
+          } else {
+            logger.info(`Session ${sessionId} not found in store, creating new session`);
+          }
+        }
+        if (!transport) {
+          transport = new StreamableHTTPServerTransport2({
+            sessionIdGenerator: /* @__PURE__ */ __name(() => randomUUID(), "sessionIdGenerator"),
+            onsessioninitialized: /* @__PURE__ */ __name(async (newSessionId) => {
+              transports[newSessionId] = transport;
+              logger.info(`Session initialized: ${newSessionId}`);
+              if (httpOptions.sessionStore) {
+                await httpOptions.sessionStore.createSession(newSessionId);
+              }
+            }, "onsessioninitialized")
+          });
+          transport.onclose = async () => {
+            if (transport.sessionId) {
+              delete transports[transport.sessionId];
+              logger.debug(`Session cleaned up: ${transport.sessionId}`);
+              if (httpOptions.sessionStore) {
+                await httpOptions.sessionStore.deleteSession(transport.sessionId);
+              }
+            }
+          };
+          if (!mcpServer) {
+            throw new Error("MCP server not initialized");
+          }
+          await mcpServer.connect(transport);
+        }
       } else if (sessionId && !isInitializeRequest(req.body)) {
         logger.info(`Transport missing for session ${sessionId}, checking session store...`);
         if (httpOptions.sessionStore) {
@@ -698,11 +754,12 @@ async function createHTTPServer(serverInput, options) {
             });
             return;
           }
+          logger.info(`Auto-restoring session: ${sessionId}`);
           transport = new StreamableHTTPServerTransport2({
             sessionIdGenerator: /* @__PURE__ */ __name(() => sessionId, "sessionIdGenerator"),
             onsessioninitialized: /* @__PURE__ */ __name((sid) => {
               transports[sid] = transport;
-              logger.info(`Transport recreated for session: ${sid}`);
+              logger.info(`Session auto-restored: ${sid}`);
             }, "onsessioninitialized")
           });
           transport.onclose = async () => {
@@ -1424,14 +1481,14 @@ var MCPServer = class {
   /**
   * Auto-register all services from the mcp directory
   * Scans the directory recursively and registers all exported classes
-  * 
+  *
   * @param mcpDir - Path to the mcp directory containing service files
   * @param serviceFactories - Optional map of service class names to factory functions for dependency injection
-  * 
+  *
   * @example
   * // Auto-register services with no dependencies
   * await server.autoRegisterServices('./mcp');
-  * 
+  *
   * @example
   * // Auto-register with dependency injection
   * await server.autoRegisterServices('./mcp', {
@@ -1584,7 +1641,7 @@ var MCPServer = class {
   }
   /**
   * Watch UI manifest for changes and reload resources dynamically
-  * 
+  *
   * CRITICAL: Only for stateful mode. In stateless mode, each request
   * creates a fresh server that reads the manifest directly, making
   * watchers both unnecessary and a memory leak source.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leanmcp/core",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "Core library implementing decorators, reflection, and MCP runtime server",
   "main": "dist/index.js",
   "module": "dist/index.mjs",