@matyah00/openpi 0.1.2

package/extensions/themeMap.ts

@@ -0,0 +1,145 @@
+/**
+ * themeMap.ts — Per-extension default theme assignments
+ *
+ * Themes live in .pi/themes/ and are mapped by extension filename (no extension).
+ * Each extension calls applyExtensionTheme(import.meta.url, ctx) in its session_start
+ * hook to automatically load its designated theme on boot.
+ *
+ * Available themes (.pi/themes/):
+ *   catppuccin-mocha · cyberpunk · dracula · everforest · gruvbox
+ *   midnight-ocean   · nord      · ocean-breeze · rose-pine
+ *   synthwave        · tokyo-night
+ */
+
+import type { ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { basename } from "path";
+import { fileURLToPath } from "url";
+
+// ── Theme assignments ──────────────────────────────────────────────────────
+//
+// Key   = extension filename without extension (matches extensions/<key>.ts)
+// Value = theme name from .pi/themes/<value>.json
+//
+export const THEME_MAP: Record<string, string> = {
+	"agent-chain":        "midnight-ocean",   // deep sequential pipeline
+	"agent-team":         "dracula",          // rich orchestration palette
+	"coms":               "ocean-breeze",     // peer-to-peer messaging, cross-boundary
+	"coms-net":           "ocean-breeze",     // peer-to-peer messaging, cross-boundary
+	"cross-agent":        "ocean-breeze",     // cross-boundary, connecting
+	"damage-control":     "gruvbox",          // grounded, earthy safety
+	"minimal":            "synthwave",        // synthwave by default now!
+	"pi-pi":              "rose-pine",        // warm creative meta-agent
+	"pure-focus":         "everforest",       // calm, distraction-free
+	"purpose-gate":       "tokyo-night",      // intentional, sharp focus
+	"session-replay":     "catppuccin-mocha", // soft, reflective history
+	"subagent-widget":    "cyberpunk",        // multi-agent futuristic
+	"system-select":      "catppuccin-mocha", // soft selection UI
+	"theme-cycler":       "synthwave",        // neon, it's a theme tool
+	"tilldone":           "everforest",       // task-focused calm
+	"tool-counter":       "synthwave",        // techy metrics
+	"tool-counter-widget":"synthwave",        // same family
+};
+
+// ── Helpers ───────────────────────────────────────────────────────────────
+
+/** Derive the extension name (e.g. "minimal") from its import.meta.url. */
+function extensionName(fileUrl: string): string {
+	const filePath = fileUrl.startsWith("file://") ? fileURLToPath(fileUrl) : fileUrl;
+	return basename(filePath).replace(/\.[^.]+$/, "");
+}
+
+// ── Theme ──────────────────────────────────────────────────────────────────
+
+/**
+ * Apply the mapped theme for an extension on session boot.
+ *
+ * @param fileUrl   Pass `import.meta.url` from the calling extension file.
+ * @param ctx       The ExtensionContext from the session_start handler.
+ * @returns         true if the theme was applied successfully, false otherwise.
+ */
+export function applyExtensionTheme(fileUrl: string, ctx: ExtensionContext): boolean {
+	if (!ctx.hasUI) return false;
+
+	const name = extensionName(fileUrl);
+	
+	// If there are multiple extensions stacked in 'ipi', they each fire session_start
+	// and try to apply their own mapped theme. The LAST one to fire wins.
+	// Since system-select is last in the ipi alias array, it was setting 'catppuccin-mocha'.
+	
+	// We want to skip theme application for all secondary extensions if they are stacked,
+	// so the primary extension (first in the array) dictates the theme.
+	const primaryExt = primaryExtensionName();
+	if (primaryExt && primaryExt !== name) {
+		return true; // Pretend we succeeded, but don't overwrite the primary theme
+	}
+
+	let themeName = THEME_MAP[name];
+	
+	if (!themeName) {
+		themeName = "synthwave";
+	}
+
+	const result = ctx.ui.setTheme(themeName);
+	
+	if (!result.success && themeName !== "synthwave") {
+		return ctx.ui.setTheme("synthwave").success;
+	}
+	
+	return result.success;
+}
+// ── Title ──────────────────────────────────────────────────────────────────
+
+/**
+ * Read process.argv to find the first -e / --extension flag value.
+ *
+ * When Pi is launched as:
+ *   pi -e extensions/subagent-widget.ts -e extensions/pure-focus.ts
+ *
+ * process.argv contains those paths verbatim. Every stacked extension calls
+ * this and gets the same answer ("subagent-widget"), so all setTitle calls
+ * are idempotent — no shared state or deduplication needed.
+ *
+ * Returns null if no -e flag is present (e.g. plain `pi` with no extensions).
+ */
+function primaryExtensionName(): string | null {
+	const argv = process.argv;
+	for (let i = 0; i < argv.length - 1; i++) {
+		if (argv[i] === "-e" || argv[i] === "--extension") {
+			return basename(argv[i + 1]).replace(/\.[^.]+$/, "");
+		}
+	}
+	return null;
+}
+
+/**
+ * Set the terminal title to "π - <first-extension-name>" on session boot.
+ * Reads the title from process.argv so all stacked extensions agree on the
+ * same value — no coordination or shared state required.
+ *
+ * Deferred 150 ms to fire after Pi's own startup title-set.
+ */
+function applyExtensionTitle(ctx: ExtensionContext): void {
+	if (!ctx.hasUI) return;
+	const name = primaryExtensionName();
+	if (!name) return;
+	setTimeout(() => ctx.ui.setTitle(`π - ${name}`), 150);
+}
+
+// ── Combined default ───────────────────────────────────────────────────────
+
+/**
+ * Apply both the mapped theme AND the terminal title for an extension.
+ * Drop-in replacement for applyExtensionTheme — call this in every session_start.
+ *
+ * Usage:
+ *   import { applyExtensionDefaults } from "./themeMap.ts";
+ *
+ *   pi.on("session_start", async (_event, ctx) => {
+ *     applyExtensionDefaults(import.meta.url, ctx);
+ *     // ... rest of handler
+ *   });
+ */
+export function applyExtensionDefaults(fileUrl: string, ctx: ExtensionContext): void {
+	applyExtensionTheme(fileUrl, ctx);
+	applyExtensionTitle(ctx);
+}

package/extensions/tool-counter-widget.ts

@@ -0,0 +1,68 @@
+/**
+ * Tool Counter Widget — Tool call counts in a widget above the editor
+ *
+ * Shows a persistent, live-updating widget with per-tool background colors.
+ * Format: Tools (N): [Bash 3] [Read 7] [Write 2]
+ *
+ * Usage: pi -e extensions/tool-counter-widget.ts
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { Box, Text } from "@mariozechner/pi-tui";
+import { applyExtensionDefaults } from "./themeMap.ts";
+
+const palette = [
+	[12, 40, 80],    // deep navy
+	[50, 20, 70],    // dark purple
+	[10, 55, 45],    // dark teal
+	[70, 30, 10],    // dark rust
+	[55, 15, 40],    // dark plum
+	[15, 50, 65],    // dark ocean
+	[45, 45, 15],    // dark olive
+	[65, 18, 25],    // dark wine
+];
+
+function bg(rgb: number[], s: string): string {
+	return `\x1b[48;2;${rgb[0]};${rgb[1]};${rgb[2]}m${s}\x1b[49m`;
+}
+
+export default function (pi: ExtensionAPI) {
+	const counts: Record<string, number> = {};
+	const toolColors: Record<string, number[]> = {};
+	let total = 0;
+	let colorIdx = 0;
+
+	pi.on("tool_execution_end", async (event) => {
+		if (!(event.toolName in toolColors)) {
+			toolColors[event.toolName] = palette[colorIdx % palette.length];
+			colorIdx++;
+		}
+		counts[event.toolName] = (counts[event.toolName] || 0) + 1;
+		total++;
+	});
+
+	pi.on("session_start", async (_event, ctx) => {
+		applyExtensionDefaults(import.meta.url, ctx);
+		ctx.ui.setWidget("tool-counter", (_tui, theme) => {
+			const text = new Text("", 1, 1);
+
+			return {
+				render(width: number): string[] {
+					const entries = Object.entries(counts);
+					const parts = entries.map(([name, count]) => {
+						const rgb = toolColors[name];
+						return bg(rgb, `\x1b[38;2;220;220;220m  ${name} ${count}  \x1b[39m`);
+					});
+					text.setText(
+						theme.fg("accent", `Tools (${total}):`) +
+						(entries.length > 0 ? " " + parts.join(" ") : "")
+					);
+					return text.render(width);
+				},
+				invalidate() {
+					text.invalidate();
+				},
+			};
+		});
+	});
+}

package/extensions/tool-counter.ts

@@ -0,0 +1,102 @@
+/**
+ * Tool Counter — Rich two-line custom footer
+ *
+ * Line 1: model + context meter on left, tokens in/out + cost on right
+ * Line 2: cwd (branch) on left, tool call tally on right
+ *
+ * Demonstrates: setFooter, footerData.getGitBranch(), onBranchChange(),
+ * session branch traversal for token/cost accumulation.
+ *
+ * Usage: pi -e extensions/tool-counter.ts
+ */
+
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+import { basename } from "node:path";
+import { applyExtensionDefaults } from "./themeMap.ts";
+
+export default function (pi: ExtensionAPI) {
+	const counts: Record<string, number> = {};
+
+	pi.on("tool_execution_end", async (event) => {
+		counts[event.toolName] = (counts[event.toolName] || 0) + 1;
+	});
+
+	pi.on("session_start", async (_event, ctx) => {
+		applyExtensionDefaults(import.meta.url, ctx);
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			const unsub = footerData.onBranchChange(() => tui.requestRender());
+
+			return {
+				dispose: unsub,
+				invalidate() {},
+				render(width: number): string[] {
+					// --- Line 1: cwd + branch (left), tokens + cost (right) ---
+					let tokIn = 0;
+					let tokOut = 0;
+					let cost = 0;
+					for (const entry of ctx.sessionManager.getBranch()) {
+						if (entry.type === "message" && entry.message.role === "assistant") {
+							const m = entry.message as AssistantMessage;
+							tokIn += m.usage.input;
+							tokOut += m.usage.output;
+							cost += m.usage.cost.total;
+						}
+					}
+
+					const fmt = (n: number) => n < 1000 ? `${n}` : `${(n / 1000).toFixed(1)}k`;
+					const dir = basename(ctx.cwd);
+					const branch = footerData.getGitBranch();
+
+					// --- Line 1: model + context meter (left), tokens + cost (right) ---
+					const usage = ctx.getContextUsage();
+					const pct = usage ? usage.percent : 0;
+					const filled = Math.round(pct / 10) || 1;
+					const bar = "#".repeat(filled) + "-".repeat(10 - filled);
+					const model = ctx.model?.id || "no-model";
+
+					const l1Left =
+						theme.fg("dim", ` ${model} `) +
+						theme.fg("warning", "[") +
+						theme.fg("success", "#".repeat(filled)) +
+						theme.fg("dim", "-".repeat(10 - filled)) +
+						theme.fg("warning", "]") +
+						theme.fg("dim", " ") +
+						theme.fg("accent", `${Math.round(pct)}%`);
+
+					const l1Right =
+						theme.fg("success", `${fmt(tokIn)}`) +
+						theme.fg("dim", " in ") +
+						theme.fg("accent", `${fmt(tokOut)}`) +
+						theme.fg("dim", " out ") +
+						theme.fg("warning", `$${cost.toFixed(4)}`) +
+						theme.fg("dim", " ");
+
+					const pad1 = " ".repeat(Math.max(1, width - visibleWidth(l1Left) - visibleWidth(l1Right)));
+					const line1 = truncateToWidth(l1Left + pad1 + l1Right, width, "");
+
+					// --- Line 2: cwd + branch (left), tool tally (right) ---
+					const l2Left =
+						theme.fg("dim", ` ${dir}`) +
+						(branch
+							? theme.fg("dim", " ") + theme.fg("warning", "(") + theme.fg("success", branch) + theme.fg("warning", ")")
+							: "");
+
+					const entries = Object.entries(counts);
+					const l2Right = entries.length === 0
+						? theme.fg("dim", "waiting for tools ")
+						: entries.map(
+							([name, count]) =>
+								theme.fg("accent", name) + theme.fg("dim", " ") + theme.fg("success", `${count}`)
+						).join(theme.fg("warning", " | ")) + theme.fg("dim", " ");
+
+					const pad2 = " ".repeat(Math.max(1, width - visibleWidth(l2Left) - visibleWidth(l2Right)));
+					const line2 = truncateToWidth(l2Left + pad2 + l2Right, width, "");
+
+					return [line1, line2];
+				},
+			};
+		});
+	});
+}