@badliveware/pi-footer-framework 0.2.2 → 0.3.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## 0.3.0
+
+- Added cache read/write token and cost fields to the footer `stats` source so custom renderers can show cache activity alongside input, output, totals, and cost.
+- Improved footer render performance by caching session stats and text-width metadata for repeated renders of unchanged content.
+- Reduced startup and render work by deferring slower extension probes and rendering styled footer text in contiguous cell runs instead of per-cell wrappers.
+
 ## 0.2.2
 
 - Fixed custom footer rendering so layouts with all visible items on line 1 render as a single footer line instead of reserving an empty second line.

package/index.ts

@@ -2,9 +2,12 @@ import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
 import { fileURLToPath, pathToFileURL } from "node:url";
-import { Type } from "@mariozechner/pi-ai";
-import { defineTool, type ExtensionAPI, type ExtensionContext } from "@mariozechner/pi-coding-agent";
-import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+import { Type } from "@earendil-works/pi-ai";
+import { defineTool, type ExtensionAPI, type ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import { renderFooterCellRuns } from "./src/cell-runs.ts";
+import { createFooterStatsCache, type FooterStats } from "./src/stats-cache.ts";
+import { createFooterTextMetricsCache } from "./src/text-metrics-cache.ts";
 
 type ChecksState = "pass" | "fail" | "running" | "unknown";
 export type FooterAnchorMode = "gap" | "left" | "center" | "right" | "spread";
@@ -58,6 +61,8 @@ export interface FooterCell {
 	itemId?: string;
 	continuation?: boolean;
 	filler?: boolean;
+	prefix?: string;
+	suffix?: string;
 }
 
 export interface FooterColumnItem {
@@ -148,7 +153,7 @@ export function footerCellsFromText(text: string, itemId?: string): FooterCell[]
 				pendingZeroWidthPlain += cluster;
 			}
 		} else {
-			cells.push({ raw: `${pendingZeroWidthRaw}${activeCellPrefix(state)}${cluster}${activeCellSuffix(state)}`, plainText: `${pendingZeroWidthPlain}${cluster}`, itemId });
+			cells.push({ raw: `${pendingZeroWidthRaw}${cluster}`, plainText: `${pendingZeroWidthPlain}${cluster}`, itemId, prefix: activeCellPrefix(state), suffix: activeCellSuffix(state) });
 			pendingZeroWidthRaw = "";
 			pendingZeroWidthPlain = "";
 			for (let i = 1; i < clusterWidth; i += 1) cells.push({ raw: "", plainText: "", itemId, continuation: true });
@@ -219,15 +224,7 @@ export function writeFooterText(cells: FooterCell[], start: number, text: string
 	}
 }
 
-export function renderFooterCells(cells: FooterCell[]): string {
-	let end = cells.length;
-	while (end > 0) {
-		const cell = cells[end - 1];
-		if (cell.continuation || !cell.filler) break;
-		end -= 1;
-	}
-	return cells.slice(0, end).map((cell) => (cell.continuation ? "" : cell.raw)).join("");
-}
+export const renderFooterCells = renderFooterCellRuns;
 
 export function resolveFooterColumn(column: FooterColumn | undefined, width: number, itemWidth: number): number | undefined {
 	if (column === undefined) return undefined;
@@ -1823,11 +1820,11 @@ export default function footerFramework(pi: ExtensionAPI): void {
 		return sorted.sort((a, b) => a.placement.order - b.placement.order || a.id.localeCompare(b.id));
 	}
 
-	function renderVisibilityDiagnostics(items: FooterItem[], lines: Array<{ line: FooterLine; plainText: string }>): FooterRenderDiagnostic[] {
+	function renderVisibilityDiagnostics(items: FooterItem[], lines: Array<{ line: FooterLine; plainText: string }>, plainTextFor: (text: string) => string = plainFooterText): FooterRenderDiagnostic[] {
 		const lineTextByNumber = new Map(lines.map((line) => [line.line, line.plainText]));
 		const diagnostics: FooterRenderDiagnostic[] = [];
 		for (const item of items) {
-			const itemPlainText = plainFooterText(item.text).trim();
+			const itemPlainText = plainTextFor(item.text).trim();
 			if (!itemPlainText) continue;
 			const linePlainText = lineTextByNumber.get(item.placement.line);
 			if (linePlainText === undefined) {
@@ -1898,7 +1895,7 @@ export default function footerFramework(pi: ExtensionAPI): void {
 
 	function buildPiSources(
 		footerData: { getGitBranch(): string | null; getExtensionStatuses(): ReadonlyMap<string, string> },
-		stats: { input: number; output: number; cost: number; inputText: string; outputText: string; costText: string; value: string },
+		stats: FooterStats,
 	): Record<string, FooterAdapterSourceValue> {
 		const sources: Record<string, FooterAdapterSourceValue> = {
 			cwd: { label: "cwd", value: currentCtx?.cwd ?? "", tone: "muted", data: { path: currentCtx?.cwd ?? "" } },
@@ -2004,7 +2001,7 @@ export default function footerFramework(pi: ExtensionAPI): void {
 	function collectItems(
 		theme: ExtensionContext["ui"]["theme"],
 		footerData: { getGitBranch(): string | null; getExtensionStatuses(): ReadonlyMap<string, string> },
-		stats: { input: number; output: number; cost: number; inputText: string; outputText: string; costText: string; value: string },
+		stats: FooterStats,
 		diagnostics: FooterTemplateDiagnostic[],
 	): FooterItem[] {
 		const items: FooterItem[] = [];
@@ -2141,6 +2138,8 @@ export default function footerFramework(pi: ExtensionAPI): void {
 		ctx.ui.setFooter((tui, theme, footerData) => {
 			requestRender = () => tui.requestRender();
 			const unsubscribe = footerData.onBranchChange(() => tui.requestRender());
+			const footerStats = createFooterStatsCache(formatTokens);
+			const textMetrics = createFooterTextMetricsCache(plainFooterText, visibleWidth);
 
 			return {
 				dispose() {
@@ -2149,25 +2148,7 @@ export default function footerFramework(pi: ExtensionAPI): void {
 				},
 				invalidate() {},
 					render(width: number): string[] {
-					let input = 0;
-					let output = 0;
-					let cost = 0;
-					for (const entry of ctx.sessionManager.getEntries()) {
-						if (entry.type !== "message" || entry.message.role !== "assistant") continue;
-						input += entry.message.usage.input;
-						output += entry.message.usage.output;
-						cost += entry.message.usage.cost.total;
-					}
-
-					const stats = {
-						input,
-						output,
-						cost,
-						inputText: formatTokens(input),
-						outputText: formatTokens(output),
-						costText: cost.toFixed(3),
-						value: `↑${formatTokens(input)} ↓${formatTokens(output)} $${cost.toFixed(3)}`,
-					};
+					const stats = footerStats(ctx.sessionManager.getEntries());
 					const diagnostics: FooterTemplateDiagnostic[] = [];
 					const items = collectItems(theme, footerData, stats, diagnostics);
 					lastTemplateDiagnostics = diagnostics;
@@ -2175,21 +2156,24 @@ export default function footerFramework(pi: ExtensionAPI): void {
 					const lineResults = Array.from({ length: maxLine }, (_, index) => {
 						const line = index + 1;
 						const result = renderFooterLine(theme, width, items, line, getLineAnchor(settings, line));
-						return { line, text: result.line, plainText: plainFooterText(result.line), layout: result.layout };
+						return { line, text: result.line, plainText: textMetrics(result.line).plainText, layout: result.layout };
 					});
 					const line1Result = lineResults[0];
 					const line2Result = lineResults[1];
-					const renderedItems = items.map((item) => ({
-						id: item.id,
-						line: item.placement.line,
-						zone: item.placement.zone,
-						order: item.placement.order,
-						column: item.placement.column,
-						width: visibleWidth(item.text),
-						plainText: plainFooterText(item.text),
-						renderSource: item.renderSource,
-						tokens: item.tokens,
-					}));
+					const renderedItems = items.map((item) => {
+						const metrics = textMetrics(item.text);
+						return {
+							id: item.id,
+							line: item.placement.line,
+							zone: item.placement.zone,
+							order: item.placement.order,
+							column: item.placement.column,
+							width: metrics.width,
+							plainText: metrics.plainText,
+							renderSource: item.renderSource,
+							tokens: item.tokens,
+						};
+					});
 
 					lastFooterSnapshot = {
 						width,
@@ -2202,7 +2186,7 @@ export default function footerFramework(pi: ExtensionAPI): void {
 						line2Layout: line2Result?.layout ?? renderFooterLine(theme, width, [], 2, getLineAnchor(settings, 2)).layout,
 						gitBranch: footerData.getGitBranch(),
 						renderedItems,
-						renderDiagnostics: renderVisibilityDiagnostics(items, lineResults),
+						renderDiagnostics: renderVisibilityDiagnostics(items, lineResults, (text) => textMetrics(text).plainText),
 						extensionStatuses: Array.from(footerData.getExtensionStatuses().entries()).map(([key, value]) => ({ key, value })),
 						model: ctx.model?.id ?? "no-model",
 						contextUsage: ctx.getContextUsage() ?? null,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@badliveware/pi-footer-framework",
-  "version": "0.2.2",
+  "version": "0.3.0",
   "description": "Configurable footer framework extension for Pi.",
   "type": "module",
   "types": "./index.ts",
@@ -43,9 +43,9 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-ai": "*",
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   },
   "engines": {
     "node": ">=20"