@astrosheep/pi-quota 0.8.1 → 0.9.0

package/README.md

@@ -18,7 +18,7 @@ Reload Pi after installing or changing settings. Use `/usage` for a persistent,
 
 A percentage always means **remaining**, not used. Unknown is `[?] ?`; an expired reset is `↺ due` with the stale percentage hidden until the next successful query. A wallet balance never becomes a percentage. Historical spend is not a quota.
 
-The footer shows at most two windows and `+N` for additional ones, for the current provider only. `/usage` queries all configured supported providers and shows successful results with all windows, grouped by provider; it does not open a modal or send messages to the model. Snapshots remain frozen at capture time. Query failures replace stale values in the footer and appear in `/usage --all`.
+The footer shows at most two windows — set `quotaUsage.footerWindows` to a positive integer or `"all"` to show more — and `+N` for additional ones, for the current provider only. `/usage` queries all configured supported providers and shows successful results with all windows, grouped by provider; it does not open a modal or send messages to the model. Snapshots remain frozen at capture time. Query failures replace stale values in the footer and appear in `/usage --all`.
 
 ## Configure providers
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@astrosheep/pi-quota",
-  "version": "0.8.1",
+  "version": "0.9.0",
   "publishConfig": {
     "access": "public"
   },

package/src/bar/quota.ts

@@ -24,14 +24,18 @@ function windowSpans(window: QuotaWindow, now: number): Span[] {
   ];
 }
 
-export function quotaElement(state: QuotaState, now = Date.now()): BarElement {
+// Footer window cap: a positive count, or 'all' for every shared window.
+export type FooterWindows = number | 'all';
+
+export function quotaElement(state: QuotaState, now = Date.now(), maxWindows: FooterWindows = 2): BarElement {
   const element: BarElement = { id: 'quota', spans: [] };
   if (state.kind === 'hidden') return element;
   if (state.kind === 'loading') return { ...element, spans: [{ text: 'Quota …', tone: 'dim' }] };
   if (state.kind === 'error') return { ...element, spans: [{ text: `Quota ! ${errorText[state.code]}`, tone: 'warning' }] };
   const windows = state.snapshot.windows;
   const shared = windows.filter(window => !window.scope);
-  const shown = (shared.length ? shared : windows).slice(0, 2);
+  const eligible = shared.length ? shared : windows;
+  const shown = maxWindows === 'all' ? eligible : eligible.slice(0, maxWindows);
   const spans: Span[] = state.snapshot.balance ? balanceSpans(state.snapshot.balance)
     : state.snapshot.allowance ? allowanceSpans(state.snapshot.allowance) : [];
   for (const window of shown) {
@@ -42,8 +46,8 @@ export function quotaElement(state: QuotaState, now = Date.now()): BarElement {
   return { ...element, spans };
 }
 
-export function renderFooter(state: QuotaState, paint: Paint = plain, now = Date.now()): string {
-  return renderBar([quotaElement(state, now)], paint);
+export function renderFooter(state: QuotaState, paint: Paint = plain, now = Date.now(), maxWindows: FooterWindows = 2): string {
+  return renderBar([quotaElement(state, now, maxWindows)], paint);
 }
 
 export function renderUsage(state: QuotaState, paint: Paint = plain, now = Date.now(), barWidth = 16): string[] {

package/src/config.ts

@@ -6,12 +6,13 @@ import { createYesCodeAdapter, validateYesCodeOptions } from './query/yescode.ts
 import type { NewApiOptions } from './query/new-api.ts';
 import type { YesCodeOptions } from './query/yescode.ts';
 import type { QuotaAdapter } from './query/types.ts';
+import type { FooterWindows } from './bar/quota.ts';
 
 export type ProviderQuotaConfig =
   | ({ adapter: 'new-api'; quotaPerUnit: number; currency: string } & Pick<NewApiOptions, 'dashboardAccessToken' | 'dashboardUserId'>)
   | ({ adapter: 'yescode' } & YesCodeOptions)
   | { adapter: 'deepseek' | 'sub2api' | 'opencode-go' };
-export interface QuotaConfig { providers: Record<string, ProviderQuotaConfig> }
+export interface QuotaConfig { providers: Record<string, ProviderQuotaConfig>; footerWindows?: FooterWindows }
 
 const DASHBOARD_KEYS = ['dashboardAccessToken', 'dashboardUserId'] as const;
 
@@ -29,10 +30,16 @@ function parseDashboardOptions(item: Record<string, unknown>): Pick<NewApiOption
   return result;
 }
 
+function parseFooterWindows(value: unknown): FooterWindows {
+  if (value === 'all') return value;
+  if (typeof value === 'number' && Number.isSafeInteger(value) && value >= 1) return value;
+  throw new QuotaConfigError();
+}
+
 export class QuotaConfigError extends Error {
   constructor() {
     // Settings may accidentally contain secrets. Never echo content/values.
-    super('Invalid settings.json quotaUsage. Use providers with adapter "new-api", "deepseek", "sub2api", "opencode-go" or "yescode".');
+    super('Invalid settings.json quotaUsage. Use providers with adapter "new-api", "deepseek", "sub2api", "opencode-go" or "yescode", and optionally footerWindows as a positive integer or "all".');
   }
 }
 
@@ -41,7 +48,7 @@ function record(value: unknown): value is Record<string, unknown> {
 }
 
 export function parseQuotaConfig(value: unknown): QuotaConfig {
-  if (!record(value) || Object.keys(value).some(key => key !== 'providers')
+  if (!record(value) || Object.keys(value).some(key => key !== 'providers' && key !== 'footerWindows')
     || !record(value.providers) || Object.keys(value.providers).length > 64) throw new QuotaConfigError();
   const providers: Record<string, ProviderQuotaConfig> = Object.create(null);
   for (const [provider, item] of Object.entries(value.providers)) {
@@ -76,7 +83,14 @@ export function parseQuotaConfig(value: unknown): QuotaConfig {
       providers[provider] = { adapter: 'new-api', ...settings, ...parseDashboardOptions(item) };
     } catch { throw new QuotaConfigError(); }
   }
-  return { providers };
+  const config: QuotaConfig = { providers };
+  if (value.footerWindows !== undefined) config.footerWindows = parseFooterWindows(value.footerWindows);
+  return config;
+}
+
+export function loadQuotaFooterWindows(settings: unknown): FooterWindows | undefined {
+  if (!record(settings) || settings.quotaUsage === undefined) return undefined;
+  return parseQuotaConfig(settings.quotaUsage).footerWindows;
 }
 
 export function loadQuotaAdaptersFromSettings(settings: unknown): QuotaAdapter[] {

package/src/extension.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI, ExtensionContext } from '@earendil-works/pi-coding-agent';
 import { renderFooter } from './bar/quota.ts';
+import type { FooterWindows } from './bar/quota.ts';
 import { captureAllUsage, USAGE_ENTRY, usageCardComponent, usageLines, compactUsageLines } from './bar/card.ts';
 import type { UsageCard } from './bar/card.ts';
 import { codexAdapter } from './query/codex.ts';
@@ -23,6 +24,7 @@ export interface QuotaExtensionOptions {
   adapters?: readonly QuotaAdapter[];
   providers?: readonly string[]; // Extra configured quota providers to include in /usage
   footer?: boolean; // false: query + /usage + structured events only
+  footerWindows?: FooterWindows; // footer window cap; default 2, 'all' shows every window
   intervalMs?: number;
   timeoutMs?: number;
 }
@@ -57,7 +59,7 @@ export function createQuotaExtension(options: QuotaExtensionOptions = {}) {
         const ctx = current;
         const paint = process.env.NO_COLOR !== undefined ? undefined
           : (tone: Parameters<typeof ctx.ui.theme.fg>[0], text: string) => ctx.ui.theme.fg(tone, text);
-        const text = renderFooter(controller.state, paint) || undefined;
+        const text = renderFooter(controller.state, paint, Date.now(), options.footerWindows) || undefined;
         if (text !== lastStatus) {
           ctx.ui.setStatus(STATUS_KEY, text);
           lastStatus = text;

package/src/index.ts

@@ -1,11 +1,13 @@
 import { getAgentDir, SettingsManager } from '@earendil-works/pi-coding-agent';
 import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
 import { createQuotaExtension } from './extension.ts';
-import { loadQuotaAdaptersFromSettings, QuotaConfigError } from './config.ts';
+import { loadQuotaAdaptersFromSettings, loadQuotaFooterWindows, QuotaConfigError } from './config.ts';
+import type { FooterWindows } from './bar/quota.ts';
 import type { QuotaAdapter } from './query/types.ts';
 
 export default function (pi: ExtensionAPI): void {
   let adapters: QuotaAdapter[] = [];
+  let footerWindows: FooterWindows | undefined;
   try {
     // Quota provider bindings belong to the user's global Pi settings. Project
     // settings are intentionally not read here: a project must not redirect a
@@ -14,11 +16,12 @@ export default function (pi: ExtensionAPI): void {
       projectTrusted: false,
     }).getGlobalSettings() as unknown;
     adapters = loadQuotaAdaptersFromSettings(settings);
+    footerWindows = loadQuotaFooterWindows(settings);
   } catch {
     pi.on('session_start', (_event, ctx) => {
       if (ctx.hasUI) ctx.ui.notify(new QuotaConfigError().message, 'warning');
     });
   }
   // Invalid optional settings never disable the built-in Codex/Kimi/OpenCode Go adapters.
-  createQuotaExtension({ adapters })(pi);
+  createQuotaExtension({ adapters, footerWindows })(pi);
 }