ultimate-pi 0.1.7 → 0.2.2

package/.pi/extensions/custom-footer.ts

@@ -5,9 +5,70 @@
  * ~/project (main)  ↑12k ↓8k  $0.042  [████████░░] 72%  anthropic • claude-sonnet-4-20250514 • xhigh
  */
 
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type {
+	ExtensionAPI,
+	ExtensionContext,
+	ThemeColor,
+} from "@mariozechner/pi-coding-agent";
 import type { TUI } from "@mariozechner/pi-tui";
-import { Box, Text, truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+import {
+	Box,
+	Text,
+	truncateToWidth,
+	visibleWidth,
+} from "@mariozechner/pi-tui";
+
+// ── router decision reader ──────────────────────────────────────────
+
+/** Extract last routing decision from model-router session entries. */
+function readRouterDecision(
+	ctx: ExtensionContext,
+): {
+	targetProvider: string;
+	targetModelId: string;
+	thinking: string;
+	profile: string;
+	enabled: boolean;
+} | null {
+	try {
+		const entries = ctx.sessionManager.getEntries();
+		for (let i = entries.length - 1; i >= 0; i--) {
+			const e = entries[i];
+			if (
+				e.type === "custom" &&
+				(e as any).customType === "router-state" &&
+				(e as any).data
+			) {
+				const data = (e as any).data;
+				const d = data.lastDecision;
+				if (d?.targetProvider && d?.targetModelId) {
+					return {
+						targetProvider: d.targetProvider,
+						targetModelId: d.targetModelId,
+						thinking: d.thinking,
+						profile: data.selectedProfile ?? d.profile,
+						enabled: data.enabled ?? true,
+					};
+				}
+			}
+		}
+	} catch {
+		// session not ready
+	}
+	return null;
+}
+
+// ── profile colors ────────────────────────────────────────────────
+
+const PROFILE_COLORS: Record<string, ThemeColor> = {
+	auto: "accent",
+	cheap: "success",
+	deep: "toolTitle",
+};
+
+function profileColor(profile: string): ThemeColor {
+	return PROFILE_COLORS[profile] ?? "muted";
+}
 
 // ── token formatting ──────────────────────────────────────────────
 
@@ -85,13 +146,31 @@ interface ModelInfo {
 function modelInfo(ctx: ExtensionContext): ModelInfo {
 	const m = ctx.model;
 	if (!m) return { id: "no-model", provider: "unknown", reasoning: false };
+
+	// Router provider — extract actual model from routing decision
+	if (m.provider === "router") {
+		const decision = readRouterDecision(ctx);
+		if (decision) {
+			return {
+				id: decision.targetModelId,
+				provider: decision.targetProvider,
+				reasoning: true,
+			};
+		}
+		// No decision yet — show profile name so user knows router is active
+		return { id: m.id, provider: "router", reasoning: m.reasoning };
+	}
+
 	return { id: m.id, provider: m.provider, reasoning: m.reasoning };
 }
 
 // ── thinking label ────────────────────────────────────────────────
 
 /** Returns thinking level display string, or null if model doesn't support reasoning. */
-function thinkingLabel(level: string | null, reasoning: boolean): string | null {
+function thinkingLabel(
+	level: string | null,
+	reasoning: boolean,
+): string | null {
 	if (!reasoning || level == null) return null;
 	if (level === "minimal") return "off";
 	return level;
@@ -99,7 +178,10 @@ function thinkingLabel(level: string | null, reasoning: boolean): string | null
 
 // ── cwd + branch ──────────────────────────────────────────────────
 
-function pathLabel(cwd: string, branch: string | null): { cwd: string; branch: string | null } {
+function pathLabel(
+	cwd: string,
+	branch: string | null,
+): { cwd: string; branch: string | null } {
 	let pwd = cwd;
 	const home = process.env.HOME || process.env.USERPROFILE;
 	if (home && pwd.startsWith(home)) {
@@ -124,6 +206,8 @@ export default function customFooter(pi: ExtensionAPI) {
 		modelProvider: string;
 		modelReasoning: boolean;
 		thinkingLevel: string | null;
+		routerProfile: string | null;
+		routerActive: boolean;
 		branch: string | null;
 	} = {
 		tui: null,
@@ -136,6 +220,8 @@ export default function customFooter(pi: ExtensionAPI) {
 		modelProvider: "…",
 		modelReasoning: false,
 		thinkingLevel: null,
+		routerProfile: null,
+		routerActive: false,
 		branch: null,
 	};
 
@@ -150,7 +236,18 @@ export default function customFooter(pi: ExtensionAPI) {
 		state.modelId = mi.id;
 		state.modelProvider = mi.provider;
 		state.modelReasoning = mi.reasoning;
-		state.thinkingLevel = pi.getThinkingLevel();
+
+		// When router active, use thinking from the routing decision
+		if (ctx.model?.provider === "router") {
+			const routerDecision = readRouterDecision(ctx);
+			state.thinkingLevel = routerDecision?.thinking ?? pi.getThinkingLevel();
+			state.routerProfile = routerDecision?.profile ?? ctx.model?.id;
+			state.routerActive = routerDecision?.enabled ?? true;
+		} else {
+			state.thinkingLevel = pi.getThinkingLevel();
+			state.routerProfile = null;
+			state.routerActive = false;
+		}
 	}
 
 	pi.on("session_start", (_event, ctx) => {
@@ -159,7 +256,10 @@ export default function customFooter(pi: ExtensionAPI) {
 		state.cost = computeCost(ctx);
 
 		ctx.ui.setFooter((tui, theme, footerData) => {
-			if (unsubBranch) { unsubBranch(); unsubBranch = null; }
+			if (unsubBranch) {
+				unsubBranch();
+				unsubBranch = null;
+			}
 
 			state.tui = tui;
 			state.branch = footerData.getGitBranch();
@@ -178,7 +278,10 @@ export default function customFooter(pi: ExtensionAPI) {
 
 			return {
 				dispose() {
-					if (unsubBranch) { unsubBranch(); unsubBranch = null; }
+					if (unsubBranch) {
+						unsubBranch();
+						unsubBranch = null;
+					}
 					state.tui = null;
 					state.box = null;
 					state.textLine = null;
@@ -200,7 +303,10 @@ export default function customFooter(pi: ExtensionAPI) {
 					const innerW = Math.max(1, width - 2 - SAFETY_MARGIN);
 
 					// ── left: cwd/branch  tokens  cost ──
-					const { cwd: cwdPath, branch: gitBranch } = pathLabel(state.cwd, state.branch);
+					const { cwd: cwdPath, branch: gitBranch } = pathLabel(
+						state.cwd,
+						state.branch,
+					);
 					const pathDisplay = gitBranch
 						? `${cwdPath} (${theme.fg("error", gitBranch)})`
 						: cwdPath;
@@ -223,26 +329,58 @@ export default function customFooter(pi: ExtensionAPI) {
 					// ── right: provider • modelId thinkingLevel ──
 					const tl = thinkingLabel(state.thinkingLevel, state.modelReasoning);
 					const modelDisplay = tl ? `${state.modelId} ${tl}` : state.modelId;
-					const rightParts: string[] = [state.modelProvider, modelDisplay];
-					const rightStr = rightParts.join(" • ");
+
+					// Colorize: router profile gets mode color, model gets thinking color
+					const rightStr = state.routerActive
+						? (() => {
+								const pColor = profileColor(
+									state.routerProfile ?? "router",
+								);
+								const profileStr = theme.fg(pColor, state.routerProfile ?? "router");
+								const thinkingColorFn = theme.getThinkingBorderColor(
+									(state.thinkingLevel as any) ?? "medium",
+								);
+								const modelStr = thinkingColorFn(modelDisplay);
+								return `${profileStr} • ${modelStr}`;
+							})()
+						: `${state.modelProvider} • ${modelDisplay}`;
 
 					// ── compose single line ──
 					const colLeft = leftStr;
-					const dimRight = dim(rightStr);
+					const finalRight = state.routerActive ? rightStr : dim(rightStr);
 					const lw = visibleWidth(colLeft);
-					const rw = visibleWidth(dimRight);
+					const rw = visibleWidth(finalRight);
 					const bw = visibleWidth(barFull);
 					const gap = 2;
 
 					if (lw + gap + bw + gap + rw <= innerW) {
 						const pad = innerW - lw - gap - bw - gap - rw;
-						const line = colLeft + " ".repeat(gap + pad) + barFull + " ".repeat(gap) + dimRight;
+						const line =
+							colLeft +
+							" ".repeat(gap + pad) +
+							barFull +
+							" ".repeat(gap) +
+							finalRight;
 						textLine.setText(truncateToWidth(line, innerW));
 					} else {
 						// Priority: keep bar visible, keep left (cwd) intact, truncate modelId first
-						const tlNow = thinkingLabel(state.thinkingLevel, state.modelReasoning);
+						const tlNow = thinkingLabel(
+							state.thinkingLevel,
+							state.modelReasoning,
+						);
+						const thinkingColorFn = theme.getThinkingBorderColor(
+							(state.thinkingLevel as any) ?? "medium",
+						);
 						const buildRight = (mid: string) => {
 							const modelPart = tlNow ? `${mid} ${tlNow}` : mid;
+							if (state.routerActive) {
+								const pColor = profileColor(
+									state.routerProfile ?? "router",
+								);
+								const profileStr = theme.fg(pColor, state.routerProfile ?? "router");
+								const modelStr = thinkingColorFn(modelPart);
+								return `${profileStr} • ${modelStr}`;
+							}
 							const parts: string[] = [state.modelProvider];
 							if (modelPart) parts.push(modelPart);
 							return dim(parts.join(" • "));
@@ -252,7 +390,10 @@ export default function customFooter(pi: ExtensionAPI) {
 						let dimR = buildRight(truncMid);
 						let rwNow = visibleWidth(dimR);
 
-						while (truncMid.length > 0 && lw + gap + bw + gap + rwNow > innerW) {
+						while (
+							truncMid.length > 0 &&
+							lw + gap + bw + gap + rwNow > innerW
+						) {
 							truncMid = truncMid.slice(0, -1);
 							dimR = buildRight(truncMid);
 							rwNow = visibleWidth(dimR);
@@ -260,7 +401,12 @@ export default function customFooter(pi: ExtensionAPI) {
 
 						if (lw + gap + bw + gap + rwNow <= innerW) {
 							const pad = innerW - lw - gap - bw - gap - rwNow;
-							const line = colLeft + " ".repeat(gap + pad) + barFull + " ".repeat(gap) + dimR;
+							const line =
+								colLeft +
+								" ".repeat(gap + pad) +
+								barFull +
+								" ".repeat(gap) +
+								dimR;
 							textLine.setText(truncateToWidth(line, innerW));
 						} else {
 							// ModelId gone, still overflows: truncate leftStr
@@ -269,7 +415,12 @@ export default function customFooter(pi: ExtensionAPI) {
 								textLine.setText(truncateToWidth(dimR, innerW));
 							} else {
 								const truncLeft = truncateToWidth(colLeft, Math.max(1, avail));
-								const line = truncLeft + " ".repeat(gap) + barFull + " ".repeat(gap) + dimR;
+								const line =
+									truncLeft +
+									" ".repeat(gap) +
+									barFull +
+									" ".repeat(gap) +
+									dimR;
 								textLine.setText(truncateToWidth(line, innerW));
 							}
 						}
@@ -291,11 +442,32 @@ export default function customFooter(pi: ExtensionAPI) {
 	});
 
 	// Track model changes
-	pi.on("model_select", (event) => {
-		state.modelId = event.model?.id ?? "…";
-		state.modelProvider = event.model?.provider ?? "…";
-		state.modelReasoning = event.model?.reasoning ?? false;
-		state.thinkingLevel = pi.getThinkingLevel();
+	pi.on("model_select", (event, ctx) => {
+		if (event.model?.provider === "router") {
+			const routerDecision = readRouterDecision(ctx);
+			if (routerDecision) {
+				state.modelId = routerDecision.targetModelId;
+				state.modelProvider = routerDecision.targetProvider;
+				state.modelReasoning = true;
+				state.thinkingLevel = routerDecision.thinking;
+				state.routerProfile = routerDecision.profile;
+				state.routerActive = routerDecision.enabled;
+			} else {
+				state.modelId = event.model.id ?? "…";
+				state.modelProvider = "router";
+				state.modelReasoning = event.model?.reasoning ?? false;
+				state.thinkingLevel = pi.getThinkingLevel();
+				state.routerProfile = event.model.id;
+				state.routerActive = true;
+			}
+		} else {
+			state.modelId = event.model?.id ?? "…";
+			state.modelProvider = event.model?.provider ?? "…";
+			state.modelReasoning = event.model?.reasoning ?? false;
+			state.thinkingLevel = pi.getThinkingLevel();
+			state.routerProfile = null;
+			state.routerActive = false;
+		}
 		invalidate();
 	});
 

package/.pi/extensions/custom-header.ts

@@ -6,30 +6,67 @@
  * doubling vertical resolution in the same terminal footprint.
  */
 
+import { join } from "node:path";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { truncateToWidth } from "@mariozechner/pi-tui";
-import Jimp from "jimp";
-import { join } from "node:path";
+import * as JimpModule from "jimp";
 
 const imagePath = join(process.cwd(), ".pi", "extensions", "banner.png");
 
 // Terminal footprint — keep a safety margin so we never crash on narrow terminals
 const SAFETY_MARGIN = 2;
-const TERM_WIDTH = 120;
-const TERM_HEIGHT = 24;
+const TERM_WIDTH = 100;
+const TERM_HEIGHT = 15;
 
 // Pixel grid: 1 char wide × 2 pixels tall (half-block)
 const PIXEL_WIDTH = TERM_WIDTH;
-const PIXEL_HEIGHT = TERM_HEIGHT * 2;
+const PIXEL_HEIGHT = TERM_HEIGHT;
 
 const HALF_BLOCK = "\u2584"; // lower half block: fg = bottom, bg = top
 
 let cachedBanner: string[] | null = null;
 let loadPromise: Promise<string[]> | null = null;
 
+function getJimpRuntime(): { read(path: string): Promise<any> } {
+	const mod = JimpModule as Record<string, unknown>;
+	const candidates = [mod.default, mod.Jimp, mod];
+	for (const candidate of candidates) {
+		if (
+			candidate &&
+			(typeof candidate === "object" || typeof candidate === "function") &&
+			typeof (candidate as { read?: unknown }).read === "function"
+		) {
+			return candidate as { read(path: string): Promise<any> };
+		}
+	}
+	throw new TypeError("Jimp runtime missing read() export");
+}
+
+function intToRgba(pixel: number): {
+	r: number;
+	g: number;
+	b: number;
+	a: number;
+} {
+	return {
+		r: (pixel >> 24) & 255,
+		g: (pixel >> 16) & 255,
+		b: (pixel >> 8) & 255,
+		a: pixel & 255,
+	};
+}
+
+function resizeImageCompat(image: any, w: number, h: number): void {
+	try {
+		image.resize({ w, h });
+	} catch {
+		image.resize(w, h);
+	}
+}
+
 function ansiCell(
 	top: { r: number; g: number; b: number; a: number },
-	bottom: { r: number; g: number; b: number; a: number }
+	bottom: { r: number; g: number; b: number; a: number },
 ): string {
 	const topTransparent = top.a < 128;
 	const bottomTransparent = bottom.a < 128;
@@ -46,15 +83,16 @@ function ansiCell(
 }
 
 async function loadBanner(): Promise<string[]> {
+	const Jimp = getJimpRuntime();
 	const image = await Jimp.read(imagePath);
-	image.resize(PIXEL_WIDTH, PIXEL_HEIGHT);
+	resizeImageCompat(image, PIXEL_WIDTH, PIXEL_HEIGHT);
 
 	const lines: string[] = [];
 	for (let row = 0; row < PIXEL_HEIGHT; row += 2) {
 		let line = "";
 		for (let col = 0; col < PIXEL_WIDTH; col++) {
-			const top = Jimp.intToRGBA(image.getPixelColor(col, row));
-			const bottom = Jimp.intToRGBA(image.getPixelColor(col, row + 1));
+			const top = intToRgba(image.getPixelColor(col, row));
+			const bottom = intToRgba(image.getPixelColor(col, row + 1));
 			line += ansiCell(top, bottom);
 		}
 		lines.push(line);