token-goat 2.9.29 → 2.9.31

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

@@ -1,368 +0,0 @@
-import { createRequire as __cjsRequire } from 'node:module';
-const require = __cjsRequire(import.meta.url);
-import {
-  deliveredOutputBytes
-} from "./token-goat-chunk-L7H3U27L.mjs";
-import {
-  PassthroughFilter,
-  compressedTokensSaved,
-  deliverCompressed,
-  dispatchArgv,
-  filterByName,
-  selectFilter
-} from "./token-goat-chunk-ZER6EAXZ.mjs";
-import {
-  storeBashOutputSync
-} from "./token-goat-chunk-O6I2XU4P.mjs";
-import {
-  capTokens,
-  shlexSplit
-} from "./token-goat-chunk-2WOUYWLM.mjs";
-import {
-  recordStat
-} from "./token-goat-chunk-CIQ2FTY3.mjs";
-import {
-  loadConfig
-} from "./token-goat-chunk-TGOXNAHW.mjs";
-import {
-  init_define_import_meta_env
-} from "./token-goat-chunk-A37V4PBF.mjs";
-
-// src/bash_runner.ts
-init_define_import_meta_env();
-import { spawn, spawnSync } from "node:child_process";
-import * as os from "node:os";
-
-// src/shell.ts
-init_define_import_meta_env();
-import * as fs from "node:fs";
-import * as path from "node:path";
-var WSL_LAUNCHER_SEGMENTS = /* @__PURE__ */ new Set(["system32", "syswow64", "windowsapps"]);
-function isExecutable(p) {
-  try {
-    return fs.statSync(p).isFile();
-  } catch {
-    return false;
-  }
-}
-function isWslLauncherDir(dir) {
-  return dir.toLowerCase().split(/[\\/]+/).some((seg) => WSL_LAUNCHER_SEGMENTS.has(seg));
-}
-function locateBashOnPath(pathValue, isExe = isExecutable) {
-  for (const dir of (pathValue ?? "").split(path.delimiter)) {
-    if (!dir || isWslLauncherDir(dir)) continue;
-    const cand = path.join(dir, "bash.exe");
-    if (isExe(cand)) return cand;
-  }
-  return null;
-}
-function knownGitBashPaths() {
-  const localPrograms = process.env["LOCALAPPDATA"] ? path.join(process.env["LOCALAPPDATA"], "Programs") : void 0;
-  const bases = [
-    process.env["ProgramFiles"],
-    process.env["ProgramFiles(x86)"],
-    process.env["ProgramW6432"],
-    localPrograms
-  ];
-  const out = [];
-  for (const base of bases) {
-    if (!base) continue;
-    out.push(path.join(base, "Git", "bin", "bash.exe"));
-    out.push(path.join(base, "Git", "usr", "bin", "bash.exe"));
-  }
-  return out;
-}
-function resolveWindowsBash() {
-  if (process.platform !== "win32") return null;
-  const override = process.env["TOKEN_GOAT_BASH"];
-  if (override && isExecutable(override)) return override;
-  const onPath = locateBashOnPath(process.env["PATH"]);
-  if (onPath) return onPath;
-  for (const cand of knownGitBashPaths()) {
-    if (isExecutable(cand)) return cand;
-  }
-  return null;
-}
-function wrappedShell() {
-  if (process.platform !== "win32") return true;
-  return resolveWindowsBash() ?? true;
-}
-function canRunWrappedShell() {
-  return process.platform !== "win32" || resolveWindowsBash() !== null;
-}
-function resolvePowerShell(env = process.env, isExe = isExecutable) {
-  const override = env["TOKEN_GOAT_POWERSHELL"] || env["TOKEN_GOAT_PWSH"];
-  if (override && isExe(override)) return override;
-  if (process.platform === "win32") {
-    for (const dir of (env["PATH"] ?? "").split(path.delimiter)) {
-      if (!dir) continue;
-      const cand = path.join(dir, "pwsh.exe");
-      if (isExe(cand)) return cand;
-    }
-    const systemRoot = env["SystemRoot"] ?? env["windir"] ?? "C:\\Windows";
-    const defaultPs = path.join(systemRoot, "System32", "WindowsPowerShell", "v1.0", "powershell.exe");
-    if (isExe(defaultPs)) return defaultPs;
-    return "powershell.exe";
-  }
-  for (const dir of (env["PATH"] ?? "").split(path.delimiter)) {
-    if (!dir) continue;
-    const cand = path.join(dir, "pwsh");
-    if (isExe(cand)) return cand;
-  }
-  for (const cand of ["/usr/bin/pwsh", "/usr/local/bin/pwsh", "/opt/microsoft/powershell/7/pwsh"]) {
-    if (isExe(cand)) return cand;
-  }
-  return "pwsh";
-}
-function canRunPowerShell(env = process.env, isExe = isExecutable) {
-  if (process.platform === "win32") return true;
-  try {
-    const ps = resolvePowerShell(env, isExe);
-    return isExe(ps);
-  } catch {
-    return false;
-  }
-}
-
-// src/bash_runner.ts
-var DEFAULT_TIMEOUT_SECONDS = 600;
-var MAX_CAPTURE_BYTES = 32 * 1024 * 1024;
-var MIN_RECORD_STAT_BYTES = 32;
-var DEFAULT_HEARTBEAT_INTERVAL_MS = 15e3;
-var CaptureSink = class {
-  constructor(cap) {
-    this.cap = cap;
-  }
-  cap;
-  chunks = [];
-  bytes = 0;
-  overflowed = false;
-  push(chunk) {
-    if (this.overflowed) return;
-    this.bytes += chunk.length;
-    if (this.bytes > this.cap) {
-      this.overflowed = true;
-      return;
-    }
-    this.chunks.push(chunk);
-  }
-  text() {
-    return decode(Buffer.concat(this.chunks));
-  }
-};
-function scheduleHeartbeat(startTime, intervalMs, write) {
-  let interval = intervalMs;
-  let timer;
-  const arm = () => {
-    timer = setTimeout(() => {
-      write(Math.round((Date.now() - startTime) / 1e3));
-      interval *= 2;
-      arm();
-    }, interval);
-    timer.unref();
-  };
-  arm();
-  return {
-    onOutput: () => {
-      interval = intervalMs;
-      if (timer !== void 0) clearTimeout(timer);
-      arm();
-    },
-    stop: () => {
-      if (timer !== void 0) clearTimeout(timer);
-    }
-  };
-}
-function resolveFilter(command, filterName, cwd) {
-  const effectiveCwd = cwd ?? process.cwd();
-  let split;
-  try {
-    split = shlexSplit(command);
-  } catch {
-    split = null;
-  }
-  const argv = split === null ? [command] : dispatchArgv(split, effectiveCwd).argv;
-  if (filterName) {
-    const named = filterByName(filterName);
-    if (named !== null) return { filter: named, argv };
-  }
-  if (split === null) return { filter: null, argv };
-  return { filter: selectFilter(split, effectiveCwd), argv };
-}
-function baseSpawnOptions(timeout, cwd) {
-  return { timeout: timeout * 1e3, cwd };
-}
-function spawnTarget(command, nativeShell, shellType) {
-  if (shellType === "pwsh" || shellType === "powershell") {
-    const ps = resolvePowerShell();
-    return {
-      file: ps,
-      args: [
-        "-NoProfile",
-        "-NonInteractive",
-        "-Command",
-        "$c = $env:TG_CMD; Remove-Item Env:\\TG_CMD -ErrorAction SilentlyContinue; [scriptblock]::Create($c).Invoke(); if ($LASTEXITCODE -ne $null -and $LASTEXITCODE -ne 0) { exit $LASTEXITCODE }"
-      ],
-      shell: false,
-      cmdEnv: { TG_CMD: command }
-    };
-  }
-  const shell = nativeShell || shellType === "native" ? true : wrappedShell();
-  if (typeof shell === "string") return { file: shell, args: ["-c", 'eval "$TG_CMD"'], shell: false, cmdEnv: { TG_CMD: command } };
-  return { file: command, args: [], shell };
-}
-function mergeSpawnEnv(env, cmdEnv) {
-  return cmdEnv ? { ...env ?? process.env, ...cmdEnv } : env;
-}
-function signalExitCode(signal) {
-  const num = os.constants.signals[signal];
-  return 128 + (num ?? 0);
-}
-function resolveProfile(explicit) {
-  if (explicit) return explicit;
-  try {
-    const p = loadConfig().compression.profile;
-    return p && p !== "auto" ? p : "balanced";
-  } catch {
-    return "balanced";
-  }
-}
-function resolveCompressLimits() {
-  try {
-    const bc = loadConfig().bash_compress;
-    return { maxLines: bc.max_lines, maxBytes: bc.max_bytes };
-  } catch {
-    return { maxLines: 1e3, maxBytes: 64 * 1024 };
-  }
-}
-async function run(command, opts = {}) {
-  const timeout = opts.timeout ?? DEFAULT_TIMEOUT_SECONDS;
-  const { filter, argv } = resolveFilter(command, opts.filterName, opts.cwd);
-  if (filter === null) {
-    if ((opts.maxTokens ?? 0) > 0 || opts.quietSuccess) {
-      return wrapAndCompress(command, argv, new PassthroughFilter(), timeout, resolveProfile(opts.compressionProfile), opts);
-    }
-    return passthrough(command, timeout, opts.cwd, opts.env, opts.nativeShell, opts.shellType);
-  }
-  return wrapAndCompress(command, argv, filter, timeout, resolveProfile(opts.compressionProfile), opts);
-}
-function runRaw(command, timeout = DEFAULT_TIMEOUT_SECONDS, nativeShell, shellType) {
-  return passthrough(command, timeout, void 0, void 0, nativeShell, shellType);
-}
-function passthrough(command, timeout, cwd, env, nativeShell, shellType) {
-  const { file, args, shell, cmdEnv } = spawnTarget(command, nativeShell, shellType);
-  const result = spawnSync(file, args, {
-    ...baseSpawnOptions(timeout, cwd),
-    shell,
-    env: mergeSpawnEnv(env, cmdEnv),
-    stdio: "inherit"
-  });
-  if (isTimeout(result.error)) return 124;
-  if (result.status !== null) return result.status;
-  if (result.signal) return signalExitCode(result.signal);
-  return 0;
-}
-function isTimeout(error) {
-  return !!error && error.code === "ETIMEDOUT";
-}
-function decode(buf) {
-  if (buf == null) return "";
-  return typeof buf === "string" ? buf : buf.toString("utf8");
-}
-async function wrapAndCompress(command, argv, filter, timeout, profile, opts) {
-  const writeStdout = opts.writeStdout ?? ((s) => process.stdout.write(s));
-  const writeStderr = opts.writeStderr ?? ((s) => process.stderr.write(s));
-  const startTime = Date.now();
-  const { file, args, shell, cmdEnv } = spawnTarget(command, opts.nativeShell, opts.shellType);
-  const stdoutSink = new CaptureSink(MAX_CAPTURE_BYTES);
-  const stderrSink = new CaptureSink(MAX_CAPTURE_BYTES);
-  let timedOut = false;
-  let overflowed = false;
-  const child = spawn(file, args, { cwd: opts.cwd, shell, env: mergeSpawnEnv(opts.env, cmdEnv), stdio: ["ignore", "pipe", "pipe"] });
-  const heartbeat = scheduleHeartbeat(startTime, opts.heartbeatIntervalMs ?? DEFAULT_HEARTBEAT_INTERVAL_MS, (elapsed) => {
-    writeStderr(`[token-goat compress] still running, ${elapsed}s elapsed
-`);
-  });
-  const timeoutHandle = setTimeout(() => {
-    timedOut = true;
-    child.kill();
-  }, timeout * 1e3);
-  const onData = (sink) => (chunk) => {
-    heartbeat.onOutput();
-    sink.push(chunk);
-    if (sink.overflowed) {
-      overflowed = true;
-      child.kill();
-    }
-  };
-  child.stdout?.on("data", onData(stdoutSink));
-  child.stderr?.on("data", onData(stderrSink));
-  const { code, signal } = await new Promise((resolve) => {
-    child.on("close", (code2, signal2) => resolve({ code: code2, signal: signal2 }));
-  });
-  heartbeat.stop();
-  clearTimeout(timeoutHandle);
-  let stdoutText = stdoutSink.text();
-  let stderrText = stderrSink.text();
-  if (overflowed) {
-    stdoutText += `
-[token-goat: capture capped at ${MAX_CAPTURE_BYTES / (1024 * 1024)} MiB]`;
-  }
-  let exitCode;
-  if (timedOut) {
-    exitCode = 124;
-    stderrText = (stderrText ? stderrText + "\n" : "") + `[token-goat: command exceeded ${timeout}s timeout and was killed]`;
-  } else if (code !== null) {
-    exitCode = code;
-  } else if (signal) {
-    exitCode = signalExitCode(signal);
-  } else {
-    exitCode = 0;
-  }
-  if (opts.quietSuccess && exitCode === 0) {
-    const duration = ((Date.now() - startTime) / 1e3).toFixed(1) + "s";
-    const fullRaw = (stdoutText + (stderrText ? "\n" + stderrText : "")).trim();
-    const id = storeBashOutputSync(command, fullRaw, 0, opts.cwd ?? null);
-    const summary = `[tg: ok] ${command} (${duration}, recall: token-goat bash-output ${id})
-`;
-    writeStdout(summary);
-    return 0;
-  }
-  const limits = resolveCompressLimits();
-  const { applied, marker, compressed, text: delivered } = deliverCompressed(filter, stdoutText, stderrText, exitCode, argv, {
-    compressionProfile: profile,
-    maxLines: limits.maxLines,
-    maxBytes: limits.maxBytes
-  });
-  let text = delivered;
-  const maxTokens = opts.maxTokens ?? 0;
-  if (maxTokens > 0) text = capTokens(text, maxTokens);
-  const capped = text !== delivered;
-  let body = text + marker;
-  if (applied || capped) {
-    const fullRaw = (stdoutText + (stderrText ? "\n" + stderrText : "")).trim();
-    const recallId = storeBashOutputSync(command, fullRaw, exitCode, opts.cwd ?? null);
-    body += "\n[token-goat] full output: bash-output " + recallId + " --full";
-    if (opts.capHint) body += "\n[token-goat] " + opts.capHint;
-  }
-  writeStdout(body.endsWith("\n") ? body : body + "\n");
-  if (applied) recordSavings(compressed);
-  return exitCode;
-}
-function recordSavings(result) {
-  const delivered = deliveredOutputBytes(result.originalBytes);
-  const bytesSaved = Math.max(0, delivered - result.compressedBytes);
-  if (bytesSaved < MIN_RECORD_STAT_BYTES) return;
-  recordStat(`bash_compress:${result.filterName}`, bytesSaved, compressedTokensSaved(bytesSaved));
-}
-
-export {
-  canRunWrappedShell,
-  canRunPowerShell,
-  DEFAULT_TIMEOUT_SECONDS,
-  MAX_CAPTURE_BYTES,
-  DEFAULT_HEARTBEAT_INTERVAL_MS,
-  resolveFilter,
-  run,
-  runRaw
-};