Jeffery
98 lines
3.5 KiB
JavaScript
98 lines
3.5 KiB
JavaScript
import fs from 'fs';
|
||
import path from 'path';
|
||
import { fileURLToPath } from 'url';
|
||
import yaml from 'js-yaml';
|
||
import { warn } from './log.js';
|
||
|
||
const ROLES_DIR = path.join(fileURLToPath(import.meta.url), '..', 'prompts', 'roles');
|
||
|
||
/**
|
||
* 解析單一角色 .md 檔：前置 YAML frontmatter（徽章、代表色、面向、個性等）＋ 本文（審查重點）。
|
||
* 回傳合併後的角色物件：{ name, side, focus, badge, color, personality, body }。
|
||
*/
|
||
export function parseRoleFile(content) {
|
||
const normalized = content.replace(/\r\n/g, '\n');
|
||
const match = normalized.match(/^---\n([\s\S]*?)\n---\n?([\s\S]*)$/);
|
||
if (!match) throw new Error('角色檔缺少 frontmatter');
|
||
const meta = yaml.load(match[1]) || {};
|
||
return { ...meta, body: match[2].trim() };
|
||
}
|
||
|
||
let cachedRoles = null;
|
||
|
||
/**
|
||
* 讀取並解析所有角色 .md，結果快取於模組層級（單次程序生命週期內檔案不變）。
|
||
* 單一檔案解析失敗（壞 YAML、缺 frontmatter 等）時記錄警告並略過，不讓整個流程崩潰。
|
||
*/
|
||
function readRoleFiles() {
|
||
if (cachedRoles) return cachedRoles;
|
||
const roles = [];
|
||
for (const f of fs.readdirSync(ROLES_DIR).filter(f => f.endsWith('.md')).sort()) {
|
||
try {
|
||
roles.push(parseRoleFile(fs.readFileSync(path.join(ROLES_DIR, f), 'utf8')));
|
||
} catch (e) {
|
||
warn(`角色檔解析失敗，已略過: ${f}（${e.message}）`);
|
||
}
|
||
}
|
||
cachedRoles = roles;
|
||
return cachedRoles;
|
||
}
|
||
|
||
/**
|
||
* 載入攻擊方角色（Step2 產生 findings 用），依檔名排序。
|
||
* 防守方（如 Paladin）不在此列，裁決邏輯由去重/誤報過濾流程承擔。
|
||
*/
|
||
export function loadRoles() {
|
||
return readRoleFiles().filter(r => r.side === 'attack');
|
||
}
|
||
|
||
/** 依 frontmatter name 取得單一角色（不分大小寫），找不到回傳 null。 */
|
||
export function loadRole(name) {
|
||
const target = String(name).toLowerCase();
|
||
return readRoleFiles().find(r => String(r.name).toLowerCase() === target) || null;
|
||
}
|
||
|
||
/**
|
||
* 由角色定義組出攻擊方的 system prompt：
|
||
* 套用其個性與審查重點本文，並要求以固定 JSON 陣列格式回傳 findings。
|
||
*/
|
||
export function buildAnalysisPrompt(role) {
|
||
return [
|
||
`你是 ${role.badge ? role.badge + ' ' : ''}${role.name}，負責「${role.focus || '綜合'}」面向的程式碼審查（攻擊方）。`,
|
||
role.personality ? `個性：${role.personality}` : '',
|
||
'',
|
||
role.body,
|
||
'',
|
||
'---',
|
||
'',
|
||
'請分析以下 Git Diff，只針對新增/修改處，依你的面向找出所有問題。',
|
||
'回傳 JSON 陣列，每個問題格式如下：',
|
||
'{',
|
||
' "level": "critical|warning|info",',
|
||
` "role": "${role.name}",`,
|
||
' "location": "檔案路徑:行號 或 檔案路徑",',
|
||
' "suggestion": "繁體中文（台灣用語）的具體修改建議"',
|
||
'}',
|
||
'',
|
||
'等級定義：',
|
||
'- critical：嚴重且應立即處理的問題',
|
||
'- warning：建議修正的問題',
|
||
'- info：可選的改善建議',
|
||
'',
|
||
'只回傳 JSON 陣列，不要有其他文字。如果沒有問題，回傳空陣列 []。',
|
||
].filter(l => l !== '').join('\n');
|
||
}
|
||
|
||
export function getRoleIntro(roles) {
|
||
const lines = [
|
||
'## 🤖 AI Code Review 團隊', '',
|
||
'| 👤 角色 | 🎯 面向 | 🧠 個性 |',
|
||
'|--------|--------|--------|',
|
||
];
|
||
for (const r of roles) {
|
||
const badge = r.badge ? `${r.badge} ` : '';
|
||
lines.push(`| **${badge}${r.name}** | ${r.focus || ''} | ${r.personality || ''} |`);
|
||
}
|
||
return lines.join('\n');
|
||
}
|