@janvitos/pi-plan-build 0.1.5 → 0.1.7

package/README.md

@@ -21,6 +21,7 @@ A global [Pi coding agent](https://github.com/badlogic/pi-mono) extension that a
   - **Stay in Plan mode**
 - Staying in Plan mode produces a durable acknowledgement and stops the run until the user responds.
 - Mode state survives reloads, resumes, and forks.
+- When Pi recreates the custom editor, the latest 100 user prompts from the active session branch are restored for Up/Down history navigation.
 
 ## Requirements
 
@@ -110,7 +111,7 @@ npm test
 npm pack --dry-run
 ```
 
-The tests cover state decoding, safe plan paths, mutation restrictions, deferred transitions, complete plan rendering, approval decisions, stop behavior, fresh-session handoff content, and question formatting.
+The tests cover state decoding, safe plan paths, mutation restrictions, deferred transitions, mode rendering, session-based prompt history restoration, complete plan rendering, approval decisions, stop behavior, fresh-session handoff content, and question formatting.
 
 ### Publishing
 

package/index.ts

@@ -19,6 +19,7 @@ import {
 	buildPlanReviewMessage,
 	classifyPlanExitChoice,
 	decodeModeState,
+	extractPromptHistory,
 	formatModeStatus,
 	isAllowedPlanMutation,
 	makePlanPath,
@@ -366,7 +367,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 		updateModeIndicator(ctx);
 	});
 
-	pi.on("session_start", async (_event, ctx) => {
+	pi.on("session_start", async (event, ctx) => {
 		currentContext = ctx;
 		const entries = ctx.sessionManager.getEntries();
 		const latest = entries
@@ -389,6 +390,8 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 		updateModeIndicator(ctx);
 
 		if (ctx.mode === "tui") {
+			// Startup history is populated after session_start; replacement flows recreate the editor after that step.
+			const promptHistory = event.reason === "startup" ? [] : extractPromptHistory(ctx.sessionManager.getBranch());
 			class ModeEditor extends CustomEditor {
 				onCycle?: () => void;
 
@@ -397,7 +400,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 				}
 
 				override render(width: number): string[] {
-					const badge = formatModeStatus(selectedMode, ctx.ui.theme);
+					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)));
@@ -421,6 +424,7 @@ export default function planBuildModes(pi: ExtensionAPI): void {
 			}
 			ctx.ui.setEditorComponent((tui, theme, keybindings) => {
 				const editor = new ModeEditor(tui, theme, keybindings);
+				for (const prompt of promptHistory) editor.addToHistory(prompt);
 				requestEditorRender = () => editor.requestModeRender();
 				editor.onCycle = () => {
 					if (currentContext) void selectMode(nextMode(selectedMode), currentContext, "manual");

package/package.json

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

package/utils.ts

@@ -9,14 +9,13 @@ const MODE_LABELS: Record<Mode, { color: string; text: string }> = {
 };
 
 export interface ModeStatusTheme {
-	fg(color: "dim", text: string): string;
 	bold(text: string): string;
 }
 
-export function formatModeStatus(mode: Mode, theme: ModeStatusTheme): 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 theme.fg("dim", "[") + modeText + theme.fg("dim", "]");
+	return borderColor("[") + modeText + borderColor("]");
 }
 
 export const PLAN_EXIT_APPROVE_CHOICE = "Switch to Build and implement here";
@@ -122,6 +121,34 @@ export function formatQuestionAnswers(answers: QuestionAnswerData[]): string {
 	return answers.map((answer) => `"${answer.question}"="${answer.answers.length ? answer.answers.join(", ") : "Unanswered"}"`).join(", ");
 }
 
+export function extractPromptHistory(entries: readonly unknown[], limit = 100): string[] {
+	const prompts: string[] = [];
+	for (const entry of entries) {
+		if (!entry || typeof entry !== "object") continue;
+		const candidate = entry as {
+			type?: unknown;
+			message?: { role?: unknown; content?: unknown };
+		};
+		if (candidate.type !== "message" || candidate.message?.role !== "user") continue;
+
+		const content = candidate.message.content;
+		const text = typeof content === "string"
+			? content
+			: Array.isArray(content)
+				? content
+					.filter((block): block is { type: "text"; text: string } =>
+						!!block && typeof block === "object" && (block as { type?: unknown }).type === "text" && typeof (block as { text?: unknown }).text === "string")
+					.map((block) => block.text)
+					.join("")
+				: "";
+		const trimmed = text.trim();
+		if (!trimmed || prompts.at(-1) === trimmed) continue;
+		prompts.push(trimmed);
+	}
+	const maxEntries = Math.max(0, Math.floor(limit));
+	return maxEntries === 0 ? [] : prompts.slice(-maxEntries);
+}
+
 export function nextMode(mode: Mode): Mode {
 	return mode === "build" ? "plan" : "build";
 }