billion-context 0.1.140 → 0.1.141

package/dist/index.js

@@ -43467,6 +43467,10 @@ var require_lib = __commonJS({
   }
 });
 
+// src/index.ts
+import { realpathSync as realpathSync2 } from "fs";
+import { fileURLToPath as fileURLToPath8 } from "url";
+
 // node_modules/acp-kernel/dist/chunk-6RNVHY67.js
 import { createRequire } from "module";
 var require2 = createRequire(import.meta.url);
@@ -73243,6 +73247,10 @@ async function startServer(opts) {
   const log2 = (level, msg) => logMsg(opts, level, msg);
   const instanceId = randomUUID4();
   const instanceStartedAt = Date.now();
+  const parentWatchPid = Number.parseInt(process.env.BILI_PARENT_PID ?? "", 10);
+  const initialWatcherPid = Number.isInteger(parentWatchPid) && parentWatchPid > 0 && parentWatchPid !== process.pid ? parentWatchPid : null;
+  const proxyWatchers = /* @__PURE__ */ new Set();
+  if (initialWatcherPid !== null) proxyWatchers.add(initialWatcherPid);
   await initSessions();
   loadConversations();
   log2("info", `[persist] ${getStore().enabled ? "enabled" : "disabled"}`);
@@ -73263,7 +73271,7 @@ async function startServer(opts) {
   const server = http.createServer(async (req, res) => {
     armRequestWatchdog(req, res, log2);
     try {
-      await handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt);
+      await handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt, proxyWatchers, initialWatcherPid);
     } catch (err2) {
       const msg = String(err2);
       const e = err2;
@@ -73419,10 +73427,30 @@ Content-Length: ${Buffer.byteLength(body)}\r
   if (process.platform === "win32") {
     process.on("SIGBREAK", () => shutdown("SIGBREAK"));
   }
-  const parentPid = Number.parseInt(process.env.BILI_PARENT_PID ?? "", 10);
-  if (Number.isInteger(parentPid) && parentPid > 0 && parentPid !== process.pid) {
+  if (initialWatcherPid !== null) {
+    let idleSince = null;
+    let lastDead = [];
     const watcher = setInterval(() => {
-      if (!isPidAlive(parentPid)) shutdown(`parent-gone (pid ${parentPid})`);
+      const dead = [];
+      for (const pid of proxyWatchers) {
+        if (!isPidAlive(pid)) {
+          proxyWatchers.delete(pid);
+          dead.push(pid);
+        }
+      }
+      if (dead.length > 0) lastDead = dead;
+      if (proxyWatchers.size > 0) {
+        idleSince = null;
+        return;
+      }
+      if (idleSince === null) {
+        idleSince = Date.now();
+        return;
+      }
+      if (Date.now() - idleSince >= WATCHER_IDLE_GRACE_MS) {
+        const reason = lastDead.length === 1 && lastDead[0] === initialWatcherPid ? `parent-gone (pid ${lastDead[0]})` : `watchers-gone (pids: ${lastDead.join(", ")})`;
+        shutdown(reason);
+      }
     }, 2e3);
     watcher.unref?.();
   }
@@ -73458,7 +73486,8 @@ function adminTrustedHosts(bindHost, port) {
   return set;
 }
 var headroomFallbackLogged = /* @__PURE__ */ new Set();
-async function handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt) {
+var WATCHER_IDLE_GRACE_MS = 5e3;
+async function handle(req, res, opts, core, config, log2, instanceId, instanceStartedAt, proxyWatchers, initialWatcherPid) {
   const selfProbePath = selfAdminProbePath(req.url ?? "", req.socket.localPort);
   if (selfProbePath !== void 0) req.url = selfProbePath;
   const isAdminPath = req.url === "/__bili/" || req.url?.startsWith("/__bili/") || req.url === "/__acp/" || req.url?.startsWith("/__acp/");
@@ -73587,6 +73616,29 @@ async function handle(req, res, opts, core, config, log2, instanceId, instanceSt
     }
     return handlePluginStatus(conversationId2, res, { core, config, log: log2 }, params.get("fallback") === "latest");
   }
+  if (req.method === "POST" && req.url === "/__bili/watcher") {
+    try {
+      const body = await readBody(req);
+      const parsed2 = JSON.parse(body.toString("utf8"));
+      const pid = parsed2.pid;
+      if (typeof pid !== "number" || !Number.isInteger(pid) || pid <= 1 || pid === process.pid) {
+        res.writeHead(400, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: false, error: "expected { pid: <integer> }" }));
+      } else if (initialWatcherPid === null) {
+        res.writeHead(409, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: false, error: "watchdog not armed (no parent pid) \u2014 this proxy does not take watchers" }));
+      } else {
+        proxyWatchers.add(pid);
+        res.writeHead(200, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: true, watchers: proxyWatchers.size }));
+      }
+      return;
+    } catch (err2) {
+      res.writeHead(err2 instanceof BodyTooLargeError ? 413 : 400, { "content-type": "application/json" });
+      res.end(JSON.stringify({ ok: false, error: String(err2) }));
+      return;
+    }
+  }
   if (req.method === "POST" && req.url === "/__bili/plugin/tool") {
     try {
       const body = await readBody(req);
@@ -81073,10 +81125,26 @@ async function main() {
 }
 
 // src/index.ts
-main().catch((err2) => {
-  console.error("bili: failed to start:", err2);
-  process.exit(1);
-});
+function invokedAsScript() {
+  const argv1 = process.argv[1];
+  if (!argv1) return false;
+  let self2;
+  let invoked;
+  try {
+    self2 = realpathSync2(fileURLToPath8(import.meta.url));
+    invoked = realpathSync2(argv1);
+  } catch {
+    return false;
+  }
+  if (process.platform === "win32") return self2.toLowerCase() === invoked.toLowerCase();
+  return self2 === invoked;
+}
+if (invokedAsScript()) {
+  main().catch((err2) => {
+    console.error("bili: failed to start:", err2);
+    process.exit(1);
+  });
+}
 /*! Bundled license information:
 
 undici/lib/web/fetch/body.js: