agentic-flow-kit 2.0.0 → 2.1.0

package/src/models/resolve.mjs

@@ -0,0 +1,312 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { HOSTS, modelRoleForAgent } from '../hosts/registry.mjs';
+import { getField } from '../install/frontmatter.mjs';
+import { probeModels } from './probe.mjs';
+import { knownFor } from './known.mjs';
+
+export const ROLE_NAMES = ['code', 'judgment', 'review'];
+export const LOCAL_REL = '.agents/config/model-roles.local.json';
+const KIT_ROOT = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '..', '..');
+
+// --model 支持两种形式：code=sonnet（全局）、claude:code=sonnet（宿主级）。
+export function parseModelFlags(values = []) {
+  const out = { global: {}, hosts: {} };
+  for (const raw of values) {
+    const s = String(raw).trim();
+    const eq = s.indexOf('=');
+    if (eq <= 0) {
+      const err = new Error(`invalid --model: ${raw}`);
+      err.code = 'BAD_MODEL';
+      err.value = raw;
+      throw err;
+    }
+    const left = s.slice(0, eq).trim();
+    const id = s.slice(eq + 1).trim();
+    const [host, role] = left.includes(':') ? left.split(':', 2) : [null, left];
+    if (!id || !ROLE_NAMES.includes(role)) {
+      const err = new Error(`invalid --model: ${raw}`);
+      err.code = 'BAD_MODEL';
+      err.value = raw;
+      throw err;
+    }
+    if (host) {
+      const key = host.trim().toLowerCase();
+      if (!HOSTS[key]) {
+        const err = new Error(`unknown host in --model: ${host}`);
+        err.code = 'BAD_MODEL';
+        err.value = raw;
+        throw err;
+      }
+      out.hosts[key] = out.hosts[key] || {};
+      out.hosts[key][role] = id;
+    } else {
+      out.global[role] = id;
+    }
+  }
+  return out;
+}
+
+// 升级时本地模型配置可能因 gitignore 丢失；从已安装宿主子代理回收 model 字段，
+// 避免把用户已配置的模型静默改写回 inherit。只在未重新探测时作为兜底。
+export function readInjectedModels({ target, hosts = [] } = {}) {
+  const out = {};
+  for (const host of hosts) {
+    const h = HOSTS[host];
+    if (!h?.agents?.modelField) continue;
+    const dir = path.join(target, h.dir, h.agents.dir);
+    let names;
+    try {
+      names = fs.readdirSync(dir);
+    } catch {
+      continue;
+    }
+    const values = {};
+    for (const name of names) {
+      if (!name.endsWith(h.agents.ext)) continue;
+      let text;
+      try {
+        text = fs.readFileSync(path.join(dir, name), 'utf8');
+      } catch {
+        continue;
+      }
+      const value = getField(text, h.agents.modelField);
+      if (!value || value === h.agents.inheritValue) continue;
+      const role = modelRoleForAgent(name);
+      if (!values[role]) values[role] = value;
+    }
+    if (Object.keys(values).length) out[host] = values;
+  }
+  return out;
+}
+
+export function loadRoleConfig(target) {
+  for (const file of [
+    path.join(target, '.agents/config/model-roles.json'),
+    path.join(KIT_ROOT, 'templates/engine/.agents/config/model-roles.json'),
+  ]) {
+    try {
+      return JSON.parse(fs.readFileSync(file, 'utf8'));
+    } catch {
+      /* try next */
+    }
+  }
+  return { roles: { review: { heterogeneous: true } } };
+}
+
+export function readLocalModelRoles(target) {
+  const file = path.join(target, LOCAL_REL);
+  if (!fs.existsSync(file)) return { exists: false, file, roles: {}, hosts: {} };
+  try {
+    const json = JSON.parse(fs.readFileSync(file, 'utf8'));
+    return { exists: true, file, roles: json.roles || {}, hosts: json.hosts || {}, raw: json };
+  } catch {
+    return { exists: true, file, error: true, roles: {}, hosts: {} };
+  }
+}
+
+function roleValue(roles, role) {
+  const entry = roles?.[role];
+  if (typeof entry === 'string') return entry;
+  return entry?.model || null;
+}
+
+function hostRoleValue(hosts, host, role) {
+  const entry = hosts?.[host];
+  if (!entry) return null;
+  return typeof entry[role] === 'string' ? entry[role] : null;
+}
+
+const ROLE_HINTS = {
+  code: [/haiku/i, /flash/i, /mini/i, /small/i, /coder/i, /sonnet/i, /qwen/i],
+  judgment: [/opus/i, /o3/i, /r1/i, /reason/i, /thinking/i, /pro/i, /max/i, /glm-5/i, /deepseek/i],
+  review: [/opus/i, /pro/i, /max/i, /r1/i, /thinking/i, /glm-5/i, /deepseek/i],
+};
+
+function scoreModel(id, role) {
+  const s = String(id);
+  return (ROLE_HINTS[role] || []).reduce((score, re) => score + (re.test(s) ? 1 : 0), 0);
+}
+
+function unique(list) {
+  return [...new Set(list.filter(Boolean))];
+}
+
+// 按角色推荐：先探测值，再用已核实的内置别名兜底。
+export function recommendRoles({ host, candidates = [], roleConfig = null } = {}) {
+  const detected = candidates.map((c) => (typeof c === 'string' ? c : c.id));
+  const out = {};
+  for (const role of ROLE_NAMES) {
+    const pool = unique([...detected, ...knownFor(host, role)]);
+    let best = null;
+    let bestScore = 0;
+    for (const id of pool) {
+      const score = scoreModel(id, role);
+      if (score > bestScore) {
+        best = id;
+        bestScore = score;
+      }
+    }
+    out[role] = best || pool[0] || roleConfig?.roles?.[role]?.model || null;
+  }
+  if (out.review && out.code && out.review === out.code) {
+    out.review = unique([...detected, ...knownFor(host, 'review')]).find((id) => id !== out.code) || null;
+  }
+  return out;
+}
+
+export function resolveModels({
+  hosts,
+  target = process.cwd(),
+  cli = { global: {}, hosts: {} },
+  tui = {},
+  previous = {},
+  mode = 'recommend',
+  probe = true,
+  allowCli = false,
+  home,
+  env,
+  exec,
+  roleConfig = null,
+} = {}) {
+  const config = roleConfig || loadRoleConfig(target);
+  const local = readLocalModelRoles(target);
+  const warnings = [];
+  const perHost = {};
+  const candidates = {};
+  const sources = new Set();
+
+  for (const host of hosts) {
+    const h = HOSTS[host];
+    if (!h) continue;
+
+    const found = probe && mode !== 'inherit'
+      ? probeModels({ host, target, home, env, allowCli, exec })
+      : [];
+    candidates[host] = found;
+    const recommended = recommendRoles({ host, candidates: found, roleConfig: config });
+    const values = {};
+
+    for (const role of ROLE_NAMES) {
+      const chain = [];
+      const cliHost = hostRoleValue(cli.hosts, host, role);
+      const cliGlobal = cli.global?.[role] || null;
+      const tuiValue = hostRoleValue(tui, host, role);
+      if (tuiValue === '__none__') {
+        values[role] = null;
+        sources.add('tui');
+        continue;
+      }
+      if (cliHost) chain.push([cliHost, 'cli']);
+      if (cliGlobal) chain.push([cliGlobal, 'cli']);
+      if (tuiValue) chain.push([tuiValue, 'tui']);
+      if (mode !== 'inherit') {
+        const localHost = hostRoleValue(local.hosts, host, role);
+        const localGlobal = roleValue(local.roles, role);
+        if (localHost) chain.push([localHost, 'local']);
+        if (localGlobal) chain.push([localGlobal, 'local']);
+        // 未重新探测时，保留已安装子代理的模型；显式探测/推荐优先。
+        const previousHost = !probe ? hostRoleValue(previous, host, role) : null;
+        if (previousHost) chain.push([previousHost, 'previous']);
+        if (probe && recommended[role]) chain.push([recommended[role], 'detected']);
+      }
+      if (chain.length) {
+        values[role] = chain[0][0];
+        sources.add(chain[0][1]);
+      } else {
+        values[role] = h.agents?.inheritValue || null;
+        if (values[role]) sources.add('inherit');
+      }
+    }
+
+    // 独立复核默认要求异构；显式指定不覆盖，只告警。
+    // inherit / null 不是具体模型，不能拿来做异构比较（否则会把 inherit 改成推荐模型）。
+    const concrete = (v) => typeof v === 'string' && v !== 'inherit';
+    if (mode !== 'inherit'
+      && concrete(values.review) && concrete(values.code) && values.review === values.code
+      && config?.roles?.review?.heterogeneous !== false) {
+      // CLI / TUI / 本地配置都是用户显式选择：同模型只告警，不擅自改写。
+      const explicit = Boolean(
+        hostRoleValue(cli.hosts, host, 'review') || cli.global?.review
+        || hostRoleValue(tui, host, 'review')
+        || hostRoleValue(local.hosts, host, 'review') || roleValue(local.roles, 'review'),
+      );
+      if (!explicit) {
+        const alt = unique([...found.map((c) => c.id), ...knownFor(host, 'review')]).find((id) => id !== values.code);
+        if (alt) {
+          values.review = alt;
+          sources.add('detected');
+        } else if (h.agents?.inheritValue) {
+          values.review = h.agents.inheritValue;
+          warnings.push({ host, role: 'review', code: 'REVIEW_SAME_AS_CODE' });
+        } else {
+          values.review = null;
+          warnings.push({ host, role: 'review', code: 'REVIEW_SAME_AS_CODE' });
+        }
+      } else {
+        warnings.push({ host, role: 'review', code: 'REVIEW_SAME_AS_CODE' });
+      }
+    }
+
+    perHost[host] = values;
+  }
+
+  const roles = {};
+  for (const role of ROLE_NAMES) {
+    // 角色摘要不依赖宿主顺序：取第一个真正解析出值的宿主。
+    const hostValue = hosts.map((host) => perHost[host]?.[role]).find(Boolean) || null;
+    const value = cli.global?.[role]
+      || roleValue(local.roles, role)
+      || hostValue
+      || null;
+    roles[role] = { model: value };
+  }
+
+  return {
+    roles,
+    hosts: perHost,
+    source: [...sources].join('+') || 'inherit',
+    warnings,
+    candidates,
+    local,
+  };
+}
+
+export function summarizeModels(models) {
+  if (!models?.hosts) return '-';
+  return Object.entries(models.hosts)
+    .map(([host, roles]) => `${host}: ${ROLE_NAMES.map((r) => `${r}=${roles[r] || 'inherit'}`).join(' ')}`)
+    .join('; ');
+}
+
+export function writeLocalModelRoles(target, models, { overwrite = false } = {}) {
+  if (!models) return { written: false, reason: 'no-models' };
+  const file = path.join(target, LOCAL_REL);
+  if (fs.existsSync(file) && !overwrite) return { written: false, file, reason: 'exists' };
+
+  const roles = {};
+  for (const role of ROLE_NAMES) roles[role] = { model: models.roles?.[role]?.model || null };
+  if (!ROLE_NAMES.some((role) => roles[role].model)) {
+    return { written: false, file, reason: 'empty' };
+  }
+
+  let existing = {};
+  if (fs.existsSync(file)) {
+    try {
+      existing = JSON.parse(fs.readFileSync(file, 'utf8'));
+    } catch {
+      existing = {};
+    }
+  }
+
+  const data = {
+    version: 1,
+    ...existing,
+    roles: { ...(existing.roles || {}), ...roles },
+    hosts: { ...(existing.hosts || {}), ...(models.hosts || {}) },
+  };
+  fs.mkdirSync(path.dirname(file), { recursive: true });
+  fs.writeFileSync(file, `${JSON.stringify(data, null, 2)}\n`, 'utf8');
+  return { written: true, file };
+}

package/src/render/tokens.mjs

@@ -1,8 +1,6 @@
 import { HOSTS } from '../hosts/registry.mjs';
 
-// 模板变量：只替换已知 token，未知 {{...}} 原样保留（避免误伤代码模板字符串）。
-export function buildVars({ target, projectName, stack, language, hosts, kitVersion, installedAt, trackHosts = false }) {
-  const stackDefaults = {
+const STACK_DEFAULTS = {
     dotnet: {
       description: 'ASP.NET Core 后端 + Vue 3 前端 + SQL Server。',
       descriptionEn: 'ASP.NET Core backend + Vue 3 frontend + SQL Server.',
@@ -10,20 +8,66 @@ export function buildVars({ target, projectName, stack, language, hosts, kitVers
       webPort: 5203,
     },
     node: {
-      description: 'Node.js / TypeScript 项目。',
-      descriptionEn: 'A Node.js / TypeScript project.',
+      description: 'Node.js 项目。',
+      descriptionEn: 'A Node.js project.',
+      apiPort: 3000,
+      webPort: 5173,
+    },
+    typescript: {
+      description: 'TypeScript 项目。',
+      descriptionEn: 'A TypeScript project.',
       apiPort: 3000,
       webPort: 5173,
     },
+    rust: {
+      description: 'Rust 项目（Cargo）。',
+      descriptionEn: 'A Rust project (Cargo).',
+      apiPort: 8080,
+      webPort: 5173,
+    },
+    go: {
+      description: 'Go 项目（Go modules）。',
+      descriptionEn: 'A Go project (Go modules).',
+      apiPort: 8080,
+      webPort: 5173,
+    },
+    java: {
+      description: 'Java 项目（Maven 或 Gradle）。',
+      descriptionEn: 'A Java project (Maven or Gradle).',
+      apiPort: 8080,
+      webPort: 5173,
+    },
     empty: {
       description: '项目技术栈按实际填写。',
       descriptionEn: 'Fill in the project stack description.',
       apiPort: 3000,
       webPort: 5173,
     },
-  };
-  const d = stackDefaults[stack] || stackDefaults.empty;
-  const stackDescription = language === 'en' ? d.descriptionEn : d.description;
+};
+
+export function stackDescription(stack, language) {
+  const d = STACK_DEFAULTS[stack] || STACK_DEFAULTS.empty;
+  return language === 'en' ? d.descriptionEn : d.description;
+}
+
+// 宿主权限白名单里随栈变化的命令。npm / node / git 是引擎运行时，留在各宿主模板的固定项里。
+const STACK_SHELL = {
+  dotnet: ['dotnet build', 'dotnet run'],
+  rust: ['cargo build', 'cargo run', 'cargo test', 'cargo check'],
+  go: ['go build', 'go run', 'go test'],
+  java: ['mvn compile', 'mvn test', 'mvn -q test', 'gradle test', 'gradlew test', 'gradlew.bat test', './gradlew test'],
+};
+
+function stackAllowFragment(stack, tool, sep) {
+  const commands = STACK_SHELL[stack] || [];
+  if (!commands.length) return '';
+  return commands.map((cmd) => `      "${tool}(${cmd}${sep}*)",`).join('\n') + '\n';
+}
+
+// 模板变量：只替换已知 token，未知 {{...}} 原样保留（避免误伤代码模板字符串）。
+export function buildVars({ target, projectName, stack, language, hosts, kitVersion, installedAt, trackHosts = false }) {
+  const d = STACK_DEFAULTS[stack] || STACK_DEFAULTS.empty;
+  const description = stackDescription(stack, language);
   const hostIgnores = trackHosts
     ? ''
     : hosts
@@ -33,13 +77,16 @@ export function buildVars({ target, projectName, stack, language, hosts, kitVers
   return {
     projectName: projectName || pathBase(target),
     stack,
-    stackDescription,
+    stackDescription: description,
     language,
     apiPort: String(d.apiPort),
     webPort: String(d.webPort),
     boardPort: '8933',
     hosts: hosts.join(', '),
     hostsJson: JSON.stringify(hosts),
+    stackAllowBashColon: stackAllowFragment(stack, 'Bash', ':'),
+    stackAllowBashSpace: stackAllowFragment(stack, 'Bash', ' '),
+    stackAllowShellSpace: stackAllowFragment(stack, 'Shell', ' '),
     kitVersion,
     installedAt,
     nodeMinimum: '22',

package/src/tui/ansi.mjs

@@ -0,0 +1,55 @@
+// 零依赖 ANSI 渲染原语：转义序列、可视宽度、着色、进度条与圆角边框。
+const ESC = '\x1b'; // 转义起始符（与 tui 其余文件的 \x1b 字面量写法统一）
+// CSI 序列（ESC[ ... 字母）与 OSC 序列（ESC] ... BEL/ST）
+const ANSI_RE = new RegExp(ESC + '\\[[0-9;]*[A-Za-z]|' + ESC + '\\][^\\x07]*(?:\\x07|' + ESC + '\\\\)', 'g');
+
+// 移除 ANSI 转义序列
+export function stripAnsi(str) {
+  return str.replace(ANSI_RE, '');
+}
+
+// 东亚宽字符区间（CJK/全角），可视宽度按 2 计
+const WIDE_RE = /[\u1100-\u115F\u2E80-\uA4CF\uAC00-\uD7A3\uF900-\uFAFF\uFE30-\uFE6F\uFF00-\uFF60\uFFE0-\uFFE6]/;
+
+export function visualWidth(str) {
+  let width = 0;
+  for (const ch of stripAnsi(str)) width += WIDE_RE.test(ch) ? 2 : 1;
+  return width;
+}
+
+const paint = (code, reset) => (str) => ESC + code + str + ESC + reset;
+
+export const cyan = paint('[36m', '[39m');
+export const dim = paint('[2m', '[22m');
+export const green = paint('[32m', '[39m');
+export const red = paint('[31m', '[39m');
+
+export const SPINNER_FRAMES = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+
+// 第 n 帧 spinner（按帧表长度自动回绕）
+export const spinner = (n) => SPINNER_FRAMES[n % SPINNER_FRAMES.length];
+
+// 单行进度：回车 + 清行 + spinner + 进度条（TUI 逐文件进度渲染用）。detail 是当前文件，随下一帧被清掉。
+export function progressLine(done, total, detail = '') {
+  const tail = detail ? ` ${detail}` : '';
+  return `\r\x1b[2K${spinner(done)} ${progress(done, total)}${tail}`;
+}
+
+export function progress(current, total, width = 24) {
+  // total=0 时 filled=0，避免除零
+  const filled = total > 0 ? Math.round((width * current) / total) : 0;
+  return '█'.repeat(filled) + '░'.repeat(width - filled) + ` ${current}/${total}`;
+}
+
+// 圆角边框盒：对齐按 stripAnsi 后的可视宽度计算，正文含 CJK/色码不歪
+export function box(title, bodyLines) {
+  const lines = bodyLines.map(String);
+  const titleW = visualWidth(title);
+  const inner = Math.max(titleW + 1, ...lines.map((l) => visualWidth(l)));
+  const out = [`╭─ ${title} ${'─'.repeat(inner - titleW - 1)}╮`];
+  for (const line of lines) {
+    out.push(`│ ${line}${' '.repeat(inner - visualWidth(line))} │`);
+  }
+  out.push(`╰${'─'.repeat(inner + 2)}╯`);
+  return out;
+}