@hk_net/pi-usage-bars 0.3.0 → 0.4.1

package/extensions/usage-bars/index.ts

@@ -1,4 +1,4 @@
-/** Usage bars for subscription-backed providers supported by current Pi releases. */
+/** Quota, balance, and spend indicators for providers supported by current Pi releases. */
 
 import {
   DynamicBorder,
@@ -22,9 +22,16 @@ import {
   fetchAllUsages,
   fetchClaudeUsageWithFallback,
   fetchCodexUsage,
+  fetchDeepSeekBalance,
+  fetchKimiUsage,
+  fetchMiniMaxUsage,
+  fetchMoonshotBalance,
+  fetchOpenRouterUsage,
   fetchZaiUsage,
   providerToPiProviderId,
   resolveUsageEndpoints,
+  type AccountBalance,
+  type AccountSpend,
   type ProviderKey,
   type UsageByProvider,
   type UsageData,
@@ -33,15 +40,63 @@ import {
 
 const POLL_INTERVAL_MS = 2 * 60 * 1000;
 const STATUS_KEY = "usage-bars";
-const PROVIDERS: readonly ProviderKey[] = ["codex", "claude", "zai", "zai-cn"];
+const PROVIDERS: readonly ProviderKey[] = [
+  "codex",
+  "claude",
+  "zai",
+  "zai-cn",
+  "kimi",
+  "minimax",
+  "minimax-cn",
+  "openrouter",
+  "deepseek",
+  "moonshot",
+  "moonshot-cn",
+];
 
 const PROVIDER_LABELS: Record<ProviderKey, string> = {
   codex: "Codex",
   claude: "Claude",
   zai: "ZAI Coding Plan (Global)",
   "zai-cn": "ZAI Coding Plan (China)",
+  kimi: "Kimi For Coding",
+  minimax: "MiniMax Coding Plan (Global)",
+  "minimax-cn": "MiniMax Coding Plan (China)",
+  openrouter: "OpenRouter",
+  deepseek: "DeepSeek",
+  moonshot: "Moonshot/Kimi API (Global)",
+  "moonshot-cn": "Moonshot/Kimi API (China)",
 };
 
+function formatFinancialAmount(amount: number, unit: string): string {
+  if (/^[A-Z]{3}$/.test(unit)) {
+    return new Intl.NumberFormat("en-US", {
+      style: "currency",
+      currency: unit,
+      minimumFractionDigits: 2,
+      maximumFractionDigits: 5,
+    }).format(amount);
+  }
+  const formatted = new Intl.NumberFormat("en-US", { maximumFractionDigits: 2 }).format(amount);
+  return `${formatted} ${unit}`;
+}
+
+function formatAccountBalance(balance: AccountBalance): string {
+  return `${balance.label} · ${formatFinancialAmount(balance.amount, balance.unit)}`;
+}
+
+function formatAccountSpend(spend: AccountSpend): string {
+  const values = [
+    spend.daily === undefined ? undefined : `today ${formatFinancialAmount(spend.daily, spend.unit)}`,
+    spend.weekly === undefined ? undefined : `week ${formatFinancialAmount(spend.weekly, spend.unit)}`,
+    spend.monthly === undefined ? undefined : `month ${formatFinancialAmount(spend.monthly, spend.unit)}`,
+  ].filter((value): value is string => Boolean(value));
+  if (values.length === 0 && spend.lifetime !== undefined) {
+    values.push(`lifetime ${formatFinancialAmount(spend.lifetime, spend.unit)}`);
+  }
+  return `Spent · ${values.join(" · ")}`;
+}
+
 interface SubscriptionItem {
   name: string;
   provider: ProviderKey;
@@ -135,12 +190,12 @@ class UsageSelectorComponent extends Container implements Focusable {
 
   private updateHint(): void {
     if (this.hint === "loading") {
-      this.hintText.setText(this.theme.fg("dim", "Fetching usage from configured subscription providers…"));
+      this.hintText.setText(this.theme.fg("dim", "Fetching quota, balance, and spend from configured providers…"));
     } else if (this.hint === "error") {
       this.hintText.setText(this.theme.fg("error", "Failed to fetch usage data"));
     } else {
       this.hintText.setText(
-        this.theme.fg("muted", "Only showing configured subscription providers. ") +
+        this.theme.fg("muted", "Only showing configured usage providers. ") +
           this.theme.fg("dim", "✓ = active provider"),
       );
     }
@@ -197,19 +252,51 @@ class UsageSelectorComponent extends Container implements Focusable {
       const weeklyReset = item.data.weeklyResetsIn
         ? theme.fg("dim", `  resets in ${item.data.weeklyResetsIn}`)
         : "";
-
-      this.listContainer.addChild(new Text(
-        indent + theme.fg("muted", "Session  ") + this.renderBar(session) + " " +
-          theme.fg(colorForPercent(session), `${session}%`.padStart(4)) + sessionReset,
-        0,
-        0,
-      ));
-      this.listContainer.addChild(new Text(
-        indent + theme.fg("muted", "Weekly   ") + this.renderBar(weekly) + " " +
-          theme.fg(colorForPercent(weekly), `${weekly}%`.padStart(4)) + weeklyReset,
-        0,
-        0,
-      ));
+      const sessionLabel = (item.data.sessionLabel ?? "Session").slice(0, 9).padEnd(10);
+      const weeklyLabel = (item.data.weeklyLabel ?? "Weekly").slice(0, 9).padEnd(10);
+
+      if (!item.data.quotaHidden) {
+        if (!item.data.sessionHidden) {
+          this.listContainer.addChild(new Text(
+            indent + theme.fg("muted", sessionLabel) + this.renderBar(session) + " " +
+              theme.fg(colorForPercent(session), `${session}%`.padStart(4)) + sessionReset,
+            0,
+            0,
+          ));
+        }
+        if (!item.data.weeklyHidden) {
+          this.listContainer.addChild(new Text(
+            indent + theme.fg("muted", weeklyLabel) + this.renderBar(weekly) + " " +
+              theme.fg(colorForPercent(weekly), `${weekly}%`.padStart(4)) + weeklyReset,
+            0,
+            0,
+          ));
+        }
+      }
+      if (item.data.accountBalance) {
+        this.listContainer.addChild(new Text(
+          indent + theme.fg("muted", formatAccountBalance(item.data.accountBalance)),
+          0,
+          0,
+        ));
+      }
+      for (const balance of item.data.accountBalanceDetails ?? []) {
+        this.listContainer.addChild(new Text(
+          indent + theme.fg("dim", formatAccountBalance(balance)),
+          0,
+          0,
+        ));
+      }
+      if (item.data.accountSpend) {
+        this.listContainer.addChild(new Text(
+          indent + theme.fg("muted", formatAccountSpend(item.data.accountSpend)),
+          0,
+          0,
+        ));
+      }
+      if (item.data.notice) {
+        this.listContainer.addChild(new Text(indent + theme.fg("muted", item.data.notice), 0, 0));
+      }
 
       if (typeof item.data.extraSpend === "number" && typeof item.data.extraLimit === "number") {
         this.listContainer.addChild(new Text(
@@ -300,6 +387,13 @@ export default function (pi: ExtensionAPI): void {
     claude: null,
     zai: null,
     "zai-cn": null,
+    kimi: null,
+    minimax: null,
+    "minimax-cn": null,
+    openrouter: null,
+    deepseek: null,
+    moonshot: null,
+    "moonshot-cn": null,
     activeProvider: null,
     available: {},
   };
@@ -343,15 +437,50 @@ export default function (pi: ExtensionAPI): void {
       ctx.ui.setStatus(STATUS_KEY, theme.fg("warning", `${label} usage unavailable (${data.error})`));
       return;
     }
+    if (data.quotaHidden) {
+      const financial = [
+        data.accountBalance ? formatAccountBalance(data.accountBalance) : undefined,
+        data.accountSpend?.monthly === undefined
+          ? undefined
+          : `Month · ${formatFinancialAmount(data.accountSpend.monthly, data.accountSpend.unit)}`,
+      ].filter((value): value is string => Boolean(value));
+      const summary = financial.length > 0 ? financial.join(" · ") : data.notice;
+      ctx.ui.setStatus(
+        STATUS_KEY,
+        summary ? theme.fg("dim", `${label} `) + theme.fg("muted", summary) : undefined,
+      );
+      return;
+    }
 
     const session = clampPercent(data.session);
     const weekly = clampPercent(data.weekly);
+    const sessionPrefix = data.sessionLabel === "5-hour"
+      ? "5h "
+      : data.sessionLabel === "Interval"
+        ? "I "
+        : data.sessionLabel === "Key limit"
+          ? "L "
+          : "S ";
+    const quotaLanes: string[] = [];
+    if (!data.sessionHidden) {
+      quotaLanes.push(
+        theme.fg("muted", sessionPrefix) + renderBar(theme, session) + " " + renderPercent(theme, session) +
+          (data.sessionResetsIn ? theme.fg("dim", ` ⟳ ${data.sessionResetsIn}`) : ""),
+      );
+    }
+    if (!data.weeklyHidden) {
+      quotaLanes.push(
+        theme.fg("muted", "W ") + renderBar(theme, weekly) + " " + renderPercent(theme, weekly) +
+          (data.weeklyResetsIn ? theme.fg("dim", ` ⟳ ${data.weeklyResetsIn}`) : ""),
+      );
+    }
     const status =
       theme.fg("dim", `${label} `) +
-      theme.fg("muted", "S ") + renderBar(theme, session) + " " + renderPercent(theme, session) +
-      (data.sessionResetsIn ? theme.fg("dim", ` ⟳ ${data.sessionResetsIn}`) : "") +
-      theme.fg("muted", " W ") + renderBar(theme, weekly) + " " + renderPercent(theme, weekly) +
-      (data.weeklyResetsIn ? theme.fg("dim", ` ⟳ ${data.weeklyResetsIn}`) : "") +
+      quotaLanes.join(" ") +
+      (data.accountBalance ? theme.fg("muted", ` · ${formatAccountBalance(data.accountBalance)}`) : "") +
+      (data.accountSpend?.monthly === undefined
+        ? ""
+        : theme.fg("muted", ` · Month ${formatFinancialAmount(data.accountSpend.monthly, data.accountSpend.unit)}`)) +
       (data.stale ? theme.fg("warning", " stale") : "") +
       (data.warning && !data.stale ? theme.fg("warning", " ⚠") : "");
     ctx.ui.setStatus(STATUS_KEY, status);
@@ -404,6 +533,25 @@ export default function (pi: ExtensionAPI): void {
     if (provider === "claude") state.claude = await fetchClaudeUsageWithFallback(credential.token, { signal });
     if (provider === "zai") state.zai = await fetchZaiUsage(credential.token, "zai", { endpoints, signal });
     if (provider === "zai-cn") state["zai-cn"] = await fetchZaiUsage(credential.token, "zai-cn", { endpoints, signal });
+    if (provider === "kimi") state.kimi = await fetchKimiUsage(credential.token, { endpoints, signal });
+    if (provider === "minimax") {
+      state.minimax = await fetchMiniMaxUsage(credential.token, "minimax", { endpoints, signal });
+    }
+    if (provider === "minimax-cn") {
+      state["minimax-cn"] = await fetchMiniMaxUsage(credential.token, "minimax-cn", { endpoints, signal });
+    }
+    if (provider === "openrouter") {
+      state.openrouter = await fetchOpenRouterUsage(credential.token, { endpoints, signal });
+    }
+    if (provider === "deepseek") {
+      state.deepseek = await fetchDeepSeekBalance(credential.token, { endpoints, signal });
+    }
+    if (provider === "moonshot") {
+      state.moonshot = await fetchMoonshotBalance(credential.token, "moonshot", { endpoints, signal });
+    }
+    if (provider === "moonshot-cn") {
+      state["moonshot-cn"] = await fetchMoonshotBalance(credential.token, "moonshot-cn", { endpoints, signal });
+    }
   }
 
   async function runPoll(): Promise<void> {
@@ -493,7 +641,7 @@ export default function (pi: ExtensionAPI): void {
   });
 
   pi.registerCommand("usage", {
-    description: "Show subscription usage for configured providers",
+    description: "Show quota, balance, and spend for configured providers",
     handler: async (_args, ctx) => {
       currentContext = ctx;
       updateProviderFrom(ctx.model);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@hk_net/pi-usage-bars",
-  "version": "0.3.0",
-  "description": "Subscription usage bars for current Pi providers (Codex, Claude, and ZAI Coding Plan)",
+  "version": "0.4.1",
+  "description": "Quota, balance, and spend indicators for current Pi providers",
   "keywords": [
     "pi-package",
     "pi",
@@ -9,7 +9,12 @@
     "usage",
     "codex",
     "anthropic",
-    "zai"
+    "zai",
+    "kimi",
+    "minimax",
+    "openrouter",
+    "deepseek",
+    "moonshot"
   ],
   "license": "MIT",
   "repository": {
@@ -27,6 +32,7 @@
   "type": "module",
   "files": [
     "extensions",
+    "docs",
     "README.md",
     "CHANGELOG.md",
     "LICENSE"
@@ -34,8 +40,7 @@
   "scripts": {
     "test": "bun test tests",
     "typecheck": "tsc --noEmit",
-    "smoke": "pi -e ./extensions/usage-bars/index.ts --list-models",
-    "check": "npm run typecheck && npm test && npm run smoke",
+    "check": "npm run typecheck && npm test",
     "prepublishOnly": "npm run check"
   },
   "engines": {
@@ -46,8 +51,8 @@
     "@earendil-works/pi-tui": ">=0.81.1"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "0.81.1",
-    "@earendil-works/pi-tui": "0.81.1",
+    "@earendil-works/pi-coding-agent": "0.83.0",
+    "@earendil-works/pi-tui": "0.83.0",
     "@types/bun": "1.3.14",
     "@types/node": "24.13.3",
     "typescript": "5.9.3"