@cortexkit/opencode-magic-context 0.49.19 → 0.49.20

package/src/shared/worker-shutdown.ts

@@ -0,0 +1,160 @@
+import type { MessagePort, Worker } from "node:worker_threads";
+
+/**
+ * Stopping a worker thread that still has a SQLite connection open.
+ *
+ * On Bun 1.4.4, `worker.terminate()` closes the worker's open `bun:sqlite`
+ * connections while it tears the thread down, and for that whole teardown
+ * (measured from a few ms up to about 460 ms) every other connection to the same
+ * file gets SQLITE_BUSY when it asks for the writer lock. Reads keep working. A
+ * caller that terminates a worker and then writes at once, such as
+ * `clearSession()` cancelling a session's auto-search backfill, then fails its
+ * own write. Bun 1.4.3 and earlier did not do this, and a worker that closes
+ * its connection itself never causes it on any version.
+ *
+ * So an owner asks the worker to stop: it posts {@link WORKER_CLOSE_MESSAGE},
+ * the worker closes its connections and exits (see {@link exitOnWorkerClose}),
+ * and `terminate()` runs only if the worker has not exited by a deadline. The
+ * deadline is chosen per worker from the longest synchronous SQLite work it may
+ * be in the middle of, because a worker handles the message only between
+ * statements. Bun 1.3 queues parent messages until a worker's entry module has
+ * finished evaluating, so a close posted while the worker is still loading is
+ * handled as soon as it has loaded.
+ */
+export const WORKER_CLOSE_MESSAGE = { kind: "close" } as const;
+
+export function isWorkerCloseMessage(message: unknown): boolean {
+    return (
+        typeof message === "object" &&
+        message !== null &&
+        (message as { kind?: unknown }).kind === WORKER_CLOSE_MESSAGE.kind
+    );
+}
+
+/**
+ * Worker side: when the owner asks this worker to stop, close its connections
+ * and end the thread. `process.exit()` in a worker thread ends only that
+ * thread, on Node and on Bun. Other `message` listeners in the worker see the
+ * same message and must ignore its kind.
+ *
+ * On Bun 1.3.14 a worker whose parent port still has a `message` listener does
+ * not exit after `port.close()`, so this listener must never outlive the port.
+ * A worker that installs it closes its port only through the returned
+ * `closePort`, which removes the listener first (on the next timer turn). The port's own `close()`
+ * cannot be wrapped instead: on Bun 1.3.14 it is a read-only, non-configurable
+ * property. `worker-shutdown-fence.test.ts` rejects any direct port close in a
+ * file that calls this function.
+ */
+export function exitOnWorkerClose(
+    port: WorkerClosePort | null,
+    closeConnections: () => void,
+): { closePort(): void } {
+    const listener = (message: unknown) => {
+        if (!isWorkerCloseMessage(message)) return;
+        try {
+            closeConnections();
+        } finally {
+            // Exit even if a close threw: the thread is ending and a connection
+            // that could not close is released by the thread's teardown instead.
+            process.exit(0);
+        }
+    };
+    if (!port) return { closePort() {} };
+    port.on("message", listener);
+    let closed = false;
+    return {
+        closePort() {
+            if (closed) return;
+            closed = true;
+            // Close on a later timer turn, not synchronously: on Bun 1.3.14 a
+            // port closed while an entry module with a top-level await is still
+            // evaluating does not end the thread. Every current caller closes
+            // its port at the end of the module body or later, so the timer runs
+            // after evaluation has finished.
+            setTimeout(() => {
+                port.off("message", listener);
+                port.close();
+            });
+        },
+    };
+}
+
+/** The parts of a worker's parent port that {@link exitOnWorkerClose} uses. */
+export type WorkerClosePort = Pick<MessagePort, "on" | "off" | "close">;
+
+export interface WorkerShutdown {
+    /**
+     * Ask the worker to close its connections and exit; terminate it if it has
+     * not exited after the deadline. Calling it again returns the same promise.
+     * Resolves once the worker has exited.
+     */
+    close(): Promise<void>;
+}
+
+const outcomes = new Map<string, { graceful: number; forced: number }>();
+
+/** How each site's workers ended after a close, by themselves or terminated. A snapshot. */
+export function getWorkerShutdownStats(): ReadonlyMap<
+    string,
+    { readonly graceful: number; readonly forced: number }
+> {
+    return new Map([...outcomes].map(([site, counts]) => [site, { ...counts }]));
+}
+
+/**
+ * Owner side. Call right after creating the worker, so its exit is observed
+ * even when it exits before anyone asks it to close.
+ *
+ * @param site names the worker in {@link getWorkerShutdownStats}.
+ * @param deadlineMs how long the worker may take to close and exit before it
+ *   is terminated.
+ */
+export function createWorkerShutdown(
+    worker: Worker,
+    site: string,
+    deadlineMs: number,
+): WorkerShutdown {
+    let exited = false;
+    let forced = false;
+    let closing: Promise<void> | undefined;
+    const exit = new Promise<void>((resolve) => {
+        worker.once("exit", () => {
+            exited = true;
+            resolve();
+        });
+    });
+    const record = () => {
+        const counts = outcomes.get(site) ?? { graceful: 0, forced: 0 };
+        if (forced) counts.forced += 1;
+        else counts.graceful += 1;
+        outcomes.set(site, counts);
+    };
+    const terminate = () => {
+        forced = true;
+        void worker.terminate().catch(() => {});
+    };
+    return {
+        close() {
+            if (closing) return closing;
+            if (exited) {
+                closing = Promise.resolve();
+                return closing;
+            }
+            try {
+                worker.postMessage(WORKER_CLOSE_MESSAGE);
+            } catch {
+                terminate();
+            }
+            const fallback = setTimeout(() => {
+                if (!exited) terminate();
+            }, deadlineMs);
+            // A host that is exiting must not wait for this timer.
+            fallback.unref?.();
+            closing = exit.then(() => {
+                clearTimeout(fallback);
+                record();
+            });
+            return closing;
+        },
+    };
+}

package/dist/embedding-backfill-fmz50tsy.js

@@ -1,24 +0,0 @@
-import {
-  ensureMemoryEmbeddings2
-} from "./index-24fezq97.js";
-import"./index-ze7mvk54.js";
-import"./index-4dhv7mqz.js";
-import"./index-qt1qqasc.js";
-import"./index-bjcnntmz.js";
-import"./index-wgypt3cs.js";
-import"./index-rr2bb45z.js";
-import"./index-b5vqckbw.js";
-import"./index-6x57ar45.js";
-import"./index-2vxtcb95.js";
-import"./index-a6mc0v4d.js";
-import"./index-s734jcsr.js";
-import"./index-v0p37hm2.js";
-import"./index-k29z4aqh.js";
-import"./index-c5pytzne.js";
-import"./index-beht37jy.js";
-import"./index-j75v2476.js";
-import"./index-g2q4htwp.js";
-import"./index-pavzsdc6.js";
-export {
-  ensureMemoryEmbeddings2 as ensureMemoryEmbeddings
-};

package/dist/recomp-orchestrator-999jj4bc.js

@@ -1,61 +0,0 @@
-import {
-  isRecompFailure2,
-  isRecompSkip2,
-  extractRecompReason2,
-  setRecompStarting2,
-  setRecompNote2,
-  setRecompTerminal2,
-  buildRecompDeps2,
-  runManagedRecomp2
-} from "./index-c1p88z1y.js";
-import"./index-rpxdjkm0.js";
-import"./index-4s3t0nd0.js";
-import"./index-3h4dp9wb.js";
-import"./index-se4m2g6w.js";
-import"./index-sd3egckz.js";
-import"./index-24fezq97.js";
-import"./index-ze7mvk54.js";
-import"./index-n37sebck.js";
-import"./index-4dhv7mqz.js";
-import"./index-v10f1rjv.js";
-import"./index-s74zqahq.js";
-import"./index-qt1qqasc.js";
-import"./index-9zcsy3hh.js";
-import"./index-cf74k5ak.js";
-import"./index-1zx87sdm.js";
-import"./index-kq82mtg0.js";
-import"./index-x8a7jmkh.js";
-import"./index-y9f76s3f.js";
-import"./index-bjcnntmz.js";
-import"./index-wgypt3cs.js";
-import"./index-rr2bb45z.js";
-import"./index-b5vqckbw.js";
-import"./index-6x57ar45.js";
-import"./index-jrr131e0.js";
-import"./index-fb3zxn06.js";
-import"./index-63k48rv9.js";
-import"./index-2vxtcb95.js";
-import"./index-a6mc0v4d.js";
-import"./index-fr0wbhdn.js";
-import"./index-nj7yykkx.js";
-import"./index-ke416mdc.js";
-import"./index-bp5083x2.js";
-import"./index-58e3y90n.js";
-import"./index-s734jcsr.js";
-import"./index-v0p37hm2.js";
-import"./index-k29z4aqh.js";
-import"./index-c5pytzne.js";
-import"./index-beht37jy.js";
-import"./index-j75v2476.js";
-import"./index-g2q4htwp.js";
-import"./index-pavzsdc6.js";
-export {
-  buildRecompDeps2 as buildRecompDeps,
-  extractRecompReason2 as extractRecompReason,
-  isRecompFailure2 as isRecompFailure,
-  isRecompSkip2 as isRecompSkip,
-  runManagedRecomp2 as runManagedRecomp,
-  setRecompNote2 as setRecompNote,
-  setRecompStarting2 as setRecompStarting,
-  setRecompTerminal2 as setRecompTerminal
-};

package/dist/v2/embedding-backfill-7c7yr69y.js

@@ -1,23 +0,0 @@
-import {
-  ensureMemoryEmbeddings2
-} from "./server-zzkj2fch.js";
-import"./server-8by8qfec.js";
-import"./server-4dhv7mqz.js";
-import"./server-qt1qqasc.js";
-import"./server-h1v0pyxb.js";
-import"./server-271qxsf5.js";
-import"./server-hy3d3vsx.js";
-import"./server-46neh5fa.js";
-import"./server-6tmj0s7j.js";
-import"./server-rmygf5pn.js";
-import"./server-1wy7b9de.js";
-import"./server-xj136kt8.js";
-import"./server-51779zs2.js";
-import"./server-6pmqg9m4.js";
-import"./server-tgnbmwah.js";
-import"./server-beht37jy.js";
-import"./server-f2xf5gcb.js";
-import"./server-pavzsdc6.js";
-export {
-  ensureMemoryEmbeddings2 as ensureMemoryEmbeddings
-};

package/dist/v2/recomp-orchestrator-f6m9jr7p.js

@@ -1,60 +0,0 @@
-import {
-  isRecompFailure2,
-  isRecompSkip2,
-  extractRecompReason2,
-  setRecompStarting2,
-  setRecompNote2,
-  setRecompTerminal2,
-  buildRecompDeps2,
-  runManagedRecomp2
-} from "./server-6vdzn1td.js";
-import"./server-d04vrx0j.js";
-import"./server-hm38q7r0.js";
-import"./server-2ar62dyx.js";
-import"./server-zzkj2fch.js";
-import"./server-n37sebck.js";
-import"./server-8by8qfec.js";
-import"./server-4dhv7mqz.js";
-import"./server-34kwfsr4.js";
-import"./server-a9xqj88f.js";
-import"./server-qt1qqasc.js";
-import"./server-9hx7g5jp.js";
-import"./server-27gq4mfz.js";
-import"./server-1zx87sdm.js";
-import"./server-n1prf1g6.js";
-import"./server-h0410h0v.js";
-import"./server-nj659894.js";
-import"./server-tbe1femw.js";
-import"./server-x8a7jmkh.js";
-import"./server-h1v0pyxb.js";
-import"./server-271qxsf5.js";
-import"./server-hy3d3vsx.js";
-import"./server-46neh5fa.js";
-import"./server-6tmj0s7j.js";
-import"./server-bmymkx73.js";
-import"./server-fb3zxn06.js";
-import"./server-xq7dfjke.js";
-import"./server-fr0wbhdn.js";
-import"./server-rmygf5pn.js";
-import"./server-1wy7b9de.js";
-import"./server-da07v4xp.js";
-import"./server-g3qma5c0.js";
-import"./server-pwv4ezb0.js";
-import"./server-2ra6pwf3.js";
-import"./server-xj136kt8.js";
-import"./server-51779zs2.js";
-import"./server-6pmqg9m4.js";
-import"./server-tgnbmwah.js";
-import"./server-beht37jy.js";
-import"./server-f2xf5gcb.js";
-import"./server-pavzsdc6.js";
-export {
-  buildRecompDeps2 as buildRecompDeps,
-  extractRecompReason2 as extractRecompReason,
-  isRecompFailure2 as isRecompFailure,
-  isRecompSkip2 as isRecompSkip,
-  runManagedRecomp2 as runManagedRecomp,
-  setRecompNote2 as setRecompNote,
-  setRecompStarting2 as setRecompStarting,
-  setRecompTerminal2 as setRecompTerminal
-};

package/dist/wrapup-orchestrator-whhbjmaz.js

@@ -1,48 +0,0 @@
-import {
-  runManagedWrapup2
-} from "./index-4911rn0b.js";
-import"./index-c1p88z1y.js";
-import"./index-rpxdjkm0.js";
-import"./index-4s3t0nd0.js";
-import"./index-3h4dp9wb.js";
-import"./index-se4m2g6w.js";
-import"./index-sd3egckz.js";
-import"./index-24fezq97.js";
-import"./index-ze7mvk54.js";
-import"./index-n37sebck.js";
-import"./index-4dhv7mqz.js";
-import"./index-v10f1rjv.js";
-import"./index-s74zqahq.js";
-import"./index-qt1qqasc.js";
-import"./index-9zcsy3hh.js";
-import"./index-cf74k5ak.js";
-import"./index-1zx87sdm.js";
-import"./index-kq82mtg0.js";
-import"./index-x8a7jmkh.js";
-import"./index-y9f76s3f.js";
-import"./index-bjcnntmz.js";
-import"./index-wgypt3cs.js";
-import"./index-rr2bb45z.js";
-import"./index-b5vqckbw.js";
-import"./index-6x57ar45.js";
-import"./index-jrr131e0.js";
-import"./index-fb3zxn06.js";
-import"./index-63k48rv9.js";
-import"./index-2vxtcb95.js";
-import"./index-a6mc0v4d.js";
-import"./index-fr0wbhdn.js";
-import"./index-nj7yykkx.js";
-import"./index-ke416mdc.js";
-import"./index-bp5083x2.js";
-import"./index-58e3y90n.js";
-import"./index-s734jcsr.js";
-import"./index-v0p37hm2.js";
-import"./index-k29z4aqh.js";
-import"./index-c5pytzne.js";
-import"./index-beht37jy.js";
-import"./index-j75v2476.js";
-import"./index-g2q4htwp.js";
-import"./index-pavzsdc6.js";
-export {
-  runManagedWrapup2 as runManagedWrapup
-};