@agnishc/edb-global-footer 0.10.8 → 0.12.0

package/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## [0.12.0] - 2026-05-22
+
+## [0.10.9] - 2026-05-18
+
 ## [0.10.8] - 2026-05-18
 
 ## [0.10.6] - 2026-05-15

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@agnishc/edb-global-footer",
-	"version": "0.10.8",
+	"version": "0.12.0",
 	"description": "Pi extension: two-line status footer showing path, git branch, token usage, and model",
 	"keywords": [
 		"pi-package",

package/src/footer.ts

@@ -19,6 +19,28 @@ import type { GitStatus } from "./types";
 //   - Keys for extensions whose status you want to hide
 const STATUS_KEY_BLACKLIST = new Set<string>(["extmgr", "sm"]);
 
+// ── Agent mode helper ─────────────────────────────────────────────────────────
+
+const AGENT_MODE_ENTRY_TYPE = "agent-mode:active";
+
+function getActiveAgentMode(entries: any[]): string | null {
+	// Scan from most recent to oldest
+	for (let i = entries.length - 1; i >= 0; i--) {
+		const entry = entries[i];
+		if (entry.type === "custom" && entry.customType === AGENT_MODE_ENTRY_TYPE) {
+			const mode = entry.data?.mode;
+			if (mode === null) {
+				// Clear marker — stop scanning, return no mode
+				return null;
+			}
+			if (typeof mode === "string" && mode.length > 0) {
+				return mode;
+			}
+		}
+	}
+	return null;
+}
+
 // ── Live state types ───────────────────────────────────────────────────────────
 
 interface TpsState {
@@ -169,6 +191,12 @@ export function createFooterRenderer(
 							modelText += `${sep}${theme.fg("dim", thinkLabel)}`;
 						}
 					}
+
+					// Append active agent mode after thinking label
+					const agentMode = getActiveAgentMode(ctx.sessionManager.getEntries());
+					if (agentMode) {
+						modelText += `${sep}${theme.fg("muted", `◈ ${agentMode}`)}`;
+					}
 					rightPartsLine2.push(modelText);
 				}
 				const statsLine = renderFooterLine(width, leftStats, rightPartsLine2.join(sep));

package/src/git.ts

@@ -13,7 +13,7 @@ export function parseGitStatus(output: string): GitStatus {
 		if (!line) continue;
 		if (line.startsWith("# branch.head ")) {
 			const head = line.slice("# branch.head ".length).trim();
-			branch = head && head !== "(detached)" ? head : null;
+			branch = head && !head.startsWith("(detached") ? head : null;
 			continue;
 		}
 		if (line.startsWith("# branch.ab ")) {

package/src/index.test.ts

@@ -0,0 +1,134 @@
+import { describe, expect, it } from "vitest";
+import { formatThinkingLabel, formatTokens, sanitizeStatusText } from "./format.js";
+import { parseGitStatus } from "./git.js";
+import { withIcon } from "./icons.js";
+import { formatTps } from "./tps.js";
+
+describe("formatTokens", () => {
+	it("formats small numbers without suffix", () => {
+		expect(formatTokens(0)).toBe("0");
+		expect(formatTokens(999)).toBe("999");
+	});
+
+	it("formats K suffix for thousands", () => {
+		expect(formatTokens(1000)).toBe("1.0k");
+		expect(formatTokens(1500)).toBe("1.5k");
+		expect(formatTokens(9500)).toBe("9.5k");
+	});
+
+	it("formats M suffix for millions", () => {
+		expect(formatTokens(1_000_000)).toBe("1.0M");
+		expect(formatTokens(1_500_000)).toBe("1.5M");
+	});
+
+	it("formats M suffix for billions", () => {
+		expect(formatTokens(1_000_000_000)).toBe("1000M");
+	});
+});
+
+describe("formatThinkingLabel", () => {
+	it("returns MI for minimal", () => {
+		expect(formatThinkingLabel("minimal")).toBe("MI");
+	});
+
+	it("returns L for low", () => {
+		expect(formatThinkingLabel("low")).toBe("L");
+	});
+
+	it("returns M for medium", () => {
+		expect(formatThinkingLabel("medium")).toBe("M");
+	});
+
+	it("returns H for high", () => {
+		expect(formatThinkingLabel("high")).toBe("H");
+	});
+
+	it("returns XH for xhigh", () => {
+		expect(formatThinkingLabel("xhigh")).toBe("XH");
+	});
+
+	it("returns empty string for off", () => {
+		expect(formatThinkingLabel("off")).toBe("");
+	});
+});
+
+describe("sanitizeStatusText", () => {
+	it("replaces newlines with spaces", () => {
+		expect(sanitizeStatusText("hello\nworld")).toBe("hello world");
+	});
+
+	it("replaces carriage returns with spaces", () => {
+		expect(sanitizeStatusText("hello\rworld")).toBe("hello world");
+	});
+
+	it("replaces tabs with spaces", () => {
+		expect(sanitizeStatusText("hello\tworld")).toBe("hello world");
+	});
+
+	it("collapses multiple spaces", () => {
+		expect(sanitizeStatusText("hello    world")).toBe("hello world");
+	});
+
+	it("handles combined edge cases", () => {
+		expect(sanitizeStatusText("hello\r\n\tworld")).toBe("hello world");
+	});
+});
+
+describe("parseGitStatus", () => {
+	it("parses branch from porcelain=v2 format", () => {
+		const output = `# branch.oid abc1234
+# branch.head main
+# branch.upstream origin/main
+# branch.ab +1 -2
+ M file.txt`;
+		const result = parseGitStatus(output);
+		expect(result.branch).toBe("main");
+		expect(result.ahead).toBe(1);
+		expect(result.behind).toBe(2);
+		expect(result.dirty).toBe(true);
+	});
+
+	it("handles detached HEAD", () => {
+		const output = `# branch.oid abc1234
+# branch.head (detached at abc1234)
+`;
+		const result = parseGitStatus(output);
+		expect(result.branch).toBeNull();
+	});
+
+	it("marks dirty when non-comment lines present", () => {
+		const output = `# branch.oid abc1234
+# branch.head main
+ M modified.txt`;
+		const result = parseGitStatus(output);
+		expect(result.dirty).toBe(true);
+	});
+});
+
+describe("withIcon", () => {
+	it("combines icon and text", () => {
+		expect(withIcon("🚀", "Launch")).toBe("🚀 Launch");
+	});
+
+	it("returns text only when icon is empty", () => {
+		expect(withIcon("", "Launch")).toBe("Launch");
+	});
+});
+
+describe("formatTps", () => {
+	it("formats 0 tps as 0", () => {
+		expect(formatTps(0)).toBe("0");
+	});
+
+	it("formats single tps", () => {
+		expect(formatTps(1)).toBe("1.0");
+	});
+
+	it("formats multiple tps", () => {
+		expect(formatTps(10)).toBe("10");
+	});
+
+	it("rounds large tps", () => {
+		expect(formatTps(15.7)).toBe("16");
+	});
+});

package/src/workingIndicator.ts

@@ -455,6 +455,32 @@ function formatElapsed(ms: number): string {
 	return `${seconds}s`;
 }
 
+function stripAnsi(text: string): string {
+	return text.replace(/\x1b\[[0-9;]*m/g, "");
+}
+
+function isCompletionLine(line: string): boolean {
+	return /^✓ · .+ · \d/.test(stripAnsi(line).trim());
+}
+
+function stripCompletionLine(text: string): string {
+	return text
+		.split(/\r?\n/)
+		.filter((line) => !isCompletionLine(line))
+		.join("\n");
+}
+
+function appendCompletionToFinalAssistantMessage(message: any, completionLine: string): void {
+	if (!message || message.role !== "assistant" || !Array.isArray(message.content)) return;
+	const textBlocks = message.content.filter(
+		(block: any) => block?.type === "text" && typeof block.text === "string" && block.text.trim(),
+	);
+	const lastText = textBlocks[textBlocks.length - 1];
+	if (!lastText) return;
+	const text = stripCompletionLine(lastText.text);
+	lastText.text = `${text.trimEnd()}\n\n${completionLine}`;
+}
+
 // ── Shimmer ───────────────────────────────────────────────────────────────────
 
 const SHIMMER_BAND_WIDTH = 4;
@@ -629,26 +655,18 @@ export function installWorkingIndicator(pi: ExtensionAPI): WorkingIndicatorRef {
 
 	// ── Completion ────────────────────────────────────────────────────────────
 
-	function showCompletion(totalMs: number): void {
-		if (!ctx) return;
-		cancelCompletionTimer();
-
-		const theme = ctx.ui.theme;
-		const check = theme.fg("success", "✓");
-		const verb = theme.fg("accent", randomItem(COMPLETION_VERBS));
-		const time = theme.fg("dim", formatElapsed(totalMs));
-		const sep = theme.fg("dim", " · ");
-		const completionLine = `${check}${sep}${verb}${sep}${time}`;
-
-		// The Loader is torn down by pi on agent_end, so setWorkingMessage is a no-op.
-		// Use setWidget to briefly show the completion state above the editor.
-		ctx.ui.setWidget("wi-completion", [completionLine]);
+	function completionLine(totalMs: number): string {
+		// This string is persisted into assistant message content, so it must stay plain text.
+		// ANSI escapes inside persisted markdown can render as literal control-code fragments.
+		return `✓ · ${randomItem(COMPLETION_VERBS)} · ${formatElapsed(totalMs)}`;
+	}
 
-		completionTimer = setTimeout(() => {
-			completionTimer = null;
-			ref.currentLine = undefined;
-			ctx?.ui.setWidget("wi-completion", undefined);
-		}, 3000);
+	function showCompletion(_totalMs: number): void {
+		// Completion is now persisted into the final assistant message on message_end.
+		// Do not use the old temporary wi-completion widget.
+		cancelCompletionTimer();
+		ref.currentLine = undefined;
+		ctx?.ui.setWidget("wi-completion", undefined);
 	}
 
 	// ── Spinner frames (apply to pi's working indicator widget) ───────────────
@@ -687,6 +705,15 @@ export function installWorkingIndicator(pi: ExtensionAPI): WorkingIndicatorRef {
 		startTimers();
 	});
 
+	pi.on("message_end", async (event: any, messageCtx) => {
+		ctx = messageCtx;
+		const message = event?.message;
+		if (message?.role !== "assistant") return;
+		if (message.stopReason === "toolUse") return;
+		const totalMs = startedAt > 0 ? Date.now() - startedAt : 0;
+		appendCompletionToFinalAssistantMessage(message, completionLine(totalMs));
+	});
+
 	pi.on("tool_execution_start", (event) => {
 		if (!isActive) return;
 		const e = event as { toolCallId?: string; toolName?: string };