token-goat 2.9.26 → 2.9.29

package/dist/{token-goat-chunk-G6EYO5CD.mjs → token-goat-chunk-ONXH6NZ5.mjs}

@@ -1,13 +1,8 @@
 import { createRequire as __cjsRequire } from 'node:module';
 const require = __cjsRequire(import.meta.url);
 import {
-  classifyContent,
-  guardDivisor,
-  stripAnsiEscapes
-} from "./token-goat-chunk-D2X7XN5M.mjs";
-import {
-  safeSlice
-} from "./token-goat-chunk-6E2IDLHF.mjs";
+  estimateTokensFromLength
+} from "./token-goat-chunk-52NSDWYY.mjs";
 import {
   displaySafeText
 } from "./token-goat-chunk-NMTKNYGF.mjs";
@@ -25,82 +20,6 @@ function claudeConfigDir(homeDir = os.homedir()) {
   return path.join(homeDir, ".claude");
 }
 
-// src/overflow_guard.ts
-init_define_import_meta_env();
-function estimateTokensFromLength(length, cls = "text") {
-  return Math.max(1, Math.floor(Math.max(0, length) / guardDivisor(cls)) + 1);
-}
-function estimateTokens(text) {
-  const stripped = stripAnsiEscapes(text);
-  return estimateTokensFromLength(stripped.length, classifyContent(stripped));
-}
-function trimToBudget(text, budgetTokens, command) {
-  const markerMarginTokens = 64;
-  const strippedAll = stripAnsiEscapes(text);
-  const contentClass = classifyContent(strippedAll);
-  const totalTokens = estimateTokensFromLength(strippedAll.length, contentClass);
-  if (totalTokens <= budgetTokens) {
-    return text;
-  }
-  const lines = text.split("\n");
-  if (lines.length > 1 && lines[lines.length - 1] === "") lines.pop();
-  const totalLines = lines.length;
-  const bodyBudget = Math.max(1, budgetTokens - markerMarginTokens);
-  const charBudget = bodyBudget * guardDivisor(contentClass);
-  const kept = [];
-  let used = 0;
-  for (const ln of lines) {
-    const stripped = stripAnsiEscapes(ln);
-    const cost = ln.length + 1;
-    if (kept.length === 0 && cost > charBudget) {
-      const truncated = safeSlice(stripped, charBudget);
-      kept.push(truncated);
-      break;
-    }
-    if (kept.length > 0 && used + cost > charBudget) {
-      break;
-    }
-    kept.push(ln);
-    used += cost;
-  }
-  const shown = kept.length;
-  const hint = getHintFor(command);
-  const marker = `[token-goat: output capped at ~${budgetTokens} tokens to protect context \u2014 showing ${shown} of ${totalLines} lines. ${hint}]`;
-  return kept.join("\n") + "\n" + marker;
-}
-function capJsonRows(items, budgetTokens) {
-  const totalCount = items.length;
-  const charBudget = Math.max(1, budgetTokens * guardDivisor());
-  const kept = [];
-  let used = 0;
-  for (const item of items) {
-    const cost = JSON.stringify(item).length + 2;
-    if (kept.length > 0 && used + cost > charBudget) break;
-    kept.push(item);
-    used += cost;
-  }
-  return { items: kept, truncated: kept.length < totalCount, totalCount };
-}
-function getHintFor(command) {
-  const cmd = (command || "").toLowerCase().trim();
-  if (cmd === "symbol") {
-    return "Request a specific method (file.py::Class.method) or use --json for structured access.";
-  }
-  if (cmd === "heading" || cmd === "section") {
-    return "Request a narrower sub-heading, e.g. 'doc.md::Section#2'.";
-  }
-  if (cmd === "lines") {
-    return "Request a smaller line range, e.g. 'file.py@100-150'.";
-  }
-  if (cmd === "bash-output" || cmd === "web-output") {
-    return "Use --grep PATTERN, --section HEADING, or --tail N to narrow the cached output.";
-  }
-  if (cmd === "semantic") {
-    return "Narrow your query text or pass --limit to reduce the number of matches returned.";
-  }
-  return "Narrow your query or raise overflow_guard max_tokens in config.";
-}
-
 // src/resident_context.ts
 init_define_import_meta_env();
 import * as fs from "node:fs";
@@ -307,10 +226,6 @@ function accumulateResidentLines(lines) {
 
 export {
   claudeConfigDir,
-  estimateTokensFromLength,
-  estimateTokens,
-  trimToBudget,
-  capJsonRows,
   createResidentContextStats,
   accumulateResidentLine,
   summarizeResidentContext,

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

@@ -0,0 +1,295 @@
+import { createRequire as __cjsRequire } from 'node:module';
+const require = __cjsRequire(import.meta.url);
+import {
+  serveOne,
+  swapEnv
+} from "./token-goat-chunk-S4VSXURL.mjs";
+import {
+  relayInProcess
+} from "./token-goat-chunk-QO3LXVVR.mjs";
+import "./token-goat-chunk-IVWI2OCK.mjs";
+import "./token-goat-chunk-DNRPXHYH.mjs";
+import "./token-goat-chunk-RTMVLHWX.mjs";
+import "./token-goat-chunk-L7H3U27L.mjs";
+import "./token-goat-chunk-RRNZMM3A.mjs";
+import "./token-goat-chunk-E4ZOJIZK.mjs";
+import "./token-goat-chunk-E5TK5DNO.mjs";
+import "./token-goat-chunk-ZER6EAXZ.mjs";
+import "./token-goat-chunk-44BM7DDQ.mjs";
+import "./token-goat-chunk-EUSDXRXU.mjs";
+import "./token-goat-chunk-G7IWXCFF.mjs";
+import "./token-goat-chunk-H5J3C56Q.mjs";
+import "./token-goat-chunk-TDUT2CL6.mjs";
+import "./token-goat-chunk-ONXH6NZ5.mjs";
+import "./token-goat-chunk-O6I2XU4P.mjs";
+import "./token-goat-chunk-D4ETHJD6.mjs";
+import "./token-goat-chunk-52NSDWYY.mjs";
+import "./token-goat-chunk-2WOUYWLM.mjs";
+import "./token-goat-chunk-CIQ2FTY3.mjs";
+import {
+  PROTOCOL_VERSION,
+  SERVER_SLOTS,
+  bundleEntryFiles,
+  configStamp,
+  endpointFor,
+  ensureServerKey,
+  mac,
+  macMatches,
+  nonce,
+  readFrames,
+  removeMarker,
+  serverKeyPath,
+  touchMarker,
+  writeFrame
+} from "./token-goat-chunk-DPNLUJWB.mjs";
+import "./token-goat-chunk-3BTK54F3.mjs";
+import "./token-goat-chunk-VMFWNHGR.mjs";
+import "./token-goat-chunk-435UQTKD.mjs";
+import "./token-goat-chunk-7CPMDFGB.mjs";
+import {
+  hookServerEnabled
+} from "./token-goat-chunk-TGOXNAHW.mjs";
+import "./token-goat-chunk-XB5KKOCG.mjs";
+import {
+  clearPerRequestCaches
+} from "./token-goat-chunk-YZAAZU2S.mjs";
+import "./token-goat-chunk-IVE6U7KV.mjs";
+import "./token-goat-chunk-XYWPOSCN.mjs";
+import "./token-goat-chunk-TCQANXYC.mjs";
+import "./token-goat-chunk-NMTKNYGF.mjs";
+import {
+  VERSION,
+  configPath,
+  ensureDataDirPrivate
+} from "./token-goat-chunk-MELOKFDJ.mjs";
+import "./token-goat-chunk-Y4AFKTHK.mjs";
+import "./token-goat-chunk-JTKCQMQA.mjs";
+import {
+  init_define_import_meta_env
+} from "./token-goat-chunk-A37V4PBF.mjs";
+
+// src/hook_server.ts
+init_define_import_meta_env();
+import * as fs from "node:fs";
+import * as net from "node:net";
+import * as os from "node:os";
+var IDLE_EXIT_MS = { primary: 30 * 6e4, overflow: 3 * 6e4 };
+var KEY_CHECK_MS = 2e3;
+var HANDSHAKE_IDLE_MS = 1e4;
+function bundleStamp() {
+  return bundleEntryFiles().map((f) => {
+    try {
+      const st = fs.statSync(f);
+      return `${st.size}:${st.mtimeMs}`;
+    } catch {
+      return "-";
+    }
+  }).join("|");
+}
+var enabledMemo;
+function stillEnabled() {
+  const key = `${process.env["TOKEN_GOAT_HOOK_SERVER"] ?? ""}|${configPath()}|${configStamp()}`;
+  if (enabledMemo?.key !== key) enabledMemo = { key, enabled: hookServerEnabled() };
+  return enabledMemo.enabled;
+}
+function retirementReason(loadedStamp) {
+  if (bundleStamp() !== loadedStamp) return "bundle replaced";
+  if (!stillEnabled()) return "disabled";
+  return void 0;
+}
+async function handle(request, runCli, send) {
+  clearPerRequestCaches();
+  if (request.kind === "cli") {
+    const res = await serveOne({ id: 0, argv: request.argv, cwd: request.cwd, env: request.env }, runCli, clearPerRequestCaches);
+    return send({ ok: true, stdout: res.stdout, stderr: res.stderr, status: res.status });
+  }
+  const startedAt = performance.now();
+  const cwdBefore = process.cwd();
+  const restoreEnv = swapEnv(request.env);
+  const afterReply = [];
+  try {
+    process.chdir(request.cwd);
+    const payload = JSON.parse(request.input);
+    const stdout = await relayInProcess(request.event, payload, request.harnessWaitMs, { elapsedMs: () => request.elapsedMs + (performance.now() - startedAt), afterReply: (work) => afterReply.push(work) });
+    send({ ok: true, stdout });
+    for (const work of afterReply) work();
+  } finally {
+    restoreEnv();
+    try {
+      process.chdir(cwdBefore);
+    } catch {
+    }
+    clearPerRequestCaches();
+  }
+}
+function listen(server, endpoint) {
+  const tryListen = () => new Promise((resolve) => {
+    server.once("error", resolve);
+    server.listen({ path: endpoint, readableAll: false, writableAll: false }, () => {
+      server.removeListener("error", resolve);
+      resolve(void 0);
+    });
+  });
+  return tryListen().then(async (err) => {
+    if (err === void 0) return true;
+    if (err.code !== "EADDRINUSE" || process.platform === "win32") return false;
+    const alive = await new Promise((resolve) => {
+      const probe = net.connect(endpoint);
+      probe.once("connect", () => {
+        probe.destroy();
+        resolve(true);
+      });
+      probe.once("error", () => resolve(false));
+    });
+    if (alive) return false;
+    fs.rmSync(endpoint, { force: true });
+    return await tryListen() === void 0;
+  });
+}
+async function runHookServer(slot, runCli) {
+  if (!Number.isInteger(slot) || slot < 0 || slot >= SERVER_SLOTS) throw new Error(`--slot must be an integer from 0 to ${SERVER_SLOTS - 1}`);
+  if (!hookServerEnabled()) {
+    touchMarker("disabled", configStamp());
+    return;
+  }
+  ensureDataDirPrivate();
+  const key = ensureServerKey();
+  const keyPath = serverKeyPath();
+  const loadedStamp = bundleStamp();
+  const endpoint = endpointFor(slot);
+  process.chdir(os.tmpdir());
+  const status = { pid: process.pid, slot, version: VERSION, startedAt: Date.now(), lastUsedAt: Date.now(), served: 0, errors: 0 };
+  let busy = false;
+  let retiring = false;
+  let idleTimer;
+  const flushing = /* @__PURE__ */ new Set();
+  const awaitingRequest = /* @__PURE__ */ new Set();
+  const endWith = (socket, frame) => {
+    const flushed = new Promise((resolve) => {
+      socket.once("close", () => resolve());
+      writeFrame(socket, frame);
+      socket.end(() => resolve());
+    });
+    flushing.add(flushed);
+    void flushed.then(() => flushing.delete(flushed));
+  };
+  let socketIno;
+  const ownsEndpoint = () => {
+    if (socketIno === void 0) return true;
+    try {
+      return fs.statSync(endpoint, { bigint: true, throwIfNoEntry: false })?.ino === socketIno;
+    } catch {
+      return true;
+    }
+  };
+  const server = net.createServer((socket) => serveConnection(socket));
+  const exitOnceFlushed = () => void Promise.all(flushing).then(() => process.exit(0));
+  const stop = () => {
+    retiring = true;
+    if (idleTimer !== void 0) clearTimeout(idleTimer);
+    if (ownsEndpoint()) server.close();
+    for (const socket of awaitingRequest) endWith(socket, { t: "stale" });
+    awaitingRequest.clear();
+    if (!busy) exitOnceFlushed();
+  };
+  const armIdle = () => {
+    if (idleTimer !== void 0) clearTimeout(idleTimer);
+    idleTimer = setTimeout(stop, slot === 0 ? IDLE_EXIT_MS.primary : IDLE_EXIT_MS.overflow);
+  };
+  function serveConnection(socket) {
+    let nc = "";
+    let ns = "";
+    socket.setTimeout(HANDSHAKE_IDLE_MS, () => socket.destroy());
+    socket.on("error", () => socket.destroy());
+    socket.once("close", () => awaitingRequest.delete(socket));
+    const onMessage = (msg) => {
+      if (msg["t"] === "hello" && nc === "") {
+        if (msg["v"] !== PROTOCOL_VERSION || typeof msg["nc"] !== "string" || msg["nc"] === "") return void socket.destroy();
+        const reason = retiring ? "retiring" : retirementReason(loadedStamp);
+        if (reason !== void 0) {
+          if (reason === "disabled") touchMarker("disabled", configStamp());
+          if (reason === "bundle replaced") removeMarker(`spawn-${slot}`);
+          writeFrame(socket, { t: "stale" });
+          socket.end(() => {
+            if (!retiring) stop();
+          });
+          return;
+        }
+        if (busy && msg["ctl"] !== true) {
+          writeFrame(socket, { t: "busy" });
+          return void socket.end();
+        }
+        nc = msg["nc"];
+        ns = nonce();
+        writeFrame(socket, { t: "challenge", v: PROTOCOL_VERSION, ns, mac: mac(key, "S", nc, ns) });
+        awaitingRequest.add(socket);
+        return;
+      }
+      if (msg["t"] === "req" && ns !== "") {
+        const body = msg["body"];
+        if (typeof body !== "string" || !macMatches(mac(key, "C", nc, ns, body), msg["mac"])) return void socket.destroy();
+        if (!awaitingRequest.delete(socket)) return;
+        let request;
+        let parseError;
+        try {
+          request = JSON.parse(body);
+        } catch (e) {
+          parseError = e;
+        }
+        const kind = request?.kind;
+        if (kind === "status" || kind === "stop") {
+          const resBody = JSON.stringify({ ok: true, info: { ...status } });
+          endWith(socket, { t: "res", mac: mac(key, "R", nc, ns, resBody), body: resBody });
+          if (kind === "stop" && !retiring) stop();
+          return;
+        }
+        if (busy || retiring) {
+          writeFrame(socket, { t: retiring ? "stale" : "busy" });
+          return void socket.end();
+        }
+        busy = true;
+        socket.setTimeout(0);
+        let replied = false;
+        const reply = (res) => {
+          if (replied) return;
+          replied = true;
+          const resBody = JSON.stringify(res);
+          endWith(socket, { t: "res", mac: mac(key, "R", nc, ns, resBody), body: resBody });
+        };
+        void (async () => {
+          try {
+            if (request === void 0) throw parseError;
+            await handle(request, runCli, reply);
+            status.served++;
+          } catch (e) {
+            status.errors++;
+            reply({ ok: false, error: e instanceof Error ? e.message : String(e) });
+          }
+          busy = false;
+          status.lastUsedAt = Date.now();
+          if (retiring) stop();
+          else armIdle();
+        })();
+        return;
+      }
+      socket.destroy();
+    };
+    readFrames(socket, onMessage, () => void 0);
+  }
+  if (!await listen(server, endpoint)) return;
+  if (process.platform !== "win32") socketIno = fs.statSync(endpoint, { bigint: true }).ino;
+  removeMarker("failed");
+  setInterval(() => {
+    if (!fs.existsSync(keyPath) || !ownsEndpoint()) stop();
+  }, KEY_CHECK_MS).unref();
+  if (process.platform !== "win32") {
+    try {
+      fs.chmodSync(endpoint, 384);
+    } catch {
+    }
+  }
+  armIdle();
+}
+export {
+  runHookServer
+};

package/dist/{token-goat-chunk-LHUP4DXZ.mjs → token-goat-chunk-PGTH2U4C.mjs}

@@ -10,7 +10,7 @@ import {
   stripMultilineStringSpan,
   stripNestedBlockCommentSpan,
   stripStringLiterals
-} from "./token-goat-chunk-7SXPJOSZ.mjs";
+} from "./token-goat-chunk-7CPMDFGB.mjs";
 import {
   init_define_import_meta_env
 } from "./token-goat-chunk-A37V4PBF.mjs";