@ghyper9023/pi-dev-workflow 0.3.1 → 0.4.0

package/extensions/grill-me-agent.ts

@@ -3,7 +3,7 @@
  *
  * 职责：
  *   1. runGrillPhase()  — 启动 sub-agent 生成评审问题，TUI 逐题呈现（选项 + 自定义输入）
- *   2. runPRDPhase()    — 启动 sub-agent 生成 PRD，保存到 pi-dev-output/pi-prd/
+ *   2. runPRDPhase()    — 启动 sub-agent 生成 PRD，保存到 .pi-dev-output/pi-prd/
  *
  * 关键设计决策（修复 #2）：
  *   sub-agent 通过 `write` 工具将评审问题写入临时文件，主进程事后读取。
@@ -14,7 +14,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
 import { BorderedLoader, DynamicBorder } from "@earendil-works/pi-coding-agent";
-import { spawnSubagent, extractFinalOutput, type AgentDef } from "./sub-agents";
+import { spawnSubagent, extractFinalOutput, discoverAgents, type AgentDef } from "./sub-agents";
 export type { AgentDef };
 import {
 	Container,
@@ -23,6 +23,7 @@ import {
 	Spacer,
 	type SelectItem,
 } from "@earendil-works/pi-tui";
+import { uiSelect, uiConfirm, uiInput } from "./ui-helpers";
 
 // ── Types ────────────────────────────────────────────────────
 
@@ -53,31 +54,67 @@ export interface PRDResult {
 
 // ── Output dirs ──────────────────────────────────────────────
 
-const DEV_OUTPUT_DIR = "pi-dev-output";
+const DEV_OUTPUT_DIR = ".pi-dev-output";
 const GRILL_DIRNAME = "pi-grill";
+const GRILL_ANSWERS_DIRNAME = "answers";
+const GRILL_QUESTIONS_DIRNAME = "questions";
 const PRD_DIRNAME = "pi-prd";
 
-/** Ensure an output subdirectory exists and write .gitignore. */
+/** Ensure an output subdirectory exists. */
 function ensureOutputDir(cwd: string, subdir: string): string {
 	const dir = path.join(cwd, DEV_OUTPUT_DIR, subdir);
 	fs.mkdirSync(dir, { recursive: true });
-	const gitignorePath = path.join(cwd, DEV_OUTPUT_DIR, ".gitignore");
-	try {
-		const existing = fs.readFileSync(gitignorePath, "utf-8").trim();
-		if (!existing.includes("*")) {
-			fs.writeFileSync(gitignorePath, "*\n!.gitignore\n");
-		}
-	} catch {
-		fs.writeFileSync(gitignorePath, "*\n!.gitignore\n");
-	}
 	return dir;
 }
 
-/** Generate a safe temp filename for grill output. */
+/** Format current time as YYYYMMDD-HHmm for human-readable timestamps. */
+function formatTimestamp(): string {
+	const now = new Date();
+	const Y = now.getFullYear().toString();
+	const M = (now.getMonth() + 1).toString().padStart(2, "0");
+	const D = now.getDate().toString().padStart(2, "0");
+	const h = now.getHours().toString().padStart(2, "0");
+	const m = now.getMinutes().toString().padStart(2, "0");
+	return `${Y}${M}${D}-${h}${m}`;
+}
+
+/** Generate a safe temp filename for grill output (pi-grill/questions/questions-<id>-<YYYYMMDD-HHmm>.json). */
 function grillOutputPath(cwd: string): string {
-	const dir = ensureOutputDir(cwd, GRILL_DIRNAME);
+	const dir = ensureOutputDir(cwd, path.join(GRILL_DIRNAME, GRILL_QUESTIONS_DIRNAME));
 	const ts = Date.now().toString(36);
-	return path.join(dir, `questions-${ts}.json`);
+	return path.join(dir, `questions-${ts}-${formatTimestamp()}.json`);
+}
+
+/**
+ * Save the final assembled prompt to a timestamped answer file (pi-grill/answers/answer-<id>-<YYYYMMDD-HHmm>.md).
+ * Returns the relative path from cwd (for display in notifications).
+ */
+export function saveAnswerFile(cwd: string, content: string): string {
+	const dir = ensureOutputDir(cwd, path.join(GRILL_DIRNAME, GRILL_ANSWERS_DIRNAME));
+	const ts = Date.now().toString(36);
+	const filename = `answer-${ts}-${formatTimestamp()}.md`;
+	fs.writeFileSync(path.join(dir, filename), content, "utf-8");
+	return path.join(DEV_OUTPUT_DIR, GRILL_DIRNAME, GRILL_ANSWERS_DIRNAME, filename);
+}
+
+/**
+ * Find the most recent answer backup file and read its content.
+ * Returns undefined if no backup exists or read fails.
+ * Now reads from pi-grill/answers/ subdirectory.
+ */
+export function recoverFromBackup(cwd: string): string | undefined {
+	const dir = path.join(cwd, DEV_OUTPUT_DIR, GRILL_DIRNAME, GRILL_ANSWERS_DIRNAME);
+	try {
+		if (!fs.existsSync(dir)) return undefined;
+		const files = fs.readdirSync(dir)
+			.filter(f => f.startsWith("answer-") && f.endsWith(".md"))
+			.map(f => ({ name: f, mtime: fs.statSync(path.join(dir, f)).mtimeMs }))
+			.sort((a, b) => b.mtime - a.mtime);
+		if (files.length === 0) return undefined;
+		return fs.readFileSync(path.join(dir, files[0].name), "utf-8");
+	} catch {
+		return undefined;
+	}
 }
 
 /** Generate a safe PRD filename. */
@@ -107,96 +144,81 @@ function writeToolPromptSuffix(outputFilePath: string): string {
 		"Instead, use the `write` tool to save the questions to a file.",
 		`Write to this exact path: ${outputFilePath}`,
 		"",
-		"The file content must be a valid JSON object with a single key \"questions\":",
-		'```json',
+		"The file content must be a **valid** JSON object conforming to this JSON Schema (Draft-07):",
+		"",
+		"```json",
+		'{',
+		'  "$schema": "http://json-schema.org/draft-07/schema#",',
+		'  "type": "object",',
+		'  "required": ["questions"],',
+		'  "properties": {',
+		'    "questions": {',
+		'      "type": "array",',
+		'      "items": {',
+		'        "type": "object",',
+		'        "required": ["id", "question", "options"],',
+		'        "properties": {',
+		'          "id": { "type": "integer" },',
+		'          "question": { "type": "string" },',
+		'          "options": {',
+		'            "type": "array",',
+		'            "items": { "type": "string" },',
+		'            "minItems": 1',
+		'          }',
+		'        }',
+		'      }',
+		'    }',
+		'  }',
+		'}',
+		"```",
+		"",
+		"### \u26a0\ufe0f CRITICAL: String escaping rules",
+		"",
+		"Every string value (question text, option text) MUST be valid JSON-escaped:",
+		"- Double quotes inside text \u2192 \\\"",
+		"- Newlines \u2192 \\n",
+		"- Backslashes \u2192 \\\\",
+		"- Tabs \u2192 \\t",
+		"",
+		"### \u2705 Self-review before writing",
+		"",
+		"Before calling the `write` tool, mentally validate your JSON.",
+		"Check that all strings are properly escaped and the structure matches the schema above.",
+		"If you are unsure, write a quick test with `bash` (e.g. `node -e \"JSON.parse(...)\"`).",
+		"",
+		"Example output:",
+		"```json",
 		'{',
 		'  "questions": [',
 		'    {',
 		'      "id": 1,',
-		'      "question": "问题文本?",',
-		'      "options": ["选项A", "选项B", "选项C"]',
+		'      "question": "\\u9879\\u76ee\\u662f\\u5426\\u5b58\\u5728\\u6a21\\u5757\\u7ed3\\u6784\\uff1f",',
+		'      "options": [',
+		'        "\\u5df2\\u5b58\\u5728\\uff0c\\u4f8b\\u5982 src/controller/example.rs",',
+		'        "\\u4ece\\u96f6\\u521b\\u5efa"',
+		'      ]',
+		'    },',
+		'    {',
+		'      "id": 2,',
+		'      "question": "\\u4ed6\\u8bf4\\u201c\\u8fd9\\u4e2a\\u4e0d\\u884c\\u201d\\uff0c\\u8be5\\u5982\\u4f55\\u5904\\u7406\\uff1f",',
+		'      "options": [',
+		'        "\\u5ffd\\u7565",',
+		'        "\\u4fee\\u590d"',
+		'      ]',
 		'    }',
 		'  ]',
 		'}',
-		'```',
+		"```",
 		"",
 		"After writing, you may include a brief summary in your chat response.",
 		"But the JSON MUST be in the file, NOT in the chat.",
 	].join("\n");
 }
 
-const GRILL_AGENT_DEF: AgentDef = {
-	name: "grill-agent",
-	description: "Design review agent — interviews the developer about a feature plan",
-	tools: ["read", "bash", "write"],
-	systemPrompt: [
-		"You are an expert design reviewer. Interview the developer relentlessly about every aspect of the feature plan until reaching shared understanding.",
-		"",
-		"## Rules",
-		"- For EACH question, provide recommended answer OPTIONS (a/b/c format) that the user can pick from",
-		"- Walk down each branch of the design tree, resolving dependencies between decisions one-by-one",
-		"- Be specific — refer to actual code modules, file paths, and architecture decisions",
-		"- Explore the codebase (use read/bash tools) when a question can be answered by looking at existing code",
-		"- Ask questions about: architecture, data flow, edge cases, security, testing, module boundaries, dependencies, error handling, performance, scalability",
-		"- If terminology conflicts with existing project glossary (CONTEXT.md), call it out",
-		"- Sharpen fuzzy language — propose precise canonical terms",
-		"- Stress-test scenarios with specific edge cases",
-		"- Cross-reference with existing code — surface contradictions",
-		"",
-		"## Quantity",
-		"Ask as many questions as needed to thoroughly review the design.",
-		"Do not artificially limit the number — cover architecture, data flow, edge cases, security, testing, module boundaries, dependencies, error handling, and more.",
-		"Typically 15-40 questions for a moderate feature.",
-		"",
-		"## Language",
-		"Questions and options should be in the same language as the feature request (default: Chinese).",
-	].join("\n"),
-	timeoutMs: 300_000, // 5 min
-};
-
-const PRD_AGENT_DEF: AgentDef = {
-	name: "prd-agent",
-	description: "PRD writer — synthesizes a PRD from conversation context",
-	tools: ["read", "bash"],
-	systemPrompt: [
-		"You are an expert product spec writer.",
-		"Your task is to create a PRD from the provided conversation context.",
-		"",
-		"## Rules",
-		"- Do NOT ask any questions — just synthesize what you already know",
-		"- Explore the repo to understand the current state of the codebase",
-		"- Use the project's domain vocabulary throughout",
-		"- Use the template below and output ONLY the Markdown content (no JSON wrapper, no preamble)",
-		"",
-		"## Template",
-		"",
-		"# {Feature Name} — PRD",
-		"",
-		"## Problem Statement",
-		"The problem that the user is facing, from the user's perspective.",
-		"",
-		"## Solution",
-		"The solution to the problem, from the user's perspective.",
-		"",
-		"## User Stories",
-		"A numbered list of user stories:",
-		"1. As an <actor>, I want a <feature>, so that <benefit>",
-		"",
-		"## Implementation Decisions",
-		"A list of implementation decisions including modules to build/modify, architectural decisions, schema changes, API contracts.",
-		"Do NOT include specific file paths or code snippets (may become outdated).",
-		"",
-		"## Testing Decisions",
-		"A description of what makes a good test, which modules will be tested, prior art.",
-		"",
-		"## Out of Scope",
-		"Things explicitly out of scope.",
-		"",
-		"## Further Notes",
-		"Any further notes about the feature.",
-	].join("\n"),
-	timeoutMs: 300_000,
-};
+// ── Default agent definitions (loaded from agents/ directory) ────
+
+const _defaultGrillAgent = discoverAgents().find(a => a.name === "dev-grill-agent")!;
+const _defaultPrdAgent = discoverAgents().find(a => a.name === "dev-prd-agent")!;
 
 // ── File-based question extraction ───────────────────────────
 
@@ -220,7 +242,13 @@ function readQuestionsFromFile(filePath: string): GrillQuestion[] {
 				question: q.question,
 				options: q.options,
 			}));
-	} catch {
+	} catch (e) {
+		// Log parse errors for debugging (development feedback)
+		try {
+			const content = fs.existsSync(filePath) ? fs.readFileSync(filePath, "utf-8").slice(0, 1000) : "(file not found)";
+			console.error(`[grill-me-agent] JSON parse error in ${filePath}:`, (e as Error).message);
+			console.error(`[grill-me-agent] First 1000 chars:`, content);
+		} catch { /* ignore secondary errors */ }
 		return [];
 	}
 }
@@ -339,16 +367,17 @@ export async function runGrillPhase(
 		enhancedPrompt: assembledPrompt,
 	};
 
-	const agentDef = options?.agentDef ?? GRILL_AGENT_DEF;
+	const agentDef = options?.agentDef ?? _defaultGrillAgent;
 	const confirmTitle = options?.title ?? "🔍 设计方案评审";
-	const confirmDesc = options?.description ?? "是否进入设计评审 (Grill) 模式？\nAI 会从架构、数据流、边界条件、安全等多个维度挑战你的设计。";
+	const confirmDesc = options?.description ?? "AI 会从架构、数据流、边界条件、安全等多个维度挑战你的设计。";
 	const qTitlePrefix = options?.questionTitle ?? "设计方案评审";
-	const loaderLabel = options?.loaderLabel ?? "🧠 AI 正在分析代码并生成评审问题...";
+	const loaderLabel = options?.loaderLabel ?? "🧠 AI 子代理正在分析代码并生成评审问题...";
 
 	// ── Step 1: Confirm entering grill mode ──────────────────
-	const enterGrill = await ctx.ui.confirm(confirmTitle, confirmDesc);
+	const enterGrill = await uiConfirm(ctx, confirmTitle, confirmDesc);
 	if (!enterGrill) {
-		return { ...defaultResult, cancelled: true };
+		// Skip grill but continue the workflow (not a cancellation)
+		return defaultResult;
 	}
 
 	// ── Step 2: Prepare output file + enhanced prompt ─────────
@@ -369,12 +398,13 @@ export async function runGrillPhase(
 			ctx.cwd,
 			loader.signal,
 			undefined,
-			(progress) => { loader.setText(`🧠 ${progress.slice(0, 60)}`); },
+			(progress) => {
+					const inner = (loader as unknown as { loader?: { setText?: (t: string) => void } }).loader;
+					inner?.setText?.(`🧠 ${progress.slice(0, 60)}`);
+				},
 		)
 			.then((result) => {
-				// Primary: read from file (sub-agent wrote via `write` tool)
 				let qs = readQuestionsFromFile(outputFilePath);
-				// Fallback: parse from NDJSON response text
 				if (qs.length === 0) {
 					const output = extractFinalOutput(result.output);
 					qs = parseGrillQuestions(output);
@@ -382,7 +412,6 @@ export async function runGrillPhase(
 				done(qs);
 			})
 			.catch(() => {
-				// On error, still try to read file (sub-agent may have written before error)
 				const qs = readQuestionsFromFile(outputFilePath);
 				done(qs);
 			});
@@ -390,12 +419,21 @@ export async function runGrillPhase(
 		return loader;
 	});
 
-	// ── Step 4: Clean up temp file ───────────────────────────
-	try { fs.unlinkSync(outputFilePath); } catch { /* ignore */ }
-
-	// ── Step 4b: Retry dialog if no questions generated ──────
+	// ── Step 4: Retry dialog if no questions generated ──────
 	if (questions.length === 0) {
-		const choice = await ctx.ui.select(
+		let failedFileContent = "";
+		let parseErrorMsg = "";
+		try {
+			if (fs.existsSync(outputFilePath)) {
+				failedFileContent = fs.readFileSync(outputFilePath, "utf-8").slice(0, 2000);
+				JSON.parse(failedFileContent);
+			}
+		} catch (e) {
+			parseErrorMsg = (e as Error).message;
+		}
+
+		const choice = await uiSelect(
+			ctx,
 			"⚠️ AI 未能成功生成评审问题",
 			[
 				"🔄 重新尝试生成评审问题",
@@ -407,8 +445,29 @@ export async function runGrillPhase(
 		switch (choice) {
 			case "🔄 重新尝试生成评审问题": {
 				const retryPath = grillOutputPath(ctx.cwd);
-				const retryPrompt = [assembledPrompt, writeToolPromptSuffix(retryPath)].join("\n\n");
-				questions = await ctx.ui.custom<GrillQuestion[]>((tui, theme, _kb, done) => {
+				const errorFeedback = parseErrorMsg
+					? [
+						"",
+						"### ⚠️ Previous attempt had JSON errors — fix them now",
+						"",
+						`The previous attempt wrote to \`${outputFilePath}\` but the JSON was invalid.`,
+						"",
+						`JSON parse error: ${parseErrorMsg}`,
+						"",
+						"Invalid file content (first 2000 chars):",
+						"```",
+						failedFileContent.slice(0, 1000),
+						"```",
+						"",
+						"Please write valid JSON to the new path below. Make sure all strings are properly JSON-escaped.",
+					].join("\n")
+					: "";
+				const retryPrompt = [
+					assembledPrompt,
+					writeToolPromptSuffix(retryPath),
+					errorFeedback,
+				].filter(Boolean).join("\n\n");
+				const retryQuestions = await ctx.ui.custom<GrillQuestion[]>((tui, theme, _kb, done) => {
 					const loader = new BorderedLoader(tui, theme, loaderLabel);
 					loader.onAbort = () => done([]);
 					spawnSubagent(agentDef, retryPrompt, ctx.cwd, loader.signal, undefined)
@@ -420,38 +479,91 @@ export async function runGrillPhase(
 						.catch(() => done([]));
 					return loader;
 				});
-				try { fs.unlinkSync(retryPath); } catch { /* ignore */ }
-				if (questions.length === 0) {
-					ctx.ui.notify("⚠️ 再次尝试仍然失败，跳过 Grill 阶段", "warning");
+				if (retryQuestions.length === 0) {
 					return defaultResult;
 				}
-				break;
+				// Replace questions with retry results (with back support)
+				const pairs: Array<{ question: string; answer: string }> = [];
+				let rIdx = 0;
+				while (rIdx >= 0 && rIdx < retryQuestions.length) {
+					const q = retryQuestions[rIdx]!;
+					const previousAnswer = pairs[rIdx]?.answer;
+					const answer = await showQuestionTUI(ctx, q, rIdx + 1, retryQuestions.length, qTitlePrefix,
+						rIdx > 0, previousAnswer);
+					if (answer === null) {
+						return { ...defaultResult, cancelled: true, pairs };
+					}
+					if (answer === "__BACK__") {
+						if (rIdx > 0) {
+							rIdx--;
+							continue;
+						}
+						return { ...defaultResult, cancelled: true, pairs };
+					}
+					// Overwrite if re-answering, otherwise append
+					if (rIdx < pairs.length) {
+						pairs[rIdx] = { question: q.question, answer };
+					} else {
+						pairs.push({ question: q.question, answer });
+					}
+					rIdx++;
+				}
+				const qaBlock = pairs
+					.map((p, i) => `[评审问题 ${i + 1}]\n问题: ${p.question}\n回答: ${p.answer}`)
+					.join("\n\n");
+				const finalEnhancedPrompt = [
+					assembledPrompt,
+					"",
+					"---",
+					"## 设计评审记录",
+					"",
+					"以下是在开发前进行的设计评审问答，所有决策已确认：",
+					"",
+					qaBlock,
+				].join("\n");
+				return {
+					cancelled: false,
+					pairs,
+					enhancedPrompt: finalEnhancedPrompt,
+				};
 			}
 			case "⏭️ 跳过 Grill，直接发送 Prompt":
-				ctx.ui.notify("⏭️ 已跳过 Grill 阶段", "info");
 				return defaultResult;
 			case "❌ 取消 (Esc)":
 			default:
-				ctx.ui.notify("❌ 操作已取消", "warning");
 				return { ...defaultResult, cancelled: true };
 		}
 	}
 
-	ctx.ui.notify(`✅ AI 生成了 ${questions.length} 个评审问题`, "success");
-
-	// ── Step 5: TUI — present questions one by one ───────────
+	// ── Step 5: TUI — present questions one by one (with back support) ──
 	const pairs: Array<{ question: string; answer: string }> = [];
+	let qIdx = 0;
 
-	for (let idx = 0; idx < questions.length; idx++) {
-		const q = questions[idx];
-		const answer = await showQuestionTUI(ctx, q, idx + 1, questions.length, qTitlePrefix);
+	while (qIdx >= 0 && qIdx < questions.length) {
+		const q = questions[qIdx]!;
+		const previousAnswer = pairs[qIdx]?.answer;
+		const answer = await showQuestionTUI(ctx, q, qIdx + 1, questions.length, qTitlePrefix,
+			qIdx > 0, previousAnswer);
 
 		if (answer === null) {
-			ctx.ui.notify("❌ 评审已取消", "warning");
 			return { ...defaultResult, cancelled: true, pairs };
 		}
 
-		pairs.push({ question: q.question, answer });
+		if (answer === "__BACK__") {
+			if (qIdx > 0) {
+				qIdx--;
+				continue;
+			}
+			return { ...defaultResult, cancelled: true, pairs };
+		}
+
+		// Overwrite if re-answering (back then forward), otherwise append
+		if (qIdx < pairs.length) {
+			pairs[qIdx] = { question: q.question, answer };
+		} else {
+			pairs.push({ question: q.question, answer });
+		}
+		qIdx++;
 	}
 
 	// ── Step 6: Assemble enhanced prompt ─────────────────────
@@ -470,8 +582,6 @@ export async function runGrillPhase(
 		qaBlock,
 	].join("\n");
 
-	ctx.ui.notify(`✅ 评审完成，共 ${pairs.length} 道问题`, "success");
-
 	return {
 		cancelled: false,
 		pairs,
@@ -486,17 +596,32 @@ async function showQuestionTUI(
 	currentIndex: number,
 	totalCount: number,
 	titlePrefix = "设计方案评审",
+	backable = false,
+	previousAnswer?: string,
 ): Promise<string | null> {
 	const selectItems: SelectItem[] = q.options.map((opt, i) => ({
 		value: `opt-${i}`,
-		label: `(${String.fromCharCode(97 + i)}) ${opt}`,
+		label: opt === previousAnswer
+			? `(${String.fromCharCode(97 + i)}) ${opt} - 上次选择`
+			: `(${String.fromCharCode(97 + i)}) ${opt}`,
 	}));
+
+	const customLabel = previousAnswer && !q.options.includes(previousAnswer)
+		? `✏️  自定义输入 - 上次选择`
+		: `✏️  自定义输入`;
 	selectItems.push({
 		value: "__custom__",
-		label: "✏️  自定义输入",
+		label: customLabel,
 		description: "输入你自己的回答，不受选项限制",
 	});
 
+	if (backable && currentIndex > 1) {
+		selectItems.push({
+			value: "__back__",
+			label: "← 返回上一题",
+		});
+	}
+
 	const title = `${titlePrefix} (问题 ${currentIndex}/${totalCount})`;
 
 	const value = await ctx.ui.custom<string | null>((tui, theme, _kb, done) => {
@@ -520,8 +645,11 @@ async function showQuestionTUI(
 		container.addChild(selectList);
 
 		container.addChild(new Spacer(1));
+		const hint = backable && currentIndex > 1
+			? "  ↑↓ 导航 • Enter 选择 • 选择←返回上一题 • Esc 取消全部评审"
+			: "  ↑↓ 导航 • Enter 选择 • Esc 取消全部评审";
 		container.addChild(
-			new Text(theme.fg("dim", "  ↑↓ 导航 • Enter 选择 • Esc 取消全部评审"), 0, 0),
+			new Text(theme.fg("dim", hint), 0, 0),
 		);
 		container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
 
@@ -535,13 +663,18 @@ async function showQuestionTUI(
 		};
 	});
 
+	if (value === "__back__") return "__BACK__";
 	if (value === null) return null;
 	if (value === "__custom__") {
-		const custom = await ctx.ui.input("✏️ 自定义回答", {
-			placeholder: "输入你的回答内容（Esc 取消本题，回到选项）",
-			required: false,
-		});
-		if (custom === undefined) return showQuestionTUI(ctx, q, currentIndex, totalCount);
+		const custom = await uiInput(ctx, "✏️ 自定义回答",
+			previousAnswer && !q.options.includes(previousAnswer)
+				? `(上次: ${previousAnswer.slice(0, 60)})`
+				: "输入你的回答内容（Esc 取消本题，回到选项）",
+			false, true,
+			previousAnswer && !q.options.includes(previousAnswer) ? previousAnswer : "",
+		);
+		if (custom === "__BACK__") return "__BACK__";
+		if (custom === undefined) return showQuestionTUI(ctx, q, currentIndex, totalCount, titlePrefix, backable, previousAnswer);
 		return custom.trim() || "(空)";
 	}
 
@@ -555,7 +688,7 @@ async function showQuestionTUI(
  * Run the PRD phase:
  * 1. Ask user if they want to create a PRD
  * 2. Call sub-agent → gets PRD Markdown
- * 3. Save to pi-dev-output/pi-prd/<name>.md
+ * 3. Save to .pi-dev-output/pi-prd/<name>.md
  * 4. Ask if user wants to start development
  */
 export async function runPRDPhase(
@@ -564,9 +697,10 @@ export async function runPRDPhase(
 	pi: ExtensionAPI,
 	ctx: ExtensionCommandContext,
 ): Promise<PRDResult | null> {
-	const wantPrd = await ctx.ui.confirm(
+	const wantPrd = await uiConfirm(
+		ctx,
 		"📋 创建 PRD",
-		"是否为此功能创建 PRD 文档？\nPRD 将保存到 pi-dev-output/pi-prd/ 目录。",
+		"PRD 将保存到 .pi-dev-output/pi-prd/ 目录。",
 	);
 	if (!wantPrd) return null;
 
@@ -582,7 +716,7 @@ export async function runPRDPhase(
 			context,
 		].join("\n");
 
-		spawnSubagent(PRD_AGENT_DEF, prdTask, ctx.cwd, loader.signal)
+		spawnSubagent(_defaultPrdAgent, prdTask, ctx.cwd, loader.signal)
 			.then((result) => {
 				const output = extractFinalOutput(result.output);
 				done(output && output.length >= 50 ? output : null);
@@ -593,7 +727,6 @@ export async function runPRDPhase(
 	});
 
 	if (!prdContent) {
-		ctx.ui.notify("⚠️ PRD 生成失败", "error");
 		return null;
 	}
 
@@ -602,7 +735,6 @@ export async function runPRDPhase(
 	const filePath = path.join(DEV_OUTPUT_DIR, PRD_DIRNAME, filename);
 	const fullPath = path.join(prdDir, filename);
 	fs.writeFileSync(fullPath, prdContent, "utf-8");
-	ctx.ui.notify(`✅ PRD 已保存到 ${filePath}`, "success");
 
 	await askDevelopmentStart(pi, ctx, prdContent, filePath);
 	return { content: prdContent, filePath };
@@ -614,7 +746,8 @@ async function askDevelopmentStart(
 	prdContent: string,
 	prdFilePath: string,
 ): Promise<void> {
-	const choice = await ctx.ui.select(
+	const choice = await uiSelect(
+		ctx,
 		"🚀 是否开始开发？",
 		[
 			"是 — 根据 PRD 开始开发",
@@ -638,18 +771,13 @@ async function askDevelopmentStart(
 				"",
 				"请按照上述 PRD 逐步实现。先分析代码库结构，给出实施计划，确认后再编写代码。",
 			].join("\n");
-			pi.sendUserMessage(devMsg);
-			ctx.ui.notify("🚀 已发送开发指令给主代理", "success");
+			pi.sendUserMessage(devMsg, { deliverAs: "followUp" });
 			break;
 		}
 		case "否 — 稍后手动开始":
-			ctx.ui.notify(`📋 PRD 已保存在 ${prdFilePath}，可随时手动引用`, "info");
 			break;
 		default: {
-			const customMsg = await ctx.ui.input("✏️ 自定义开发指令", {
-				placeholder: "输入你的开发指令（将结合 PRD 一起发送给主代理）",
-				required: false,
-			});
+			const customMsg = await uiInput(ctx, "✏️ 自定义开发指令", "输入你的开发指令（将结合 PRD 一起发送给主代理）");
 			if (customMsg === undefined) return askDevelopmentStart(pi, ctx, prdContent, prdFilePath);
 			const finalMsg = customMsg.trim()
 				? [
@@ -668,8 +796,7 @@ async function askDevelopmentStart(
 					"--- PRD 全文 ---",
 					prdContent,
 				].join("\n");
-			pi.sendUserMessage(finalMsg);
-			ctx.ui.notify("🚀 已发送自定义开发指令给主代理", "success");
+			pi.sendUserMessage(finalMsg, { deliverAs: "followUp" });
 			break;
 		}
 	}