pi-bar 0.3.1 → 0.3.3

package/README.md

@@ -3,21 +3,22 @@
 [![npm version](https://img.shields.io/npm/v/pi-bar.svg)](https://www.npmjs.com/package/pi-bar)
 [![npm downloads](https://img.shields.io/npm/dm/pi-bar.svg)](https://www.npmjs.com/package/pi-bar)
 
-**Never accidentally run Opus on a typo again.** pi-bar puts the active model, thinking level, and context usage in pi's footer so you catch wrong-model / wrong-reasoning / context-pressure problems before they cost you.
+**Never accidentally run Opus on a typo again.** pi-bar puts the active model, thinking level, context usage, and extension statuses in pi's footer so you catch wrong-model / wrong-reasoning / context-pressure problems before they cost you.
 
 ```text
-◈ claude-opus-4.7  ❯  ✦ think:med  ❯  ◷ 2.6% / 1.0M
+◈ claude-opus-4.7  ❯  ✦ think:med  ❯  ◷ 2.6% / 1.0M  ❯  ✧ plan:active
 ```
 
-![pi-bar screenshot, three rows showing green, yellow, and red context usage](https://cdn.jsdelivr.net/npm/pi-bar@0.3.0/assets/screenshot.png)
+![pi-bar screenshot, three rows showing green, yellow, and red context usage](https://cdn.jsdelivr.net/npm/pi-bar@0.3.3/assets/screenshot.png)
 
 ## Why use it?
 
 - **See the active model at a glance** — catch accidental model switches before an expensive or sensitive task starts.
 - **Track thinking level in place** — immediately notice when you are using the wrong reasoning setting.
 - **Watch context pressure early** — context usage turns green, yellow, then red as you approach the limit.
+- **Keep extension statuses visible** — statuses set with `ctx.ui.setStatus()` still appear in the custom footer.
 
-pi-bar is intentionally tiny: one small extension, no broad behavior changes, and no commands to learn. It replaces pi's built-in footer with a compact model / thinking / context status line that refreshes when the model changes, thinking level changes, and after assistant turns.
+pi-bar is intentionally tiny: one small extension, no broad behavior changes, and no commands to learn. It replaces pi's built-in footer with a compact model / thinking / context / extension-status line that refreshes when model, thinking level, context usage, or extension statuses change.
 
 ## Install
 
@@ -38,11 +39,27 @@ pi-bar works out of the box, but you can tune it with environment variables befo
 ### Show or hide segments
 
 ```bash
-PI_BAR_SHOW=model,thinking,context pi
+PI_BAR_SHOW=model,thinking,context,extensions pi
 PI_BAR_SHOW=model,context pi
 ```
 
-Allowed segments are `model`, `thinking`, and `context`.
+Allowed segments are `model`, `thinking`, `context`, and `extensions`. The `extensions` segment is hidden when no extension has set a status.
+
+### Configure extension statuses
+
+pi-bar shows statuses set by other extensions with `ctx.ui.setStatus()`.
+Use `/bar status` inside pi to choose which status keys appear.
+
+```text
+/bar status list
+/bar status hide plan-mode
+/bar status show plan-mode
+/bar status only plan-mode,preset
+/bar status none
+/bar status all
+```
+
+Status filter choices persist in the current session.
 
 ### Change context thresholds
 
@@ -62,10 +79,10 @@ PI_BAR_ICONS=nerdfont pi
 
 Supported presets:
 
-- `unicode` — default, e.g. `◈ gpt-5.5  ❯  ✦ think:medium  ❯  ◷ 42.7% / 272k`
-- `plain` — maximum compatibility, e.g. `model:gpt-5.5 | think:medium | ctx:42.7% / 272k`
-- `emoji` — colorful emoji markers, e.g. `🤖 gpt-5.5  ›  ✨ think:medium  ›  🧭 42.7% / 272k`
-- `nerdfont` — icon-font glyphs for terminals configured with a Nerd Font, e.g. `󰊩 gpt-5.5  󰮱  󰉑 think:medium  󰮱  󰍛 42.7% / 272k`
+- `unicode` — default, e.g. `◈ gpt-5.5  ❯  ✦ think:medium  ❯  ◷ 42.7% / 272k  ❯  ✧ plan:active`
+- `plain` — maximum compatibility, e.g. `model:gpt-5.5 | think:medium | ctx:42.7% / 272k | ext:plan:active`
+- `emoji` — colorful emoji markers, e.g. `🤖 gpt-5.5  ›  ✨ think:medium  ›  🧭 42.7% / 272k  ›  🧩 plan:active`
+- `nerdfont` — icon-font glyphs for terminals configured with a Nerd Font, e.g. `󰚩 gpt-5.5    󰧑 think:medium    󰍛 42.7% / 272k    󰐱 plan:active`
 
 ## Compatibility
 

package/extensions/status-footer.ts

@@ -1,39 +1,82 @@
 /**
  * pi-bar — footer/statusline extension.
  *
- * Replaces the built-in footer with three left-aligned segments:
- *   ◈ <model name> ❯ ✦ think:<level> ❯ ◷ <context% / window>
+ * Replaces the built-in footer with left-aligned segments:
+ *   ◈ <model name> ❯ ✦ think:<level> ❯ ◷ <context% / window> ❯ ✧ <extension statuses>
  *
  * Examples:
  *   ◈ claude-opus-4.7  ❯  ✦ think:med  ❯  ◷ 2.6% / 1.0M
  *
- * Re-renders on model change, thinking-level change, and after each
- * assistant turn so context usage stays current.
+ * Re-renders on model change, thinking-level change, status updates,
+ * and after each assistant turn so context usage stays current.
  *
  * Configuration env vars:
- *   PI_BAR_SHOW=model,thinking,context
+ *   PI_BAR_SHOW=model,thinking,context,extensions
  *   PI_BAR_THRESHOLDS=70,90
  *   PI_BAR_ICONS=unicode|plain|emoji|nerdfont
  */
 
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { truncateToWidth } from "@earendil-works/pi-tui";
 
-type SegmentName = "model" | "thinking" | "context";
+type SegmentName = "model" | "thinking" | "context" | "extensions";
 type IconPreset = "unicode" | "plain" | "emoji" | "nerdfont";
+type StatusFilter =
+	| { mode: "all"; hidden: Set<string> }
+	| { mode: "only"; shown: Set<string> };
+type SerializedStatusFilter =
+	| { mode: "all"; hidden: string[] }
+	| { mode: "only"; shown: string[] };
 
-const DEFAULT_SEGMENTS: SegmentName[] = ["model", "thinking", "context"];
+const STATUS_FILTER_ENTRY_TYPE = "pi-bar-status-filter";
+
+const DEFAULT_SEGMENTS: SegmentName[] = [
+	"model",
+	"thinking",
+	"context",
+	"extensions",
+];
 const DEFAULT_WARNING_THRESHOLD = 70;
 const DEFAULT_ERROR_THRESHOLD = 90;
 
 const ICON_PRESETS: Record<
 	IconPreset,
-	{ model: string; thinking: string; context: string; separator: string }
+	{
+		model: string;
+		thinking: string;
+		context: string;
+		extensions: string;
+		separator: string;
+	}
 > = {
-	unicode: { model: "◈", thinking: "✦", context: "◷", separator: "❯" },
-	plain: { model: "model", thinking: "think", context: "ctx", separator: "|" },
-	emoji: { model: "🤖", thinking: "✨", context: "🧭", separator: "›" },
-	nerdfont: { model: "󰚩", thinking: "󰧑", context: "󰍛", separator: "" },
+	unicode: {
+		model: "◈",
+		thinking: "✦",
+		context: "◷",
+		extensions: "✧",
+		separator: "❯",
+	},
+	plain: {
+		model: "model",
+		thinking: "think",
+		context: "ctx",
+		extensions: "ext",
+		separator: "|",
+	},
+	emoji: {
+		model: "🤖",
+		thinking: "✨",
+		context: "🧭",
+		extensions: "🧩",
+		separator: "›",
+	},
+	nerdfont: {
+		model: "󰚩",
+		thinking: "󰧑",
+		context: "󰍛",
+		extensions: "󰐱",
+		separator: "",
+	},
 };
 
 function formatTokens(n: number): string {
@@ -95,7 +138,7 @@ function parseSegments(): SegmentName[] {
 		.split(",")
 		.map((segment) => segment.trim().toLowerCase())
 		.filter((segment): segment is SegmentName =>
-			["model", "thinking", "context"].includes(segment),
+			["model", "thinking", "context", "extensions"].includes(segment),
 		);
 
 	return requested.length > 0 ? requested : DEFAULT_SEGMENTS;
@@ -140,16 +183,153 @@ function prefixed(icon: string, text: string, preset: IconPreset): string {
 	return `${icon} ${text}`;
 }
 
+function shouldShowStatus(key: string, filter: StatusFilter): boolean {
+	if (filter.mode === "only") return filter.shown.has(key);
+	return !filter.hidden.has(key);
+}
+
+function formatExtensionStatuses(
+	statuses: ReadonlyMap<string, string>,
+	filter: StatusFilter,
+	seenStatusKeys: Set<string>,
+): string | null {
+	const parts = Array.from(statuses.entries())
+		.filter(([, text]) => text.trim().length > 0)
+		.filter(([key]) => {
+			seenStatusKeys.add(key);
+			return shouldShowStatus(key, filter);
+		})
+		.map(([key, text]) => `${key}:${text}`);
+
+	return parts.length > 0 ? parts.join(" ") : null;
+}
+
+function serializeStatusFilter(filter: StatusFilter): SerializedStatusFilter {
+	if (filter.mode === "only") {
+		return { mode: "only", shown: Array.from(filter.shown).sort() };
+	}
+	return { mode: "all", hidden: Array.from(filter.hidden).sort() };
+}
+
+function parseSerializedStatusFilter(value: unknown): StatusFilter | null {
+	if (!value || typeof value !== "object") return null;
+	const data = value as Partial<SerializedStatusFilter>;
+
+	if (data.mode === "only" && Array.isArray(data.shown)) {
+		return { mode: "only", shown: new Set(data.shown) };
+	}
+	if (data.mode === "all" && Array.isArray(data.hidden)) {
+		return { mode: "all", hidden: new Set(data.hidden) };
+	}
+	return null;
+}
+
+function splitStatusKeys(raw: string): string[] {
+	return raw
+		.split(/[\s,]+/)
+		.map((key) => key.trim())
+		.filter(Boolean);
+}
+
+function describeStatusFilter(filter: StatusFilter): string {
+	if (filter.mode === "only") {
+		const shown = Array.from(filter.shown).sort();
+		return shown.length > 0 ? `showing only: ${shown.join(", ")}` : "showing none";
+	}
+
+	const hidden = Array.from(filter.hidden).sort();
+	return hidden.length > 0 ? `showing all except: ${hidden.join(", ")}` : "showing all";
+}
+
 export default function (pi: ExtensionAPI) {
 	let requestRender: (() => void) | undefined;
+	let statusFilter: StatusFilter = { mode: "all", hidden: new Set() };
+	const seenStatusKeys = new Set<string>();
 	const refresh = () => requestRender?.();
+	const restoreStatusFilter = (ctx: ExtensionContext) => {
+		let restoredFilter: StatusFilter | null = null;
+		for (const entry of ctx.sessionManager.getBranch()) {
+			if (entry.type === "custom" && entry.customType === STATUS_FILTER_ENTRY_TYPE) {
+				restoredFilter = parseSerializedStatusFilter(entry.data);
+			}
+		}
+		statusFilter = restoredFilter ?? { mode: "all", hidden: new Set() };
+	};
+
+	pi.registerCommand("bar", {
+		description: "Configure pi-bar footer status segments",
+		handler: async (args, ctx) => {
+			const [section, action, ...rest] = args.trim().split(/\s+/).filter(Boolean);
+			if (!section || section === "status" || section === "statuses") {
+				const keys = splitStatusKeys(rest.join(" "));
+
+				switch (action) {
+					case undefined:
+					case "list":
+					case "ls": {
+						const known = Array.from(seenStatusKeys).sort();
+						ctx.ui.notify(
+							`pi-bar statuses: ${describeStatusFilter(statusFilter)}${known.length > 0 ? `; known: ${known.join(", ")}` : "; known: none yet"}`,
+							"info",
+						);
+						return;
+					}
+					case "all":
+						statusFilter = { mode: "all", hidden: new Set() };
+						break;
+					case "none":
+						statusFilter = { mode: "only", shown: new Set() };
+						break;
+					case "only":
+						statusFilter = { mode: "only", shown: new Set(keys) };
+						break;
+					case "hide":
+						if (statusFilter.mode === "only") {
+							for (const key of keys) statusFilter.shown.delete(key);
+						} else {
+							for (const key of keys) statusFilter.hidden.add(key);
+						}
+						break;
+					case "show":
+						if (statusFilter.mode === "only") {
+							for (const key of keys) statusFilter.shown.add(key);
+						} else {
+							for (const key of keys) statusFilter.hidden.delete(key);
+						}
+						break;
+					default:
+						ctx.ui.notify(
+							"Usage: /bar status [list|all|none|only <keys>|show <keys>|hide <keys>]",
+							"warning",
+						);
+						return;
+				}
+
+				pi.appendEntry(STATUS_FILTER_ENTRY_TYPE, serializeStatusFilter(statusFilter));
+				refresh();
+				ctx.ui.notify(`pi-bar statuses: ${describeStatusFilter(statusFilter)}`, "info");
+				return;
+			}
+
+			ctx.ui.notify(
+				"Usage: /bar status [list|all|none|only <keys>|show <keys>|hide <keys>]",
+				"warning",
+			);
+		},
+	});
 
 	pi.on("model_select", async () => refresh());
 	pi.on("thinking_level_select", async () => refresh());
 	pi.on("turn_end", async () => refresh());
 	pi.on("message_end", async () => refresh());
+	pi.on("session_tree", async (_event, ctx) => {
+		restoreStatusFilter(ctx);
+		refresh();
+	});
 
 	pi.on("session_start", async (_event, ctx) => {
+		restoreStatusFilter(ctx);
+
 		if (!ctx.hasUI) return;
 
 		const visibleSegments = parseSegments();
@@ -157,7 +337,7 @@ export default function (pi: ExtensionAPI) {
 		const icons = ICON_PRESETS[iconPreset];
 		const { warningThreshold, errorThreshold } = parseThresholds();
 
-		ctx.ui.setFooter((tui, theme) => {
+		ctx.ui.setFooter((tui, theme, footerData) => {
 			requestRender = () => tui.requestRender();
 
 			return {
@@ -168,6 +348,11 @@ export default function (pi: ExtensionAPI) {
 				render(width: number): string[] {
 					const modelName = formatModelName(ctx.model?.id);
 					const thinkingLevel = String(pi.getThinkingLevel());
+					const extensionStatuses = formatExtensionStatuses(
+						footerData?.getExtensionStatuses?.() ?? new Map(),
+						statusFilter,
+						seenStatusKeys,
+					);
 					const usage = ctx.getContextUsage();
 					const contextSegmentColor = contextColor(
 						usage?.percent,
@@ -179,7 +364,7 @@ export default function (pi: ExtensionAPI) {
 						? `${usage.percent !== null ? `${usage.percent.toFixed(1)}%` : "—%"} / ${formatTokens(usage.contextWindow)}`
 						: "—";
 
-					const segmentRenderers: Record<SegmentName, string> = {
+					const segmentRenderers: Record<SegmentName, string | null> = {
 						model: theme.fg(
 							"accent",
 							prefixed(icons.model, modelName, iconPreset),
@@ -192,11 +377,18 @@ export default function (pi: ExtensionAPI) {
 							contextSegmentColor,
 							prefixed(icons.context, contextText, iconPreset),
 						),
+						extensions: extensionStatuses
+							? theme.fg(
+									"muted",
+									prefixed(icons.extensions, extensionStatuses, iconPreset),
+								)
+							: null,
 					};
 
 					const separator = `  ${theme.fg("dim", icons.separator)}  `;
 					const line = visibleSegments
 						.map((segment) => segmentRenderers[segment])
+						.filter((segment): segment is string => segment !== null)
 						.join(separator);
 
 					return [truncateToWidth(line, width)];

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-bar",
-  "version": "0.3.1",
-  "description": "Never accidentally run Opus on a typo. Pi footer/statusline showing active model, thinking level, and context usage.",
+  "version": "0.3.3",
+  "description": "Never accidentally run Opus on a typo. Pi footer/statusline showing active model, thinking level, context usage, and extension statuses.",
   "type": "module",
   "keywords": [
     "pi-package",
@@ -46,6 +46,6 @@
     "extensions": [
       "./extensions/status-footer.ts"
     ],
-    "image": "https://cdn.jsdelivr.net/npm/pi-bar@0.3.0/assets/screenshot.png"
+    "image": "https://cdn.jsdelivr.net/npm/pi-bar@0.3.3/assets/screenshot.png"
   }
 }