@huicai/cli 2026.711.0 → 2026.711.24616

package/src/clipboard.js

@@ -1,34 +1 @@
-// 跨平台剪贴板写入（/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 });
-    }
-  });
-}
+import{spawn as o}from"node:child_process";export function copyToClipboard(r){const{cmd:c,args:t}="darwin"===process.platform?{cmd:"pbcopy",args:[]}:"win32"===process.platform?{cmd:"clip",args:[]}:process.env.WAYLAND_DISPLAY?{cmd:"wl-copy",args:[]}:{cmd:"xclip",args:["-selection","clipboard"]};return new Promise(s=>{let e;try{e=o(c,t)}catch{return void s({ok:!1,tool:c})}e.on("error",()=>s({ok:!1,tool:c})),e.on("close",o=>s({ok:0===o,tool:c})),e.stdin.on("error",()=>{});try{e.stdin.end(String(r))}catch{s({ok:!1,tool:c})}})}

package/src/codeindex.js

@@ -1,229 +1 @@
-// 代码检索引擎（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');
-}
+import t from"node:fs";import e from"node:path";import{embed as n}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 s=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"]);export const MAX_FILE_BYTES=524288;const o=new Map;export function tokenize(t){const e=String(t),n=e.replace(/([a-z0-9])([A-Z])/g,"$1 $2").toLowerCase().split(/[^a-z0-9_]+/).filter(t=>t.length>=2&&t.length<=40),s=e.match(/[一-鿿]/g);if(s)for(const t of s)n.push(t);const o=e.match(/[一-鿿]{2,}/g);if(o)for(const t of o)for(let e=0;e<t.length-1;e++)n.push(t.slice(e,e+2));return n}function*i(n,o=0,r=14){let c;try{c=t.readdirSync(n,{withFileTypes:!0})}catch{return}for(const t of c){if(IGNORE.has(t.name)||t.name.startsWith("."))continue;const c=e.join(n,t.name);t.isDirectory()?o<r&&(yield*i(c,o+1,r)):t.isFile()&&s.has(e.extname(t.name).toLowerCase())&&(yield c)}}export function buildIndex(n){const s=[],o=new Map;let r=0;t:for(const c of i(n)){let i;try{if(t.statSync(c).size>524288)continue;const e=t.readFileSync(c);if(e.includes(0))continue;i=e.toString("utf8")}catch{continue}const l=e.relative(n,c).split(e.sep).join("/"),a=i.split("\n");for(let t=0;t<a.length;t+=40){const e=Math.min(a.length,t+50),n=a.slice(t,e),i=tokenize(n.join("\n"));if(!i.length)continue;const c=new Map;for(const t of i)c.set(t,(c.get(t)||0)+1);for(const t of c.keys())o.set(t,(o.get(t)||0)+1);if(s.push({file:l,start:t+1,end:e,tf:c,len:i.length,text:n.join("\n")}),r+=i.length,s.length>=6e3)break t;if(e>=a.length)break}}return{chunks:s,df:o,N:s.length,avgLen:s.length?r/s.length:1}}export function getIndex(t){const e=o.get(t);if(e&&Date.now()-e.builtAt<3e5)return e.index;const n=buildIndex(t);return o.set(t,{index:n,builtAt:Date.now()}),n}export function invalidate(t){o.delete(t)}export function search(t,e,n=8){const s=[...new Set(tokenize(e))];if(!s.length||!t.N)return[];const{chunks:o,df:i,N:r,avgLen:c}=t,l=[];for(const t of o){let e=0;for(const n of s){const s=t.tf.get(n);if(!s)continue;const o=i.get(n)||1;e+=Math.log(1+(r-o+.5)/(o+.5))*(2.5*s/(s+1.5*(.25+t.len/c*.75)))}e>0&&l.push({file:t.file,start:t.start,end:t.end,text:t.text,score:e})}return l.sort((t,e)=>e.score-t.score),l.slice(0,n)}export function cosine(t,e){if(!t||!e)return 0;let n=0,s=0,o=0;const i=Math.min(t.length,e.length);for(let r=0;r<i;r++)n+=t[r]*e[r],s+=t[r]*t[r],o+=e[r]*e[r];return s&&o?n/(Math.sqrt(s)*Math.sqrt(o)):0}export async function searchCode(t,e,s){const o=String(e?.query||"").trim();if(!o)return"缺少 query（要检索的概念/功能描述）";const i=Number.isFinite(e?.k)&&e.k>0?Math.min(e.k,20):8,r=getIndex(t);if(!r.N)return"（当前目录未索引到代码文件）";const c=!!(s&&s.embeddingModel&&s.baseUrl&&s.apiKey);let l=search(r,o,c?Math.min(30,Math.max(3*i,15)):i);if(!l.length)return`（未找到与「${o}」明显相关的代码；可换关键词，或用 grep 精确匹配）`;let a=!1;if(c&&l.length>1)try{const t=l.map(t=>t.text.slice(0,2e3)),e=await n(s.baseUrl,s.apiKey,s.embeddingModel,[o,...t]);if(e&&e.length===t.length+1&&e[0]){const t=e[0];l=l.map((n,s)=>({...n,score:cosine(t,e[s+1])})).sort((t,e)=>e.score-t.score),a=!0}}catch{}return(a?"（向量语义重排）\n":"")+l.slice(0,i).map(t=>{const e=t.text.length>1200?t.text.slice(0,1200)+"\n…（本块过长已截断）":t.text;return`${t.file}:${t.start}-${t.end}\n${e}`}).join("\n\n─────\n\n")}function r(t){const e=new Set,n=[/(?: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 s of n){let n;for(;(n=s.exec(t))&&e.size<12;)e.add(n[1])}return[...e]}export function buildRepoMap(n,{maxFiles:s=40,maxChars:o=2500}={}){const c=[];for(const s of i(n))try{const o=t.statSync(s);if(o.size>524288)continue;c.push({abs:s,size:o.size,rel:e.relative(n,s).split(e.sep).join("/")})}catch{}if(!c.length)return"";c.sort((t,e)=>e.size-t.size);const l=[];let a=0;for(const e of c.slice(0,s)){let n=[];try{n=r(t.readFileSync(e.abs,"utf8").slice(0,4e4))}catch{}const s=`- ${e.rel}${n.length?"："+n.slice(0,8).join(", "):""}`;if(a+s.length>o)break;l.push(s),a+=s.length+1}return l.join("\n")}

package/src/colors.js

@@ -1,47 +1 @@
-// 极简 ANSI 颜色工具（零依赖）。尊重 NO_COLOR 与非 TTY（管道/重定向时自动关闭颜色）。
-// 运行时可切换：setColorMode('mono') 强制无色（/theme）；setAccent(name) 设强调色（/color）。
-
-const baseEnabled = process.env.NO_COLOR == null && process.env.TERM !== 'dumb' && process.stdout.isTTY;
-let mono = false;              // /theme mono：强制关闭颜色
-let accentName = 'cyan';       // /color：强调色（提示符/高亮）
-
-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',
-  bold: wrap('1'),
-  dim: wrap('2'),
-  italic: wrap('3'),
-  underline: wrap('4'),
-  red: wrap('31'),
-  green: wrap('32'),
-  yellow: wrap('33'),
-  blue: wrap('34'),
-  magenta: wrap('35'),
-  cyan: wrap('36'),
-  gray: wrap('90'),
-  /** 强调色（随 /color 变化）。 */
-  accent: (s) => (c[accentName] ? c[accentName](s) : String(s)),
-};
-
-// 兼容旧引用：静态基线（不随运行时主题变化；如需动态请用 c.* / on()）
-export const colorEnabled = baseEnabled;
+const e=null==process.env.NO_COLOR&&"dumb"!==process.env.TERM&&process.stdout.isTTY;let n=!1,t="cyan";const o=t=>o=>e&&!n?`[${t}m${o}`:String(o);export const ACCENTS=["cyan","green","magenta","blue","yellow","red"];export function setColorMode(e){n="mono"===e}export function setAccent(e){return!!ACCENTS.includes(e)&&(t=e,!0)}export function getAccent(){return t}export const c={reset:"",bold:o("1"),dim:o("2"),italic:o("3"),underline:o("4"),red:o("31"),green:o("32"),yellow:o("33"),blue:o("34"),magenta:o("35"),cyan:o("36"),gray:o("90"),accent:e=>c[t]?c[t](e):String(e)};export const colorEnabled=e;