opencode-usage-coach 0.1.1 → 0.1.2

package/dist/index.js

@@ -1,16 +1,28 @@
 // src/index.ts
 import { mkdirSync, writeFileSync, appendFileSync, readFileSync, existsSync } from "fs";
 import { spawn } from "child_process";
+import { createHash } from "crypto";
 import { homedir } from "os";
-import { join } from "path";
+import { join, resolve } from "path";
 import { tool } from "@opencode-ai/plugin";
 var PLUGIN_NAME = "opencode-usage-coach";
-var STATE_DIR = process.env.UC_STATE_DIR ?? join(homedir(), ".cache", "opencode-usage-coach");
+var DEBUG = process.env.UC_DEBUG === "1";
+var TTL_MS = Number(process.env.UC_TTL_MS ?? 6e4);
+var STATE_DIR = join(homedir(), ".cache", "opencode-usage-coach");
 var STATE_FILE = join(STATE_DIR, "state.json");
 var HARNESS_FILE = join(STATE_DIR, "harness.json");
 var LOG_FILE = join(STATE_DIR, "coach.log");
-var DEBUG = process.env.UC_DEBUG === "1";
-var TTL_MS = Number(process.env.UC_TTL_MS ?? 6e4);
+function projectStateDir(dir) {
+  const abs = resolve(dir || ".");
+  const h = createHash("sha1").update(abs).digest("hex").slice(0, 12);
+  return join(homedir(), ".cache", "opencode-usage-coach", "projects", h);
+}
+function setStateDir(dir) {
+  STATE_DIR = process.env.UC_STATE_DIR ?? projectStateDir(dir);
+  STATE_FILE = join(STATE_DIR, "state.json");
+  HARNESS_FILE = join(STATE_DIR, "harness.json");
+  LOG_FILE = join(STATE_DIR, "coach.log");
+}
 var NOOP_HOOKS = {};
 function log(msg) {
   if (DEBUG) try {
@@ -63,14 +75,72 @@ function humanRemaining(iso) {
     const mins = Math.floor((new Date(iso).getTime() - 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`;
-    return `resets in ${Math.floor(mins / 1440)}d`;
+    if (mins < 1440) return `resets in ${Math.floor(mins / 60)}h ${mins % 60}m`;
+    return `${Math.floor(mins / 1440)}d left`;
   } catch {
     return "";
   }
 }
+function captureStdout(args) {
+  return new Promise((resolve2) => {
+    let out = "";
+    let p;
+    try {
+      p = spawn("codexbar", args, { stdio: ["ignore", "pipe", "ignore"] });
+    } catch {
+      return resolve2("");
+    }
+    p.stdout?.on("data", (d) => {
+      out += d.toString();
+    });
+    p.on("error", () => resolve2(""));
+    p.on("close", () => resolve2(out));
+  });
+}
+async function fetchEnabledProviders() {
+  const out = await captureStdout(["config", "providers"]);
+  const ids = [];
+  for (const line of out.split("\n")) {
+    const m = line.match(/^\s*([a-zA-Z0-9_-]+):\s*enabled/);
+    if (m) ids.push(m[1]);
+  }
+  return ids;
+}
+function providerAdvice(h5, wk) {
+  const S5H = STOP_5H, SWK = STOP_WK, T5H = THR_5H, TWK = THR_WK;
+  if (h5 >= S5H || wk >= SWK) return "STOP \u2014 finish current only";
+  if (h5 >= T5H && wk >= TWK) return "small tasks only \u2014 big ones will hit both limits";
+  if (h5 >= T5H) return "small tasks only \u2014 5h window nearly full, big tasks after reset";
+  if (wk >= TWK) return "small tasks only \u2014 big ones will strain late-week";
+  if (h5 >= 50 || wk >= 50) return "moderate tasks OK \u2014 save big ones for headroom";
+  return "big tasks OK \u2014 short & long limits comfortable";
+}
+async function fetchProvidersCoach() {
+  const ids = await fetchEnabledProviders();
+  const results = await Promise.all(ids.map(async (id) => {
+    try {
+      const out = await captureStdout(["usage", "--provider", id, "--json"]);
+      const u = JSON.parse(out)[0]?.usage;
+      if (!u) return null;
+      const h5 = Math.round(u.tertiary?.usedPercent ?? 0);
+      const wk = Math.round(u.primary?.usedPercent ?? 0);
+      return {
+        id,
+        name: id,
+        fiveHour: h5,
+        weekly: wk,
+        fiveHourReset: humanRemaining(u.tertiary?.resetsAt),
+        weeklyReset: humanRemaining(u.primary?.resetsAt),
+        advice: providerAdvice(h5, wk)
+      };
+    } catch {
+      return null;
+    }
+  }));
+  return results.filter(Boolean);
+}
 function fetchQuota() {
-  return new Promise((resolve) => {
+  return new Promise((resolve2) => {
     let out = "";
     let p;
     try {
@@ -78,21 +148,21 @@ function fetchQuota() {
         stdio: ["ignore", "pipe", "ignore"]
       });
     } catch {
-      return resolve(null);
+      return resolve2(null);
     }
     p.stdout?.on("data", (d) => {
       out += d.toString();
     });
-    p.on("error", () => resolve(null));
+    p.on("error", () => resolve2(null));
     p.on("close", () => {
       try {
         const text = out.trim();
-        if (!text || text === "[]") return resolve(null);
+        if (!text || text === "[]") return resolve2(null);
         const u = JSON.parse(text)[0]?.usage;
-        if (!u) return resolve(null);
-        resolve({ weekly: u.primary ?? { usedPercent: 0 }, monthly: u.secondary ?? { usedPercent: 0 }, fiveHour: u.tertiary ?? { usedPercent: 0 } });
+        if (!u) return resolve2(null);
+        resolve2({ weekly: u.primary ?? { usedPercent: 0 }, monthly: u.secondary ?? { usedPercent: 0 }, fiveHour: u.tertiary ?? { usedPercent: 0 } });
       } catch {
-        resolve(null);
+        resolve2(null);
       }
     });
   });
@@ -113,6 +183,7 @@ function coach(q) {
 var LOADING = { decision: "GO", advice: "quota loading\u2026", weekly: -1, monthly: -1, fiveHour: -1 };
 async function UsageCoachPlugin(input) {
   try {
+    setStateDir(input.directory);
     let last = null;
     let lastFetchedAt = 0;
     let refreshing = false;
@@ -121,12 +192,17 @@ async function UsageCoachPlugin(input) {
         if (refreshing) return;
         if (last && Date.now() - lastFetchedAt < TTL_MS) return;
         refreshing = true;
-        fetchQuota().then((q) => {
+        fetchQuota().then(async (q) => {
           try {
             last = coach(q);
             lastFetchedAt = Date.now();
-            writeState(last);
-            log(`${last.decision} | weekly=${last.weekly}% 5h=${last.fiveHour}%`);
+            let providers = [];
+            try {
+              providers = await fetchProvidersCoach();
+            } catch {
+            }
+            writeState({ ...last, providers, updatedAt: (/* @__PURE__ */ new Date()).toISOString() });
+            log(`${last.decision} | weekly=${last.weekly}% 5h=${last.fiveHour}% | providers=${providers.length}`);
           } catch (e) {
             log(`refresh-in-then err: ${String(e)}`);
           }

package/dist/tui.js

@@ -1,14 +1,21 @@
 // src/tui.tsx
 import { setProp as _$setProp } from "@opentui/solid";
+import { effect as _$effect } from "@opentui/solid";
 import { insert as _$insert } from "@opentui/solid";
 import { createTextNode as _$createTextNode } from "@opentui/solid";
 import { insertNode as _$insertNode } from "@opentui/solid";
 import { createElement as _$createElement } from "@opentui/solid";
 import { readFileSync, existsSync, writeFileSync, mkdirSync } from "fs";
+import { createHash } from "crypto";
 import { homedir } from "os";
-import { join } from "path";
+import { join, resolve } from "path";
 import { createRoot, createSignal, onCleanup } from "solid-js";
-var STATE_DIR = process.env.UC_STATE_DIR ?? join(homedir(), ".cache", "opencode-usage-coach");
+function projectStateDir(dir) {
+  const abs = resolve(dir || ".");
+  const h = createHash("sha1").update(abs).digest("hex").slice(0, 12);
+  return join(homedir(), ".cache", "opencode-usage-coach", "projects", h);
+}
+var STATE_DIR = join(homedir(), ".cache", "opencode-usage-coach");
 var STATE_FILE = join(STATE_DIR, "state.json");
 var HARNESS_FILE = join(STATE_DIR, "harness.json");
 var MARKER = join(STATE_DIR, "tui-loaded.txt");
@@ -50,11 +57,15 @@ function short(s, n) {
   return s.length <= n ? s : s.slice(0, n - 1) + "\u2026";
 }
 function initializeTui(api, disposeRoot) {
+  STATE_DIR = process.env.UC_STATE_DIR ?? projectStateDir(api.state.path.directory);
+  STATE_FILE = join(STATE_DIR, "state.json");
+  HARNESS_FILE = join(STATE_DIR, "harness.json");
+  MARKER = join(STATE_DIR, "tui-loaded.txt");
   try {
     mkdirSync(STATE_DIR, {
       recursive: true
     });
-    writeFileSync(MARKER, `loaded ${(/* @__PURE__ */ new Date()).toISOString()}`);
+    writeFileSync(MARKER, `loaded ${(/* @__PURE__ */ new Date()).toISOString()} @ ${api.state.path.directory}`);
   } catch {
   }
   const [getState, setState] = createSignal(readState());
@@ -68,16 +79,20 @@ function initializeTui(api, disposeRoot) {
   }, 3e3);
   onCleanup(() => clearInterval(timer));
   onCleanup(() => disposeRoot());
-  const statusDot = {
-    generating: "\u25CF",
-    grading: "\u25CF",
-    revising: "\u25CF",
-    completed: "\u25CF",
-    failed: "\u25CF",
-    timed_out: "\u25CF",
-    halted_quota: "\u25CF"
+  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) => ({
+      fg: th[k]
+    });
     let s = null;
     try {
       s = getState();
@@ -92,98 +107,148 @@ function initializeTui(api, disposeRoot) {
     }
     const nodes = [];
     if (s) {
+      const dKey = s.decision === "GO" ? "success" : s.decision === "THROTTLE" ? "warning" : "error";
       nodes.push((() => {
         var _el$ = _$createElement("text"), _el$2 = _$createTextNode(`usage-coach [`), _el$3 = _$createTextNode(`]`);
         _$insertNode(_el$, _el$2);
         _$insertNode(_el$, _el$3);
         _$insert(_el$, () => TAG[s.decision], _el$3);
+        _$effect((_$p) => _$setProp(_el$, "style", st(dKey), _$p));
         return _el$;
       })());
+      if (s.providers && s.providers.length > 0) {
+        for (const p of s.providers) {
+          nodes.push((() => {
+            var _el$4 = _$createElement("text"), _el$5 = _$createTextNode(` `);
+            _$insertNode(_el$4, _el$5);
+            _$insert(_el$4, () => p.name, null);
+            _$effect((_$p) => _$setProp(_el$4, "style", st("textMuted"), _$p));
+            return _el$4;
+          })());
+          const k5 = p.fiveHour >= 70 ? "error" : p.fiveHour >= 40 ? "warning" : "success";
+          const kw = p.weekly >= 85 ? "error" : p.weekly >= 60 ? "warning" : "success";
+          nodes.push((() => {
+            var _el$6 = _$createElement("text"), _el$7 = _$createTextNode(` 5h `), _el$8 = _$createTextNode(` `), _el$9 = _$createTextNode(`% `);
+            _$insertNode(_el$6, _el$7);
+            _$insertNode(_el$6, _el$8);
+            _$insertNode(_el$6, _el$9);
+            _$insert(_el$6, () => bar(p.fiveHour), _el$8);
+            _$insert(_el$6, () => p.fiveHour, _el$9);
+            _$insert(_el$6, () => p.fiveHourReset, null);
+            _$effect((_$p) => _$setProp(_el$6, "style", st(k5), _$p));
+            return _el$6;
+          })());
+          nodes.push((() => {
+            var _el$0 = _$createElement("text"), _el$1 = _$createTextNode(` wk `), _el$10 = _$createTextNode(` `), _el$11 = _$createTextNode(`% `);
+            _$insertNode(_el$0, _el$1);
+            _$insertNode(_el$0, _el$10);
+            _$insertNode(_el$0, _el$11);
+            _$insert(_el$0, () => bar(p.weekly), _el$10);
+            _$insert(_el$0, () => p.weekly, _el$11);
+            _$insert(_el$0, () => p.weeklyReset, null);
+            _$effect((_$p) => _$setProp(_el$0, "style", st(kw), _$p));
+            return _el$0;
+          })());
+          nodes.push((() => {
+            var _el$12 = _$createElement("text"), _el$13 = _$createTextNode(` -> `);
+            _$insertNode(_el$12, _el$13);
+            _$insert(_el$12, () => p.advice, null);
+            _$effect((_$p) => _$setProp(_el$12, "style", st(dKey), _$p));
+            return _el$12;
+          })());
+        }
+      } else {
+        nodes.push((() => {
+          var _el$16 = _$createElement("text"), _el$17 = _$createTextNode(` 5h `), _el$18 = _$createTextNode(` `), _el$19 = _$createTextNode(`%`);
+          _$insertNode(_el$16, _el$17);
+          _$insertNode(_el$16, _el$18);
+          _$insertNode(_el$16, _el$19);
+          _$insert(_el$16, () => bar(s.fiveHour), _el$18);
+          _$insert(_el$16, () => s.fiveHour, _el$19);
+          return _el$16;
+        })());
+        nodes.push((() => {
+          var _el$20 = _$createElement("text"), _el$21 = _$createTextNode(` wk `), _el$22 = _$createTextNode(` `), _el$23 = _$createTextNode(`%`);
+          _$insertNode(_el$20, _el$21);
+          _$insertNode(_el$20, _el$22);
+          _$insertNode(_el$20, _el$23);
+          _$insert(_el$20, () => bar(s.weekly), _el$22);
+          _$insert(_el$20, () => s.weekly, _el$23);
+          return _el$20;
+        })());
+        nodes.push((() => {
+          var _el$24 = _$createElement("text"), _el$25 = _$createTextNode(` mo `), _el$26 = _$createTextNode(` `), _el$27 = _$createTextNode(`%`);
+          _$insertNode(_el$24, _el$25);
+          _$insertNode(_el$24, _el$26);
+          _$insertNode(_el$24, _el$27);
+          _$insert(_el$24, () => bar(s.monthly), _el$26);
+          _$insert(_el$24, () => s.monthly, _el$27);
+          return _el$24;
+        })());
+      }
+    } else {
       nodes.push((() => {
-        var _el$4 = _$createElement("text"), _el$5 = _$createTextNode(` 5h `), _el$6 = _$createTextNode(` `), _el$7 = _$createTextNode(`%`);
-        _$insertNode(_el$4, _el$5);
-        _$insertNode(_el$4, _el$6);
-        _$insertNode(_el$4, _el$7);
-        _$insert(_el$4, () => bar(s.fiveHour), _el$6);
-        _$insert(_el$4, () => s.fiveHour, _el$7);
-        return _el$4;
-      })());
-      nodes.push((() => {
-        var _el$8 = _$createElement("text"), _el$9 = _$createTextNode(` wk `), _el$0 = _$createTextNode(` `), _el$1 = _$createTextNode(`%`);
-        _$insertNode(_el$8, _el$9);
-        _$insertNode(_el$8, _el$0);
-        _$insertNode(_el$8, _el$1);
-        _$insert(_el$8, () => bar(s.weekly), _el$0);
-        _$insert(_el$8, () => s.weekly, _el$1);
-        return _el$8;
+        var _el$28 = _$createElement("text");
+        _$insertNode(_el$28, _$createTextNode(`usage-coach: ...`));
+        return _el$28;
       })());
+    }
+    if (h && h.active !== false && h.tasks.length > 0) {
       nodes.push((() => {
-        var _el$10 = _$createElement("text"), _el$11 = _$createTextNode(` mo `), _el$12 = _$createTextNode(` `), _el$13 = _$createTextNode(`%`);
-        _$insertNode(_el$10, _el$11);
-        _$insertNode(_el$10, _el$12);
-        _$insertNode(_el$10, _el$13);
-        _$insert(_el$10, () => bar(s.monthly), _el$12);
-        _$insert(_el$10, () => s.monthly, _el$13);
-        return _el$10;
+        var _el$30 = _$createElement("text");
+        _$insertNode(_el$30, _$createTextNode(` `));
+        return _el$30;
       })());
-    } else {
       nodes.push((() => {
-        var _el$14 = _$createElement("text");
-        _$insertNode(_el$14, _$createTextNode(`usage-coach: ...`));
-        return _el$14;
+        var _el$32 = _$createElement("text"), _el$33 = _$createTextNode(`harness: `), _el$34 = _$createTextNode(` `), _el$35 = _$createTextNode(`/`);
+        _$insertNode(_el$32, _el$33);
+        _$insertNode(_el$32, _el$34);
+        _$insertNode(_el$32, _el$35);
+        _$insert(_el$32, () => h.name, _el$34);
+        _$insert(_el$32, () => h.current, _el$35);
+        _$insert(_el$32, () => h.total, null);
+        _$effect((_$p) => _$setProp(_el$32, "style", st("textMuted"), _$p));
+        return _el$32;
       })());
-    }
-    const hStatus = h && h.tasks.length > 0 ? `${h.name} ${h.current}/${h.total}` : "idle";
-    nodes.push((() => {
-      var _el$16 = _$createElement("text");
-      _$insertNode(_el$16, _$createTextNode(` `));
-      return _el$16;
-    })());
-    nodes.push((() => {
-      var _el$18 = _$createElement("text"), _el$19 = _$createTextNode(`harness: `);
-      _$insertNode(_el$18, _el$19);
-      _$insert(_el$18, hStatus, null);
-      return _el$18;
-    })());
-    if (h && h.tasks.length > 0) {
       for (const t of h.tasks) {
-        const dot = statusDot[t.status] ?? "\xB7";
+        const sKey = statusKey[t.status] ?? "text";
         const lbl = TLABEL[t.status] ?? t.status;
         const rev = t.revisions > 0 && t.status === "revising" ? `(${t.revisions})` : "";
         const mdl = t.model ? ` ${t.model.split("/").pop() ?? t.model}` : "";
         nodes.push((() => {
-          var _el$20 = _$createElement("text"), _el$21 = _$createTextNode(` `), _el$22 = _$createTextNode(` `), _el$23 = _$createTextNode(` `), _el$24 = _$createTextNode(` `);
-          _$insertNode(_el$20, _el$21);
-          _$insertNode(_el$20, _el$22);
-          _$insertNode(_el$20, _el$23);
-          _$insertNode(_el$20, _el$24);
-          _$insert(_el$20, dot, _el$22);
-          _$insert(_el$20, () => t.id, _el$23);
-          _$insert(_el$20, mdl, _el$23);
-          _$insert(_el$20, lbl, _el$24);
-          _$insert(_el$20, rev, _el$24);
-          _$insert(_el$20, () => short(t.title, 12), null);
-          return _el$20;
+          var _el$36 = _$createElement("text"), _el$37 = _$createTextNode(` \u25CF `), _el$38 = _$createTextNode(` `), _el$39 = _$createTextNode(` `);
+          _$insertNode(_el$36, _el$37);
+          _$insertNode(_el$36, _el$38);
+          _$insertNode(_el$36, _el$39);
+          _$insert(_el$36, () => t.id, _el$38);
+          _$insert(_el$36, mdl, _el$38);
+          _$insert(_el$36, lbl, _el$39);
+          _$insert(_el$36, rev, _el$39);
+          _$insert(_el$36, () => short(t.title, 12), null);
+          _$effect((_$p) => _$setProp(_el$36, "style", st(sKey), _$p));
+          return _el$36;
         })());
         const pv = t.model ? t.model.startsWith("zai") ? "zai" : (t.model.split("/")[0] ?? "").split("-")[0] : "";
         const q = pv && h.quotas?.[pv] ? h.quotas[pv] : null;
         const pct = q ? q.fiveHour : 0;
+        const qKey = pct >= 70 ? "error" : pct >= 40 ? "warning" : "success";
         nodes.push((() => {
-          var _el$25 = _$createElement("text"), _el$26 = _$createTextNode(` 5h `), _el$27 = _$createTextNode(` `), _el$28 = _$createTextNode(`%`);
-          _$insertNode(_el$25, _el$26);
-          _$insertNode(_el$25, _el$27);
-          _$insertNode(_el$25, _el$28);
-          _$insert(_el$25, () => bar(pct), _el$27);
-          _$insert(_el$25, pct, _el$28);
-          return _el$25;
+          var _el$40 = _$createElement("text"), _el$41 = _$createTextNode(` 5h `), _el$42 = _$createTextNode(` `), _el$43 = _$createTextNode(`%`);
+          _$insertNode(_el$40, _el$41);
+          _$insertNode(_el$40, _el$42);
+          _$insertNode(_el$40, _el$43);
+          _$insert(_el$40, () => bar(pct), _el$42);
+          _$insert(_el$40, pct, _el$43);
+          _$effect((_$p) => _$setProp(_el$40, "style", st(qKey), _$p));
+          return _el$40;
         })());
       }
     }
     return (() => {
-      var _el$29 = _$createElement("box");
-      _$setProp(_el$29, "flexDirection", "column");
-      _$insert(_el$29, nodes);
-      return _el$29;
+      var _el$44 = _$createElement("box");
+      _$setProp(_el$44, "flexDirection", "column");
+      _$insert(_el$44, nodes);
+      return _el$44;
     })();
   };
   api.slots.register({
@@ -194,9 +259,9 @@ function initializeTui(api, disposeRoot) {
           return panel(ctx);
         } catch {
           return (() => {
-            var _el$30 = _$createElement("text");
-            _$insertNode(_el$30, _$createTextNode(`usage-coach`));
-            return _el$30;
+            var _el$45 = _$createElement("text");
+            _$insertNode(_el$45, _$createTextNode(`usage-coach`));
+            return _el$45;
           })();
         }
       },
@@ -205,9 +270,9 @@ function initializeTui(api, disposeRoot) {
           return panel(ctx);
         } catch {
           return (() => {
-            var _el$32 = _$createElement("text");
-            _$insertNode(_el$32, _$createTextNode(`usage-coach`));
-            return _el$32;
+            var _el$47 = _$createElement("text");
+            _$insertNode(_el$47, _$createTextNode(`usage-coach`));
+            return _el$47;
           })();
         }
       }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-usage-coach",
-  "version": "0.1.1",
-  "description": "opencode closed-loop usage coach — quota SENSE -> coaching DECIDE -> loop ACT + TUI integration",
+  "version": "0.1.2",
+  "description": "opencode closed-loop usage coach \u2014 quota SENSE -> coaching DECIDE -> loop ACT + TUI integration",
   "type": "module",
   "main": "./dist/index.js",
   "module": "./dist/index.js",
@@ -26,7 +26,16 @@
     "README.md",
     "LICENSE"
   ],
-  "keywords": ["opencode", "opencode-plugin", "quota", "usage", "coach", "loop", "zai", "glm"],
+  "keywords": [
+    "opencode",
+    "opencode-plugin",
+    "quota",
+    "usage",
+    "coach",
+    "loop",
+    "zai",
+    "glm"
+  ],
   "license": "MIT",
   "peerDependencies": {
     "@opencode-ai/plugin": ">=1.14",
@@ -43,4 +52,4 @@
     "tsup": "^8.5",
     "typescript": "^5"
   }
-}
+}