@narumitw/pi-plan-mode 0.1.31 → 0.1.33

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@narumitw/pi-plan-mode",
-	"version": "0.1.31",
+	"version": "0.1.33",
 	"description": "Pi extension that adds a Codex-like read-only /plan collaboration mode.",
 	"type": "module",
 	"license": "MIT",

package/src/plan-mode.ts

@@ -5,7 +5,6 @@ const STATUS_KEY = "plan-mode";
 const PLAN_WIDGET_KEY = "plan-mode-plan";
 const PLAN_CONTEXT_MESSAGE_TYPE = "plan-mode-context";
 const PROPOSED_PLAN_MESSAGE_TYPE = "proposed-plan";
-const PLAN_IMPLEMENTATION_MESSAGE_TYPE = "plan-mode-implementation";
 const PLAN_CONTEXT_MARKER = "[CODEX-LIKE PLAN MODE ACTIVE]";
 const SAFE_BUILTIN_PLAN_TOOLS = new Set(["read", "bash", "grep", "find", "ls"]);
 const BLOCKED_BUILTIN_TOOLS = new Set(["edit", "write"]);
@@ -214,8 +213,7 @@ export default function planMode(pi: ExtensionAPI) {
 		if (!wasEnabled) {
 			ctx.ui.notify("Plan mode enabled. I will explore and plan, but not modify files.", "info");
 		}
-		if (ctx.isIdle()) pi.sendUserMessage(prompt);
-		else pi.sendUserMessage(prompt, { deliverAs: "followUp" });
+		sendPlanModeUserMessage(prompt, ctx);
 	}
 
 	function exitPlanMode(ctx: ExtensionContext) {
@@ -226,6 +224,11 @@ export default function planMode(pi: ExtensionAPI) {
 		updateUi(ctx);
 	}
 
+	function sendPlanModeUserMessage(message: string, ctx: ExtensionContext) {
+		if (ctx.isIdle()) pi.sendUserMessage(message);
+		else pi.sendUserMessage(message, { deliverAs: "followUp" });
+	}
+
 	function startImplementation(ctx: ExtensionContext) {
 		const plan = state.latestPlan?.trim();
 		exitPlanMode(ctx);
@@ -235,13 +238,9 @@ export default function planMode(pi: ExtensionAPI) {
 			return;
 		}
 
-		pi.sendMessage(
-			{
-				customType: PLAN_IMPLEMENTATION_MESSAGE_TYPE,
-				content: `Plan mode is now disabled. Full tool access is restored. Implement this proposed plan now:\n\n${plan}`,
-				display: true,
-			},
-			{ triggerTurn: true },
+		sendPlanModeUserMessage(
+			`Plan mode is now disabled. Full tool access is restored. Implement this proposed plan now:\n\n${plan}`,
+			ctx,
 		);
 	}
 
@@ -299,7 +298,7 @@ export default function planMode(pi: ExtensionAPI) {
 		}
 		if (choice === "Revise plan") {
 			const refinement = await ctx.ui.editor("Revise the plan", "");
-			if (refinement?.trim()) pi.sendUserMessage(refinement.trim());
+			if (refinement?.trim()) sendPlanModeUserMessage(refinement.trim(), ctx);
 			return;
 		}
 		if (choice === "Exit Plan mode") {