@gavin-hu/my-pi-agent 0.7.0 → 0.8.1

package/extensions/plan/README.md

@@ -42,8 +42,9 @@ pi --plan                          # start in plan mode
   them into `todo`.
 - **A navigable review.** The review screen splits the body at its headings,
   names the section you are in, and jumps between sections with `n`/`p`.
-- **Kept current.** After approval, `set_plan_step` flips a step's checkbox in
-  the file, so the plan mirrors what has actually been done.
+- **Kept current.** After approval, `set_plan_steps` flips the checkboxes of
+  the steps you name in the file, so the plan mirrors what has actually been
+  done.
 - **Reviewable plan.** `exit_plan_mode` reads the plan file back and presents
   it for approval. The user approves, keeps planning, or asks for a refinement.
 - **Browser review on demand.** Press `w` on the review screen (or run
@@ -68,7 +69,7 @@ pi --plan                          # start in plan mode
 | `enter_plan_mode` | normal mode | Ask to enter plan mode; empty schema, confirmed with `ctx.ui.confirm`. |
 | `write_plan` | plan mode | Save the plan: `title` (file-name slug), `content` (full markdown in the documented outline), optional `plan_path` to overwrite a refinement. Returns the absolute path. |
 | `exit_plan_mode` | plan mode | Read the plan file (`plan_path`) and present it for approval. |
-| `set_plan_step` | normal mode | Keep an approved plan current: set one top-level step (`step`, 1-based, in the order `exit_plan_mode` listed) to `pending` or `completed`, in place. Only the checkbox changes. |
+| `set_plan_steps` | normal mode | Keep an approved plan current: set the checkboxes of the given top-level steps (`updates`, each a 1-based `step` in the order `exit_plan_mode` listed, plus `status`) to `pending` or `completed` in one write. Only the checkboxes change, and a reopened step drops its `[DONE:n]` marker. |
 
 ### Commands and flags
 
@@ -249,6 +250,8 @@ agent file that overrides it.
 
 ## Browser review
 
+<img src="../../assets/plan-review.png" alt="The browser-based plan review, showing the plan's sections, a rendered code block, and an inline comment box" width="100%">
+
 Press `w` on the review screen — or run `/plan review [path]` — to open the plan
 in the default browser on `127.0.0.1`. The page is a review surface: a sticky
 header with the plan title, theme toggle, and Save/Close; a sticky **section
@@ -314,7 +317,7 @@ returns them to the model.
 
 - **Not a sandbox.** Extensions share Pi's OS permissions.
 - **No progress tracker of record.** `todo` owns completion state; the plan
-  file only mirrors it (`set_plan_step`), and plan mode does not read completion
+  file only mirrors it (`set_plan_steps`), and plan mode does not read completion
   back out.
 - **No auto-execution.** Approval hands control back to the model.
 
@@ -322,7 +325,7 @@ returns them to the model.
 
 | Integration point | How |
 |---|---|
-| Tools | `enter_plan_mode`, `write_plan`, `exit_plan_mode`, and `set_plan_step` registered with `executionMode: "sequential"`. `enter_plan_mode` is active normally with `readOnlyHint: true`; `write_plan` (`readOnlyHint: false`) and `exit_plan_mode` (`readOnlyHint: true`) are `defaultActive: false`. `set_plan_step` is active normally (`readOnlyHint: false`, `idempotentHint: true`); the read-only policy default-denies it in plan mode. Each returns typed `details` (`EnterPlanModeDetails`, `WritePlanDetails`, `ExitPlanModeDetails`, `SetPlanStepDetails`); no `outputSchema`/`structuredContent`. |
+| Tools | `enter_plan_mode`, `write_plan`, `exit_plan_mode`, and `set_plan_steps` registered with `executionMode: "sequential"`. `enter_plan_mode` is active normally with `readOnlyHint: true`; `write_plan` (`readOnlyHint: false`) and `exit_plan_mode` (`readOnlyHint: true`) are `defaultActive: false`. `set_plan_steps` is active normally (`readOnlyHint: false`, `idempotentHint: true`); the read-only policy default-denies it in plan mode. Each returns typed `details` (`EnterPlanModeDetails`, `WritePlanDetails`, `ExitPlanModeDetails`, `SetPlanStepsDetails`); no `outputSchema`/`structuredContent`. |
 | Commands and flags | `/plan` (toggle, or enter and send a task), `/plan review [path]` (browser review), `--plan` launch flag, `Ctrl+Alt+P` shortcut. |
 | State | Enabled flag and the tools it hid persisted with `pi.appendEntry("plan-mode", { enabled, removed })` and rebuilt from `ctx.sessionManager.getBranch()`; the last plan path lives in runtime memory for the footer chip. |
 | Hooks | `tool_call` blocks non-read-only calls and forces `subagent` read-only; `before_agent_start` injects the hidden `[PLAN MODE ACTIVE]` context; `context` keeps only the newest plan context while enabled and drops stale ones otherwise. |
@@ -360,10 +363,11 @@ returns them to the model.
   source-line map, so `tui.ts` splits the body at its headings and renders each
   section separately. That gives exact offsets for the current-section label and
   `n`/`p`; the single-section case is byte-for-byte the old render.
-- **`set_plan_step` is a second contained write.** It changes only a step's
-  checkbox, in an existing markdown plan inside the plans directory
-  (`isContainedPlan`), and refuses while plan mode is on — so the read-only
-  guarantee and the path backstop are unchanged.
+- **`set_plan_steps` is a second contained write.** It changes only the named
+  steps' checkboxes, in an existing markdown plan inside the plans directory
+  (`isContainedPlan`), in a single all-or-nothing write — refusing every entry
+  if any step is out of range — and refuses while plan mode is on, so the
+  read-only guarantee and the path backstop are unchanged.
 - **Direction: enforce capabilities at the tool boundary.** (1) A shared
   default-deny policy classifies every call as read-only or mutating instead of
   maintaining a mutator list. (2) Read-only delegation is enforced in the child
@@ -394,13 +398,13 @@ returns them to the model.
 | File | Responsibility |
 |---|---|
 | `index.ts` | Wiring: tools, command, flag, shortcut, events, context injection. |
-| `types.ts` | `PlanModeEntry`, `EnterPlanModeDetails`, `WritePlanDetails`, `ExitPlanModeDetails`, `SetPlanStepDetails`. |
+| `types.ts` | `PlanModeEntry`, `EnterPlanModeDetails`, `WritePlanDetails`, `ExitPlanModeDetails`, `SetPlanStepsDetails`. |
 | `policy.ts` | Plan mode's read-only policy, the shared prompt summary, and the bash-refusal guidance. |
 | `git-bash.ts` | `checkReadOnlyGit`: the pure allowlist that restricts `bash` to read-only git commands. |
-| `plans.ts` | Plan-file store: slug/stamp naming, directory resolution, containment, write/read/`setStep`/`saveFeedback`. |
+| `plans.ts` | Plan-file store: slug/stamp naming, directory resolution, containment, write/read/`setSteps`/`saveFeedback`. |
 | `steps.ts` | Plan-document structure (pure): the outline, step extraction, heading sections, and the checkbox edit. |
 | `runtime.ts` | Enabled state, tool gating, persistence, control-tool identity, footer status. |
-| `tools.ts` | `enter_plan_mode`, `write_plan`, `exit_plan_mode`, and `set_plan_step`. |
+| `tools.ts` | `enter_plan_mode`, `write_plan`, `exit_plan_mode`, and `set_plan_steps`. |
 | `commands.ts` | `/plan` (enter, optional task) and the `/plan review [path]` browser subcommand. |
 | `tui.ts` | `PlanViewComponent` (sectioned Markdown body, section jump, scroll gauge, approve/refine/browser/keep). |
 | `blocks.ts` | Source-line block model (fence-aware headings, paragraphs, list items, code, quotes, blanks). |

package/extensions/plan/plans.ts

@@ -20,7 +20,7 @@ import { CONFIG_DIR_NAME, getAgentDir, type ExtensionAPI } from "@earendil-works
 import { repoRootFor } from "../../lib/git/index.ts";
 import { isInside, isInsideReal } from "../../lib/path.ts";
 import { formatFeedback, type PlanAnnotation } from "./feedback.ts";
-import { extractPlanSteps, setStepStatus, type PlanStep } from "./steps.ts";
+import { extractPlanSteps, setStepStatuses, type StepUpdate } from "./steps.ts";
 
 /** Largest plan accepted, in UTF-8 bytes. */
 export const MAX_PLAN_BYTES = 256 * 1024;
@@ -57,8 +57,8 @@ export interface PlanStore {
 	write(cwd: string, input: WritePlanInput): Promise<WrittenPlan>;
 	/** Read a plan inside the plans directory, or undefined when missing/outside. */
 	read(cwd: string, planPath: string): Promise<StoredPlan | undefined>;
-	/** Set a top-level step's checkbox in an existing plan. Throws when the file or step is invalid. */
-	setStep(cwd: string, planPath: string, step: number, status: PlanStep["status"]): Promise<StoredPlan>;
+	/** Set the given top-level steps' checkboxes in an existing plan. Throws when the file or any step is invalid. */
+	setSteps(cwd: string, planPath: string, updates: StepUpdate[]): Promise<StoredPlan>;
 	/** Replace the plan's stored browser feedback. Throws when the file is missing or outside the plans directory. */
 	saveFeedback(cwd: string, planPath: string, annotations: PlanAnnotation[]): Promise<StoredPlan>;
 }
@@ -222,26 +222,25 @@ export function createPlanStore(pi: ExtensionAPI, options: { now?: () => Date }
 		}
 	};
 
-	const setStep = async (
-		cwd: string,
-		planPath: string,
-		step: number,
-		status: PlanStep["status"],
-	): Promise<StoredPlan> => {
+	const setSteps = async (cwd: string, planPath: string, updates: StepUpdate[]): Promise<StoredPlan> => {
 		const plan = await read(cwd, planPath);
 		if (!plan) throw new Error("No plan file there; pass a plan_path written by write_plan.");
-		const updated = setStepStatus(plan.content, step, status);
-		if (!updated) {
-			const total = extractPlanSteps(plan.content).length;
+		const total = extractPlanSteps(plan.content).length;
+		const outOfRange = updates.find(
+			(update) => !Number.isInteger(update.step) || update.step < 1 || update.step > total,
+		);
+		if (outOfRange) {
 			const range = total === 0 ? "it has no steps" : `valid steps are 1 to ${total}`;
-			throw new Error(`Step ${step} is out of range; ${range}.`);
+			throw new Error(`Step ${outOfRange.step} is out of range; ${range}.`);
 		}
-		writeFileSync(plan.path, updated.plan, "utf-8");
+		const updated = setStepStatuses(plan.content, updates);
+		if (!updated) throw new Error("Could not update the plan steps; pass steps from the plan file.");
+		writeFileSync(plan.path, updated, "utf-8");
 		return {
 			path: plan.path,
 			relativePath: plan.relativePath,
-			content: updated.plan,
-			bytes: Buffer.byteLength(updated.plan, "utf-8"),
+			content: updated,
+			bytes: Buffer.byteLength(updated, "utf-8"),
 		};
 	};
 
@@ -263,5 +262,5 @@ export function createPlanStore(pi: ExtensionAPI, options: { now?: () => Date }
 		};
 	};
 
-	return { dirFor, write, read, setStep, saveFeedback };
+	return { dirFor, write, read, setSteps, saveFeedback };
 }

package/extensions/plan/runtime.ts

@@ -31,7 +31,7 @@ export const ENTER_TOOL = "enter_plan_mode";
 /** Tool the model calls to save the plan artifact. */
 export const WRITE_PLAN_TOOL = "write_plan";
 /** Tool the model calls to keep an approved plan's steps current. */
-export const SET_PLAN_STEP_TOOL = "set_plan_step";
+export const SET_PLAN_STEPS_TOOL = "set_plan_steps";
 /** Custom-entry type used for persistence. */
 export const STATE_TYPE = "plan-mode";
 

package/extensions/plan/steps.ts

@@ -166,28 +166,40 @@ export function extractPlanSteps(plan: string): PlanStep[] {
 	return topLevelItems(plan).map((item) => ({ content: elide(item.step.content), status: item.step.status }));
 }
 
+/** One checkbox change: a 1-based step number and the state to set. */
+export interface StepUpdate {
+	step: number;
+	status: PlanStep["status"];
+}
+
 /**
- * Set the checkbox of the `step`th top-level step (1-based, matching
- * {@link extractPlanSteps}) and return the rewritten plan with the updated step,
- * or undefined when the step number is out of range. Only the marker changes: a
- * plain `1.`/`-` item gains a checkbox, an existing one is set, and the step
- * text is left exactly as written.
+ * Apply a batch of checkbox changes to the given top-level steps (1-based,
+ * matching {@link extractPlanSteps}) and return the rewritten plan, or
+ * undefined when any step number is out of range. All-or-nothing: every step is
+ * validated before any edit, so one bad entry changes nothing. Only the marker
+ * changes: a plain `1.`/`-` item gains a checkbox, an existing one is set, a
+ * `[DONE:n]` marker is dropped when the step is reopened, and the step text is
+ * otherwise left exactly as written. When a step appears twice, the last entry
+ * wins.
  */
-export function setStepStatus(
-	plan: string,
-	step: number,
-	status: "pending" | "completed",
-): { plan: string; step: PlanStep } | undefined {
-	if (!Number.isInteger(step) || step < 1) return undefined;
-	const target = topLevelItems(plan)[step - 1];
-	if (!target) return undefined;
-
-	const existing = target.raw.match(CHECKBOX);
-	const text = existing ? existing[2] : target.raw;
-	const checkbox = status === "completed" ? "[x]" : "[ ]";
+export function setStepStatuses(plan: string, updates: StepUpdate[]): string | undefined {
+	const items = topLevelItems(plan);
+	for (const update of updates) {
+		if (!Number.isInteger(update.step) || update.step < 1 || !items[update.step - 1]) return undefined;
+	}
+
 	const lines = plan.split("\n");
-	lines[target.line] = `${" ".repeat(target.indent)}${target.lead}${checkbox} ${text}`;
-	return { plan: lines.join("\n"), step: parseStep(`${checkbox} ${text}`) };
+	for (const update of updates) {
+		const target = items[update.step - 1];
+		const existing = target.raw.match(CHECKBOX);
+		let text = existing ? existing[2] : target.raw;
+		// A reopened step must lose its marker too: the marker alone reads as
+		// completed, so keeping it would contradict the checkbox.
+		if (update.status === "pending") text = text.replace(DONE_MARKER, "").trimEnd();
+		const checkbox = update.status === "completed" ? "[x]" : "[ ]";
+		lines[target.line] = `${" ".repeat(target.indent)}${target.lead}${checkbox} ${text}`;
+	}
+	return lines.join("\n");
 }
 
 /**

package/extensions/plan/tools.ts

@@ -7,7 +7,7 @@
  * approval (a scrollable review screen in the TUI, a select dialog elsewhere),
  * and on approval seeds the `todo` tool with the plan's steps (through
  * `ctx.executeTool`, so the todo widget and validation run normally).
- * `set_plan_step` then keeps the file current: after approval the model ticks
+ * `set_plan_steps` then keeps the file current: after approval the model ticks
  * each step off in place as it finishes it. The file is the source of truth:
  * the model executes from it.
  *
@@ -26,10 +26,10 @@ import { FULL_SCREEN_OVERLAY, reuseText, toolDetails } from "../../lib/tui.ts";
 import { expandHint } from "../../lib/ui.ts";
 import { READ_ONLY_SUMMARY } from "./policy.ts";
 import { feedbackSummary } from "./feedback.ts";
-import { ENTER_TOOL, EXIT_TOOL, SET_PLAN_STEP_TOOL, WRITE_PLAN_TOOL, type PlanRuntime } from "./runtime.ts";
-import { PLAN_OUTLINE, extractPlanSteps, type PlanStep } from "./steps.ts";
+import { ENTER_TOOL, EXIT_TOOL, SET_PLAN_STEPS_TOOL, WRITE_PLAN_TOOL, type PlanRuntime } from "./runtime.ts";
+import { PLAN_OUTLINE, extractPlanSteps, type PlanStep, type StepUpdate } from "./steps.ts";
 import { PlanViewComponent, sanitizePlanText, type PlanViewAction } from "./tui.ts";
-import type { EnterPlanModeDetails, ExitPlanModeDetails, SetPlanStepDetails, WritePlanDetails } from "./types.ts";
+import type { EnterPlanModeDetails, ExitPlanModeDetails, SetPlanStepsDetails, WritePlanDetails } from "./types.ts";
 import { createPlanWebReview, type PlanWebReview } from "./web.ts";
 
 const ExitPlanModeParams = Type.Object({
@@ -54,10 +54,7 @@ const WritePlanParams = Type.Object({
 	),
 });
 
-const SetPlanStepParams = Type.Object({
-	plan_path: Type.String({
-		description: "Path to the approved plan file, as returned by write_plan or exit_plan_mode.",
-	}),
+const StepUpdateParams = Type.Object({
 	step: Type.Integer({
 		minimum: 1,
 		description: "1-based step number, in the order the plan was approved.",
@@ -67,9 +64,19 @@ const SetPlanStepParams = Type.Object({
 	}),
 });
 
+const SetPlanStepsParams = Type.Object({
+	plan_path: Type.String({
+		description: "Path to the approved plan file, as returned by write_plan or exit_plan_mode.",
+	}),
+	updates: Type.Array(StepUpdateParams, {
+		minItems: 1,
+		description: "The steps to set; each entry sets one step's checkbox, and the last entry wins for a duplicate step.",
+	}),
+});
+
 type ExitPlanModeArgs = Static<typeof ExitPlanModeParams>;
 type WritePlanArgs = Static<typeof WritePlanParams>;
-type SetPlanStepArgs = Static<typeof SetPlanStepParams>;
+type SetPlanStepsArgs = Static<typeof SetPlanStepsParams>;
 
 /** Lines of plan shown in an unexpanded result. */
 const PREVIEW_LINES = 8;
@@ -89,6 +96,21 @@ function preview(plan: string): { body: string; more: number } {
 	return { body: lines.slice(0, PREVIEW_LINES).join("\n"), more: lines.length - PREVIEW_LINES };
 }
 
+/** "Step 2" or "Steps 1, 2" for a set of step numbers. */
+function stepLabel(steps: number[]): string {
+	return `${steps.length === 1 ? "Step" : "Steps"} ${steps.join(", ")}`;
+}
+
+/** "Steps 1, 2 completed; step 3 reopened" for the tool's text result. */
+function describeUpdates(updates: StepUpdate[]): string {
+	const completed = updates.filter((update) => update.status === "completed").map((update) => update.step);
+	const pending = updates.filter((update) => update.status === "pending").map((update) => update.step);
+	const parts: string[] = [];
+	if (completed.length > 0) parts.push(`${stepLabel(completed)} completed`);
+	if (pending.length > 0) parts.push(`${stepLabel(pending)} reopened`);
+	return parts.join("; ");
+}
+
 /** Seed the todo list with the plan's steps; returns what was recorded. */
 async function seedTodos(
 	ctx: ExtensionToolContext,
@@ -464,9 +486,9 @@ export function registerTools(
 					: "";
 			const tail =
 				recorded > 0
-					? ` Recorded ${recorded} step${recorded === 1 ? "" : "s"} in the todo list. As you finish each step, mark it done there and tick it off in the file with ${SET_PLAN_STEP_TOOL}.`
+					? ` Recorded ${recorded} step${recorded === 1 ? "" : "s"} in the todo list. As you finish each step, mark it done there and tick it off in the file with ${SET_PLAN_STEPS_TOOL}.`
 					: steps.length > 0
-						? ` Could not record the steps in the todo list${error ? `: ${error}` : ""}. Keep these steps in mind and tick them off with ${SET_PLAN_STEP_TOOL} as you go.`
+						? ` Could not record the steps in the todo list${error ? `: ${error}` : ""}. Keep these steps in mind and tick them off with ${SET_PLAN_STEPS_TOOL} as you go.`
 						: " If the todo tool is available, record the steps with it before you start.";
 			return {
 				content: [
@@ -525,22 +547,23 @@ export function registerTools(
 	});
 
 	pi.registerTool({
-		name: SET_PLAN_STEP_TOOL,
-		label: "Set plan step",
+		name: SET_PLAN_STEPS_TOOL,
+		label: "Set plan steps",
 		description:
-			"Keep an approved plan file current by setting one top-level step's checkbox in place. The step number matches " +
-			"the order the plan was approved in. Only the checkbox changes; the step text is left as written. Call it as you " +
-			"finish (or reopen) each step so the file mirrors reality.",
-		promptSnippet: "Tick a step off in the approved plan file.",
+			"Keep an approved plan file current by setting the checkboxes of the given top-level steps in place. Each step " +
+			"number matches the order the plan was approved in; a step can be marked completed or reopened. Only the " +
+			"checkboxes change; the step text is left as written. The file is rewritten once, and nothing changes when any " +
+			"step is out of range.",
+		promptSnippet: "Tick steps off in the approved plan file.",
 		promptGuidelines: [
-			"After a plan is approved, call set_plan_step as you complete each step so the plan file stays current.",
+			"After a plan is approved, call set_plan_steps as you complete steps so the plan file stays current.",
 		],
-		parameters: SetPlanStepParams,
+		parameters: SetPlanStepsParams,
 		executionMode: "sequential",
 		annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false },
 
 		async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
-			const args = params as SetPlanStepArgs;
+			const args = params as SetPlanStepsArgs;
 			if (runtime.isEnabled()) {
 				return {
 					content: [{ type: "text", text: "Not while planning; refine the plan with write_plan instead." }],
@@ -549,24 +572,23 @@ export function registerTools(
 				};
 			}
 			try {
-				const plan = await runtime.plans.setStep(ctx.cwd, args.plan_path, args.step, args.status);
+				const plan = await runtime.plans.setSteps(ctx.cwd, args.plan_path, args.updates);
 				const steps = extractPlanSteps(plan.content);
 				const done = steps.filter((step) => step.status === "completed").length;
 				return {
 					content: [
 						{
 							type: "text",
-							text: `Step ${args.step} is ${args.status} (${done}/${steps.length} done) in ${plan.relativePath}.`,
+							text: `${describeUpdates(args.updates)} (${done}/${steps.length} done) in ${plan.relativePath}.`,
 						},
 					],
 					details: {
 						path: plan.path,
 						relativePath: plan.relativePath,
-						step: args.step,
-						status: args.status,
+						updates: args.updates,
 						done,
 						total: steps.length,
-					} satisfies SetPlanStepDetails,
+					} satisfies SetPlanStepsDetails,
 				};
 			} catch (error) {
 				const message = error instanceof Error ? error.message : String(error);
@@ -580,24 +602,32 @@ export function registerTools(
 
 		renderCall(args, theme, context) {
 			const text = reuseText(context?.lastComponent);
-			const step = (args as Partial<SetPlanStepArgs>).step;
-			const suffix = theme.fg("muted", step === undefined ? "a step" : `step ${step}`);
-			text.setText(theme.fg("toolTitle", theme.bold(`${SET_PLAN_STEP_TOOL} `)) + suffix);
+			const updates = (args as Partial<SetPlanStepsArgs>).updates;
+			const label =
+				updates && updates.length === 1 && updates[0]?.step !== undefined
+					? `step ${updates[0].step}`
+					: updates && updates.length > 1
+						? `${updates.length} steps`
+						: "steps";
+			text.setText(theme.fg("toolTitle", theme.bold(`${SET_PLAN_STEPS_TOOL} `)) + theme.fg("muted", label));
 			return text;
 		},
 
 		renderResult(result, _options, theme, context) {
 			const text = reuseText(context?.lastComponent);
-			const details = toolDetails<SetPlanStepDetails>(result.details);
+			const details = toolDetails<SetPlanStepsDetails>(result.details);
 			if (!details) {
 				const first = result.content[0];
 				text.setText(first?.type === "text" ? first.text : "");
 				return text;
 			}
+			const completed = details.updates.filter((update) => update.status === "completed").map((u) => u.step);
+			const pending = details.updates.filter((update) => update.status === "pending").map((u) => u.step);
 			const heading =
-				details.status === "completed"
-					? theme.fg("success", `✓ Step ${details.step} done`)
-					: theme.fg("muted", `Step ${details.step} reopened`);
+				completed.length > 0
+					? theme.fg("success", `✓ ${stepLabel(completed)} done`) +
+						(pending.length > 0 ? theme.fg("muted", ` · ${stepLabel(pending)} reopened`) : "")
+					: theme.fg("muted", `${stepLabel(pending)} reopened`);
 			text.setText(heading + theme.fg("dim", ` · ${details.done}/${details.total}`));
 			return text;
 		},

package/extensions/plan/types.ts

@@ -2,6 +2,8 @@
  * Shared types for the plan-mode extension.
  */
 
+import type { StepUpdate } from "./steps.ts";
+
 /** Persisted (custom entry) plan-mode state. */
 export interface PlanModeEntry {
 	enabled: boolean;
@@ -32,16 +34,14 @@ export interface WritePlanDetails {
 	bytes: number;
 }
 
-/** Structured result carried in `set_plan_step`'s `details`. */
-export interface SetPlanStepDetails {
+/** Structured result carried in `set_plan_steps`'s `details`. */
+export interface SetPlanStepsDetails {
 	/** Absolute path of the plan file that was updated. */
 	path: string;
 	/** Path relative to the working directory, for transcript display. */
 	relativePath: string;
-	/** The 1-based step number that was set. */
-	step: number;
-	/** The step's new status. */
-	status: "pending" | "completed";
+	/** The checkbox changes that were applied, in call order. */
+	updates: StepUpdate[];
 	/** Completed steps after the update. */
 	done: number;
 	/** Total top-level steps in the plan. */

package/lib/README.md

@@ -24,7 +24,7 @@ belongs in that extension instead — see `extensions/web-access/http.ts` and
 | [`interaction.ts`](./interaction.ts) | Turn-affinity prompt contract: the channel a remote adapter installs on `ctx.ui` under a string key, the `InteractionChannel` interface (`request` / `post` / `postFile`), and `remoteChannel` / `isRemoteTurn` / `supportsCustomUI` / `askHuman` (with an optional file/text preface) plus the shared “Other” label. |
 | [`git/`](./git/) | Shared git plumbing. [`git/runner.ts`](./git/runner.ts) is the `RunGit` seam (`createExecRunner` with timeout/env handling, `runGitOrThrow`, `GitError`); [`git/read.ts`](./git/read.ts) holds the read helpers (`repoRoot`, `repoRootFor`, `gitDir`, `revParse`, `hasCommits`, `currentBranch`); [`git/index.ts`](./git/index.ts) is the barrel. |
 | [`highlight.ts`](./highlight.ts) | First-party, escape-first syntax tokenizer (`detectLanguage`, `fenceLanguage`, `highlightLines`, `HighlightLanguage`, `TokenClass`) for js/ts, json, css, html, shell, python, java, yaml, and sql; shared by `file-browse` and plan's code blocks. |
-| [`list-cursor.ts`](./list-cursor.ts) | Scrollable-list behavior for the `/rewind`, `/extensions`, `/doctor`, `/todos`, and `/jobs` screens: `fitRows`, `keepVisible`, `clampScroll`, `navIntent`, `navigateList`, `navigateScroll`, `wheelDelta`, `scrollList`, `isCloseKey`, `selectionMarker`, `formatRange`, and the `ListCursor` state. |
+| [`list-cursor.ts`](./list-cursor.ts) | Scrollable-list behavior for the `/rewind`, `/bytefly`, `/todos`, and `/jobs` screens: `fitRows`, `keepVisible`, `clampScroll`, `navIntent`, `navigateList`, `navigateScroll`, `wheelDelta`, `scrollList`, `isCloseKey`, `selectionMarker`, `formatRange`, and the `ListCursor` state. |
 | [`path.ts`](./path.ts) | `isInside` (string containment) plus `realPathOfNearest` / `isInsideReal` (symlink-aware): the containment predicates the file-browse, worktree, and doc guards share. |
 | [`pi-invocation.ts`](./pi-invocation.ts) | `getPiInvocation`: how to launch a nested `pi` process, shared by `subagent` and bytefly's doctor surface. |
 | [`policy.ts`](./policy.ts) | `createReadOnlyPolicy`: default-deny tool classification from `readOnlyHint` + explicit allow/deny. |

package/lib/requirements.ts

@@ -194,9 +194,9 @@ export const REQUIREMENTS: Readonly<Record<string, ExtensionRequirements>> = {
 	bytefly: {
 		configFiles: [SHARED_CONFIG_FILE],
 		notes: [
-			"TUI only for the header, footer, and turn rule; `/extensions` prints a listing and `/doctor` prints its report in RPC, JSON, and print modes.",
-			"The `/extensions` picker writes through Pi's own settings; project scope requires a trusted project.",
-			"`/doctor`'s list screen and the proactive missing-prerequisite warning need an interactive UI.",
+			"TUI only for the header, footer, and turn rule; `/bytefly` (and the `/extensions` and `/doctor` aliases) prints a listing and report in RPC, JSON, and print modes.",
+			"The panel writes extension enablement through Pi's own settings; project scope requires a trusted project.",
+			"`/bytefly`'s list screen and the proactive missing-prerequisite warning need an interactive UI.",
 		],
 	},
 };

package/lib/tui.ts

@@ -3,9 +3,9 @@
  * todo).
  *
  * `FULL_SCREEN_OVERLAY` is for the one surface that must own the whole terminal:
- * the plan read/review screen. Every list screen (`/todos`, `/jobs`,
- * `/rewind`) stays in the dock's editor slot so they match each other and
- * leave the transcript visible.
+ * the plan read/review screen. Every list screen (`/bytefly`, `/todos`, `/jobs`,
+ * `/rewind`) stays in the dock's editor slot so they match each other and leave
+ * the transcript visible.
  *
  * Screens are full-width, terminal-height-aware components. These helpers build
  * the top rule, clamp a screen's body to the terminal, and render the key-hint
@@ -128,11 +128,10 @@ export interface DockListFrame {
  * its blank line, the body, an optional range row, optional detail rows, and
  * the blank-line-padded key-hint footer.
  *
- * This is the convention `/extensions` shares with `/jobs` and `/todos`.
- * `/doctor` (no summary, no trailing blank, range last) and `/rewind` (a blank
- * after the header and no blank before the hint) place their blanks
- * differently, so they stay hand-rolled rather than changing their output to
- * fit this helper.
+ * This is the convention the bytefly panel (`/bytefly`, and the `/extensions`
+ * and `/doctor` aliases) shares with `/jobs` and `/todos`. `/rewind` (a blank
+ * after the header and no blank before the hint) places its blanks differently,
+ * so it stays hand-rolled rather than changing its output to fit this helper.
  */
 export function dockListFrame(frame: DockListFrame): string[] {
 	const width = Math.max(1, frame.width);

package/lib/web/components.ts

@@ -113,6 +113,17 @@ th[aria-sort="descending"] .bf-sort::after{content:" \\25BC";font-size:9px;color
 .bf-md pre{border:1px solid var(--border);border-radius:var(--radius-md);padding:var(--space-3) var(--space-4);overflow:auto;background:var(--canvas)}
 .bf-md pre code{background:none;border:0;padding:0}
 .bf-md blockquote{margin-left:0;padding:0 1em;border-left:3px solid var(--border);color:var(--muted)}
+.bf-md li.task{list-style:none;margin-left:-1.4em}
+.bf-md .md-check{color:var(--success)}
+.bf-md table{margin:0;border-collapse:collapse;table-layout:auto;width:auto;font-size:var(--fs-sm)}
+.bf-md .md-table-wrap{overflow-x:auto;margin:0 0 1em}
+.bf-md th,.bf-md td{border:1px solid var(--border);padding:6px 10px;text-align:left;vertical-align:top}
+.bf-md thead th{background:var(--surface);font-weight:var(--fw-medium)}
+.bf-md .md-left{text-align:left}
+.bf-md .md-center{text-align:center}
+.bf-md .md-right{text-align:right}
+.bf-md img{max-width:100%;height:auto;border-radius:var(--radius-sm)}
+.bf-md hr{border:0;border-top:1px solid var(--border);margin:1.4em 0}
 .bf-comment{padding:var(--space-3) var(--space-4);background:var(--surface);border:1px solid var(--border);border-left:3px solid var(--warn);border-radius:var(--radius-md)}
 .bf-comment__quote{margin-bottom:var(--space-1);color:var(--muted);font-size:var(--fs-xs);overflow-wrap:anywhere}
 .bf-comment__body{white-space:pre-wrap;overflow-wrap:anywhere}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gavin-hu/my-pi-agent",
-  "version": "0.7.0",
+  "version": "0.8.1",
   "description": "Personal Pi customizations, packaged as a Pi package: worktree isolation and rewind snapshots, question, todo, goal, plan, subagent, job, watch, web-access, file-browse, file-open, document, memory, wechat, and bytefly.",
   "keywords": [
     "pi-package",
@@ -90,7 +90,7 @@
     "themes": [
       "./themes/*.json"
     ],
-    "image": "./assets/bytefly-chrome.png"
+    "image": "https://raw.githubusercontent.com/gavin-hu/my-pi-agent/main/assets/bytefly-chrome.png"
   },
   "scripts": {
     "test": "bun test --parallel=4 --timeout 30000",

package/extensions/bytefly/doctor/commands.ts

@@ -1,68 +0,0 @@
-/**
- * The `/doctor` command.
- *
- * One command for the whole surface: `/doctor` reports everything, `/doctor
- * document` narrows it, `/doctor refresh` re-probes. The TUI opens a docked
- * list screen; every other mode prints the same report as text.
- */
-
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { withRailsSuppressed } from "../../../lib/rails.ts";
-import { REQUIREMENTS, type ExtensionReport } from "../../../lib/requirements.ts";
-import { formatExtension, formatReport, hasProblems } from "./format.ts";
-import type { DoctorRuntime } from "./runtime.ts";
-import { DoctorScreen } from "./tui.ts";
-
-/** The report text for a set of reports, narrowed when one extension was named. */
-function reportText(name: string | undefined, reports: readonly ExtensionReport[]): string {
-	const first = reports[0];
-	return name && first ? formatExtension(first) : formatReport(reports);
-}
-
-/** Register the `/doctor` command. */
-export function registerCommands(pi: ExtensionAPI, runtime: DoctorRuntime): void {
-	pi.registerCommand("doctor", {
-		description: "Check that every extension has what it needs: /doctor [extension] [refresh]",
-		getArgumentCompletions: (prefix: string) => {
-			const options = [...Object.keys(REQUIREMENTS), "refresh"].filter((option) => option.startsWith(prefix));
-			return options.length > 0 ? options.map((value) => ({ value, label: value })) : null;
-		},
-		handler: async (args, ctx) => {
-			const parts = args.trim().split(/\s+/).filter(Boolean);
-			const refresh = parts.includes("refresh");
-			const name = parts.find((part) => part !== "refresh");
-
-			const reports = await runtime.probeAll({ refresh });
-			if (name && !reports.some((report) => report.extension === name)) {
-				ctx.ui.notify(`doctor: no extension named "${name}"`, "warning");
-				return;
-			}
-			const selected = name ? reports.filter((report) => report.extension === name) : reports;
-
-			if (ctx.mode !== "tui") {
-				ctx.ui.notify(reportText(name, selected), hasProblems(selected) ? "warning" : "info");
-				return;
-			}
-
-			// The dock screen owns the editor slot; hide the rails for its lifetime.
-			let current = selected;
-			await withRailsSuppressed(pi, () =>
-				ctx.ui.custom<void>(
-					(tui, theme, _keybindings, done) =>
-						new DoctorScreen({
-							reports: () => current,
-							rows: () => tui.terminal?.rows,
-							theme,
-							requestRender: () => tui.requestRender(),
-							refresh: async () => {
-								const next = await runtime.probeAll({ refresh: true });
-								current = name ? next.filter((report) => report.extension === name) : next;
-								tui.requestRender();
-							},
-							done,
-						}),
-				),
-			);
-		},
-	});
-}