/** * gbrain-local-status — classify the local gbrain engine into 5 states. * * Shared between bin/gstack-gbrain-detect (preamble probe on every skill start) * and bin/gstack-gbrain-sync.ts (orchestrator SKIP-when-not-ok semantics). * Single source of truth: same probe, same classification, same cache. * * Per the split-engine plan (D2 + D8): * - Probe: `gbrain sources list --json`. Cheap (~80ms), actually hits the DB. * Uses the same stderr patterns as lib/gbrain-sources.ts:66-67. * - Cache: 60s TTL at ~/.gstack/.gbrain-local-status-cache.json, keyed on * {home, path_hash, gbrain_bin_path, gbrain_version, config_mtime}. * - --no-cache bypass: /setup-gbrain and /sync-gbrain pass it after any * state-mutating operation so the next read sees fresh status. * * No-cli → gbrain not on PATH. * Missing → CLI present, ~/.gbrain/config.json absent. * Broken-config → config exists but `gbrain sources list` fails with config parse error * (or any non-recognized error — defensive default per codex #8). * Broken-db → config exists, DB unreachable per stderr classification. * Ok → DB reachable, sources list returned valid JSON. */ import { execFileSync } from "child_process"; import { createHash, } from "crypto"; import { existsSync, mkdirSync, readFileSync, renameSync, statSync, writeFileSync, } from "fs"; import { homedir } from "os"; import { dirname, join } from "path"; export type LocalEngineStatus = | "ok" | "no-cli" | "missing-config" | "broken-config" | "broken-db"; export interface ClassifyOptions { /** Bypass the 60s cache. Used after any state-mutating operation. */ noCache?: boolean; /** Env override for the spawned `gbrain` (used by tests to point at a fake binary). */ env?: NodeJS.ProcessEnv; } interface CacheEntry { schema_version: 1; status: LocalEngineStatus; cached_at: number; /** Cache invariants — entry is invalidated if any of these change between writes. */ key: { home: string; path_hash: string; gbrain_bin_path: string; gbrain_version: string; config_mtime: number; // 0 when config absent config_size: number; // 0 when config absent }; } export const CACHE_TTL_MS = 60_000; export const PROBE_TIMEOUT_MS = 5_000; /** Effective user home — respects HOME env override (used by tests). */ function userHome(): string { return process.env.HOME || homedir(); } /** Cache path computed fresh on each call so tests can mutate GSTACK_HOME per case. */ export function cacheFilePath(): string { return join( process.env.GSTACK_HOME || join(userHome(), ".gstack"), ".gbrain-local-status-cache.json", ); } function gbrainConfigPath(): string { return join(userHome(), ".gbrain", "config.json"); } function hashPath(p: string): string { return createHash("sha256").update(p).digest("hex").slice(0, 16); } /** * Resolve the absolute path of `gbrain` on PATH. Returns null when missing. * Memoized per-process keyed on PATH so detect's call and the classifier's * call share one fork-exec (~200ms saved per skill preamble). */ const _gbrainBinCache = new Map(); export function resolveGbrainBin(env?: NodeJS.ProcessEnv): string | null { const e = env ?? process.env; const key = e.PATH || ""; if (_gbrainBinCache.has(key)) return _gbrainBinCache.get(key)!; let result: string | null = null; try { const out = execFileSync("sh", ["-c", "command -v gbrain"], { encoding: "utf-8", timeout: 2_000, stdio: ["ignore", "pipe", "ignore"], env: e, }); result = out.trim() || null; } catch { result = null; } _gbrainBinCache.set(key, result); return result; } /** Memoized per-process. */ const _gbrainVersionCache = new Map(); export function readGbrainVersion(env?: NodeJS.ProcessEnv): string { const e = env ?? process.env; const key = `${e.PATH || ""}|${resolveGbrainBin(e) || ""}`; if (_gbrainVersionCache.has(key)) return _gbrainVersionCache.get(key)!; let result = ""; try { const out = execFileSync("gbrain", ["--version"], { encoding: "utf-8", timeout: 2_000, stdio: ["ignore", "pipe", "ignore"], env: e, }); result = out.trim().split("\n")[0] || ""; } catch { result = ""; } _gbrainVersionCache.set(key, result); return result; } function configFingerprint(): { mtime: number; size: number } { try { const st = statSync(gbrainConfigPath()); return { mtime: Math.floor(st.mtimeMs), size: st.size }; } catch { return { mtime: 0, size: 0 }; } } function buildCacheKey( gbrainBin: string | null, gbrainVersion: string, env?: NodeJS.ProcessEnv, ): CacheEntry["key"] { const e = env ?? process.env; const config = configFingerprint(); return { home: e.HOME || "", path_hash: hashPath(e.PATH || ""), gbrain_bin_path: gbrainBin || "", gbrain_version: gbrainVersion, config_mtime: config.mtime, config_size: config.size, }; } function keysEqual(a: CacheEntry["key"], b: CacheEntry["key"]): boolean { return ( a.home === b.home && a.path_hash === b.path_hash && a.gbrain_bin_path === b.gbrain_bin_path && a.gbrain_version === b.gbrain_version && a.config_mtime === b.config_mtime && a.config_size === b.config_size ); } function readCache(key: CacheEntry["key"]): LocalEngineStatus | null { if (!existsSync(cacheFilePath())) return null; try { const raw = JSON.parse(readFileSync(cacheFilePath(), "utf-8")) as CacheEntry; if (raw.schema_version !== 1) return null; if (Date.now() - raw.cached_at > CACHE_TTL_MS) return null; if (!keysEqual(raw.key, key)) return null; return raw.status; } catch { return null; } } function writeCache(status: LocalEngineStatus, key: CacheEntry["key"]): void { const entry: CacheEntry = { schema_version: 1, status, cached_at: Date.now(), key, }; try { mkdirSync(dirname(cacheFilePath()), { recursive: true }); const tmp = cacheFilePath() + ".tmp." + process.pid; writeFileSync(tmp, JSON.stringify(entry, null, 2), "utf-8"); renameSync(tmp, cacheFilePath()); } catch { // Cache write failure is non-fatal — we re-probe next call. } } /** * Probe via `gbrain sources list --json`. Classify the outcome. * * Pattern strings ("Cannot connect to database", "config.json") are deliberately * the same strings used in lib/gbrain-sources.ts:66-67. If gbrain reworks its * error messages, classifier returns broken-config defensively (codex #8). */ function freshClassify(env?: NodeJS.ProcessEnv): LocalEngineStatus { // 1. CLI on PATH? const gbrainBin = resolveGbrainBin(env); if (!gbrainBin) return "no-cli"; // 2. Config file present? if (!existsSync(gbrainConfigPath())) return "missing-config"; // 3. Probe gbrain sources list. try { execFileSync("gbrain", ["sources", "list", "--json"], { encoding: "utf-8", timeout: PROBE_TIMEOUT_MS, stdio: ["ignore", "pipe", "pipe"], env: env ?? process.env, }); return "ok"; } catch (err) { const e = err as NodeJS.ErrnoException & { stderr?: Buffer | string }; const stderr = (e.stderr ? e.stderr.toString() : "") || ""; // ENOENT can happen if gbrain disappeared between resolveGbrainBin and now. if (e.code === "ENOENT") return "no-cli"; // Pattern match against gbrain's known error strings. Order matters: // "Cannot connect to database" is the more specific DB-unreachable signal. if (stderr.includes("Cannot connect to database")) return "broken-db"; if (stderr.includes("config.json")) return "broken-config"; // Defensive default per codex #8: unrecognized failures classify as // broken-config so the user sees the raw stderr surfaced upstream. return "broken-config"; } } /** * Classify the local gbrain engine status. Cached for 60s; bypassable. * * Returns one of 5 states. Never throws — failure modes are surfaced as states. */ export function localEngineStatus(opts: ClassifyOptions = {}): LocalEngineStatus { const env = opts.env ?? process.env; const gbrainBin = resolveGbrainBin(env); const gbrainVersion = gbrainBin ? readGbrainVersion(env) : ""; const key = buildCacheKey(gbrainBin, gbrainVersion, env); if (!opts.noCache) { const cached = readCache(key); if (cached) return cached; } const fresh = freshClassify(env); writeCache(fresh, key); return fresh; }