@henryqw/pi-footer 0.2.2 → 0.4.0

package/README.md

@@ -23,17 +23,18 @@ pi install npm:@henryqw/pi-footer
 
 ```text
 pi-packages · clear-field-f8d2 · PR #123 · approved
-↑ 12.4k · ↓ 2.1k · ↺ 84.3% · $ 0.127 · ◔ 36.8%    gpt-5.6-luna • high
+↑ 12.4k · ↓ 2.1k · ↺ 84.3% · ⚡ 87.4 t/s · $ 0.127 · ◔ 36.8%    gpt-5.6-luna • high
 Codex #1 · 50% · 7d 1d 1h 22m
+●  🐴 ponytail: ⚡ FULL
 ```
 
 First line shows repository, branch, and pull request status (`pi-pr`) after clickable checkout link. Linked-worktree branches drop generated `worktree/` prefix.
 
-Second line shows cumulative input tokens, output tokens, latest cache-hit rate, estimated cost, and context usage. Unavailable values render as `—` without a misleading percent sign. Active model and thinking level are right-aligned.
+Second line shows cumulative input tokens, output tokens, latest cache-hit rate, tokens per second of the most recent assistant response, estimated cost, and context usage. Unavailable values render as `—` without a misleading percent sign. Active model and thinking level are right-aligned.
 
 `off` uses the same dim grey as the model name. Active levels use an ANSI-256 gradient: green `minimal`, yellow-green `low`, lime `medium`, yellow `high`, orange `xhigh`, and red `max`. `ultra` renders as a rainbow when the runtime supplies it. Pi 0.84.2 does not yet accept `ultra`, so that footer path remains unreachable until Pi adds it.
 
-Final line renders every non-empty status emitted through `ctx.ui.setStatus()`, sorted by status key (excluding `pi-pr`, which appears on the first line). Producer text, spacing, colors, links, and glyphs are preserved, including Ponytail mode and Codex quota.
+Third line is reserved for non-empty statuses from `@henryqw` extensions, currently Codex quota. Fourth line renders statuses from all other extensions, including Ponytail and `pi-rewind`. If only external statuses exist, third line stays blank so external output remains on line four. Statuses are sorted by key; producer text, spacing, colors, links, and glyphs are preserved.
 
 ## Clickable checkout
 

package/extensions/footer.ts

@@ -12,6 +12,7 @@ const THINKING_COLORS = {
 	xhigh: 208,
 	max: 196,
 } as const;
+const HENRY_STATUS_KEYS = new Set(["pi-multi-codex"]);
 
 function formatTokens(count: number): string {
 	if (count < 1_000) return `${count}`;
@@ -53,6 +54,19 @@ export default function footerExtension(pi: ExtensionAPI): void {
 		const commonName = commonDir && basename(commonDir) === ".git" ? basename(dirname(commonDir)) : undefined;
 		const repo = git.code === 0 ? commonName && commonName !== rootName ? commonName : rootName : basename(ctx.cwd);
 
+		let tps: number | undefined;
+		let assistantStartedAt: number | undefined;
+		pi.on("message_start", async (event) => {
+			if (event.message.role === "assistant") assistantStartedAt = performance.now();
+		});
+		pi.on("message_end", async (event) => {
+			if (event.message.role !== "assistant") return;
+			const output = event.message.usage?.output ?? 0;
+			const seconds = assistantStartedAt === undefined ? 0 : (performance.now() - assistantStartedAt) / 1000;
+			assistantStartedAt = undefined;
+			tps = seconds > 0 ? output / seconds : undefined;
+		});
+
 		// ponytail: keyed on length + last entry (sessions are append-only); revisit if entries ever mutate in place.
 		let usageKey: string | undefined;
 		let input = 0;
@@ -106,8 +120,14 @@ export default function footerExtension(pi: ExtensionAPI): void {
 					const statuses = [...extensionStatuses]
 						.filter(([key]) => key !== "pi-pr")
 						.sort(([a], [b]) => a.localeCompare(b))
-						.map(([, text]) => sanitizeStatus(text))
-						.filter(Boolean);
+						.map(([key, text]) => [key, sanitizeStatus(text)] as const)
+						.filter(([, text]) => Boolean(text));
+					const henryStatuses = statuses
+						.filter(([key]) => HENRY_STATUS_KEYS.has(key))
+						.map(([, text]) => text);
+					const externalStatuses = statuses
+						.filter(([key]) => !HENRY_STATUS_KEYS.has(key))
+						.map(([, text]) => text);
 					const thinking = String(ctx.thinkingLevel ?? "off");
 					const thinkingColor = THINKING_COLORS[thinking as keyof typeof THINKING_COLORS];
 					const ellipsis = theme.fg("dim", "…");
@@ -115,7 +135,8 @@ export default function footerExtension(pi: ExtensionAPI): void {
 						`↑ ${formatTokens(input)}`,
 						`↓ ${formatTokens(output)}`,
 						`↺ ${cacheRate === undefined ? "—" : `${cacheRate.toFixed(1)}%`}`,
-						`$ ${cost.toFixed(3)}`,
+						`⚡ ${tps === undefined ? "—" : `${tps.toFixed(1)} t/s`}`,
+					`$ ${cost.toFixed(3)}`,
 						`◔ ${context == null ? "—" : `${context.toFixed(1)}%`}`,
 					].join(" · "));
 					const thinkingText = thinking === "ultra"
@@ -130,7 +151,8 @@ export default function footerExtension(pi: ExtensionAPI): void {
 						firstLine,
 						align(usage, model, width, ellipsis),
 					];
-					if (statuses.length) lines.push(statuses.join(" "));
+					if (henryStatuses.length || externalStatuses.length) lines.push(henryStatuses.join(" "));
+					if (externalStatuses.length) lines.push(externalStatuses.join(" "));
 					return lines.map((line) => truncateToWidth(line, width, ellipsis));
 				},
 			};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-footer",
-  "version": "0.2.2",
+  "version": "0.4.0",
   "description": "Show concise repository, branch, and usage details in the Pi footer.",
   "keywords": [
     "pi-package",