opencode-usage-coach 0.8.5 → 0.9.1

package/README.md

@@ -2,7 +2,7 @@
 
 A closed-loop usage coach and harness for [OpenCode](https://opencode.ai). Built for flat-rate / quota-metered coding plans — it **senses quota → coaches → stops/advances the loop**. Provider-agnostic, configurable via `harness.config.json`.
 
-[![npm version](https://img.shields.io/npm/v/opencode-usage-coach)](https://www.npmjs.com/package/opencode-usage-coach) [![license](https://img.shields.io/npm/l/opencode-usage-coach)](./LICENSE) [![coverage](https://img.shields.io/badge/coverage-planned-yellow)](#) [![Ko-fi](https://img.shields.io/badge/Ko--fi-sponsor-FF5E5B)](https://ko-fi.com/lhjnano) [![GitHub Sponsors](https://img.shields.io/badge/GitHub-Sponsors-ea4aaa)](https://github.com/sponsors/lhjnano)
+[![npm version](https://img.shields.io/npm/v/opencode-usage-coach)](https://www.npmjs.com/package/opencode-usage-coach) [![license](https://img.shields.io/npm/l/opencode-usage-coach)](./LICENSE) [![coverage](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/lhjnano/opencode-usage-coach/main/coverage-badge.json)](./coverage-badge.json) [![Ko-fi](https://img.shields.io/badge/Ko--fi-sponsor-FF5E5B)](https://ko-fi.com/lhjnano) [![GitHub Sponsors](https://img.shields.io/badge/GitHub-Sponsors-ea4aaa)](https://github.com/sponsors/lhjnano)
 
 ## Features
 

package/dist/index.js

@@ -184,6 +184,7 @@ var PLUGIN_NAME = "opencode-usage-coach";
 var TTL_MS = Number(process.env.UC_TTL_MS ?? 6e4);
 var DEFAULT_MAX_STEPS = Number(process.env.UC_MAX_STEPS ?? 30) || 30;
 var WATCHDOG_POLL_MS = Math.max(1e3, Number(process.env.UC_WATCHDOG_POLL_MS ?? 3e3) || 3e3);
+var WALL_TIMEOUT_MS = Math.max(1, Number(process.env.UC_WALL_TIMEOUT_MIN ?? 30) || 30) * 60 * 1e3;
 var DEFAULT_MAX_QUESTIONS = Math.max(1, Math.round(Number(process.env.UC_MAX_QUESTIONS ?? 7)) || 7);
 var PIPE_LOG = join2(homedir(), ".cache", "opencode-usage-coach", "pipeline.log");
 function pipeLog(msg) {
@@ -827,7 +828,7 @@ Questions for the user (${r.questions.length}):`);
   if (r.rawAnalysis) L.push(`
 Raw analysis: ${r.rawAnalysis.slice(0, 200)}`);
   L.push("\n[usage-coach NEXT] unknowns reviewed:");
-  L.push("  - If questions are flagged, ask the user first.");
+  L.push("  - If questions are flagged, call question() to present them to the user.");
   L.push("  - If task splits are suggested, adjust via task_update.");
   L.push("  - Then proceed to generate/generate_batch.");
   return L.join("\n");
@@ -870,11 +871,27 @@ function checkScanGate(sessionID) {
   try {
     const h = readHarness(sessionID);
     if (!h || !h.scanRequired) return { warning: null, summary: null };
-    if (h.scanDone) return { warning: null, summary: h.scanSummary ?? null };
-    return {
-      warning: `\u26A0 DIAGNOSIS GATE: unknown_scan was NOT called before this generate. You are generating without pre-flight gap analysis. Blind spots (unknown unknowns) may cause wrong assumptions and waste steps. Call unknown_scan first, OR proceed consciously accepting the risk.`,
-      summary: null
-    };
+    if (!h.scanDone) {
+      return {
+        warning: `\u26A0 DIAGNOSIS GATE: unknown_scan was NOT called before this generate. You are generating without pre-flight gap analysis. Blind spots (unknown unknowns) may cause wrong assumptions and waste steps. Call unknown_scan first, OR proceed consciously accepting the risk.`,
+        summary: null
+      };
+    }
+    if (h.unknownScan?.questions?.length && !h.questionsResolved) {
+      return {
+        warning: `\u26A0 UNRESOLVED QUESTIONS: unknown_scan found ${h.unknownScan.questions.length} question(s) for the user, but the question tool was not called. You MUST call question() to present these to the user before generating. The answers materially affect the approach. Do NOT proceed without asking.`,
+        summary: h.scanSummary ?? null
+      };
+    }
+    let summary = h.scanSummary ?? null;
+    if (h.questionsResolved && h.questionAnswers) {
+      const answers = Object.entries(h.questionAnswers).map(([id, ans]) => `  ${id}: ${ans}`).join("\n");
+      summary = summary ? `${summary}
+User answers to scan questions:
+${answers}` : `User answers to scan questions:
+${answers}`;
+    }
+    return { warning: null, summary };
   } catch {
     return { warning: null, summary: null };
   }
@@ -896,8 +913,10 @@ async function runModel(client, model, prompt, directory, track, maxSteps = DEFA
   const t0 = Date.now();
   const subStart = Date.now();
   let poller = null;
+  let wallTimer = null;
   let subId = null;
   let timedOut = false;
+  let pollerDone = false;
   let signalTimeout;
   const timeoutSignal = new Promise((resolve2) => {
     signalTimeout = resolve2;
@@ -912,12 +931,13 @@ async function runModel(client, model, prompt, directory, track, maxSteps = DEFA
     subId = id;
     log(`runModel(${model}): session ${id} created, sending prompt (${prompt.length} chars), max_steps=${maxSteps}`);
     poller = setInterval(async () => {
-      if (timedOut) return;
+      if (timedOut || pollerDone) return;
       try {
         let step = 0;
         let lastTs = (/* @__PURE__ */ new Date()).toISOString();
         try {
           const msgs = await client.session.messages?.({ path: { id } });
+          if (pollerDone) return;
           const msgList = Array.isArray(msgs?.data) ? msgs.data : Array.isArray(msgs) ? msgs : [];
           if (msgList.length) {
             step = msgList.filter((m) => {
@@ -940,7 +960,7 @@ async function runModel(client, model, prompt, directory, track, maxSteps = DEFA
           signalTimeout();
           return;
         }
-        if (track) {
+        if (track && !pollerDone) {
           const elapsed2 = Math.round((Date.now() - subStart) / 1e3);
           updateSubSession(track.sessionID, track.taskId, {
             subSessionId: id,
@@ -953,6 +973,17 @@ async function runModel(client, model, prompt, directory, track, maxSteps = DEFA
         log(`runModel poller err: ${String(e)}`);
       }
     }, WATCHDOG_POLL_MS);
+    wallTimer = setTimeout(() => {
+      if (!timedOut) {
+        timedOut = true;
+        log(`runModel(${model}): WALL-CLOCK timeout after ${WALL_TIMEOUT_MS / 1e3}s, aborting session ${id}`);
+        try {
+          client.session.abort?.({ path: { id } });
+        } catch {
+        }
+        signalTimeout();
+      }
+    }, WALL_TIMEOUT_MS);
     const promptP = client.session.prompt({
       path: { id },
       body: { model: { providerID, modelID }, parts: [{ type: "text", text: prompt }] }
@@ -997,7 +1028,9 @@ async function runModel(client, model, prompt, directory, track, maxSteps = DEFA
     log(`runModel err (${model}, ${elapsed}s): ${String(e)}`);
     return `ERROR: runModel exception after ${elapsed}s: ${String(e)}`;
   } finally {
+    pollerDone = true;
     if (poller) clearInterval(poller);
+    if (wallTimer) clearTimeout(wallTimer);
     if (track) {
       try {
         clearSubSession(track.sessionID, track.taskId);
@@ -1031,7 +1064,9 @@ var WORM_MAX_NODES = num("UC_WORM_MAX_NODES", 1e5);
 function humanRemaining(iso) {
   try {
     if (!iso) return "";
-    const mins = Math.floor((new Date(iso).getTime() - Date.now()) / 6e4);
+    const ms = new Date(iso).getTime();
+    if (Number.isNaN(ms)) return "";
+    const mins = Math.floor((ms - Date.now()) / 6e4);
     if (mins < 0) return "resets soon";
     if (mins < 60) return `resets in ${mins}m`;
     if (mins < 1440) return `resets in ${Math.floor(mins / 60)}h ${mins % 60}m`;
@@ -1098,6 +1133,21 @@ async function fetchProvidersCoach() {
   }));
   return results.filter(Boolean);
 }
+function parseQuotaResponse(rawText) {
+  try {
+    const text = (rawText || "").trim();
+    if (!text || text === "[]") return null;
+    const u = JSON.parse(text)[0]?.usage;
+    if (!u) return null;
+    return {
+      weekly: u.primary ?? { usedPercent: 0 },
+      monthly: u.secondary ?? { usedPercent: 0 },
+      fiveHour: u.tertiary ?? { usedPercent: 0 }
+    };
+  } catch {
+    return null;
+  }
+}
 function fetchQuota(provider) {
   return new Promise((resolve2) => {
     let out = "";
@@ -1115,21 +1165,24 @@ function fetchQuota(provider) {
     });
     p.on("error", () => resolve2(null));
     p.on("close", () => {
-      try {
-        const text = out.trim();
-        if (!text || text === "[]") return resolve2(null);
-        const u = JSON.parse(text)[0]?.usage;
-        if (!u) return resolve2(null);
-        resolve2({ weekly: u.primary ?? { usedPercent: 0 }, monthly: u.secondary ?? { usedPercent: 0 }, fiveHour: u.tertiary ?? { usedPercent: 0 } });
-      } catch {
-        resolve2(null);
-      }
+      resolve2(parseQuotaResponse(out));
     });
   });
 }
+async function fetchQuotaWithRetry(provider, maxRetries = 3) {
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    const q = await fetchQuota(provider);
+    if (q) return q;
+    if (attempt < maxRetries - 1) {
+      await new Promise((r) => setTimeout(r, 1e3 * (attempt + 1)));
+    }
+  }
+  return null;
+}
 function coach(q, lighter) {
-  if (!q) return { decision: "GO", advice: "quota unavailable \u2014 proceeding cautiously.", weekly: -1, monthly: -1, fiveHour: -1 };
+  if (!q) return { decision: "GO", advice: "quota unavailable \u2014 retrying. proceeding cautiously.", weekly: -2, monthly: -2, fiveHour: -2 };
   const wk = Math.round(q.weekly?.usedPercent ?? 0), mo = Math.round(q.monthly?.usedPercent ?? 0), h5 = Math.round(q.fiveHour?.usedPercent ?? 0);
+  if (Number.isNaN(wk) || Number.isNaN(mo) || Number.isNaN(h5)) return { decision: "THROTTLE", advice: "invalid quota data \u2014 proceeding with caution. switch to lighter model if available.", weekly: Number.isNaN(wk) ? 0 : wk, monthly: Number.isNaN(mo) ? 0 : mo, fiveHour: Number.isNaN(h5) ? 0 : h5 };
   const wkR = humanRemaining(q.weekly?.resetsAt), h5R = humanRemaining(q.fiveHour?.resetsAt);
   const stop = (r) => ({ decision: "STOP", advice: `STOP recommend \u2014 ${r}. window nearly exhausted. stop now or it will be force-blocked.`, weekly: wk, monthly: mo, fiveHour: h5 });
   const thr = (r) => ({ decision: "THROTTLE", advice: `Throttle recommend \u2014 ${r}. switch to lighter model (${lighter}) or wait for window reset.`, weekly: wk, monthly: mo, fiveHour: h5 });
@@ -1153,7 +1206,8 @@ function isFreeModel(model, provider) {
 }
 function providerToCodexbar(provider) {
   if (!provider) return "";
-  return provider.split("-")[0];
+  const first = provider.split("-")[0];
+  return first || provider;
 }
 async function resolveAgent(client, sessionID) {
   if (!sessionID) return "";
@@ -1198,6 +1252,7 @@ async function UsageCoachPlugin(input) {
     const PROVIDER = process.env.UC_PROVIDER ?? cfg0.provider ?? "";
     const LIGHTER = process.env.UC_LIGHTER_MODEL ?? cfg0.lighterModel ?? "a lighter model";
     let last = null;
+    let lastKnownQuota = null;
     let lastFetchedAt = 0;
     let refreshing = false;
     const refreshBackground = () => {
@@ -1215,9 +1270,11 @@ async function UsageCoachPlugin(input) {
           return;
         }
         const activeProvider = providerToCodexbar(currentProvider) || PROVIDER;
-        fetchQuota(activeProvider).then(async (q) => {
+        fetchQuotaWithRetry(activeProvider).then(async (q) => {
           try {
-            last = coach(q, LIGHTER);
+            if (q) lastKnownQuota = q;
+            const effectiveQ = q ?? lastKnownQuota;
+            last = coach(effectiveQ, LIGHTER);
             lastFetchedAt = Date.now();
             let providers = [];
             try {
@@ -1234,7 +1291,7 @@ async function UsageCoachPlugin(input) {
             log(`refresh-in-then err: ${String(e)}`);
           }
         }).catch((e) => {
-          log(`fetchQuota err: ${String(e)}`);
+          log(`fetchQuotaWithRetry err: ${String(e)}`);
         }).finally(() => {
           refreshing = false;
         });
@@ -1272,7 +1329,7 @@ async function UsageCoachPlugin(input) {
         const agent = await resolveAgent(input.client, _input.sessionID);
         currentAgent = agent;
         refreshBackground();
-        const harnessTools = ["unknown_scan", "generate", "generate_batch", "grade", "investigate", "verify_diagnosis", "generalize", "harness_start", "task_update", "harness_done", "record_failure", "reverse_interview"];
+        const harnessTools = ["unknown_scan", "question", "generate", "generate_batch", "grade", "investigate", "verify_diagnosis", "generalize", "harness_start", "task_update", "harness_done", "record_failure", "reverse_interview"];
         if (!harnessTools.includes(_input.tool)) return;
         if (!isHarnessAgent(agent)) {
           throw new Error(`[${PLUGIN_NAME}] '${_input.tool}' is restricted to agent mode ${JSON.stringify(HARNESS_AGENTS)} (current: ${JSON.stringify(agent || "unknown")}). Switch to that agent mode to use it.`);
@@ -1318,9 +1375,11 @@ async function UsageCoachPlugin(input) {
 \u26A0 DIAGNOSIS GATE \u2014 unknown_scan is REQUIRED before generate/generate_batch.
   unknown_scan({ prompt: "<user request>", tasks: [{id:1, title:"..."}, ...] })
   If you skip it, generate will inject a \u26A0 warning into the sub-session prompt.
-  Review the report: if QUESTIONS are flagged \u2192 ask the user first. If TASK
-  REFINEMENTS are suggested \u2192 apply via task_update. Unknown unknowns found will
-  be automatically injected into generate prompts as context.
+  Review the report: if QUESTIONS are flagged \u2192 call question() to present them
+  to the user BEFORE generate (it is enforced \u2014 generate will block with a
+  warning until answers are recorded). If TASK REFINEMENTS are suggested \u2192
+  apply via task_update. Unknown unknowns found will be automatically injected
+  into generate prompts as context.
 
 STEP LIMIT (default ${DEFAULT_MAX_STEPS}): each generate call creates a sub-session that is automatically aborted if it exceeds ${DEFAULT_MAX_STEPS} assistant steps. Before starting the loop, review each task: can it be completed in a focused, single-pass effort? If a task seems too broad (multiple files, multiple features, open-ended research), SPLIT it now into 2-3 smaller subtasks. A timeout wastes quota \u2014 split upfront.
 
@@ -1462,6 +1521,42 @@ Then: harness_done(). Follow the [usage-coach NEXT] directive each tool returns.
             return formatReport(result);
           }
         }),
+        question: tool({
+          description: "Present unknown_scan questions to the user. REQUIRED after unknown_scan finds questions \u2014 call BEFORE generate. First call (no answers) returns the questions formatted for presentation. Present them to the user, then call again with their answers.",
+          args: {
+            answers: tool.schema.record(tool.schema.string(), tool.schema.string()).optional().describe(`User's answers keyed by question ID (e.g. {"Q1": "yes", "Q2": "node"}). Omit on first call to get the questions.`)
+          },
+          async execute(args, ctx) {
+            const h = readHarness(ctx.sessionID);
+            if (!h || !h.unknownScan || !h.unknownScan.questions?.length) {
+              return "No questions from unknown_scan. [usage-coach NEXT] proceed to generate.";
+            }
+            const questions = h.unknownScan.questions;
+            if (!args.answers) {
+              const lines2 = [`unknown_scan found ${questions.length} question(s) that need user input before proceeding:
+`];
+              questions.forEach((q) => {
+                lines2.push(`[${q.id}] ${q.question}`);
+              });
+              lines2.push('\nPresent ALL of these questions to the user. Collect their answers, then call question({ answers: { "' + questions[0].id + '": "..." } }) with ALL answers.');
+              lines2.push("[usage-coach NEXT] Present the questions above to the user verbatim. After they respond, call question({answers:{...}}) to record answers, then proceed to generate.");
+              return lines2.join("\n");
+            }
+            h.questionsResolved = true;
+            h.questionAnswers = args.answers;
+            writeHarness(ctx.sessionID, h);
+            const answered = Object.keys(args.answers).length;
+            const lines = [`Questions resolved (${answered}/${questions.length} answered).
+`];
+            questions.forEach((q) => {
+              const ans = args.answers[q.id];
+              if (ans) lines.push(`  ${q.id}: ${q.question}
+  \u2192 ${ans}`);
+            });
+            lines.push("\n[usage-coach NEXT] Answers recorded. Proceed to generate \u2014 they will be injected into the sub-session prompt.");
+            return lines.join("\n");
+          }
+        }),
         task_update: tool({
           description: "Update a harness task's status on the panel. Call whenever a task transitions to generating/grading/revising/completed/failed.",
           args: {
@@ -1655,7 +1750,14 @@ Origin: ${args.task}
           args: { prompt: tool.schema.string(), max_steps: tool.schema.number().optional().describe("Maximum sub-session steps before timeout (default 30). Increase for complex tasks, decrease to fail fast on scope creep.") },
           async execute(args, ctx) {
             const cfg = readHarnessCfg(ctx.directory);
-            if (!cfg.generator) return 'ERROR: no generator model configured. Set "generator" in harness.config.json (see harness.config.example.json).';
+            if (!cfg.generator) {
+              const h = readHarness(ctx.sessionID);
+              if (h) {
+                h.active = false;
+                writeHarness(ctx.sessionID, h);
+              }
+              return 'ERROR: no generator model configured. Set "generator" in harness.config.json (see harness.config.example.json). HARNESS TERMINATED \u2014 configure a generator model, then restart the harness.';
+            }
             let decision = "GO";
             try {
               decision = current().decision;
@@ -1761,7 +1863,14 @@ ${gate.summary}
           args: { tasks: tool.schema.array(tool.schema.object({ id: tool.schema.number(), prompt: tool.schema.string() })), max_steps: tool.schema.number().optional().describe("Maximum sub-session steps per task before timeout (default 30).") },
           async execute(args, ctx) {
             const cfg = readHarnessCfg(ctx.directory);
-            if (!cfg.generator) return 'ERROR: no generator model configured. Set "generator" in harness.config.json (see harness.config.example.json).';
+            if (!cfg.generator) {
+              const h = readHarness(ctx.sessionID);
+              if (h) {
+                h.active = false;
+                writeHarness(ctx.sessionID, h);
+              }
+              return 'ERROR: no generator model configured. Set "generator" in harness.config.json (see harness.config.example.json). HARNESS TERMINATED \u2014 configure a generator model, then restart the harness.';
+            }
             let decision = "GO";
             try {
               decision = current().decision;
@@ -2039,5 +2148,28 @@ If the task is already well-specified with no significant ambiguities, return {"
   }
 }
 export {
-  UsageCoachPlugin as default
+  buildGapPrompt,
+  buildScanSummary,
+  checkScanGate,
+  clearSubSession,
+  coach,
+  UsageCoachPlugin as default,
+  detectLanguage,
+  extractImplNotes,
+  extractKeywords,
+  findActiveTaskId,
+  formatReport,
+  humanRemaining,
+  isFreeModel,
+  isHarnessAgent,
+  parseFileList,
+  parseGapAnalysis,
+  parseQuotaResponse,
+  providerAdvice,
+  providerToCodexbar,
+  readHarness,
+  readRules,
+  setStateDir,
+  updateSubSession,
+  writeHarness
 };

package/dist/tui.js

@@ -10,6 +10,101 @@ import { createHash } from "crypto";
 import { homedir } from "os";
 import { join, resolve } from "path";
 import { createRoot, createSignal, onCleanup } from "solid-js";
+
+// src/tui-logic.ts
+var STALE_MS = 5 * 6e4;
+var HIDE_MS = 30 * 6e4;
+var TAG = {
+  GO: "ok",
+  THROTTLE: "slow",
+  STOP: "STOP"
+};
+var TLABEL = {
+  generating: "gen",
+  grading: "grade",
+  revising: "revise",
+  completed: "done",
+  failed: "fail",
+  timed_out: "timeout",
+  halted_quota: "quota-halt",
+  stale: "STALE"
+};
+var STATUS_KEY = {
+  generating: "info",
+  grading: "accent",
+  revising: "warning",
+  completed: "success",
+  failed: "error",
+  timed_out: "error",
+  halted_quota: "error"
+};
+var TERMINAL_STATUSES = /* @__PURE__ */ new Set([
+  "completed",
+  "failed",
+  "timed_out",
+  "halted_quota"
+]);
+function barFill(p) {
+  const n = !Number.isFinite(p) || p <= 0 ? 0 : Math.max(1, Math.min(10, Math.round(p / 10)));
+  return "\u2588".repeat(n);
+}
+function barEmpty(p) {
+  const n = !Number.isFinite(p) || p <= 0 ? 0 : Math.max(1, Math.min(10, Math.round(p / 10)));
+  return "\u2591".repeat(10 - n);
+}
+function computeStaleness(h, now = Date.now()) {
+  const hAge = h.updatedAt ? now - new Date(h.updatedAt).getTime() : 0;
+  const hasActiveSub = h.tasks.some((t) => !!t.subSessionId);
+  const isStale = !hasActiveSub && hAge > STALE_MS;
+  const shouldHide = hAge > HIDE_MS && !hasActiveSub;
+  return { hAge, hasActiveSub, isStale, shouldHide };
+}
+function isHarnessVisible(h, staleness) {
+  if (!h) return false;
+  if (h.tasks.length === 0) return false;
+  if (h.active !== true) return false;
+  if (staleness.shouldHide) return false;
+  return true;
+}
+function computeTaskDisplay(t, isStale, now = Date.now()) {
+  const displayStatus = isStale && !TERMINAL_STATUSES.has(t.status) ? "stale" : t.status;
+  const sKey = STATUS_KEY[displayStatus] ?? "text";
+  const lbl = TLABEL[displayStatus] ?? displayStatus;
+  const rev = (t.revisions ?? 0) > 0 && t.status === "revising" ? `(${t.revisions})` : "";
+  const mdl = t.model ? ` ${t.model.split("/").pop() ?? t.model}` : "";
+  const hasSub = !!t.subSessionId;
+  const subStepStr = hasSub && t.subStep !== void 0 && t.subStep > 0 ? ` step:${t.subStep}` : "";
+  const subEl = hasSub && t.subElapsed !== void 0 ? ` ${t.subElapsed}s` : "";
+  const subWarn = hasSub && (t.subElapsed ?? 0) > 300;
+  const elapsed = t.startedAt ? Math.max(0, Math.round((now - new Date(t.startedAt).getTime()) / 1e3)) : 0;
+  const taskEl = t.status === "completed" || t.status === "failed" ? "" : elapsed > 0 ? ` ${elapsed}s` : "";
+  const displayEl = hasSub ? subEl : taskEl;
+  const lineKey = subWarn ? "warning" : sKey;
+  return {
+    displayStatus,
+    themeKey: lineKey,
+    label: lbl,
+    revSuffix: rev,
+    modelStr: mdl,
+    stepStr: subStepStr,
+    elapsedStr: displayEl,
+    hasSub,
+    subWarn
+  };
+}
+function decisionThemeKey(decision) {
+  return decision === "GO" ? "success" : decision === "THROTTLE" ? "warning" : "error";
+}
+function taskQuotaPct(t, s) {
+  const pv = t.model ? (t.model.split("/")[0] ?? "").split("-")[0] : "";
+  const provCoach = pv ? s?.providers?.find((p) => p.id === pv || pv && p.id.startsWith(pv) || pv && pv.startsWith(p.id)) : s?.providers?.[0];
+  const rawPct = provCoach?.fiveHour ?? s?.fiveHour ?? -1;
+  const pct = rawPct < 0 ? 0 : rawPct;
+  const label = rawPct === -1 ? "\u2026" : rawPct < 0 ? "retry" : `${rawPct}%`;
+  return { pct, label };
+}
+
+// src/tui.tsx
 function projectStateDir(dir) {
   const abs = resolve(dir || ".");
   const h = createHash("sha1").update(abs).digest("hex").slice(0, 12);
@@ -71,31 +166,6 @@ function readHarness() {
     return null;
   }
 }
-var TAG = {
-  GO: "ok",
-  THROTTLE: "slow",
-  STOP: "STOP"
-};
-var TLABEL = {
-  generating: "gen",
-  grading: "grade",
-  revising: "revise",
-  completed: "done",
-  failed: "fail",
-  timed_out: "timeout",
-  halted_quota: "quota-halt",
-  stale: "STALE"
-};
-var STALE_MS = 5 * 6e4;
-var HIDE_MS = 30 * 6e4;
-function barFill(p) {
-  const n = p <= 0 ? 0 : Math.max(1, Math.min(10, Math.round(p / 10)));
-  return "\u2588".repeat(n);
-}
-function barEmpty(p) {
-  const n = p <= 0 ? 0 : Math.max(1, Math.min(10, Math.round(p / 10)));
-  return "\u2591".repeat(10 - n);
-}
 function initializeTui(api, disposeRoot) {
   STATE_DIR = process.env.UC_STATE_DIR ?? projectStateDir(api.state.path.directory);
   STATE_FILE = join(STATE_DIR, "state.json");
@@ -162,15 +232,6 @@ function initializeTui(api, disposeRoot) {
     } catch {
     }
   });
-  const statusKey = {
-    generating: "info",
-    grading: "accent",
-    revising: "warning",
-    completed: "success",
-    failed: "error",
-    timed_out: "error",
-    halted_quota: "error"
-  };
   const panel = (ctx) => {
     const th = ctx.theme?.current ?? {};
     const st = (k) => ({
@@ -206,7 +267,7 @@ function initializeTui(api, disposeRoot) {
     }
     const nodes = [];
     if (s) {
-      const dKey = s.decision === "GO" ? "success" : s.decision === "THROTTLE" ? "warning" : "error";
+      const dKey = decisionThemeKey(s.decision);
       const modelShort = s.model ? s.model.split("/").pop() ?? s.model : "";
       if (s.isFree) {
         nodes.push((() => {
@@ -316,7 +377,7 @@ function initializeTui(api, disposeRoot) {
           }
         } else {
           nodes.push((() => {
-            var _el$32 = _$createElement("box"), _el$33 = _$createElement("text"), _el$35 = _$createElement("text"), _el$36 = _$createElement("text"), _el$37 = _$createElement("text");
+            var _el$32 = _$createElement("box"), _el$33 = _$createElement("text"), _el$35 = _$createElement("text"), _el$36 = _$createElement("text"), _el$37 = _$createElement("text"), _el$38 = _$createTextNode(` `), _el$39 = _$createTextNode(`%`);
             _$insertNode(_el$32, _el$33);
             _$insertNode(_el$32, _el$35);
             _$insertNode(_el$32, _el$36);
@@ -325,7 +386,9 @@ function initializeTui(api, disposeRoot) {
             _$insertNode(_el$33, _$createTextNode(` 5h `));
             _$insert(_el$35, () => barFill(s.fiveHour));
             _$insert(_el$36, () => barEmpty(s.fiveHour));
-            _$insertNode(_el$37, _$createTextNode(` 0%`));
+            _$insertNode(_el$37, _el$38);
+            _$insertNode(_el$37, _el$39);
+            _$insert(_el$37, () => s.fiveHour, _el$39);
             _$effect((_p$) => {
               var _v$9 = st("text"), _v$0 = st("text");
               _v$9 !== _p$.e && (_p$.e = _$setProp(_el$35, "style", _v$9, _p$.e));
@@ -338,126 +401,111 @@ function initializeTui(api, disposeRoot) {
             return _el$32;
           })());
           nodes.push((() => {
-            var _el$39 = _$createElement("box"), _el$40 = _$createElement("text"), _el$42 = _$createElement("text"), _el$43 = _$createElement("text"), _el$44 = _$createElement("text");
-            _$insertNode(_el$39, _el$40);
-            _$insertNode(_el$39, _el$42);
-            _$insertNode(_el$39, _el$43);
-            _$insertNode(_el$39, _el$44);
-            _$setProp(_el$39, "flexDirection", "row");
-            _$insertNode(_el$40, _$createTextNode(` 1w `));
-            _$insert(_el$42, () => barFill(s.weekly));
-            _$insert(_el$43, () => barEmpty(s.weekly));
-            _$insertNode(_el$44, _$createTextNode(` 0%`));
+            var _el$40 = _$createElement("box"), _el$41 = _$createElement("text"), _el$43 = _$createElement("text"), _el$44 = _$createElement("text"), _el$45 = _$createElement("text"), _el$46 = _$createTextNode(` `), _el$47 = _$createTextNode(`%`);
+            _$insertNode(_el$40, _el$41);
+            _$insertNode(_el$40, _el$43);
+            _$insertNode(_el$40, _el$44);
+            _$insertNode(_el$40, _el$45);
+            _$setProp(_el$40, "flexDirection", "row");
+            _$insertNode(_el$41, _$createTextNode(` 1w `));
+            _$insert(_el$43, () => barFill(s.weekly));
+            _$insert(_el$44, () => barEmpty(s.weekly));
+            _$insertNode(_el$45, _el$46);
+            _$insertNode(_el$45, _el$47);
+            _$insert(_el$45, () => s.weekly, _el$47);
             _$effect((_p$) => {
               var _v$1 = st("text"), _v$10 = st("text");
-              _v$1 !== _p$.e && (_p$.e = _$setProp(_el$42, "style", _v$1, _p$.e));
-              _v$10 !== _p$.t && (_p$.t = _$setProp(_el$43, "style", _v$10, _p$.t));
+              _v$1 !== _p$.e && (_p$.e = _$setProp(_el$43, "style", _v$1, _p$.e));
+              _v$10 !== _p$.t && (_p$.t = _$setProp(_el$44, "style", _v$10, _p$.t));
               return _p$;
             }, {
               e: void 0,
               t: void 0
             });
-            return _el$39;
+            return _el$40;
           })());
         }
       }
     } else {
       nodes.push((() => {
-        var _el$46 = _$createElement("text");
-        _$insertNode(_el$46, _$createTextNode(`usage-coach: ...`));
-        return _el$46;
+        var _el$48 = _$createElement("text");
+        _$insertNode(_el$48, _$createTextNode(`usage-coach: ...`));
+        return _el$48;
       })());
     }
-    if (h && h.tasks.length > 0 && h.active !== false) {
-      const hAge = h.updatedAt ? Date.now() - new Date(h.updatedAt).getTime() : 0;
-      const hasActiveSub = h.tasks.some((t) => !!t.subSessionId);
-      const isStale = !hasActiveSub && hAge > STALE_MS;
-      const shouldHide = hAge > HIDE_MS && !hasActiveSub;
-      if (shouldHide) {
-      } else {
+    if (h) {
+      const staleness = computeStaleness(h);
+      if (isHarnessVisible(h, staleness)) {
+        const isStale = staleness.isStale;
         nodes.push((() => {
-          var _el$48 = _$createElement("text");
-          _$insertNode(_el$48, _$createTextNode(` `));
-          return _el$48;
+          var _el$50 = _$createElement("text");
+          _$insertNode(_el$50, _$createTextNode(` `));
+          return _el$50;
         })());
         nodes.push((() => {
-          var _el$50 = _$createElement("text"), _el$51 = _$createTextNode(`harness: `), _el$52 = _$createTextNode(` `), _el$53 = _$createTextNode(`/`);
-          _$insertNode(_el$50, _el$51);
-          _$insertNode(_el$50, _el$52);
-          _$insertNode(_el$50, _el$53);
-          _$insert(_el$50, () => h.name, _el$52);
-          _$insert(_el$50, () => h.current, _el$53);
-          _$insert(_el$50, () => h.total, null);
-          _$insert(_el$50, isStale ? " (stale)" : "", null);
-          _$effect((_$p) => _$setProp(_el$50, "style", st("textMuted"), _$p));
-          return _el$50;
+          var _el$52 = _$createElement("text"), _el$53 = _$createTextNode(`harness: `), _el$54 = _$createTextNode(` `), _el$55 = _$createTextNode(`/`);
+          _$insertNode(_el$52, _el$53);
+          _$insertNode(_el$52, _el$54);
+          _$insertNode(_el$52, _el$55);
+          _$insert(_el$52, () => h.name, _el$54);
+          _$insert(_el$52, () => h.current, _el$55);
+          _$insert(_el$52, () => h.total, null);
+          _$insert(_el$52, isStale ? " (stale)" : "", null);
+          _$effect((_$p) => _$setProp(_el$52, "style", st("textMuted"), _$p));
+          return _el$52;
         })());
         for (const t of h.tasks) {
-          const TERMINAL = /* @__PURE__ */ new Set(["completed", "failed", "timed_out", "halted_quota"]);
-          const displayStatus = isStale && !TERMINAL.has(t.status) ? "stale" : t.status;
-          const sKey = statusKey[displayStatus] ?? "text";
-          const lbl = TLABEL[displayStatus] ?? displayStatus;
-          const rev = t.revisions > 0 && t.status === "revising" ? `(${t.revisions})` : "";
-          const mdl = t.model ? ` ${t.model.split("/").pop() ?? t.model}` : "";
-          const hasSub = !!t.subSessionId;
-          const subStepStr = hasSub && t.subStep !== void 0 && t.subStep > 0 ? ` step:${t.subStep}` : "";
-          const subEl = hasSub && t.subElapsed !== void 0 ? ` ${t.subElapsed}s` : "";
-          const subWarn = hasSub && (t.subElapsed ?? 0) > 300;
-          const elapsed = t.startedAt ? Math.max(0, Math.round((Date.now() - new Date(t.startedAt).getTime()) / 1e3)) : 0;
-          const taskEl = t.status === "completed" || t.status === "failed" ? "" : elapsed > 0 ? ` ${elapsed}s` : "";
-          const displayEl = hasSub ? subEl : taskEl;
-          const lineKey = subWarn ? "warning" : sKey;
+          const td = computeTaskDisplay(t, isStale);
           nodes.push((() => {
-            var _el$54 = _$createElement("text"), _el$55 = _$createTextNode(` \u25CF `), _el$56 = _$createTextNode(` `), _el$57 = _$createTextNode(` `);
-            _$insertNode(_el$54, _el$55);
-            _$insertNode(_el$54, _el$56);
-            _$insertNode(_el$54, _el$57);
-            _$insert(_el$54, () => t.id, _el$56);
-            _$insert(_el$54, mdl, _el$56);
-            _$insert(_el$54, lbl, _el$57);
-            _$insert(_el$54, rev, _el$57);
-            _$insert(_el$54, subStepStr, _el$57);
-            _$insert(_el$54, displayEl, _el$57);
-            _$insert(_el$54, () => t.title, null);
-            _$effect((_$p) => _$setProp(_el$54, "style", st(lineKey), _$p));
-            return _el$54;
+            var _el$56 = _$createElement("text"), _el$57 = _$createTextNode(` \u25CF `), _el$58 = _$createTextNode(` `), _el$59 = _$createTextNode(` `);
+            _$insertNode(_el$56, _el$57);
+            _$insertNode(_el$56, _el$58);
+            _$insertNode(_el$56, _el$59);
+            _$insert(_el$56, () => t.id, _el$58);
+            _$insert(_el$56, () => td.modelStr, _el$58);
+            _$insert(_el$56, () => td.label, _el$59);
+            _$insert(_el$56, () => td.revSuffix, _el$59);
+            _$insert(_el$56, () => td.stepStr, _el$59);
+            _$insert(_el$56, () => td.elapsedStr, _el$59);
+            _$insert(_el$56, () => t.title, null);
+            _$effect((_$p) => _$setProp(_el$56, "style", st(td.themeKey), _$p));
+            return _el$56;
           })());
-          const pv = t.model ? (t.model.split("/")[0] ?? "").split("-")[0] : "";
-          const provCoach = pv ? s?.providers?.find((p) => p.id === pv || pv && p.id.startsWith(pv) || pv && pv.startsWith(p.id)) : s?.providers?.[0];
-          const rawPct = provCoach?.fiveHour ?? s?.fiveHour ?? -1;
-          const pct = rawPct < 0 ? 0 : rawPct;
-          const pctLabel = rawPct < 0 ? "n/a" : `${rawPct}%`;
+          const {
+            pct,
+            label: pctLabel
+          } = taskQuotaPct(t, s);
           nodes.push((() => {
-            var _el$58 = _$createElement("box"), _el$59 = _$createElement("text"), _el$61 = _$createElement("text"), _el$62 = _$createElement("text"), _el$63 = _$createElement("text"), _el$64 = _$createTextNode(` `);
-            _$insertNode(_el$58, _el$59);
-            _$insertNode(_el$58, _el$61);
-            _$insertNode(_el$58, _el$62);
-            _$insertNode(_el$58, _el$63);
-            _$setProp(_el$58, "flexDirection", "row");
-            _$insertNode(_el$59, _$createTextNode(` 5h `));
-            _$insert(_el$61, () => barFill(pct));
-            _$insert(_el$62, () => barEmpty(pct));
-            _$insertNode(_el$63, _el$64);
-            _$insert(_el$63, pctLabel, null);
+            var _el$60 = _$createElement("box"), _el$61 = _$createElement("text"), _el$63 = _$createElement("text"), _el$64 = _$createElement("text"), _el$65 = _$createElement("text"), _el$66 = _$createTextNode(` `);
+            _$insertNode(_el$60, _el$61);
+            _$insertNode(_el$60, _el$63);
+            _$insertNode(_el$60, _el$64);
+            _$insertNode(_el$60, _el$65);
+            _$setProp(_el$60, "flexDirection", "row");
+            _$insertNode(_el$61, _$createTextNode(` 5h `));
+            _$insert(_el$63, () => barFill(pct));
+            _$insert(_el$64, () => barEmpty(pct));
+            _$insertNode(_el$65, _el$66);
+            _$insert(_el$65, pctLabel, null);
             _$effect((_p$) => {
               var _v$11 = st("text"), _v$12 = st("text");
-              _v$11 !== _p$.e && (_p$.e = _$setProp(_el$61, "style", _v$11, _p$.e));
-              _v$12 !== _p$.t && (_p$.t = _$setProp(_el$62, "style", _v$12, _p$.t));
+              _v$11 !== _p$.e && (_p$.e = _$setProp(_el$63, "style", _v$11, _p$.e));
+              _v$12 !== _p$.t && (_p$.t = _$setProp(_el$64, "style", _v$12, _p$.t));
               return _p$;
             }, {
               e: void 0,
               t: void 0
             });
-            return _el$58;
+            return _el$60;
           })());
         }
       }
     }
     return (() => {
-      var _el$65 = _$createElement("box");
-      _$setProp(_el$65, "flexDirection", "column");
-      _$insert(_el$65, nodes);
-      return _el$65;
+      var _el$67 = _$createElement("box");
+      _$setProp(_el$67, "flexDirection", "column");
+      _$insert(_el$67, nodes);
+      return _el$67;
     })();
   };
   tlog("registering slots");
@@ -472,9 +520,9 @@ function initializeTui(api, disposeRoot) {
         } catch (e) {
           tlog(`sidebar_footer err: ${String(e)}`);
           result = (() => {
-            var _el$66 = _$createElement("text");
-            _$insertNode(_el$66, _$createTextNode(`usage-coach`));
-            return _el$66;
+            var _el$68 = _$createElement("text");
+            _$insertNode(_el$68, _$createTextNode(`usage-coach`));
+            return _el$68;
           })();
         }
         return result;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-usage-coach",
-  "version": "0.8.5",
+  "version": "0.9.1",
   "description": "opencode closed-loop usage coach — quota SENSE -> coaching DECIDE -> loop ACT + TUI integration",
   "type": "module",
   "main": "./dist/index.js",
@@ -20,6 +20,9 @@
     "typecheck": "tsc --noEmit",
     "lint": "eslint .",
     "lint:fix": "eslint . --fix",
+    "test": "node --import tsx --test test/**/*.test.ts",
+    "test:coverage": "c8 --reporter=text --reporter=lcov --reporter=json-summary node --import tsx --test test/**/*.test.ts",
+    "coverage:badge": "node scripts/coverage-badge.mjs",
     "prepack": "tsup"
   },
   "files": [
@@ -50,12 +53,15 @@
     "@opencode-ai/plugin": "*",
     "@opentui/core": ">=0.4.0",
     "@opentui/solid": ">=0.4.0",
+    "@types/node": "^26.1.1",
+    "c8": "^11.0.0",
     "esbuild-plugin-solid": "^0.6.0",
     "eslint": "^10.6.0",
     "eslint-plugin-solid": "^0.14.5",
     "globals": "^17.7.0",
     "solid-js": "^1.9",
     "tsup": "^8.5",
+    "tsx": "^4.23.0",
     "typescript": "^5",
     "typescript-eslint": "^8.63.0"
   }