cross-agent-teams-mcp 0.5.10 → 0.5.11

package/dist/cli.js

@@ -3712,12 +3712,14 @@ function mountMcp(app, db, fanout, channelWakeFanout, opts = {}) {
       sessionIdGenerator: () => randomUUID5(),
       onsessioninitialized: (sid) => {
         sessionIdForCaller = sid;
+        const now2 = Date.now();
         sessions.set(sid, {
           transport,
           server,
           sessionId: sid,
           agentIdHolder,
-          createdAt: Date.now(),
+          createdAt: now2,
+          lastActivityAt: now2,
           clientInfo: void 0,
           originInfo: { origin: "local", remote_addr: null }
         });
@@ -3771,12 +3773,14 @@ function mountMcp(app, db, fanout, channelWakeFanout, opts = {}) {
       registerSvc
     );
     server.connect(transport);
+    const now = Date.now();
     return {
       transport,
       server,
       sessionId: "",
       agentIdHolder,
-      createdAt: Date.now(),
+      createdAt: now,
+      lastActivityAt: now,
       originInfo: { origin: "local", remote_addr: null }
     };
   }
@@ -3820,6 +3824,7 @@ function mountMcp(app, db, fanout, channelWakeFanout, opts = {}) {
     }
     if (session) {
       session.originInfo = originInfo;
+      session.lastActivityAt = Date.now();
     }
     if (body?.method === "initialize") {
       const params = body.params;
@@ -3842,6 +3847,7 @@ function mountMcp(app, db, fanout, channelWakeFanout, opts = {}) {
     const sid = req.headers["mcp-session-id"];
     const session = sid ? sessions.get(sid) : void 0;
     if (!session) return reply.code(400).send({ error: "unknown_session" });
+    session.lastActivityAt = Date.now();
     await session.transport.handleRequest(req.raw, reply.raw);
     return reply;
   });
@@ -3849,14 +3855,15 @@ function mountMcp(app, db, fanout, channelWakeFanout, opts = {}) {
     const sid = req.headers["mcp-session-id"];
     const session = sid ? sessions.get(sid) : void 0;
     if (!session) return reply.code(400).send({ error: "unknown_session" });
+    session.lastActivityAt = Date.now();
     await session.transport.handleRequest(req.raw, reply.raw);
     return reply;
   });
-  function reapOrphanSessions(now, graceMs = 6e4) {
+  function reapOrphanSessions(now, graceMs = 18e5) {
     for (const session of sessions.values()) {
       if (session.agentIdHolder.current !== void 0) continue;
-      const ageMs = now - session.createdAt;
-      if (ageMs < graceMs) continue;
+      const idleMs = now - session.lastActivityAt;
+      if (idleMs < graceMs) continue;
       try {
         void session.transport.close();
       } catch {
@@ -4040,6 +4047,7 @@ function isLoopbackHost(host) {
 // src/daemon/server.ts
 var DEFAULT_KEEP_ALIVE_TIMEOUT_MS = 12e4;
 var DEFAULT_ORPHAN_GC_INTERVAL_MS = 6e4;
+var DEFAULT_ORPHAN_GC_IDLE_MS = 18e5;
 function parsePositiveInt(raw, fallback) {
   const n = Number(raw);
   return Number.isInteger(n) && n > 0 ? n : fallback;
@@ -4073,9 +4081,10 @@ async function buildServer(opts) {
   }, cleanupIntervalMs);
   if (typeof interval.unref === "function") interval.unref();
   const orphanGcIntervalMs = opts.orphanGcIntervalMs ?? parsePositiveInt(process.env.ORPHAN_GC_INTERVAL_MS, DEFAULT_ORPHAN_GC_INTERVAL_MS);
+  const orphanGcIdleMs = opts.orphanGcIdleMs ?? parsePositiveInt(process.env.ORPHAN_GC_IDLE_MS, DEFAULT_ORPHAN_GC_IDLE_MS);
   const orphanGcInterval = setInterval(() => {
     try {
-      mcp.reapOrphanSessions(Date.now());
+      mcp.reapOrphanSessions(Date.now(), orphanGcIdleMs);
     } catch {
     }
   }, orphanGcIntervalMs);
@@ -4130,17 +4139,69 @@ function releasePidFile(path) {
 }
 
 // src/daemon/shutdown.ts
-function wireShutdown(app, pidPath) {
-  const handler = async (_signal) => {
+var DEFAULT_GRACE_MS = 5e3;
+function resolveGraceMs(explicit) {
+  if (typeof explicit === "number" && Number.isFinite(explicit)) {
+    return explicit < 0 ? 0 : Math.floor(explicit);
+  }
+  const raw = process.env.XATS_SHUTDOWN_GRACE_MS;
+  if (raw === void 0 || raw === "") return DEFAULT_GRACE_MS;
+  const n = Number(raw);
+  if (!Number.isFinite(n)) return DEFAULT_GRACE_MS;
+  return n < 0 ? 0 : Math.floor(n);
+}
+function wireShutdown(app, pidPath, opts = {}) {
+  const graceMs = resolveGraceMs(opts.graceMs);
+  const exit = opts.exit ?? ((code) => {
+    process.exit(code);
+  });
+  let shuttingDown = false;
+  const handler = (_signal) => {
+    if (shuttingDown) {
+      releasePidFile(pidPath);
+      exit(0);
+      return;
+    }
+    shuttingDown = true;
+    void runDrain(app, pidPath, graceMs, exit);
+  };
+  process.on("SIGTERM", handler);
+  process.on("SIGINT", handler);
+}
+async function runDrain(app, pidPath, graceMs, exit) {
+  if (graceMs <= 0) {
+    try {
+      app.server.closeAllConnections();
+    } catch {
+    }
     try {
       await app.close();
     } catch {
     }
     releasePidFile(pidPath);
-    process.exit(0);
-  };
-  process.once("SIGTERM", handler);
-  process.once("SIGINT", handler);
+    exit(0);
+    return;
+  }
+  let timer;
+  const deadline = new Promise((resolve) => {
+    timer = setTimeout(() => resolve("timeout"), graceMs);
+    if (typeof timer.unref === "function") timer.unref();
+  });
+  const closed = app.close().then(() => "closed").catch(() => "closed");
+  const winner = await Promise.race([closed, deadline]);
+  if (winner === "timeout") {
+    try {
+      app.server.closeAllConnections();
+    } catch {
+    }
+    try {
+      await app.close();
+    } catch {
+    }
+  }
+  if (timer) clearTimeout(timer);
+  releasePidFile(pidPath);
+  exit(0);
 }
 
 // src/daemon/port.ts