@sma1lboy/rove 0.9.71 → 0.9.73

package/dist/cli/pty-host-node.mjs

@@ -186,6 +186,7 @@ async function nodePtyDriver(spawn) {
 }
 
 // ../kobe-daemon/src/daemon/pty-server.ts
+import { readFileSync as readFileSync4 } from "node:fs";
 import { mkdir as mkdir2, unlink as unlink2, writeFile } from "node:fs/promises";
 import { createServer } from "node:net";
 import { dirname as dirname3 } from "node:path";
@@ -1285,6 +1286,14 @@ class PtyHost {
 
 // ../kobe-daemon/src/daemon/pty-server.ts
 var DEFAULT_IDLE_EXIT_MS = 60000;
+var DEFAULT_ORPHAN_CHECK_MS = 30000;
+function resolveMaxLifetimeMs() {
+  const raw = process.env.KOBE_PTY_MAX_LIFETIME_MS;
+  if (raw === undefined)
+    return null;
+  const n = Number(raw);
+  return Number.isFinite(n) && n > 0 ? n : null;
+}
 function resolveIdleExitMs() {
   const raw = process.env.KOBE_PTY_IDLE_EXIT_MS;
   if (raw === undefined)
@@ -1297,6 +1306,9 @@ async function startPtyHostServer(options = {}) {
   const pidPath = options.pidPath ?? defaultPtyHostPidPath();
   const freezeDir = options.freezeDir ?? defaultPtyFreezeDir();
   const idleExitMs = options.idleExitMs || resolveIdleExitMs();
+  const orphanCheckMs = options.orphanCheckMs || DEFAULT_ORPHAN_CHECK_MS;
+  const maxLifetimeMs = options.maxLifetimeMs ?? resolveMaxLifetimeMs();
+  const bootedAtMs = Date.now();
   const log = options.log ?? (() => {});
   const clients = new Set;
   let stopping = false;
@@ -1318,6 +1330,31 @@ async function startPtyHostServer(options = {}) {
       stop();
     }, idleExitMs);
   };
+  const orphaned = () => {
+    if (maxLifetimeMs !== null && Date.now() - bootedAtMs >= maxLifetimeMs) {
+      return `exceeded its ${maxLifetimeMs}ms lifetime ceiling`;
+    }
+    let owner;
+    try {
+      owner = readFileSync4(pidPath, "utf8").trim();
+    } catch {
+      return `pidfile ${pidPath} is gone`;
+    }
+    const pid = Number.parseInt(owner, 10);
+    if (pid === process.pid)
+      return null;
+    return Number.isInteger(pid) ? `pidfile ${pidPath} now names pid ${pid}` : `pidfile ${pidPath} is unreadable`;
+  };
+  const orphanTimer = setInterval(() => {
+    if (stopping)
+      return;
+    const reason = orphaned();
+    if (!reason)
+      return;
+    log("orphan", `${reason} — exiting rather than outliving my owner`);
+    stop();
+  }, orphanCheckMs);
+  orphanTimer.unref?.();
   const ptys = new PtyHost({
     onSessionStart: cancelIdle,
     onSessionEnd: () => {
@@ -1371,6 +1408,7 @@ async function startPtyHostServer(options = {}) {
         return;
       stopping = true;
       cancelIdle();
+      clearInterval(orphanTimer);
       await ptys.shutdown();
       if (wipeFreezeOnStop)
         clearFrozenSessions(freezeDir);