@llblab/pi-codex-usage 0.9.4 → 0.10.1

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## Unreleased
+
+## 0.10.1: Softer Quota Bar
+
+- Changed the normal and loading dual-quota bar background from `selectedBg` to the softer themed `userMessageBg`; exhausted quotas still use `toolErrorBg`. The bar is less prominent across dark and light themes without altering quota values or countdowns.
+
+## 0.10.0: Business Credit Usage
+
+- Added Business-account credit usage support for responses that do not expose Codex rate-limit windows. Impact: the statusline now shows rounded remaining credit percentage and an optional reset countdown while preserving existing rate-limit displays when both forms are available.
+- Scheduled credit reset countdown redraws at their exact display boundaries and retained credits-only reports in the active cache. Impact: credit status remains locally responsive between API refreshes without displacing model-specific quota windows.
+
 ## 0.9.4: Responsive Quota Polling
 
 - Increased the automatic Codex quota refresh interval from 30 to 60 seconds. Impact: status values remain responsive down to the locally rendered seconds countdown while reducing repeated OpenAI requests, especially when several Pi instances are open.

package/README.md

@@ -68,6 +68,12 @@ Before the first usable report for the active quota bucket arrives, two half-hei
 
 When the weekly reset time is available, it follows either the single-window percentage or the dual-window bar. More than a day remains is shown in 144-minute day-tenth steps such as `7d`, `6.9d`, `6.6d`, `5.1d`, `5d`, `3.7d`, `3d`, `2d`, `1.9d`, `1.5d`, and `1.1d`, rounded upward to the next tenth. At 24 hours and below it switches to upward-rounded 6-minute hour-tenth steps such as `24h`, `23.7h`, `20.1h`, `20h`, `19.9h`, `1.4h`, `1.3h`, `1.2h`, `1.1h`, and `1h`. Under an hour it switches to floored minutes, and under a minute to seconds. After the reset timestamp passes, `0s` is held until the next successful quota refresh reports the new weekly window.
 
+Business accounts that return credits instead of Codex rate-limit windows show the rounded remaining percentage. When the response includes a credit reset cycle, its countdown is shown after the percentage:
+
+```text
+codex 81% 2.8d
+```
+
 When the 5-hour window is exhausted and exposes its own reset time, the statusline adds the 5-hour reset before the weekly reset:
 
 ```text

package/index.ts

@@ -90,6 +90,12 @@ export type UsageQueryError = {
 
 export type CodexUsageReport = {
   snapshots: NormalizedRateLimitSnapshot[];
+  credits?: NormalizedCreditUsage;
+};
+
+export type NormalizedCreditUsage = {
+  remainingPercent: number;
+  resetAt?: number;
 };
 
 export type NormalizedRateLimitSnapshot = {
@@ -106,6 +112,8 @@ export type NormalizedRateLimitWindow = {
 type RateLimitStatusPayload = {
   rate_limit?: unknown;
   additional_rate_limits?: unknown;
+  credits?: unknown;
+  spend_control?: unknown;
 };
 
 type BackendAdditionalRateLimit = {
@@ -593,7 +601,10 @@ async function queryUsage(
         source === "pi-auth"
           ? await queryViaPiAuth(ctx, options.timeoutMs)
           : await queryViaCodexAppServer(options.timeoutMs);
-      if (selectUsageSnapshot(report, activeUsageLimitId(model))) {
+      if (
+        selectUsageSnapshot(report, activeUsageLimitId(model)) ||
+        report.credits
+      ) {
         return { ok: true, report };
       }
       errors.push({
@@ -920,12 +931,56 @@ export function normalizeBackendPayload(
     }
   }
 
-  if (snapshots.length === 0) {
+  const credits = normalizeBackendCredits(payload, _capturedAt);
+  if (snapshots.length === 0 && !credits) {
     throw new Error(
-      "Codex usage endpoint returned no displayable rate-limit windows.",
+      "Codex usage endpoint returned no displayable rate-limit windows or credits.",
     );
   }
-  return { snapshots };
+  return credits ? { snapshots, credits } : { snapshots };
+}
+
+function normalizeBackendCredits(
+  payload: RateLimitStatusPayload,
+  capturedAt: number,
+): NormalizedCreditUsage | undefined {
+  const spendControl = payload.spend_control as
+    | Record<string, unknown>
+    | undefined;
+  const individualLimit = spendControl?.individual_limit as
+    | Record<string, unknown>
+    | undefined;
+  const credits = payload.credits as Record<string, unknown> | undefined;
+  const limit = asNumber(individualLimit?.limit);
+  const used = asNumber(individualLimit?.used);
+  const remaining = asNumber(individualLimit?.remaining);
+  const resetAt = asResetTime(
+    [
+      individualLimit?.reset_at,
+      individualLimit?.resets_at,
+      individualLimit?.reset_time,
+      spendControl?.reset_at,
+      spendControl?.resets_at,
+      credits?.reset_at,
+      credits?.resets_at,
+    ],
+    individualLimit?.reset_after_seconds ?? spendControl?.reset_after_seconds,
+    capturedAt,
+  );
+  if (
+    limit === undefined ||
+    used === undefined ||
+    remaining === undefined ||
+    limit <= 0
+  )
+    return undefined;
+  const remainingPercent = Math.min(
+    100,
+    Math.max(0, (remaining / limit) * 100),
+  );
+  return resetAt === undefined
+    ? { remainingPercent }
+    : { remainingPercent, resetAt };
 }
 
 function backendAdditionalLimitId(limit: BackendAdditionalRateLimit): string {
@@ -1081,7 +1136,7 @@ export function formatCodexUsageStatusline(
   const barText = formatStatuslineBarText(
     ctx,
     bar ?? "",
-    hasExhaustedQuotaWindow(report, model) ? "toolErrorBg" : "selectedBg",
+    hasExhaustedQuotaWindow(report, model) ? "toolErrorBg" : "userMessageBg",
     model,
   );
   return countdown
@@ -1103,7 +1158,9 @@ export function formatCodexUsageStatusValue(
   const model = typeof modelOrNow === "number" ? undefined : modelOrNow;
   const capturedNow = typeof modelOrNow === "number" ? modelOrNow : now;
   const snapshot = selectActiveUsageSnapshot(report, model);
-  if (!snapshot || (!snapshot.primary && !snapshot.secondary)) return undefined;
+  if (!snapshot || (!snapshot.primary && !snapshot.secondary)) {
+    return formatCreditUsage(report.credits, capturedNow);
+  }
   if (!snapshot.primary || !snapshot.secondary) {
     const window = snapshot.primary ?? snapshot.secondary;
     if (!window) return undefined;
@@ -1130,6 +1187,17 @@ export function formatCodexUsageStatusValue(
   return countdown ? `${bar} ${countdown}` : bar;
 }
 
+function formatCreditUsage(
+  credits: NormalizedCreditUsage | undefined,
+  now: number,
+): string | undefined {
+  if (!credits) return undefined;
+  const percentage = `${Math.round(credits.remainingPercent)}%`;
+  return credits.resetAt
+    ? `${percentage} ${formatResetCountdown(credits.resetAt, now)}`
+    : percentage;
+}
+
 export function formatWeeklyResetCountdown(
   report: CodexUsageReport,
   modelOrNow?: CodexUsageModel | number,
@@ -1167,7 +1235,10 @@ export function nextResetCountdownDelayMs(
   model?: CodexUsageModel,
 ): number | undefined {
   const snapshot = selectActiveUsageSnapshot(report, model);
-  const resetTimes = [weeklyWindow(snapshot)?.resetAt];
+  const hasRateLimitWindow = Boolean(snapshot?.primary || snapshot?.secondary);
+  const resetTimes = [
+    hasRateLimitWindow ? weeklyWindow(snapshot)?.resetAt : report.credits?.resetAt,
+  ];
   if (snapshot?.secondary && isQuotaWindowExhausted(snapshot.primary)) {
     resetTimes.push(snapshot.primary?.resetAt);
   }
@@ -1228,7 +1299,7 @@ function formatStatuslineText(
 function formatStatuslineBarText(
   ctx: ExtensionContext,
   bar: string,
-  background: "selectedBg" | "toolErrorBg" = "selectedBg",
+  background: "userMessageBg" | "toolErrorBg",
   model?: CodexUsageModel,
 ): string {
   const label = ctx.ui.theme.fg("accent", activeUsageLabel(model));
@@ -1260,7 +1331,7 @@ function formatStatuslineLoading(
   return formatStatuslineBarText(
     ctx,
     formatCodexUsageLoadingBar(frame),
-    "selectedBg",
+    "userMessageBg",
     model,
   );
 }
@@ -1391,7 +1462,10 @@ export function canReuseCachedReport(
   report: CodexUsageReport,
   model: CodexUsageModel | undefined,
 ): boolean {
-  return selectActiveUsageSnapshot(report, model) !== undefined;
+  return (
+    selectActiveUsageSnapshot(report, model) !== undefined ||
+    report.credits !== undefined
+  );
 }
 
 export function isFullyAvailableReport(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llblab/pi-codex-usage",
-  "version": "0.9.4",
+  "version": "0.10.1",
   "private": false,
   "description": "Minimal Pi extension that shows primary Codex ChatGPT subscription usage limits",
   "keywords": [