From e5bc3adc620d240414929c54832064c5b64bd64a Mon Sep 17 00:00:00 2001
From: lanyuanxiaoyao <lanyuanxiaoyao@gmail.com>
Date: Thu, 11 Jun 2026 16:40:13 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20task=20=E5=92=8C=20build=20?=
 =?UTF-8?q?=E9=98=B6=E6=AE=B5=E6=94=B9=E7=94=A8=20PromptBuilder=20?=
 =?UTF-8?q?=E7=BB=84=E8=A3=85=E6=8F=90=E7=A4=BA=E8=AF=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/core/assembler.ts | 35 ++++++++++++++++++-----------------
 1 file changed, 18 insertions(+), 17 deletions(-)

diff --git a/src/core/assembler.ts b/src/core/assembler.ts
index 50e8152..b554a00 100644
--- a/src/core/assembler.ts
+++ b/src/core/assembler.ts
@@ -106,23 +106,22 @@ export async function assembleTaskPrompt(
     });
   }
 
-  const parts: string[] = [];
-  parts.push(`# 任务拆解阶段：${changeName}\n`);
-  parts.push(task.prompt);
+  const builder = new PromptBuilder();
+
+  builder.head(`# 任务拆解阶段：${changeName}`);
+  builder.prompt(task.prompt);
 
   const planDocPaths = plan.documents.map((d) => join(changeDir, `${d.name}.md`));
-  parts.push(`\n请先读取以下规划文档：`);
-  for (const p of planDocPaths) {
-    parts.push(`- ${p}`);
-  }
+  builder.context(`请先读取以下规划文档：\n${planDocPaths.map((p) => `- ${p}`).join("\n")}`);
 
   const taskPath = join(changeDir, "task.md");
   if (existsSync(taskPath)) {
-    parts.push(`\ntask.md 已存在，请先读取 ${taskPath} 查看已有内容，在此基础上修订。`);
+    builder.context(`task.md 已存在，请先读取 ${taskPath} 查看已有内容，在此基础上修订。`);
   }
 
-  parts.push(`\n请将任务列表写入 ${taskPath}`);
-  return applyCommandPrefix(parts.join("\n"), config);
+  builder.guide(`请将任务列表写入 ${taskPath}`);
+
+  return applyCommandPrefix(builder.build(), config);
 }
 
 export async function assembleBuildPrompt(
@@ -168,14 +167,16 @@ export async function assembleBuildPrompt(
     return `所有任务已完成。变更 "${changeName}" 可以归档。`;
   }
 
-  const parts: string[] = [];
-  parts.push(`# 构建阶段：${changeName}\n`);
-  parts.push(build.prompt);
-  parts.push(`\n请先读取 ${taskPath} 查看任务列表。`);
-  parts.push(`当前有 ${pendingTasks.length} 个待执行任务，从第一个未完成的任务开始。`);
-  parts.push(`完成后更新 ${taskPath} 中的 checkbox。`);
+  const builder = new PromptBuilder();
 
-  return applyCommandPrefix(parts.join("\n"), config);
+  builder.head(`# 构建阶段：${changeName}`);
+  builder.prompt(build.prompt);
+  builder.context(
+    `请先读取 ${taskPath} 查看任务列表。\n当前有 ${pendingTasks.length} 个待执行任务，从第一个未完成的任务开始。`,
+  );
+  builder.guide(`完成后更新 ${taskPath} 中的 checkbox。`);
+
+  return applyCommandPrefix(builder.build(), config);
 }
 
 export async function assembleArchivePrompt(