@janvitos/pi-plan-build 0.1.2 → 0.1.4

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.
-- Fixed-width, bold, color-coded **Build** (blue) and **Plan** (yellow) status badges 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 {
@@ -64,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", {
@@ -94,8 +95,10 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 		pi.appendEntry(STATE_TYPE, stateData());
 	}
 
-	function updateStatus(ctx: ExtensionContext): void {
-		ctx.ui.setStatus(STATUS_KEY, formatModeStatus(selectedMode));
+	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 {
@@ -134,7 +137,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 			pendingReminder = undefined;
 			applyTools(mode);
 		}
-		updateStatus(ctx);
+		updateModeIndicator(ctx);
 		persist();
 	}
 
@@ -360,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) => {
@@ -383,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, this.borderColor);
+					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?.();
@@ -398,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");
 				};
@@ -409,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.2",
+  "version": "0.1.4",
   "description": "Plan safely, approve explicitly, then implement here or in a clean session.",
   "type": "module",
   "license": "MIT",

package/utils.ts

@@ -3,17 +3,19 @@ import path from "node:path";
 export type Mode = "build" | "plan";
 
 const ANSI_RESET = "\x1b[0m";
-const BOLD_BLACK_FOREGROUND = "1;38;2;0;0;0";
-const MODE_BADGES: Record<Mode, { background: string; text: string; reservedSpace: string }> = {
-	// A trailing reset keeps Pi's status sanitizer from trimming the uncolored reserved cell.
-	plan: { background: "48;2;255;215;0", text: " PLAN ", reservedSpace: "\u00a0" },
-	build: { background: "48;2;59;130;246", text: " BUILD ", reservedSpace: "" },
+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 function formatModeStatus(mode: Mode): string {
-	const badge = MODE_BADGES[mode];
-	const reservedSuffix = badge.reservedSpace ? `${badge.reservedSpace}${ANSI_RESET}` : "";
-	return `\x1b[${BOLD_BLACK_FOREGROUND};${badge.background}m${badge.text}${ANSI_RESET}${reservedSuffix}`;
+export interface ModeStatusTheme {
+	bold(text: string): string;
+}
+
+export function formatModeStatus(mode: Mode, theme: ModeStatusTheme, borderColor: (text: string) => string): string {
+	const label = MODE_LABELS[mode];
+	const modeText = `\x1b[${label.color}m${theme.bold(label.text)}${ANSI_RESET}`;
+	return borderColor("[") + modeText + borderColor("]");
 }
 
 export const PLAN_EXIT_APPROVE_CHOICE = "Switch to Build and implement here";