@zhushanwen/pi-plan 0.2.0 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-plan",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Lightweight plan mode for Pi coding agent",
   "type": "module",
   "main": "index.ts",

package/src/__tests__/compact-handler.test.ts

@@ -76,7 +76,6 @@ describe("handlePlanComplete", () => {
     expect(pi.sendUserMessage).toHaveBeenCalledWith(expect.any(String), { deliverAs: "steer" });
     expect((pi as unknown as Record<string, unknown>).__goalInit).toHaveBeenCalledWith(
       expect.any(String),
-      expect.arrayContaining([expect.any(String)]),
       undefined,
       ctx,
     );
@@ -107,7 +106,6 @@ describe("handlePlanComplete", () => {
     expect(ctx.ui.notify).not.toHaveBeenCalled();
     expect((pi as unknown as Record<string, unknown>).__goalInit).toHaveBeenCalledWith(
       expect.any(String),
-      expect.arrayContaining([expect.any(String)]),
       undefined,
       ctx,
     );

package/src/command.ts

@@ -175,3 +175,24 @@ function handleEnterPlanMode(
     `3. After plan complete: check subagent capability → suggest goal + wave or single-agent execution.`,
   );
 }
+
+/**
+ * Programmatic entry to start plan mode (for `pi.__planStart`).
+ *
+ * Mirrors the `/plan` command's enter logic. Returns false if plan mode is
+ * already active (caller can then surface a message or wait).
+ *
+ * @returns true if plan mode started; false if already active
+ */
+export function startPlanMode(
+  pi: ExtensionAPI,
+  sessions: PlanSessionMap,
+  ctx: ExtensionContext,
+  requirement: string,
+): boolean {
+  const sessionId = ctx.sessionManager.getSessionId();
+  const state = getPlanState(sessions, sessionId, ctx);
+  if (state.isActive) return false;
+  handleEnterPlanMode(pi, sessions, sessionId, ctx, state, requirement);
+  return true;
+}

package/src/compact.ts

@@ -76,11 +76,12 @@ export function detectGoalCapability(pi: ExtensionAPI): boolean {
 
 /** Try to initialize goal via programming interface */
 function tryGoalInit(pi: ExtensionAPI, planFilePath: string, ctx: ExtensionContext): boolean {
+  // Inline alias mirrors @zhushanwen/pi-goal's GoalInitFn (exported from goal/src/index.ts).
+  // Kept inline due to optional duck-typed coupling (pi.__goalInit) — update both if signature changes.
   type GoalInitFn = (
     objective: string,
-    tasks: string[],
-    budget?: { tokenBudget?: number; timeBudgetMinutes?: number; maxTurns?: number },
-    ctx?: ExtensionContext,
+    budget: { tokenBudget?: number; timeBudgetMinutes?: number } | undefined,
+    ctx: ExtensionContext,
   ) => boolean;
 
   try {
@@ -95,7 +96,7 @@ function tryGoalInit(pi: ExtensionAPI, planFilePath: string, ctx: ExtensionConte
     const tasks = extractPlanSteps(planContent);
     if (tasks.length === 0) return false;
 
-    return goalInit(objective, tasks, undefined, ctx);
+    return goalInit(objective, undefined, ctx);
   } catch {
     return false;
   }

package/src/index.ts

@@ -1,6 +1,6 @@
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 
-import { registerPlanCommand } from "./command.js";
+import { registerPlanCommand, startPlanMode } from "./command.js";
 import { type PlanSessionMap, reconstructPlanState } from "./state.js";
 import { registerPlanTool } from "./tool.js";
 import { updatePlanWidget } from "./widget.js";
@@ -13,6 +13,12 @@ export default function planExtension(pi: ExtensionAPI) {
   registerPlanTool(pi, sessions);
   registerPlanCommand(pi, sessions);
 
+  // External API: __planStart（allow other extensions to start plan mode programmatically, #9）
+  const api = pi as unknown as Record<string, unknown>;
+  api.__planStart = (requirement: string, ctx: ExtensionContext): boolean => {
+    return startPlanMode(pi, sessions, ctx, requirement);
+  };
+
   // Dynamic import compact handlers — avoids cross-group static import
   import("./compact.js").then(({ registerPlanEventHandlers }) => {
     registerPlanEventHandlers(pi, sessions);