@sassywares/pi-extensions 0.3.21 → 0.3.22

package/README.md

@@ -151,7 +151,7 @@ Writes do not fail quietly and they are not matched by pattern. `rm`, `sed -i`,
 
 A backend is only trusted after it proves both halves of the contract: reads inside the working directory must succeed **and** a probe write there must fail. A backend that is installed but cannot enforce (for example bubblewrap on a distribution that blocks unprivileged user namespaces) is discarded, and ask mode continues without bash rather than falling back to a command allowlist. The working directory always wins over the temp allowance: if the project itself lives under the temp directory, temp writes are denied too and `/modes` reports `temp dir is read-only`.
 
-`/modes` shows the verdict, and the footer status line shows `⏸ ask · ro(sandbox-exec)`, `⏸ ask · bash off`, or `▶ work`.
+`/modes` shows the verdict. The mode badge sits on a footer line of its own, aligned to the right edge: `▶ work`, `⏸ ask · ro(sandbox-exec)`, or `⏸ ask · bash off`. To place it there without reimplementing pi's footer, the extension delegates to pi's own `FooterComponent` and appends the badge line, so directory, session name, token stats, model, and context usage stay exactly as pi renders them. If a future pi release changes what that component reads, the extension restores the native footer and falls back to a left-aligned status line.
 
 - `--ask` starts a session in ask mode; the mode persists across resume.
 - `shift+tab` is not used, because pi binds it to the thinking level. To match an editor that uses it for modes, bind `app.thinking.cycle` to something else in `~/.pi/agent/keybindings.json` and add a shortcut there.

package/extensions/modes/footer.ts

@@ -0,0 +1,88 @@
+/**
+ * Right-aligned mode badge.
+ *
+ * `ctx.ui.setStatus()` has no placement: pi appends status texts as a line of
+ * their own, left-aligned, and its sanitizer collapses runs of spaces, so a
+ * badge cannot be pushed to the right edge from there.
+ *
+ * This footer keeps pi's own rendering (it delegates to the exported
+ * `FooterComponent`) and appends one extra line with the badge aligned to the
+ * right edge. Nothing about the directory, session name, token stats, or model
+ * line is reimplemented here, so it stays correct as pi's footer evolves.
+ *
+ * If a future pi version changes what `FooterComponent` reads from its session
+ * argument, the render below throws; `onError` then tells the extension to
+ * restore the native footer and fall back to the plain status line.
+ */
+
+import type { Component } from "@earendil-works/pi-tui";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import type { ExtensionContext, ReadonlyFooterDataProvider, Theme } from "@earendil-works/pi-coding-agent";
+import { FooterComponent } from "@earendil-works/pi-coding-agent";
+
+export interface ModeBadge {
+	text: string;
+	tone: "warning" | "error" | "accent";
+}
+
+export interface ModesFooterOptions {
+	/** Session auto-compaction, shown by the footer as the `(auto)` suffix. */
+	autoCompact: boolean;
+	/** Called once if pi's footer cannot be rendered through this shim. */
+	onError: () => void;
+}
+
+/**
+ * The subset of `AgentSession` that `FooterComponent` reads. Everything except
+ * a few computed values is already on the extension context.
+ */
+function sessionShim(ctx: ExtensionContext) {
+	return {
+		model: ctx.model,
+		state: { model: ctx.model, thinkingLevel: ctx.thinkingLevel },
+		routedModel: undefined,
+		getContextUsage: () => ctx.getContextUsage(),
+		modelRuntime: {
+			// Extensions cannot observe subscription backing; the provider name is
+			// the only signal available.
+			isUsingSubscription: (provider: string) => provider === "kimi-coding",
+		},
+		sessionManager: ctx.sessionManager,
+	} as unknown as ConstructorParameters<typeof FooterComponent>[0];
+}
+
+export function createModesFooter(
+	ctx: ExtensionContext,
+	theme: Theme,
+	footerData: ReadonlyFooterDataProvider,
+	badge: () => ModeBadge,
+	options: ModesFooterOptions,
+): Component & { dispose?(): void } {
+	const inner = new FooterComponent(sessionShim(ctx), footerData);
+	inner.setAutoCompactEnabled(options.autoCompact);
+	let failed = false;
+
+	function badgeLine(width: number): string {
+		const current = badge();
+		const text = current.text;
+		if (visibleWidth(text) >= width) return theme.fg(current.tone, truncateToWidth(text, width, "…"));
+		const padding = Math.max(1, width - visibleWidth(text));
+		return " ".repeat(padding) + theme.fg(current.tone, text);
+	}
+
+	return {
+		invalidate() {
+			inner.invalidate();
+		},
+		render(width: number): string[] {
+			if (failed) return [badgeLine(width)];
+			try {
+				return [...inner.render(width), badgeLine(width)];
+			} catch {
+				failed = true;
+				options.onError();
+				return [badgeLine(width)];
+			}
+		},
+	};
+}

package/extensions/modes/index.ts

@@ -17,6 +17,7 @@
 import type { AgentMessage } from "@earendil-works/pi-agent-core";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { createWrappedOperations, detectBackend, type SandboxBackend } from "./backends.ts";
+import { createModesFooter, type ModeBadge } from "./footer.ts";
 
 type Mode = "work" | "ask";
 
@@ -51,6 +52,8 @@ export default function modesExtension(pi: ExtensionAPI): void {
 	let backendReason = "not detected yet";
 	let backendChecked = false;
 	let workTools: string[] | undefined;
+	let footerStatus: "none" | "custom" | "fallback" = "none";
+	let footerTui: { requestRender(): void } | undefined;
 	const wrappedCalls = new Set<string>();
 
 	function configuredAskTools(): string[] {
@@ -71,13 +74,55 @@ export default function modesExtension(pi: ExtensionAPI): void {
 		pi.appendEntry(STATE_CUSTOM_TYPE, { mode } satisfies PersistedState);
 	}
 
+	function badgeSpec(): ModeBadge {
+		if (mode === "work") return { text: "▶ work", tone: "accent" };
+		if (backend) return { text: `⏸ ask · ro(${backend.id})`, tone: "warning" };
+		return { text: "⏸ ask · bash off", tone: "error" };
+	}
+
 	function updateStatus(ctx: ExtensionContext): void {
-		if (mode === "work") {
-			ctx.ui.setStatus("modes", ctx.ui.theme.fg("accent", "▶ work"));
+		// The custom footer draws the badge itself, right-aligned.
+		if (footerStatus === "custom") {
+			ctx.ui.setStatus("modes", undefined);
+			footerTui?.requestRender();
 			return;
 		}
-		const text = backend ? `⏸ ask · ro(${backend.id})` : "⏸ ask · bash off";
-		ctx.ui.setStatus("modes", ctx.ui.theme.fg(backend ? "warning" : "error", text));
+		const badge = badgeSpec();
+		ctx.ui.setStatus("modes", ctx.ui.theme.fg(badge.tone, badge.text));
+	}
+
+	/**
+	 * Install a footer that keeps pi's rendering and appends the badge aligned to
+	 * the right edge. Falls back to the native footer plus a status line if pi's
+	 * footer cannot be driven from here.
+	 */
+	function installFooter(ctx: ExtensionContext): void {
+		if (ctx.mode !== "tui" || footerStatus === "custom") return;
+		const autoCompact = pi.getSettings().compaction?.enabled !== false;
+
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			footerTui = tui;
+			const unsubscribe = footerData.onBranchChange(() => tui.requestRender());
+			return {
+				...createModesFooter(ctx, theme, footerData, badgeSpec, {
+					autoCompact,
+					onError: () => {
+						footerStatus = "fallback";
+						footerTui = undefined;
+						queueMicrotask(() => {
+							ctx.ui.setFooter(undefined);
+							updateStatus(ctx);
+						});
+					},
+				}),
+				dispose() {
+					unsubscribe();
+					footerTui = undefined;
+				},
+			};
+		});
+		footerStatus = "custom";
+		ctx.ui.setStatus("modes", undefined);
 	}
 
 	function describe(): string {
@@ -240,6 +285,7 @@ export default function modesExtension(pi: ExtensionAPI): void {
 			workTools = pi.getActiveTools();
 			pi.setActiveTools(askToolset());
 		}
+		installFooter(ctx);
 		updateStatus(ctx);
 	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sassywares/pi-extensions",
-  "version": "0.3.21",
+  "version": "0.3.22",
   "description": "A zesty power pack for pi: auto-titled sessions, real web access, clean handoffs, isolated subagents, and ask/work modes.",
   "type": "module",
   "license": "MIT",
@@ -36,6 +36,7 @@
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*",
     "typebox": "*"
   },
   "pi": {