golem-kit 0.1.1 → 0.2.1

package/src/runtime/harness/agent-status.js

@@ -0,0 +1,388 @@
+// Vendored from bridge-commander 5bf87e4b harness/agent-status.js (zero-dependency). Local patches are marked 'golem:'.
+'use strict';
+// agent-status — session status (model, context usage, rate limits) read from
+// the files the harnesses ALREADY write; no statusline dependency, nothing new
+// is persisted. Backs the OPTIONAL `status`/`runCommand` capability verbs
+// (port.js): claude-tmux reads the session transcript under ~/.claude/projects,
+// codex-tmux reads the rollout log under ~/.codex/sessions. Everything here is
+// best-effort by contract: a missing/unreadable/foreign-shaped file returns
+// null, never a throw.
+//
+// Status shape (the port's `status(ref)` return value):
+//   { model, contextUsed, contextWindow, effort?, rateLimits? }
+//   effort (reasoning level, e.g. "high") — from the claude sidecar or the codex
+//   turn_context; absent when unknown (e.g. the claude transcript fallback).
+//   rateLimits (codex only — claude does not persist them): { primary?,
+//   secondary? } each { usedPercent, windowMinutes, resetsAt (epoch secs) }.
+//
+// Files can grow to tens of MB, so reads are TAIL reads (last N bytes, from
+// the end); the interesting lines — claude's last assistant message, codex's
+// last token_count event — always sit near the bottom. claude alone gets one
+// escalation step: a huge tool-result line can push the last assistant line
+// past a small tail window.
+
+const fs = require('node:fs');
+const os = require('node:os');
+const path = require('node:path');
+
+const TAIL_BYTES = 256 * 1024;
+
+// tailRead — the last maxBytes of a file, decoded; null when missing/unreadable.
+// When the file is smaller than maxBytes the whole file comes back.
+function tailRead(file, maxBytes) {
+  let fd;
+  try {
+    fd = fs.openSync(file, 'r');
+  } catch {
+    return null;
+  }
+  try {
+    const size = fs.fstatSync(fd).size;
+    const want = Math.min(size, maxBytes);
+    const buf = Buffer.alloc(want);
+    fs.readSync(fd, buf, 0, want, size - want);
+    return buf.toString('utf8');
+  } catch {
+    return null;
+  } finally {
+    fs.closeSync(fd);
+  }
+}
+
+// ---------- claude ----------
+// Transcript path: ~/.claude/projects/<slug(cwd)>/<sessionId>.jsonl — the slug
+// replaces every non-alphanumeric cwd character with '-' (verified against
+// real transcript dirs: /home/ai/.treehouse/x → -home-ai--treehouse-x).
+function claudeProjectSlug(cwd) {
+  return String(cwd).replace(/[^A-Za-z0-9]/g, '-');
+}
+
+// Context window per model — matched by substring so versioned ids
+// (claude-fable-5, claude-opus-4-8, …) hit without an exhaustive list.
+// Extend by adding a pair; unknown models get the conservative default.
+const CLAUDE_CONTEXT_WINDOWS = [
+  ['fable', 1000000],
+  ['opus', 200000],
+  ['sonnet', 200000],
+  ['haiku', 200000],
+];
+const CLAUDE_DEFAULT_WINDOW = 200000;
+function claudeContextWindow(model) {
+  const m = String(model || '').toLowerCase();
+  for (const [needle, window] of CLAUDE_CONTEXT_WINDOWS) {
+    if (m.includes(needle)) return window;
+  }
+  return CLAUDE_DEFAULT_WINDOW;
+}
+
+// ---------- claude statusline sidecar ----------
+// harness/statusline.js (the workspace's Claude Code `statusLine` command) tees
+// each stdin payload to <workspace>/.bridge-commander/statusline/<session_id>.json
+// — the ONLY source of the REAL context window (context_window_size), the
+// account rate limits, and the model display name. Lieutenants run with cwd =
+// workspace root, so the sidecar exists for them; workers in worktree cwds have
+// none and fall through to the transcript+map path below (accepted).
+const STATE_DIR_NAME = '.bridge-commander';
+function findBridgeWorkspace(startDir) {
+  if (!startDir) return null;
+  let dir = path.resolve(startDir);
+  for (;;) {
+    try {
+      if (fs.statSync(path.join(dir, STATE_DIR_NAME)).isDirectory()) return dir;
+    } catch { /* keep walking up */ }
+    const parent = path.dirname(dir);
+    if (parent === dir) return null;
+    dir = parent;
+  }
+}
+
+// Map the statusline payload's rate_limits (five_hour/seven_day, each
+// {used_percentage, resets_at}) onto the shared status rateLimits shape
+// (primary/secondary, each {usedPercent, windowMinutes, resetsAt epoch secs}),
+// so formatStatus labels them 5h / 1w just like codex.
+function claudeSidecarRateLimits(rl) {
+  if (!rl || typeof rl !== 'object') return null;
+  const toEpochSecs = (v) => {
+    if (v == null) return undefined;
+    if (typeof v === 'number' && Number.isFinite(v)) return v > 1e11 ? Math.floor(v / 1000) : Math.floor(v);
+    if (typeof v === 'string' && v.trim() !== '') {
+      const n = Number(v);
+      if (Number.isFinite(n)) return n > 1e11 ? Math.floor(n / 1000) : Math.floor(n);
+      const p = Date.parse(v);
+      if (!Number.isNaN(p)) return Math.floor(p / 1000);
+    }
+    return undefined;
+  };
+  const pick = (w, windowMinutes) => {
+    if (!w || typeof w !== 'object' || w.used_percentage == null) return undefined;
+    const out = { usedPercent: Number(w.used_percentage), windowMinutes };
+    const r = toEpochSecs(w.resets_at);
+    if (r !== undefined) out.resetsAt = r;
+    return out;
+  };
+  const out = {};
+  const primary = pick(rl.five_hour, 300);
+  const secondary = pick(rl.seven_day, 10080);
+  if (primary) out.primary = primary;
+  if (secondary) out.secondary = secondary;
+  return Object.keys(out).length ? out : null;
+}
+
+// claudeSidecarStatus(ref, opts?) -> status | null
+// Prefer the sidecar when one exists for this session (file name == resumeId)
+// and it carries a real context_window_size. Any missing file / bad JSON /
+// absent window → null, so the caller falls back to the transcript path.
+function claudeSidecarStatus(ref, opts = {}) {
+  const dir = opts.sidecarDir
+    || (() => {
+      const ws = opts.workspace || findBridgeWorkspace(ref.cwd);
+      return ws ? path.join(ws, STATE_DIR_NAME, 'statusline') : null;
+    })();
+  if (!dir) return null;
+  let doc;
+  try {
+    doc = JSON.parse(fs.readFileSync(path.join(dir, ref.resumeId + '.json'), 'utf8'));
+  } catch {
+    return null;
+  }
+  const p = doc && doc.payload;
+  const cw = p && p.context_window;
+  const window = cw && Number(cw.context_window_size);
+  if (!p || !cw || !Number.isFinite(window) || window <= 0) return null;
+  const out = {
+    model: (p.model && (p.model.id || p.model.display_name)) || null,
+    contextUsed: Number(cw.total_input_tokens) || 0,
+    contextWindow: window,
+  };
+  const effort = p.effort && p.effort.level;
+  if (effort) out.effort = String(effort);
+  const rl = claudeSidecarRateLimits(p.rate_limits);
+  if (rl) out.rateLimits = rl;
+  return out;
+}
+
+// claudeStatus(ref, opts?) -> status | null
+// Prefer the statusline sidecar (real window + rate limits); else the last
+// assistant line's message.usage: contextUsed = input + cache_read +
+// cache_creation + output (what the next turn starts from), with the context
+// window guessed from the model→window map.
+function claudeStatus(ref, opts = {}) {
+  if (!ref || !ref.cwd || !ref.resumeId) return null;
+  const sidecar = claudeSidecarStatus(ref, opts);
+  if (sidecar) return sidecar;
+  const projectsDir = opts.projectsDir || process.env.BC_CLAUDE_PROJECTS_DIR
+    || path.join(os.homedir(), '.claude', 'projects');
+  const file = path.join(projectsDir, claudeProjectSlug(ref.cwd), ref.resumeId + '.jsonl');
+  let size = 0;
+  try {
+    size = fs.statSync(file).size;
+  } catch {
+    return null;
+  }
+  // One escalation: 256KB tail first, 4MB if no assistant line surfaced.
+  for (const maxBytes of [TAIL_BYTES, 16 * TAIL_BYTES]) {
+    const text = tailRead(file, maxBytes);
+    if (text === null) return null;
+    const lines = text.split('\n');
+    for (let i = lines.length - 1; i >= 0; i--) {
+      if (!lines[i].includes('"type":"assistant"')) continue;
+      let doc;
+      try {
+        doc = JSON.parse(lines[i]);
+      } catch {
+        continue; // the tail window's first line may be cut mid-JSON
+      }
+      const msg = doc && doc.type === 'assistant' && doc.message;
+      const u = msg && msg.usage;
+      if (!u || typeof u !== 'object') continue;
+      const used = (u.input_tokens || 0) + (u.cache_read_input_tokens || 0)
+        + (u.cache_creation_input_tokens || 0) + (u.output_tokens || 0);
+      return {
+        model: msg.model || null,
+        contextUsed: used,
+        contextWindow: claudeContextWindow(msg.model),
+      };
+    }
+    if (size <= maxBytes) break; // whole file scanned — a bigger tail finds nothing new
+  }
+  return null;
+}
+
+// ---------- codex ----------
+// Rollout path: ~/.codex/sessions/YYYY/MM/DD/rollout-<ts>-<threadId>.jsonl
+// (threadId — see codexThreadId). The date dirs are walked newest-first and the
+// first match wins, so a thread resumed on a later day resolves to its newest
+// rollout file.
+function codexRolloutFile(threadId, sessionsDir) {
+  const suffix = '-' + threadId + '.jsonl';
+  const listDesc = (dir) => {
+    try {
+      return fs.readdirSync(dir).sort().reverse();
+    } catch {
+      return [];
+    }
+  };
+  for (const year of listDesc(sessionsDir)) {
+    for (const month of listDesc(path.join(sessionsDir, year))) {
+      for (const day of listDesc(path.join(sessionsDir, year, month))) {
+        const dir = path.join(sessionsDir, year, month, day);
+        const hit = listDesc(dir).find((f) => f.startsWith('rollout-') && f.endsWith(suffix));
+        if (hit) return path.join(dir, hit);
+      }
+    }
+  }
+  return null;
+}
+
+function codexRateLimits(rl) {
+  if (!rl || typeof rl !== 'object') return null;
+  const pick = (w) => (w && typeof w === 'object' ? {
+    usedPercent: w.used_percent,
+    windowMinutes: w.window_minutes,
+    resetsAt: w.resets_at,
+  } : undefined);
+  const out = {};
+  if (pick(rl.primary)) out.primary = pick(rl.primary);
+  if (pick(rl.secondary)) out.secondary = pick(rl.secondary);
+  return Object.keys(out).length ? out : null;
+}
+
+// codexStatus(ref, opts?) -> status | null
+// The thread-id comes from codexThreadId (recorded session-id file, then the
+// ref) — NOT from ref.resumeId alone: a codex ref is born without one.
+// The last token_count event carries current context occupancy
+// (info.last_token_usage.total_tokens = input+cached+output of the last turn)
+// and the model context window; the model rides every turn_context line, so the
+// tail always has a fresh one. rate_limits come from the same token_count.
+// NOTE: info.total_token_usage is the CUMULATIVE session total (grows forever,
+// exceeds the window) — it is NOT occupancy. Selecting on last_token_usage is
+// deliberate: rollouts where info is populated always carry it (verified on
+// real rollouts), and the only ones missing it have info === null, which the
+// null-guards below already reject — so no total_token_usage fallback is needed.
+// codexThreadId(ref, opts) -> the thread-id whose rollout to read, or null.
+// The notify relay rewrites <stateDir>/<key>.session-id at every turn, so that
+// file is ground truth; ref.resumeId is adopted once from a turn-end POST and a
+// ref can live its whole life without one (codex assigns the id itself — see
+// codex-tmux.js). Same order resume() uses: recorded file first, ref second.
+// opts.stateDir comes from the caller that knows where harness state lives
+// (codex-tmux status()); without it only the ref can answer.
+function codexThreadId(ref, opts = {}) {
+  if (ref && ref.session && opts.stateDir) {
+    const key = ref.window ? ref.session + ':' + ref.window : ref.session;
+    try {
+      const rec = fs.readFileSync(path.join(opts.stateDir, key + '.session-id'), 'utf8').trim();
+      if (rec) return rec;
+    } catch {
+      // no recorded id — the ref's is all there is
+    }
+  }
+  return (ref && ref.resumeId) || null;
+}
+
+function codexStatus(ref, opts = {}) {
+  const threadId = codexThreadId(ref, opts);
+  if (!threadId) return null;
+  const sessionsDir = opts.sessionsDir || process.env.BC_CODEX_SESSIONS_DIR
+    || path.join(os.homedir(), '.codex', 'sessions');
+  const file = codexRolloutFile(threadId, sessionsDir);
+  if (!file) return null;
+  const text = tailRead(file, TAIL_BYTES);
+  if (text === null) return null;
+  const lines = text.split('\n');
+  let usage = null;
+  let model = null;
+  let effort = null;
+  for (let i = lines.length - 1; i >= 0 && !(usage && model); i--) {
+    const line = lines[i];
+    let doc = null;
+    if (!usage && line.includes('"token_count"')) {
+      try { doc = JSON.parse(line); } catch { continue; }
+      const p = doc && doc.payload;
+      if (p && p.type === 'token_count' && p.info && p.info.last_token_usage) usage = p;
+    } else if (!model && line.includes('"turn_context"')) {
+      try { doc = JSON.parse(line); } catch { continue; }
+      const p = doc && doc.payload;
+      if (doc.type === 'turn_context' && p && p.model) {
+        model = String(p.model);
+        if (p.effort) effort = String(p.effort);
+      }
+    }
+  }
+  if (!usage) return null;
+  const out = {
+    model,
+    contextUsed: usage.info.last_token_usage.total_tokens || 0,
+    contextWindow: usage.info.model_context_window || null,
+  };
+  if (effort) out.effort = effort;
+  const rl = codexRateLimits(usage.rate_limits);
+  if (rl) out.rateLimits = rl;
+  return out;
+}
+
+// ---------- shared slash-command surface ----------
+// The commands every status-capable harness answers; runCommand semantics:
+// /status formats status(), /compact rides the verified-submit send path
+// (the harness's OWN /compact runs in-session), /help renders this list.
+const SLASH_COMMANDS = [
+  { name: '/status', description: 'model, context usage and rate limits' },
+  { name: '/compact', description: 'compact the conversation to free context' },
+  { name: '/help', description: 'list the available commands' },
+];
+
+// Replies render as markdown in the chat thread, where a single newline
+// collapses — blank-line separators keep each line its own paragraph.
+function helpText(cmds) {
+  return cmds.map((c) => c.name + ' — ' + c.description).join('\n\n');
+}
+
+function fmtInt(n) {
+  return Number.isFinite(n) ? Math.round(n).toLocaleString('en-US') : '?';
+}
+
+function fmtWindowLabel(minutes) {
+  if (!Number.isFinite(minutes)) return 'rate';
+  if (minutes % 10080 === 0) return (minutes / 10080) + 'w';
+  if (minutes % 1440 === 0) return (minutes / 1440) + 'd';
+  if (minutes % 60 === 0) return (minutes / 60) + 'h';
+  return minutes + 'min';
+}
+
+// formatStatus(status) -> the human /status reply.
+function formatStatus(st) {
+  const modelLine = 'model: ' + (st.model || 'unknown') + (st.effort ? ' (' + st.effort + ')' : '');
+  const lines = [modelLine];
+  if (Number.isFinite(st.contextUsed) && st.contextWindow > 0) {
+    const pct = Math.round((st.contextUsed / st.contextWindow) * 100);
+    lines.push('context: ' + fmtInt(st.contextUsed) + ' / ' + fmtInt(st.contextWindow)
+      + ' tokens (' + pct + '%)');
+  } else if (Number.isFinite(st.contextUsed)) {
+    lines.push('context: ' + fmtInt(st.contextUsed) + ' tokens');
+  }
+  const rl = st.rateLimits || {};
+  for (const key of ['primary', 'secondary']) {
+    const w = rl[key];
+    if (!w) continue;
+    let line = fmtWindowLabel(w.windowMinutes) + ' limit: ' + (Number.isFinite(w.usedPercent) ? Math.round(w.usedPercent) : '?') + '% used';
+    if (Number.isFinite(w.resetsAt)) {
+      line += ' (resets ' + new Date(w.resetsAt * 1000).toISOString().replace('T', ' ').slice(0, 16) + ' UTC)';
+    }
+    lines.push(line);
+  }
+  return lines.join('\n\n');
+}
+
+module.exports = {
+  tailRead,
+  claudeProjectSlug,
+  claudeContextWindow,
+  findBridgeWorkspace,
+  claudeSidecarStatus,
+  claudeStatus,
+  codexRolloutFile,
+  codexThreadId,
+  codexStatus,
+  SLASH_COMMANDS,
+  helpText,
+  formatStatus,
+};