@ghyper9023/pi-dev-workflow 0.3.1 → 0.4.0

package/extensions/dev-prompts.ts

@@ -24,17 +24,22 @@
  */
 
 import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
-import { runGrillPhase, runPRDPhase, type AgentDef, type GrillOptions } from "./grill-me-agent";
+import { runGrillPhase, runPRDPhase, saveAnswerFile, recoverFromBackup, type GrillOptions } from "./grill-me-agent";
+import { discoverAgents } from "./sub-agents";
+import { runWorkflow, loadCheckpointFromFile, type WorkflowStepDef } from "./workflow-engine";
+import { uiSelect, uiConfirm, uiInput, BACK_MARKER } from "./ui-helpers";
 
 // ── Helpers ──────────────────────────────────────────────────
 
-/** Ask a single question. Returns `undefined` on cancel (Esc). */
+/** Ask a single question with proper wrapping. Returns `undefined` on cancel (Esc), or BACK_MARKER for back. */
 async function ask(
 	ctx: ExtensionCommandContext,
 	label: string,
 	placeholder: string,
+	backable = false,
+	initialValue = "",
 ): Promise<string | undefined> {
-	return ctx.ui.input(label, { placeholder, required: false });
+	return uiInput(ctx, label, placeholder, false, backable, initialValue);
 }
 
 /** Check if a field value is empty or explicitly "无". */
@@ -351,170 +356,284 @@ function assembleComparePrompt(f: CompareFields): string {
 	return lines.join("\n");
 }
 
-// ── Specialized Agent Definitions for Grill ─────────────────
-
-/** Bug fix root cause analysis & reproduction review. */
-const FIX_GRILL_AGENT_DEF: AgentDef = {
-	name: "fix-grill-agent",
-	description: "Bug fix review agent — challenges the developer on understanding the bug root cause",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert debugger reviewing a bug fix plan. Interview the developer relentlessly.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format)",
-		"- Focus on: reproduction steps, environment conditions, input variations, error messages, logs, attempted fixes",
-		"- Push for precise root cause identification — ask 'why' at least 3 levels deep",
-		"- Check if the developer has considered: edge cases in inputs, race conditions, state leakage, timeout, memory",
-		"- Verify the proposed fix addresses the root cause, not just the symptom",
-		"- If the codebase has logs/metrics available, suggest checking specific patterns",
-		"- Stress-test regression risk: could the fix break other parts of the system?",
-		"- Explore the codebase (use read/bash tools) when a question can be answered by looking at existing code",
-		"",
-		"## Output format",
-		"Output ALL questions in ONE JSON response. No preamble or explanation.",
-		'Only output the JSON object: { "questions": [{ "id": 1, "question": "...", "options": ["..."] }] }',
-		"",
-		"## Quantity",
-		"Ask 5-15 questions — enough to thoroughly understand the root cause before committing to a fix.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the bug report (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
-
-/** Document outline & structure review. */
-const DOC_GRILL_AGENT_DEF: AgentDef = {
-	name: "doc-grill-agent",
-	description: "Documentation review agent — reviews document outline before writing",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert technical writer reviewing a documentation plan. Interview the developer.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format)",
-		"- Focus on: target audience level, document structure, what examples to include, what NOT to cover",
-		"- Check if the proposed outline covers: overview → quick start → detailed usage → FAQ/troubleshooting",
-		"- Ask about existing documentation that should be linked or consolidated",
-		"- Clarify terminology preferences and API naming conventions",
-		"- Identify potential gaps: error handling, security considerations, performance notes, deprecation notices",
-		"- Explore the codebase (use read/bash tools) to see existing docs for consistency",
-		"",
-		"## Output format",
-		"Output ALL questions in ONE JSON response. No preamble or explanation.",
-		'Only output the JSON object: { "questions": [{ "id": 1, "question": "...", "options": ["..."] }] }',
-		"",
-		"## Quantity",
-		"Ask 3-10 questions — enough to define the scope and structure.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the documentation request (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
-
-/** Refactoring plan review. */
-const REFACTOR_GRILL_AGENT_DEF: AgentDef = {
-	name: "refactor-grill-agent",
-	description: "Refactoring review agent — reviews refactoring plan before implementation",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert software architect reviewing a refactoring plan. Interview the developer.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format)",
-		"- Focus on: module boundaries, API compatibility, dependency inversion, test strategy, migration steps",
-		"- Verify that behavior is preserved — no hidden logic changes disguised as refactoring",
-		"- Ask about: interface contracts, error handling behavior, logging behavior, side effects",
-		"- Check for risk: what's the rollback plan? Can we refactor incrementally?",
-		"- Ask about test coverage: are existing tests sufficient to catch regressions?",
-		"- Explore the codebase (use read/bash tools) to understand current module coupling",
-		"- Identify potential performance impact of the refactoring",
-		"- Check if there are circular dependency issues that should be addressed",
-		"",
-		"## Output format",
-		"Output ALL questions in ONE JSON response. No preamble or explanation.",
-		'Only output the JSON object: { "questions": [{ "id": 1, "question": "...", "options": ["..."] }] }',
-		"",
-		"## Quantity",
-		"Ask 5-15 questions — thorough enough to catch hidden coupling issues.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the refactoring request (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
-
-/** Test plan coverage review. */
-const TEST_GRILL_AGENT_DEF: AgentDef = {
-	name: "test-grill-agent",
-	description: "Test plan review agent — reviews test coverage and approach before writing tests",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert QA engineer reviewing a test plan. Interview the developer.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format)",
-		"- Focus on: coverage dimensions, edge cases not yet considered, mocking strategy, test isolation",
-		"- Ask about: null/empty inputs, timeouts, idempotency, retry logic, success/failure paths",
-		"- Check if the developer has considered: 4xx/5xx HTTP responses, concurrent access, partial failures",
-		"- Ask about testing infrastructure: how to run tests, CI integration, test data management",
-		"- Verify testability: are dependencies injected? Can we mock external services?",
-		"- Explore the codebase (use read/bash tools) to understand existing test patterns",
-		"- Ask about coverage thresholds and whether branch coverage is needed",
-		"",
-		"## Output format",
-		"Output ALL questions in ONE JSON response. No preamble or explanation.",
-		'Only output the JSON object: { "questions": [{ "id": 1, "question": "...", "options": ["..."] }] }',
-		"",
-		"## Quantity",
-		"Ask 5-12 questions — enough to define a thorough test strategy.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the test request (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
-
-/** Performance optimization plan review. */
-const PERF_GRILL_AGENT_DEF: AgentDef = {
-	name: "perf-grill-agent",
-	description: "Performance review agent — reviews optimization approach and benchmarking strategy",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert performance engineer reviewing an optimization plan. Interview the developer.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format)",
-		"- Focus on: benchmarking methodology, measurement tools, baseline metrics, optimization approaches",
-		"- Verify the bottleneck identification: is it really the bottleneck? What's the evidence?",
-		"- Ask about: existing profiling data, hot paths, memory vs CPU trade-offs",
-		"- Check if simpler solutions exist (e.g., caching before algorithm rewrite)",
-		"- Ask about regression risk: could the optimization introduce correctness issues?",
-		"- Explore the codebase (use read/bash tools) to understand current implementation",
-		"- Ask about monitoring: how will we measure the improvement in production?",
-		"- Challenge assumptions: is this optimization premature? What's the user-facing impact?",
-		"",
-		"## Output format",
-		"Output ALL questions in ONE JSON response. No preamble or explanation.",
-		'Only output the JSON object: { "questions": [{ "id": 1, "question": "...", "options": ["..."] }] }',
-		"",
-		"## Quantity",
-		"Ask 5-12 questions — enough to validate the approach before implementation.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the optimization request (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
+// ── Specialized Agent Definitions (loaded from agents/grill/ directory) ──
+
+const _fixGrillAgent = discoverAgents().find(a => a.name === "dev-fix-grill-agent")!;
+const _docGrillAgent = discoverAgents().find(a => a.name === "dev-doc-grill-agent")!;
+const _refactorGrillAgent = discoverAgents().find(a => a.name === "dev-refactor-grill-agent")!;
+const _testGrillAgent = discoverAgents().find(a => a.name === "dev-test-grill-agent")!;
+const _perfGrillAgent = discoverAgents().find(a => a.name === "dev-perf-grill-agent")!;
+
+// ── Workflow configurations ──────────────────────────────────
+
+const FEAT_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 生成实施计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 实施代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 精简代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 300_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 更新文档",
+		type: "confirm",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const FIX_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析根因并制定修复计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 修复代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 更新文档",
+		type: "confirm",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const REFACTOR_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析重构计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 重构代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 精简代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 300_000,
+	},
+];
+
+const PERF_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析性能问题并制定优化计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "⚡ 优化代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+];
+
+const TEST_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析测试计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🧪 编写测试 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+];
+
+const DOC_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析文档需求",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 撰写文档",
+		type: "auto",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const STYLE_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 风格调整 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 2,
+		timeoutMs: 300_000,
+	},
+];
+
+const SECURITY_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "reviewer",
+		label: "🔒 安全审查",
+		type: "auto",
+		agentName: "reviewer",
+		timeoutMs: 300_000,
+	},
+];
 
 // ── Command runner ───────────────────────────────────────────
 
+/** Format workflow steps into a readable list. */
+function formatWorkflowSteps(steps: WorkflowStepDef[]): string {
+	return steps.map((s, i) => `${i + 1}. ${s.label}`).join("\n");
+}
+
 /**
- * Run a wizard with an optional Grill phase:
- * Wizard → Assemble → (Grill?) → Send
+ * Prompt user to choose workflow mode and optionally customize sub-agent chain.
+ *
+ * Returns true if workflow was started and handled (caller should return immediately).
+ * Returns false if caller should fall through to direct prompt sending.
  */
+async function promptWorkflowDecision(
+	ctx: ExtensionCommandContext,
+	pi: ExtensionAPI,
+	finalPrompt: string,
+	defaultSteps: WorkflowStepDef[],
+): Promise<boolean> {
+	if (!defaultSteps || defaultSteps.length === 0) return false;
+
+	const choice = await uiSelect(
+		ctx,
+		"🚀 选择工作流模式",
+		[
+			"1. 使用默认链式子代理（推荐）",
+			"2. 自定义链式子代理",
+			"3. 退出工作流（直接发送 prompt 给主代理）",
+		],
+	);
+
+	if (!choice || choice.startsWith("3")) {
+		return false;
+	}
+
+	if (choice.startsWith("1")) {
+		saveAnswerFile(ctx.cwd, finalPrompt);
+		await runWorkflow(ctx, pi, finalPrompt, { steps: defaultSteps }, "快速链式");
+		return true;
+	}
+
+	// ── Custom mode: let user pick steps and set timeouts (with back support) ──
+	const customSteps: WorkflowStepDef[] = [];
+	let stepIdx = 0;
+	while (stepIdx >= 0 && stepIdx < defaultSteps.length) {
+		const step = defaultSteps[stepIdx]!;
+
+		// If this step was already added, show current config, else show default
+		const existingStep = customSteps.find(cs => cs.id === step.id);
+		const include = await uiConfirm(
+			ctx,
+			`📌 ${step.label}${existingStep ? ' (已添加)' : ''}`,
+			`类型: ${step.type}\n默认超时: ${(step.timeoutMs / 60000).toFixed(0)} 分钟`,
+			stepIdx > 0, // backable only after first step
+		);
+		if (include === "back") {
+			// Go back: remove the last added step
+			if (customSteps.length > 0) {
+				customSteps.pop();
+				stepIdx--;
+				continue;
+			}
+			return false; // nothing to go back to
+		}
+		if (include === undefined) return false; // Esc
+
+		if (!include) {
+			stepIdx++;
+			continue; // skip this step
+		}
+
+		const existingTimeout = existingStep?.timeoutMs;
+		const timeoutStr = await uiInput(
+			ctx,
+			`⏱️ ${step.label} - 超时时间(分钟)`,
+			existingTimeout
+				? `当前: ${(existingTimeout / 60000).toFixed(0)} 分钟，留空保持`
+				: `留空保持默认 (${(step.timeoutMs / 60000).toFixed(0)} 分钟)`,
+		);
+		if (timeoutStr === undefined) { stepIdx++; continue; } // Esc on input = skip
+
+		if (existingStep) {
+			// Update existing step's timeout
+			existingStep.timeoutMs = timeoutStr ? parseInt(timeoutStr, 10) * 60 * 1000 || step.timeoutMs : step.timeoutMs;
+		} else {
+			customSteps.push({
+				...step,
+				timeoutMs: timeoutStr ? parseInt(timeoutStr, 10) * 60 * 1000 || step.timeoutMs : step.timeoutMs,
+			});
+		}
+		stepIdx++;
+	}
+
+	if (customSteps.length === 0) {
+		return false;
+	}
+
+	saveAnswerFile(ctx.cwd, finalPrompt);
+	await runWorkflow(ctx, pi, finalPrompt, { steps: customSteps }, "自定义");
+	return true;
+}
+
 async function runWizardWithGrill(
 	ctx: ExtensionCommandContext,
 	pi: ExtensionAPI,
@@ -523,17 +642,30 @@ async function runWizardWithGrill(
 	questions: Array<{ label: string; placeholder: string; key: string }>,
 	assembler: (answers: Record<string, string>) => string,
 	grillOptions?: GrillOptions,
+	workflowConfig?: { steps: WorkflowStepDef[] },
 ): Promise<void> {
-	ctx.ui.notify(`📋 /dev-${type} — ${label}，请逐项填写以下信息（留空跳过对应段落，Esc 取消）`, "info");
-
 	const answers: Record<string, string> = {};
-	for (const q of questions) {
-		const val = await ask(ctx, q.label, q.placeholder);
+	let idx = 0;
+
+	while (idx >= 0 && idx < questions.length) {
+		const q = questions[idx]!;
+		const existingVal = answers[q.key];
+		const placeholder = existingVal
+			? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+			: q.placeholder;
+		const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 		if (val === undefined) {
-			ctx.ui.notify("❌ 已取消", "warning");
+			return;
+		}
+		if (val === BACK_MARKER) {
+			if (idx > 0) {
+				idx--;
+				continue;
+			}
 			return;
 		}
 		answers[q.key] = val;
+		idx++;
 	}
 
 	const basePrompt = assembler(answers);
@@ -549,25 +681,32 @@ async function runWizardWithGrill(
 			loaderLabel: grillOptions.loaderLabel,
 		});
 		if (grillResult.cancelled) {
-			ctx.ui.notify("❌ 操作已取消", "warning");
 			return;
 		}
 		finalPrompt = grillResult.enhancedPrompt;
 	}
 
-	ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
-	pi.sendUserMessage(finalPrompt);
-	ctx.ui.notify(`📝 /dev-${type} 提示词已投递，主代理正在处理`, "info");
+	// ── Workflow phase ───────────────────────────
+	if (workflowConfig && workflowConfig.steps.length > 0) {
+		const handled = await promptWorkflowDecision(ctx, pi, finalPrompt, workflowConfig.steps);
+		if (handled) return;
+	}
+
+	// ── Guard & persist before sending ──────────────────────
+	if (!finalPrompt) {
+		const recovered = recoverFromBackup(ctx.cwd);
+		if (recovered) {
+			finalPrompt = recovered;
+		} else {
+			return;
+		}
+	}
+	const answerPath = saveAnswerFile(ctx.cwd, finalPrompt);
+	pi.sendUserMessage(finalPrompt, { deliverAs: "followUp" });
 }
 
 /**
  * Run a wizard: ask questions, assemble prompt, send to agent.
- * @param ctx     Command context
- * @param pi      Extension API (for sendUserMessage)
- * @param type    Display name of the prompt type (e.g. "feat")
- * @param label   Friendly label (e.g. "新功能/创意生成")
- * @param questions  Array of [label, placeholder] pairs
- * @param assembler  Function that takes answers and returns the prompt string
  */
 async function runWizard(
 	ctx: ExtensionCommandContext,
@@ -576,28 +715,48 @@ async function runWizard(
 	label: string,
 	questions: Array<{ label: string; placeholder: string; key: string }>,
 	assembler: (answers: Record<string, string>) => string,
+	workflowConfig?: { steps: WorkflowStepDef[] },
 ): Promise<void> {
-	ctx.ui.notify(`📋 /dev-${type} — ${label}，请逐项填写以下信息（留空跳过对应段落，Esc 取消）`, "info");
-
 	const answers: Record<string, string> = {};
-
-	for (const q of questions) {
-		const val = await ask(ctx, q.label, q.placeholder);
+	let idx = 0;
+
+	while (idx >= 0 && idx < questions.length) {
+		const q = questions[idx]!;
+		const existingVal = answers[q.key];
+		const placeholder = existingVal
+			? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+			: q.placeholder;
+		const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 		if (val === undefined) {
-			ctx.ui.notify("❌ 已取消", "warning");
+			// Esc → cancel whole wizard
+			return;
+		}
+		if (val === BACK_MARKER) {
+			// Go back to previous question
+			if (idx > 0) {
+				idx--;
+				continue;
+			}
+			// Already at first question → cancel
 			return;
 		}
 		answers[q.key] = val;
+		idx++;
 	}
 
 	const prompt = assembler(answers);
-	ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
 
-	// Send the assembled prompt to the main agent
-	pi.sendUserMessage(prompt);
+	// ── Workflow phase ───────────────────────────
+	if (workflowConfig && workflowConfig.steps.length > 0) {
+		const handled = await promptWorkflowDecision(ctx, pi, prompt, workflowConfig.steps);
+		if (handled) return;
+	}
 
-	// Also show the prompt in a notification for reference
-	ctx.ui.notify(`📝 /dev-${type} 提示词已投递，主代理正在处理`, "info");
+	// Persist prompt before sending
+	saveAnswerFile(ctx.cwd, prompt);
+
+	// Send the assembled prompt to the main agent
+	pi.sendUserMessage(prompt, { deliverAs: "followUp" });
 }
 
 // ── Questions for each command ────────────────────────────────
@@ -690,45 +849,55 @@ const COMPARE_QUESTIONS = [
 export default function (pi: ExtensionAPI) {
 	// ── /dev-feat ──────────────────────────────────────────────
 	pi.registerCommand("dev-feat", {
-		description: "(prompt wizard) 新功能/创意生成 — 支持设计评审 (Grill) + PRD 生成",
+		description: "(prompt wizard) 新功能/创意生成 — 支持设计评审 (Grill) + 自动化工作流",
 		handler: async (_args, ctx) => {
-			// ── Phase 1: Wizard ──────────────────────────────────
-			ctx.ui.notify("📋 /dev-feat — 新功能/创意生成，请逐项填写以下信息（留空跳过对应段落，Esc 取消）", "info");
-
 			const answers: Record<string, string> = {};
-			for (const q of FEAT_QUESTIONS) {
-				const val = await ask(ctx, q.label, q.placeholder);
+			let featIdx = 0;
+			while (featIdx >= 0 && featIdx < FEAT_QUESTIONS.length) {
+				const q = FEAT_QUESTIONS[featIdx]!;
+				const existingVal = answers[q.key];
+				const placeholder = existingVal
+					? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+					: q.placeholder;
+				const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 				if (val === undefined) {
-					ctx.ui.notify("❌ 已取消", "warning");
+					return;
+				}
+				if (val === BACK_MARKER) {
+					if (featIdx > 0) {
+						featIdx--;
+						continue;
+					}
 					return;
 				}
 				answers[q.key] = val;
+				featIdx++;
 			}
 
-			// ── Phase 2: Assemble base prompt ───────────────────
 			const basePrompt = assembleFeatPrompt(answers as FeatFields);
-			ctx.ui.notify(`✅ 基本信息已收集，共 ${FEAT_QUESTIONS.length} 项`, "success");
 
-			// ── Phase 3: Grill (设计评审) ───────────────────────
 			const grillResult = await runGrillPhase(basePrompt, ctx);
 			if (grillResult.cancelled) {
-				ctx.ui.notify("❌ 操作已取消", "warning");
 				return;
 			}
 			const finalPrompt = grillResult.enhancedPrompt;
 
-			// ── Phase 4: Send to main agent ─────────────────────
-			ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
-			pi.sendUserMessage(finalPrompt);
-			ctx.ui.notify(`📝 /dev-feat 提示词已投递，主代理正在处理`, "info");
-
-			// ── Phase 5: Wait for agent to finish ───────────────
-			await ctx.waitForIdle();
+			if (FEAT_WORKFLOW_STEPS.length > 0) {
+				const handled = await promptWorkflowDecision(ctx, pi, finalPrompt, FEAT_WORKFLOW_STEPS);
+				if (handled) return;
+			}
 
-			// ── Phase 6: PRD generation ─────────────────────────
-			const moduleHint = (answers as FeatFields).module || "feature";
-			const grillContext = finalPrompt;
-			await runPRDPhase(grillContext, moduleHint, pi, ctx);
+			if (!finalPrompt) {
+				const recovered = recoverFromBackup(ctx.cwd);
+				if (recovered) {
+					const answerPath = saveAnswerFile(ctx.cwd, recovered);
+					pi.sendUserMessage(recovered, { deliverAs: "followUp" });
+					return;
+				}
+				return;
+			}
+			const answerPath = saveAnswerFile(ctx.cwd, finalPrompt);
+			pi.sendUserMessage(finalPrompt, { deliverAs: "followUp" });
 		},
 	});
 
@@ -740,12 +909,13 @@ export default function (pi: ExtensionAPI) {
 				ctx, pi, "fix", "问题排查/错误修正",
 				FIX_QUESTIONS, assembleFixPrompt,
 				{
-					agentDef: FIX_GRILL_AGENT_DEF,
+					agentDef: _fixGrillAgent,
 					title: "🐛 Bug 根因分析评审",
-					description: "是否进入 Bug 根因分析评审 (Grill) 模式？\nAI 会从复现条件、根因推理、修复方案、回归风险等维度挑战你的理解。",
+					description: "AI 会从复现条件、根因推理、修复方案、回归风险等维度挑战你的理解。",
 					questionTitle: "Bug 根因分析",
 					loaderLabel: "🧠 AI 正在分析代码并生成根因评审问题...",
 				},
+				{ steps: FIX_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -758,12 +928,13 @@ export default function (pi: ExtensionAPI) {
 				ctx, pi, "doc", "文档生成/总结",
 				DOC_QUESTIONS, assembleDocPrompt,
 				{
-					agentDef: DOC_GRILL_AGENT_DEF,
+					agentDef: _docGrillAgent,
 					title: "📄 文档大纲评审",
-					description: "是否进入文档大纲评审 (Grill) 模式？\nAI 会从受众定位、结构安排、示例选择等维度审视你的文档计划。",
+					description: "AI 会从受众定位、结构安排、示例选择等维度审视你的文档计划。",
 					questionTitle: "文档大纲评审",
 					loaderLabel: "🧠 AI 正在分析并生成文档大纲评审问题...",
 				},
+				{ steps: DOC_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -776,12 +947,13 @@ export default function (pi: ExtensionAPI) {
 				ctx, pi, "refactor", "重构/优化",
 				REFACTOR_QUESTIONS, assembleRefactorPrompt,
 				{
-					agentDef: REFACTOR_GRILL_AGENT_DEF,
+					agentDef: _refactorGrillAgent,
 					title: "🔧 重构方案评审",
-					description: "是否进入重构方案评审 (Grill) 模式？\nAI 会从模块边界、API 兼容性、测试策略、迁移风险等维度审视你的重构计划。",
+					description: "AI 会从模块边界、API 兼容性、测试策略、迁移风险等维度审视你的重构计划。",
 					questionTitle: "重构方案评审",
 					loaderLabel: "🧠 AI 正在分析代码并生成重构评审问题...",
 				},
+				{ steps: REFACTOR_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -794,12 +966,13 @@ export default function (pi: ExtensionAPI) {
 				ctx, pi, "test", "测试用例/评估",
 				TEST_QUESTIONS, assembleTestPrompt,
 				{
-					agentDef: TEST_GRILL_AGENT_DEF,
+					agentDef: _testGrillAgent,
 					title: "🧪 测试计划评审",
-					description: "是否进入测试计划评审 (Grill) 模式？\nAI 会从覆盖维度、边界条件、模拟策略等角度审视你的测试方案。",
+					description: "AI 会从覆盖维度、边界条件、模拟策略等角度审视你的测试方案。",
 					questionTitle: "测试计划评审",
 					loaderLabel: "🧠 AI 正在分析并生成测试评审问题...",
 				},
+				{ steps: TEST_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -820,12 +993,13 @@ export default function (pi: ExtensionAPI) {
 				ctx, pi, "perf", "性能优化",
 				PERF_QUESTIONS, assemblePerfPrompt,
 				{
-					agentDef: PERF_GRILL_AGENT_DEF,
+					agentDef: _perfGrillAgent,
 					title: "⚡ 性能优化方案评审",
-					description: "是否进入性能优化方案评审 (Grill) 模式？\nAI 会从基准测试方法、优化方向、回归风险等维度审视你的方案。",
+					description: "AI 会从基准测试方法、优化方向、回归风险等维度审视你的方案。",
 					questionTitle: "性能优化方案评审",
 					loaderLabel: "🧠 AI 正在分析并生成性能优化评审问题...",
 				},
+				{ steps: PERF_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -834,7 +1008,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("dev-style", {
 		description: "(prompt wizard) 风格/格式调整 — 交互填写后发送优化提示词给主代理",
 		handler: async (_args, ctx) => {
-			await runWizard(ctx, pi, "style", "风格/格式调整", STYLE_QUESTIONS, assembleStylePrompt);
+			await runWizard(ctx, pi, "style", "风格/格式调整", STYLE_QUESTIONS, assembleStylePrompt, { steps: STYLE_WORKFLOW_STEPS });
 		},
 	});
 
@@ -842,7 +1016,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("dev-security", {
 		description: "(prompt wizard) 安全审查 — 交互填写后发送优化提示词给主代理",
 		handler: async (_args, ctx) => {
-			await runWizard(ctx, pi, "security", "安全审查", SECURITY_QUESTIONS, assembleSecurityPrompt);
+			await runWizard(ctx, pi, "security", "安全审查", SECURITY_QUESTIONS, assembleSecurityPrompt, { steps: SECURITY_WORKFLOW_STEPS });
 		},
 	});
 
@@ -861,4 +1035,16 @@ export default function (pi: ExtensionAPI) {
 			await runWizard(ctx, pi, "compare", "对比评估", COMPARE_QUESTIONS, assembleComparePrompt);
 		},
 	});
+
+	// ── /dev-workflow-continue — 恢复中断的工作流 ─────────────
+	pi.registerCommand("dev-workflow-continue", {
+		description: "恢复上次中断的自动化工作流（从 checkpoint 继续）",
+		handler: async (_args, ctx) => {
+			const cp = loadCheckpointFromFile(ctx.cwd);
+			if (!cp) {
+				return;
+			}
+			await runWorkflow(ctx, pi, cp.prompt, { steps: FEAT_WORKFLOW_STEPS }, "恢复");
+		},
+	});
 }