kibi-runtime 2.13.1 → 2.13.3

package/dist/engine-daemon.js

@@ -3,7 +3,7 @@ import { createRequire } from "node:module";
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 
 // ../cli/src/engine-daemon.ts
-import { writeFileSync as writeFileSync3 } from "node:fs";
+import { writeFileSync as writeFileSync4 } from "node:fs";
 import path9 from "node:path";
 import { pathToFileURL } from "node:url";
 
@@ -24,7 +24,7 @@ import {
   rmSync as rmSync2,
   statSync as statSync4,
   unlinkSync,
-  writeFileSync as writeFileSync2
+  writeFileSync as writeFileSync3
 } from "node:fs";
 import * as net from "node:net";
 import * as os from "node:os";
@@ -102,7 +102,7 @@ function manifestVersion(manifestPath, expectedName) {
 }
 function cliVersion() {
   if (true)
-    return "2.16.0";
+    return "2.18.0";
   return manifestVersion(path.join(moduleDirectory, "..", "package.json"), "kibi-cli") ?? "unknown";
 }
 function coreVersion() {
@@ -1576,7 +1576,7 @@ writeln('${INTERACTIVE_QUERY_FRAME_END}').
       child.kill(signal);
     } catch {}
   }
-  async terminateProcessTree(child) {
+  async terminateProcessTree(child, graceMs = 0) {
     let closed = false;
     const exited = new Promise((resolve) => {
       const markExited = () => {
@@ -1586,6 +1586,19 @@ writeln('${INTERACTIVE_QUERY_FRAME_END}').
       child.once("close", markExited);
       child.once("error", markExited);
     });
+    if (graceMs > 0 && child.exitCode === null && child.signalCode === null) {
+      let graceTimer;
+      await Promise.race([
+        exited,
+        new Promise((resolve) => {
+          graceTimer = setTimeout(resolve, graceMs);
+        })
+      ]);
+      if (graceTimer)
+        clearTimeout(graceTimer);
+      if (closed)
+        return;
+    }
     if (child.exitCode === null)
       this.signalProcessTree(child, "SIGTERM");
     await Promise.race([
@@ -1707,7 +1720,7 @@ writeln('${INTERACTIVE_QUERY_FRAME_END}').
     }
     return new PrologProcessTerminatedError(this.terminationReason ?? "process exited");
   }
-  async terminate() {
+  async terminate(options = {}) {
     if (this.interactiveStarted) {
       this.terminationReason ??= "terminated";
     }
@@ -1727,7 +1740,7 @@ writeln('${INTERACTIVE_QUERY_FRAME_END}').
       this.terminationPromise = (async () => {
         if (current) {
           current.stdin?.end();
-          await this.terminateProcessTree(current);
+          await this.terminateProcessTree(current, options.graceMs ?? 0);
         }
         await Promise.all(oneShots.map((child) => this.terminateProcessTree(child)));
       })();
@@ -1741,7 +1754,7 @@ writeln('${INTERACTIVE_QUERY_FRAME_END}').
 }
 
 // ../cli/src/prolog/store-lock.ts
-import { existsSync as existsSync4, rmSync } from "node:fs";
+import { existsSync as existsSync4, rmSync, writeFileSync } from "node:fs";
 import { readFileSync as readFileSync3 } from "node:fs";
 import { dirname, join } from "node:path";
 var cachedBootId;
@@ -1797,6 +1810,14 @@ function breakStoreLock(rdfDirectory) {
     }
   }
 }
+var STORE_LOCK_TAKEOVER_FILE = ".kibi-lock-takeover.json";
+var STORE_LOCK_TAKEOVER_VISIBLE_MS = 24 * 60 * 60 * 1000;
+function recordStoreLockTakeover(storePath, brokenPid, reason, at = new Date) {
+  try {
+    writeFileSync(join(storePath, STORE_LOCK_TAKEOVER_FILE), `${JSON.stringify({ brokenPid, reason, at: at.toISOString() })}
+`, { mode: 384 });
+  } catch {}
+}
 function decideStoreLockTakeover(owner) {
   const state = classifyStoreLockHolder(owner);
   switch (state) {
@@ -1833,6 +1854,7 @@ Store lock holder: ${holderSummary}. Close that session, or run 'kibi engine sto
     };
   }
   const lockDirectory = storeLocked.lockDirectory !== "" ? storeLocked.lockDirectory : join(kbPath, "rdf");
+  recordStoreLockTakeover(kbPath, storeLocked.owner?.pid ?? null, decision.reason);
   breakStoreLock(lockDirectory);
   const retried = await prolog.query(`kb_attach('${escapeAtom(kbPath)}')`);
   if (retried.success) {
@@ -1856,7 +1878,7 @@ import {
   readFileSync as readFileSync4,
   readdirSync,
   statSync as statSync3,
-  writeFileSync
+  writeFileSync as writeFileSync2
 } from "node:fs";
 import path5 from "node:path";
 var BRANCH_STORE_MANIFEST_VERSION = 1;
@@ -1918,7 +1940,7 @@ function ensureBranchStoreManifest(workspaceRoot, branch) {
     if (entries.length > 0) {
       throw new Error(`Branch store at ${storePath} has compiled data but no valid identity manifest; refusing to adopt it.`);
     }
-    writeFileSync(manifestPath, `${JSON.stringify(expectedBranchStoreManifest(branch), null, 2)}
+    writeFileSync2(manifestPath, `${JSON.stringify(expectedBranchStoreManifest(branch), null, 2)}
 `, { mode: 384 });
   }
   return storePath;
@@ -2005,6 +2027,8 @@ var READ_LIMITED_METHODS = new Set([
 ]);
 var ENGINE_PROTOCOL_VERSION = 1;
 var ENGINE_IDLE_TIMEOUT_MS = 10 * 60 * 1000;
+var ENGINE_IDLE_CHECK_INTERVAL_MS = 30 * 1000;
+var ENGINE_PROLOG_EXIT_GRACE_MS = 1000;
 function enginePackageVersions() {
   return kibiPackageVersions();
 }
@@ -2174,7 +2198,7 @@ function runtimeDirectory() {
       try {
         mkdirSync2(candidate, { recursive: true, mode: 448 });
         const probe = path8.join(candidate, `.kibi-write-${process.pid}`);
-        writeFileSync2(probe, "", { mode: 384 });
+        writeFileSync3(probe, "", { mode: 384 });
         unlinkSync(probe);
         return candidate;
       } catch {}
@@ -2214,7 +2238,7 @@ function runtimeDirectory() {
       if (Buffer.byteLength(path8.join(canonical, socketName), "utf8") > maximumBytes)
         continue;
       const probe = path8.join(canonical, `.kibi-write-${randomUUID()}`);
-      writeFileSync2(probe, "", { mode: 384, flag: "wx" });
+      writeFileSync3(probe, "", { mode: 384, flag: "wx" });
       unlinkSync(probe);
       return canonical;
     } catch {}
@@ -2340,7 +2364,7 @@ async function ensureJournaledBranchStoreAsync(branchPath) {
         recoverStaleJournalLock(branchPath);
         archiveLegacyFiles(branchPath);
         if (!existsSync6(sentinelPath)) {
-          writeFileSync2(sentinelPath, `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
+          writeFileSync3(sentinelPath, `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
 `, { mode: 384 });
         }
         return;
@@ -2359,7 +2383,7 @@ async function ensureJournaledBranchStoreAsync(branchPath) {
       recoverStaleJournalLock(branchPath);
       archiveLegacyFiles(branchPath);
       if (!existsSync6(sentinelPath)) {
-        writeFileSync2(sentinelPath, `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
+        writeFileSync3(sentinelPath, `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
 `, { mode: 384 });
       }
       return;
@@ -2369,13 +2393,13 @@ async function ensureJournaledBranchStoreAsync(branchPath) {
   const legacyAudit = path8.join(branchPath, "audit.log");
   if (!existsSync6(legacyRdf) && !existsSync6(legacyAudit)) {
     mkdirSync2(path8.join(branchPath, "rdf"), { recursive: true });
-    writeFileSync2(markerPath, `{"format":"kibi.rdf-journal.v1","schemaVersion":1}
+    writeFileSync3(markerPath, `{"format":"kibi.rdf-journal.v1","schemaVersion":1}
 `, { mode: 384 });
-    writeFileSync2(path8.join(branchPath, "CURRENT"), `generation-1:0
+    writeFileSync3(path8.join(branchPath, "CURRENT"), `generation-1:0
 `, {
       mode: 384
     });
-    writeFileSync2(path8.join(branchPath, "kb.rdf"), `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
+    writeFileSync3(path8.join(branchPath, "kb.rdf"), `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
 `, { mode: 384 });
     return;
   }
@@ -2424,7 +2448,7 @@ async function ensureJournaledBranchStoreAsync(branchPath) {
     renameSync(source, target);
   }
   archiveLegacyFiles(branchPath);
-  writeFileSync2(path8.join(branchPath, "kb.rdf"), `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
+  writeFileSync3(path8.join(branchPath, "kb.rdf"), `KIBI_STORAGE_FORMAT=kibi.rdf-journal.v1
 `, { mode: 384 });
   fsyncJournaledBranchStore(branchPath);
   rmSync2(staging, { recursive: true, force: true });
@@ -2617,11 +2641,12 @@ async function runEngineDaemon(requestedOptions) {
       rmSync2(options.socketPath, { force: true });
     } catch {}
   }
-  writeFileSync2(enginePidPath(options.workspaceRoot, options.branch), `${process.pid}
+  writeFileSync3(enginePidPath(options.workspaceRoot, options.branch), `${process.pid}
 `, { mode: 384 });
   const server = net.createServer();
   let activeClients = 0;
-  let idleTimer = null;
+  let lastActivityAt = Date.now();
+  let idleWatch = null;
   let queue = Promise.resolve();
   let idleCompactionQueued = false;
   const queryCache = new Map;
@@ -2698,12 +2723,15 @@ async function runEngineDaemon(requestedOptions) {
         idleCompactionQueued = false;
       });
     }
-    if (idleTimer)
-      clearTimeout(idleTimer);
-    idleTimer = setTimeout(() => {
-      if (activeClients === 0)
+    lastActivityAt = Date.now();
+    if (idleWatch)
+      return;
+    const idleMs = engineIdleTimeoutMs();
+    idleWatch = setInterval(() => {
+      if (!shuttingDown && activeClients === 0 && Date.now() - lastActivityAt >= idleMs)
         shutdown();
-    }, engineIdleTimeoutMs());
+    }, Math.min(idleMs, ENGINE_IDLE_CHECK_INTERVAL_MS));
+    idleWatch.unref();
   };
   const handle = async (request, cancelledRequests) => {
     if (request.protocolVersion !== ENGINE_PROTOCOL_VERSION) {
@@ -2958,20 +2986,22 @@ async function runEngineDaemon(requestedOptions) {
     shuttingDown = true;
     process.off("SIGTERM", requestSignalShutdown);
     process.off("SIGINT", requestSignalShutdown);
-    if (idleTimer)
-      clearTimeout(idleTimer);
+    if (idleWatch)
+      clearInterval(idleWatch);
     clearInterval(workspaceWatchdog);
     server.close();
     await queue.catch(() => {
       return;
     });
     const saved = session.needsRecycle() ? null : await prolog.query("kb_save").catch(() => null);
+    let detached = false;
     if (saved?.success) {
       try {
         fsyncJournaledBranchStore(branchPath);
       } catch {}
+      detached = await prolog.query("kb_detach").then((result) => result.success).catch(() => false);
     }
-    await prolog.terminate().catch(() => {
+    await prolog.terminate({ graceMs: detached ? 0 : ENGINE_PROLOG_EXIT_GRACE_MS }).catch(() => {
       return;
     });
     for (const client of clients)
@@ -3008,6 +3038,7 @@ async function runEngineDaemon(requestedOptions) {
         const bytes = typeof chunk === "string" ? Buffer.from(chunk) : chunk;
         buffer = parseFrames(Buffer.concat([buffer, bytes]), (value) => {
           const request = value;
+          lastActivityAt = Date.now();
           if (shuttingDown) {
             writeSocketFrame(socket, {
               id: request.id,
@@ -3114,7 +3145,7 @@ async function runEngineDaemonCli(args = process.argv.slice(2)) {
     const socketPath = socketIndex >= 0 ? args[socketIndex + 1] : undefined;
     if (socketPath) {
       try {
-        writeFileSync3(`${socketPath}.error`, `${message}
+        writeFileSync4(`${socketPath}.error`, `${message}
 `, { mode: 384 });
       } catch {}
     }