@huicai/cli 2026.710.97949 → 2026.711.0

package/src/checkpoints.js

@@ -0,0 +1,99 @@
+// 会话检查点（对标 Claude Code /rewind）：每个用户回合前快照「对话位置 + 该回合将改动文件的原始内容」，
+// 可把对话/文件/两者回滚到某个检查点之前。文件原文由 runTurn 经 fileJournal 在写入前记录（见 agent.js）。
+//
+// 检查点存活于当前运行的会话（内存），不落盘（避免把大段文件内容写进会话 JSON）。
+// 语义：检查点 N 记录「第 N 个用户回合开始前」的状态；回滚到 N 即回到该回合之前。
+
+import fs from 'node:fs';
+import path from 'node:path';
+
+const MAX_CHECKPOINTS = 100;
+
+function safeClone(o) {
+  try {
+    return o ? JSON.parse(JSON.stringify(o)) : {};
+  } catch {
+    return {};
+  }
+}
+
+function deriveLabel(text) {
+  const first = String(text || '').replace(/\s+/g, ' ').trim();
+  return first ? (first.length > 60 ? first.slice(0, 60) + '…' : first) : '(空)';
+}
+
+export function createCheckpoints() {
+  const list = []; // {id, time, label, messagesLen, compaction, journal:[{path,before,existed,tooLarge?}]}
+  let seq = 0;
+
+  /** 回合开始前调用：记录对话位置，返回一个 journal 数组交给 runTurn 累积本回合的文件原文。 */
+  function snapshot({ session, label }) {
+    const cp = {
+      id: ++seq,
+      time: Date.now(),
+      label: deriveLabel(label),
+      messagesLen: session.messages.length,
+      compaction: safeClone(session.compaction),
+      journal: [],
+    };
+    list.push(cp);
+    // 上限保护：超出则丢弃最早的（其对应文件原文一并释放）
+    if (list.length > MAX_CHECKPOINTS) list.splice(0, list.length - MAX_CHECKPOINTS);
+    return cp.journal;
+  }
+
+  /** 从某检查点（含）起、各文件最早一次记录的原始内容（用于回滚到该点之前的文件状态）。 */
+  function changedFilesSince(pos) {
+    const map = new Map(); // relPath -> journal entry（最早）
+    for (let i = pos; i < list.length; i++) {
+      for (const e of list[i].journal) {
+        if (e && e.path && !map.has(e.path)) map.set(e.path, e);
+      }
+    }
+    return map;
+  }
+
+  /**
+   * 回滚到检查点 id。
+   * @param {number} id 检查点编号
+   * @param {'chat'|'code'|'both'} mode
+   * @param {{session:object, cwd:string}} p
+   */
+  function restore(id, mode, { session, cwd }) {
+    const pos = list.findIndex((c) => c.id === id);
+    if (pos < 0) return { ok: false, error: `检查点 ${id} 不存在（/rewind 查看可用检查点）` };
+    const cp = list[pos];
+    const summary = { ok: true, id, mode, conversation: false, filesRestored: [], filesDeleted: [], filesFailed: [], filesSkipped: [] };
+
+    if (mode === 'code' || mode === 'both') {
+      for (const [relPath, e] of changedFilesSince(pos)) {
+        if (e.tooLarge) { summary.filesSkipped.push(relPath); continue; }
+        const abs = path.resolve(cwd, relPath);
+        try {
+          if (!e.existed) {
+            fs.rmSync(abs, { force: true });
+            summary.filesDeleted.push(relPath);
+          } else {
+            fs.mkdirSync(path.dirname(abs), { recursive: true });
+            fs.writeFileSync(abs, e.before, 'utf8');
+            summary.filesRestored.push(relPath);
+          }
+        } catch (err) {
+          summary.filesFailed.push(`${relPath}：${err.message}`);
+        }
+      }
+    }
+
+    if (mode === 'chat' || mode === 'both') {
+      if (cp.messagesLen <= session.messages.length) session.messages.length = cp.messagesLen;
+      session.compaction = safeClone(cp.compaction);
+      session.lastPromptTokens = 0;
+      summary.conversation = true;
+      // 对话被截断 → 该点及之后的检查点失效
+      list.length = pos;
+    }
+    return summary;
+  }
+
+  return { list, snapshot, restore, changedFilesSince };
+}

package/src/clipboard.js

@@ -0,0 +1,34 @@
+// 跨平台剪贴板写入（/copy）：macOS pbcopy、Windows clip、Linux wl-copy/xclip。
+// 零依赖，best-effort：命令缺失或失败时返回 {ok:false}，由调用方回退（如直接打印内容）。
+
+import { spawn } from 'node:child_process';
+
+/** 选取本平台的剪贴板写入命令。 */
+function clipboardCmd() {
+  if (process.platform === 'darwin') return { cmd: 'pbcopy', args: [] };
+  if (process.platform === 'win32') return { cmd: 'clip', args: [] };
+  if (process.env.WAYLAND_DISPLAY) return { cmd: 'wl-copy', args: [] };
+  return { cmd: 'xclip', args: ['-selection', 'clipboard'] };
+}
+
+/** 把文本写入系统剪贴板。返回 {ok, tool}。 */
+export function copyToClipboard(text) {
+  const { cmd, args } = clipboardCmd();
+  return new Promise((resolve) => {
+    let child;
+    try {
+      child = spawn(cmd, args);
+    } catch {
+      resolve({ ok: false, tool: cmd });
+      return;
+    }
+    child.on('error', () => resolve({ ok: false, tool: cmd }));
+    child.on('close', (code) => resolve({ ok: code === 0, tool: cmd }));
+    child.stdin.on('error', () => {}); // 命令不存在时 stdin 写入会报错，已由 error/close 处理
+    try {
+      child.stdin.end(String(text));
+    } catch {
+      resolve({ ok: false, tool: cmd });
+    }
+  });
+}

package/src/codeindex.js

@@ -0,0 +1,229 @@
+// 代码检索引擎（search_code 工具 + 项目地图）：把仓库按行窗口分块，建 BM25 排序索引，
+// 按「概念/多词查询」检索最相关的代码块——比 grep（纯词法逐行）更懂「相关性」，且零依赖。
+// 可选：若平台提供 /v1/embeddings，则对候选块做向量重排，逼近真语义（api.embed 失败自动退化为纯 BM25）。
+// 索引按 cwd 缓存于内存，过期或首次使用时惰性重建。
+
+import fs from 'node:fs';
+import path from 'node:path';
+import { embed } from './api.js';
+
+export const IGNORE = new Set([
+  '.git', 'node_modules', 'dist', 'build', 'target', 'out', '.next', '.nuxt',
+  '.venv', 'venv', '__pycache__', '.idea', '.gradle', '.mvn', 'coverage', '.cache',
+]);
+const CODE_EXT = new Set([
+  '.js', '.mjs', '.cjs', '.ts', '.tsx', '.jsx', '.java', '.kt', '.py', '.go', '.rs',
+  '.rb', '.php', '.c', '.h', '.cpp', '.hpp', '.cs', '.swift', '.scala', '.sql',
+  '.vue', '.svelte', '.sh', '.css', '.scss', '.html', '.md', '.yaml', '.yml', '.json', '.toml',
+]);
+const WINDOW = 50;   // 每块行数
+const STEP = 40;     // 步长（10 行重叠，避免边界切断）
+const MAX_CHUNKS = 6000;
+export const MAX_FILE_BYTES = 512 * 1024;
+const CACHE_TTL = 5 * 60 * 1000;
+
+const cache = new Map(); // cwd -> { index, builtAt }
+
+/** 代码分词：驼峰拆分 + 非字母数字切分 + 小写（拉丁）；CJK 出单字(召回)+相邻二元(精度)。
+ *  索引与查询共用同一分词，故中文注释/中文查询也能命中（代码常是中文注释 + 英文标识符混排）。 */
+export function tokenize(s) {
+  const str = String(s);
+  const tokens = str
+    .replace(/([a-z0-9])([A-Z])/g, '$1 $2')
+    .toLowerCase()
+    .split(/[^a-z0-9_]+/)
+    .filter((t) => t.length >= 2 && t.length <= 40);
+  const singles = str.match(/[一-鿿]/g);
+  if (singles) for (const ch of singles) tokens.push(ch);
+  const runs = str.match(/[一-鿿]{2,}/g);
+  if (runs) for (const run of runs) for (let i = 0; i < run.length - 1; i++) tokens.push(run.slice(i, i + 2));
+  return tokens;
+}
+
+function* walk(dir, depth = 0, maxDepth = 14) {
+  let entries;
+  try {
+    entries = fs.readdirSync(dir, { withFileTypes: true });
+  } catch {
+    return;
+  }
+  for (const e of entries) {
+    if (IGNORE.has(e.name) || e.name.startsWith('.')) continue;
+    const full = path.join(dir, e.name);
+    if (e.isDirectory()) {
+      if (depth < maxDepth) yield* walk(full, depth + 1, maxDepth);
+    } else if (e.isFile() && CODE_EXT.has(path.extname(e.name).toLowerCase())) {
+      yield full;
+    }
+  }
+}
+
+/** 构建 BM25 索引（惰性、可缓存）。 */
+export function buildIndex(cwd) {
+  const chunks = [];
+  const df = new Map();
+  let totalLen = 0;
+  outer: for (const abs of walk(cwd)) {
+    let text;
+    try {
+      const st = fs.statSync(abs);
+      if (st.size > MAX_FILE_BYTES) continue;
+      const buf = fs.readFileSync(abs);
+      if (buf.includes(0)) continue; // 二进制
+      text = buf.toString('utf8');
+    } catch {
+      continue;
+    }
+    const rel = path.relative(cwd, abs).split(path.sep).join('/');
+    const lines = text.split('\n');
+    for (let start = 0; start < lines.length; start += STEP) {
+      const end = Math.min(lines.length, start + WINDOW);
+      const slice = lines.slice(start, end);
+      const tokens = tokenize(slice.join('\n'));
+      if (!tokens.length) continue;
+      const tf = new Map();
+      for (const t of tokens) tf.set(t, (tf.get(t) || 0) + 1);
+      for (const t of tf.keys()) df.set(t, (df.get(t) || 0) + 1);
+      chunks.push({ file: rel, start: start + 1, end, tf, len: tokens.length, text: slice.join('\n') });
+      totalLen += tokens.length;
+      if (chunks.length >= MAX_CHUNKS) break outer;
+      if (end >= lines.length) break;
+    }
+  }
+  return { chunks, df, N: chunks.length, avgLen: chunks.length ? totalLen / chunks.length : 1 };
+}
+
+/** 取（缓存的）索引。 */
+export function getIndex(cwd) {
+  const hit = cache.get(cwd);
+  if (hit && Date.now() - hit.builtAt < CACHE_TTL) return hit.index;
+  const index = buildIndex(cwd);
+  cache.set(cwd, { index, builtAt: Date.now() });
+  return index;
+}
+
+/** 使某目录索引失效（写文件后可调用；此处保守，仅供扩展）。 */
+export function invalidate(cwd) {
+  cache.delete(cwd);
+}
+
+/** BM25 检索，返回前 k 个相关代码块。 */
+export function search(index, query, k = 8) {
+  const qtokens = [...new Set(tokenize(query))];
+  if (!qtokens.length || !index.N) return [];
+  const { chunks, df, N, avgLen } = index;
+  const k1 = 1.5;
+  const b = 0.75;
+  const scored = [];
+  for (const c of chunks) {
+    let score = 0;
+    for (const qt of qtokens) {
+      const f = c.tf.get(qt);
+      if (!f) continue;
+      const n = df.get(qt) || 1;
+      const idf = Math.log(1 + (N - n + 0.5) / (n + 0.5));
+      score += idf * ((f * (k1 + 1)) / (f + k1 * (1 - b + b * (c.len / avgLen))));
+    }
+    if (score > 0) scored.push({ file: c.file, start: c.start, end: c.end, text: c.text, score });
+  }
+  scored.sort((a, b2) => b2.score - a.score);
+  return scored.slice(0, k);
+}
+
+/** 余弦相似度。 */
+export function cosine(a, b) {
+  if (!a || !b) return 0;
+  let dot = 0;
+  let na = 0;
+  let nb = 0;
+  const n = Math.min(a.length, b.length);
+  for (let i = 0; i < n; i++) {
+    dot += a[i] * b[i];
+    na += a[i] * a[i];
+    nb += b[i] * b[i];
+  }
+  return na && nb ? dot / (Math.sqrt(na) * Math.sqrt(nb)) : 0;
+}
+
+/**
+ * search_code 工具执行器：BM25 召回 + （配置了嵌入模型时）向量语义重排，格式化为「文件:行范围 + 代码」。
+ * 配了 config.embeddingModel 才走 /v1/embeddings 做重排（真语义）；未配或向量化失败则纯 BM25（离线可用）。
+ */
+export async function searchCode(cwd, args, config) {
+  const query = String(args?.query || '').trim();
+  if (!query) return '缺少 query（要检索的概念/功能描述）';
+  const k = Number.isFinite(args?.k) && args.k > 0 ? Math.min(args.k, 20) : 8;
+  const index = getIndex(cwd);
+  if (!index.N) return '（当前目录未索引到代码文件）';
+  const rerank = !!(config && config.embeddingModel && config.baseUrl && config.apiKey);
+  const recall = rerank ? Math.min(30, Math.max(k * 3, 15)) : k; // 重排时多召回候选
+  let hits = search(index, query, recall);
+  if (!hits.length) return `（未找到与「${query}」明显相关的代码；可换关键词，或用 grep 精确匹配）`;
+  let reranked = false;
+  if (rerank && hits.length > 1) {
+    try {
+      const texts = hits.map((h) => h.text.slice(0, 2000));
+      const vecs = await embed(config.baseUrl, config.apiKey, config.embeddingModel, [query, ...texts]);
+      if (vecs && vecs.length === texts.length + 1 && vecs[0]) {
+        const qv = vecs[0];
+        hits = hits.map((h, i) => ({ ...h, score: cosine(qv, vecs[i + 1]) })).sort((a, b) => b.score - a.score);
+        reranked = true;
+      }
+    } catch {
+      /* 向量化失败 → 退化为 BM25 排序，不影响可用性 */
+    }
+  }
+  const head = reranked ? '（向量语义重排）\n' : '';
+  return head + hits.slice(0, k).map((h) => {
+    const body = h.text.length > 1200 ? h.text.slice(0, 1200) + '\n…（本块过长已截断）' : h.text;
+    return `${h.file}:${h.start}-${h.end}\n${body}`;
+  }).join('\n\n─────\n\n');
+}
+
+/** 提取一段代码里的关键符号名（函数/类/def/导出/顶层箭头常量），供项目地图用。 */
+function extractSymbols(text) {
+  const syms = new Set();
+  const res = [
+    /(?:export\s+)?(?:async\s+)?function\s+([A-Za-z_]\w*)/g,
+    /\bclass\s+([A-Za-z_]\w*)/g,
+    /\bdef\s+([A-Za-z_]\w*)/g,
+    /\b(?:export\s+)?const\s+([A-Za-z_]\w*)\s*=\s*(?:async\s*)?(?:function|\([^)]*\)\s*=>)/g,
+    /@(?:Get|Post|Put|Delete|Request)Mapping\(\s*"([^"]+)"/g,
+    /\bpublic\s+[\w<>[\]]+\s+([A-Za-z_]\w*)\s*\(/g,
+  ];
+  for (const re of res) {
+    let m;
+    while ((m = re.exec(text)) && syms.size < 12) syms.add(m[1]);
+  }
+  return [...syms];
+}
+
+/**
+ * 项目地图：关键源文件 + 其主要符号的紧凑概览（注入系统提示词，帮智能体快速建立全局观）。
+ * 按文件大小近似「重要性」取 top N，控制总字符数。
+ */
+export function buildRepoMap(cwd, { maxFiles = 40, maxChars = 2500 } = {}) {
+  const files = [];
+  for (const abs of walk(cwd)) {
+    try {
+      const st = fs.statSync(abs);
+      if (st.size > MAX_FILE_BYTES) continue;
+      files.push({ abs, size: st.size, rel: path.relative(cwd, abs).split(path.sep).join('/') });
+    } catch { /* skip */ }
+  }
+  if (!files.length) return '';
+  files.sort((a, b) => b.size - a.size);
+  const lines = [];
+  let used = 0;
+  for (const f of files.slice(0, maxFiles)) {
+    let syms = [];
+    try {
+      syms = extractSymbols(fs.readFileSync(f.abs, 'utf8').slice(0, 40000));
+    } catch { /* skip */ }
+    const line = `- ${f.rel}${syms.length ? '：' + syms.slice(0, 8).join(', ') : ''}`;
+    if (used + line.length > maxChars) break;
+    lines.push(line);
+    used += line.length + 1;
+  }
+  return lines.join('\n');
+}

package/src/colors.js

@@ -1,8 +1,30 @@
 // 极简 ANSI 颜色工具（零依赖）。尊重 NO_COLOR 与非 TTY（管道/重定向时自动关闭颜色）。
+// 运行时可切换：setColorMode('mono') 强制无色（/theme）；setAccent(name) 设强调色（/color）。
 
-const enabled = process.env.NO_COLOR == null && process.env.TERM !== 'dumb' && process.stdout.isTTY;
+const baseEnabled = process.env.NO_COLOR == null && process.env.TERM !== 'dumb' && process.stdout.isTTY;
+let mono = false;              // /theme mono：强制关闭颜色
+let accentName = 'cyan';       // /color：强调色（提示符/高亮）
 
-const wrap = (code) => (s) => (enabled ? `\x1b[${code}m${s}\x1b[0m` : String(s));
+const on = () => baseEnabled && !mono;
+const wrap = (code) => (s) => (on() ? `\x1b[${code}m${s}\x1b[0m` : String(s));
+
+export const ACCENTS = ['cyan', 'green', 'magenta', 'blue', 'yellow', 'red'];
+
+/** 切换主题：'mono' 关闭颜色，其它值恢复（按 env/TTY 判定）。 */
+export function setColorMode(mode) {
+  mono = mode === 'mono';
+}
+/** 设置强调色（cyan/green/magenta/blue/yellow/red）；非法值忽略。返回是否生效。 */
+export function setAccent(name) {
+  if (ACCENTS.includes(name)) {
+    accentName = name;
+    return true;
+  }
+  return false;
+}
+export function getAccent() {
+  return accentName;
+}
 
 export const c = {
   reset: '\x1b[0m',
@@ -17,6 +39,9 @@ export const c = {
   magenta: wrap('35'),
   cyan: wrap('36'),
   gray: wrap('90'),
+  /** 强调色（随 /color 变化）。 */
+  accent: (s) => (c[accentName] ? c[accentName](s) : String(s)),
 };
 
-export const colorEnabled = enabled;
+// 兼容旧引用：静态基线（不随运行时主题变化；如需动态请用 c.* / on()）
+export const colorEnabled = baseEnabled;