@janvitos/pi-plan-build 0.1.1 → 0.1.3

package/README.md

@@ -8,7 +8,7 @@ A global [Pi coding agent](https://github.com/badlogic/pi-mono) extension that a
 
 - New sessions start in **Build** mode.
 - Bare `Tab` cycles **Build → Plan → Build** and replaces normal Tab autocomplete.
-- `[Build]` and `[Plan]` status indicators in Pi's footer.
+- Bracketed **build** (blue) and **plan** (yellow) indicators rendered as bold prompt prefixes inside Pi's text input box.
 - `/plan`, `/build`, and the `--plan` startup flag.
 - Per-session plans at `~/.pi/agent/plans/<session-id>.md`.
 - In Plan mode, built-in `edit` and `write` are restricted to the exact plan file.

package/index.ts

@@ -2,7 +2,7 @@ import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
 import { CustomEditor, getAgentDir, getMarkdownTheme, type EntryRenderer, type ExtensionAPI, type ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { Key, Markdown, matchesKey, Text } from "@earendil-works/pi-tui";
+import { Key, Markdown, matchesKey, Text, visibleWidth } from "@earendil-works/pi-tui";
 import { Type } from "typebox";
 import { registerQuestionTool } from "./question-ui.ts";
 import {
@@ -19,6 +19,7 @@ import {
 	buildPlanReviewMessage,
 	classifyPlanExitChoice,
 	decodeModeState,
+	formatModeStatus,
 	isAllowedPlanMutation,
 	makePlanPath,
 	nextMode,
@@ -63,6 +64,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 	let planPath = "";
 	let toolsBeforeModes: string[] = [];
 	let currentContext: ExtensionContext | undefined;
+	let requestEditorRender: (() => void) | undefined;
 	let freshImplementationPlan: string | undefined;
 
 	pi.registerFlag("plan", {
@@ -93,11 +95,10 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 		pi.appendEntry(STATE_TYPE, stateData());
 	}
 
-	function updateStatus(ctx: ExtensionContext): void {
-		const label = selectedMode === "plan"
-			? ctx.ui.theme.fg("warning", "Plan")
-			: ctx.ui.theme.fg("border", "Build");
-		ctx.ui.setStatus(STATUS_KEY, `[${label}]`);
+	function updateModeIndicator(ctx: ExtensionContext): void {
+		// Clear the legacy footer status; the mode is now rendered inside the editor.
+		ctx.ui.setStatus(STATUS_KEY, undefined);
+		requestEditorRender?.();
 	}
 
 	function discoverUnmanagedTools(): void {
@@ -136,7 +137,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 			pendingReminder = undefined;
 			applyTools(mode);
 		}
-		updateStatus(ctx);
+		updateModeIndicator(ctx);
 		persist();
 	}
 
@@ -362,7 +363,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 	pi.on("agent_settled", async (_event, ctx) => {
 		runMode = undefined;
 		applyTools(selectedMode);
-		updateStatus(ctx);
+		updateModeIndicator(ctx);
 	});
 
 	pi.on("session_start", async (_event, ctx) => {
@@ -385,11 +386,31 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 		planPath = makePlanPath(path.join(getAgentDir(), "plans"), ctx.sessionManager.getSessionId());
 		if (selectedMode === "plan") await ensurePlanDirectory();
 		applyTools(selectedMode);
-		updateStatus(ctx);
+		updateModeIndicator(ctx);
 
 		if (ctx.mode === "tui") {
 			class ModeEditor extends CustomEditor {
 				onCycle?: () => void;
+
+				requestModeRender(): void {
+					this.tui.requestRender();
+				}
+
+				override render(width: number): string[] {
+					const badge = formatModeStatus(selectedMode, ctx.ui.theme);
+					const prompt = `${badge} `;
+					const promptWidth = visibleWidth(prompt);
+					const paddingWidth = Math.min(promptWidth, Math.max(0, Math.floor((width - 1) / 2)));
+					if (this.getPaddingX() !== paddingWidth) this.setPaddingX(paddingWidth);
+
+					const lines = super.render(width);
+					const reservedPrefix = " ".repeat(paddingWidth);
+					if (paddingWidth === promptWidth && lines[1]?.startsWith(reservedPrefix)) {
+						lines[1] = prompt + lines[1].slice(reservedPrefix.length);
+					}
+					return lines;
+				}
+
 				override handleInput(data: string): void {
 					if (matchesKey(data, Key.tab)) {
 						this.onCycle?.();
@@ -400,6 +421,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 			}
 			ctx.ui.setEditorComponent((tui, theme, keybindings) => {
 				const editor = new ModeEditor(tui, theme, keybindings);
+				requestEditorRender = () => editor.requestModeRender();
 				editor.onCycle = () => {
 					if (currentContext) void selectMode(nextMode(selectedMode), currentContext, "manual");
 				};
@@ -411,6 +433,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 	pi.on("session_shutdown", async (_event, ctx) => {
 		ctx.ui.setStatus(STATUS_KEY, undefined);
 		ctx.ui.setEditorComponent(undefined);
+		requestEditorRender = undefined;
 		currentContext = undefined;
 	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@janvitos/pi-plan-build",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Plan safely, approve explicitly, then implement here or in a clean session.",
   "type": "module",
   "license": "MIT",

package/utils.ts

@@ -2,6 +2,23 @@ import path from "node:path";
 
 export type Mode = "build" | "plan";
 
+const ANSI_RESET = "\x1b[0m";
+const MODE_LABELS: Record<Mode, { color: string; text: string }> = {
+	plan: { color: "38;2;255;215;0", text: "plan" },
+	build: { color: "38;2;59;130;246", text: "build" },
+};
+
+export interface ModeStatusTheme {
+	fg(color: "dim", text: string): string;
+	bold(text: string): string;
+}
+
+export function formatModeStatus(mode: Mode, theme: ModeStatusTheme): string {
+	const label = MODE_LABELS[mode];
+	const modeText = `\x1b[${label.color}m${theme.bold(label.text)}${ANSI_RESET}`;
+	return theme.fg("dim", "[") + modeText + theme.fg("dim", "]");
+}
+
 export const PLAN_EXIT_APPROVE_CHOICE = "Switch to Build and implement here";
 export const PLAN_EXIT_FRESH_CHOICE = "Start fresh and implement";
 export const PLAN_EXIT_STAY_CHOICE = "Stay in Plan mode";