pi-zentui 0.1.2 → 0.1.4

package/README.md

@@ -84,9 +84,10 @@ On first run, Zentui creates a config file at:
 
 ```json
 {
+  "projectRefreshIntervalMs": 30000,
   "icons": {
     "cwd": "󰝰",
-    "git": "",
+    "git": "",
     "ahead": "↑",
     "behind": "↓",
     "diverged": "⇕",
@@ -113,6 +114,8 @@ On first run, Zentui creates a config file at:
 }
 ```
 
+`projectRefreshIntervalMs` controls how often Zentui refreshes project status (git/runtime) while Pi is idle. Set it to `0` to disable polling; invalid values or values below 5000 ms fall back to `30000`.
+
 ### Color values
 
 Colors can be:
@@ -129,23 +132,28 @@ This means Zentui works with any Pi theme — it uses your theme's colors by def
 
 ## Development
 
-If you use [mise](https://mise.jdx.dev/):
+```bash
+npm install
+npm run verify
+npm run fmt
+npm run pack:check
+```
+
+### Test in Pi
+
+The project keeps Pi core packages as peer dependencies for runtime and dev dependencies for
+typechecking. To avoid accidentally running the local `node_modules/.bin/pi` shim, the dev scripts use
+the globally installed Pi binary by default:
 
 ```bash
-mise install
-mise run setup
-mise run verify
-mise run fmt
-mise run ci
+npm run pi:dev
+npm run pi:install-local
 ```
 
-Without mise:
+Override the binary if your Pi install is somewhere else:
 
 ```bash
-npm install
-npm run verify
-npm run fmt
-npm run pack:check
+PI_BIN=/path/to/pi npm run pi:dev
 ```
 
 ## Credits

package/extensions/zentui/config.ts

@@ -4,7 +4,11 @@ import { getAgentDir } from "@mariozechner/pi-coding-agent";
 
 export type ColorSpec = string;
 
+const DEFAULT_PROJECT_REFRESH_INTERVAL_MS = 30_000;
+const MIN_PROJECT_REFRESH_INTERVAL_MS = 5_000;
+
 export type PolishedTuiConfig = {
+	projectRefreshIntervalMs: number;
 	icons: {
 		cwd: string;
 		git: string;
@@ -84,6 +88,7 @@ const themeColorTokens = new Set([
 ]);
 
 export const defaultConfig: PolishedTuiConfig = {
+	projectRefreshIntervalMs: DEFAULT_PROJECT_REFRESH_INTERVAL_MS,
 	icons: {
 		cwd: "󰝰",
 		git: "",
@@ -128,6 +133,24 @@ type ThemeLike = {
 	fg(color: string, text: string): string;
 };
 
+type ConfigRecord = Record<string, unknown>;
+
+function isRecord(value: unknown): value is ConfigRecord {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+function parseProjectRefreshIntervalMs(value: unknown): number {
+	if (value === 0) return 0;
+	if (typeof value !== "number" || !Number.isFinite(value)) {
+		return defaultConfig.projectRefreshIntervalMs;
+	}
+
+	const interval = Math.round(value);
+	return interval >= MIN_PROJECT_REFRESH_INTERVAL_MS
+		? interval
+		: defaultConfig.projectRefreshIntervalMs;
+}
+
 export function colorize(theme: ThemeLike, color: ColorSpec, text: string): string {
 	if (themeColorTokens.has(color)) {
 		return theme.fg(color, text);
@@ -148,20 +171,29 @@ export function ensureConfigExists(): void {
 	}
 }
 
+export function mergeConfig(parsed: unknown): PolishedTuiConfig {
+	const config = isRecord(parsed) ? parsed : {};
+	const icons = isRecord(config.icons) ? (config.icons as Partial<PolishedTuiConfig["icons"]>) : {};
+	const colors = isRecord(config.colors)
+		? (config.colors as Partial<PolishedTuiConfig["colors"]>)
+		: {};
+	return {
+		projectRefreshIntervalMs: parseProjectRefreshIntervalMs(config.projectRefreshIntervalMs),
+		icons: {
+			...defaultConfig.icons,
+			...icons,
+		},
+		colors: {
+			...defaultConfig.colors,
+			...colors,
+		},
+	};
+}
+
 export function loadConfig(): PolishedTuiConfig {
 	try {
 		if (!existsSync(configPath)) return defaultConfig;
-		const parsed = JSON.parse(readFileSync(configPath, "utf8")) as Partial<PolishedTuiConfig>;
-		return {
-			icons: {
-				...defaultConfig.icons,
-				...(parsed.icons ?? {}),
-			},
-			colors: {
-				...defaultConfig.colors,
-				...(parsed.colors ?? {}),
-			},
-		};
+		return mergeConfig(JSON.parse(readFileSync(configPath, "utf8")));
 	} catch {
 		return defaultConfig;
 	}

package/extensions/zentui/index.ts

@@ -8,11 +8,11 @@ import type {
 import { type EditorTheme, type TUI, truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
 import { type PolishedTuiConfig, colorize, ensureConfigExists, loadConfig } from "./config";
 import { type GitStatusSummary, emptyGitStatus, readGitStatus } from "./git";
+import { type StopProjectRefreshInterval, startProjectRefreshInterval } from "./project-refresh";
 import { type RuntimeInfo, readRuntimeInfo } from "./runtime";
-import { PolishedEditor, patchUserMessageComponent } from "./ui";
+import { PolishedEditor, patchUserMessageComponent, restoreUserMessageComponent } from "./ui";
 
 type FooterState = GitStatusSummary & {
-	busy: boolean;
 	modelLabel: string;
 	providerLabel: string;
 	contextLabel: string;
@@ -128,7 +128,6 @@ function formatCwdLabel(cwd: string, cwdIcon: string): string {
 
 export default function (pi: ExtensionAPI) {
 	const state: FooterState = {
-		busy: false,
 		modelLabel: "no-model",
 		providerLabel: "Unknown",
 		contextLabel: "--",
@@ -140,6 +139,7 @@ export default function (pi: ExtensionAPI) {
 
 	let currentConfig: PolishedTuiConfig = loadConfig();
 	let requestFooterRender: (() => void) | undefined;
+	let stopProjectRefreshInterval: StopProjectRefreshInterval = () => {};
 	let projectRefreshInFlight = false;
 	let projectRefreshPending = false;
 
@@ -267,15 +267,8 @@ export default function (pi: ExtensionAPI) {
 	const installEditor = (ctx: ExtensionContext) => {
 		syncState(ctx);
 
-		let currentEditor: PolishedEditor | undefined;
-		let autocompleteFixed = false;
-
-		type AutocompleteEditorInternals = {
-			autocompleteProvider?: unknown;
-		};
-
-		const editorFactory = (tui: TUI, theme: EditorTheme, keybindings: KeybindingsManager) => {
-			const editor = new PolishedEditor(
+		const editorFactory = (tui: TUI, theme: EditorTheme, keybindings: KeybindingsManager) =>
+			new PolishedEditor(
 				tui,
 				theme,
 				keybindings,
@@ -287,22 +280,6 @@ export default function (pi: ExtensionAPI) {
 					].join(ctx.ui.theme.fg("borderMuted", "  ")),
 				() => pi.getThinkingLevel(),
 			);
-			currentEditor = editor;
-
-			const originalHandleInput = editor.handleInput.bind(editor);
-			editor.handleInput = (data: string) => {
-				const editorInternals = editor as unknown as AutocompleteEditorInternals;
-				if (!autocompleteFixed && !editorInternals.autocompleteProvider) {
-					autocompleteFixed = true;
-					ctx.ui.setEditorComponent(editorFactory);
-					currentEditor?.handleInput(data);
-					return;
-				}
-				originalHandleInput(data);
-			};
-
-			return editor;
-		};
 
 		ctx.ui.setEditorComponent(editorFactory);
 	};
@@ -313,6 +290,11 @@ export default function (pi: ExtensionAPI) {
 		patchUserMessageComponent(ctx.ui.theme);
 		installFooter(ctx);
 		installEditor(ctx);
+		stopProjectRefreshInterval();
+		stopProjectRefreshInterval = startProjectRefreshInterval(
+			currentConfig.projectRefreshIntervalMs,
+			() => scheduleProjectRefresh(ctx),
+		);
 		scheduleProjectRefresh(ctx);
 		refresh();
 	};
@@ -321,14 +303,18 @@ export default function (pi: ExtensionAPI) {
 		installUi(ctx);
 	});
 
+	pi.on("session_shutdown", async () => {
+		stopProjectRefreshInterval();
+		stopProjectRefreshInterval = () => {};
+		restoreUserMessageComponent();
+	});
+
 	pi.on("agent_start", async (_event, ctx) => {
-		state.busy = true;
 		syncState(ctx);
 		refresh();
 	});
 
 	pi.on("agent_end", async (_event, ctx) => {
-		state.busy = false;
 		syncState(ctx);
 		scheduleProjectRefresh(ctx);
 		refresh();

package/extensions/zentui/project-refresh.ts

@@ -0,0 +1,13 @@
+export type StopProjectRefreshInterval = () => void;
+
+export function startProjectRefreshInterval(
+	intervalMs: number,
+	refresh: () => void,
+): StopProjectRefreshInterval {
+	if (intervalMs <= 0) return () => {};
+
+	const timer = setInterval(refresh, intervalMs);
+	timer.unref?.();
+
+	return () => clearInterval(timer);
+}

package/extensions/zentui/ui.ts

@@ -16,13 +16,22 @@ import {
 const OSC133_ZONE_START = "\x1b]133;A\x07";
 const OSC133_ZONE_END = "\x1b]133;B\x07";
 const OSC133_ZONE_FINAL = "\x1b]133;C\x07";
-const originalUserMessageRender = UserMessageComponent.prototype.render;
+const userMessagePatchKey = "__zentuiUserMessagePatch";
 
 type AutocompleteEditorInternals = {
 	autocompleteList?: Pick<Component, "render">;
 	isShowingAutocomplete?: () => boolean;
 };
 
+type UserMessageRender = (this: UserMessageComponent, width: number) => string[];
+type UserMessagePatchState = {
+	originalRender: UserMessageRender;
+	patchedRender: UserMessageRender;
+};
+type PatchableUserMessagePrototype = typeof UserMessageComponent.prototype & {
+	[userMessagePatchKey]?: UserMessagePatchState;
+};
+
 let currentUiTheme: Theme | undefined;
 
 const TRUECOLOR_BACKGROUND_ANSI = /\x1b\[48;2;\d+;\d+;\d+m/g;
@@ -36,51 +45,74 @@ function stripBackgroundAnsi(text: string): string {
 		.replace(SIMPLE_BACKGROUND_ANSI, "");
 }
 
-function fillStyledLine(
-	content: string,
-	width: number,
-	background?: (text: string) => string,
-): string {
+function fillStyledLine(content: string, width: number): string {
 	const truncated = truncateToWidth(stripBackgroundAnsi(content), width, "");
 	const padWidth = Math.max(0, width - visibleWidth(truncated));
-	const pad =
-		padWidth > 0 ? (background ? background(" ".repeat(padWidth)) : " ".repeat(padWidth)) : "";
+	const pad = padWidth > 0 ? " ".repeat(padWidth) : "";
 	return `${truncated}${pad}`;
 }
 
+function userMessagePrototype(): PatchableUserMessagePrototype {
+	return UserMessageComponent.prototype as PatchableUserMessagePrototype;
+}
+
+function renderPatchedUserMessage(this: UserMessageComponent, width: number): string[] {
+	const originalRender = userMessagePrototype()[userMessagePatchKey]?.originalRender;
+	if (!currentUiTheme || !originalRender) {
+		return originalRender
+			? originalRender.call(this, width)
+			: (Container.prototype.render.call(this, width) as string[]);
+	}
+
+	const railWidth = 2;
+	const innerWidth = Math.max(1, width - railWidth);
+	const baseLines = Container.prototype.render.call(this, innerWidth) as string[];
+	if (baseLines.length === 0) return baseLines;
+
+	const hasLeadingSpacer = baseLines.length > 1 && visibleWidth(baseLines[0] ?? "") === 0;
+	const leadingLines = hasLeadingSpacer ? [baseLines[0] ?? ""] : [];
+	const contentLines = hasLeadingSpacer ? baseLines.slice(1) : baseLines;
+	const rail = `${currentUiTheme.fg("accent", "│")}\x1b[0m `;
+	const border = currentUiTheme.fg("border", "─".repeat(width));
+	const styledLines = contentLines.map((line) => `${rail}${fillStyledLine(line, innerWidth)}`);
+
+	if (styledLines.length === 0) {
+		return leadingLines;
+	}
+
+	const framedLines = [border, ...styledLines, border];
+	framedLines[0] = OSC133_ZONE_START + framedLines[0];
+	framedLines[framedLines.length - 1] =
+		framedLines[framedLines.length - 1] + OSC133_ZONE_END + OSC133_ZONE_FINAL;
+	return [...leadingLines, ...framedLines];
+}
+
 export function patchUserMessageComponent(uiTheme: Theme): void {
 	currentUiTheme = uiTheme;
 
-	const prototype = UserMessageComponent.prototype as {
-		render(width: number): string[];
+	const prototype = userMessagePrototype();
+	const patchState = prototype[userMessagePatchKey] ?? {
+		originalRender: prototype.render,
+		patchedRender: renderPatchedUserMessage,
 	};
-	prototype.render = function (this: UserMessageComponent, width: number): string[] {
-		if (!currentUiTheme) {
-			return originalUserMessageRender.call(this, width);
-		}
+	patchState.patchedRender = renderPatchedUserMessage;
 
-		const railWidth = 2;
-		const innerWidth = Math.max(1, width - railWidth);
-		const baseLines = Container.prototype.render.call(this, innerWidth) as string[];
-		if (baseLines.length === 0) return baseLines;
+	Object.defineProperty(prototype, userMessagePatchKey, {
+		value: patchState,
+		configurable: true,
+	});
+	prototype.render = renderPatchedUserMessage;
+}
 
-		const hasLeadingSpacer = baseLines.length > 1 && visibleWidth(baseLines[0] ?? "") === 0;
-		const leadingLines = hasLeadingSpacer ? [baseLines[0] ?? ""] : [];
-		const contentLines = hasLeadingSpacer ? baseLines.slice(1) : baseLines;
-		const rail = `${currentUiTheme.fg("accent", "│")}\x1b[0m `;
-		const border = currentUiTheme.fg("border", "─".repeat(width));
-		const styledLines = contentLines.map((line) => `${rail}${fillStyledLine(line, innerWidth)}`);
+export function restoreUserMessageComponent(): void {
+	currentUiTheme = undefined;
 
-		if (styledLines.length === 0) {
-			return leadingLines;
-		}
+	const prototype = userMessagePrototype();
+	const patchState = prototype[userMessagePatchKey];
+	if (!patchState || prototype.render !== patchState.patchedRender) return;
 
-		const framedLines = [border, ...styledLines, border];
-		framedLines[0] = OSC133_ZONE_START + framedLines[0];
-		framedLines[framedLines.length - 1] =
-			framedLines[framedLines.length - 1] + OSC133_ZONE_END + OSC133_ZONE_FINAL;
-		return [...leadingLines, ...framedLines];
-	};
+	prototype.render = patchState.originalRender;
+	delete prototype[userMessagePatchKey];
 }
 
 export class PolishedEditor extends CustomEditor {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-zentui",
-	"version": "0.1.2",
+	"version": "0.1.4",
 	"description": "A Starship-inspired statusline and Opencode-style TUI for Pi.",
 	"type": "module",
 	"license": "MIT",
@@ -18,6 +18,8 @@
 		"lint": "biome check .",
 		"typecheck": "tsc --noEmit",
 		"test": "vitest run",
+		"pi:dev": "${PI_BIN:-/opt/homebrew/bin/pi} --no-extensions -e ./extensions/zentui/index.ts",
+		"pi:install-local": "${PI_BIN:-/opt/homebrew/bin/pi} install ./ -l",
 		"fmt": "biome format --write .",
 		"fmt:check": "biome format --check .",
 		"verify": "npm run lint && npm run typecheck && npm run test",
@@ -38,9 +40,9 @@
 	},
 	"devDependencies": {
 		"@biomejs/biome": "^1.9.4",
-		"@mariozechner/pi-ai": "^0.65.2",
-		"@mariozechner/pi-coding-agent": "^0.65.2",
-		"@mariozechner/pi-tui": "^0.65.2",
+		"@mariozechner/pi-ai": "^0.73.0",
+		"@mariozechner/pi-coding-agent": "^0.73.0",
+		"@mariozechner/pi-tui": "^0.73.0",
 		"@types/node": "^25.5.2",
 		"typescript": "^6.0.2",
 		"vitest": "^3.2.4"