@arabold/docs-mcp-server 1.31.1 → 1.32.0

package/dist/index.js

@@ -10056,6 +10056,8 @@ async function registerMcpService(server, docService, pipeline, readOnly = false
   const mcpServer = createMcpServerInstance(mcpTools, readOnly);
   const authMiddleware = authManager ? createAuthMiddleware(authManager) : null;
   const sseTransports = {};
+  const heartbeatIntervals = {};
+  const HEARTBEAT_INTERVAL_MS = 3e4;
   server.route({
     method: "GET",
     url: "/sse",
@@ -10067,12 +10069,31 @@ async function registerMcpService(server, docService, pipeline, readOnly = false
         if (telemetry.isEnabled()) {
           logger.info(`🔗 MCP client connected: ${transport.sessionId}`);
         }
-        reply.raw.on("close", () => {
+        const heartbeatInterval = setInterval(() => {
+          try {
+            reply.raw.write(": heartbeat\n\n");
+          } catch {
+            clearInterval(heartbeatInterval);
+            delete heartbeatIntervals[transport.sessionId];
+          }
+        }, HEARTBEAT_INTERVAL_MS);
+        heartbeatIntervals[transport.sessionId] = heartbeatInterval;
+        const cleanupConnection = () => {
+          const interval = heartbeatIntervals[transport.sessionId];
+          if (interval) {
+            clearInterval(interval);
+            delete heartbeatIntervals[transport.sessionId];
+          }
           delete sseTransports[transport.sessionId];
           transport.close();
           if (telemetry.isEnabled()) {
             logger.info(`🔗 MCP client disconnected: ${transport.sessionId}`);
           }
+        };
+        reply.raw.on("close", cleanupConnection);
+        reply.raw.on("error", (error) => {
+          logger.debug(`SSE connection error: ${error}`);
+          cleanupConnection();
         });
         await mcpServer.connect(transport);
       } catch (error) {
@@ -10114,10 +10135,15 @@ async function registerMcpService(server, docService, pipeline, readOnly = false
         const requestTransport = new StreamableHTTPServerTransport({
           sessionIdGenerator: void 0
         });
-        reply.raw.on("close", () => {
+        const cleanupRequest = () => {
           logger.debug("Streamable HTTP request closed");
           requestTransport.close();
           requestServer.close();
+        };
+        reply.raw.on("close", cleanupRequest);
+        reply.raw.on("error", (error) => {
+          logger.debug(`Streamable HTTP connection error: ${error}`);
+          cleanupRequest();
         });
         await requestServer.connect(requestTransport);
         await requestTransport.handleRequest(request.raw, reply.raw, request.body);
@@ -10130,10 +10156,17 @@ async function registerMcpService(server, docService, pipeline, readOnly = false
     }
   });
   mcpServer._sseTransports = sseTransports;
+  mcpServer._heartbeatIntervals = heartbeatIntervals;
   return mcpServer;
 }
 async function cleanupMcpService(mcpServer) {
   try {
+    const heartbeatIntervals = mcpServer._heartbeatIntervals;
+    if (heartbeatIntervals) {
+      for (const interval of Object.values(heartbeatIntervals)) {
+        clearInterval(interval);
+      }
+    }
     const sseTransports = mcpServer._sseTransports;
     if (sseTransports) {
       for (const transport of Object.values(sseTransports)) {
@@ -10737,7 +10770,7 @@ const Layout = ({
   children,
   eventClientConfig
 }) => {
-  const versionString = version || "1.31.1";
+  const versionString = version || "1.32.0";
   const versionInitializer = `versionUpdate({ currentVersion: ${`'${versionString}'`} })`;
   return /* @__PURE__ */ jsxs("html", { lang: "en", children: [
     /* @__PURE__ */ jsxs("head", { children: [
@@ -13049,7 +13082,7 @@ class AppServer {
       try {
         if (telemetry.isEnabled()) {
           telemetry.setGlobalContext({
-            appVersion: "1.31.1",
+            appVersion: "1.32.0",
             appPlatform: process.platform,
             appNodeVersion: process.version,
             appServicesEnabled: this.getActiveServicesList(),
@@ -16536,7 +16569,7 @@ function createCliProgram() {
   const commandStartTimes = /* @__PURE__ */ new Map();
   let globalEventBus = null;
   let globalTelemetryService = null;
-  program.name("docs-mcp-server").description("Unified CLI, MCP Server, and Web Interface for Docs MCP Server.").version("1.31.1").addOption(
+  program.name("docs-mcp-server").description("Unified CLI, MCP Server, and Web Interface for Docs MCP Server.").version("1.32.0").addOption(
     new Option("--verbose", "Enable verbose (debug) logging").conflicts("silent")
   ).addOption(new Option("--silent", "Disable all logging except errors")).addOption(
     new Option("--telemetry", "Enable telemetry collection").env("DOCS_MCP_TELEMETRY").argParser((value) => {
@@ -16570,7 +16603,7 @@ function createCliProgram() {
     if (shouldEnableTelemetry()) {
       if (telemetry.isEnabled()) {
         telemetry.setGlobalContext({
-          appVersion: "1.31.1",
+          appVersion: "1.32.0",
           appPlatform: process.platform,
           appNodeVersion: process.version,
           appInterface: "cli",