feat: 提示词拼装使用动态命令前缀

This commit is contained in:
2026-06-09 20:37:10 +08:00
parent 2feea7a74f
commit 78caec6449
2 changed files with 52 additions and 6 deletions

View File

@@ -5,6 +5,7 @@ import type { RuneConfig } from "../types.ts";
import { CommandError } from "../cli/errors.ts";
import { getChangeDir } from "./config.ts";
import { parseTasks } from "./task-parser.ts";
import { applyCommandPrefix, getPmPrefix } from "./pm.ts";
export function assembleDiscussPrompt(config: RuneConfig): string {
const discuss = config.stages.discuss;
@@ -12,7 +13,7 @@ export function assembleDiscussPrompt(config: RuneConfig): string {
throw new CommandError("讨论阶段未配置", {
hint: "请在 .rune/config.yaml 中配置 stages.discuss",
});
return discuss.prompt;
return applyCommandPrefix(discuss.prompt, config);
}
export async function assemblePlanPrompt(
@@ -69,7 +70,7 @@ export async function assemblePlanPrompt(
}
parts.push(`\n请将文档写入目录${changeDir}`);
return parts.join("\n");
return applyCommandPrefix(parts.join("\n"), config);
}
export async function assembleBuildPrompt(
@@ -91,8 +92,9 @@ export async function assembleBuildPrompt(
try {
taskContent = await readFile(taskPath, "utf-8");
} catch {
const prefix = getPmPrefix(config);
throw new CommandError(`变更 "${changeName}" 尚未完成规划task.md 不存在`, {
hint: `请先完成规划阶段:rune plan ${changeName} 生成所有规划文档`,
hint: `请先完成规划阶段:${prefix} plan ${changeName} 生成所有规划文档`,
});
}
@@ -114,7 +116,7 @@ export async function assembleBuildPrompt(
}
parts.push(`\n请从第一个待执行任务开始。完成后更新 ${taskPath} 中的 checkbox。`);
return parts.join("\n");
return applyCommandPrefix(parts.join("\n"), config);
}
export async function assembleArchivePrompt(
@@ -154,5 +156,5 @@ export async function assembleArchivePrompt(
}
parts.push(archive.prompt);
return parts.join("\n");
return applyCommandPrefix(parts.join("\n"), config);
}

View File

@@ -149,7 +149,7 @@ describe("assembleBuildPrompt", () => {
} catch (e: any) {
expect(e.message).toContain("尚未完成规划");
expect(e.message).toContain("nonexistent");
expect(e.hint).toContain("rune plan");
expect(e.hint).toContain("plan nonexistent");
}
});
});
@@ -163,3 +163,47 @@ describe("assembleArchivePrompt", () => {
expect(prompt).toContain("归档");
});
});
describe("命令前缀替换", () => {
it("assembleDiscussPrompt 替换 rune 为配置前缀", () => {
const config: RuneConfig = {
stages: { discuss: { prompt: "执行 rune status 查看状态" } },
metadata: { command: "bunx @lanyuanxiaoyao/rune" },
};
const prompt = assembleDiscussPrompt(config);
expect(prompt).toContain("bunx @lanyuanxiaoyao/rune status");
expect(prompt).not.toContain("执行 rune ");
});
it("assembleDiscussPrompt 无配置时追加降级说明", () => {
const config: RuneConfig = {
stages: { discuss: { prompt: "执行 rune status 查看" } },
};
const prompt = assembleDiscussPrompt(config);
expect(prompt).toContain("bunx @lanyuanxiaoyao/rune status");
expect(prompt).toContain("如果没有安装 bun");
});
it("assembleBuildPrompt 错误提示使用动态前缀", async () => {
const config: RuneConfig = {
stages: { build: { prompt: "构建阶段" } },
metadata: { command: "pnpx @lanyuanxiaoyao/rune" },
};
try {
await assembleBuildPrompt(config, TMP_DIR, "nonexistent-build");
expect.unreachable();
} catch (e: any) {
expect(e.hint).toContain("pnpx @lanyuanxiaoyao/rune plan nonexistent-build");
}
});
it("不替换 /rune- 形式", () => {
const config: RuneConfig = {
stages: { discuss: { prompt: "使用 /rune-plan 进入规划,然后 rune build 构建" } },
metadata: { command: "bunx @lanyuanxiaoyao/rune" },
};
const prompt = assembleDiscussPrompt(config);
expect(prompt).toContain("/rune-plan");
expect(prompt).toContain("bunx @lanyuanxiaoyao/rune build");
});
});