@diegopetrucci/pi-extensions 0.1.10 → 0.1.11

package/README.md

@@ -1,39 +1,33 @@
 # pi-extensions
 
-A collection of [pi](https://github.com/badlogic/pi-mono) agent extensions I made.
+A collection of [pi](https://github.com/badlogic/pi-mono) agent extensions I made:
 
-## Included extensions
+- [`minimal-footer`](./extensions/minimal-footer): Replaces pi's built-in footer with a minimal two-line layout: branch/repo on the first line, context/model on the second, plus OpenAI Codex 5-hour and 7-day usage when available.
+- [`oracle`](./extensions/oracle): Adds an Amp-style read-only oracle tool that auto-selects the strongest reasoning model on the current provider/subscription, covers pi’s built-in providers with hardcoded rankings, sets reasoning to xhigh by default, and shows live status while running.
+- [`permission-gate`](./extensions/permission-gate): Prompts for confirmation before dangerous bash commands like `rm -rf`, `sudo`, and `chmod 777`.
+- [`confirm-destructive`](./extensions/confirm-destructive): Confirms before destructive session actions like clear, switch, and fork.
+- [`notify`](./extensions/notify): Sends configurable terminal, desktop, bell, and sound notifications when pi finishes and is ready for input.
 
-| Extension | Description |
-|---|---|
-| [`minimal-footer`](./extensions/minimal-footer) | Replaces pi's built-in footer with a minimal two-line layout: branch/repo on the first line, context/model on the second. |
-| [`oracle`](./extensions/oracle) | Adds an Amp-style read-only oracle tool that auto-selects the strongest reasoning model on the current provider/subscription, covers pi’s built-in providers with hardcoded rankings, sets reasoning to xhigh by default, and shows live status while running. |
-| [`permission-gate`](./extensions/permission-gate) | Prompts for confirmation before dangerous bash commands like `rm -rf`, `sudo`, and `chmod 777`. |
-| [`confirm-destructive`](./extensions/confirm-destructive) | Confirms before destructive session actions like clear, switch, and fork. |
-| [`notify`](./extensions/notify) | Sends configurable terminal, desktop, bell, and sound notifications when pi finishes and is ready for input. |
+(For the full list of pi extensions I use, [check out my dotfiles](https://github.com/diegopetrucci/dot/blob/main/.pi/agent/settings.json).)
 
 ## Install
 
-### GitHub
-
-Install the repo as a pi package:
+Full collection:
 
 ```bash
-pi install git:github.com/diegopetrucci/pi-extensions
+pi install npm:@diegopetrucci/pi-extensions
 ```
 
-Or pin to a tagged version:
+Or pin the GitHub package to this release:
 
 ```bash
-pi install git:github.com/diegopetrucci/pi-extensions@v0.1.10
+pi install git:github.com/diegopetrucci/pi-extensions@v0.1.11
 ```
 
-### npm
-
-Install the full collection from npm:
+Or a specific extension:
 
 ```bash
-pi install npm:@diegopetrucci/pi-extensions
+pi install npm:@diegopetrucci/pi-oracle
 ```
 
 Then reload pi:
@@ -41,105 +35,3 @@ Then reload pi:
 ```text
 /reload
 ```
-
-## Install only one extension
-
-If you only want one extension, you have two options.
-
-### Option 1: install the standalone npm package
-
-```bash
-pi install npm:@diegopetrucci/pi-minimal-footer
-```
-
-```bash
-pi install npm:@diegopetrucci/pi-oracle
-```
-
-```bash
-pi install npm:@diegopetrucci/pi-permission-gate
-```
-
-```bash
-pi install npm:@diegopetrucci/pi-confirm-destructive
-```
-
-```bash
-pi install npm:@diegopetrucci/pi-notify
-```
-
-### Option 2: filter the repo package
-
-If you prefer the collection package, you can filter it in your pi settings.
-
-Minimal footer only:
-
-```json
-{
-  "packages": [
-    {
-      "source": "npm:@diegopetrucci/pi-extensions",
-      "extensions": ["extensions/minimal-footer/index.ts"]
-    }
-  ]
-}
-```
-
-Oracle only:
-
-```json
-{
-  "packages": [
-    {
-      "source": "npm:@diegopetrucci/pi-extensions",
-      "extensions": ["extensions/oracle/index.ts"]
-    }
-  ]
-}
-```
-
-Permission gate only:
-
-```json
-{
-  "packages": [
-    {
-      "source": "npm:@diegopetrucci/pi-extensions",
-      "extensions": ["extensions/permission-gate/index.ts"]
-    }
-  ]
-}
-```
-
-Confirm destructive only:
-
-```json
-{
-  "packages": [
-    {
-      "source": "npm:@diegopetrucci/pi-extensions",
-      "extensions": ["extensions/confirm-destructive/index.ts"]
-    }
-  ]
-}
-```
-
-Notify only:
-
-```json
-{
-  "packages": [
-    {
-      "source": "npm:@diegopetrucci/pi-extensions",
-      "extensions": ["extensions/notify/index.ts"]
-    }
-  ]
-}
-```
-
-## npm publishing
-
-The repo is set up to support both:
-
-- the collection package: `@diegopetrucci/pi-extensions`
-- standalone extension packages like `@diegopetrucci/pi-minimal-footer`, `@diegopetrucci/pi-oracle`, `@diegopetrucci/pi-permission-gate`, `@diegopetrucci/pi-confirm-destructive`, and `@diegopetrucci/pi-notify`

package/extensions/minimal-footer/README.md

@@ -10,6 +10,7 @@ It replaces pi's built-in footer with a cleaner two-line layout that focuses on
 - current repo name
 - current context percentage
 - current model and thinking level
+- OpenAI Codex 5-hour and 7-day usage when available
 
 ## Layout
 
@@ -27,6 +28,12 @@ fix/remove-detached-image-tasks                     SendItToMy
 44.1%                                              gpt-5.4 high
 ```
 
+When using `openai-codex`, the bottom-left line also includes subscription usage:
+
+```text
+44.1% · 5h 12% · 7d 38%
+```
+
 On narrow terminals it falls back to one item per line.
 
 ## Install
@@ -60,6 +67,7 @@ Then reload pi:
 - **Top left:** current git branch
 - **Top right:** current repo directory name
 - **Bottom left:** current context usage percentage
+- **Bottom left on `openai-codex`:** current context usage percentage plus 5-hour and 7-day Codex usage
 - **Bottom right:** model id and thinking level
 
 ## Publishing notes
@@ -72,3 +80,5 @@ This extension also lives inside the broader [`pi-extensions`](../../README.md)
 - Uses pi footer data for git branch updates.
 - Shows only context percentage, not context window size.
 - Shows the model id rather than a provider-specific display label.
+- For `openai-codex`, reads pi's stored OAuth login and fetches usage from ChatGPT's backend usage endpoint.
+- Usage is cached briefly in memory and refreshed after turns.

package/extensions/minimal-footer/index.ts

@@ -1,14 +1,109 @@
 import { basename } from "node:path";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import {
+	AuthStorage,
+	type ExtensionAPI,
+	type ExtensionContext,
+} from "@mariozechner/pi-coding-agent";
 import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+import {
+	fetchOpenAICodexUsage,
+	formatUsageSummary,
+	isOpenAICodexProvider,
+	type UsageSnapshot,
+} from "./openai-usage";
+
+const USAGE_CACHE_TTL_MS = 5 * 60 * 1000;
+const USAGE_REQUEST_TIMEOUT_MS = 10 * 1000;
+
+type UsageSessionState = {
+	authStorage: AuthStorage;
+	snapshot?: UsageSnapshot;
+	lastFetchedAt?: number;
+	loading: boolean;
+	error?: string;
+	inflight?: Promise<void>;
+	requestRender?: () => void;
+};
+
+function clearUsageState(state: UsageSessionState): void {
+	state.snapshot = undefined;
+	state.lastFetchedAt = undefined;
+	state.loading = false;
+	state.error = undefined;
+}
+
+async function refreshUsageIfNeeded(
+	ctx: ExtensionContext,
+	state: UsageSessionState,
+	force = false,
+): Promise<void> {
+	if (!isOpenAICodexProvider(ctx.model?.provider)) {
+		clearUsageState(state);
+		state.requestRender?.();
+		return;
+	}
+
+	const now = Date.now();
+	if (
+		!force &&
+		state.lastFetchedAt &&
+		now - state.lastFetchedAt < USAGE_CACHE_TTL_MS
+	) {
+		return;
+	}
+
+	if (state.inflight) {
+		return state.inflight;
+	}
+
+	state.loading = true;
+	state.requestRender?.();
+	state.inflight = (async () => {
+		try {
+			const snapshot = await fetchOpenAICodexUsage(state.authStorage, {
+				timeoutMs: USAGE_REQUEST_TIMEOUT_MS,
+			});
+			if (snapshot) {
+				state.snapshot = snapshot;
+				state.lastFetchedAt = snapshot.fetchedAt;
+				state.error = undefined;
+			} else {
+				state.snapshot = undefined;
+				state.lastFetchedAt = Date.now();
+				state.error = undefined;
+			}
+		} catch (error) {
+			state.error = error instanceof Error ? error.message : String(error);
+		} finally {
+			state.loading = false;
+			state.inflight = undefined;
+			state.requestRender?.();
+		}
+	})();
+
+	return state.inflight;
+}
 
 export default function (pi: ExtensionAPI) {
-	pi.on("session_start", async (_event, ctx) => {
+	const states = new WeakMap<object, UsageSessionState>();
+
+	pi.on("session_start", (_event, ctx) => {
+		const state: UsageSessionState = {
+			authStorage: AuthStorage.create(),
+			loading: false,
+		};
+		states.set(ctx.sessionManager, state);
+
 		ctx.ui.setFooter((tui, theme, footerData) => {
 			const unsub = footerData.onBranchChange(() => tui.requestRender());
+			state.requestRender = () => tui.requestRender();
+			void refreshUsageIfNeeded(ctx, state);
 
 			return {
-				dispose: unsub,
+				dispose() {
+					if (state.requestRender) state.requestRender = undefined;
+					unsub();
+				},
 				invalidate() {},
 				render(width: number): string[] {
 					const repo = basename(ctx.cwd);
@@ -16,6 +111,10 @@ export default function (pi: ExtensionAPI) {
 
 					const usage = ctx.getContextUsage();
 					const context = usage?.percent == null ? "?" : `${usage.percent.toFixed(1)}%`;
+					const usageSummary = isOpenAICodexProvider(ctx.model?.provider)
+						? formatUsageSummary(state.snapshot)
+						: undefined;
+					const contextText = usageSummary ? `${context} · ${usageSummary}` : context;
 
 					const model = ctx.model?.id ?? "no-model";
 					const thinking = pi.getThinkingLevel();
@@ -23,7 +122,7 @@ export default function (pi: ExtensionAPI) {
 
 					const branchStyled = theme.fg("dim", branch);
 					const repoStyled = theme.fg("dim", repo);
-					const contextStyled = theme.fg("dim", context);
+					const contextStyled = theme.fg("dim", contextText);
 					const modelStyled = theme.fg("dim", modelText);
 
 					const renderSplitLine = (left: string, right: string): string => {
@@ -51,4 +150,16 @@ export default function (pi: ExtensionAPI) {
 			};
 		});
 	});
+
+	pi.on("model_select", (_event, ctx) => {
+		const state = states.get(ctx.sessionManager);
+		if (!state) return;
+		void refreshUsageIfNeeded(ctx, state, true);
+	});
+
+	pi.on("turn_end", (_event, ctx) => {
+		const state = states.get(ctx.sessionManager);
+		if (!state) return;
+		void refreshUsageIfNeeded(ctx, state);
+	});
 }

package/extensions/minimal-footer/openai-usage.ts

@@ -0,0 +1,121 @@
+import { AuthStorage } from "@mariozechner/pi-coding-agent";
+
+const PROVIDER_ID = "openai-codex";
+
+interface WhamUsageWindow {
+	reset_at?: number;
+	used_percent?: number;
+}
+
+interface WhamUsageResponse {
+	rate_limit?: {
+		primary_window?: WhamUsageWindow;
+		secondary_window?: WhamUsageWindow;
+	};
+}
+
+export interface UsageWindow {
+	usedPercent?: number;
+	resetAt?: number;
+}
+
+export interface UsageSnapshot {
+	primary?: UsageWindow;
+	secondary?: UsageWindow;
+	fetchedAt: number;
+}
+
+function normalizeUsedPercent(value?: number): number | undefined {
+	if (typeof value !== "number" || !Number.isFinite(value)) return undefined;
+	return Math.min(100, Math.max(0, value));
+}
+
+function normalizeResetAt(value?: number): number | undefined {
+	if (typeof value !== "number" || !Number.isFinite(value)) return undefined;
+	return value * 1000;
+}
+
+function parseUsageWindow(window?: WhamUsageWindow): UsageWindow | undefined {
+	if (!window) return undefined;
+	const usedPercent = normalizeUsedPercent(window.used_percent);
+	const resetAt = normalizeResetAt(window.reset_at);
+	if (usedPercent === undefined && resetAt === undefined) return undefined;
+	return { usedPercent, resetAt };
+}
+
+function parseUsageSnapshot(data: WhamUsageResponse): Omit<UsageSnapshot, "fetchedAt"> {
+	return {
+		primary: parseUsageWindow(data.rate_limit?.primary_window),
+		secondary: parseUsageWindow(data.rate_limit?.secondary_window),
+	};
+}
+
+function getOAuthAccountId(authStorage: AuthStorage): string | undefined {
+	const credential = authStorage.get(PROVIDER_ID);
+	if (!credential || credential.type !== "oauth") return undefined;
+	const accountId = (credential as { accountId?: unknown }).accountId;
+	return typeof accountId === "string" && accountId.trim()
+		? accountId.trim()
+		: undefined;
+}
+
+function formatUsagePercent(value?: number): string | undefined {
+	if (typeof value !== "number" || !Number.isFinite(value)) return undefined;
+	return `${Math.round(value)}%`;
+}
+
+export function isOpenAICodexProvider(provider?: string): boolean {
+	return provider === PROVIDER_ID;
+}
+
+export function formatUsageSummary(snapshot?: UsageSnapshot): string | undefined {
+	if (!snapshot) return undefined;
+
+	const primary = formatUsagePercent(snapshot.primary?.usedPercent);
+	const secondary = formatUsagePercent(snapshot.secondary?.usedPercent);
+	const parts: string[] = [];
+
+	if (primary) parts.push(`5h ${primary}`);
+	if (secondary) parts.push(`7d ${secondary}`);
+
+	return parts.length > 0 ? parts.join(" · ") : undefined;
+}
+
+export async function fetchOpenAICodexUsage(
+	authStorage: AuthStorage,
+	options?: { timeoutMs?: number },
+): Promise<UsageSnapshot | undefined> {
+	const accessToken = await authStorage.getApiKey(PROVIDER_ID, {
+		includeFallback: false,
+	});
+	if (!accessToken) return undefined;
+
+	authStorage.reload();
+	const accountId = getOAuthAccountId(authStorage);
+	const controller = new AbortController();
+	const timeoutMs = options?.timeoutMs ?? 10_000;
+	const timeout = setTimeout(() => controller.abort(), timeoutMs);
+
+	try {
+		const headers: Record<string, string> = {
+			Authorization: `Bearer ${accessToken}`,
+			Accept: "application/json",
+		};
+		if (accountId) {
+			headers["ChatGPT-Account-Id"] = accountId;
+		}
+
+		const response = await fetch("https://chatgpt.com/backend-api/wham/usage", {
+			headers,
+			signal: controller.signal,
+		});
+		if (!response.ok) {
+			throw new Error(`Usage request failed: ${response.status}`);
+		}
+
+		const data = (await response.json()) as WhamUsageResponse;
+		return { ...parseUsageSnapshot(data), fetchedAt: Date.now() };
+	} finally {
+		clearTimeout(timeout);
+	}
+}

package/extensions/minimal-footer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-minimal-footer",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "A minimal custom footer for pi.",
   "keywords": ["pi-package", "pi", "terminal", "footer"],
   "license": "MIT",
@@ -11,6 +11,7 @@
   },
   "files": [
     "index.ts",
+    "openai-usage.ts",
     "README.md"
   ],
   "publishConfig": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "A collection of pi extensions, including a minimal custom footer, an Amp-style oracle, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
   "keywords": ["pi-package", "pi", "terminal", "agent"],
   "license": "MIT",