pi-zentui 0.3.3 → 0.5.0

package/README.md

@@ -21,8 +21,10 @@ Zentui brings two popular aesthetics to Pi:
 - `on  branch` — git branch with icon
 - `[!?↑]` — git status indicators (modified, untracked, ahead/behind, stashed, etc.)
 - `via  v5.5.0` — runtime detection with version and Starship-style Nerd Font runtime/language modules
+- Optional segments (off by default): `user@host`, current time, OS icon, and session duration
 - Right side shows context usage, token counts, and cost
 - Built-in footer segments can be shown or hidden individually from `/zentui`
+- Fully custom Starship-style layout via a `footerFormat` template string — see [Footer Format Template](#footer-format-template)
 - Third-party Pi extension statuses from `ctx.ui.setStatus()` can be shown on the left,
   middle, or right side, or hidden per status key from `/zentui`
 
@@ -127,7 +129,7 @@ pi install git:github.com/lmilojevicc/pi-zentui
 
 User config lives at `~/.pi/agent/zentui.json`. The file is optional: missing or invalid known values fall back to Zentui defaults, unknown keys are ignored at runtime, and `/zentui` can patch color-source settings, UI feature toggles, built-in footer segment visibility, and active third-party status placements.
 
-The interactive `/zentui` menu is split into four sections. Use `Tab` and `Shift+Tab` to switch between `Coloring`, `Features`, `Built-in segments`, and `Extension segments`.
+The interactive `/zentui` menu is split into five sections. Use `Tab` and `Shift+Tab` to switch between `Coloring`, `Features`, `Layout`, `Built-in segments`, and `Extension segments`.
 
 Useful slash-command shortcuts:
 
@@ -141,6 +143,8 @@ Useful slash-command shortcuts:
 /zentui copy-friendly enable
 /zentui copy-friendly disable
 /zentui copy-friendly toggle
+/zentui format "$cwd on branch $git_branch$git_status using $runtime $fill $context"
+/zentui format clear
 ```
 
 Default config values — copy this and change any value you want:
@@ -148,7 +152,14 @@ Default config values — copy this and change any value you want:
 ```json
 {
 	"projectRefreshIntervalMs": 30000,
+	"footerFormat": "",
+	"contextStyle": "text",
+	"contextThresholds": {
+		"warning": 70,
+		"error": 90
+	},
 	"icons": {
+		"mode": "auto",
 		"cwd": "󰝰",
 		"git": "",
 		"ahead": "↑",
@@ -164,7 +175,10 @@ Default config values — copy this and change any value you want:
 		"typechanged": "T",
 		"cacheHit": "󰆼",
 		"editorPrompt": "",
-		"rail": "│"
+		"rail": "│",
+		"username": "",
+		"time": "",
+		"os": ""
 	},
 	"colors": {
 		"cwd": "bold cyan",
@@ -178,6 +192,10 @@ Default config values — copy this and change any value you want:
 		"extensionStatus": "bright-black",
 		"separator": "bright-black",
 		"runtimePrefix": "",
+		"sessionDuration": "yellow",
+		"username": "bold yellow",
+		"time": "bold yellow",
+		"os": "bold white",
 		"editorAccent": "accent",
 		"editorPrompt": "accent",
 		"editorBorder": "borderMuted",
@@ -204,10 +222,15 @@ Default config values — copy this and change any value you want:
 		"cwd": true,
 		"gitBranch": true,
 		"gitStatus": true,
+		"gitCounts": false,
 		"runtime": true,
 		"context": true,
 		"tokens": true,
-		"cost": true
+		"cost": true,
+		"sessionDuration": false,
+		"username": false,
+		"time": false,
+		"os": false
 	},
 	"extensionStatuses": {
 		"defaultPlacement": "right",
@@ -217,12 +240,15 @@ Default config values — copy this and change any value you want:
 }
 ```
 
-- Style values can be Starship/terminal strings (`bold purple`, `fg:202`, `#89b4fa`, `bg:blue fg:bright-green`) or Pi theme tokens (`accent`, `borderMuted`, `thinkingHigh`).
-- `projectRefreshIntervalMs`: project status polling interval; `0` disables polling.
-- `icons`: every shown icon key is configurable; omit any key to use the Zentui default. `rail` sets the vertical glyph drawn as the left rail of the active editor frame and previous user messages when `copyFriendly` is disabled (default `│`; any single Unicode vertical or block glyph). `editorPrompt` controls an optional copy-friendly editor prompt glyph; the default is `""` so copy-friendly mode stays rail-free.
+- Style values can be Starship/terminal strings (`bold purple`, `fg:202`, `#89b` / `#89b4fa`, `bg:blue fg:bright-green`) or Pi theme tokens (`accent`, `borderMuted`, `thinkingHigh`). Short `#rgb` hex values expand to `#rrggbb`.
+- `projectRefreshIntervalMs`: project status polling interval; `0` disables polling. Values `1..4999` clamp up to `5000` (minimum 5s); invalid/non-finite values fall back to `30000`.
+- `contextStyle`: `text` (default), `gauge`, or `text+gauge` for the context segment.
+- `contextThresholds`: `{ warning, error }` percentages (default `70` / `90`) that select contextNormal / contextWarning / contextError colors.
+- `icons`: every shown icon key is configurable; omit any key to use the Zentui default. `icons.mode` is `auto` | `nerd` | `ascii` (default `auto`, same glyphs as nerd). ASCII mode swaps in plain fallbacks for statusline icons and runtime symbols — useful without a Nerd Font. Custom per-icon strings always win over mode defaults. Custom `icons.os` always wins; when left at the mode default, Zentui maps the OS icon by platform. `rail` sets the vertical glyph drawn as the left rail of the active editor frame and previous user messages when `copyFriendly` is disabled (default `│`; any single Unicode vertical or block glyph). `editorPrompt` controls an optional copy-friendly editor prompt glyph; the default is `""` so copy-friendly mode stays rail-free.
 - `colorSources`: `theme` maps styles through Pi theme tokens; `terminal` emits terminal colors. `/zentui` switches these sources; manual JSON controls specific style values.
 - `features`: `editor` enables Zentui's custom editor, selector borders, and previous-message chrome. `statusLine` enables Zentui's custom footer/status line. `copyFriendly` hides editor and previous-message rail glyphs so native terminal selection copies less chrome. All three can be changed from `/zentui` or direct slash-command arguments.
-- `footerSegments`: show or hide individual built-in footer segments (`cwd`, `gitBranch`, `gitStatus`, `runtime`, `context`, `tokens`, `cost`). Toggle them from the `Built-in segments` tab in `/zentui`.
+- `footerSegments`: show or hide individual built-in footer segments (`cwd`, `gitBranch`, `gitStatus`, `gitCounts`, `runtime`, `sessionDuration`, `username`, `time`, `os`, `context`, `tokens`, `cost`). Toggle them from the `Built-in segments` tab in `/zentui`.
+- `footerFormat`: optional Starship-style template string that fully controls the footer layout. When set, it overrides `footerSegments`. See [Footer Format Template](#footer-format-template) below. The `/zentui` **Layout** tab configures context style and icon mode; set or clear custom formats with `/zentui format`.
 - `extensionStatuses`: controls third-party statuses published by other Pi extensions through `ctx.ui.setStatus()`. `defaultPlacement` and each `placements` value can be `off`, `left`, `middle`, or `right`. The `Extension segments` tab in `/zentui` lists only statuses that are currently active.
 - The shown `editor*` values match the default `theme` source. Omit those keys to keep Zentui's source-aware defaults when switching between `theme` and `terminal`.
 - `editorAccent` styles the active editor rail and previous user-message rail when `features.copyFriendly` is disabled.
@@ -232,10 +258,64 @@ Default config values — copy this and change any value you want:
 
 Tip: when using copy-friendly mode, setting Pi's `editorPaddingX` to `1` in `~/.pi/agent/settings.json` keeps a small left gutter without copying a rail glyph.
 
+## Footer Format Template
+
+For full control, set a Starship-style `footerFormat` template string. It supports `$variable` and `${variable}` tokens, a special `$fill` token that splits the line into left and right zones, and conditional groups `( ... )` that drop entirely when every nested variable is empty. When set, it overrides the built-in `footerSegments` layout; when empty or omitted, the segment layout above is used.
+
+A second `$fill` creates a **centered middle zone** — content between the two fills is true-centered (`floor((gap - middle) / 2)`), just like third-party statuses placed `middle`.
+
+```json
+{
+	"footerFormat": "$os $username $cwd( on $git_branch)( $git_status)( via $runtime)$fill($context)($sep$tokens)($sep$cost)($sep$time)"
+}
+```
+
+Center the branch between directory and cost:
+
+```json
+{
+	"footerFormat": "$cwd $fill $git_branch $fill $cost"
+}
+```
+
+### Variables
+
+| Token               | Aliases      | Renders                                          |
+| ------------------- | ------------ | ------------------------------------------------ |
+| `$cwd`              | `$directory` | current directory                                |
+| `$git_branch`       | `$branch`    | git branch with icon                             |
+| `$git_status`       | `$status`    | `[!?↑]` status block                             |
+| `$git_state`        | `$state`     | `REBASING` / `MERGING` / … (optional `n/m`)      |
+| `$runtime`          |              | runtime icon + version                           |
+| `$session_duration` | `$duration`  | session running time                             |
+| `$username`         |              | `user@host`                                      |
+| `$os`               |              | operating-system icon                            |
+| `$time`             |              | current time `HH:MM`                             |
+| `$context`          |              | context usage (text and/or gauge via config)     |
+| `$tokens`           |              | input/output token counts                        |
+| `$cost`             |              | session cost                                     |
+| `$sep`              | `$separator` | themed ` | ` using `colors.separator`            |
+| `$fill`             | —            | special: splits zones                            |
+
+### `$fill` behavior
+
+| `$fill` count | Layout                                                                   |
+| ------------- | ------------------------------------------------------------------------ |
+| 0             | everything left-aligned                                                  |
+| 1             | tokens before → left, tokens after → right                               |
+| 2             | before first → left, between → **centered middle**, after second → right |
+| 3+            | first two count; extras ignored                                          |
+
+- Literal text (`on branch`, `using`, `\|`, spaces) is rendered verbatim — you control all spacing.
+- Each variable renders its core value only (no `on`/`via` prefixes); add those words as literal text.
+- Conditional groups: wrap optional pieces in parentheses, e.g. `$cwd( on $git_branch)($git_status)$fill($context)`. If every `$var` inside a group is empty, the whole group (including its literals) is dropped.
+- Unknown `$variables` render empty.
+- Set or clear at runtime: `/zentui format "<template>"` and `/zentui format clear`.
+
 ## Requirements
 
 - [Pi](https://pi.dev) coding agent 0.79 or newer
-- A [Nerd Font](https://www.nerdfonts.com/) for icons
+- A [Nerd Font](https://www.nerdfonts.com/) for icons (or set `icons.mode` to `"ascii"`)
 
 ## Development
 

package/extensions/zentui/config.ts

@@ -1,10 +1,27 @@
 import { existsSync, readFileSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
 import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import {
+	ICON_GLYPH_KEYS,
+	type IconGlyphs,
+	type IconMode,
+	NERD_DEFAULT_ICONS,
+	normalizeIconMode,
+	type ResolvedIcons,
+	resolveConfiguredIcons,
+} from "./icons";
 import { isSupportedColorSpec } from "./style";
 
 export type ColorSpec = string;
 export type ColorSource = "theme" | "terminal";
+export type { IconMode } from "./icons";
+
+export type ContextStyle = "text" | "gauge" | "text+gauge";
+
+export type ContextThresholds = {
+	warning: number;
+	error: number;
+};
 
 export type ColorSourcesConfig = {
 	starship: ColorSource;
@@ -28,6 +45,9 @@ export type FooterSegmentsConfig = {
 	tokens: boolean;
 	cost: boolean;
 	sessionDuration: boolean;
+	username: boolean;
+	time: boolean;
+	os: boolean;
 };
 
 export type ExtensionStatusPlacement = "off" | "left" | "middle" | "right";
@@ -46,24 +66,10 @@ const MIN_PROJECT_REFRESH_INTERVAL_MS = 5_000;
 
 export type PolishedTuiConfig = {
 	projectRefreshIntervalMs: number;
-	icons: {
-		cwd: string;
-		git: string;
-		ahead: string;
-		behind: string;
-		diverged: string;
-		conflicted: string;
-		untracked: string;
-		stashed: string;
-		modified: string;
-		staged: string;
-		renamed: string;
-		deleted: string;
-		typechanged: string;
-		cacheHit: string;
-		editorPrompt: string;
-		rail: string;
-	};
+	footerFormat: string;
+	contextStyle: ContextStyle;
+	contextThresholds: ContextThresholds;
+	icons: ResolvedIcons;
 	colors: {
 		cwd: ColorSpec;
 		gitBranch: ColorSpec;
@@ -77,6 +83,9 @@ export type PolishedTuiConfig = {
 		runtimePrefix: ColorSpec;
 		extensionStatus: ColorSpec;
 		sessionDuration: ColorSpec;
+		username: ColorSpec;
+		time: ColorSpec;
+		os: ColorSpec;
 		editorAccent?: ColorSpec;
 		editorPrompt?: ColorSpec;
 		editorBorder?: ColorSpec;
@@ -95,27 +104,49 @@ export type PolishedTuiConfig = {
 	extensionStatuses: ExtensionStatusesConfig;
 };
 
+/**
+ * Canonical footer format variable names. In a `footerFormat` string these
+ * are written as `$name` or `${name}`.
+ */
+export const FOOTER_FORMAT_VARIABLES = [
+	"cwd",
+	"git_branch",
+	"git_status",
+	"git_state",
+	"runtime",
+	"session_duration",
+	"username",
+	"os",
+	"time",
+	"context",
+	"tokens",
+	"cost",
+	"sep",
+] as const;
+
+/**
+ * Alias → canonical variable name mapping for `footerFormat`.
+ * `$fill` is special (not a variable) and handled by the parser.
+ */
+export const FOOTER_FORMAT_ALIASES: Record<string, string> = {
+	directory: "cwd",
+	branch: "git_branch",
+	status: "git_status",
+	state: "git_state",
+	duration: "session_duration",
+	separator: "sep",
+};
+
 export const configPath = join(getAgentDir(), "zentui.json");
 
 export const defaultConfig: PolishedTuiConfig = {
 	projectRefreshIntervalMs: DEFAULT_PROJECT_REFRESH_INTERVAL_MS,
+	footerFormat: "",
+	contextStyle: "text",
+	contextThresholds: { warning: 70, error: 90 },
 	icons: {
-		cwd: "󰝰",
-		git: "",
-		ahead: "↑",
-		behind: "↓",
-		diverged: "⇕",
-		conflicted: "=",
-		untracked: "?",
-		stashed: "$",
-		modified: "!",
-		staged: "+",
-		renamed: "»",
-		deleted: "✘",
-		typechanged: "T",
-		cacheHit: "󰆼",
-		editorPrompt: "",
-		rail: "│",
+		mode: "auto",
+		...NERD_DEFAULT_ICONS,
 	},
 	colors: {
 		cwd: "bold cyan",
@@ -130,6 +161,9 @@ export const defaultConfig: PolishedTuiConfig = {
 		runtimePrefix: "",
 		extensionStatus: "bright-black",
 		sessionDuration: "yellow",
+		username: "bold yellow",
+		time: "bold yellow",
+		os: "bold white",
 	},
 	colorSources: {
 		starship: "theme",
@@ -151,6 +185,9 @@ export const defaultConfig: PolishedTuiConfig = {
 		tokens: true,
 		cost: true,
 		sessionDuration: false,
+		username: false,
+		time: false,
+		os: false,
 	},
 	extensionStatuses: {
 		defaultPlacement: "right",
@@ -159,24 +196,6 @@ export const defaultConfig: PolishedTuiConfig = {
 	},
 };
 
-const iconKeys = [
-	"cwd",
-	"git",
-	"ahead",
-	"behind",
-	"diverged",
-	"conflicted",
-	"untracked",
-	"stashed",
-	"modified",
-	"staged",
-	"renamed",
-	"deleted",
-	"typechanged",
-	"cacheHit",
-	"editorPrompt",
-] as const satisfies readonly (keyof PolishedTuiConfig["icons"])[];
-
 type ConfigRecord = Record<string, unknown>;
 
 function isRecord(value: unknown): value is ConfigRecord {
@@ -186,17 +205,43 @@ function isRecord(value: unknown): value is ConfigRecord {
 function parseProjectRefreshIntervalMs(value: unknown): number {
 	if (value === 0) return 0;
 	if (typeof value !== "number" || !Number.isFinite(value)) {
-		return defaultConfig.projectRefreshIntervalMs;
+		return DEFAULT_PROJECT_REFRESH_INTERVAL_MS;
 	}
 
 	const interval = Math.round(value);
-	return interval >= MIN_PROJECT_REFRESH_INTERVAL_MS
-		? interval
-		: defaultConfig.projectRefreshIntervalMs;
+	if (interval <= 0) return 0;
+	return Math.max(MIN_PROJECT_REFRESH_INTERVAL_MS, interval);
+}
+
+function clampPercent(value: number): number {
+	return Math.max(0, Math.min(100, value));
+}
+
+function parseContextStyle(value: unknown): ContextStyle {
+	if (value === "text" || value === "gauge" || value === "text+gauge") return value;
+	return defaultConfig.contextStyle;
 }
 
-function railValue(value: unknown): string {
-	return typeof value === "string" && value.trim().length > 0 ? value : defaultConfig.icons.rail;
+function parseContextThresholds(value: unknown): ContextThresholds {
+	const defaults = defaultConfig.contextThresholds;
+	if (!isRecord(value)) return { ...defaults };
+
+	const warningRaw = value.warning;
+	const errorRaw = value.error;
+	let warning =
+		typeof warningRaw === "number" && Number.isFinite(warningRaw)
+			? clampPercent(Math.round(warningRaw))
+			: defaults.warning;
+	let error =
+		typeof errorRaw === "number" && Number.isFinite(errorRaw)
+			? clampPercent(Math.round(errorRaw))
+			: defaults.error;
+	if (error < warning) {
+		const swapped = warning;
+		warning = error;
+		error = swapped;
+	}
+	return { warning, error };
 }
 
 function stringValue(record: Record<string, unknown>, key: string): string | undefined {
@@ -240,13 +285,13 @@ function definedColors(
 	) as Partial<PolishedTuiConfig["colors"]>;
 }
 
-function normalizeIcons(record: Record<string, unknown>): Partial<PolishedTuiConfig["icons"]> {
+function normalizeIconOverrides(record: Record<string, unknown>): Partial<IconGlyphs> {
 	return Object.fromEntries(
-		iconKeys.flatMap((key) => {
+		ICON_GLYPH_KEYS.flatMap((key) => {
 			const value = stringValue(record, key);
 			return value === undefined ? [] : [[key, value]];
 		}),
-	) as Partial<PolishedTuiConfig["icons"]>;
+	) as Partial<IconGlyphs>;
 }
 
 function normalizeColors(record: Record<string, unknown>): Partial<PolishedTuiConfig["colors"]> {
@@ -263,6 +308,9 @@ function normalizeColors(record: Record<string, unknown>): Partial<PolishedTuiCo
 		runtimePrefix: colorValue(record, "runtimePrefix"),
 		extensionStatus: colorValue(record, "extensionStatus"),
 		sessionDuration: colorValue(record, "sessionDuration"),
+		username: colorValue(record, "username"),
+		time: colorValue(record, "time"),
+		os: colorValue(record, "os"),
 		editorAccent: colorValue(record, "editorAccent"),
 		editorPrompt: colorValue(record, "editorPrompt"),
 		editorBorder: colorValue(record, "editorBorder"),
@@ -304,6 +352,9 @@ function normalizeFooterSegments(record: Record<string, unknown>): FooterSegment
 		tokens: footerSegmentValue(record, "tokens"),
 		cost: footerSegmentValue(record, "cost"),
 		sessionDuration: footerSegmentValue(record, "sessionDuration"),
+		username: footerSegmentValue(record, "username"),
+		time: footerSegmentValue(record, "time"),
+		os: footerSegmentValue(record, "os"),
 	};
 }
 
@@ -361,7 +412,10 @@ function isFooterSegmentKey(value: string): value is keyof FooterSegmentsConfig
 		value === "context" ||
 		value === "tokens" ||
 		value === "cost" ||
-		value === "sessionDuration"
+		value === "sessionDuration" ||
+		value === "username" ||
+		value === "time" ||
+		value === "os"
 	);
 }
 
@@ -412,7 +466,8 @@ export function ensureConfigExists(): void {
 export function mergeConfig(parsed: unknown): PolishedTuiConfig {
 	const config = isRecord(parsed) ? parsed : {};
 	const iconsRecord = isRecord(config.icons) ? (config.icons as Record<string, unknown>) : {};
-	const icons = normalizeIcons(iconsRecord);
+	const iconMode = normalizeIconMode(iconsRecord.mode);
+	const iconOverrides = normalizeIconOverrides(iconsRecord);
 	const colors = isRecord(config.colors)
 		? normalizeColors(config.colors as Record<string, unknown>)
 		: {};
@@ -430,11 +485,10 @@ export function mergeConfig(parsed: unknown): PolishedTuiConfig {
 		: defaultConfig.extensionStatuses;
 	return {
 		projectRefreshIntervalMs: parseProjectRefreshIntervalMs(config.projectRefreshIntervalMs),
-		icons: {
-			...defaultConfig.icons,
-			...icons,
-			rail: railValue(iconsRecord.rail),
-		},
+		footerFormat: stringValue(config, "footerFormat") ?? "",
+		contextStyle: parseContextStyle(config.contextStyle),
+		contextThresholds: parseContextThresholds(config.contextThresholds),
+		icons: resolveConfiguredIcons(iconMode, iconOverrides),
 		colors: {
 			...defaultConfig.colors,
 			...colors,
@@ -521,6 +575,47 @@ export function saveFooterSegmentsPatch(
 	return mergeConfig(record);
 }
 
+export function saveFooterFormatPatch(value: string, path = configPath): PolishedTuiConfig {
+	const record = readConfigRecord(path);
+	record.footerFormat = typeof value === "string" ? value : "";
+	writeFileSync(path, `${JSON.stringify(record, null, 2)}\n`, "utf8");
+	return mergeConfig(record);
+}
+
+export function saveIconsModePatch(mode: IconMode, path = configPath): PolishedTuiConfig {
+	const record = readConfigRecord(path);
+	const existing = isRecord(record.icons) ? { ...(record.icons as Record<string, unknown>) } : {};
+	record.icons = {
+		...existing,
+		mode: normalizeIconMode(mode),
+	};
+	writeFileSync(path, `${JSON.stringify(record, null, 2)}\n`, "utf8");
+	return mergeConfig(record);
+}
+
+export function saveContextStylePatch(style: ContextStyle, path = configPath): PolishedTuiConfig {
+	const record = readConfigRecord(path);
+	record.contextStyle = parseContextStyle(style);
+	writeFileSync(path, `${JSON.stringify(record, null, 2)}\n`, "utf8");
+	return mergeConfig(record);
+}
+
+export function saveContextThresholdsPatch(
+	thresholds: Partial<ContextThresholds>,
+	path = configPath,
+): PolishedTuiConfig {
+	const record = readConfigRecord(path);
+	const existing = isRecord(record.contextThresholds)
+		? { ...(record.contextThresholds as Record<string, unknown>) }
+		: {};
+	record.contextThresholds = {
+		...existing,
+		...thresholds,
+	};
+	writeFileSync(path, `${JSON.stringify(record, null, 2)}\n`, "utf8");
+	return mergeConfig(record);
+}
+
 export function saveExtensionStatusPlacement(
 	key: string,
 	placement: ExtensionStatusPlacement,