token-goat 2.9.29 → 2.9.30

package/dist/token-goat-chunk-COTXZHTS.mjs

@@ -0,0 +1,39 @@
+import { createRequire as __cjsRequire } from 'node:module';
+const require = __cjsRequire(import.meta.url);
+import {
+  DownloadCooldownError,
+  DownloadFailedError,
+  MODEL_DOWNLOAD_HOST,
+  RUNTIME_DOWNLOAD_HOST,
+  activeDownloadCooldown,
+  clearDownloadFailure,
+  describeCause,
+  downloadFailureRecordPath,
+  downloadHoldMs,
+  failedAtOf,
+  isExplicitDownload,
+  lastDownloadFailure,
+  recordDownloadFailure,
+  withExplicitDownload
+} from "./token-goat-chunk-SMVRN2HU.mjs";
+import "./token-goat-chunk-GOLEOXZP.mjs";
+import "./token-goat-chunk-AM5V6GSX.mjs";
+import "./token-goat-chunk-FNFBKLRD.mjs";
+import "./token-goat-chunk-VES7EL7O.mjs";
+import "./token-goat-chunk-A37V4PBF.mjs";
+export {
+  DownloadCooldownError,
+  DownloadFailedError,
+  MODEL_DOWNLOAD_HOST,
+  RUNTIME_DOWNLOAD_HOST,
+  activeDownloadCooldown,
+  clearDownloadFailure,
+  describeCause,
+  downloadFailureRecordPath,
+  downloadHoldMs,
+  failedAtOf,
+  isExplicitDownload,
+  lastDownloadFailure,
+  recordDownloadFailure,
+  withExplicitDownload
+};

package/dist/{token-goat-chunk-RQGGMPCR.mjs → token-goat-chunk-CYCEYAEO.mjs}

@@ -18,7 +18,7 @@ import {
   readServerKey,
   touchMarker,
   writeFrame
-} from "./token-goat-chunk-DPNLUJWB.mjs";
+} from "./token-goat-chunk-LOBGHXPB.mjs";
 import {
   envBool
 } from "./token-goat-chunk-IVE6U7KV.mjs";
@@ -31,13 +31,15 @@ init_define_import_meta_env();
 import { spawn } from "node:child_process";
 import * as fs from "node:fs";
 import * as net from "node:net";
+import * as os from "node:os";
 var HANDSHAKE_TIMEOUT_MS = 150;
 var FIND_BUDGET_MS = 300;
+var HEDGE_MS = 40;
 var RESPONSE_TIMEOUT_MS = 12e4;
 var CONTROL_WAIT_MS = 1e4;
 var SPAWN_RETRY_MS = 3e4;
 var DISABLED_MARKER_TTL_MS = 10 * 6e4;
-function attempt(slot, key, request, handshakeMs, dir) {
+function attempt(slot, key, request, handshakeMs, dir, race) {
   return new Promise((resolve) => {
     const socket = net.connect(endpointFor(slot, dir));
     const nc = nonce();
@@ -51,6 +53,10 @@ function attempt(slot, key, request, handshakeMs, dir) {
       resolve(outcome);
     };
     let timer = setTimeout(() => finish({ kind: "busy" }), handshakeMs);
+    const cancel = () => {
+      if (!dispatched) finish({ kind: "busy" });
+    };
+    race?.onCancel(cancel);
     socket.on("error", (e) => {
       if (dispatched) finish({ kind: "lost" });
       else finish({ kind: e.code === "ENOENT" || e.code === "ECONNREFUSED" ? "absent" : "refused" });
@@ -70,9 +76,10 @@ function attempt(slot, key, request, handshakeMs, dir) {
           const body = JSON.stringify(request);
           const frame = encodeFrame({ t: "req", mac: mac(key, "C", nc, ns, body), body });
           if (!frameFits(frame)) return finish({ kind: "oversize" });
+          if (race !== void 0 && !race.claim(cancel)) return finish({ kind: "busy" });
           dispatched = true;
           clearTimeout(timer);
-          timer = setTimeout(() => finish({ kind: "lost" }), RESPONSE_TIMEOUT_MS);
+          if (request.kind !== "cli") timer = setTimeout(() => finish({ kind: "lost" }), RESPONSE_TIMEOUT_MS);
           socket.write(frame);
           return;
         }
@@ -97,7 +104,9 @@ function startServer(slot) {
   if (disabled || markerAgeMs(`spawn-${slot}`) < SPAWN_RETRY_MS || !fs.existsSync(launcher)) return;
   touchMarker(`spawn-${slot}`);
   try {
-    spawn(process.execPath, [launcher, "hook-server", "run", "--slot", String(slot)], { detached: true, stdio: "ignore", windowsHide: true }).unref();
+    const child = spawn(process.execPath, [launcher, "hook-server", "run", "--slot", String(slot)], { cwd: os.tmpdir(), detached: true, stdio: "ignore", windowsHide: true });
+    child.on("error", () => void 0);
+    child.unref();
   } catch {
   }
 }
@@ -113,17 +122,56 @@ async function callServer(request, opts = {}) {
     return void 0;
   }
   const deadline = Date.now() + FIND_BUDGET_MS;
-  for (let slot = 0; slot < SERVER_SLOTS && Date.now() < deadline; slot++) {
-    const outcome = await attempt(slot, key, request, opts.handshakeMs ?? HANDSHAKE_TIMEOUT_MS);
-    if (outcome.kind === "served") return outcome.reply;
-    if (outcome.kind === "lost") return "lost";
-    if (outcome.kind === "absent") {
-      if (autostart) startServer(slot);
-      return void 0;
-    }
-    if (outcome.kind === "stale" || outcome.kind === "oversize") return void 0;
-  }
-  return void 0;
+  const handshakeMs = opts.handshakeMs ?? HANDSHAKE_TIMEOUT_MS;
+  return new Promise((resolve) => {
+    let claimed = false;
+    let settled = false;
+    let exhausted = false;
+    let next = 0;
+    let pending = 0;
+    let hedge;
+    const cancels = [];
+    const settle = (value) => {
+      if (settled) return;
+      settled = true;
+      clearTimeout(hedge);
+      for (const cancel of cancels) cancel();
+      resolve(value);
+    };
+    const race = {
+      claim: (self) => {
+        if (claimed || settled) return false;
+        claimed = true;
+        clearTimeout(hedge);
+        for (const cancel of cancels) if (cancel !== self) cancel();
+        return true;
+      },
+      onCancel: (cancel) => cancels.push(cancel)
+    };
+    const askNext = () => {
+      clearTimeout(hedge);
+      if (settled || claimed || exhausted || next >= SERVER_SLOTS || Date.now() >= deadline) {
+        if (pending === 0 && !claimed) settle(void 0);
+        return;
+      }
+      const slot = next++;
+      pending++;
+      void attempt(slot, key, request, handshakeMs, void 0, race).then((outcome) => {
+        pending--;
+        if (outcome.kind === "served") return settle(outcome.reply);
+        if (outcome.kind === "lost") return settle("lost");
+        if (claimed) return;
+        if (outcome.kind === "stale" || outcome.kind === "oversize") return settle(void 0);
+        if (outcome.kind === "absent") {
+          if (autostart) startServer(slot);
+          exhausted = true;
+        }
+        askNext();
+      });
+      if (next < SERVER_SLOTS) hedge = setTimeout(askNext, HEDGE_MS);
+    };
+    askNext();
+  });
 }
 async function relayViaServer(event, input, harnessWaitMs) {
   try {
@@ -147,17 +195,29 @@ async function relayViaServer(event, input, harnessWaitMs) {
 var WARM_CLI_COMMANDS = /* @__PURE__ */ new Set([
   "answer",
   "brief",
+  "call-chain",
+  "callers",
   "changed",
+  "context-for",
+  "dead",
+  "deps",
   "exports",
+  "find",
+  "impact",
   "imports",
+  "locate",
   "map",
   "outline",
   "read",
   "refs",
+  "scope",
   "section",
   "semantic",
+  "similar",
   "skeleton",
-  "symbol"
+  "symbol",
+  "test-for",
+  "types"
 ]);
 function warmCliEligible(argv) {
   const command = argv[2];