@khalilgharbaoui/opencode-claude-code-plugin 0.54.2 → 0.56.0

package/dist/tui.js

@@ -271,6 +271,157 @@ function createSpawnRecordReader(file = spawnRecordStorePath) {
   };
 }
 
+// src/account-usage.ts
+import { mkdirSync as mkdirSync2, readFileSync as readFileSync2, renameSync as renameSync2, statSync as statSync2, writeFileSync as writeFileSync2 } from "fs";
+import { homedir as homedir2 } from "os";
+import { join as join2 } from "path";
+var UNTIMED_READING_TTL_MS = 5 * 60 * 60 * 1e3;
+var USAGE_READING_MAX_AGE_MS = 8 * 24 * 60 * 60 * 1e3;
+var NAME = /^[a-z0-9_-]{1,64}$/;
+var STATUS = /^[a-z_]{1,32}$/;
+var MAX_RESETS_TEXT = 80;
+var pathOverride2 = null;
+function accountUsageStorePath() {
+  if (pathOverride2) return pathOverride2;
+  const stateRoot = process.env.XDG_STATE_HOME || join2(homedir2(), ".local", "state");
+  return join2(stateRoot, "opencode-claude-code-plugin", "account-usage.json");
+}
+function finite(value) {
+  return typeof value === "number" && Number.isFinite(value);
+}
+function validEntry(value) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) return null;
+  const entry = value;
+  if (!finite(entry.observedAt)) return null;
+  if (typeof entry.status !== "string" || !STATUS.test(entry.status)) return null;
+  if (entry.utilization !== null && !finite(entry.utilization)) return null;
+  if (entry.resetsAt !== void 0 && !finite(entry.resetsAt)) return null;
+  if (entry.resetsText !== void 0 && (typeof entry.resetsText !== "string" || entry.resetsText.length > MAX_RESETS_TEXT)) {
+    return null;
+  }
+  const noted = Array.isArray(entry.noted) ? [...new Set(entry.noted.filter((t) => finite(t) && t > 0 && t <= 100))].sort(
+    (a, b) => a - b
+  ) : [];
+  return {
+    utilization: entry.utilization,
+    ...entry.resetsAt === void 0 ? {} : { resetsAt: entry.resetsAt },
+    ...entry.resetsText === void 0 ? {} : { resetsText: entry.resetsText },
+    status: entry.status,
+    observedAt: entry.observedAt,
+    noted
+  };
+}
+function readStore(file, now) {
+  const out = /* @__PURE__ */ new Map();
+  try {
+    const parsed = JSON.parse(readFileSync2(file, "utf8"));
+    const accounts = parsed && typeof parsed === "object" ? parsed.accounts : null;
+    if (!accounts || typeof accounts !== "object" || Array.isArray(accounts)) return out;
+    const cutoff = now - USAGE_READING_MAX_AGE_MS;
+    for (const [account, windows] of Object.entries(accounts)) {
+      if (!NAME.test(account) || !windows || typeof windows !== "object") continue;
+      const kept = /* @__PURE__ */ new Map();
+      for (const [window, raw] of Object.entries(windows)) {
+        if (!NAME.test(window)) continue;
+        const entry = validEntry(raw);
+        if (entry && entry.observedAt >= cutoff) kept.set(window, entry);
+      }
+      if (kept.size > 0) out.set(account, kept);
+    }
+  } catch {
+  }
+  return out;
+}
+function isReadingCurrent(entry, now = Date.now()) {
+  if (entry.resetsAt !== void 0) return entry.resetsAt > now;
+  return now - entry.observedAt < UNTIMED_READING_TTL_MS;
+}
+function usagePercent(utilization) {
+  return Math.round(utilization * 1e3) / 10;
+}
+function createAccountUsageReader(file = accountUsageStorePath) {
+  let identity = "";
+  let usage = /* @__PURE__ */ new Map();
+  return {
+    read() {
+      const path = file();
+      let next = "";
+      try {
+        const stats = statSync2(path);
+        next = `${stats.mtimeMs}:${stats.size}:${stats.ino}`;
+      } catch {
+        next = "missing";
+      }
+      if (next !== identity) {
+        identity = next;
+        usage = next === "missing" ? /* @__PURE__ */ new Map() : readStore(path, Date.now());
+      }
+      return usage;
+    }
+  };
+}
+
+// src/account-sidebar.ts
+var WARNING_PERCENT = 75;
+var SHORT_WINDOWS = {
+  five_hour: "5h",
+  seven_day: "7d",
+  seven_day_opus: "7d opus",
+  seven_day_sonnet: "7d sonnet",
+  seven_day_overage_included: "7d extra",
+  overage: "extra"
+};
+function shortWindowName(window) {
+  if (window === "unknown") return "";
+  return SHORT_WINDOWS[window] ?? window.replace(/_/g, " ");
+}
+var MONTHS = ["Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"];
+function formatResetClock(resetsAt, now) {
+  const when = new Date(resetsAt);
+  const today = new Date(now);
+  const pad = (value) => String(value).padStart(2, "0");
+  const clock = `${pad(when.getHours())}:${pad(when.getMinutes())}`;
+  const sameDay = when.getFullYear() === today.getFullYear() && when.getMonth() === today.getMonth() && when.getDate() === today.getDate();
+  return sameDay ? clock : `${MONTHS[when.getMonth()]} ${when.getDate()} ${clock}`;
+}
+function rank(entry) {
+  if (entry.status === "rejected") return 1e3;
+  if (entry.utilization === null) return -1;
+  return usagePercent(entry.utilization);
+}
+function pressingWindow(windows, now) {
+  let best;
+  for (const [window, entry] of windows) {
+    if (!isReadingCurrent(entry, now)) continue;
+    if (!best || rank(entry) > rank(best[1])) best = [window, entry];
+  }
+  return best;
+}
+function accountUsageLines(usage, now, width = SIDEBAR_ROW_WIDTH) {
+  const lines = [];
+  for (const account of [...usage.keys()].sort()) {
+    const pressing = pressingWindow(usage.get(account), now);
+    if (!pressing) continue;
+    const [window, entry] = pressing;
+    const state = entry.status === "rejected" ? "limited" : entry.utilization !== null && usagePercent(entry.utilization) >= WARNING_PERCENT ? "warning" : "ok";
+    const short = shortWindowName(window);
+    const level = state === "limited" ? "limited" : entry.utilization === null ? `${short ? `${short} ` : ""}ok` : `${short ? `${short} ` : ""}${Math.round(usagePercent(entry.utilization))}%`;
+    const resets = entry.resetsAt !== void 0 ? ` \xB7 resets ${formatResetClock(entry.resetsAt, now)}` : entry.resetsText ? ` \xB7 resets ${entry.resetsText}` : "";
+    const name = truncate(account, Math.max(1, Math.floor(width / 2)));
+    const rest = ` ${truncate(`\xB7 ${level}${resets}`, Math.max(0, width - Array.from(name).length - 1))}`;
+    lines.push({
+      account,
+      window,
+      state,
+      segments: [
+        { text: name, role: "agent" },
+        { text: rest, role: state === "limited" ? "error" : state === "warning" ? "effort" : "muted" }
+      ]
+    });
+  }
+  return lines;
+}
+
 // src/tui.ts
 var SUBAGENTS_TUI_PLUGIN_ID = "opencode-claude-code-plugin.subagents";
 var SECTION_ORDER = 900;
@@ -320,8 +471,12 @@ function createSidebar(host) {
   const records = createSpawnRecordReader();
   let recordMap = attempt(() => records.read(), /* @__PURE__ */ new Map());
   const [recordVersion, setRecordVersion] = createSignal(0);
+  const usageReader = createAccountUsageReader();
+  const [usage, setUsage] = createSignal(attempt(() => usageReader.read(), /* @__PURE__ */ new Map()));
+  const [usageNow, setUsageNow] = createSignal(Date.now());
   const sectionsRunning = /* @__PURE__ */ new Set();
   const sectionsShowing = /* @__PURE__ */ new Set();
+  const sectionsMounted = /* @__PURE__ */ new Set();
   let spinner;
   let refresh;
   const syncClocks = () => {
@@ -331,16 +486,22 @@ function createSidebar(host) {
       clearInterval(spinner);
       spinner = void 0;
     }
-    if (sectionsShowing.size > 0 && !refresh) {
+    const wanted = sectionsShowing.size > 0 || sectionsMounted.size > 0;
+    if (wanted && !refresh) {
       refresh = setInterval(() => {
-        const next = attempt(() => records.read(), recordMap);
-        if (next !== recordMap) {
-          recordMap = next;
-          setRecordVersion((value) => value + 1);
+        if (sectionsShowing.size > 0) {
+          const next = attempt(() => records.read(), recordMap);
+          if (next !== recordMap) {
+            recordMap = next;
+            setRecordVersion((value) => value + 1);
+          }
+          setNow(Date.now());
         }
-        setNow(Date.now());
+        setUsage(attempt(() => usageReader.read(), usage()));
+        const minute = Math.floor(Date.now() / 6e4) * 6e4;
+        if (minute !== usageNow()) setUsageNow(minute);
       }, REFRESH_INTERVAL_MS);
-    } else if (sectionsShowing.size === 0 && refresh) {
+    } else if (!wanted && refresh) {
       clearInterval(refresh);
       refresh = void 0;
     }
@@ -424,9 +585,35 @@ function createSidebar(host) {
     insertNode(section, body);
     return section;
   };
+  const buildAccounts = (lines) => {
+    const section = createElement("box");
+    const title = createElement("text");
+    const bold = createElement("b");
+    insertNode(bold, createTextNode("Accounts"));
+    insertNode(title, bold);
+    effect(() => setProp(title, "fg", host.colors().text));
+    insertNode(section, title);
+    for (const line of lines) {
+      const text2 = createElement("text");
+      setProp(text2, "wrapMode", "none");
+      for (const segment of line.segments) {
+        const span = createElement("span");
+        insertNode(span, createTextNode(segment.text));
+        effect(() => setProp(span, "style", { fg: roleColor(host.colors(), segment.role) }));
+        insertNode(text2, span);
+      }
+      insertNode(section, text2);
+    }
+    return section;
+  };
   const render = (sessionID) => {
     const root = createElement("box");
+    const subagentsBox = createElement("box");
+    const accountsBox = createElement("box");
+    insertNode(root, subagentsBox);
+    insertNode(root, accountsBox);
     const key = /* @__PURE__ */ Symbol("subagents-section");
+    sectionsMounted.add(key);
     const [expanded, setExpanded] = createSignal(true);
     const rows = createMemo(() => {
       const parentID = sessionID();
@@ -444,18 +631,29 @@ function createSidebar(host) {
     onCleanup(() => {
       sectionsRunning.delete(key);
       sectionsShowing.delete(key);
+      sectionsMounted.delete(key);
       syncClocks();
     });
-    insert(root, () => {
+    insert(subagentsBox, () => {
       const list = rows();
       if (list.length === 0) return null;
       return attempt(() => buildSection(list, expanded, setExpanded), null);
     });
+    const accountLines = createMemo(
+      () => attempt(() => accountUsageLines(usage(), usageNow()), [])
+    );
+    insert(accountsBox, () => {
+      const lines = accountLines();
+      if (lines.length === 0) return null;
+      return attempt(() => buildAccounts(lines), null);
+    });
+    syncClocks();
     return root;
   };
   const dispose = () => {
     sectionsRunning.clear();
     sectionsShowing.clear();
+    sectionsMounted.clear();
     syncClocks();
   };
   return { render, dispose };