@henryqw/pi-footer 0.5.5 → 0.5.6

package/README.md

@@ -32,7 +32,7 @@ pi-packages · clear-field-f8d2 · PR #123 · approved    Codex #1 · 50% · 7d
 ```
 
 - The first line shows the repository, branch, and `pi-pr` pull request status after the clickable checkout link. Linked-worktree branches drop the generated `worktree/` prefix.
-- The second line shows cumulative input tokens, output tokens, latest cache-hit rate, and tokens per second for the most recent assistant response. It also shows estimated cost and context usage. The active model and thinking level are right-aligned.
+- The second line shows cumulative input tokens, output tokens, latest cache-hit rate, and tokens per second for the most recent assistant response. It also shows estimated cost and context usage. Totals include reported tool usage and finished `pi-subagent` background workflows. The active model and thinking level are right-aligned.
 - The third line shows other extension statuses on the left and cumulative agent-work time on the right, beneath the active model.
 
 Unavailable values render as `—` without a misleading percent sign.

package/extensions/footer.ts

@@ -1,5 +1,4 @@
 import { basename, dirname } from "node:path";
-import type { Usage } from "@earendil-works/pi-ai";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { getCapabilities, hyperlink, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 import { configuredOpenUri } from "@henryqw/pi-open-in/open-uri";
@@ -14,11 +13,26 @@ const THINKING_COLORS = {
 } as const;
 const HENRY_STATUS_KEY = "pi-multi-codex";
 const AGENT_TIME_ENTRY = "pi-footer:agent-work";
+const SUBAGENT_BACKGROUND_RESULT = "subagent-background-result";
 
-function isValidMilliseconds(value: unknown): value is number {
+type CountedUsage = { input: number; output: number; cost: { total: number } };
+
+function isNonNegativeNumber(value: unknown): value is number {
 	return typeof value === "number" && Number.isFinite(value) && value >= 0;
 }
 
+function subagentBackgroundUsage(details: unknown): CountedUsage | undefined {
+	if (!details || typeof details !== "object" || Array.isArray(details)) return;
+	const usage = (details as Record<string, unknown>).usage;
+	if (!usage || typeof usage !== "object" || Array.isArray(usage)) return;
+	const record = usage as Record<string, unknown>;
+	const cost = record.cost;
+	if (!cost || typeof cost !== "object" || Array.isArray(cost)) return;
+	const total = (cost as Record<string, unknown>).total;
+	if (!isNonNegativeNumber(record.input) || !isNonNegativeNumber(record.output) || !isNonNegativeNumber(total)) return;
+	return { input: record.input, output: record.output, cost: { total } };
+}
+
 function formatTokens(count: number): string {
 	if (count < 1_000) return `${count}`;
 	if (count < 1_000_000) return `${(count / 1_000).toFixed(1)}k`;
@@ -129,7 +143,7 @@ export default function footerExtension(pi: ExtensionAPI): void {
 		// Latest valid entry wins; stored data is untrusted.
 		activeMilliseconds = 0;
 		for (const entry of ctx.sessionManager.getEntries()) {
-			if (entry.type === "custom" && entry.customType === AGENT_TIME_ENTRY && isValidMilliseconds(entry.data)) {
+			if (entry.type === "custom" && entry.customType === AGENT_TIME_ENTRY && isNonNegativeNumber(entry.data)) {
 				activeMilliseconds = entry.data;
 			}
 		}
@@ -169,7 +183,7 @@ export default function footerExtension(pi: ExtensionAPI): void {
 			output = 0;
 			cost = 0;
 			cacheRate = undefined;
-			const add = (usage: Usage | undefined) => {
+			const add = (usage: CountedUsage | undefined) => {
 				if (!usage) return;
 				input += usage.input;
 				output += usage.output;
@@ -184,6 +198,8 @@ export default function footerExtension(pi: ExtensionAPI): void {
 					add(usage);
 				} else if (entry.type === "message" && entry.message.role === "toolResult") {
 					add(entry.message.usage);
+				} else if (entry.type === "custom_message" && entry.customType === SUBAGENT_BACKGROUND_RESULT) {
+					add(subagentBackgroundUsage(entry.details));
 				} else if (entry.type === "branch_summary" || entry.type === "compaction") {
 					add(entry.usage);
 				}

package/package.json

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