pi-agent-extensions 0.3.5 → 0.4.0

package/README.md

@@ -34,6 +34,8 @@ Original repository: https://github.com/mitsuhiko/agent-stuff
 | **cwd_history** | Tracker | Tracks directory changes in context | ✅ Stable |
 | **btw** | Command | Quick side questions without history | ✅ Stable |
 | **nvidia-nim** | Command | Nvidia NIM auth & config | ✅ Stable |
+| **powerline-footer** | UI | Custom powerline-style footer bar | ✅ Stable |
+| **session-breakdown** | Command | Session analytics dashboard | ✅ Stable |
 
 ## Install
 
@@ -290,7 +292,7 @@ Ask quick "by the way" side questions without polluting your conversation histor
 - ✅ Ephemeral overlay — nothing enters session history
 - ✅ Zero context cost — no tokens wasted
 - ✅ Scrollable answer (↑↓/j/k, PgUp/PgDn)
-- ✅ Prefers cheap/fast models (Codex mini → Haiku → current)
+- ✅ Uses your currently selected model
 - ✅ Non-UI fallback (prints to stdout)
 
 **When to use `/btw` vs normal prompts:**
@@ -338,6 +340,23 @@ Authenticate and configure Nvidia NIM as an LLM provider.
 - Adds configured models to `~/.pi/agent/settings.json` `enabledModels` for scoped `/model` + Ctrl+P cycling
 - Model IDs must be `org/model` (exactly one `/`), e.g. `moonshotai/kimi-k2.5` (not `nvidia/moonshotai/kimi-k2.5`)
 
+**Powerline Footer**
+Custom powerline-style footer replacing the default pi footer with richer information.
+- Git branch + working tree status (staged/unstaged/untracked/ahead/behind)
+- Model name + context usage with color-coded percentage
+- Estimated session cost from model pricing
+- Session duration timer
+- Python virtualenv / conda environment detection
+- Extension statuses + session name
+- Auto-refreshes every 10 seconds via async git commands
+
+**Session Breakdown (`/session-breakdown`)**
+Interactive analytics dashboard for your pi sessions.
+- Analyzes all sessions in `~/.pi/agent/sessions/`
+- Shows sessions/day, messages/day, tokens/day, cost/day
+- Model breakdown with per-model usage stats
+- Filterable by 7/30/90 day windows
+
 ## Development
 
 ```bash

package/extensions/btw/index.ts

@@ -17,7 +17,7 @@
  * - Zero context cost — no tokens wasted on history
  */
 
-import { complete, type Model, type Api, type UserMessage } from "@mariozechner/pi-ai";
+import { complete, type UserMessage } from "@mariozechner/pi-ai";
 import type {
   ExtensionAPI,
   ExtensionCommandContext,
@@ -38,7 +38,7 @@ import {
   type TUI,
 } from "@mariozechner/pi-tui";
 
-import { getRequestAuth, hasRequestAuth } from "../shared/auth.js";
+import { getRequestAuth } from "../shared/auth.js";
 import {
   BTW_SYSTEM_PROMPT,
   buildBtwUserMessage,
@@ -46,40 +46,6 @@ import {
   extractResponseText,
 } from "./btw.js";
 
-const HAIKU_MODEL_ID = "claude-haiku-4-5";
-const CODEX_MODEL_ID = "gpt-5.1-codex-mini";
-
-/**
- * Select a cheap/fast model for BTW side questions.
- * Prefers Codex mini → Haiku → current model.
- */
-async function selectBtwModel(
-  currentModel: Model<Api>,
-  modelRegistry: {
-    find: (provider: string, modelId: string) => Model<Api> | undefined;
-    getApiKeyAndHeaders: (
-      model: Model<Api>,
-    ) => Promise<
-      | { ok: true; apiKey?: string; headers?: Record<string, string> }
-      | { ok: false; error: string }
-    >;
-  },
-): Promise<Model<Api>> {
-  // Try Codex mini first (cheapest)
-  const codexModel = modelRegistry.find("openai-codex", CODEX_MODEL_ID);
-  if (codexModel && (await hasRequestAuth(modelRegistry, codexModel))) {
-    return codexModel;
-  }
-
-  // Try Haiku (fast & cheap)
-  const haikuModel = modelRegistry.find("anthropic", HAIKU_MODEL_ID);
-  if (haikuModel && (await hasRequestAuth(modelRegistry, haikuModel))) {
-    return haikuModel;
-  }
-
-  // Fallback to current model
-  return currentModel;
-}
 
 /**
  * Overlay component that displays the BTW answer.
@@ -299,8 +265,8 @@ async function runBtwCommand(
     conversationText = serializeConversation(llmMessages);
   }
 
-  // Select model (prefer cheap models)
-  const btwModel = await selectBtwModel(ctx.model, ctx.modelRegistry);
+  // Use the currently selected model
+  const btwModel = ctx.model;
 
   // Build LLM messages
   const userMessage: UserMessage = {

package/extensions/control/index.ts

@@ -189,11 +189,11 @@ function getSocketPath(sessionId: string): string {
 }
 
 function isSafeSessionId(sessionId: string): boolean {
-	return !sessionId.includes("/") && !sessionId.includes("\\") && !sessionId.includes("..") && sessionId.length > 0;
+	return /^[a-zA-Z0-9_-]+$/.test(sessionId);
 }
 
 function isSafeAlias(alias: string): boolean {
-	return !alias.includes("/") && !alias.includes("\\") && !alias.includes("..") && alias.length > 0;
+	return /^[a-zA-Z0-9_ -]+$/.test(alias);
 }
 
 function getAliasPath(alias: string): string {

package/extensions/notify/index.ts

@@ -15,8 +15,12 @@ import { Markdown, type MarkdownTheme } from "@mariozechner/pi-tui";
  * Send a desktop notification via OSC 777 escape sequence.
  */
 const notify = (title: string, body: string): void => {
+	// Sanitize to prevent terminal escape sequence injection (remove control characters like ESC and BEL)
+	const safeTitle = title.replace(/[\x00-\x1f\x7f]/g, "");
+	const safeBody = body.replace(/[\x00-\x1f\x7f]/g, "");
+
 	// OSC 777 format: ESC ] 777 ; notify ; title ; body BEL
-	process.stdout.write(`\x1b]777;notify;${title};${body}\x07`);
+	process.stdout.write(`\x1b]777;notify;${safeTitle};${safeBody}\x07`);
 };
 
 const isTextPart = (part: unknown): part is { type: "text"; text: string } =>

package/extensions/powerline-footer/index.ts

@@ -0,0 +1,220 @@
+import type { ExtensionAPI, ExtensionContext, ContextUsage, ReadonlyFooterDataProvider } from "@mariozechner/pi-coding-agent";
+import { type Component, type Theme, type TUI, visibleWidth } from "@mariozechner/pi-tui";
+import * as child_process from "child_process";
+
+class PowerlineFooter implements Component {
+	private tui: TUI;
+	private theme: Theme;
+	private footerData: ReadonlyFooterDataProvider;
+	private ctx: ExtensionContext;
+	private interval?: ReturnType<typeof setInterval>;
+	private sessionStartTime: number;
+
+	// Cached git state (updated async every 10s)
+	private gitStatusExtras: string = "";
+
+	constructor(tui: TUI, theme: Theme, footerData: ReadonlyFooterDataProvider, ctx: ExtensionContext) {
+		this.tui = tui;
+		this.theme = theme;
+		this.footerData = footerData;
+		this.ctx = ctx;
+		this.sessionStartTime = Date.now();
+
+		this.fetchAsyncData();
+
+		this.interval = setInterval(() => {
+			this.fetchAsyncData();
+			this.tui.requestRender();
+		}, 10000);
+	}
+
+	dispose() {
+		if (this.interval) {
+			clearInterval(this.interval);
+		}
+	}
+
+	private fetchAsyncData() {
+		const cwd = this.ctx.cwd;
+		const branch = this.footerData.getGitBranch();
+
+		if (branch) {
+			child_process.exec("git --no-optional-locks status --porcelain", { encoding: "utf8", cwd }, (err, stdout) => {
+				if (err) return;
+				const staged = (stdout.match(/^[AMDRC]/gm) || []).length;
+				const unstaged = (stdout.match(/^.[MD]/gm) || []).length;
+				const untracked = (stdout.match(/^\?\?/gm) || []).length;
+
+				child_process.exec("git --no-optional-locks rev-list --count --left-right @{u}...HEAD", { encoding: "utf8", cwd }, (err2, revListOut) => {
+					let ahead = 0;
+					let behind = 0;
+					if (!err2 && revListOut && revListOut.includes("\t")) {
+						const [b, a] = revListOut.trim().split("\t");
+						behind = parseInt(b, 10);
+						ahead = parseInt(a, 10);
+					}
+
+					let statusExtras = "";
+					if (staged > 0) statusExtras += `+${staged}`;
+					if (unstaged > 0) statusExtras += `!${unstaged}`;
+					if (untracked > 0) statusExtras += `?${untracked}`;
+					if (ahead > 0) statusExtras += `⇡${ahead}`;
+					if (behind > 0) statusExtras += `⇣${behind}`;
+					this.gitStatusExtras = statusExtras;
+					this.tui.requestRender();
+				});
+			});
+		} else {
+			this.gitStatusExtras = "";
+		}
+	}
+
+	invalidate() {}
+	handleInput(_data: string) {}
+
+	private formatTokens(num: number): string {
+		if (num >= 1_000_000) {
+			return (num / 1_000_000).toFixed(1) + "M";
+		}
+		if (num >= 1000) {
+			return Math.floor(num / 1000) + "K";
+		}
+		return String(num);
+	}
+
+	render(width: number): string[] {
+		const RESET = "\x1b[0m";
+		const DIM = "\x1b[2m";
+		const BOLD = "\x1b[1m";
+		const MAUVE = "\x1b[38;5;183m";
+		const BLUE = "\x1b[38;5;111m";
+		const GREEN = "\x1b[38;5;150m";
+		const YELLOW = "\x1b[38;5;222m";
+		const RED = "\x1b[38;5;211m";
+		const SKY = "\x1b[38;5;117m";
+		const PEACH = "\x1b[38;5;216m";
+		const OVERLAY2 = "\x1b[38;5;103m";
+
+		// --- Directory ---
+		const cwd = this.ctx.cwd;
+		const homeDir = process.env.HOME || process.env.USERPROFILE || "";
+		let shortDir = cwd;
+		if (homeDir && shortDir.startsWith(homeDir)) {
+			shortDir = "~" + shortDir.slice(homeDir.length);
+		}
+		const parts = shortDir.split("/");
+		if (parts.length > 4) {
+			shortDir = parts.slice(parts.length - 4).join("/");
+		}
+
+		// --- Git ---
+		let gitInfo = "";
+		const branch = this.footerData.getGitBranch();
+		if (branch) {
+			gitInfo = ` ${DIM}on${RESET} ${MAUVE} ${branch}${RESET}`;
+			if (this.gitStatusExtras) {
+				gitInfo += ` ${RED}${this.gitStatusExtras}${RESET}`;
+			}
+		}
+
+		// --- Model ---
+		const model = this.ctx.model;
+		const modelShort = model?.name || model?.id || "unknown";
+
+		// --- Context usage ---
+		const contextUsage = this.ctx.getContextUsage();
+		let contextInfo = "";
+		let costInfo = "";
+
+		if (contextUsage) {
+			const used = contextUsage.tokens;
+			const total = contextUsage.contextWindow;
+			const pct = contextUsage.percent;
+			const remaining = 100 - pct;
+
+			let ctxColor = GREEN;
+			if (remaining < 20) {
+				ctxColor = RED;
+			} else if (remaining < 50) {
+				ctxColor = YELLOW;
+			}
+
+			contextInfo = `${ctxColor}${this.formatTokens(used)}/${this.formatTokens(total)} (${pct}%)${RESET}`;
+
+			// Cost estimation from model pricing
+			if (model?.cost && contextUsage.usageTokens > 0) {
+				// Rough estimate: treat usageTokens as input, trailingTokens as recent output
+				const inputTokens = contextUsage.usageTokens;
+				const outputTokens = contextUsage.trailingTokens;
+				const cost =
+					(inputTokens * model.cost.input) / 1_000_000 +
+					(outputTokens * model.cost.output) / 1_000_000;
+
+				if (cost > 0.001) {
+					costInfo = ` ${DIM}|${RESET} ${PEACH}$${cost.toFixed(2)}${RESET}`;
+				}
+			}
+		}
+
+		// --- Session duration ---
+		let durationInfo = "";
+		const durationMs = Date.now() - this.sessionStartTime;
+		if (durationMs > 0) {
+			const durS = Math.floor(durationMs / 1000);
+			let durFmt = "";
+			if (durS >= 3600) {
+				const durH = Math.floor(durS / 3600);
+				const durM = Math.floor((durS % 3600) / 60);
+				durFmt = `${durH}h${durM}m`;
+			} else if (durS >= 60) {
+				durFmt = `${Math.floor(durS / 60)}m`;
+			} else {
+				durFmt = `${durS}s`;
+			}
+			durationInfo = ` ${DIM}|${RESET} ${OVERLAY2}${durFmt}${RESET}`;
+		}
+
+		// --- Python env ---
+		let envInfo = "";
+		if (process.env.CONDA_DEFAULT_ENV) {
+			envInfo = ` ${SKY}(${process.env.CONDA_DEFAULT_ENV})${RESET}`;
+		} else if (process.env.VIRTUAL_ENV) {
+			const venvName = process.env.VIRTUAL_ENV.split("/").pop() || "";
+			envInfo = ` ${SKY}(${venvName})${RESET}`;
+		}
+
+		// --- Time ---
+		const date = new Date();
+		const currentTime = `${String(date.getHours()).padStart(2, "0")}:${String(date.getMinutes()).padStart(2, "0")}`;
+
+		// --- Session name ---
+		const sessionName = this.ctx.sessionManager.getSessionName();
+		const sessionInfo = sessionName ? `${MAUVE}[${sessionName}]${RESET} ` : "";
+
+		// --- Extension statuses ---
+		let statusInfo = "";
+		const statuses = this.footerData.getExtensionStatuses();
+		if (statuses.size > 0) {
+			const parts: string[] = [];
+			for (const [, text] of statuses) {
+				if (text) parts.push(text);
+			}
+			if (parts.length > 0) {
+				statusInfo = ` ${DIM}|${RESET} ${parts.join(" ")}`;
+			}
+		}
+
+		const line = `${sessionInfo}${BOLD}${BLUE} ${shortDir}${RESET}${gitInfo} ${DIM}|${RESET} ${OVERLAY2}${modelShort}${RESET} ${contextInfo}${costInfo}${durationInfo}${envInfo}${statusInfo} ${DIM}${currentTime}${RESET}`;
+
+		return [line];
+	}
+}
+
+export default function (pi: ExtensionAPI) {
+	pi.on("session_start", async (_event, ctx) => {
+		if (!ctx.hasUI) return;
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			return new PowerlineFooter(tui, theme, footerData, ctx);
+		});
+	});
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-extensions",
-  "version": "0.3.5",
+  "version": "0.4.0",
   "description": "Collection of extensions for pi coding agent",
   "type": "module",
   "repository": {
@@ -47,7 +47,8 @@
       "./extensions/todos/index.ts",
       "./extensions/whimsical/index.ts",
       "./extensions/nvidia-nim/index.ts",
-      "./extensions/btw/index.ts"
+      "./extensions/btw/index.ts",
+      "./extensions/powerline-footer/index.ts"
     ],
     "themes": [
       "./themes/nightowl.json",