@huicai/cli 2026.1021.0 → 2026.1022.59980
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/bin/huicai.js +45 -10
- package/package.json +2 -2
- package/src/account-scope.js +1 -0
- package/src/agent.js +1 -1
- package/src/assets-import.js +1 -1
- package/src/automation-sync.js +1 -0
- package/src/codeindex.js +1 -1
- package/src/commands.js +1 -1
- package/src/config.js +1 -1
- package/src/context.js +1 -1
- package/src/daemon.js +1 -1
- package/src/decisions.js +1 -1
- package/src/doccache.js +1 -1
- package/src/experts-store.js +1 -1
- package/src/experts.js +1 -1
- package/src/input.js +1 -1
- package/src/lsp.js +1 -1
- package/src/mcp.js +1 -1
- package/src/memory.js +1 -1
- package/src/plan-limits.js +1 -1
- package/src/privatedir.js +1 -1
- package/src/repl.js +1 -1
- package/src/schedule.js +1 -1
- package/src/session.js +1 -1
- package/src/sessionindex.js +1 -1
- package/src/skills.js +1 -1
- package/src/squad.js +1 -1
- package/src/subagent.js +1 -1
- package/src/toolpipe.js +1 -1
- package/src/uninstall.js +1 -1
- package/src/workflow.js +1 -1
- package/src/worklog.js +1 -1
package/src/doccache.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import t from"node:path";import n from"node:crypto";import{
|
|
1
|
+
import e from"node:fs";import t from"node:path";import n from"node:crypto";import{accountDataDir as r}from"./account-scope.js";import{extractDocument as c}from"./api.js";const a=()=>t.join(r(),"cache","docs"),o=new Map;function s(e){return n.createHash("sha1").update(e).digest("hex")}export function fileKey(e,t,n){return s(`${r()}|f|${e}|${t.size}|${Math.floor(t.mtimeMs)}|${n||0}`)}export function contentKey(e,t){return s(`${r()}|c|${n.createHash("sha1").update(e).digest("hex")}|${e.length}|${t||0}`)}function i(e){return t.join(a(),e+".json")}export function getCached(t){if(o.has(t))return o.get(t);try{const n=i(t),r=JSON.parse(e.readFileSync(n,"utf8")),c=new Date;return e.utimesSync(n,c,c),o.set(t,r),r}catch{return null}}export function putCached(n,r){o.set(n,r);try{e.mkdirSync(a(),{recursive:!0}),e.writeFileSync(i(n),JSON.stringify(r)),function(){try{const n=e.readdirSync(a()).filter(e=>e.endsWith(".json"));if(n.length<=64)return;const r=n.map(n=>{try{return{f:n,t:e.statSync(t.join(a(),n)).mtimeMs}}catch{return null}}).filter(Boolean).sort((e,t)=>e.t-t.t);for(const{f:n}of r.slice(0,r.length-64))try{e.unlinkSync(t.join(a(),n))}catch{}}catch{}}()}catch{}}export async function extractFileCached(n,r,a,o={}){const s=o.maxChars||6e4,i=fileKey(r,a,s),u=getCached(i);if(u)return{result:u,cached:!0};o.onMiss&&o.onMiss();const h=o.fetch||c,f=await h(n.baseUrl,n.apiKey,{filename:t.basename(r),bytes:e.readFileSync(r),maxChars:s},o.onUpload);return putCached(i,f),{result:f,cached:!1}}export async function extractContentCached(e,t,n,r={}){const a=r.maxChars||0,o=contentKey(n,a),s=getCached(o);if(s)return{result:s,cached:!0};r.onMiss&&r.onMiss();const i=r.fetch||c,u={filename:t,bytes:n};a&&(u.maxChars=a);const h=await i(e.baseUrl,e.apiKey,u,r.onUpload);return putCached(o,h),{result:h,cached:!1}}export function clearDocCache(){o.clear();try{e.rmSync(a(),{recursive:!0,force:!0})}catch{}}
|
package/src/experts-store.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import t from"node:fs";import r
|
|
1
|
+
import t from"node:fs";import{accountDataDir as r}from"./account-scope.js";import e from"node:path";import{parseFrontmatter as n}from"./skills.js";import{loadSquads as s,countUserSquads as i}from"./squad.js";import{discoverExperts as o}from"./experts.js";import{loadConfig as c}from"./config.js";import{assertLimit as m}from"./plan-limits.js";const a=()=>e.join(r(),"agents"),u=()=>e.join(r(),"squads"),p=/^[\w一-龥-]{1,64}$/;function l(t){const r=String(t||"").trim();if(!p.test(r))throw new Error("非法名称：只能包含中文、字母、数字、下划线与连字符");return r}export function countUserAgents(){try{return t.readdirSync(a()).filter(t=>t.endsWith(".md")).length}catch{return 0}}function d(){const r=a();let s=[];try{s=t.readdirSync(r).filter(t=>t.endsWith(".md"))}catch{return[]}const i=[];for(const o of s){const s=o.replace(/\.md$/,"").trim();if(s)try{const c=t.readFileSync(e.join(r,o),"utf8"),m=n(c);i.push({slug:s,name:(m.name||s).trim(),description:m.description||""})}catch{}}return i}function f(){let t=[];try{t=s(null)||[]}catch{t=[]}return t.map(t=>({slug:t.file?e.basename(t.file).replace(/\.md$/,""):t.name,name:t.name,description:t.description||"",leader:t.leader||"",mode:t.mode||"dispatch",when:t.when||"",memberCount:(t.members||[]).length,builtin:"builtin"===t.scope}))}export function listMyExperts(){return{agents:d(),squads:f()}}export function readExpert(r,n){const s=l(n),i=e.join("squad"===r?u():a(),`${s}.md`);let o="";try{o=t.readFileSync(i,"utf8")}catch{throw new Error("未找到该定义（可能是内置项，编辑将另存为可写副本）")}return{kind:r,slug:s,content:o}}export function saveAgent({slug:r,content:s,overwrite:i=!0}={}){const o=l(r),c=String(s||""),u=n(c);if(!(u.name||"").trim())throw new Error("定义缺少name（frontmatter）");if(!(u.body||"").trim())throw new Error("专家的系统提示词正文不能为空");const p=e.join(a(),`${o}.md`);if(!i&&t.existsSync(p))return{status:"skipped",slug:o,name:(u.name||o).trim(),...listMyExperts()};const d=t.existsSync(p);return d||m("agents",countUserAgents()),t.mkdirSync(a(),{recursive:!0}),t.writeFileSync(p,c),{status:d?"overwritten":"created",slug:o,name:(u.name||o).trim(),...listMyExperts()}}export function saveSquad({slug:r,content:s,overwrite:o=!0}={}){const c=l(r),a=String(s||""),p=n(a);if(!(p.name||"").trim())throw new Error("定义缺少name（frontmatter）");const d=e.join(u(),`${c}.md`);if(!o&&t.existsSync(d))return{status:"skipped",slug:c,name:(p.name||c).trim(),...listMyExperts()};const f=t.existsSync(d);return f||m("squads",i()),t.mkdirSync(u(),{recursive:!0}),t.writeFileSync(d,a),{status:f?"overwritten":"created",slug:c,name:(p.name||c).trim(),...listMyExperts()}}export function removeAgent(r){const n=l(r),s=e.join(a(),`${n}.md`);if(!t.existsSync(s))throw new Error("该专家不存在或不可删除");return t.rmSync(s),listMyExperts()}export function removeSquad(r){const n=l(r),s=e.join(u(),`${n}.md`);if(!t.existsSync(s))throw new Error("内置专家团不可删除；可召唤后另存为副本再改");return t.rmSync(s),listMyExperts()}export async function platformExperts({force:t=!1,config:r}={}){try{const e=r||c();return e&&e.baseUrl&&e.apiKey?await o({baseUrl:e.baseUrl,apiKey:e.apiKey},{force:t}):[]}catch{return[]}}
|
package/src/experts.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import
|
|
1
|
+
import{accountDataDir as e}from"./account-scope.js";import t from"node:fs";import n from"node:path";import r from"node:os";import i from"node:crypto";import{listModels as a,streamChat as s}from"./api.js";const o=6e5,c=/^(agent|workflow|datahuman|squad)-/,p={squad:0,workflow:1,agent:2,datahuman:3},u={squad:"专家团",workflow:"工作流",agent:"智能体",datahuman:"数字人"};function l(e){if(e.hc_type&&u[e.hc_type])return e.hc_type;const t=/^(agent|workflow|datahuman|squad)-/.exec(String(e.id||""));return t?t[1]:"agent"}export function expertsCachePath(t){const r=i.createHash("sha1").update(String(t.baseUrl||"")+"|"+String(t.apiKey||"")).digest("hex").slice(0,12);return n.join(e(),"cache",`experts-${r}.json`)}function y(e){try{const n=JSON.parse(t.readFileSync(e,"utf8"));if(Array.isArray(n?.experts))return{fetchedAt:n.fetchedAt||0,experts:n.experts}}catch{}return null}function d(e,r){try{t.mkdirSync(n.dirname(e),{recursive:!0}),t.writeFileSync(e,JSON.stringify({fetchedAt:Date.now(),experts:r}))}catch{}}async function m(e){return(await a(e.baseUrl,e.apiKey)).filter(e=>c.test(String(e.id||""))).map(e=>({id:e.id,name:e.display_name||e.id,description:String(e.description||"").replace(/\s+/g," ").trim(),type:l(e)})).sort((e,t)=>(p[e.type]??9)-(p[t.type]??9)||String(e.name).localeCompare(t.name,"zh"))}const f=new Set;export async function discoverExperts(e,{force:t=!1}={}){if(!e||!e.baseUrl||!e.apiKey)return[];const n=expertsCachePath(e),r=t?null:y(n);if(r)return Date.now()-r.fetchedAt>6e5&&function(e,t){f.has(t)||(f.add(t),m(e).then(e=>d(t,e)).catch(()=>{}).finally(()=>f.delete(t)))}(e,n),r.experts;try{const r=await(i=m(e),a=t?15e3:3500,new Promise((e,t)=>{const n=setTimeout(()=>t(new Error(`发现超时（${a}ms）`)),a);i.then(t=>{clearTimeout(n),e(t)},e=>{clearTimeout(n),t(e)})}));return d(n,r),r}catch{const e=y(n);return e?e.experts:[]}var i,a}export async function makeExpertTools({config:e}){if(!e||!e.baseUrl||!e.apiKey)return[];if(!1===e.autoExperts)return[];if(c.test(String(e.model||"")))return[];const t=await discoverExperts(e);if(!t.length)return[];const n=new Map(t.map(e=>[e.id,e])),r=function(e){const t=e.slice(0,24),n=t.map(e=>`${e.id} — [${u[e.type]||e.type}]${e.name}${e.description?"："+e.description.slice(0,80):""}`);return e.length>t.length&&n.push(`…另有 ${e.length-t.length} 个（/experts查看全部）`),n.join("\n")}(t),i={fails:0,openUntil:0},a=()=>Date.now()<i.openUntil?"平台专家暂时不可用（连续失败已熔断，稍后自动恢复）。请先用本地工具与自身知识完成任务，稍后可重试咨询。":null,p=async(r,a,c)=>{const p=String(r||"").trim(),l=n.get(p)||t.find(e=>e.name===p);if(!l)return`未找到平台专家「${p}」，请从目录中选择（/experts可查看全部）`;if(!String(a||"").trim())return"缺少query（要咨询/交办的自然语言内容）";try{const t=await async function(e,t,n,r){const i=r&&String(r).trim()?`${n}\n\n【背景/上游材料】\n${r}`:String(n);return(await s({baseUrl:e.baseUrl,apiKey:e.apiKey,model:t,messages:[{role:"user",content:i}],client:e.client,sessionId:e.sessionId},{},AbortSignal.timeout(o))).content||"(平台无返回)"}(e,l.id,a,c);return i.fails=0,`平台${u[l.type]||""}「${l.name}」的答复：\n${t}`}catch(e){const t=e&&"AbortError"===e.name?`超过 ${Math.round(10)} 分钟未完成，已中止`:e.message;return i.fails++,i.fails>=3&&(i.openUntil=Date.now()+3e4),`咨询「${l.name}」失败：${t}${i.openUntil>Date.now()?"（平台已熔断，请先用本地能力完成）":""}`}};return[{readOnly:!0,spec:{type:"function",function:{name:"consult_expert",description:"咨询/交办给一个平台专家（当前账号可用的专家团/工作流/智能体/数字人，服务端编排：知识库RAG/工具/多智能体协作/NL→SQL）。遇到专家擅长领域的问题优先交给它产出权威结论，再用本机工具把结论落地到项目。可多次调用（多个独立咨询会自动并行）。\n可用专家目录：\n"+r,parameters:{type:"object",properties:{expert:{type:"string",description:"专家id（目录第一列）",...t.length<=24?{enum:t.map(e=>e.id)}:{}},query:{type:"string",description:"自然语言的咨询/任务输入（专家看不到本对话，需自足）"},context:{type:"string",description:"可选：完成该咨询所需的背景/上游材料"}},required:["expert","query"]}}},run:async e=>a()||p(e?.expert,e?.query,e?.context)},{readOnly:!0,spec:{type:"function",function:{name:"consult_experts",description:"一次性把多个**独立**的咨询/子任务并行交给多个平台专家（同一专家也可多份），聚合各自答复。适合跨领域任务：如「专家团出方案 + 数字人查数据 + 智能体审内容」同时推进，显著减少串行等待。",parameters:{type:"object",properties:{tasks:{type:"array",description:"并行咨询数组（建议 2-6 个，上限 8）",items:{type:"object",properties:{expert:{type:"string",description:"专家id（见consult_expert目录）"},query:{type:"string",description:"该专家的自足任务/咨询"},context:{type:"string",description:"可选背景"}},required:["expert","query"]}}},required:["tasks"]}}},run:async e=>{const t=a();if(t)return t;const n=Array.isArray(e?.tasks)?e.tasks.slice(0,8):[];if(!n.length)return"未提供tasks（需要一个咨询数组）";return(await async function(e,t,n){const r=new Array(e.length);let i=0;return await Promise.all(Array.from({length:Math.max(1,Math.min(t,e.length))},async()=>{for(;i<e.length;){const t=i++;r[t]=await n(e[t],t)}})),r}(n,4,async e=>({expert:e?.expert,out:await p(e?.expert,e?.query,e?.context)}))).map((e,t)=>`【咨询 ${t+1} · ${e.expert}】\n${e.out}`).join("\n\n---\n\n")}}]}
|
package/src/input.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import t from"node:path";import n from"node:os";import{
|
|
1
|
+
import e from"node:fs";import t from"node:path";import n from"node:os";import{accountDataDir as o}from"./account-scope.js";import{runBash as r}from"./executors.js";import{extractFileCached as s}from"./doccache.js";const i=()=>t.join(o(),"history.json"),a={png:"image/png",jpg:"image/jpeg",jpeg:"image/jpeg",gif:"image/gif",webp:"image/webp",bmp:"image/bmp",svg:"image/svg+xml"},c=new Set(["pdf","doc","docx","xls","xlsx","ppt","pptx","rtf","odt","ods","odp","csv"]),l=8388608,p=20971520;export async function expandAtMentions(o,r,i=null,u={}){const g=/(^|[\s("'`（【,，:：])@([~A-Za-z0-9_./\\-]+)/g,d=new Set,f=[];let m;for(;(m=g.exec(o))&&f.length<5;){const e=m[2].replace(/[.,;:。，；]+$/,"");e&&!d.has(e)&&(d.add(e),f.push(e))}const $=[],y=[],x=[],w=[];for(const o of f){const s=o.startsWith("~")?t.join(n.homedir(),o.slice(1)):t.resolve(r,o);let u;try{u=e.statSync(s)}catch{continue}if(x.push(o),u.isDirectory()){try{const t=e.readdirSync(s).slice(0,100).join("\n");$.push(`【引用目录 @${o}】\n${t||"（空目录）"}`)}catch{}continue}const g=t.extname(s).slice(1).toLowerCase();if(a[g])if(u.size>l)$.push(`【图片 @${o}】（超过 ${Math.round(8)}MB，未附加）`);else try{const t=e.readFileSync(s).toString("base64");y.push({label:o,dataUri:`data:${a[g]};base64,${t}`}),$.push(`【已附加图片 @${o}】`)}catch{}else if(c.has(g))i&&i.baseUrl&&i.apiKey?u.size>p?$.push(`【文档 @${o}】（超过 ${Math.round(20)}MB，未解析）`):w.push({index:$.push("")-1,raw:o,abs:s,st:u}):$.push(`【文档 @${o}】（需登录后才能解析为文本）`);else if(u.size>524288)$.push(`【引用文件 @${o}】\n（文件过大：${Math.round(u.size/1024)}KB，未展开；可让我用read_file分段读取）`);else try{const t=e.readFileSync(s);if(h(t)){$.push(`【引用文件 @${o}】\n（二进制文件，未展开）`);continue}let n=t.toString("utf8");n.length>12e3&&(n=n.slice(0,12e3)+"\n…（已截断）"),$.push(`【引用文件 @${o}】\n\`\`\`\n${n}\n\`\`\``)}catch{}}if(w.length){const e=/^(agent|workflow|datahuman|squad)-/.test(String(i?.model||""));let t=0;const n=async({index:n,raw:o,abs:r,st:a})=>{try{const{result:c,cached:l}=await s(i,r,a,{maxChars:6e4,fetch:u.extract,onMiss:()=>u.onDocStart&&u.onDocStart(o,++t,w.length)}),p=c.truncated?e?"·已截断(头尾摘要)":"·已截断(头尾摘要，被省略部分可用read_document工具按offset分段读取)":"",h=`${c.pages?c.pages+"页·":""}${c.chars??0}字${p}`;$[n]=`【引用文档 @${o}（${h}）】\n${c.text||"（空）"}`,u.onDocDone&&u.onDocDone(o,{cached:l})}catch(e){$[n]=`【文档 @${o}】（解析失败：${e.message}）`,u.onDocDone&&u.onDocDone(o,{error:e.message})}},o=w.slice();await Promise.all(Array.from({length:Math.min(3,o.length)},async()=>{for(;o.length;)await n(o.shift())}))}return{text:$.length?o+"\n\n"+$.join("\n\n"):o,mentions:x,images:y}}function h(e){const t=Math.min(e.length,4096);for(let n=0;n<t;n++)if(0===e[n])return!0;return!1}const u=new RegExp("(^|\\n\\n)((?:【(?:引用文档|引用文件|引用目录) @[^】\\n]*】|【上下文：[^】\\n]*】|【文档附件：[^】\\n]*】))\\n([\\s\\S]*?)(?=\\n\\n(?:【(?:引用文档|引用文件|引用目录|文档|图片|已附加图片) @|【上下文：|【文档附件：)|$)","g");export function collapseMentionBlocks(e){const t=String(e||"");if(!/【(?:引用文档|引用文件|引用目录) @|【上下文：|【文档附件：/.test(t))return t;const n=t.startsWith("【文档附件：");return t.replace(u,(e,o,r,s,i)=>n&&i+e.length===t.length||s.length<=200?e:`${o}${r}（${s.length} 字已折叠，继续对话不受影响）`)}export function buildUserContent(e,t){return t&&t.length?[{type:"text",text:e},...t.map(e=>({type:"image_url",image_url:{url:e.dataUri}}))]:e}export async function runBangCommand(e,t){const n=await r(t,{command:e}),o=n.length>8e3?n.slice(0,8e3)+"\n…（已截断）":n;return{output:n,record:`我在终端直接执行了命令：$ ${e}\n${o}`}}function g(){try{const t=JSON.parse(e.readFileSync(i(),"utf8"));return Array.isArray(t?.entries)?t.entries:[]}catch{return[]}}export function loadHistory(e,t=200){return g().filter(t=>t&&t.cwd===e&&"string"==typeof t.line).slice(-t).map(e=>e.line)}export function appendHistory(t,n){const r=String(n||"").trim();if(r)try{let n=g();const s=[...n].reverse().find(e=>e.cwd===t);if(s&&s.line===r)return;n.push({cwd:t,line:r,ts:Date.now()}),n.length>1e3&&(n=n.slice(-1e3)),e.mkdirSync(o(),{recursive:!0}),e.writeFileSync(i(),JSON.stringify({entries:n}),{mode:384})}catch{}}
|
package/src/lsp.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import t from"node:path";import r from"node:os";import{spawn as n}from"node:child_process";import{
|
|
1
|
+
import e from"node:fs";import t from"node:path";import r from"node:os";import{spawn as n}from"node:child_process";import{accountDataDir as o}from"./account-scope.js";export const OPERATIONS=["goToDefinition","findReferences","goToImplementation","hover"];const i={typescript:{command:"typescript-language-server",args:["--stdio"],extensions:[".ts",".tsx",".mts",".cts",".js",".jsx",".mjs",".cjs"]},python:{command:"pylsp",args:[],extensions:[".py"]},go:{command:"gopls",args:[],extensions:[".go"]},rust:{command:"rust-analyzer",args:[],extensions:[".rs"]},java:{command:"jdtls",args:[],extensions:[".java"]},cpp:{command:"clangd",args:[],extensions:[".c",".h",".cpp",".hpp",".cc",".cxx"]}};let s=null;const c=new Map;export function setProvider(e){s=e&&"function"==typeof e.query?e:null}export function loadServers(){let r={};try{const n=JSON.parse(e.readFileSync(t.join(o(),"lsp.json"),"utf8"));n&&"object"==typeof n.servers&&(r=n.servers)}catch{}return{...i,...r}}const a=e=>"file://"+("win32"===process.platform?"/"+e.replace(/\\/g,"/"):e),p=e=>{const t=String(e||"").replace(/^file:\/\//,""),r="win32"===process.platform?t.replace(/^\//,"").replace(/\//g,"\\"):t;try{return decodeURIComponent(r)}catch{return r}};function u({key:e,def:t,cwd:r}){const o=n(t.command,t.args||[],{cwd:r,stdio:["pipe","pipe","pipe"],env:{...process.env,...t.env||{}}});!function(){if(m)return;m=!0,process.once("exit",()=>{try{shutdownLsp()}catch{}})}(),o.unref(),o.stdout?.unref?.(),o.stderr?.unref?.(),o.stdin?.unref?.();const i=new Map;let s=0,a=Buffer.alloc(0),p=!1;const u=new Set;o.stdout.on("data",e=>{for(a=Buffer.concat([a,e]);;){const e=a.indexOf("\r\n\r\n");if(e<0)break;const t=a.subarray(0,e).toString("utf8"),r=/Content-Length:\s*(\d+)/i.exec(t);if(!r){a=a.subarray(e+4);continue}const n=Number(r[1]);if(a.length<e+4+n)break;const o=a.subarray(e+4,e+4+n).toString("utf8");let s;a=a.subarray(e+4+n);try{s=JSON.parse(o)}catch{continue}if(void 0!==s.id&&i.has(s.id)){const{resolve:e}=i.get(s.id);i.delete(s.id),e(s)}}}),o.stderr.on("data",()=>{});const f=()=>{p=!0;for(const{resolve:e}of i.values())e({error:{message:"语言服务器已退出"}});i.clear(),c.delete(e)};o.on("exit",f),o.on("error",f);const l=e=>{if(p)return;const t=JSON.stringify(e);try{o.stdin.write(`Content-Length: ${Buffer.byteLength(t,"utf8")}\r\n\r\n${t}`)}catch{f()}};return{key:e,child:o,opened:u,request:(e,t,r)=>new Promise(n=>{if(p)return n({error:{message:"语言服务器不可用"}});const o=++s,c=setTimeout(()=>{i.delete(o),n({error:{message:`语言服务器 ${r}ms未响应（大仓库首次索引较慢，稍后重试）`}})},r);i.set(o,{resolve:e=>{clearTimeout(c),n(e)}}),l({jsonrpc:"2.0",id:o,method:e,params:t})}),notify:(e,t)=>l({jsonrpc:"2.0",method:e,params:t}),get dead(){return p},dispose(){try{o.kill("SIGTERM")}catch{}f()}}}async function f(r,n){const o=function(e){const r=t.extname(String(e||"")).toLowerCase();if(!r)return null;const n=loadServers();for(const[e,t]of Object.entries(n))if((t?.extensions||[]).some(e=>String(e).toLowerCase()===r))return{key:e,def:t};return null}(r);if(!o)return{error:`没有为 ${t.extname(r)||"该类型"} 配置语言服务器。可在 ~/.huicai/lsp.json的servers里加一条。`};const{key:i,def:s}=o,p=c.get(i);if(p&&!p.dead)return{client:p};if(!function(r){if(!r)return!1;if(t.isAbsolute(r))try{return e.existsSync(r)}catch{return!1}const n="win32"===process.platform?[".exe",".cmd",".bat",""]:[""];for(const o of String(process.env.PATH||"").split(t.delimiter))if(o)for(const i of n)try{if(e.existsSync(t.join(o,r+i)))return!0}catch{}return!1}(s.command))return{error:`未找到语言服务器可执行文件 ${s.command}（${i}）。请自行安装后重试；也可在 ~/.huicai/lsp.json里把command指向你机器上的实际路径。本工具不会代为安装。`};const f=u({key:i,def:s,cwd:n});c.set(i,f);const l=await f.request("initialize",{processId:process.pid,rootUri:a(n),workspaceFolders:[{uri:a(n),name:t.basename(n)}],capabilities:{textDocument:{definition:{dynamicRegistration:!1},references:{dynamicRegistration:!1},implementation:{dynamicRegistration:!1},hover:{dynamicRegistration:!1,contentFormat:["plaintext","markdown"]},synchronization:{dynamicRegistration:!1,didSave:!1}}}},6e4);return l?.error?(f.dispose(),{error:"语言服务器初始化失败："+(l.error.message||"")}):(f.notify("initialized",{}),{client:f})}export function normalizeLocations(t){const r=Array.isArray(t)?t:t?[t]:[],n=[],o=new Set,i=t=>{try{return e.realpathSync(t)}catch{return t}};for(const t of r){const r=t.uri||t.targetUri,s=t.range||t.targetSelectionRange||t.targetRange;if(!r||!s)continue;const c=p(r),a=(s.start?.line??0)+1,u=(s.start?.character??0)+1,f=`${i(c)}:${a}:${u}`;if(o.has(f))continue;if(o.add(f),n.length>=60)break;let l="";try{l=e.readFileSync(c,"utf8").split("\n")[a-1]?.trim().slice(0,160)||""}catch{}n.push({file:c,line:a,character:u,preview:l})}return n}function l(e){const t=e?.contents;return t?"string"==typeof t?t:Array.isArray(t)?t.map(e=>"string"==typeof e?e:e?.value||"").join("\n"):t.value||"":""}const d={name:"stdio",async query({operation:r,file:n,line:o,character:i,cwd:s}){const c=t.isAbsolute(n)?n:t.resolve(s,n);if(!e.existsSync(c))return{ok:!1,error:`文件不存在：${n}`};const p=await f(c,s);if(p.error)return{ok:!1,error:p.error};!function(r,n){if(r.opened.has(n))return;let o="";try{o=e.readFileSync(n,"utf8")}catch{return}const i=t.extname(n).toLowerCase(),s=".py"===i?"python":".go"===i?"go":".rs"===i?"rust":".java"===i?"java":/\.tsx?$/.test(i)?"typescript":/\.jsx?$/.test(i)?"javascript":"plaintext";r.notify("textDocument/didOpen",{textDocument:{uri:a(n),languageId:s,version:1,text:o}}),r.opened.add(n)}(p.client,c);const u={line:Math.max(0,o-1),character:Math.max(0,i-1)},d={textDocument:{uri:a(c)},position:u},m="goToDefinition"===r?"textDocument/definition":"findReferences"===r?"textDocument/references":"goToImplementation"===r?"textDocument/implementation":"textDocument/hover";"findReferences"===r&&(d.context={includeDeclaration:!0});const g=await p.client.request(m,d,2e4);return g?.error?{ok:!1,error:g.error.message||"查询失败"}:"hover"===r?{ok:!0,hover:l(g?.result)}:{ok:!0,locations:normalizeLocations(g?.result)}}};export function shutdownLsp(){for(const e of[...c.values()])e.dispose();c.clear()}export function resetLsp(){s=null,shutdownLsp()}let m=!1;export function makeLspTool({cwd:e}={}){return{readOnly:!0,timeoutMs:9e4,spec:{type:"function",function:{name:"lsp",description:"语义代码导航：精确回答「这个符号在哪定义的」「谁调用了它」「这个接口有哪些实现」「它是什么类型」。比grep（纯文本匹配）和search_code（相关性召回）精确——重名符号、跨文件调用、接口实现都能准确定位。line与character是**1 基**坐标，指向你要查的那个符号所在位置（通常先用grep/read_file找到它在第几行第几列）。若返回LSP_UNAVAILABLE，说明该语言没有可用的语言服务器，请改用search_code或grep。",parameters:{type:"object",properties:{operation:{type:"string",enum:OPERATIONS,description:"要做的查询：跳定义 / 找引用 / 找实现 / 悬停说明"},file_path:{type:"string",description:"源文件路径（相对项目根或绝对路径）"},line:{type:"number",description:"符号所在行号（1 基）"},character:{type:"number",description:"符号所在列号（1 基）"}},required:["operation","file_path","line","character"]}}},run:async t=>{const r=String(t?.operation||"");if(!OPERATIONS.includes(r))return`不支持的operation：${t?.operation}。可用：${OPERATIONS.join(" / ")}`;const n=String(t?.file_path||"").trim();if(!n)return"缺少file_path。";const o=Number(t?.line),i=Number(t?.character);if(!Number.isFinite(o)||o<1||!Number.isFinite(i)||i<1)return"line与character必须是 ≥1 的数字（1 基坐标）。";let c;try{c=await(s||d).query({operation:r,file:n,line:o,character:i,cwd:e})}catch(e){c={ok:!1,error:e.message}}return function(e,t){if(!t.ok)return`【LSP_UNAVAILABLE】${t.error}\n（这不代表符号不存在。可改用search_code按概念检索，或grep精确匹配。）`;if("hover"===e)return t.hover?t.hover.slice(0,4e3):"（该位置没有可显示的说明）";const r=t.locations||[];return r.length?({goToDefinition:"定义",findReferences:`引用（${r.length} 处）`,goToImplementation:"实现"}[e]||"结果")+"：\n"+r.map(e=>`· ${e.file}:${e.line}:${e.character}${e.preview?"\n "+e.preview:""}`).join("\n"):"findReferences"===e?"没有找到引用。":"没有找到对应位置（可能光标不在符号上，或该符号来自未索引的依赖）。"}(r,c)}}}
|
package/src/mcp.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import t from"node:fs";import e from"node:path";import{spawn as r}from"node:child_process";import{CONFIG_DIR as s}from"./config.js";import{c as n}from"./colors.js";import{compactJsonText as o}from"./toon.js";import{scanUntrustedText as i}from"./skill-scan.js";import{uiResourceUriOf as c}from"./mcp-apps.js";import{assertLimit as a}from"./plan-limits.js";export const MCP_CONFIG_PATH=e.join(s,"mcp.json");export function projectMcpConfigPath(t){return t?e.join(t,".huicai","mcp.json"):null}export const BUILTIN_MCP_CATALOG={filesystem:{description:"读写指定目录下的文件（沙箱化的文件系统访问）",command:"npx",args:["-y","@modelcontextprotocol/server-filesystem"],needsArg:"要开放的目录绝对路径",defaultsToCwd:!0},fetch:{description:"抓取网页并转成适合模型阅读的文本",command:"npx",args:["-y","@modelcontextprotocol/server-fetch"]},git:{description:"读取Git仓库历史、diff、分支信息",command:"npx",args:["-y","@modelcontextprotocol/server-git","--repository"],needsArg:"仓库目录",defaultsToCwd:!0},memory:{description:"基于知识图谱的长期记忆存取",command:"npx",args:["-y","@modelcontextprotocol/server-memory"]},sequentialthinking:{description:"结构化的分步推理辅助",command:"npx",args:["-y","@modelcontextprotocol/server-sequential-thinking"]},sqlite:{description:"查询/操作本地SQLite数据库",command:"npx",args:["-y","@modelcontextprotocol/server-sqlite","--db-path"],needsArg:"数据库文件路径"}};const l=2e4,p=12e4,u="2024-11-05";export class McpStdioClient{constructor(t,e){this.name=t,this.spec=e,this.child=null,this.buffer="",this.nextId=1,this.pending=new Map,this.closed=!1}async start(){const t={...process.env,...this.spec.env||{}};this.child=r(this.spec.command,this.spec.args||[],{env:t,stdio:["pipe","pipe","pipe"]}),this.child.on("error",t=>this._failAll(new Error(`MCP[${this.name}] 进程错误：${t.message}`))),this.child.on("exit",()=>{this.closed=!0,this._failAll(new Error(`MCP[${this.name}] 进程已退出`))}),this.child.stdout.setEncoding("utf8"),this.child.stdout.on("data",t=>this._onData(t)),this.child.stderr.on("data",()=>{}),await this._request("initialize",{protocolVersion:u,capabilities:{},clientInfo:{name:"huicai-cli",version:"0.2.0"}},l),this._notify("notifications/initialized",{})}async listTools(){return h(t=>this._request("tools/list",t,l),this.name)}async callTool(t,e){return d(await this._request("tools/call",{name:t,arguments:e||{}},p),t)}async readResource(t){const e=await this._request("resources/read",{uri:t},l);return(Array.isArray(e?.contents)?e.contents:[])[0]||null}close(){this.closed=!0;try{this.child?.kill("SIGTERM")}catch{}}_onData(t){let e;for(this.buffer+=t;(e=this.buffer.indexOf("\n"))>=0;){const t=this.buffer.slice(0,e).trim();if(this.buffer=this.buffer.slice(e+1),!t)continue;let r;try{r=JSON.parse(t)}catch{continue}if(null!=r.id&&this.pending.has(r.id)){const{resolve:t,reject:e,timer:s}=this.pending.get(r.id);clearTimeout(s),this.pending.delete(r.id),r.error?e(new Error(r.error.message||"MCP错误")):t(r.result)}}}_request(t,e,r){if(this.closed)return Promise.reject(new Error(`MCP[${this.name}] 已关闭`));const s=this.nextId++,n=JSON.stringify({jsonrpc:"2.0",id:s,method:t,params:e})+"\n";return new Promise((e,o)=>{const i=setTimeout(()=>{this.pending.delete(s),o(new Error(`MCP[${this.name}] ${t} 超时`))},r);this.pending.set(s,{resolve:e,reject:o,timer:i});try{this.child.stdin.write(n)}catch(t){clearTimeout(i),this.pending.delete(s),o(t)}})}_notify(t,e){try{this.child.stdin.write(JSON.stringify({jsonrpc:"2.0",method:t,params:e})+"\n")}catch{}}_failAll(t){for(const{reject:e,timer:r}of this.pending.values())clearTimeout(r),e(t);this.pending.clear()}}async function h(t,e){const r=[];let s;for(let e=0;e<20;e++){const e=await t(s?{cursor:s}:{});if(Array.isArray(e?.tools)&&r.push(...e.tools),s=e?.nextCursor,!s)return r}return process.emitWarning?.(`MCP[${e}] 工具清单页数超过 20，仅取前 ${r.length} 个`),r}function d(t,e){const r=(Array.isArray(t?.content)?t.content:[]).map(t=>"text"===t?.type?t.text:t?.text||JSON.stringify(t)).filter(Boolean).join("\n");return t?.isError?`工具报错：${r||"(无详情)"}`:r?o(r,{name:e.replace(/[^\w.-]/g,"_")}):"(无输出)"}export function parseSseFrame(t){const e=String(t||"").split(/\r?\n/).filter(t=>t.startsWith("data:")).map(t=>t.slice(5).trim()).join("");if(!e)return null;try{return JSON.parse(e)}catch{return null}}export class McpHttpClient{constructor(t,e){this.name=t,this.spec=e,this.url=String(e.url||"").trim(),this.sessionId="",this.protocolVersion=u,this.nextId=1,this.closed=!1}async start(){if(!/^https?:\/\//i.test(this.url))throw new Error(`仅支持http/https的MCP地址（当前：${this.url||"(空)"}）`);const t=await this._request("initialize",{protocolVersion:u,capabilities:{},clientInfo:{name:"huicai-cli",version:"0.2.0"}},l);t?.protocolVersion&&(this.protocolVersion=String(t.protocolVersion)),await this._notify("notifications/initialized",{})}async listTools(){return h(t=>this._request("tools/list",t,l),this.name)}async callTool(t,e){return d(await this._request("tools/call",{name:t,arguments:e||{}},p),t)}async readResource(t){const e=await this._request("resources/read",{uri:t},l);return(Array.isArray(e?.contents)?e.contents:[])[0]||null}close(){if(this.closed=!0,this.sessionId)try{fetch(this.url,{method:"DELETE",headers:this._headers()}).catch(()=>{})}catch{}}host(){try{return new URL(this.url).host}catch{return""}}_headers(){const t={"Content-Type":"application/json",Accept:"application/json, text/event-stream","User-Agent":"huicai-cli",...this.spec.headers||{}};return this.sessionId&&(t["Mcp-Session-Id"]=this.sessionId),this.protocolVersion&&(t["MCP-Protocol-Version"]=this.protocolVersion),t}async _post(t,e){if(this.closed)throw new Error(`MCP[${this.name}] 已关闭`);let r;try{r=await fetch(this.url,{method:"POST",headers:this._headers(),body:JSON.stringify(t),signal:AbortSignal.timeout(e)})}catch(t){throw new Error(`MCP[${this.name}] 请求失败：${t?.cause?.message||t.message}`)}const s=r.headers.get("mcp-session-id");return s&&(this.sessionId=s),r}async _notify(t,e){try{await this._post({jsonrpc:"2.0",method:t,params:e},l)}catch{}}async _request(t,e,r){const s=this.nextId++,n=await this._post({jsonrpc:"2.0",id:s,method:t,params:e},r);if(!n.ok){const e=await n.text().catch(()=>"");throw new Error(`MCP[${this.name}] ${t} HTTP ${n.status}`+(e?`：${e.replace(/\s+/g," ").slice(0,200)}`:""))}const o=(n.headers.get("content-type")||"").toLowerCase().includes("text/event-stream")?await this._readSse(n,s,t):await n.json().catch(()=>null);if(!o)throw new Error(`MCP[${this.name}] ${t} 无有效响应`);if(o.error)throw new Error(o.error.message||`MCP[${this.name}] ${t} 出错`);return o.result}async _readSse(t,e,r){const s=t.body?.getReader?.();if(!s)return null;const n=new TextDecoder;let o="";try{for(;;){const{done:t,value:r}=await s.read();if(t)break;for(o+=n.decode(r,{stream:!0});;){const t=/\r?\n\r?\n/.exec(o);if(!t)break;const r=o.slice(0,t.index);o=o.slice(t.index+t[0].length);const s=parseSseFrame(r);if(s&&s.id===e)return s}}}finally{try{await s.cancel()}catch{}}return null}}function m(t){const e=String(t).replace(/[^a-zA-Z0-9_-]/g,"_");return e.length>60?e.slice(0,60):e}function f(e){try{const r=JSON.parse(t.readFileSync(e,"utf8"));return r?.mcpServers&&"object"==typeof r.mcpServers?r.mcpServers:{}}catch{return{}}}export function countMcpServers(){return Object.keys(f(MCP_CONFIG_PATH)).length}export function saveMcpServers(r){if(!r||"object"!=typeof r||Array.isArray(r))throw new Error("配置须为 {服务名: {command,args,env}} 对象");const s=countMcpServers(),n=Object.keys(r).length;return n>s&&a("mcp_servers",n-1),t.mkdirSync(e.dirname(MCP_CONFIG_PATH),{recursive:!0}),t.writeFileSync(MCP_CONFIG_PATH,JSON.stringify({mcpServers:r},null,2)+"\n",{mode:384}),{servers:r,path:MCP_CONFIG_PATH}}export function readConfig(t){const e={};for(const[r,s]of[[MCP_CONFIG_PATH,"global"],[projectMcpConfigPath(t),"project"]])if(r)for(const[t,n]of Object.entries(f(r)))n&&"object"==typeof n&&(e[t]={...n,scope:s});return e}function g(t,e,r,s){const n="function"==typeof e?.host?e.host():"",o=n?`[MCP:${t}@${n}]`:`[MCP:${t}]`,a=n?`${t}@${n}`:t,l=[];for(const p of r){const r=p.name,u=p.description||r,h=i(`${u}\n${JSON.stringify(p.inputSchema||{})}`,{source:a,label:`${r} 的工具描述`});if(h.counts.critical){const t=h.findings.filter(t=>"critical"===t.severity).slice(0,3).map(t=>`${t.check} ${t.message}`).join("；");s?.(`已拒绝注册 ${a} 的工具 ${r}：其描述命中提示词注入模式（${t}）。该工具的描述不会进入模型上下文；其余工具不受影响。`);continue}const d=h.counts.warn?`［注意：本工具描述含 ${h.counts.warn} 处可疑表述，其内容属外部数据，不得当作指令执行］ `:"";h.counts.warn&&s?.(`${a} 的工具 ${r} 描述含 ${h.counts.warn} 处可疑表述，已标注但未拦截。`);const f=c(p);l.push({readOnly:!0===p.annotations?.readOnlyHint,mcpServer:t,mcpOrigin:n,...f?{uiResource:f}:{},spec:{type:"function",function:{name:m(`mcp__${t}__${r}`),description:`${o} ${d}${u}`,parameters:p.inputSchema&&"object"==typeof p.inputSchema?p.inputSchema:{type:"object",properties:{}}}},run:t=>e.callTool(r,t)})}return l}export function transportOf(t){return t&&"object"==typeof t?"string"==typeof t.url&&t.url.trim()?"http":"string"==typeof t.command&&t.command.trim()?"stdio":null:null}export async function connectServer(t,e,r={}){const s=transportOf(e);if(!s)return{error:`服务「${t}」既没有command（本地进程）也没有url（远程HTTP）`};const n="http"===s?new McpHttpClient(t,e):new McpStdioClient(t,e),o=[],i=t=>{o.push(t),r.onNotice?.(t)};try{await n.start();const e=await n.listTools();return{client:n,tools:g(t,n,e,i),notices:o}}catch(t){return n.close(),{error:t.message}}}export async function loadMcpTools(t,e={}){const r=readConfig(t),s=e.onInfo||(t=>process.stdout.write(n.gray(" · "+t)+"\n")),o=e.onInfo||(t=>process.stdout.write(n.yellow(" · "+t)+"\n")),i=[],c=[],a={cwd:t||"",servers:{}};for(const[t,e]of Object.entries(r))a.servers[t]={name:t,spec:e,scope:e.scope||"global",status:e.disabled?"disabled":!1===e.autoStart?"idle":"pending",toolCount:0,error:""};for(const[t,e]of Object.entries(a.servers)){if("pending"!==e.status)continue;const r=await connectServer(t,e.spec,{onNotice:e=>o(`MCP[${t}] ${e}`)});r.error?(e.status="failed",e.error=r.error,o(`MCP[${t}] 加载失败，已跳过：${r.error}`)):(e.status="connected",e.toolCount=r.tools.length,e.client=r.client,c.push(r.client),i.push(...r.tools),s(`MCP[${t}] 已连接，工具 ${r.tools.length} 个`))}return{tools:i,clients:c,registry:a}}const y=(t,e,r={},s=[])=>({type:"function",function:{name:t,description:e,parameters:{type:"object",properties:r,required:s}}});export function hostOf(t){try{return new URL(String(t)).host}catch{return""}}export function mcpCatalogText(t){const e=Object.values(t?.servers||{}),r=[];for(const t of e){const e="connected"===t.status?`已连接（${t.toolCount} 个工具）`:"idle"===t.status?"已登记，未连接（需要时connect_mcp连上）":"failed"===t.status?`连接失败：${String(t.error||"").slice(0,60)}`:"disabled"===t.status?"已禁用":"未连接",s="http"===transportOf(t.spec)?` · 远程 ${hostOf(t.spec?.url)}`:"";r.push(`- ${t.name}（${"project"===t.scope?"项目级":"全局"}${s}）：${e}`)}const s=Object.entries(BUILTIN_MCP_CATALOG).filter(([e])=>!t?.servers?.[e]).map(([t,e])=>`- ${t}：${e.description}（未安装，可connect_mcp一键装上）`);return s.length&&r.push(...s),r.join("\n")}export function makeMcpConnectTools({registry:t,register:e,clients:r=[],cwd:s=""}){return t&&"function"==typeof e?[{readOnly:!0,spec:y("list_mcp_servers","列出本机可用的MCP服务（外部能力连接器）：哪些已连上、哪些登记了没连、哪些是内置目录里可一键装的。当任务需要当前工具覆盖不到的能力（访问某个系统、某类数据源）时先看这里。"),run:async()=>{const e=mcpCatalogText(t);return e?"MCP服务：\n"+e+"\n\n用connect_mcp(name) 连接后，该服务的工具会以mcp__<服务名>__<工具名> 出现，下一轮即可调用。":"本机没有配置任何MCP服务，也没有内置可装的服务。"}},{readOnly:!1,spec:y("connect_mcp","连接一个MCP服务并把它的工具挂进当前会话（下一轮就能调用）。用于会话进行中发现「这活儿需要某个外部能力」时按需接入，不必重开会话。已登记但未自动连接的服务直接给名字即可；内置目录里的服务会用npx拉起（首次会下载，需用户同意）。",{name:{type:"string",description:"服务名（见list_mcp_servers）"},argument:{type:"string",description:"（可选）该服务需要的参数，如filesystem的目录、sqlite的库文件路径"},url:{type:"string",description:"（可选）远程MCP服务的http(s) 地址；给了它就按远程连接，不拉本地进程"}},["name"]),run:async n=>{const o=String(n?.name||"").trim();if(!o)return"缺少服务名。";const i=t.servers[o];if("connected"===i?.status)return`MCP[${o}] 已经连接（${i.toolCount} 个工具），可直接调用mcp__${o}__* 工具。`;if("disabled"===i?.status)return`MCP[${o}] 已被禁用（在mcp.json里去掉 "disabled": true后可用）。`;let c=i?.spec;const a=String(n?.url||"").trim();if(a){if(!/^https?:\/\//i.test(a))return`远程MCP地址必须是http(s)：${a}`;c={url:a,headers:i?.spec?.headers||{}},t.servers[o]={name:o,spec:c,scope:i?.scope||"session",status:"pending",toolCount:0,error:""}}if(!c){const e=BUILTIN_MCP_CATALOG[o];if(!e)return`未找到MCP服务「${o}」。可用的见list_mcp_servers；要接入新服务请在 ~/.huicai/mcp.json或 <项目>/.huicai/mcp.json的mcpServers里登记。`;const r=String(n?.argument||"").trim()||(e.defaultsToCwd?s:"");if(e.needsArg&&!r)return`服务「${o}」需要参数：${e.needsArg}。请带argument再调一次。`;c={command:e.command,args:r?[...e.args,r]:[...e.args]},t.servers[o]={name:o,spec:c,scope:"builtin",status:"pending",toolCount:0,error:""}}const l=await connectServer(o,c),p=t.servers[o],u=(l.notices||[]).length?"\n⚠️ "+l.notices.join("\n⚠️ "):"";if(l.error)return p.status="failed",p.error=l.error,`连接MCP[${o}] 失败：${l.error}。（若是内置服务，可能是网络受限导致npx拉包失败；可改用本机已有工具完成任务。）`;p.status="connected",p.toolCount=l.tools.length,p.client=l.client,r.push(l.client),e(l.tools);const h=l.tools.map(t=>t.spec.function.name).slice(0,12).join("、");return`✅ 已连接MCP[${o}]${hostOf(c.url)?`（远程 ${hostOf(c.url)}）`:""}，新增 ${l.tools.length} 个工具：${h}`+(l.tools.length>12?" …":"")+"\n下一轮起即可直接调用它们。"+u}}]:[]}export function closeMcp(t){for(const e of t||[])e.close()}
|
|
1
|
+
import t from"node:fs";import{accountDataDir as e}from"./account-scope.js";import r from"node:path";import{spawn as s}from"node:child_process";import{c as n}from"./colors.js";import{compactJsonText as o}from"./toon.js";import{scanUntrustedText as i}from"./skill-scan.js";import{uiResourceUriOf as c}from"./mcp-apps.js";import{assertLimit as a}from"./plan-limits.js";export const mcpConfigPath=()=>r.join(e(),"mcp.json");export const MCP_CONFIG_PATH=mcpConfigPath();export function projectMcpConfigPath(t){return t?r.join(t,".huicai","mcp.json"):null}export const BUILTIN_MCP_CATALOG={filesystem:{description:"读写指定目录下的文件（沙箱化的文件系统访问）",command:"npx",args:["-y","@modelcontextprotocol/server-filesystem"],needsArg:"要开放的目录绝对路径",defaultsToCwd:!0},fetch:{description:"抓取网页并转成适合模型阅读的文本",command:"npx",args:["-y","@modelcontextprotocol/server-fetch"]},git:{description:"读取Git仓库历史、diff、分支信息",command:"npx",args:["-y","@modelcontextprotocol/server-git","--repository"],needsArg:"仓库目录",defaultsToCwd:!0},memory:{description:"基于知识图谱的长期记忆存取",command:"npx",args:["-y","@modelcontextprotocol/server-memory"]},sequentialthinking:{description:"结构化的分步推理辅助",command:"npx",args:["-y","@modelcontextprotocol/server-sequential-thinking"]},sqlite:{description:"查询/操作本地SQLite数据库",command:"npx",args:["-y","@modelcontextprotocol/server-sqlite","--db-path"],needsArg:"数据库文件路径"}};const l=2e4,p=12e4,u="2024-11-05";export class McpStdioClient{constructor(t,e){this.name=t,this.spec=e,this.child=null,this.buffer="",this.nextId=1,this.pending=new Map,this.closed=!1}async start(){const t={...process.env,...this.spec.env||{}};this.child=s(this.spec.command,this.spec.args||[],{env:t,stdio:["pipe","pipe","pipe"]}),this.child.on("error",t=>this._failAll(new Error(`MCP[${this.name}] 进程错误：${t.message}`))),this.child.on("exit",()=>{this.closed=!0,this._failAll(new Error(`MCP[${this.name}] 进程已退出`))}),this.child.stdout.setEncoding("utf8"),this.child.stdout.on("data",t=>this._onData(t)),this.child.stderr.on("data",()=>{}),await this._request("initialize",{protocolVersion:u,capabilities:{},clientInfo:{name:"huicai-cli",version:"0.2.0"}},l),this._notify("notifications/initialized",{})}async listTools(){return h(t=>this._request("tools/list",t,l),this.name)}async callTool(t,e){return d(await this._request("tools/call",{name:t,arguments:e||{}},p),t)}async readResource(t){const e=await this._request("resources/read",{uri:t},l);return(Array.isArray(e?.contents)?e.contents:[])[0]||null}close(){this.closed=!0;try{this.child?.kill("SIGTERM")}catch{}}_onData(t){let e;for(this.buffer+=t;(e=this.buffer.indexOf("\n"))>=0;){const t=this.buffer.slice(0,e).trim();if(this.buffer=this.buffer.slice(e+1),!t)continue;let r;try{r=JSON.parse(t)}catch{continue}if(null!=r.id&&this.pending.has(r.id)){const{resolve:t,reject:e,timer:s}=this.pending.get(r.id);clearTimeout(s),this.pending.delete(r.id),r.error?e(new Error(r.error.message||"MCP错误")):t(r.result)}}}_request(t,e,r){if(this.closed)return Promise.reject(new Error(`MCP[${this.name}] 已关闭`));const s=this.nextId++,n=JSON.stringify({jsonrpc:"2.0",id:s,method:t,params:e})+"\n";return new Promise((e,o)=>{const i=setTimeout(()=>{this.pending.delete(s),o(new Error(`MCP[${this.name}] ${t} 超时`))},r);this.pending.set(s,{resolve:e,reject:o,timer:i});try{this.child.stdin.write(n)}catch(t){clearTimeout(i),this.pending.delete(s),o(t)}})}_notify(t,e){try{this.child.stdin.write(JSON.stringify({jsonrpc:"2.0",method:t,params:e})+"\n")}catch{}}_failAll(t){for(const{reject:e,timer:r}of this.pending.values())clearTimeout(r),e(t);this.pending.clear()}}async function h(t,e){const r=[];let s;for(let e=0;e<20;e++){const e=await t(s?{cursor:s}:{});if(Array.isArray(e?.tools)&&r.push(...e.tools),s=e?.nextCursor,!s)return r}return process.emitWarning?.(`MCP[${e}] 工具清单页数超过 20，仅取前 ${r.length} 个`),r}function d(t,e){const r=(Array.isArray(t?.content)?t.content:[]).map(t=>"text"===t?.type?t.text:t?.text||JSON.stringify(t)).filter(Boolean).join("\n");return t?.isError?`工具报错：${r||"(无详情)"}`:r?o(r,{name:e.replace(/[^\w.-]/g,"_")}):"(无输出)"}export function parseSseFrame(t){const e=String(t||"").split(/\r?\n/).filter(t=>t.startsWith("data:")).map(t=>t.slice(5).trim()).join("");if(!e)return null;try{return JSON.parse(e)}catch{return null}}export class McpHttpClient{constructor(t,e){this.name=t,this.spec=e,this.url=String(e.url||"").trim(),this.sessionId="",this.protocolVersion=u,this.nextId=1,this.closed=!1}async start(){if(!/^https?:\/\//i.test(this.url))throw new Error(`仅支持http/https的MCP地址（当前：${this.url||"(空)"}）`);const t=await this._request("initialize",{protocolVersion:u,capabilities:{},clientInfo:{name:"huicai-cli",version:"0.2.0"}},l);t?.protocolVersion&&(this.protocolVersion=String(t.protocolVersion)),await this._notify("notifications/initialized",{})}async listTools(){return h(t=>this._request("tools/list",t,l),this.name)}async callTool(t,e){return d(await this._request("tools/call",{name:t,arguments:e||{}},p),t)}async readResource(t){const e=await this._request("resources/read",{uri:t},l);return(Array.isArray(e?.contents)?e.contents:[])[0]||null}close(){if(this.closed=!0,this.sessionId)try{fetch(this.url,{method:"DELETE",headers:this._headers()}).catch(()=>{})}catch{}}host(){try{return new URL(this.url).host}catch{return""}}_headers(){const t={"Content-Type":"application/json",Accept:"application/json, text/event-stream","User-Agent":"huicai-cli",...this.spec.headers||{}};return this.sessionId&&(t["Mcp-Session-Id"]=this.sessionId),this.protocolVersion&&(t["MCP-Protocol-Version"]=this.protocolVersion),t}async _post(t,e){if(this.closed)throw new Error(`MCP[${this.name}] 已关闭`);let r;try{r=await fetch(this.url,{method:"POST",headers:this._headers(),body:JSON.stringify(t),signal:AbortSignal.timeout(e)})}catch(t){throw new Error(`MCP[${this.name}] 请求失败：${t?.cause?.message||t.message}`)}const s=r.headers.get("mcp-session-id");return s&&(this.sessionId=s),r}async _notify(t,e){try{await this._post({jsonrpc:"2.0",method:t,params:e},l)}catch{}}async _request(t,e,r){const s=this.nextId++,n=await this._post({jsonrpc:"2.0",id:s,method:t,params:e},r);if(!n.ok){const e=await n.text().catch(()=>"");throw new Error(`MCP[${this.name}] ${t} HTTP ${n.status}`+(e?`：${e.replace(/\s+/g," ").slice(0,200)}`:""))}const o=(n.headers.get("content-type")||"").toLowerCase().includes("text/event-stream")?await this._readSse(n,s,t):await n.json().catch(()=>null);if(!o)throw new Error(`MCP[${this.name}] ${t} 无有效响应`);if(o.error)throw new Error(o.error.message||`MCP[${this.name}] ${t} 出错`);return o.result}async _readSse(t,e,r){const s=t.body?.getReader?.();if(!s)return null;const n=new TextDecoder;let o="";try{for(;;){const{done:t,value:r}=await s.read();if(t)break;for(o+=n.decode(r,{stream:!0});;){const t=/\r?\n\r?\n/.exec(o);if(!t)break;const r=o.slice(0,t.index);o=o.slice(t.index+t[0].length);const s=parseSseFrame(r);if(s&&s.id===e)return s}}}finally{try{await s.cancel()}catch{}}return null}}function m(t){const e=String(t).replace(/[^a-zA-Z0-9_-]/g,"_");return e.length>60?e.slice(0,60):e}function f(e){try{const r=JSON.parse(t.readFileSync(e,"utf8"));return r?.mcpServers&&"object"==typeof r.mcpServers?r.mcpServers:{}}catch{return{}}}export function countMcpServers(){return Object.keys(f(mcpConfigPath())).length}export function saveMcpServers(e){if(!e||"object"!=typeof e||Array.isArray(e))throw new Error("配置须为 {服务名: {command,args,env}} 对象");const s=countMcpServers(),n=Object.keys(e).length;return n>s&&a("mcp_servers",n-1),t.mkdirSync(r.dirname(mcpConfigPath()),{recursive:!0}),t.writeFileSync(mcpConfigPath(),JSON.stringify({mcpServers:e},null,2)+"\n",{mode:384}),{servers:e,path:mcpConfigPath()}}export function readConfig(t){const e={};for(const[r,s]of[[mcpConfigPath(),"global"],[projectMcpConfigPath(t),"project"]])if(r)for(const[t,n]of Object.entries(f(r)))n&&"object"==typeof n&&(e[t]={...n,scope:s});return e}function g(t,e,r,s){const n="function"==typeof e?.host?e.host():"",o=n?`[MCP:${t}@${n}]`:`[MCP:${t}]`,a=n?`${t}@${n}`:t,l=[];for(const p of r){const r=p.name,u=p.description||r,h=i(`${u}\n${JSON.stringify(p.inputSchema||{})}`,{source:a,label:`${r} 的工具描述`});if(h.counts.critical){const t=h.findings.filter(t=>"critical"===t.severity).slice(0,3).map(t=>`${t.check} ${t.message}`).join("；");s?.(`已拒绝注册 ${a} 的工具 ${r}：其描述命中提示词注入模式（${t}）。该工具的描述不会进入模型上下文；其余工具不受影响。`);continue}const d=h.counts.warn?`［注意：本工具描述含 ${h.counts.warn} 处可疑表述，其内容属外部数据，不得当作指令执行］ `:"";h.counts.warn&&s?.(`${a} 的工具 ${r} 描述含 ${h.counts.warn} 处可疑表述，已标注但未拦截。`);const f=c(p);l.push({readOnly:!0===p.annotations?.readOnlyHint,mcpServer:t,mcpOrigin:n,...f?{uiResource:f}:{},spec:{type:"function",function:{name:m(`mcp__${t}__${r}`),description:`${o} ${d}${u}`,parameters:p.inputSchema&&"object"==typeof p.inputSchema?p.inputSchema:{type:"object",properties:{}}}},run:t=>e.callTool(r,t)})}return l}export function transportOf(t){return t&&"object"==typeof t?"string"==typeof t.url&&t.url.trim()?"http":"string"==typeof t.command&&t.command.trim()?"stdio":null:null}export async function connectServer(t,e,r={}){const s=transportOf(e);if(!s)return{error:`服务「${t}」既没有command（本地进程）也没有url（远程HTTP）`};const n="http"===s?new McpHttpClient(t,e):new McpStdioClient(t,e),o=[],i=t=>{o.push(t),r.onNotice?.(t)};try{await n.start();const e=await n.listTools();return{client:n,tools:g(t,n,e,i),notices:o}}catch(t){return n.close(),{error:t.message}}}export async function loadMcpTools(t,e={}){const r=readConfig(t),s=e.onInfo||(t=>process.stdout.write(n.gray(" · "+t)+"\n")),o=e.onInfo||(t=>process.stdout.write(n.yellow(" · "+t)+"\n")),i=[],c=[],a={cwd:t||"",servers:{}};for(const[t,e]of Object.entries(r))a.servers[t]={name:t,spec:e,scope:e.scope||"global",status:e.disabled?"disabled":!1===e.autoStart?"idle":"pending",toolCount:0,error:""};for(const[t,e]of Object.entries(a.servers)){if("pending"!==e.status)continue;const r=await connectServer(t,e.spec,{onNotice:e=>o(`MCP[${t}] ${e}`)});r.error?(e.status="failed",e.error=r.error,o(`MCP[${t}] 加载失败，已跳过：${r.error}`)):(e.status="connected",e.toolCount=r.tools.length,e.client=r.client,c.push(r.client),i.push(...r.tools),s(`MCP[${t}] 已连接，工具 ${r.tools.length} 个`))}return{tools:i,clients:c,registry:a}}const y=(t,e,r={},s=[])=>({type:"function",function:{name:t,description:e,parameters:{type:"object",properties:r,required:s}}});export function hostOf(t){try{return new URL(String(t)).host}catch{return""}}export function mcpCatalogText(t){const e=Object.values(t?.servers||{}),r=[];for(const t of e){const e="connected"===t.status?`已连接（${t.toolCount} 个工具）`:"idle"===t.status?"已登记，未连接（需要时connect_mcp连上）":"failed"===t.status?`连接失败：${String(t.error||"").slice(0,60)}`:"disabled"===t.status?"已禁用":"未连接",s="http"===transportOf(t.spec)?` · 远程 ${hostOf(t.spec?.url)}`:"";r.push(`- ${t.name}（${"project"===t.scope?"项目级":"全局"}${s}）：${e}`)}const s=Object.entries(BUILTIN_MCP_CATALOG).filter(([e])=>!t?.servers?.[e]).map(([t,e])=>`- ${t}：${e.description}（未安装，可connect_mcp一键装上）`);return s.length&&r.push(...s),r.join("\n")}export function makeMcpConnectTools({registry:t,register:e,clients:r=[],cwd:s=""}){return t&&"function"==typeof e?[{readOnly:!0,spec:y("list_mcp_servers","列出本机可用的MCP服务（外部能力连接器）：哪些已连上、哪些登记了没连、哪些是内置目录里可一键装的。当任务需要当前工具覆盖不到的能力（访问某个系统、某类数据源）时先看这里。"),run:async()=>{const e=mcpCatalogText(t);return e?"MCP服务：\n"+e+"\n\n用connect_mcp(name) 连接后，该服务的工具会以mcp__<服务名>__<工具名> 出现，下一轮即可调用。":"本机没有配置任何MCP服务，也没有内置可装的服务。"}},{readOnly:!1,spec:y("connect_mcp","连接一个MCP服务并把它的工具挂进当前会话（下一轮就能调用）。用于会话进行中发现「这活儿需要某个外部能力」时按需接入，不必重开会话。已登记但未自动连接的服务直接给名字即可；内置目录里的服务会用npx拉起（首次会下载，需用户同意）。",{name:{type:"string",description:"服务名（见list_mcp_servers）"},argument:{type:"string",description:"（可选）该服务需要的参数，如filesystem的目录、sqlite的库文件路径"},url:{type:"string",description:"（可选）远程MCP服务的http(s) 地址；给了它就按远程连接，不拉本地进程"}},["name"]),run:async n=>{const o=String(n?.name||"").trim();if(!o)return"缺少服务名。";const i=t.servers[o];if("connected"===i?.status)return`MCP[${o}] 已经连接（${i.toolCount} 个工具），可直接调用mcp__${o}__* 工具。`;if("disabled"===i?.status)return`MCP[${o}] 已被禁用（在mcp.json里去掉 "disabled": true后可用）。`;let c=i?.spec;const a=String(n?.url||"").trim();if(a){if(!/^https?:\/\//i.test(a))return`远程MCP地址必须是http(s)：${a}`;c={url:a,headers:i?.spec?.headers||{}},t.servers[o]={name:o,spec:c,scope:i?.scope||"session",status:"pending",toolCount:0,error:""}}if(!c){const e=BUILTIN_MCP_CATALOG[o];if(!e)return`未找到MCP服务「${o}」。可用的见list_mcp_servers；要接入新服务请在 ~/.huicai/mcp.json或 <项目>/.huicai/mcp.json的mcpServers里登记。`;const r=String(n?.argument||"").trim()||(e.defaultsToCwd?s:"");if(e.needsArg&&!r)return`服务「${o}」需要参数：${e.needsArg}。请带argument再调一次。`;c={command:e.command,args:r?[...e.args,r]:[...e.args]},t.servers[o]={name:o,spec:c,scope:"builtin",status:"pending",toolCount:0,error:""}}const l=await connectServer(o,c),p=t.servers[o],u=(l.notices||[]).length?"\n⚠️ "+l.notices.join("\n⚠️ "):"";if(l.error)return p.status="failed",p.error=l.error,`连接MCP[${o}] 失败：${l.error}。（若是内置服务，可能是网络受限导致npx拉包失败；可改用本机已有工具完成任务。）`;p.status="connected",p.toolCount=l.tools.length,p.client=l.client,r.push(l.client),e(l.tools);const h=l.tools.map(t=>t.spec.function.name).slice(0,12).join("、");return`✅ 已连接MCP[${o}]${hostOf(c.url)?`（远程 ${hostOf(c.url)}）`:""}，新增 ${l.tools.length} 个工具：${h}`+(l.tools.length>12?" …":"")+"\n下一轮起即可直接调用它们。"+u}}]:[]}export function closeMcp(t){for(const e of t||[])e.close()}
|
package/src/memory.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import r
|
|
1
|
+
import e from"node:fs";import{accountDataDir as r}from"./account-scope.js";import t from"node:path";import n from"node:crypto";import{tokenize as o}from"./codeindex.js";export const memoryDir=()=>t.join(r(),"memory");export const MEMORY_DIR=memoryDir();const i=6e3;export function memoryPath(e){const r=n.createHash("sha1").update(String(e||"")).digest("hex").slice(0,12);return t.join(memoryDir(),r+".md")}export function loadMemory(r){try{const t=e.readFileSync(memoryPath(r),"utf8");if(t.length<=i)return t.trim();const n=t.slice(-6e3),o=n.indexOf("\n");return("…（较早记忆已省略）\n"+(o>=0?n.slice(o+1):n)).trim()}catch{return""}}export function appendMemory(r,t){const n=String(t||"").trim();if(!n)return!1;try{e.mkdirSync(memoryDir(),{recursive:!0});const t=`- [${(new Date).toISOString().slice(0,10)}] ${n.slice(0,1e3)}\n`,o=memoryPath(r);return e.existsSync(o)?e.appendFileSync(o,t):e.writeFileSync(o,`# 长期记忆（${r}）\n\n${t}`,{mode:384}),!0}catch{return!1}}export function replaceMemory(r,t){try{return e.mkdirSync(memoryDir(),{recursive:!0}),e.writeFileSync(memoryPath(r),String(t||"").slice(0,12e3),{mode:384}),!0}catch{return!1}}export function clearMemory(r){try{return e.unlinkSync(memoryPath(r)),!0}catch{return!1}}export function readMemoryRaw(r){try{return e.readFileSync(memoryPath(r),"utf8")}catch{return""}}export function makeRecallTool({cwd:e}){return{readOnly:!0,spec:{type:"function",function:{name:"recall_memory",description:"在本项目的长期记忆里按关键词检索相关的过往结论/踩坑/约定（连超出常驻注入预算的历史也能查到）。开始一个任务前，若怀疑「以前处理过类似问题」，可先recall一下，少走弯路。",parameters:{type:"object",properties:{query:{type:"string",description:"关键词/概念"}},required:["query"]}}},run:async r=>{const t=String(r?.query||"").trim();if(!t)return"缺少query";const n=readMemoryRaw(e).split("\n").map(e=>e.trim()).filter(e=>e.startsWith("- "));if(!n.length)return"本项目暂无可检索的长期记忆。";const i=new Set(o(t));if(!i.size)return"查询词无有效token。";const c=n.map(e=>{let r=0;for(const t of new Set(o(e)))i.has(t)&&r++;return{e:e,s:r}}).filter(e=>e.s>0).sort((e,r)=>r.s-e.s).slice(0,6);return c.length?"相关的历史记忆：\n"+c.map(e=>e.e).join("\n"):`未在长期记忆中找到与「${t}」相关的条目。`}}}export function makeMemoryTool({cwd:e,emit:r}){return{readOnly:!0,spec:{type:"function",function:{name:"memory_write",description:"把值得跨会话记住的信息写入本项目的长期记忆（用户偏好、项目关键事实/约定、踩坑结论等）。新会话会自动注入这些记忆。仅记「以后仍然有用」的稳定信息；不要记一次性细节或大段代码。mode=append追加一条（默认）；mode=replace用整理后的全文替换（做记忆瘦身时用）。",parameters:{type:"object",properties:{content:{type:"string",description:"要记住的内容（简洁的一条事实/偏好/结论）"},mode:{type:"string",enum:["append","replace"],description:"追加或整体替换（默认append）"}},required:["content"]}}},run:async t=>{const n=String(t?.content||"").trim();if(!n)return"memory_write需要非空content";if("replace"===t?.mode?replaceMemory(e,n):appendMemory(e,n)){try{r?.(n)}catch{}return`已写入长期记忆（${"replace"===t?.mode?"整体替换":"追加"}）。`}return"长期记忆写入失败（磁盘不可写？），可忽略并继续任务。"}}}
|
package/src/plan-limits.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import{getPlatformConfig as e}from"./api.js";import{loadConfig as t,saveConfig as r}from"./config.js";export const PLAN_RESOURCES=Object.freeze({skills:Object.freeze({label:"技能",gate:""}),mcp_servers:Object.freeze({label:"MCP服务",gate:""}),scheduled_tasks:Object.freeze({label:"自动化任务",gate:"agent_tasks"}),agents:Object.freeze({label:"专家",gate:""}),squads:Object.freeze({label:"专家团",gate:""}),projects:Object.freeze({label:"项目",gate:"project"})});export const PLAN_FEATURES=Object.freeze({agent_tasks:"Agent自动化任务",knowledge_base:"知识库",project:"项目"});export function upgradeUrl(e){const t=String(e||"").trim().replace(/\/+$/,"");return t?`${t}/console/membership`:""}const n=["免费版","Plus会员","Pro会员","Max会员","团队版","企业版"],a=3e4;let i={key:"",at:0,plan:null,fetched:!1,failAt:0},l=null,
|
|
1
|
+
import{getPlatformConfig as e}from"./api.js";import{loadConfig as t,saveConfig as r}from"./config.js";export const PLAN_RESOURCES=Object.freeze({skills:Object.freeze({label:"技能",gate:""}),mcp_servers:Object.freeze({label:"MCP服务",gate:""}),scheduled_tasks:Object.freeze({label:"自动化任务",gate:"agent_tasks"}),agents:Object.freeze({label:"专家",gate:""}),squads:Object.freeze({label:"专家团",gate:""}),projects:Object.freeze({label:"项目",gate:"project"})});export const PLAN_FEATURES=Object.freeze({agent_tasks:"Agent自动化任务",knowledge_base:"知识库",project:"项目"});export function upgradeUrl(e){const t=String(e||"").trim().replace(/\/+$/,"");return t?`${t}/console/membership`:""}const n=["免费版","Plus会员","Pro会员","Max会员","团队版","企业版"],a=3e4;let i={key:"",at:0,plan:null,fetched:!1,failAt:0},l=null,u=null;const o=e=>`${String(e?.baseUrl||"")}\n${String(e?.apiKey||"")}`,s=e=>e&&"object"==typeof e&&!Array.isArray(e)?e:null;export function planFromUser(e){if(!e||"object"!=typeof e)return null;const t=s(e.limits),r=Array.isArray(e.entitlements)?e.entitlements.map(String):null;if(!t&&!r)return null;const n=e.tier_level??e.tierLevel,a=null==n||""===n?NaN:Number(n),i=Number(e.context_limit??e.contextLimit);return{code:String(e.plan_code??e.planCode??"").trim().toLowerCase(),name:String(e.plan_name??e.planName??"").trim(),tier:Number.isInteger(a)?a:null,limits:t,entitlements:r,featureNames:s(e.feature_names??e.featureNames),gates:s(e.limit_gates??e.limitGates),contextLimit:Number.isFinite(i)&&i>=0?Math.floor(i):null}}export function contextLimitOf(e=currentPlan()){const t=Number(e?.contextLimit);return Number.isFinite(t)&&t>0?t:0}export function currentPlan(e=t()){return i.fetched&&i.key===o(e)?i.plan:planFromUser(e?.user)}export async function refreshPlan(r=t(),{force:n=!1}={}){if(!r?.baseUrl||!r?.apiKey)return currentPlan(r);const s=o(r),c=Date.now();if(!n&&i.key===s){if(i.fetched&&c-i.at<a)return i.plan;if(i.failAt&&c-i.failAt<a)return currentPlan(r)}if(l&&l.key===s)return l.promise;const f=(async()=>{let t={};try{t=await e(r.baseUrl,r.apiKey,{timeoutMs:6e3})}catch{t={}}return t&&"object"==typeof t&&t.user&&"object"==typeof t.user?(i={key:s,at:Date.now(),plan:planFromUser(t.user),fetched:!0,failAt:0},u=null,i.plan):(i=i.key===s?{...i,failAt:Date.now()}:{key:s,at:0,plan:null,fetched:!1,failAt:Date.now()},currentPlan(r))})().finally(()=>{l&&l.promise===f&&(l=null)});return l={key:s,promise:f},f}export async function syncPlanUser(n=t()){if(!n?.baseUrl||!n?.apiKey)return currentPlan(n);let a={};try{a=await e(n.baseUrl,n.apiKey,{timeoutMs:6e3})}catch{a={}}const l=t();if(l.apiKey!==n.apiKey||l.baseUrl!==n.baseUrl)return currentPlan(l);const u=a&&"object"==typeof a.user?a.user:null;if(!u)return currentPlan(n);primePlan(n,u);try{JSON.stringify(n.user??null)!==JSON.stringify(u)&&(r({user:u}),n.user=u)}catch{}return i.plan}export function primePlan(e,t){e?.baseUrl&&e?.apiKey&&t&&"object"==typeof t&&(i={key:o(e),at:Date.now(),plan:planFromUser(t),fetched:!0,failAt:0},u=null)}export function noteContextLimit(e,t){const r=Math.floor(Number(t));r>0&&e&&(u={key:o(e),limit:r})}export function contextLimitHint(e){return u&&e&&u.key===o(e)?u.limit:0}export function resetPlanCache(){i={key:"",at:0,plan:null,fetched:!1,failAt:0},l=null,u=null}export async function planVerdict(e,r=t()){await refreshPlan(r);const n=i.fetched&&i.key===o(r)&&Date.now()-i.at<a?i.plan?.entitlements:null;return Array.isArray(n)?n.includes(e)?"granted":"denied":"unknown"}export function featureName(e,t=currentPlan()){const r=t?.featureNames?.[e];return"string"==typeof r&&r.trim()||PLAN_FEATURES[e]||e}export function featureMessage(e,t=currentPlan()){return`当前会员等级未开通「${featureName(e,t)}」，请升级会员后使用`}export function automationSuspendedMessage(e=currentPlan()){return`当前会员等级未开通「${featureName("agent_tasks",e)}」，任务已自动暂停；升级会员后可重新启用`}export function gateOf(e,t=currentPlan()){return t?.gates?String(t.gates[e]||""):PLAN_RESOURCES[e]?.gate||""}export function limitOf(e,t=currentPlan()){const r=t?.limits?.[e];if(null==r||""===r)return null;const n=Number(r);return Number.isFinite(n)?n<0?-1:Math.floor(n):null}export function remainingOf(e,t,r=currentPlan()){const n=limitOf(e,r);return null===n||n<0?1/0:Math.max(0,n-Math.max(0,Number(t)||0))}export function checkLimit(e,t,r=currentPlan()){const a=PLAN_RESOURCES[e],i=a?limitOf(e,r):null;if(null===i||i<0)return null;const l=Math.max(0,Number(t)||0);if(l<i)return null;const u=gateOf(e,r),o=i<=0&&!!u&&Array.isArray(r?.entitlements)&&!r.entitlements.includes(u);let s;return s=o?featureMessage(u,r):i<=0?`当前会员等级暂不支持创建${a.label}，请升级会员后使用`:`${function(e){return e?.name?["free","team","enterprise"].includes(e.code)?e.name:`${e.name}会员`:Number.isInteger(e?.tier)&&n[e.tier]?n[e.tier]:"当前会员等级"}(r)}最多可创建 ${i} 个${a.label}，如需更多请升级会员`,{code:o?"SUBSCRIPTION_REQUIRED":"RESOURCE_LIMIT_EXCEEDED",resource:e,limit:i,count:l,message:s}}export function checkFeature(e,t=currentPlan()){return!Array.isArray(t?.entitlements)||t.entitlements.includes(e)?null:{code:"SUBSCRIPTION_REQUIRED",feature:e,message:featureMessage(e,t)}}export class PlanLimitError extends Error{constructor(e){super(e.message),this.name="PlanLimitError",this.code=e.code,this.resource=e.resource||"",this.feature=e.feature||"",this.limit=e.limit??null,this.count=e.count??null}}export function isPlanLimitError(e){return!!e&&(e instanceof PlanLimitError||"PlanLimitError"===e.name)}export function assertLimit(e,t,r=currentPlan()){const n=checkLimit(e,t,r);if(n)throw new PlanLimitError(n)}export function assertFeature(e,t=currentPlan()){const r=checkFeature(e,t);if(r)throw new PlanLimitError(r)}export function planDeniedForModel(e){return`${String(e?.message||"").replace(/[。.]$/,"")}。这是当前账号会员等级的限制，本端无法绕过——请如实告诉用户，不要改用写文件、命令行等其它方式变相创建；需要更多额度请到平台控制台「会员订阅」升级。`}
|
package/src/privatedir.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import t from"node:fs";import r from"node:path";import n from"node:crypto";import{
|
|
1
|
+
import t from"node:fs";import r from"node:path";import n from"node:crypto";import{accountDataDir as e}from"./account-scope.js";export function bucketOf(t){const r=String(t||"anonymous");return n.createHash("sha256").update(r).digest("hex").slice(0,16)}export function safeSegment(t,r="item",n=40){const e=String(t||"").replace(/[^A-Za-z0-9_.-]+/g,"_").slice(0,n);return!e||/^\.+$/.test(e)?r:e}export function areaDir(t){return r.join(e(),t)}export function sessionDir(t,n){return r.join(areaDir(t),bucketOf(n))}export function createPrivateFile(e,o,c,i=""){try{const s=sessionDir(e,o);t.mkdirSync(s,{recursive:!0,mode:448});const a=Date.now().toString(36),u=n.randomBytes(4).toString("hex"),f=r.join(s,`${a}-${u}-${safeSegment(c)}.txt`),p=t.openSync(f,"wx",384);try{i&&t.writeFileSync(p,String(i),"utf8")}finally{t.closeSync(p)}return{path:f}}catch{return null}}export function dropSessionDir(r,n){try{return t.rmSync(sessionDir(r,n),{recursive:!0,force:!0}),!0}catch{return!1}}export function pruneArea(n,e){const o=Date.now(),c=areaDir(n);try{const n=t.readdirSync(c).slice(0,200);for(const i of n){const n=r.join(c,i);let s;try{s=t.readdirSync(n)}catch{continue}let a=s.length;for(const c of s){const i=r.join(n,c);try{o-t.statSync(i).mtimeMs>e&&(t.unlinkSync(i),a--)}catch{}}if(a<=0)try{t.rmdirSync(n)}catch{}}}catch{}}export function makeThrottle(t){let r=0;return()=>{const n=Date.now();return!(n-r<t)&&(r=n,!0)}}
|
package/src/repl.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import{terminalSelect as s}from"./terminal-select.js";import{questionSelection as o}from"./selection.js";import{parseAskAnswer as e}from"./ask-ui.js";import t from"node:readline";import{syncPlanUser as r}from"./plan-limits.js";import{c as n}from"./colors.js";import{createPermissions as i}from"./permissions.js";import{runTurn as a}from"./agent.js";import{createSession as c,saveSession as l,turnCount as u}from"./session.js";import{syncSessionLog as p,rewindSessionLog as m}from"./sessionlog.js";import{ensureSessionTitle as d}from"./title.js";import{createCheckpoints as f}from"./checkpoints.js";import{buildSystemPrompt as g}from"./context.js";import{runVerifyLoop as w}from"./autofix.js";import{runAutoFlow as h}from"./autorun.js";import{loadMcpTools as j,closeMcp as y,makeMcpConnectTools as b}from"./mcp.js";import{makeJobTools as k,onSettled as x,describe as I}from"./jobs.js";import{makeTerminalTools as T,closeAllTerminals as S,dropSessionTerminals as v}from"./terminal.js";import{makeGoalTools as $,runGoalRounds as q,setRoundKind as O,getGoal as P,viewGoal as U}from"./goalround.js";import{makeCodeModeTool as C}from"./codemode.js";import{makeWorkflowTool as _}from"./workflow.js";import{makeLspTool as M,setProvider as A}from"./lsp.js";import{ensureVerified as E}from"./sandbox.js";import{makeAutomationTools as R}from"./automation-tools.js";import{makeNotifyTool as H,notifyTurnDone as L}from"./notify.js";import{makePresenceTools as N,acquireWakeLock as W,releaseWakeLock as B}from"./presence.js";import{makeComputerTools as G}from"./computer.js";import{makeDiagramTools as V}from"./diagram.js";import{makePptxTools as D}from"./pptx-tools.js";import{makeDocxTools as Y}from"./docx-tools.js";import{makeReportTools as J}from"./report-tools.js";import{makeXlsxTools as K}from"./xlsx-tools.js";import{makeCodeTools as Q}from"./code-tools.js";import{makeTestTools as z}from"./test-runner.js";import{makeEngineeringTools as F}from"./engineering.js";import{makeWebTools as X}from"./web-inspect.js";import{browserAvailable as Z}from"./web-cdp.js";import{makeUiTools as ss}from"./ui-spec.js";import{makeAskUiTool as os}from"./ask-ui.js";import{makeSurfaceTools as es}from"./surface-state.js";import{makeVoiceFormTool as ts}from"./voice-form.js";import{makeDepAuditTool as rs}from"./depaudit.js";import{makeScopeCheckTool as ns}from"./scopecreep.js";import{makeArchaeologyTool as is}from"./archaeology.js";import{streamChat as as}from"./api.js";import{execSync as cs}from"node:child_process";import{createVisualBridge as ls}from"./visual.js";import{autoSyncPlatformSkills as us}from"./skills.js";import{makeSubagentTool as ps,makeDispatchAgentsTool as ms,makeContinuableAgentTools as ds}from"./subagent.js";import{makeDispatchMemberTool as fs}from"./squad.js";import{makeConsultTool as gs}from"./consult.js";import{makeExpertTools as ws}from"./experts.js";import{listWorkspaceFiles as hs}from"./executors.js";import{makeAdvisorTool as js}from"./advisor.js";import{makeRecallTool as ys,readMemoryRaw as bs}from"./memory.js";import{detectScanners as ks,makeScanTool as xs}from"./scan.js";import{runSlash as Is,listCommands as Ts}from"./commands.js";import{handleAccountError as Ss,accountErrorKind as vs}from"./account.js";import{expandAtMentions as $s,buildUserContent as qs,runBangCommand as Os,loadHistory as Ps,appendHistory as Us}from"./input.js";import{makeSkillTools as Cs}from"./skills.js";import{makeSessionEventTools as _s}from"./session-events.js";import{makeExternalAgentTools as Ms}from"./external-agents.js";import{makeWebSearchTools as As}from"./websearch.js";import{renderWelcome as Es}from"./banner.js";import{createCrewBudget as Rs,withBudget as Hs,makePlanBudgetTool as Ls,makeBudgetStatusTool as Ns}from"./crew.js";import{makeEvidenceTools as Ws}from"./evidence.js";import{makeTapdTools as Bs}from"./tapd.js";import{makeWeeklyTools as Gs}from"./weekly.js";export async function buildExtraTools(s,o,e={}){const t="function"==typeof e.onSub?e.onSub:null;let r={tools:[],clients:[],registry:{servers:{}}};try{r=await j(o)}catch(s){process.stdout.write(n.yellow("MCP初始化失败（忽略）："+s.message)+"\n")}const i=[...Cs(o),...r.tools],a=s.crewBudget?Rs(s.crewBudgetPlan||{}):null,c=(s,o,e)=>a?Hs(s,a,o,e):s,l=ps({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t});l&&i.push(c(l,"dispatch"));const u=ms({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t});u&&i.push(c(u,"dispatch",s=>(s?.tasks||[]).length||1)),i.push(...ds({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t,queueInject:"function"==typeof e.queueInject?e.queueInject:null}));const p=fs({config:s,cwd:o,extraTools:r.tools,onSub:t});p&&i.push(p),i.push(...As({config:s})),i.push(...Ms({config:s,cwd:o,onSub:t}));const m=gs({config:s});m&&i.push(m);try{i.push(...await ws({config:s}))}catch{}const d=js({config:s});if(d?i.push(c(d,"advisor")):a&&a.degrade("顾问","未配置advisorModel"),a&&(i.push(Ls(a,e.onBudgetDeclared||null)),i.push(Ns(a))),!1!==s.memoryEnabled&&bs(o)&&i.push(ys({cwd:o})),ks(o).length&&i.push(xs({cwd:o})),i.push(...k({sessionId:()=>s.sessionId||""})),"function"==typeof e.queueInject){x(o=>{if(o.sessionId!==String(s.sessionId||""))return;const t="agent"===o.kind?`它已转为空闲，可用send_message继续对话；想看这一段的细节用job_output("${o.id}")。`:`用job_output("${o.id}") 读它的输出（默认只回新增）。`;e.queueInject(`【后台作业完成】${I(o)}\n${t}`)})}i.push(_({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t})),i.push(C()),"function"==typeof e.sessionOf&&i.push(...$({sessionOf:e.sessionOf})),e.lspProvider&&A(e.lspProvider),i.push(M({cwd:o})),i.push(...await T({sessionId:()=>s.sessionId||"",cwd:o,config:s})),i.push(...b({registry:r.registry,clients:r.clients,cwd:o,register:s=>i.push(...s)})),i.push(...R({cwd:o,selfSchedule:!1!==e.selfSchedule,runnerHint:e.runnerHint})),i.push(...V({cwd:o})),i.push(...D({cwd:o,config:s})),i.push(...Y({cwd:o,config:s})),i.push(...J({cwd:o,config:s})),i.push(...K({cwd:o,config:s})),i.push(...Q({cwd:o,config:s})),i.push(...z({cwd:o,config:s})),i.push(...F()),i.push(...Ws({cwd:o})),i.push(...Bs({cwd:o})),i.push(...Gs({cwd:o})),i.push(...ss({cwd:o})),i.push(os({ask:e.ask,askUi:e.askUi})),e.sessionId&&i.push(...es({sessionId:e.sessionId,emit:e.onSurface})),i.push(...rs({cwd:o})),i.push(..._s({cwd:o,config:s})),i.push(...is({cwd:o})),i.push(...ns({cwd:o,runGit:(s,o)=>{try{return cs(`git ${o.join(" ")}`,{cwd:s,encoding:"utf8",stdio:["ignore","pipe","ignore"],maxBuffer:33554432})}catch{return null}},goalOf:()=>{try{const s="function"==typeof e.sessionOf?P(e.sessionOf()):null;return s&&"active"===s.phase?s.objective:""}catch{return""}}})),i.push(...ts(async(o,e)=>{let t="";try{await as({baseUrl:s.baseUrl,apiKey:s.apiKey,model:s.model,messages:[{role:"system",content:o},{role:"user",content:e}],temperature:0,client:s.client,extraBody:{enable_thinking:!1},aux:!0},{onContent:s=>{t+=s}})}catch{return""}return t}));const f=H({enabled:!1!==s.notifications});f&&i.push(f),i.push(...N({signalRef:e.signalRef}));const g="function"==typeof e.injectVisual?e.injectVisual:"function"==typeof e.queueInject?ls({config:s,queue:e.queueInject,currentModel:e.currentModel}):null;return i.push(...G({config:s,injectVisual:g,cwd:o,omit:e.omitComputer||[],enabled:e.computerUse})),Z()&&i.push(...X({cwd:o,config:s,injectVisual:g})),us(s,{onInfo:o=>{s.debug&&process.stdout.write(n.gray(" · "+o)+"\n")}}).catch(()=>{}),{extraTools:i,clients:r.clients,mcpRegistry:r.registry,crew:a}}export async function startRepl({config:u,cwd:j,yolo:b,initialPrompt:k,session:x,version:I,...T}){E().catch(()=>{}),r(u).catch(()=>{});const $=t.createInterface({input:process.stdin,output:process.stdout,history:Ps(j).reverse(),historySize:200,completer:s=>{if(/^\/\S*$/.test(s)){const o=Ts(j).flatMap(s=>[s.name,...s.aliases||[]]);return[o.map(s=>"/"+s).filter(o=>o.startsWith(s)),s]}const o=/(?:^|\s)@([^\s@]*)$/.exec(s);if(!o)return[[],s];const e="@"+o[1],t=e.toLowerCase();return[hs(j).map(s=>"@"+s).filter(s=>s.toLowerCase().startsWith(t)).slice(0,40),e]}});let U,C=null,_=!1,M=null,A=!1;T.updater?.check.then(s=>{C=s,s&&A&&!$.line&&U?.()});let R=!1;const H=s=>new Promise((o,e)=>{if($.closed)return e(new Error("终端已关闭"));const t=new AbortController,r=()=>{U=null,e(new Error("终端已关闭"))},n=s=>{$.off("close",r),U=null,o(s)};U=()=>{t.abort(),n("")},$.once("close",r),$.question(s,{signal:t.signal},n)});let N=Promise.resolve();const G=process.stdin.isTTY&&process.stdout.isTTY?o=>{const e=z?.signal,t=N.then(async()=>{R=!0;try{return await s(o,{rl:$,signal:e,onInterrupt:()=>z?.abort()})}finally{R=!1}});return N=t.catch(()=>null),t}:void 0,V=T.askUi||(G?async s=>{const t=await G(o(s));return"date_range"===s.kind&&t?.value?e(s,t.value):t||{value:null}}:void 0);let D=!1,Y=i({yolo:b,ask:H,select:G,cwd:j});const J={prompt:0,completion:0},K={list:[]},Q=f();let z=null,F=null;const X={config:u,cwd:j,permissions:Y,usage:J,session:x,version:I,crewOf:()=>ts};$.on("SIGINT",()=>{R?(z?.abort(),U?.(),process.stdout.write("\n")):z?z.abort():process.stdout.write("\n"+n.gray("输入 /exit退出。")+"\n")}),function(s,o,e,t){process.stdout.write(Es({model:s.model,cwd:o,version:t,platform:/^(agent|workflow|datahuman|squad)-/.test(String(s.model||"")),yolo:e.isYolo(),consult:s.consult?s.consultLabel||s.consult:null})+"\n")}(u,j,Y,I),Ds(x);const Z=[],ss={current:null};let{extraTools:os,clients:es,crew:ts}=await buildExtraTools(u,j,{sessionOf:()=>X.session,queueInject:s=>Z.push(s),signalRef:ss,ask:H,askUi:V,sessionId:X.session?.id,onSurface:T.onSurface,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"});const rs=()=>{d(X.session,u).then(s=>{s&&l(X.session)}).catch(()=>{})},ns=async s=>{const{text:o,mentions:e,images:t}=await $s(s,j,u,{onDocStart:(s,o,e)=>process.stdout.write(n.gray(` · 正在解析文档 @${s}${e>1?`（${o}/${e}）`:""}…`)+"\n")});e.length&&process.stdout.write(n.gray(" · 已展开引用："+e.join("、")+(t.length?`（含 ${t.length} 张图片）`:""))+"\n");const r=Q.snapshot({session:X.session,label:s});X.session.messages.push({role:"user",content:qs(o,t)}),rs(),z=new AbortController,ss.current=z.signal,X.session.compaction||(X.session.compaction={}),u.sessionId=X.session.id,u.goal=P(X.session)?.objective||"";const i="auto"===Y.getMode?.()&&!1!==u.wakeLockInAuto?W({reason:"自动模式任务执行中"}):null,c=Date.now();let m;try{m=await a({config:u,messages:X.session.messages,permissions:Y,cwd:j,signal:z.signal,extraTools:os,todoState:K,compaction:X.session.compaction,fileJournal:r,pullQueued:()=>Z.shift()||null})}finally{i?.ok&&B()}return m?.usage&&(J.prompt+=m.usage.prompt,J.completion+=m.usage.completion),m?.lastPromptTokens&&(X.session.lastPromptTokens=m.lastPromptTokens),m?.lastUsage&&(X.session.lastUsage=m.lastUsage),z=null,ss.current=null,m?.aborted||m?.error||L({durationMs:Date.now()-c,cwd:j,summary:String(s).replace(/\s+/g," ").slice(0,80),enabled:!1!==u.notifications}).catch(()=>{}),p(X.session),l(X.session),rs(),m?.error&&"auth"===Ss(m,u)&&(D=!0),m},is=async s=>{O(X.session.id,"human");const o=await ns(s);return o?.aborted||o?.error||D?o:P(X.session)?(await q({session:X.session,doTurn:ns,stats:()=>({usedTokens:J.prompt+J.completion,budgetTokens:Number(u.budgetTokens)||0}),onInfo:s=>process.stdout.write(n.gray(" ◆ "+s)+"\n"),aborted:()=>D}),l(X.session),o):o},as=()=>({...X,signal:z?.signal,select:G,resumeSession:s=>{F&&(clearInterval(F),F=null),v(X.session.id),X.session=s,u.sessionId=s.id,s.model&&(u.model=s.model);const o=Y.getMode();Y=i({yolo:!1,ask:H,select:G,cwd:j}),Y.setMode(o),X.permissions=Y,Z.length=0,K.list=[],Q.list.length=0,Ds(s)},session:X.session,extraTools:os,checkpoints:Q,onRewind:s=>m(X.session,s),print:s=>process.stdout.write(String(s).endsWith("\n")?String(s):String(s)+"\n")});for(k&&(process.stdout.write(n.cyan("\nhuicai> ")+k+"\n"),await is(k));!D;){if(C&&!_&&G){_=!0;if("update"===await G(T.updater.menu(C))){process.stdout.write("正在更新 HuiCai AI，请稍候…\n"),z=new AbortController;try{M=await T.updater.install(C,{signal:z.signal});break}catch(s){process.stdout.write(s.message+" 输入 /update 可重试。\n")}finally{z=null}}}let s;try{A=!0,s=(await H(n.cyan("\nhuicai> "))).trim()}catch{break}finally{A=!1}if(""!==s){if(Us(j,s),"/exit"===s||"/quit"===s)break;if("/update"===s&&C&&T.updater)_=!1;else{if(s.startsWith("!")&&s.length>1){const o=s.slice(1).trim(),{output:e,record:t}=await Os(o,j);process.stdout.write(n.gray(e.split("\n").map(s=>" "+s).join("\n"))+"\n"),X.session.messages.push({role:"user",content:t}),l(X.session);continue}if(s.startsWith("/")){let o;z=new AbortController;try{o=await Is(s,as())}catch(s){process.stdout.write((z.signal.aborted?"操作已取消":"命令失败："+s.message)+"\n");continue}finally{z=null}if($.closed)break;if("exit"===o.action)break;if("clear"===o.action){F&&(clearInterval(F),F=null);const s=Y.getMode();Y=i({yolo:!1,ask:H,select:G,cwd:j}),Y.setMode(s),X.permissions=Y,l(X.session),v(X.session.id),X.session=c({cwd:j,model:u.model,messages:[X.session.messages[0]]}),u.sessionId=X.session.id,Z.length=0,K.list=[],Q.list.length=0,process.stdout.write(n.gray("已开启新会话（旧会话已保存，可用huicai --resume恢复）。\n"))}"refresh"===o.action&&Ds(X.session),"cd"===o.action&&(j=o.cwd,X.cwd=j,Y=i({yolo:Y.isYolo(),ask:H,select:G,cwd:j}),X.permissions=Y,y(es),({extraTools:os,clients:es,crew:ts}=await buildExtraTools(u,j,{ask:H,askUi:V,sessionOf:()=>X.session,queueInject:s=>Z.push(s),signalRef:ss,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"})),X.session=c({cwd:j,model:u.model,messages:[{role:"system",content:g(j,{memory:!1!==u.memoryEnabled,config:u})}]}),K.list=[],Q.list.length=0,u.sessionId=X.session.id,process.stdout.write(n.gray("已切换到 "+j+"（新会话，旧会话已保存）。\n"))),"autofix"===o.action&&await w({command:o.command,maxIters:o.maxIters,cwd:j,runTurnOnce:s=>ns(s),emit:s=>process.stdout.write(n.gray(s)+"\n")}),"autorun"===o.action&&await h({goal:o.goal,config:u,permissions:Y,cwd:j,runTurnOnce:s=>ns(s),emit:s=>process.stdout.write(n.gray(" · "+s)+"\n")}),"loop"===o.action&&(F&&(clearInterval(F),F=null),o.stop?process.stdout.write(n.gray("已停止循环。\n")):(process.stdout.write(n.gray(`已开始循环：每 ${o.seconds}s运行「${o.prompt}」（/loop off停止）。\n`)),F=setInterval(()=>{z||ns(o.prompt)},1e3*o.seconds))),o.prompt&&await is(o.prompt);continue}await is(s)}}}return F&&clearInterval(F),$.close(),y(es),S(),T.updater?.dispose(),M?{updated:M}:D?{relogin:!0}:(process.stdout.write(n.gray("\n再见。\n")),{})}function Vs(s){return"string"==typeof s?s:Array.isArray(s)?s.map(s=>"string"==typeof s?s:s&&"text"===s.type&&s.text||"").join(""):""}function Ds(s){const o=(s.messages||[]).filter(s=>"user"===s.role||"assistant"===s.role),e=u(s.messages);if(0===e)return;process.stdout.write(n.gray(` 已恢复会话「${s.title||""}」（${e} 轮）——历史对话如下：`)+"\n");const t=o.length>40?o.length-40:0;t&&process.stdout.write(n.gray(` （较早 ${t} 条已省略，完整记录在 ~/.huicai/sessions）`)+"\n");for(const s of o.slice(-40)){const o=Vs(s.content).trim();if(!o)continue;const e=o.length>800?o.slice(0,800)+"…":o,t="user"===s.role?n.bold(" 你 › "):n.gray(" 助手 › ");process.stdout.write(t+e.replace(/\n/g,"\n ")+"\n")}process.stdout.write(n.gray(" ——以上为历史，可继续对话——")+"\n")}export async function runOnce({config:s,cwd:o,yolo:e,prompt:t,session:r,persist:c,outputFormat:m,maxTurns:d}){t||(process.stderr.write('用法: huicai -p "你的指令"\n'),process.exit(1));const f="json"===m,g=i({yolo:e,ask:null,cwd:o}),w=[],h={current:null},j=new AbortController;h.current=j.signal;let b=!1;const k=()=>{b&&process.exit(130),b=!0,process.stderr.write(n.gray("\n已请求中断（再按一次Ctrl-C强制退出）…\n")),j.abort()};process.on("SIGINT",k);const{extraTools:x,clients:I}=await buildExtraTools(s,o,{sessionOf:()=>r,queueInject:s=>w.push(s),signalRef:h,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"});let T=t;if(t.trim().startsWith("/")){const{runSlash:e}=await import("./commands.js"),n=[],i=await e(t.trim(),{cwd:o,config:s,session:r,permissions:g,print:s=>n.push(String(s))}).catch(s=>({handled:!1,error:s}));if(n.length&&process.stdout.write(n.join("\n")+"\n"),i&&i.prompt)T=i.prompt;else if(i&&i.handled)return y(I),process.off("SIGINT",k),{text:n.join("\n")}}const{text:v,images:$}=await $s(T,o,s);r.messages.push({role:"user",content:qs(v,$)}),r.compaction||(r.compaction={}),s.sessionId=r.id;const q=e&&!1!==s.wakeLockInAuto?W({reason:"无人值守任务执行中"}):null;let O;try{O=await a({config:s,messages:r.messages,permissions:g,cwd:o,extraTools:x,signal:j.signal,maxIters:d,compaction:r.compaction,io:f?{write:()=>{}}:void 0,pullQueued:()=>w.shift()||null})}finally{q?.ok&&B()}O?.lastPromptTokens&&(r.lastPromptTokens=O.lastPromptTokens),O?.lastUsage&&(r.lastUsage=O.lastUsage),c&&(l(r),p(r)),process.removeListener("SIGINT",k),y(I),S();const P=O?.error?vs(O):"";if(P&&(Ss(O,s,s=>process.stderr.write(s)),"auth"===P&&process.stderr.write(n.gray("请重新运行 `huicai login` 登录后重试。")+"\n")),f){const o=[...r.messages].reverse().find(s=>"assistant"===s.role&&s.content);process.stdout.write(JSON.stringify({type:"result",is_error:!!O?.error,...O?.error?{error:O.error}:{},...P?{error_kind:P}:{},result:o?String(o.content):"",usage:{input_tokens:O?.usage?.prompt??0,output_tokens:O?.usage?.completion??0},session_id:r.id,model:s.model,num_turns:u(r.messages)})+"\n")}O?.error&&process.exit(1)}
|
|
1
|
+
import{terminalSelect as s}from"./terminal-select.js";import{questionSelection as o}from"./selection.js";import{parseAskAnswer as e}from"./ask-ui.js";import t from"node:readline";import{syncPlanUser as r}from"./plan-limits.js";import{c as n}from"./colors.js";import{createPermissions as i}from"./permissions.js";import{runTurn as a}from"./agent.js";import{createSession as c,saveSession as l,turnCount as u}from"./session.js";import{syncSessionLog as p,rewindSessionLog as m}from"./sessionlog.js";import{ensureSessionTitle as d}from"./title.js";import{createCheckpoints as f}from"./checkpoints.js";import{buildSystemPrompt as g}from"./context.js";import{runVerifyLoop as w}from"./autofix.js";import{runAutoFlow as h}from"./autorun.js";import{loadMcpTools as j,closeMcp as y,makeMcpConnectTools as b}from"./mcp.js";import{makeJobTools as k,onSettled as x,describe as I}from"./jobs.js";import{makeTerminalTools as T,closeAllTerminals as v,dropSessionTerminals as S}from"./terminal.js";import{makeGoalTools as $,runGoalRounds as q,setRoundKind as O,getGoal as C,viewGoal as P}from"./goalround.js";import{makeCodeModeTool as U}from"./codemode.js";import{makeWorkflowTool as E}from"./workflow.js";import{makeLspTool as _,setProvider as M}from"./lsp.js";import{ensureVerified as A}from"./sandbox.js";import{makeAutomationTools as R}from"./automation-tools.js";import{makeNotifyTool as L,notifyTurnDone as H}from"./notify.js";import{makePresenceTools as W,acquireWakeLock as B,releaseWakeLock as N}from"./presence.js";import{makeComputerTools as V}from"./computer.js";import{makeDiagramTools as D}from"./diagram.js";import{makePptxTools as G}from"./pptx-tools.js";import{makeDocxTools as Y}from"./docx-tools.js";import{makeReportTools as J}from"./report-tools.js";import{makeXlsxTools as K}from"./xlsx-tools.js";import{makeCodeTools as Q}from"./code-tools.js";import{makeTestTools as z}from"./test-runner.js";import{makeEngineeringTools as F}from"./engineering.js";import{makeWebTools as X}from"./web-inspect.js";import{browserAvailable as Z}from"./web-cdp.js";import{makeUiTools as ss}from"./ui-spec.js";import{makeAskUiTool as os}from"./ask-ui.js";import{makeSurfaceTools as es}from"./surface-state.js";import{makeVoiceFormTool as ts}from"./voice-form.js";import{makeDepAuditTool as rs}from"./depaudit.js";import{makeScopeCheckTool as ns}from"./scopecreep.js";import{makeArchaeologyTool as is}from"./archaeology.js";import{streamChat as as}from"./api.js";import{execSync as cs}from"node:child_process";import{createVisualBridge as ls}from"./visual.js";import{autoSyncPlatformSkills as us}from"./skills.js";import{makeSubagentTool as ps,makeDispatchAgentsTool as ms,makeContinuableAgentTools as ds}from"./subagent.js";import{makeDispatchMemberTool as fs}from"./squad.js";import{makeConsultTool as gs}from"./consult.js";import{makeExpertTools as ws}from"./experts.js";import{listWorkspaceFiles as hs}from"./executors.js";import{makeAdvisorTool as js}from"./advisor.js";import{makeRecallTool as ys,readMemoryRaw as bs}from"./memory.js";import{detectScanners as ks,makeScanTool as xs}from"./scan.js";import{runSlash as Is,listCommands as Ts}from"./commands.js";import{handleAccountError as vs,accountErrorKind as Ss}from"./account.js";import{expandAtMentions as $s,buildUserContent as qs,runBangCommand as Os,loadHistory as Cs,appendHistory as Ps}from"./input.js";import{makeSkillTools as Us}from"./skills.js";import{makeSessionEventTools as Es}from"./session-events.js";import{makeExternalAgentTools as _s}from"./external-agents.js";import{makeWebSearchTools as Ms}from"./websearch.js";import{renderWelcome as As}from"./banner.js";import{createCrewBudget as Rs,withBudget as Ls,makePlanBudgetTool as Hs,makeBudgetStatusTool as Ws}from"./crew.js";import{makeEvidenceTools as Bs}from"./evidence.js";import{makeTapdTools as Ns}from"./tapd.js";import{makeWeeklyTools as Vs}from"./weekly.js";export async function buildExtraTools(s,o,e={}){const t="function"==typeof e.onSub?e.onSub:null;let r={tools:[],clients:[],registry:{servers:{}}};try{r=await j(o)}catch(s){process.stdout.write(n.yellow("MCP初始化失败（忽略）："+s.message)+"\n")}const i=[...Us(o),...r.tools],a=s.crewBudget?Rs(s.crewBudgetPlan||{}):null,c=(s,o,e)=>a?Ls(s,a,o,e):s,l=ps({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t});l&&i.push(c(l,"dispatch"));const u=ms({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t});u&&i.push(c(u,"dispatch",s=>(s?.tasks||[]).length||1)),i.push(...ds({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t,queueInject:"function"==typeof e.queueInject?e.queueInject:null}));const p=fs({config:s,cwd:o,extraTools:r.tools,onSub:t});p&&i.push(p),i.push(...Ms({config:s})),i.push(..._s({config:s,cwd:o,onSub:t}));const m=gs({config:s});m&&i.push(m);try{i.push(...await ws({config:s}))}catch{}const d=js({config:s});if(d?i.push(c(d,"advisor")):a&&a.degrade("顾问","未配置advisorModel"),a&&(i.push(Hs(a,e.onBudgetDeclared||null)),i.push(Ws(a))),!1!==s.memoryEnabled&&bs(o)&&i.push(ys({cwd:o})),ks(o).length&&i.push(xs({cwd:o})),i.push(...k({sessionId:()=>s.sessionId||""})),"function"==typeof e.queueInject){x(o=>{if(o.sessionId!==String(s.sessionId||""))return;const t="agent"===o.kind?`它已转为空闲，可用send_message继续对话；想看这一段的细节用job_output("${o.id}")。`:`用job_output("${o.id}") 读它的输出（默认只回新增）。`;e.queueInject(`【后台作业完成】${I(o)}\n${t}`)})}i.push(E({config:s,cwd:o,extraTools:r.tools,depth:0,onSub:t})),i.push(U()),"function"==typeof e.sessionOf&&i.push(...$({sessionOf:e.sessionOf})),e.lspProvider&&M(e.lspProvider),i.push(_({cwd:o})),i.push(...await T({sessionId:()=>s.sessionId||"",cwd:o,config:s})),i.push(...b({registry:r.registry,clients:r.clients,cwd:o,register:s=>i.push(...s)})),i.push(...R({cwd:o,selfSchedule:!1!==e.selfSchedule,runnerHint:e.runnerHint})),i.push(...D({cwd:o})),i.push(...G({cwd:o,config:s})),i.push(...Y({cwd:o,config:s})),i.push(...J({cwd:o,config:s})),i.push(...K({cwd:o,config:s})),i.push(...Q({cwd:o,config:s})),i.push(...z({cwd:o,config:s})),i.push(...F()),i.push(...Bs({cwd:o})),i.push(...Ns({cwd:o})),i.push(...Vs({cwd:o})),i.push(...ss({cwd:o})),i.push(os({ask:e.ask,askUi:e.askUi})),e.sessionId&&i.push(...es({sessionId:e.sessionId,emit:e.onSurface})),i.push(...rs({cwd:o})),i.push(...Es({cwd:o,config:s})),i.push(...is({cwd:o})),i.push(...ns({cwd:o,runGit:(s,o)=>{try{return cs(`git ${o.join(" ")}`,{cwd:s,encoding:"utf8",stdio:["ignore","pipe","ignore"],maxBuffer:33554432})}catch{return null}},goalOf:()=>{try{const s="function"==typeof e.sessionOf?C(e.sessionOf()):null;return s&&"active"===s.phase?s.objective:""}catch{return""}}})),i.push(...ts(async(o,e)=>{let t="";try{await as({baseUrl:s.baseUrl,apiKey:s.apiKey,model:s.model,messages:[{role:"system",content:o},{role:"user",content:e}],temperature:0,client:s.client,extraBody:{enable_thinking:!1},aux:!0},{onContent:s=>{t+=s}})}catch{return""}return t}));const f=L({enabled:!1!==s.notifications});f&&i.push(f),i.push(...W({signalRef:e.signalRef}));const g="function"==typeof e.injectVisual?e.injectVisual:"function"==typeof e.queueInject?ls({config:s,queue:e.queueInject,currentModel:e.currentModel}):null;return i.push(...V({config:s,injectVisual:g,cwd:o,omit:e.omitComputer||[],enabled:e.computerUse})),Z()&&i.push(...X({cwd:o,config:s,injectVisual:g})),us(s,{onInfo:o=>{s.debug&&process.stdout.write(n.gray(" · "+o)+"\n")}}).catch(()=>{}),{extraTools:i,clients:r.clients,mcpRegistry:r.registry,crew:a}}export async function startRepl({config:u,cwd:j,yolo:b,initialPrompt:k,session:x,version:I,...T}){A().catch(()=>{}),r(u).catch(()=>{});const $=t.createInterface({input:process.stdin,output:process.stdout,history:Cs(j).reverse(),historySize:200,completer:s=>{if(/^\/\S*$/.test(s)){const o=Ts(j).flatMap(s=>[s.name,...s.aliases||[]]);return[o.map(s=>"/"+s).filter(o=>o.startsWith(s)),s]}const o=/(?:^|\s)@([^\s@]*)$/.exec(s);if(!o)return[[],s];const e="@"+o[1],t=e.toLowerCase();return[hs(j).map(s=>"@"+s).filter(s=>s.toLowerCase().startsWith(t)).slice(0,40),e]}});let P,U=null,E=!1,_=null,M=!1;T.updater?.check.then(s=>{U=s,s&&M&&!$.line&&P?.()});let R=!1;const L=s=>new Promise((o,e)=>{if($.closed)return e(new Error("终端已关闭"));const t=new AbortController,r=()=>{P=null,e(new Error("终端已关闭"))},n=s=>{$.off("close",r),P=null,o(s)};P=()=>{t.abort(),n("")},$.once("close",r),$.question(s,{signal:t.signal},n)});let W=Promise.resolve();const V=process.stdin.isTTY&&process.stdout.isTTY?o=>{const e=z?.signal,t=W.then(async()=>{R=!0;try{return await s(o,{rl:$,signal:e,onInterrupt:()=>z?.abort()})}finally{R=!1}});return W=t.catch(()=>null),t}:void 0,D=T.askUi||(V?async s=>{const t=await V(o(s));return"date_range"===s.kind&&t?.value?e(s,t.value):t||{value:null}}:void 0);let G=!1,Y=i({yolo:b,ask:L,select:V,cwd:j});const J={prompt:0,completion:0},K={list:[]},Q=f();let z=null,F=null;const X={config:u,cwd:j,permissions:Y,usage:J,session:x,version:I,crewOf:()=>ts};$.on("SIGINT",()=>{R?(z?.abort(),P?.(),process.stdout.write("\n")):z?z.abort():process.stdout.write("\n"+n.gray("输入 /exit退出。")+"\n")}),function(s,o,e,t){process.stdout.write(As({model:s.model,cwd:o,version:t,platform:/^(agent|workflow|datahuman|squad)-/.test(String(s.model||"")),yolo:e.isYolo(),consult:s.consult?s.consultLabel||s.consult:null})+"\n")}(u,j,Y,I),Gs(x);const Z=[],ss={current:null};let{extraTools:os,clients:es,crew:ts}=await buildExtraTools(u,j,{sessionOf:()=>X.session,queueInject:s=>Z.push(s),signalRef:ss,ask:L,askUi:D,sessionId:X.session?.id,onSurface:T.onSurface,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"});const rs=()=>{d(X.session,u).then(s=>{s&&l(X.session)}).catch(()=>{})},ns=async s=>{const{text:o,mentions:e,images:t}=await $s(s,j,u,{onDocStart:(s,o,e)=>process.stdout.write(n.gray(` · 正在解析文档 @${s}${e>1?`（${o}/${e}）`:""}…`)+"\n")});e.length&&process.stdout.write(n.gray(" · 已展开引用："+e.join("、")+(t.length?`（含 ${t.length} 张图片）`:""))+"\n");const r=Q.snapshot({session:X.session,label:s});X.session.messages.push({role:"user",content:qs(o,t)}),rs(),z=new AbortController,ss.current=z.signal,X.session.compaction||(X.session.compaction={}),u.sessionId=X.session.id,u.goal=C(X.session)?.objective||"";const i="auto"===Y.getMode?.()&&!1!==u.wakeLockInAuto?B({reason:"自动模式任务执行中"}):null,c=Date.now();let m;try{m=await a({config:u,messages:X.session.messages,permissions:Y,cwd:j,signal:z.signal,extraTools:os,todoState:K,compaction:X.session.compaction,fileJournal:r,pullQueued:()=>Z.shift()||null})}finally{i?.ok&&N()}return m?.usage&&(J.prompt+=m.usage.prompt,J.completion+=m.usage.completion),m?.lastPromptTokens&&(X.session.lastPromptTokens=m.lastPromptTokens),m?.lastUsage&&(X.session.lastUsage=m.lastUsage),z=null,ss.current=null,m?.aborted||m?.error||H({durationMs:Date.now()-c,cwd:j,summary:String(s).replace(/\s+/g," ").slice(0,80),enabled:!1!==u.notifications}).catch(()=>{}),p(X.session),l(X.session),rs(),m?.error&&"auth"===vs(m,u)&&(G=!0),m},is=async s=>{O(X.session.id,"human");const o=await ns(s);return o?.aborted||o?.error||G?o:C(X.session)?(await q({session:X.session,doTurn:ns,stats:()=>({usedTokens:J.prompt+J.completion,budgetTokens:Number(u.budgetTokens)||0}),onInfo:s=>process.stdout.write(n.gray(" ◆ "+s)+"\n"),aborted:()=>G}),l(X.session),o):o},as=()=>({...X,signal:z?.signal,select:V,resumeSession:s=>{F&&(clearInterval(F),F=null),S(X.session.id),X.session=s,u.sessionId=s.id,s.model&&(u.model=s.model);const o=Y.getMode();Y=i({yolo:!1,ask:L,select:V,cwd:j}),Y.setMode(o),X.permissions=Y,Z.length=0,K.list=[],Q.list.length=0,Gs(s)},session:X.session,extraTools:os,checkpoints:Q,onRewind:s=>m(X.session,s),print:s=>process.stdout.write(String(s).endsWith("\n")?String(s):String(s)+"\n")});for(k&&(process.stdout.write(n.cyan("\nhuicai> ")+k+"\n"),await is(k));!G;){if(U&&!E&&V){E=!0;if("update"===await V(T.updater.menu(U))){process.stdout.write("正在更新 HuiCai AI，请稍候…\n"),z=new AbortController;try{_=await T.updater.install(U,{signal:z.signal});break}catch(s){process.stdout.write(s.message+" 输入 /update 可重试。\n")}finally{z=null}}}let s;try{M=!0,s=(await L(n.cyan("\nhuicai> "))).trim()}catch{break}finally{M=!1}if(""!==s){if(Ps(j,s),"/exit"===s||"/quit"===s)break;if("/update"===s&&U&&T.updater)E=!1;else{if(s.startsWith("!")&&s.length>1){const o=s.slice(1).trim(),{output:e,record:t}=await Os(o,j);process.stdout.write(n.gray(e.split("\n").map(s=>" "+s).join("\n"))+"\n"),X.session.messages.push({role:"user",content:t}),l(X.session);continue}if(s.startsWith("/")){let o;z=new AbortController;try{o=await Is(s,as())}catch(s){process.stdout.write((z.signal.aborted?"操作已取消":"命令失败："+s.message)+"\n");continue}finally{z=null}if($.closed)break;if("exit"===o.action)break;if("clear"===o.action){F&&(clearInterval(F),F=null);const s=Y.getMode();Y=i({yolo:!1,ask:L,select:V,cwd:j}),Y.setMode(s),X.permissions=Y,l(X.session),S(X.session.id),X.session=c({cwd:j,model:u.model,messages:[X.session.messages[0]]}),u.sessionId=X.session.id,Z.length=0,K.list=[],Q.list.length=0,process.stdout.write(n.gray("已开启新会话（旧会话已保存，可用huicai --resume恢复）。\n"))}"refresh"===o.action&&Gs(X.session),"cd"===o.action&&(j=o.cwd,X.cwd=j,Y=i({yolo:Y.isYolo(),ask:L,select:V,cwd:j}),X.permissions=Y,y(es),({extraTools:os,clients:es,crew:ts}=await buildExtraTools(u,j,{ask:L,askUi:D,sessionOf:()=>X.session,queueInject:s=>Z.push(s),signalRef:ss,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"})),X.session=c({cwd:j,model:u.model,messages:[{role:"system",content:g(j,{memory:!1!==u.memoryEnabled,config:u})}]}),K.list=[],Q.list.length=0,u.sessionId=X.session.id,process.stdout.write(n.gray("已切换到 "+j+"（新会话，旧会话已保存）。\n"))),"autofix"===o.action&&await w({command:o.command,maxIters:o.maxIters,cwd:j,runTurnOnce:s=>ns(s),emit:s=>process.stdout.write(n.gray(s)+"\n")}),"autorun"===o.action&&await h({goal:o.goal,config:u,permissions:Y,cwd:j,runTurnOnce:s=>ns(s),emit:s=>process.stdout.write(n.gray(" · "+s)+"\n")}),"loop"===o.action&&(F&&(clearInterval(F),F=null),o.stop?process.stdout.write(n.gray("已停止循环。\n")):(process.stdout.write(n.gray(`已开始循环：每 ${o.seconds}s运行「${o.prompt}」（/loop off停止）。\n`)),F=setInterval(()=>{z||ns(o.prompt)},1e3*o.seconds))),o.prompt&&await is(o.prompt);continue}await is(s)}}}return F&&clearInterval(F),$.close(),y(es),v(),T.updater?.dispose(),_?{updated:_}:G?{relogin:!0}:(process.stdout.write(n.gray("\n再见。\n")),{})}function Ds(s){return"string"==typeof s?s:Array.isArray(s)?s.map(s=>"string"==typeof s?s:s&&"text"===s.type&&s.text||"").join(""):""}function Gs(s){const o=(s.messages||[]).filter(s=>"user"===s.role||"assistant"===s.role),e=u(s.messages);if(0===e)return;process.stdout.write(n.gray(` 已恢复会话「${s.title||""}」（${e} 轮）——历史对话如下：`)+"\n");const t=o.length>40?o.length-40:0;t&&process.stdout.write(n.gray(` （较早 ${t} 条已省略，完整记录在 ~/.huicai/sessions）`)+"\n");for(const s of o.slice(-40)){const o=Ds(s.content).trim();if(!o)continue;const e=o.length>800?o.slice(0,800)+"…":o,t="user"===s.role?n.bold(" 你 › "):n.gray(" 助手 › ");process.stdout.write(t+e.replace(/\n/g,"\n ")+"\n")}process.stdout.write(n.gray(" ——以上为历史，可继续对话——")+"\n")}export async function runOnce({config:s,cwd:o,yolo:e,prompt:t,session:r,persist:c,outputFormat:m,maxTurns:d,signal:f,throwOnError:g=!1}){t||(process.stderr.write('用法: huicai -p "你的指令"\n'),process.exit(1));const w="json"===m,h=i({yolo:e,ask:null,cwd:o}),j=[],b={current:null},k=new AbortController;b.current=k.signal;let x=!1;const I=()=>{x&&process.exit(130),x=!0,process.stderr.write(n.gray("\n已请求中断（再按一次Ctrl-C强制退出）…\n")),k.abort()};process.on("SIGINT",I);const T=()=>{k.abort(f?.reason),v()};f?.addEventListener("abort",T,{once:!0}),f?.aborted&&T();let S=[];try{const i=await buildExtraTools(s,o,{sessionOf:()=>r,queueInject:s=>j.push(s),signalRef:b,runnerHint:"到点由调度器在该目录自动执行（终端里需有 `huicai daemon` 常驻；桌面端保持应用运行亦可）。"});S=i.clients;const m=i.extraTools;let f=t;if(t.trim().startsWith("/")){const{runSlash:e}=await import("./commands.js"),n=[],i=await e(t.trim(),{cwd:o,config:s,session:r,permissions:h,print:s=>n.push(String(s))}).catch(s=>({handled:!1,error:s}));if(n.length&&process.stdout.write(n.join("\n")+"\n"),i&&i.prompt)f=i.prompt;else if(i&&i.handled)return{ok:!i.error,text:n.join("\n")}}const{text:y,images:x}=await $s(f,o,s);r.messages.push({role:"user",content:qs(y,x)}),r.compaction||(r.compaction={}),s.sessionId=r.id;const I=e&&!1!==s.wakeLockInAuto?B({reason:"无人值守任务执行中"}):null;let T;try{T=await a({config:s,messages:r.messages,permissions:h,cwd:o,extraTools:m,signal:k.signal,maxIters:d,compaction:r.compaction,io:w?{write:()=>{}}:void 0,pullQueued:()=>j.shift()||null})}finally{I?.ok&&N()}T?.lastPromptTokens&&(r.lastPromptTokens=T.lastPromptTokens),T?.lastUsage&&(r.lastUsage=T.lastUsage),c&&(l(r),p(r));const v=T?.error?Ss(T):"";if(v&&(vs(T,s,s=>process.stderr.write(s)),"auth"===v&&process.stderr.write(n.gray("请重新运行 `huicai login` 登录后重试。")+"\n")),w){const o=[...r.messages].reverse().find(s=>"assistant"===s.role&&s.content);process.stdout.write(JSON.stringify({type:"result",is_error:!!T?.error,...T?.error?{error:T.error}:{},...v?{error_kind:v}:{},result:o?String(o.content):"",usage:{input_tokens:T?.usage?.prompt??0,output_tokens:T?.usage?.completion??0},session_id:r.id,model:s.model,num_turns:u(r.messages)})+"\n")}if(g&&(T?.error||T?.aborted))throw new Error(String(T.error||"任务已取消"));return T?.error&&(process.exitCode=1),{ok:!T?.error&&!T?.aborted,error:T?.error?String(T.error):T?.aborted?"任务已取消":"",text:String([...r.messages].reverse().find(s=>"assistant"===s.role&&s.content)?.content||"")}}finally{f?.removeEventListener("abort",T),process.off("SIGINT",I),y(S),v()}}
|
package/src/schedule.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import
|
|
1
|
+
import e from"node:fs";import r from"node:path";import t from"node:os";import n from"node:crypto";import{accountDataDir as o,requireAccount as c,accountScope as i,assertAccountPath as s}from"./account-scope.js";import{CONFIG_DIR as d,loadConfig as a}from"./config.js";import{freqToCron as u}from"./schedule-freq.js";import{assertFeature as l,assertLimit as f}from"./plan-limits.js";export const SCHEDULE_FILE=r.join(d,"schedule.json");export const scheduleFile=()=>r.join(o(),"automations.json");export function readScheduleState(r=scheduleFile()){try{const t=JSON.parse(e.readFileSync(r,"utf8"));if(!Array.isArray(t.tasks))throw new Error("自动化任务存档格式错误");return t}catch(e){if("ENOENT"===e.code)return{tasks:[],revision:0,generation:0,dirty:!1};throw e}}export function editScheduleState(t,o=scheduleFile()){e.mkdirSync(r.dirname(o),{recursive:!0,mode:448});const c=o+".lock";let i;try{i=e.openSync(c,"wx",384)}catch(r){if("EEXIST"!==r.code)throw r;try{const r=Number(e.readFileSync(c,"utf8"));if(Number.isInteger(r)&&r>0)try{process.kill(r,0)}catch(r){"ESRCH"===r.code&&e.unlinkSync(c)}}catch{}throw new Error("自动化任务正在被其它进程更新，请重试")}e.writeFileSync(i,String(process.pid));const s=o+"."+n.randomUUID()+".tmp";try{const r=readScheduleState(o),n=t(r);return e.writeFileSync(s,JSON.stringify(r,null,2)+"\n",{mode:384}),e.renameSync(s,o),n}finally{e.closeSync(i),e.rmSync(c,{force:!0}),e.rmSync(s,{force:!0})}}function h(e){const r=c();return editScheduleState(t=>{const n=e(t.tasks,r);return t.generation++,t.dirty=!0,n})}function p(e){const r={...e};if(r.cwd&&s(r.cwd),r.freq&&"object"==typeof r.freq){const e=u(r.freq);e&&(r.cron=e)}return void 0===r.enabled&&(r.enabled=!0),null==r.permission&&(r.permission="default"),null==r.model&&(r.model="auto"),"watch"===r.kind?(r.watch={source:"deps",deliver:"dry",...r.watch||{}},["dry","notify","webhook","channel"].includes(r.watch.deliver)||(r.watch.deliver="dry"),"webhook"!==r.watch.deliver||String(r.watch.webhookUrl||"").trim()||(r.watch.deliver="dry"),"channel"!==r.watch.deliver||r.watch.channelId||(r.watch.deliver="dry")):null==r.kind&&(r.kind="action"),r}export function listTasks(){return i()?readScheduleState().tasks:[]}export function countActiveTasks(e=listTasks()){return e.filter(e=>e&&!(e.freq&&"once"===e.freq.kind&&e.lastRunAt)).length}export function getTask(e){return listTasks().find(r=>r.id===e)||null}export function addTask(e){return h((r,t)=>{const o=n.randomUUID(),c=Date.now(),i=p({...e,id:o,owner:t,user:String(a().user.id??a().user.userId??a().user.username),created:c,updated:c});if(!i.cron&&!i.freq)throw new Error("缺少执行频率（freq）或cron表达式");return f("scheduled_tasks",countActiveTasks(r)),r.push(i),{id:o}})}export function updateTask(e,r){return h(t=>{const n=t.findIndex(r=>r.id===e);if(n<0)return!1;(function(e){return!e||Object.keys(e).every(r=>"enabled"===r&&!0!==e.enabled)})(r)||l("agent_tasks");const{owner:o,user:c,created:i}=t[n];return t[n]=p({...t[n],...r,id:e,owner:o,user:c,created:i,updated:Date.now()}),!0})}export function removeTask(e){return h(r=>{const t=r.findIndex(r=>r.id===e);return!(t<0)&&(r.splice(t,1),!0)})}export function setEnabled(e,r){return updateTask(e,{enabled:!!r})}export function touchRun(e,r=Date.now()){return h(t=>{const n=t.find(r=>r.id===e);return!!n&&(n.lastRunAt=r,!0)})}export function crontabLine(e,r="huicai"){const t=getTask(e.id);if(!t||t.owner!==c())throw new Error("任务不属于当前登录用户");const n=e=>"'"+String(e).replace(/'/g,"'\"'\"'")+"'";return`${t.cron} ${n(r)} run-scheduled ${n(t.id)} ${n(t.owner)}`}
|
package/src/session.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import t from"node:fs";import e from"node:path";import{randomUUID as n}from"node:crypto";import{
|
|
1
|
+
import t from"node:fs";import e from"node:path";import{randomUUID as n}from"node:crypto";import{accountScope as r}from"./account-scope.js";import{CONFIG_DIR as o}from"./config.js";import{dropSessionSpill as i}from"./spill.js";import{dropSessionJobs as s}from"./jobs.js";import{dropSessionObservations as c}from"./fs-observe.js";import{dropSessionTerminals as l}from"./terminal.js";import{dropSessionLog as a}from"./sessionlog.js";export const SESSIONS_DIR=e.join(o,"sessions");function u(){t.mkdirSync(SESSIONS_DIR,{recursive:!0})}function f(t){if(!/^[\w-]{1,100}$/.test(String(t||"")))throw new Error("非法会话编号");return e.join(SESSIONS_DIR,t+".json")}export function userKeyOf(t){const e=t?.user;if(!e||"object"!=typeof e)return"";const n=e.id??e.userId??e.username??"";return null==n?"":String(n)}export function turnCount(t){return(t||[]).filter(t=>"user"===t.role).length}export function createSession({cwd:t,model:e,messages:n,parentSession:o,origin:i,delegationDepth:s,label:c,user:l}){const a=Date.now(),u={id:Date.now().toString(36)+"-"+Math.random().toString(36).slice(2,6),owner:r(),cwd:t,model:e,createdAt:a,updatedAt:a,title:"",messages:n||[],compaction:{}};return o&&(u.parentSession=String(o)),i&&(u.origin=String(i)),Number.isFinite(s)&&(u.delegationDepth=Math.max(0,Math.floor(s))),c&&(u.label=String(c).replace(/\s+/g," ").slice(0,120)),l&&(u.user=String(l)),u}export function saveSession(e){if(!e||!e.id)return;if((e.owner||"")!==r())return;let o;try{u(),e.updatedAt=Date.now();const r=JSON.stringify(e),i=f(e.id);o=i+"."+n()+".tmp",t.writeFileSync(o,r,{mode:384,flag:"wx"}),t.renameSync(o,i)}catch(t){process.stderr.write(`（会话保存失败，不影响对话：${t.message}）\n`)}finally{if(o)try{t.unlinkSync(o)}catch{}}}export function loadSession(e){if(!e)return null;try{const n=JSON.parse(t.readFileSync(f(e),"utf8"));return n&&Array.isArray(n.messages)&&(n.owner||"")===r()?(n.id||(n.id=e),n.compaction&&"object"==typeof n.compaction||(n.compaction={}),n):null}catch{return null}}export function listSessions(n){let o;try{u(),o=t.readdirSync(SESSIONS_DIR).filter(t=>t.endsWith(".json"))}catch{return[]}const i=[];for(const s of o)try{const o=JSON.parse(t.readFileSync(e.join(SESSIONS_DIR,s),"utf8"));if((o.owner||"")!==r())continue;if(n&&o.cwd!==n)continue;i.push({id:o.id||s.replace(/\.json$/,""),title:o.title||"(未命名会话)",cwd:o.cwd,model:o.model,updatedAt:o.updatedAt||0,turns:turnCount(o.messages),user:o.user?String(o.user):"",owner:o.owner||"",origin:o.origin||"",parentSession:o.parentSession||"",delegationDepth:Number.isFinite(o.delegationDepth)?o.delegationDepth:0,label:o.label||""})}catch{}return i.sort((t,e)=>e.updatedAt-t.updatedAt),i}export function mostRecent(t){const e=listSessions(t);return e.length?e[0].id:null}export function deleteSession(e,n){let r;try{r=f(e)}catch{return!1}if(t.existsSync(r)&&!loadSession(e))return!1;const o=n||new Set;if(o.has(e))return!1;o.add(e);for(const t of listChildSessions(e))deleteSession(t.id,o);let u=!1;try{t.unlinkSync(f(e)),u=!0}catch{u=!1}return i(e),s(e),l(e),a(e),c(e),u}export function listChildSessions(t){return t?listSessions(null).filter(e=>e.parentSession===String(t)):[]}export function relTime(t){if(!t)return"未知时间";const e=Math.max(0,Math.floor((Date.now()-t)/1e3));if(e<60)return e+" 秒前";const n=Math.floor(e/60);if(n<60)return n+" 分钟前";const r=Math.floor(n/60);if(r<24)return r+" 小时前";return Math.floor(r/24)+" 天前"}
|
package/src/sessionindex.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import e from"node:fs";import t from"node:path";import
|
|
1
|
+
import e from"node:fs";import{accountScope as t}from"./account-scope.js";import n from"node:path";import o from"node:zlib";import{CONFIG_DIR as r}from"./config.js";import{SESSIONS_DIR as i,userKeyOf as s}from"./session.js";import{tokenize as c}from"./codeindex.js";const l=n.join(r,"cache","sessions-index.json.gz");let u=null;function a(e){if(!e||"system"===e.role)return"";const t=(n=e.content,"string"==typeof n?n:Array.isArray(n)?n.filter(e=>"text"===e?.type).map(e=>e.text||"").join("\n"):"").trim();var n;return t?t.slice(0,"tool"===e.role?1e3:4e3):""}function f(e){const t=new Map;for(const n of c(e))t.set(n,(t.get(n)||0)+1);return t}function d(t){let o;try{o=JSON.parse(e.readFileSync(t,"utf8"))}catch{return null}if(!o||!Array.isArray(o.messages))return null;const r=o.messages.slice(-300),i=o.messages.length-r.length,s=[];for(let e=0;e<r.length;e++){const t=a(r[e]);if(!t)continue;const n=f(t);s.push({i:i+e,role:r[e].role,text:t,tf:[...n],len:t.length})}return s.length?{meta:{id:o.id||n.basename(t,".json"),title:o.title||"",cwd:o.cwd||"",model:o.model||"",updatedAt:o.updatedAt||0,user:o.user?String(o.user):"",owner:o.owner||"",origin:o.origin||""},docs:s}:null}export function buildIndex(){const t=function(){try{const t=JSON.parse(o.gunzipSync(e.readFileSync(l)).toString("utf8"));return t&&2===t.v&&t.files&&"object"==typeof t.files?t.files:null}catch{return null}}()||Object.create(null),r=Object.create(null),s=Date.now();let c=[];try{c=e.readdirSync(i).filter(e=>e.endsWith(".json")).map(t=>{const o=n.join(i,t);try{const n=e.statSync(o);return{full:o,key:t,mtimeMs:n.mtimeMs,size:n.size}}catch{return null}}).filter(Boolean).sort((e,t)=>t.mtimeMs-e.mtimeMs).slice(0,800)}catch{c=[]}for(const e of c){const n=t[e.key];if(n&&n.mtimeMs===e.mtimeMs&&n.size===e.size&&Array.isArray(n.docs)){r[e.key]=n;continue}if(Date.now()-s>3e3){n&&(r[e.key]=n);continue}const o=d(e.full);o&&(r[e.key]={mtimeMs:e.mtimeMs,size:e.size,meta:o.meta,docs:o.docs})}!function(t){try{e.mkdirSync(n.dirname(l),{recursive:!0});const r=l+"."+process.pid+".tmp";e.writeFileSync(r,o.gzipSync(JSON.stringify({v:2,at:Date.now(),files:t})),{mode:384}),e.renameSync(r,l)}catch{}}(r);const u=[],a=new Map;let f=0;for(const e of Object.keys(r)){const t=r[e];for(const e of t.docs){const n=new Map(e.tf);u.push({meta:t.meta,i:e.i,role:e.role,text:e.text,tf:n,len:e.len}),f+=e.len;for(const e of n.keys())a.set(e,(a.get(e)||0)+1)}}return{entries:u,df:a,N:u.length,avgLen:u.length?f/u.length:1}}export function getIndex({force:t=!1}={}){const n=function(){try{return String(e.statSync(i).mtimeMs)}catch{return"none"}}();if(!t&&u&&u.stamp===n&&Date.now()-u.builtAt<3e5)return u.index;const o=buildIndex();return u={index:o,builtAt:Date.now(),stamp:n},o}export function clearIndex(){u=null;try{e.rmSync(l,{force:!0})}catch{}}function m(e,t){const n=e.toLowerCase();let o=-1;for(const e of t){const t=n.indexOf(e);t>=0&&(o<0||t<o)&&(o=t)}o<0&&(o=0);const r=Math.max(0,o-90),i=Math.min(e.length,o+90);return(r>0?"…":"")+e.slice(r,i).replace(/\s+/g," ").trim()+(i<e.length?"…":"")}export function searchSessions(e,n={}){const o=[...new Set(c(e||""))];if(!o.length)return[];const r=n.index||getIndex();if(!r.N)return[];const{df:i,N:s,avgLen:l}=r,u=new Map;for(const e of o){const t=i.get(e)||1;u.set(e,Math.log(1+(s-t+.5)/(t+.5)))}const a=void 0===n.userKey?null:String(n.userKey||""),f=!!n.includeUnowned,d=!!n.includeSubagents,p=new Map,h=new Set;for(const e of r.entries){if((e.meta.owner||"")!==t())continue;if(null!==a){const t=e.meta.user||"";if(t!==a&&(!f||""!==t)){""===t&&a&&h.add(e.sessionId||e.meta.id||e.id);continue}}if(!d&&"subagent"===e.meta.origin)continue;if(n.cwd&&e.meta.cwd!==n.cwd)continue;let r=0;for(const t of o){const n=e.tf.get(t);n&&(r+=u.get(t)*(2.5*n/(n+1.5*(.25+e.len/l*.75))))}if(r<=0)continue;const i=new Set(c(e.meta.title));for(const e of o)i.has(e)&&(r+=.8*u.get(e));const s=p.get(e.meta.id);(!s||r>s.score)&&p.set(e.meta.id,{id:e.meta.id,title:e.meta.title,cwd:e.meta.cwd,model:e.meta.model,updatedAt:e.meta.updatedAt,role:e.role,i:e.i,score:r,snippet:m(e.text,o)})}const g=Number.isFinite(n.k)&&n.k>0?Math.min(n.k,50):10,y=[...p.values()].sort((e,t)=>t.score-e.score).slice(0,g);return Object.defineProperty(y,"unownedFiltered",{value:h.size,enumerable:!1}),y}function p(e){if(!e)return"未知时间";const t=Math.max(0,Math.floor((Date.now()-e)/1e3));if(t<60)return t+" 秒前";const n=Math.floor(t/60);if(n<60)return n+" 分钟前";const o=Math.floor(n/60);return o<24?o+" 小时前":Math.floor(o/24)+" 天前"}export function formatResults(e,t){if(!e.length){const n=e.unownedFiltered||0;return`历史会话中没有匹配「${t}」的内容。`+(n?`\n（另有 ${n} 次**登录之前留下的**旧会话命中了，因不带用户标记而未列出。用 \`/history `+t+" --legacy` 可一并查看。）":"")}const n=[`历史会话中匹配「${t}」的 ${e.length} 次对话（按相关性）：`,""];for(const t of e)n.push(`· ${t.title||"(未命名会话)"} [${t.id}] ${p(t.updatedAt)}${t.cwd?" "+t.cwd:""}`),n.push(` ${"user"===t.role?"用户":"tool"===t.role?"工具":"助手"}：${t.snippet}`);return n.push("","用 `huicai --resume <会话id>` 可继续该对话。"),n.join("\n")}export async function searchHistory(e,t,n){const o=String(t?.query||"").trim();if(!o)return"缺少query（要在历史对话里找什么）";const r=searchSessions(o,{k:Number.isFinite(t?.k)&&t.k>0?Math.min(t.k,20):8,userKey:s(n||{}),cwd:t?.this_project?e:void 0,includeUnowned:!!t?.include_unowned}),i=formatResults(r,o);return r.length||!r.unownedFiltered||t?.include_unowned?i:i+`\n（你可以带include_unowned=true重新调用search_history把这 ${r.unownedFiltered} 次旧会话一并检索。）`}
|