blade-code 0.7.0 → 0.7.2

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.
Files changed (73) hide show
  1. package/dist/blade.js +2 -2
  2. package/dist/chunk-108p1xw7.js +1 -0
  3. package/dist/{chunk-nsbhtvqd.js → chunk-1mge0pzv.js} +1 -1
  4. package/dist/{chunk-hc08zg54.js → chunk-1w719wd9.js} +1 -1
  5. package/dist/{chunk-q6c9qs9b.js → chunk-22gxb9qs.js} +2 -2
  6. package/dist/{chunk-sw6ngzbg.js → chunk-275gt6q1.js} +1 -1
  7. package/dist/chunk-42q476bq.js +1 -0
  8. package/dist/{chunk-mt1c5da4.js → chunk-4hk7wfat.js} +2 -2
  9. package/dist/{chunk-tn3xjczg.js → chunk-67zx7ymj.js} +1 -1
  10. package/dist/chunk-7d1pn8a1.js +2 -0
  11. package/dist/{chunk-kk8akm7s.js → chunk-8xdqtspw.js} +1 -1
  12. package/dist/{chunk-c2b6ekqx.js → chunk-9zwf87ne.js} +2 -2
  13. package/dist/chunk-abpvdaq2.js +6 -0
  14. package/dist/{chunk-m22zeqdy.js → chunk-b6mxeey0.js} +2 -2
  15. package/dist/chunk-cadngkwr.js +2 -0
  16. package/dist/{chunk-0afbeec0.js → chunk-cfrbsnfb.js} +1 -1
  17. package/dist/{chunk-k4behavv.js → chunk-cqrkeq20.js} +1 -1
  18. package/dist/{chunk-7z6ba6t9.js → chunk-f9b8h03c.js} +1 -1
  19. package/dist/chunk-g0h3rans.js +6 -0
  20. package/dist/chunk-hv0bzn31.js +1 -0
  21. package/dist/chunk-j3k2c6tm.js +1 -0
  22. package/dist/chunk-je6r01h4.js +1 -0
  23. package/dist/{chunk-31chcjw6.js → chunk-k18bq4mn.js} +2 -2
  24. package/dist/{chunk-ft5nt5b5.js → chunk-m2k4h3bs.js} +1 -1
  25. package/dist/chunk-m4a480k7.js +9 -0
  26. package/dist/chunk-nh37erd3.js +1 -0
  27. package/dist/{chunk-rzz6r1g9.js → chunk-p6d7k1x6.js} +3 -3
  28. package/dist/chunk-p9pthxkm.js +6 -0
  29. package/dist/{chunk-96w1szbp.js → chunk-qw2wy4jc.js} +1 -1
  30. package/dist/{chunk-zmp3vdcj.js → chunk-rmtcqc8q.js} +1 -1
  31. package/dist/{chunk-gxnwrec6.js → chunk-rr62k3ar.js} +2 -2
  32. package/dist/{chunk-pkzy724w.js → chunk-t3pyakvc.js} +2 -2
  33. package/dist/chunk-t6y57501.js +1533 -0
  34. package/dist/chunk-wc2xb6ge.js +24 -0
  35. package/dist/{chunk-f76d6jxg.js → chunk-wcm0nc84.js} +1 -1
  36. package/dist/{chunk-pf33bn8s.js → chunk-xa8szzc6.js} +1 -1
  37. package/dist/{chunk-e21htmg4.js → chunk-z0qxaky4.js} +1 -1
  38. package/dist/chunk-z72cp8f5.js +150 -0
  39. package/dist/chunk-zcreym5g.js +615 -0
  40. package/dist/chunk-ztpseq3n.js +1 -0
  41. package/dist/web/assets/{FilePreview-DLnUh-wQ.js → FilePreview-DXDzoFFc.js} +1 -1
  42. package/dist/web/assets/{MarkdownRenderer-C5KTk9XG.js → MarkdownRenderer-BVPOLK1b.js} +1 -1
  43. package/dist/web/assets/{McpModal-B9qJ4ewt.js → McpModal-DeTf34Fw.js} +1 -1
  44. package/dist/web/assets/{SettingsModal-DpeZ6cJb.js → SettingsModal-CDLkPyL-.js} +1 -1
  45. package/dist/web/assets/{SettingsStore-IzKtyAnM.js → SettingsStore-DYKvNeV_.js} +1 -1
  46. package/dist/web/assets/{SkillsModal-BDfzgJIj.js → SkillsModal-D784AUrt.js} +1 -1
  47. package/dist/web/assets/{TerminalPanel-B2iTF_CR.js → TerminalPanel-BosJuIct.js} +1 -1
  48. package/dist/web/assets/{dialog-sI3ihAze.js → dialog-CgyS_4Eo.js} +1 -1
  49. package/dist/web/assets/index-Abieffcr.js +10 -0
  50. package/dist/web/assets/index-CQ3JnuUf.css +1 -0
  51. package/dist/web/index.html +2 -2
  52. package/package.json +1 -1
  53. package/dist/chunk-2nnak657.js +0 -6
  54. package/dist/chunk-64jv62vs.js +0 -1
  55. package/dist/chunk-6f63v1mk.js +0 -2
  56. package/dist/chunk-93xhspe5.js +0 -1
  57. package/dist/chunk-fehae50v.js +0 -1
  58. package/dist/chunk-gertm8dw.js +0 -157
  59. package/dist/chunk-gscq4fmx.js +0 -49
  60. package/dist/chunk-hscq9623.js +0 -1
  61. package/dist/chunk-kpq18y6v.js +0 -6
  62. package/dist/chunk-mnw1xcwn.js +0 -76
  63. package/dist/chunk-p9s685ta.js +0 -1
  64. package/dist/chunk-phnararg.js +0 -1
  65. package/dist/chunk-v97f7je6.js +0 -2138
  66. package/dist/chunk-vzebrztg.js +0 -9
  67. package/dist/chunk-wre40mf1.js +0 -2
  68. package/dist/chunk-x46rv9sv.js +0 -6
  69. package/dist/chunk-ygta7118.js +0 -1
  70. package/dist/chunk-zd9stbct.js +0 -618
  71. package/dist/chunk-zjh9qqvq.js +0 -1
  72. package/dist/web/assets/index-D8mo0jzF.js +0 -10
  73. package/dist/web/assets/index-DmXmCQOW.css +0 -1
@@ -0,0 +1,9 @@
1
+ import{Nb as R}from"./chunk-4hk7wfat.js";var m=["anthropic","openai","deepseek","google-generative-ai","groq","openrouter"],d={anthropic:"",openai:"","google-generative-ai":"",deepseek:"",groq:"",openrouter:"",azure:"",ollama:"",together:"",fireworks:"",mistral:"",cohere:"",perplexity:"",xai:"",default:""},g={anthropic:"https://api.anthropic.com",openai:"https://api.openai.com/v1",google:"https://generativelanguage.googleapis.com/v1beta",deepseek:"https://api.deepseek.com/v1",groq:"https://api.groq.com/openai/v1",openrouter:"https://openrouter.ai/api/v1",togetherai:"https://api.together.xyz/v1","fireworks-ai":"https://api.fireworks.ai/inference/v1",mistral:"https://api.mistral.ai/v1",cohere:"https://api.cohere.ai/v1",perplexity:"https://api.perplexity.ai",xai:"https://api.x.ai/v1",cerebras:"https://api.cerebras.ai/v1","novita-ai":"https://api.novita.ai/v3/openai",nvidia:"https://integrate.api.nvidia.com/v1"},h={anthropic:{"anthropic-beta":"claude-code-20250219,interleaved-thinking-2025-05-14,fine-grained-tool-streaming-2025-05-14"},openrouter:{"HTTP-Referer":"https://github.com/anthropics/blade","X-Title":"Blade"},cerebras:{"X-Cerebras-3rd-Party-Integration":"blade"},vercel:{"http-referer":"https://github.com/anthropics/blade","x-title":"blade"},zenmux:{"HTTP-Referer":"https://github.com/anthropics/blade","X-Title":"blade"}},T=(Q)=>{return h[Q]||{}};import{createAnthropic as k}from"@ai-sdk/anthropic";import{createAzure as I}from"@ai-sdk/azure";import{createDeepSeek as A}from"@ai-sdk/deepseek";import{createGoogleGenerativeAI as U}from"@ai-sdk/google";import{createOpenAI as C}from"@ai-sdk/openai";import{createOpenAICompatible as H}from"@ai-sdk/openai-compatible";import{generateText as v,jsonSchema as f,streamText as b}from"ai";var M=new Set(["user-cancel","interrupt","sibling-error","timeout"]);function E(Q){return typeof Q==="string"&&M.has(Q)}function n(Q){if(!Q.aborted)return;if(E(Q.reason))return Q.reason;return"user-cancel"}function t(Q){if(E(Q))return!0;if(!(Q instanceof Error))return!1;if(Q.name==="AbortError")return!0;if(Q instanceof DOMException&&Q.name==="AbortError")return!0;if(Q.message.includes("aborted"))return!0;return!1}function r(...Q){let q=Q.filter(Boolean);if(q.length===0)return new AbortController().signal;if(q.length===1)return q[0];let Y=AbortSignal;if(typeof Y.any==="function")return Y.any(q);for(let $ of q)if($.aborted)return $;let J=new AbortController,X=new Map;for(let $ of q){let B=()=>{J.abort($.reason);for(let[Z,F]of X)Z.removeEventListener("abort",F)};X.set($,B)}for(let[$,B]of X)$.addEventListener("abort",B,{once:!0});return J.signal}function D(Q,q,Y){let J=Y?.throwOnAbort??!1;if(q?.aborted){if(J)return Promise.reject(new DOMException("Aborted","AbortError"));return Promise.resolve()}return new Promise((X,$)=>{let B,Z=setTimeout(()=>{if(q&&B)q.removeEventListener("abort",B);X()},Q);if(q)B=()=>{if(clearTimeout(Z),J)$(new DOMException("Aborted","AbortError"));else X()},q.addEventListener("abort",B,{once:!0})})}var z=R("Chat");function x(Q){let q=new Set;for(let Y of Q)if(Y.role==="assistant"&&Y.tool_calls)for(let J of Y.tool_calls)q.add(J.id);return Q.filter((Y)=>{if(Y.role==="tool"){if(!Y.tool_call_id)return!1;return q.has(Y.tool_call_id)}return!0})}function V(Q){if(typeof Q==="string")return Q;return Q.filter((q)=>q.type==="text").map((q)=>q.text).join(`
2
+ `)}function y(Q){let q=Q;return q.text??q.textDelta??q.delta}function u(Q,q={}){try{return JSON.parse(Q)}catch{return z.warn("[VercelAIChatService] Failed to parse JSON, using fallback",{str:Q}),q}}class O{model;config;constructor(Q){this.config=Q,this.model=this.createModel(Q),z.debug("[VercelAIChatService] Initialized",{provider:Q.provider,model:Q.model})}createModel(Q){let{provider:q,apiKey:Y,baseUrl:J,model:X,customHeaders:$,apiVersion:B}=Q;switch(q){case"openai":return C({apiKey:Y,baseURL:J||void 0,headers:$})(X);case"anthropic":return k({apiKey:Y,baseURL:J||void 0,headers:$})(X);case"gemini":{if(J&&!this.isGeminiOfficialUrl(J))return H({name:"gemini",apiKey:Y,baseURL:J,headers:$})(X);return U({apiKey:Y,baseURL:J||void 0})(X)}case"google":case"google-generative-ai":case"google-vertex":{if(J&&!this.isGeminiOfficialUrl(J))return H({name:q,apiKey:Y,baseURL:J,headers:$})(X);return U({apiKey:Y,baseURL:J||void 0})(X)}case"azure-openai":{let Z=this.extractAzureResourceName(J);if(Z)return I({apiKey:Y,resourceName:Z,apiVersion:B||"2024-08-01-preview"})(X);let F=this.buildAzureBaseUrl(J,X);return H({name:"azure-openai",apiKey:Y,baseURL:F,headers:{...$,"api-key":Y},queryParams:{"api-version":B||"2024-08-01-preview"}})(X)}case"azure":{let Z=this.extractAzureResourceName(J);if(Z)return I({apiKey:Y,resourceName:Z,apiVersion:B||"2024-08-01-preview"})(X);let F=this.buildAzureBaseUrl(J,X);return H({name:"azure-openai",apiKey:Y,baseURL:F,headers:{...$,"api-key":Y},queryParams:{"api-version":B||"2024-08-01-preview"}})(X)}case"deepseek":return A({apiKey:Y,baseURL:J||void 0,headers:$})(X);default:return H({name:q,apiKey:Y,baseURL:J,headers:$})(X)}}extractAzureResourceName(Q){if(!Q)return;let q=Q.match(/https:\/\/([^.]+)\.openai\.azure(?:\.com|\.us|\.cn|\.de)/);return q?q[1]:void 0}buildAzureBaseUrl(Q,q){if(!Q)return"";let Y=Q.replace(/\/$/,"").replace(/\?.*$/,"");if(Y.includes("/openai/deployments/"))return Y;return`${Y}/openai/deployments/${q}`}isGeminiOfficialUrl(Q){return Q.includes("generativelanguage.googleapis.com")||Q.includes("aiplatform.googleapis.com")}needsThinkingReplayMetadata(){return/deepseek/i.test([this.config.provider,this.config.model,this.config.baseUrl].join(" "))}getThinkingReplayProviderOptions(Q){let q=Q?.trim();if(!q||!this.needsThinkingReplayMetadata())return;return{openaiCompatible:{reasoning_content:q}}}shouldFlattenDeepSeekThinkingToolHistory(){return this.needsThinkingReplayMetadata()}flattenDeepSeekThinkingToolHistory(Q){let q=[],Y,J=()=>{if(!Y)return;q.push({role:"assistant",content:V(Y.content).trim()||"I will use the requested tool and continue from its result."}),Y=void 0};for(let X of Q){if(X.role==="system"){J(),q.push({role:"user",content:`<system>
3
+ ${V(X.content)}
4
+ </system>`});continue}if(X.role==="assistant"&&X.tool_calls&&X.tool_calls.length>0&&X.reasoningContent?.trim()){J(),Y=X;continue}if(X.role==="tool"&&Y){let $=Y.tool_calls?.find((Z)=>Z.id===X.tool_call_id),B=$?.type==="function"?$.function:void 0;q.push({role:"user",content:`Tool result for ${X.name||B?.name||"tool"}${B?.arguments?`(${B.arguments})`:""}:
5
+ ${V(X.content)}
6
+
7
+ Continue from this tool result.`});continue}J(),q.push(X)}return J(),q}convertMessages(Q){let q=[],Y=this.needsThinkingReplayMetadata(),J=this.shouldFlattenDeepSeekThinkingToolHistory()?this.flattenDeepSeekThinkingToolHistory(Q):Q;for(let X of J)if(X.role==="system")if(Array.isArray(X.content)){let $=X.content.find((Z)=>Z.type==="text"),B={role:"system",content:V(X.content)};if($?.providerOptions)B.providerOptions=$.providerOptions;q.push(B)}else q.push({role:"system",content:X.content});else if(X.role==="user")if(Array.isArray(X.content)){let $=X.content.map((B)=>{if(B.type==="text"){let Z={type:"text",text:B.text};if(B.providerOptions)Z.providerOptions=B.providerOptions;return Z}return{type:"image",image:B.image_url.url}});q.push({role:"user",content:$})}else q.push({role:"user",content:X.content});else if(X.role==="assistant")if(X.tool_calls&&X.tool_calls.length>0){let $=X.tool_calls.map((W)=>{let G=W.function;return{type:"tool-call",toolCallId:W.id,toolName:G?.name||"",input:u(G?.arguments||"{}",{})}}),B=V(X.content),Z=Y&&X.reasoningContent?.trim()?[{type:"reasoning",text:X.reasoningContent}]:[],F=this.getThinkingReplayProviderOptions(X.reasoningContent);if(B){let W={role:"assistant",content:[...Z,{type:"text",text:B},...$]};if(F)W.providerOptions=F;q.push(W)}else if(Z.length>0){let W={role:"assistant",content:[...Z,...$]};if(F)W.providerOptions=F;q.push(W)}else q.push({role:"assistant",content:$})}else if(Y&&X.reasoningContent?.trim()){let $=V(X.content),B=[{type:"reasoning",text:X.reasoningContent}];if($)B.push({type:"text",text:$});let Z={role:"assistant",content:B},F=this.getThinkingReplayProviderOptions(X.reasoningContent);if(F)Z.providerOptions=F;q.push(Z)}else q.push({role:"assistant",content:V(X.content)});else if(X.role==="tool")q.push({role:"tool",content:[{type:"tool-result",toolCallId:X.tool_call_id,toolName:X.name||"unknown",output:{type:"text",value:V(X.content)}}]});if(this.config.enablePromptCaching&&this.config.provider==="anthropic"){let X=0;for(let $ of q)if($.role==="system"&&X<2)$.providerOptions={anthropic:{cacheControl:{type:"ephemeral"}}},X++;for(let $=q.length-1;$>=0;$--)if(q[$].role==="user"){q[$].providerOptions={anthropic:{cacheControl:{type:"ephemeral"}}};break}}return q}convertTools(Q){if(!Q||Q.length===0)return;let q={};for(let Y of Q)q[Y.name]={description:Y.description,inputSchema:f(Y.parameters)};return q}convertToolCalls(Q){return Q.map((q)=>({id:q.toolCallId,type:"function",function:{name:q.toolName,arguments:JSON.stringify(q.args??q.input??{})}}))}convertUsage(Q,q){if(!Q)return;let Y=Q.promptTokens??Q.inputTokens??0,J=Q.completionTokens??Q.outputTokens??0,X={promptTokens:Y,completionTokens:J,totalTokens:Q.totalTokens??Y+J};if(Q.reasoningTokens)X.reasoningTokens=Q.reasoningTokens;if(q?.anthropic){if(q.anthropic.cacheCreationInputTokens!==void 0)X.cacheCreationInputTokens=q.anthropic.cacheCreationInputTokens;if(q.anthropic.cacheReadInputTokens!==void 0)X.cacheReadInputTokens=q.anthropic.cacheReadInputTokens}return X}isFallbackableError(Q){let q=[],Y=new Set,J=Q;while(J&&!Y.has(J)){if(q.push(J),Y.add(J),typeof J!=="object")break;let Z=J;J=Z.lastError??Z.cause}let X=q.filter((Z)=>Z instanceof Error).map((Z)=>Z.message.toLowerCase()),$=X.join(`
8
+ `);if(["prompt_too_long","prompt is too long","maximum context length","context length exceeded","context_length_exceeded","request too large"].some((Z)=>$.includes(Z)))return!1;let B=["timeout","timed out","econnreset","econnrefused","enotfound","eai_again","etimedout","fetch failed","network error","socket hang up","connection reset","connection refused"];if(B.some((Z)=>$.includes(Z)))return!0;for(let Z of q){if(!Z||typeof Z!=="object")continue;let F=Z,W=F.status??F.statusCode??F.response?.status??F.response?.statusCode;if(W!==void 0&&([408,409,429].includes(W)||W>=500))return!0;let G=F.code;if(typeof G==="string"&&B.some((L)=>G.toLowerCase().includes(L)))return!0}for(let Z of X){let F=Z.match(/\bstatus(?:\s+code)?[:\s]*(\d{3})\b/),W=F?Number(F[1]):void 0;if(W!==void 0&&([408,409,429].includes(W)||W>=500))return!0}return!1}async retryWithBackoff(Q,q,Y=2,J=1000){let X;for(let $=0;$<=Y;$++)try{return await Q()}catch(B){if(X=B,q?.aborted)await D(0,q,{throwOnAbort:!0});if($>=Y)break;if(!this.isFallbackableError(B))throw B;let Z=J*Math.pow(2,$);z.debug(`[VercelAIChatService] Retry ${$+1}/${Y} after ${Z}ms`),await D(Z,q,{throwOnAbort:!0})}throw X}getFallbackModelIds(){let Q=[];if(this.config.fallbackModels&&this.config.fallbackModels.length>0)Q.push(...this.config.fallbackModels);else if(this.config.fallbackModel)Q.push(this.config.fallbackModel);return Q}getThinkingProviderOptions(){if(!this.config.supportsThinking)return;let Q=this.config.thinkingMode??"budget";if(Q==="off")return;let q=Q==="adaptive",Y=this.config.thinkingBudget??1e4;if(this.config.provider==="anthropic")return{anthropic:{thinking:q?{type:"enabled",budgetTokens:0}:{type:"enabled",budgetTokens:Y}}};if(this.config.provider==="deepseek"||/deepseek/i.test(this.config.model))return{deepseek:{thinking:q?{type:"enabled",budgetTokens:0}:{type:"enabled",budgetTokens:Y}}};return}getStreamTextOptions(Q,q,Y,J){let X=J;if(this.config.timeout&&this.config.timeout>0){let Z=AbortSignal.timeout(this.config.timeout);if(J)X=AbortSignal.any([J,Z]);else X=Z}let $={model:Q,messages:q,tools:Y,maxOutputTokens:this.config.maxOutputTokens,maxRetries:0,temperature:this.config.temperature??0,abortSignal:X,allowSystemInMessages:!0},B=this.getThinkingProviderOptions();if(B)$.providerOptions=B;return $}async chat(Q,q,Y){let J=Date.now();z.debug("[VercelAIChatService] Starting chat request");let X=x(Q),$=this.convertMessages(X),B=this.convertTools(q),Z=async(F)=>{let W=await v(this.getStreamTextOptions(F,$,B,Y)),G=W.toolCalls&&W.toolCalls.length>0?this.convertToolCalls(W.toolCalls):void 0,L=Array.isArray(W.reasoning)?W.reasoning.map((P)=>P.text).join(""):void 0;return{content:W.text,reasoningContent:L,toolCalls:G,usage:this.convertUsage(W.usage,W.providerMetadata),finishReason:W.finishReason}};try{return await this.retryWithBackoff(()=>Z(this.model),Y,this.config.maxRetries)}catch(F){let W=Date.now()-J;if(z.error("[VercelAIChatService] Chat failed after",W,"ms"),Y?.aborted)await D(0,Y,{throwOnAbort:!0});if(!this.isFallbackableError(F))throw F;let G=this.getFallbackModelIds();if(G.length===0)throw F;for(let L of G){z.warn(`[VercelAIChatService] Trying fallback model: ${L}`);let P=this.createModel({...this.config,model:L});try{return await Z(P)}catch(_){if(z.warn(`[VercelAIChatService] Fallback ${L} failed: ${_ instanceof Error?_.message:_}`),Y?.aborted)await D(0,Y,{throwOnAbort:!0});if(!this.isFallbackableError(_))throw _}}throw F}}async*streamChat(Q,q,Y){let J=Date.now();z.debug("[VercelAIChatService] Starting stream request");let X=x(Q),$=this.convertMessages(X),B=this.convertTools(q),Z=async function*(_,N){let j=b(_.getStreamTextOptions(N,$,B,Y)),K=0;for await(let w of j.fullStream)switch(w.type){case"text-delta":yield{content:y(w)};break;case"reasoning-delta":yield{reasoningContent:y(w)};break;case"tool-call":yield{toolCalls:[{index:K++,id:w.toolCallId,type:"function",function:{name:w.toolName,arguments:JSON.stringify(w.args??w.input??{})}}]};break;case"finish":yield{finishReason:w.finishReason,usage:_.convertUsage(w.totalUsage,w.providerMetadata)};break}},F=this.config.maxRetries??2,W,G=!1;for(let _=0;_<=F;_++)try{for await(let j of Z(this,this.model))G=!0,yield j;let N=Date.now()-J;z.debug("[VercelAIChatService] Stream completed in",N,"ms");return}catch(N){if(W=N,Y?.aborted)await D(0,Y,{throwOnAbort:!0});if(G)throw N;if(!this.isFallbackableError(N))throw N;if(_<F){let j=1000*Math.pow(2,_);z.debug(`[VercelAIChatService] Stream retry ${_+1}/${F} after ${j}ms`),await D(j,Y,{throwOnAbort:!0});continue}}let L=Date.now()-J;z.error("[VercelAIChatService] Stream failed after",L,"ms");let P=this.getFallbackModelIds();if(P.length===0)throw W;for(let _ of P){z.warn(`[VercelAIChatService] Stream fallback: ${_}`),yield{modelFallback:!0};let N=this.createModel({...this.config,model:_}),j=!1;try{for await(let K of Z(this,N))j=!0,yield K;return}catch(K){if(z.warn(`[VercelAIChatService] Stream fallback ${_} failed`),Y?.aborted)await D(0,Y,{throwOnAbort:!0});if(j)throw K;if(!this.isFallbackableError(K))throw K}}throw W}getConfig(){return{...this.config}}updateConfig(Q){z.debug("[VercelAIChatService] Updating configuration"),this.config={...this.config,...Q},this.model=this.createModel(this.config),z.debug("[VercelAIChatService] Configuration updated")}}var c=R("Service");async function Gq(Q){let q=Q;if(q.provider){let Y=T(q.provider);if(Object.keys(Y).length>0)q={...q,customHeaders:{...Y,...q.customHeaders}},c.debug(`Injected ${q.provider} specific headers:`,Object.keys(Y))}return p(q)}function p(Q){return new O(Q)}
9
+ export{m as xb,d as yb,g as zb,n as Ab,t as Bb,r as Cb,Gq as Db};
@@ -0,0 +1 @@
1
+ import{v as a,w as b,x as c}from"./chunk-m2k4h3bs.js";import"./chunk-cqrkeq20.js";import"./chunk-z0qxaky4.js";import"./chunk-wc2xb6ge.js";import"./chunk-b6mxeey0.js";import"./chunk-cadngkwr.js";import"./chunk-4hk7wfat.js";import"./chunk-cfrbsnfb.js";export{a as validatePermissions,c as validateOutput,b as loadConfiguration};
@@ -1,4 +1,4 @@
1
- import{Eb as n}from"./chunk-wre40mf1.js";import{Ob as r}from"./chunk-mt1c5da4.js";import{Qb as b}from"./chunk-0afbeec0.js";import{nanoid as A}from"nanoid";var N;((R)=>{R.PreToolUse="PreToolUse";R.PostToolUse="PostToolUse";R.PostToolUseFailure="PostToolUseFailure";R.PermissionRequest="PermissionRequest";R.UserPromptSubmit="UserPromptSubmit";R.SessionStart="SessionStart";R.SessionEnd="SessionEnd";R.Stop="Stop";R.SubagentStop="SubagentStop";R.Notification="Notification";R.Compaction="Compaction"})(N||={});var C;((W)=>{W.Approve="approve";W.Block="block";W.Async="async"})(C||={});var S;((W)=>{W.Allow="allow";W.Deny="deny";W.Ask="ask"})(S||={});var I={enabled:!1,defaultTimeout:60,timeoutBehavior:"ignore",failureBehavior:"ignore",maxConcurrentHooks:5,httpPolicy:{},PreToolUse:[],PostToolUse:[{name:"builtin:code-review-sensor",matcher:{tools:"Edit|Write"},hooks:[{type:"prompt",prompt:"审查此代码变更,重点检查:"+"1) 安全漏洞(命令注入、XSS、SQL 注入、硬编码密钥/密码)"+"2) 明显的逻辑错误(无限循环、off-by-one、空引用)"+"3) 类型安全问题。"+"如果发现严重问题,将问题描述放在 hookSpecificOutput.additionalContext 中。"+"如果没有严重问题,返回 approve。",model:void 0,timeout:15}]}],PostToolUseFailure:[],PermissionRequest:[],UserPromptSubmit:[],SessionStart:[],SessionEnd:[],Stop:[],SubagentStop:[],Notification:[],Compaction:[]};function f(Q,Y){return{...Q,...Y,PreToolUse:Y.PreToolUse??Q.PreToolUse,PostToolUse:Y.PostToolUse??Q.PostToolUse,PostToolUseFailure:Y.PostToolUseFailure??Q.PostToolUseFailure,PermissionRequest:Y.PermissionRequest??Q.PermissionRequest,UserPromptSubmit:Y.UserPromptSubmit??Q.UserPromptSubmit,SessionStart:Y.SessionStart??Q.SessionStart,SessionEnd:Y.SessionEnd??Q.SessionEnd,Stop:Y.Stop??Q.Stop,SubagentStop:Y.SubagentStop??Q.SubagentStop,Notification:Y.Notification??Q.Notification,Compaction:Y.Compaction??Q.Compaction}}function x(){let Q={};if(process.env.BLADE_DISABLE_HOOKS==="true")Q.enabled=!1;if(process.env.BLADE_HOOK_TIMEOUT){let Y=parseInt(process.env.BLADE_HOOK_TIMEOUT,10);if(!isNaN(Y)&&Y>0)Q.defaultTimeout=Y}return Q}class z{executedHooks=new Map;canExecute(Q,Y){if(!this.executedHooks.has(Q))this.executedHooks.set(Q,new Set);if(this.executedHooks.get(Q).has(Y))return console.warn(`[HookGuard] Hook ${Y} for tool ${Q} already executed, skipping`),!1;return!0}markExecuted(Q,Y){let Z=this.executedHooks.get(Q);if(Z)Z.add(Y)}cleanup(Q){this.executedHooks.delete(Q)}cleanupAll(){this.executedHooks.clear()}}import{z as q}from"zod";var j=q.object({hook_event_name:q.nativeEnum(N),hook_execution_id:q.string(),timestamp:q.string(),project_dir:q.string(),session_id:q.string(),permission_mode:q.enum(["default","autoEdit","yolo","plan"]),_metadata:q.object({blade_version:q.string(),hook_timeout_ms:q.number()}).optional()}),o=j.extend({hook_event_name:q.literal("PreToolUse"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown())}),e=j.extend({hook_event_name:q.literal("PostToolUse"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown()),tool_response:q.unknown()}),t=j.extend({hook_event_name:q.literal("Stop"),reason:q.string().optional()}),QQ=j.extend({hook_event_name:q.literal("PostToolUseFailure"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown()),error:q.string(),error_type:q.string().optional(),is_interrupt:q.boolean(),is_timeout:q.boolean()}),YQ=j.extend({hook_event_name:q.literal("PermissionRequest"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown())}),ZQ=j.extend({hook_event_name:q.literal("UserPromptSubmit"),user_prompt:q.string(),has_images:q.boolean(),image_count:q.number()}),WQ=j.extend({hook_event_name:q.literal("SessionStart"),is_resume:q.boolean(),resume_session_id:q.string().optional()}),$Q=j.extend({hook_event_name:q.literal("SessionEnd"),reason:q.enum(["user_exit","error","max_turns","idle_timeout","ctrl_c","clear","logout","other"])}),qQ=j.extend({hook_event_name:q.literal("SubagentStop"),agent_type:q.string(),task_description:q.string().optional(),success:q.boolean(),result_summary:q.string().optional(),error:q.string().optional()}),XQ=j.extend({hook_event_name:q.literal("Notification"),notification_type:q.enum(["permission_prompt","idle_prompt","auth_success","elicitation_dialog","info","warning","error"]),title:q.string().optional(),message:q.string()}),JQ=j.extend({hook_event_name:q.literal("Compaction"),trigger:q.enum(["manual","auto"]),messages_before:q.number(),tokens_before:q.number()}),dQ=q.discriminatedUnion("hook_event_name",[o,e,t,QQ,YQ,ZQ,WQ,$Q,qQ,XQ,JQ]),GQ=q.object({hookEventName:q.literal("PreToolUse"),permissionDecision:q.nativeEnum(S).optional(),permissionDecisionReason:q.string().optional(),updatedInput:q.record(q.unknown()).optional()}),KQ=q.object({hookEventName:q.literal("PostToolUse"),additionalContext:q.string().optional(),updatedOutput:q.unknown().optional()}),VQ=q.object({hookEventName:q.literal("Stop"),continue:q.boolean().optional(),continueReason:q.string().optional()}),FQ=q.object({hookEventName:q.literal("SubagentStop"),continue:q.boolean().optional(),continueReason:q.string().optional(),additionalContext:q.string().optional()}),RQ=q.object({hookEventName:q.literal("PermissionRequest"),permissionDecision:q.enum(["approve","deny","ask"]).optional(),permissionDecisionReason:q.string().optional()}),DQ=q.object({hookEventName:q.literal("UserPromptSubmit"),updatedPrompt:q.string().optional(),contextInjection:q.string().optional()}),LQ=q.object({hookEventName:q.literal("SessionStart"),env:q.record(q.string()).optional()}),_Q=q.object({hookEventName:q.literal("Compaction"),blockCompaction:q.boolean().optional(),blockReason:q.string().optional()}),jQ=q.object({decision:q.object({behavior:q.nativeEnum(C)}).optional(),systemMessage:q.string().optional(),hookSpecificOutput:q.discriminatedUnion("hookEventName",[GQ,KQ,VQ,FQ,RQ,DQ,LQ,_Q]).optional(),suppressOutput:q.boolean().optional()}),AQ=q.object({type:q.literal("command"),command:q.string(),timeout:q.number().positive().optional(),statusMessage:q.string().optional()}),BQ=q.object({type:q.literal("prompt"),prompt:q.string(),model:q.string().optional(),timeout:q.number().positive().optional()}),OQ=q.discriminatedUnion("type",[AQ,BQ]),H=q.union([q.string(),q.array(q.string())]),UQ=q.object({tools:H.optional(),paths:H.optional(),commands:H.optional()}),_=q.object({name:q.string().optional(),matcher:UQ.optional(),hooks:q.array(OQ)}),mQ=q.object({enabled:q.boolean().optional(),defaultTimeout:q.number().positive().optional(),timeoutBehavior:q.enum(["ignore","deny","ask"]).optional(),failureBehavior:q.enum(["ignore","deny","ask"]).optional(),maxConcurrentHooks:q.number().positive().optional(),PreToolUse:q.array(_).optional(),PostToolUse:q.array(_).optional(),PostToolUseFailure:q.array(_).optional(),PermissionRequest:q.array(_).optional(),UserPromptSubmit:q.array(_).optional(),SessionStart:q.array(_).optional(),SessionEnd:q.array(_).optional(),Stop:q.array(_).optional(),SubagentStop:q.array(_).optional(),Notification:q.array(_).optional(),Compaction:q.array(_).optional()});function p(Q){let Y=jQ.safeParse(Q);if(Y.success)return{success:!0,data:Y.data};return{success:!1,error:Y.error}}class v{parse(Q,Y,Z){if(Q.timedOut){let $=Z?.timeoutBehavior||"ignore",X="Hook timeout";if($==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let W=this.tryParseJSON(Q.stdout);if(W){let $=p(W);if(!$.success){let G="error"in $?$.error.message:"Unknown validation error",J=Z?.failureBehavior||"ignore",K=`Invalid hook output JSON: ${G}`;if(J==="deny")return{success:!1,blocking:!0,error:K,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if(J==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${K}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:K,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let X=$.data;if(X.decision?.behavior==="block")return{success:!1,blocking:!0,error:X.systemMessage||"Hook blocked execution",output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};if(X.decision?.behavior==="async")return{success:!0,output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};return{success:!0,output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}return this.parseByExitCode(Q,Y,Z)}parseByExitCode(Q,Y,Z){let W=Q.exitCode;switch(W){case 0:return{success:!0,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 2:return{success:!1,blocking:!0,error:Q.stderr||Q.stdout||"Hook returned exit code 2",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 124:{let $=Z?.timeoutBehavior||"ignore",X="Hook timeout";if($==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}default:{let $=Z?.failureBehavior||"ignore",X=Q.stderr||Q.stdout||`Hook failed with exit code ${W}`;if($==="deny")return{success:!1,blocking:!0,error:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${X}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}}}tryParseJSON(Q){try{let Y=Q.trim();if(!Y)return null;let Z=Y.match(/^\s*(\{[\s\S]*\})\s*$/);if(!Z)return null;return JSON.parse(Z[1])}catch{return null}}}import{spawn as l}from"node:child_process";var PQ=500;function TQ(Q){return new Promise((Y)=>setTimeout(Y,Q))}function u(Q){return Q&&typeof Q==="object"&&"code"in Q?String(Q.code):void 0}function NQ(Q,Y){return new Promise((Z)=>{let W=["/PID",String(Q),"/T"];if(Y)W.push("/F");let $=l("taskkill",W,{stdio:"ignore",windowsHide:!0}),X=!1,G=(J)=>{if(X)return;X=!0,Z(J)};$.once("error",()=>G(!1)),$.once("close",(J)=>G(J===0))})}class a{child;state="owned";terminationPromise;terminationResult;platform;gracePeriodMs;killProcess;taskkill;wait;constructor(Q,Y={}){this.child=Q;this.platform=Y.platform??process.platform,this.gracePeriodMs=Y.gracePeriodMs??PQ,this.killProcess=Y.killProcess??process.kill.bind(process),this.taskkill=Y.taskkill??NQ,this.wait=Y.wait??TQ;let Z=()=>{if(this.state==="owned")this.state="released"};Q.once("close",Z),Q.once("error",Z)}release(){if(this.state==="owned")this.state="released"}terminate(){if(this.terminationPromise)return this.terminationPromise;if(this.terminationResult)return Promise.resolve(this.terminationResult);if(this.state==="released")return this.terminationResult={success:!0,alreadyExited:!0,forced:!1},Promise.resolve(this.terminationResult);return this.state="terminating",this.terminationPromise=this.terminateOwnedTree().then((Q)=>{return this.state="released",this.terminationResult=Q,Q}),this.terminationPromise}async terminateOwnedTree(){let Q=this.child.pid;if(!Q)return{success:!1,alreadyExited:!0,forced:!1};if(this.platform==="win32"){let W=await this.taskkill(Q,!1);await this.wait(this.gracePeriodMs);let $=await this.taskkill(Q,!0);if(!W&&!$)return{success:this.signalDirectChild("SIGKILL"),alreadyExited:!1,forced:!0};return{success:!0,alreadyExited:!1,forced:!0}}let Y=this.signalPosixTree(Q,"SIGTERM");await this.wait(this.gracePeriodMs);let Z=this.signalPosixTree(Q,"SIGKILL");return{success:Y||Z,alreadyExited:!1,forced:Z}}signalPosixTree(Q,Y){if(Q<=1||Q===process.pid)return this.signalDirectChild(Y);try{return this.killProcess(-Q,Y)}catch(Z){if(u(Z)==="ESRCH")return!0;return this.signalDirectChild(Y)}}signalDirectChild(Q){try{return this.child.kill(Q)}catch(Y){return u(Y)==="ESRCH"}}}function i(Q,Y,Z={}){let W=l(Q,Y,{...Z,detached:process.platform==="win32"?Z.detached:!0,windowsHide:Z.windowsHide??!0});return{child:W,processTree:new a(W)}}class g{content="";maxSize;constructor(Q){this.maxSize=Q}append(Q){if(this.content.length<this.maxSize){let Y=this.maxSize-this.content.length;this.content+=Q.substring(0,Y)}}getContent(){return this.content}isFull(){return this.content.length>=this.maxSize}}class d{MAX_STDOUT_SIZE=1048576;MAX_STDERR_SIZE=1048576;MAX_INPUT_SIZE=102400;async execute(Q,Y,Z,W){let $=JSON.stringify(Y);if($.length>this.MAX_INPUT_SIZE)throw Error(`Hook input too large: ${$.length} bytes (max ${this.MAX_INPUT_SIZE})`);let X=this.createSafeEnv(Y),{child:G,processTree:J}=i(Q,[],{shell:!0,env:X,cwd:Z.projectDir}),K=new g(this.MAX_STDOUT_SIZE),V=new g(this.MAX_STDERR_SIZE);G.stdout?.setEncoding("utf8"),G.stderr?.setEncoding("utf8"),G.stdout?.on("data",(D)=>{K.append(D)}),G.stderr?.on("data",(D)=>{V.append(D)});try{G.stdin?.write($),G.stdin?.end()}catch(D){throw await J.terminate(),Error(`Failed to write hook input: ${D}`)}return new Promise((D,R)=>{let L=!1,F=!1,B,O=()=>{return B??=J.terminate(),B},U=null,y=()=>{if(U&&Z.abortSignal)Z.abortSignal.removeEventListener("abort",U),U=null},M=setTimeout(()=>{L=!0,O()},W);if(G.on("close",async(w)=>{if(F)return;if(F=!0,clearTimeout(M),y(),L)await O();D({stdout:K.getContent(),stderr:V.getContent(),exitCode:L?124:w??1,timedOut:L})}),G.on("error",(w)=>{if(F)return;F=!0,clearTimeout(M),y(),R(w)}),Z.abortSignal){if(U=()=>{if(F)return;F=!0,clearTimeout(M),y(),O().then(()=>{D({stdout:K.getContent(),stderr:"Hook cancelled by abort signal",exitCode:1,timedOut:!1})})},Z.abortSignal.addEventListener("abort",U),Z.abortSignal.aborted)U()}})}createSafeEnv(Q){return{BLADE_PROJECT_DIR:Q.project_dir,BLADE_SESSION_ID:Q.session_id,BLADE_HOOK_EVENT:Q.hook_event_name,BLADE_TOOL_NAME:"tool_name"in Q?Q.tool_name:"",BLADE_TOOL_USE_ID:"tool_use_id"in Q?Q.tool_use_id:"",PATH:process.env.PATH||"",HOME:process.env.HOME||"",USER:process.env.USER||"",SHELL:process.env.SHELL||"/bin/sh"}}}class P extends Error{constructor(Q){super(Q);this.name="HttpHookSecurityError"}}function c(Q,Y={}){let Z;try{Z=new URL(Q)}catch{throw new P(`Invalid URL: ${Q}`)}if(Z.protocol!=="https:"&&Z.protocol!=="http:")throw new P(`Only http/https supported, got: ${Z.protocol}`);let W=Z.hostname.toLowerCase();if(bQ(W,Y.allowedHosts))return;if(Z.protocol==="http:"&&!Y.allowHttp)throw new P(`HTTP (non-TLS) blocked for ${W}; use https:// or add to allowedHosts`);if(yQ(W)&&!Y.allowLoopback)throw new P(`Loopback address blocked: ${W}; enable policy.allowLoopback or add to allowedHosts`);if(MQ(W)&&!Y.allowPrivateRanges)throw new P(`Private/link-local address blocked: ${W}; enable policy.allowPrivateRanges or add to allowedHosts`)}function bQ(Q,Y){if(!Y||Y.length===0)return!1;return Y.some((Z)=>{let W=Z.toLowerCase();if(W===Q)return!0;if(W.startsWith("*.")){let $=W.slice(1);return Q.endsWith($)&&Q.length>$.length}return!1})}function yQ(Q){if(Q==="localhost")return!0;if(Q==="::1"||Q==="[::1]")return!0;if(/^127\.\d+\.\d+\.\d+$/.test(Q))return!0;return!1}function MQ(Q){let Y=Q.match(/^(\d+)\.(\d+)\.(\d+)\.(\d+)$/);if(!Y)return!1;let[Z,W]=Y.slice(1).map(Number);if(Z===10)return!0;if(Z===172&&W>=16&&W<=31)return!0;if(Z===192&&W===168)return!0;if(Z===169&&W===254)return!0;return!1}function s(Q){if(!Q)return{};let Y={};for(let[Z,W]of Object.entries(Q))Y[Z]=W.replace(/\$\{([A-Z_][A-Z0-9_]*)\}/gi,($,X)=>{return process.env[X]??`\${${X}}`});return Y}var wQ=r("Execution"),CQ={PreToolUse:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "...", "updatedInput": { ... } }',PostToolUse:'{ "additionalContext": "注入到工具结果的额外信息", "updatedOutput": "修改后的工具输出" }',Stop:'{ "continue": true, "continueReason": "继续执行的原因" } // continue: true 表示阻止停止',SubagentStop:'{ "continue": true, "continueReason": "...", "additionalContext": "..." }',PermissionRequest:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "..." }',UserPromptSubmit:'{ "updatedPrompt": "修改后的提示词", "contextInjection": "注入的上下文" }',Compaction:'{ "blockCompaction": true, "blockReason": "阻止压缩的原因" }'};class m{processExecutor=new d;outputParser=new v;chatServiceCache=new Map;async executePreToolHooks(Q,Y,Z){if(Q.length===0)return{decision:"allow"};let W="tool_input"in Y?Y.tool_input:{},$=[];for(let X of Q)try{let G={...Y,...W&&{tool_input:W}},J=await this.executeHook(X,G,Z);if(!J.success){if(J.blocking)return{decision:"deny",reason:J.error};if(J.needsConfirmation)return{decision:"ask",reason:J.warning||J.error};if(J.warning)$.push(J.warning);continue}let K=J.output?.hookSpecificOutput;if(K&&"permissionDecision"in K){switch(K.permissionDecision){case"deny":return{decision:"deny",reason:K.permissionDecisionReason};case"ask":return{decision:"ask",reason:K.permissionDecisionReason};case"allow":break}if("updatedInput"in K&&K.updatedInput)W={...W,...K.updatedInput}}}catch(G){let J=G instanceof Error?G.message:String(G);if($.push(`Hook failed: ${J}`),Z.config.failureBehavior==="deny")return{decision:"deny",reason:J};else if(Z.config.failureBehavior==="ask")return{decision:"ask",reason:`Hook failed: ${J}. Continue?`}}return{decision:"allow",modifiedInput:W,warning:$.length>0?$.join(`
1
+ import{wb as n}from"./chunk-cadngkwr.js";import{Nb as r}from"./chunk-4hk7wfat.js";import{Pb as b}from"./chunk-cfrbsnfb.js";import{nanoid as A}from"nanoid";var N;((R)=>{R.PreToolUse="PreToolUse";R.PostToolUse="PostToolUse";R.PostToolUseFailure="PostToolUseFailure";R.PermissionRequest="PermissionRequest";R.UserPromptSubmit="UserPromptSubmit";R.SessionStart="SessionStart";R.SessionEnd="SessionEnd";R.Stop="Stop";R.SubagentStop="SubagentStop";R.Notification="Notification";R.Compaction="Compaction"})(N||={});var C;((W)=>{W.Approve="approve";W.Block="block";W.Async="async"})(C||={});var S;((W)=>{W.Allow="allow";W.Deny="deny";W.Ask="ask"})(S||={});var I={enabled:!1,defaultTimeout:60,timeoutBehavior:"ignore",failureBehavior:"ignore",maxConcurrentHooks:5,httpPolicy:{},PreToolUse:[],PostToolUse:[{name:"builtin:code-review-sensor",matcher:{tools:"Edit|Write"},hooks:[{type:"prompt",prompt:"审查此代码变更,重点检查:"+"1) 安全漏洞(命令注入、XSS、SQL 注入、硬编码密钥/密码)"+"2) 明显的逻辑错误(无限循环、off-by-one、空引用)"+"3) 类型安全问题。"+"如果发现严重问题,将问题描述放在 hookSpecificOutput.additionalContext 中。"+"如果没有严重问题,返回 approve。",model:void 0,timeout:15}]}],PostToolUseFailure:[],PermissionRequest:[],UserPromptSubmit:[],SessionStart:[],SessionEnd:[],Stop:[],SubagentStop:[],Notification:[],Compaction:[]};function f(Q,Y){return{...Q,...Y,PreToolUse:Y.PreToolUse??Q.PreToolUse,PostToolUse:Y.PostToolUse??Q.PostToolUse,PostToolUseFailure:Y.PostToolUseFailure??Q.PostToolUseFailure,PermissionRequest:Y.PermissionRequest??Q.PermissionRequest,UserPromptSubmit:Y.UserPromptSubmit??Q.UserPromptSubmit,SessionStart:Y.SessionStart??Q.SessionStart,SessionEnd:Y.SessionEnd??Q.SessionEnd,Stop:Y.Stop??Q.Stop,SubagentStop:Y.SubagentStop??Q.SubagentStop,Notification:Y.Notification??Q.Notification,Compaction:Y.Compaction??Q.Compaction}}function x(){let Q={};if(process.env.BLADE_DISABLE_HOOKS==="true")Q.enabled=!1;if(process.env.BLADE_HOOK_TIMEOUT){let Y=parseInt(process.env.BLADE_HOOK_TIMEOUT,10);if(!isNaN(Y)&&Y>0)Q.defaultTimeout=Y}return Q}class z{executedHooks=new Map;canExecute(Q,Y){if(!this.executedHooks.has(Q))this.executedHooks.set(Q,new Set);if(this.executedHooks.get(Q).has(Y))return console.warn(`[HookGuard] Hook ${Y} for tool ${Q} already executed, skipping`),!1;return!0}markExecuted(Q,Y){let Z=this.executedHooks.get(Q);if(Z)Z.add(Y)}cleanup(Q){this.executedHooks.delete(Q)}cleanupAll(){this.executedHooks.clear()}}import{z as q}from"zod";var j=q.object({hook_event_name:q.nativeEnum(N),hook_execution_id:q.string(),timestamp:q.string(),project_dir:q.string(),session_id:q.string(),permission_mode:q.enum(["default","autoEdit","yolo","plan"]),_metadata:q.object({blade_version:q.string(),hook_timeout_ms:q.number()}).optional()}),o=j.extend({hook_event_name:q.literal("PreToolUse"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown())}),e=j.extend({hook_event_name:q.literal("PostToolUse"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown()),tool_response:q.unknown()}),t=j.extend({hook_event_name:q.literal("Stop"),reason:q.string().optional()}),QQ=j.extend({hook_event_name:q.literal("PostToolUseFailure"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown()),error:q.string(),error_type:q.string().optional(),is_interrupt:q.boolean(),is_timeout:q.boolean()}),YQ=j.extend({hook_event_name:q.literal("PermissionRequest"),tool_name:q.string(),tool_use_id:q.string(),tool_input:q.record(q.unknown())}),ZQ=j.extend({hook_event_name:q.literal("UserPromptSubmit"),user_prompt:q.string(),has_images:q.boolean(),image_count:q.number()}),WQ=j.extend({hook_event_name:q.literal("SessionStart"),is_resume:q.boolean(),resume_session_id:q.string().optional()}),$Q=j.extend({hook_event_name:q.literal("SessionEnd"),reason:q.enum(["user_exit","error","max_turns","idle_timeout","ctrl_c","clear","logout","other"])}),qQ=j.extend({hook_event_name:q.literal("SubagentStop"),agent_type:q.string(),task_description:q.string().optional(),success:q.boolean(),result_summary:q.string().optional(),error:q.string().optional()}),XQ=j.extend({hook_event_name:q.literal("Notification"),notification_type:q.enum(["permission_prompt","idle_prompt","auth_success","elicitation_dialog","info","warning","error"]),title:q.string().optional(),message:q.string()}),JQ=j.extend({hook_event_name:q.literal("Compaction"),trigger:q.enum(["manual","auto"]),messages_before:q.number(),tokens_before:q.number()}),dQ=q.discriminatedUnion("hook_event_name",[o,e,t,QQ,YQ,ZQ,WQ,$Q,qQ,XQ,JQ]),GQ=q.object({hookEventName:q.literal("PreToolUse"),permissionDecision:q.nativeEnum(S).optional(),permissionDecisionReason:q.string().optional(),updatedInput:q.record(q.unknown()).optional()}),KQ=q.object({hookEventName:q.literal("PostToolUse"),additionalContext:q.string().optional(),updatedOutput:q.unknown().optional()}),VQ=q.object({hookEventName:q.literal("Stop"),continue:q.boolean().optional(),continueReason:q.string().optional()}),FQ=q.object({hookEventName:q.literal("SubagentStop"),continue:q.boolean().optional(),continueReason:q.string().optional(),additionalContext:q.string().optional()}),RQ=q.object({hookEventName:q.literal("PermissionRequest"),permissionDecision:q.enum(["approve","deny","ask"]).optional(),permissionDecisionReason:q.string().optional()}),DQ=q.object({hookEventName:q.literal("UserPromptSubmit"),updatedPrompt:q.string().optional(),contextInjection:q.string().optional()}),LQ=q.object({hookEventName:q.literal("SessionStart"),env:q.record(q.string()).optional()}),_Q=q.object({hookEventName:q.literal("Compaction"),blockCompaction:q.boolean().optional(),blockReason:q.string().optional()}),jQ=q.object({decision:q.object({behavior:q.nativeEnum(C)}).optional(),systemMessage:q.string().optional(),hookSpecificOutput:q.discriminatedUnion("hookEventName",[GQ,KQ,VQ,FQ,RQ,DQ,LQ,_Q]).optional(),suppressOutput:q.boolean().optional()}),AQ=q.object({type:q.literal("command"),command:q.string(),timeout:q.number().positive().optional(),statusMessage:q.string().optional()}),BQ=q.object({type:q.literal("prompt"),prompt:q.string(),model:q.string().optional(),timeout:q.number().positive().optional()}),OQ=q.discriminatedUnion("type",[AQ,BQ]),H=q.union([q.string(),q.array(q.string())]),UQ=q.object({tools:H.optional(),paths:H.optional(),commands:H.optional()}),_=q.object({name:q.string().optional(),matcher:UQ.optional(),hooks:q.array(OQ)}),mQ=q.object({enabled:q.boolean().optional(),defaultTimeout:q.number().positive().optional(),timeoutBehavior:q.enum(["ignore","deny","ask"]).optional(),failureBehavior:q.enum(["ignore","deny","ask"]).optional(),maxConcurrentHooks:q.number().positive().optional(),PreToolUse:q.array(_).optional(),PostToolUse:q.array(_).optional(),PostToolUseFailure:q.array(_).optional(),PermissionRequest:q.array(_).optional(),UserPromptSubmit:q.array(_).optional(),SessionStart:q.array(_).optional(),SessionEnd:q.array(_).optional(),Stop:q.array(_).optional(),SubagentStop:q.array(_).optional(),Notification:q.array(_).optional(),Compaction:q.array(_).optional()});function p(Q){let Y=jQ.safeParse(Q);if(Y.success)return{success:!0,data:Y.data};return{success:!1,error:Y.error}}class v{parse(Q,Y,Z){if(Q.timedOut){let $=Z?.timeoutBehavior||"ignore",X="Hook timeout";if($==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let W=this.tryParseJSON(Q.stdout);if(W){let $=p(W);if(!$.success){let G="error"in $?$.error.message:"Unknown validation error",J=Z?.failureBehavior||"ignore",K=`Invalid hook output JSON: ${G}`;if(J==="deny")return{success:!1,blocking:!0,error:K,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if(J==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${K}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:K,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let X=$.data;if(X.decision?.behavior==="block")return{success:!1,blocking:!0,error:X.systemMessage||"Hook blocked execution",output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};if(X.decision?.behavior==="async")return{success:!0,output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};return{success:!0,output:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}return this.parseByExitCode(Q,Y,Z)}parseByExitCode(Q,Y,Z){let W=Q.exitCode;switch(W){case 0:return{success:!0,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 2:return{success:!1,blocking:!0,error:Q.stderr||Q.stdout||"Hook returned exit code 2",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 124:{let $=Z?.timeoutBehavior||"ignore",X="Hook timeout";if($==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}default:{let $=Z?.failureBehavior||"ignore",X=Q.stderr||Q.stdout||`Hook failed with exit code ${W}`;if($==="deny")return{success:!1,blocking:!0,error:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if($==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${X}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:X,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}}}tryParseJSON(Q){try{let Y=Q.trim();if(!Y)return null;let Z=Y.match(/^\s*(\{[\s\S]*\})\s*$/);if(!Z)return null;return JSON.parse(Z[1])}catch{return null}}}import{spawn as l}from"node:child_process";var PQ=500;function TQ(Q){return new Promise((Y)=>setTimeout(Y,Q))}function u(Q){return Q&&typeof Q==="object"&&"code"in Q?String(Q.code):void 0}function NQ(Q,Y){return new Promise((Z)=>{let W=["/PID",String(Q),"/T"];if(Y)W.push("/F");let $=l("taskkill",W,{stdio:"ignore",windowsHide:!0}),X=!1,G=(J)=>{if(X)return;X=!0,Z(J)};$.once("error",()=>G(!1)),$.once("close",(J)=>G(J===0))})}class a{child;state="owned";terminationPromise;terminationResult;platform;gracePeriodMs;killProcess;taskkill;wait;constructor(Q,Y={}){this.child=Q;this.platform=Y.platform??process.platform,this.gracePeriodMs=Y.gracePeriodMs??PQ,this.killProcess=Y.killProcess??process.kill.bind(process),this.taskkill=Y.taskkill??NQ,this.wait=Y.wait??TQ;let Z=()=>{if(this.state==="owned")this.state="released"};Q.once("close",Z),Q.once("error",Z)}release(){if(this.state==="owned")this.state="released"}terminate(){if(this.terminationPromise)return this.terminationPromise;if(this.terminationResult)return Promise.resolve(this.terminationResult);if(this.state==="released")return this.terminationResult={success:!0,alreadyExited:!0,forced:!1},Promise.resolve(this.terminationResult);return this.state="terminating",this.terminationPromise=this.terminateOwnedTree().then((Q)=>{return this.state="released",this.terminationResult=Q,Q}),this.terminationPromise}async terminateOwnedTree(){let Q=this.child.pid;if(!Q)return{success:!1,alreadyExited:!0,forced:!1};if(this.platform==="win32"){let W=await this.taskkill(Q,!1);await this.wait(this.gracePeriodMs);let $=await this.taskkill(Q,!0);if(!W&&!$)return{success:this.signalDirectChild("SIGKILL"),alreadyExited:!1,forced:!0};return{success:!0,alreadyExited:!1,forced:!0}}let Y=this.signalPosixTree(Q,"SIGTERM");await this.wait(this.gracePeriodMs);let Z=this.signalPosixTree(Q,"SIGKILL");return{success:Y||Z,alreadyExited:!1,forced:Z}}signalPosixTree(Q,Y){if(Q<=1||Q===process.pid)return this.signalDirectChild(Y);try{return this.killProcess(-Q,Y)}catch(Z){if(u(Z)==="ESRCH")return!0;return this.signalDirectChild(Y)}}signalDirectChild(Q){try{return this.child.kill(Q)}catch(Y){return u(Y)==="ESRCH"}}}function i(Q,Y,Z={}){let W=l(Q,Y,{...Z,detached:process.platform==="win32"?Z.detached:!0,windowsHide:Z.windowsHide??!0});return{child:W,processTree:new a(W)}}class g{content="";maxSize;constructor(Q){this.maxSize=Q}append(Q){if(this.content.length<this.maxSize){let Y=this.maxSize-this.content.length;this.content+=Q.substring(0,Y)}}getContent(){return this.content}isFull(){return this.content.length>=this.maxSize}}class d{MAX_STDOUT_SIZE=1048576;MAX_STDERR_SIZE=1048576;MAX_INPUT_SIZE=102400;async execute(Q,Y,Z,W){let $=JSON.stringify(Y);if($.length>this.MAX_INPUT_SIZE)throw Error(`Hook input too large: ${$.length} bytes (max ${this.MAX_INPUT_SIZE})`);let X=this.createSafeEnv(Y),{child:G,processTree:J}=i(Q,[],{shell:!0,env:X,cwd:Z.projectDir}),K=new g(this.MAX_STDOUT_SIZE),V=new g(this.MAX_STDERR_SIZE);G.stdout?.setEncoding("utf8"),G.stderr?.setEncoding("utf8"),G.stdout?.on("data",(D)=>{K.append(D)}),G.stderr?.on("data",(D)=>{V.append(D)});try{G.stdin?.write($),G.stdin?.end()}catch(D){throw await J.terminate(),Error(`Failed to write hook input: ${D}`)}return new Promise((D,R)=>{let L=!1,F=!1,B,O=()=>{return B??=J.terminate(),B},U=null,y=()=>{if(U&&Z.abortSignal)Z.abortSignal.removeEventListener("abort",U),U=null},M=setTimeout(()=>{L=!0,O()},W);if(G.on("close",async(w)=>{if(F)return;if(F=!0,clearTimeout(M),y(),L)await O();D({stdout:K.getContent(),stderr:V.getContent(),exitCode:L?124:w??1,timedOut:L})}),G.on("error",(w)=>{if(F)return;F=!0,clearTimeout(M),y(),R(w)}),Z.abortSignal){if(U=()=>{if(F)return;F=!0,clearTimeout(M),y(),O().then(()=>{D({stdout:K.getContent(),stderr:"Hook cancelled by abort signal",exitCode:1,timedOut:!1})})},Z.abortSignal.addEventListener("abort",U),Z.abortSignal.aborted)U()}})}createSafeEnv(Q){return{BLADE_PROJECT_DIR:Q.project_dir,BLADE_SESSION_ID:Q.session_id,BLADE_HOOK_EVENT:Q.hook_event_name,BLADE_TOOL_NAME:"tool_name"in Q?Q.tool_name:"",BLADE_TOOL_USE_ID:"tool_use_id"in Q?Q.tool_use_id:"",PATH:process.env.PATH||"",HOME:process.env.HOME||"",USER:process.env.USER||"",SHELL:process.env.SHELL||"/bin/sh"}}}class P extends Error{constructor(Q){super(Q);this.name="HttpHookSecurityError"}}function c(Q,Y={}){let Z;try{Z=new URL(Q)}catch{throw new P(`Invalid URL: ${Q}`)}if(Z.protocol!=="https:"&&Z.protocol!=="http:")throw new P(`Only http/https supported, got: ${Z.protocol}`);let W=Z.hostname.toLowerCase();if(bQ(W,Y.allowedHosts))return;if(Z.protocol==="http:"&&!Y.allowHttp)throw new P(`HTTP (non-TLS) blocked for ${W}; use https:// or add to allowedHosts`);if(yQ(W)&&!Y.allowLoopback)throw new P(`Loopback address blocked: ${W}; enable policy.allowLoopback or add to allowedHosts`);if(MQ(W)&&!Y.allowPrivateRanges)throw new P(`Private/link-local address blocked: ${W}; enable policy.allowPrivateRanges or add to allowedHosts`)}function bQ(Q,Y){if(!Y||Y.length===0)return!1;return Y.some((Z)=>{let W=Z.toLowerCase();if(W===Q)return!0;if(W.startsWith("*.")){let $=W.slice(1);return Q.endsWith($)&&Q.length>$.length}return!1})}function yQ(Q){if(Q==="localhost")return!0;if(Q==="::1"||Q==="[::1]")return!0;if(/^127\.\d+\.\d+\.\d+$/.test(Q))return!0;return!1}function MQ(Q){let Y=Q.match(/^(\d+)\.(\d+)\.(\d+)\.(\d+)$/);if(!Y)return!1;let[Z,W]=Y.slice(1).map(Number);if(Z===10)return!0;if(Z===172&&W>=16&&W<=31)return!0;if(Z===192&&W===168)return!0;if(Z===169&&W===254)return!0;return!1}function s(Q){if(!Q)return{};let Y={};for(let[Z,W]of Object.entries(Q))Y[Z]=W.replace(/\$\{([A-Z_][A-Z0-9_]*)\}/gi,($,X)=>{return process.env[X]??`\${${X}}`});return Y}var wQ=r("Execution"),CQ={PreToolUse:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "...", "updatedInput": { ... } }',PostToolUse:'{ "additionalContext": "注入到工具结果的额外信息", "updatedOutput": "修改后的工具输出" }',Stop:'{ "continue": true, "continueReason": "继续执行的原因" } // continue: true 表示阻止停止',SubagentStop:'{ "continue": true, "continueReason": "...", "additionalContext": "..." }',PermissionRequest:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "..." }',UserPromptSubmit:'{ "updatedPrompt": "修改后的提示词", "contextInjection": "注入的上下文" }',Compaction:'{ "blockCompaction": true, "blockReason": "阻止压缩的原因" }'};class m{processExecutor=new d;outputParser=new v;chatServiceCache=new Map;async executePreToolHooks(Q,Y,Z){if(Q.length===0)return{decision:"allow"};let W="tool_input"in Y?Y.tool_input:{},$=[];for(let X of Q)try{let G={...Y,...W&&{tool_input:W}},J=await this.executeHook(X,G,Z);if(!J.success){if(J.blocking)return{decision:"deny",reason:J.error};if(J.needsConfirmation)return{decision:"ask",reason:J.warning||J.error};if(J.warning)$.push(J.warning);continue}let K=J.output?.hookSpecificOutput;if(K&&"permissionDecision"in K){switch(K.permissionDecision){case"deny":return{decision:"deny",reason:K.permissionDecisionReason};case"ask":return{decision:"ask",reason:K.permissionDecisionReason};case"allow":break}if("updatedInput"in K&&K.updatedInput)W={...W,...K.updatedInput}}}catch(G){let J=G instanceof Error?G.message:String(G);if($.push(`Hook failed: ${J}`),Z.config.failureBehavior==="deny")return{decision:"deny",reason:J};else if(Z.config.failureBehavior==="ask")return{decision:"ask",reason:`Hook failed: ${J}. Continue?`}}return{decision:"allow",modifiedInput:W,warning:$.length>0?$.join(`
2
2
  `):void 0}}async executePostToolHooks(Q,Y,Z){if(Q.length===0)return{};let W=Z.config.maxConcurrentHooks||5,$=await this.executeHooksConcurrently(Q,Y,Z,W),X=[],G=void 0,J=[];for(let K of $){if(!K.success&&K.warning){J.push(K.warning);continue}let V=K.output?.hookSpecificOutput;if(V&&"additionalContext"in V){if(V.additionalContext)X.push(V.additionalContext);if("updatedOutput"in V&&V.updatedOutput!==void 0)G=V.updatedOutput}}return{additionalContext:X.length>0?X.join(`
3
3
 
4
4
  `):void 0,modifiedOutput:G,warning:J.length>0?J.join(`
@@ -21,7 +21,7 @@ import{Eb as n}from"./chunk-wre40mf1.js";import{Ob as r}from"./chunk-mt1c5da4.js
21
21
  `):void 0}}async executeNotificationHooks(Q,Y,Z){let W="message"in Y?Y.message:"";if(Q.length===0)return{suppress:!1,message:W};let $=[],X=!1,G=W;for(let J of Q)try{let K=await this.executeHook(J,Y,Z);if(!K.success){if(K.warning)$.push(K.warning);continue}if(K.output?.suppressOutput){X=!0;break}if(K.stdout&&K.stdout.trim())G=K.stdout.trim()}catch(K){let V=K instanceof Error?K.message:String(K);$.push(`Hook failed: ${V}`)}return{suppress:X,message:G,warning:$.length>0?$.join(`
22
22
  `):void 0}}async executeCompactionHooks(Q,Y,Z){if(Q.length===0)return{blockCompaction:!1};let W=[];for(let $ of Q)try{let X=await this.executeHook($,Y,Z);if(!X.success){if(X.warning)W.push(X.warning);continue}let G=X.output?.hookSpecificOutput;if(G&&"blockCompaction"in G&&G.blockCompaction)return{blockCompaction:!0,blockReason:G.blockReason,warning:W.length>0?W.join(`
23
23
  `):void 0}}catch(X){let G=X instanceof Error?X.message:String(X);W.push(`Hook failed: ${G}`)}return{blockCompaction:!1,warning:W.length>0?W.join(`
24
- `):void 0}}async executeHook(Q,Y,Z){if(Q.type==="command")return this.executeCommandHook(Q,Y,Z);if(Q.type==="prompt")return this.executePromptHook(Q,Y,Z);if(Q.type==="function")return this.executeFunctionHook(Q,Y,Z);if(Q.type==="http")return this.executeHttpHook(Q,Y,Z);throw Error(`Hook type ${Q.type} not supported`)}async executeCommandHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let $=await this.processExecutor.execute(Q.command,Y,Z,W);return this.outputParser.parse($,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){return{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}async executePromptHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let $=await this.getOrCreateChatService(Q.model),X="hook_event_name"in Y?String(Y.hook_event_name):"Unknown",G=this.buildPromptHookSystemMessage(Q,X),J=JSON.stringify(Y,null,2),K=new AbortController,V=setTimeout(()=>K.abort(),W),D;try{D=(await $.chat([{role:"system",content:G},{role:"user",content:J}],void 0,K.signal)).content}catch(L){if(K.signal.aborted||L instanceof Error&&L.name==="AbortError")return this.outputParser.parse({stdout:"",stderr:"",exitCode:0,timedOut:!0},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});throw L}finally{clearTimeout(V)}let R=this.extractJsonFromLLMResponse(D);return this.outputParser.parse({stdout:R,stderr:"",exitCode:0,timedOut:!1},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){return wQ.warn(`[PromptHook] 执行失败: ${$ instanceof Error?$.message:String($)}`),{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}async executeFunctionHook(Q,Y,Z){let W=(Q.timeout??10)*1000;try{let $=await this.runFunctionWithTimeout(Q.handler,Y,Z,W);if($===void 0||$===null)return{success:!0,hook:Q};let X={stdout:JSON.stringify($),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(X,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){if($.isTimeout){let X={stdout:"",stderr:`Function hook timeout after ${W}ms`,exitCode:124,timedOut:!0};return this.outputParser.parse(X,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}return{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}runFunctionWithTimeout(Q,Y,Z,W){return new Promise(($,X)=>{let G=!1,J=setTimeout(()=>{if(G)return;G=!0;let V=Object.assign(Error(`Function hook timed out after ${W}ms`),{isTimeout:!0});X(V)},W),K=()=>{if(G)return;G=!0,clearTimeout(J),X(Error("Function hook aborted"))};Z.abortSignal?.addEventListener("abort",K,{once:!0}),Promise.resolve().then(()=>Q(Y,Z)).then((V)=>{if(G)return;G=!0,clearTimeout(J),Z.abortSignal?.removeEventListener("abort",K),$(V)},(V)=>{if(G)return;G=!0,clearTimeout(J),Z.abortSignal?.removeEventListener("abort",K),X(V)})})}async executeHttpHook(Q,Y,Z){let W=(Q.timeout??10)*1000,$=Q.maxResponseBytes??262144,X=Math.max(1,(Q.retries??0)+1);try{c(Q.url,Z.config.httpPolicy)}catch(V){return{success:!1,blocking:!0,error:V instanceof Error?V.message:String(V),hook:Q}}let G={"Content-Type":"application/json",...s(Q.headers)},J=JSON.stringify(Y),K;for(let V=0;V<X;V++){if(V>0){let F=Math.min(2**V*200,5000);await new Promise((B)=>setTimeout(B,F))}let D=new AbortController,R=setTimeout(()=>D.abort(),W),L=()=>D.abort();Z.abortSignal?.addEventListener("abort",L,{once:!0});try{let F=await fetch(Q.url,{method:"POST",headers:G,body:J,redirect:"manual",signal:D.signal});if(F.status>=300&&F.status<400){K=Error(`HTTP hook received redirect (${F.status}); redirects are disabled`);continue}if(!F.ok){if(K=Error(`HTTP hook returned ${F.status} ${F.statusText}`),F.status>=400&&F.status<500)break;continue}let O={stdout:await SQ(F,$),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(O,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(F){let B=F.name==="AbortError"||F.code==="ABORT_ERR";if(B&&Z.abortSignal?.aborted){K=Error("HTTP hook aborted by caller");break}if(B){let O={stdout:"",stderr:`HTTP hook timeout after ${W}ms`,exitCode:124,timedOut:!0};if(V===X-1)return this.outputParser.parse(O,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});K=Error(`HTTP hook timeout attempt ${V+1}`);continue}K=F}finally{clearTimeout(R),Z.abortSignal?.removeEventListener("abort",L)}}return{success:!1,blocking:!1,error:K?.message??"HTTP hook failed",hook:Q}}async getOrCreateChatService(Q){let Y=Q||"__default__",Z=this.chatServiceCache.get(Y);if(Z)return Z;let{ensureStoreInitialized:W,getCurrentModel:$,getModelById:X}=await import("./chunk-fehae50v.js"),{createChatServiceAsync:G}=await import("./chunk-93xhspe5.js");await W();let J=Q?X(Q)||$():$();if(!J)throw Error("PromptHook: 无法获取模型配置。请确保至少配置了一个模型。");let K=await G({provider:J.provider,apiKey:J.apiKey,baseUrl:J.baseUrl,model:J.model,temperature:J.temperature,maxContextTokens:J.maxContextTokens,maxOutputTokens:J.maxOutputTokens});return this.chatServiceCache.set(Y,K),K}buildPromptHookSystemMessage(Q,Y){let Z=CQ[Y]||"{ ... 根据事件类型返回相应字段 }";return`你是一个代码质量评估器,作为 Hook 在 ${Y} 事件中被触发。
24
+ `):void 0}}async executeHook(Q,Y,Z){if(Q.type==="command")return this.executeCommandHook(Q,Y,Z);if(Q.type==="prompt")return this.executePromptHook(Q,Y,Z);if(Q.type==="function")return this.executeFunctionHook(Q,Y,Z);if(Q.type==="http")return this.executeHttpHook(Q,Y,Z);throw Error(`Hook type ${Q.type} not supported`)}async executeCommandHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let $=await this.processExecutor.execute(Q.command,Y,Z,W);return this.outputParser.parse($,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){return{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}async executePromptHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let $=await this.getOrCreateChatService(Q.model),X="hook_event_name"in Y?String(Y.hook_event_name):"Unknown",G=this.buildPromptHookSystemMessage(Q,X),J=JSON.stringify(Y,null,2),K=new AbortController,V=setTimeout(()=>K.abort(),W),D;try{D=(await $.chat([{role:"system",content:G},{role:"user",content:J}],void 0,K.signal)).content}catch(L){if(K.signal.aborted||L instanceof Error&&L.name==="AbortError")return this.outputParser.parse({stdout:"",stderr:"",exitCode:0,timedOut:!0},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});throw L}finally{clearTimeout(V)}let R=this.extractJsonFromLLMResponse(D);return this.outputParser.parse({stdout:R,stderr:"",exitCode:0,timedOut:!1},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){return wQ.warn(`[PromptHook] 执行失败: ${$ instanceof Error?$.message:String($)}`),{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}async executeFunctionHook(Q,Y,Z){let W=(Q.timeout??10)*1000;try{let $=await this.runFunctionWithTimeout(Q.handler,Y,Z,W);if($===void 0||$===null)return{success:!0,hook:Q};let X={stdout:JSON.stringify($),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(X,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch($){if($.isTimeout){let X={stdout:"",stderr:`Function hook timeout after ${W}ms`,exitCode:124,timedOut:!0};return this.outputParser.parse(X,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}return{success:!1,blocking:!1,error:$ instanceof Error?$.message:String($),hook:Q}}}runFunctionWithTimeout(Q,Y,Z,W){return new Promise(($,X)=>{let G=!1,J=setTimeout(()=>{if(G)return;G=!0;let V=Object.assign(Error(`Function hook timed out after ${W}ms`),{isTimeout:!0});X(V)},W),K=()=>{if(G)return;G=!0,clearTimeout(J),X(Error("Function hook aborted"))};Z.abortSignal?.addEventListener("abort",K,{once:!0}),Promise.resolve().then(()=>Q(Y,Z)).then((V)=>{if(G)return;G=!0,clearTimeout(J),Z.abortSignal?.removeEventListener("abort",K),$(V)},(V)=>{if(G)return;G=!0,clearTimeout(J),Z.abortSignal?.removeEventListener("abort",K),X(V)})})}async executeHttpHook(Q,Y,Z){let W=(Q.timeout??10)*1000,$=Q.maxResponseBytes??262144,X=Math.max(1,(Q.retries??0)+1);try{c(Q.url,Z.config.httpPolicy)}catch(V){return{success:!1,blocking:!0,error:V instanceof Error?V.message:String(V),hook:Q}}let G={"Content-Type":"application/json",...s(Q.headers)},J=JSON.stringify(Y),K;for(let V=0;V<X;V++){if(V>0){let F=Math.min(2**V*200,5000);await new Promise((B)=>setTimeout(B,F))}let D=new AbortController,R=setTimeout(()=>D.abort(),W),L=()=>D.abort();Z.abortSignal?.addEventListener("abort",L,{once:!0});try{let F=await fetch(Q.url,{method:"POST",headers:G,body:J,redirect:"manual",signal:D.signal});if(F.status>=300&&F.status<400){K=Error(`HTTP hook received redirect (${F.status}); redirects are disabled`);continue}if(!F.ok){if(K=Error(`HTTP hook returned ${F.status} ${F.statusText}`),F.status>=400&&F.status<500)break;continue}let O={stdout:await SQ(F,$),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(O,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(F){let B=F.name==="AbortError"||F.code==="ABORT_ERR";if(B&&Z.abortSignal?.aborted){K=Error("HTTP hook aborted by caller");break}if(B){let O={stdout:"",stderr:`HTTP hook timeout after ${W}ms`,exitCode:124,timedOut:!0};if(V===X-1)return this.outputParser.parse(O,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});K=Error(`HTTP hook timeout attempt ${V+1}`);continue}K=F}finally{clearTimeout(R),Z.abortSignal?.removeEventListener("abort",L)}}return{success:!1,blocking:!1,error:K?.message??"HTTP hook failed",hook:Q}}async getOrCreateChatService(Q){let Y=Q||"__default__",Z=this.chatServiceCache.get(Y);if(Z)return Z;let{ensureStoreInitialized:W,getCurrentModel:$,getModelById:X}=await import("./chunk-42q476bq.js"),{createChatServiceAsync:G}=await import("./chunk-hv0bzn31.js");await W();let J=Q?X(Q)||$():$();if(!J)throw Error("PromptHook: 无法获取模型配置。请确保至少配置了一个模型。");let K=await G({provider:J.provider,apiKey:J.apiKey,baseUrl:J.baseUrl,model:J.model,temperature:J.temperature,maxContextTokens:J.maxContextTokens,maxOutputTokens:J.maxOutputTokens});return this.chatServiceCache.set(Y,K),K}buildPromptHookSystemMessage(Q,Y){let Z=CQ[Y]||"{ ... 根据事件类型返回相应字段 }";return`你是一个代码质量评估器,作为 Hook 在 ${Y} 事件中被触发。
25
25
 
26
26
  `+`## 评估指令
27
27
  ${Q.prompt}
@@ -38,4 +38,4 @@ ${Q.prompt}
38
38
  `+`- 只输出 JSON,不要包含 markdown 代码块或其他文本
39
39
  `+`- 如果没有发现问题,使用 "approve"
40
40
  `+'- 只在发现严重问题时使用 "block"'}extractJsonFromLLMResponse(Q){let Y=Q.trim(),Z=Y.match(/^```(?:json)?\s*\n?([\s\S]*?)\n?\s*```$/);if(Z)return Z[1].trim();return Y}async executeHooksConcurrently(Q,Y,Z,W){let $=[],X=new Set;for(let G of Q){if(X.size>=W)await Promise.race(X);let J=this.executeHook(G,Y,Z).catch((V)=>({success:!1,blocking:!1,error:V instanceof Error?V.message:String(V),hook:G})),K=J.then(()=>{X.delete(K)}).catch(()=>{X.delete(K)});X.add(K),$.push(J)}return Promise.all($)}}async function SQ(Q,Y){let Z=Q.body?.getReader();if(!Z)return"";let W=new TextDecoder,$=0,X="";while(!0){let{value:G,done:J}=await Z.read();if(J)break;if($+=G.byteLength,$>Y){try{Z.cancel()}catch{}X+=W.decode(G.slice(0,Y-($-G.byteLength)));break}X+=W.decode(G,{stream:!0})}return X+=W.decode(),X}import k from"picomatch";var EQ=/^([A-Za-z0-9_|]+)\((.+)\)$/;class h{matches(Q,Y){if(!Q)return!0;if(Q.tools&&Y.toolName){if(!this.matchTools(Q.tools,Y))return!1}if(Q.paths&&Y.filePath){if(!this.matchPaths(Q.paths,Y.filePath))return!1}if(Q.commands&&Y.command){if(!this.matchCommands(Q.commands,Y.command))return!1}return!0}matchTools(Q,Y){if(Array.isArray(Q))return Q.some((Z)=>this.matchToolWithParams(Z,Y));return this.matchToolWithParams(Q,Y)}matchPaths(Q,Y){if(Array.isArray(Q))return Q.some((W)=>{return k(W)(Y)});return k(Q)(Y)}matchCommands(Q,Y){if(Array.isArray(Q))return Q.some((Z)=>this.matchPattern(Y,Z));return this.matchPattern(Y,Q)}matchToolWithParams(Q,Y){let{toolName:Z,command:W,filePath:$}=Y,X=EQ.exec(Q);if(!X)return this.matchPattern(Z,Q);let[,G,J]=X;if(!this.matchPattern(Z,G))return!1;let K=this.getArgValue(Z,W,$);if(!K)return!1;return this.matchGlobOrPattern(K,J)}getArgValue(Q,Y,Z){if(Q==="Bash"||Q==="BashTool")return Y;if(["Read","Edit","Write","Glob","Grep"].includes(Q))return Z;return Y||Z}matchGlobOrPattern(Q,Y){if(/[*?[\]{}!]/.test(Y))try{return k(Y,{bash:!0,dot:!0})(Q)}catch{}if(Y.endsWith("*")){let Z=Y.slice(0,-1);return Q.startsWith(Z)}return Q===Y}matchPattern(Q,Y){if(Y==="*")return!0;if(!Y.includes("|")&&!/[.*+?^${}()|[\]\\]/.test(Y))return Q===Y;if(Y.includes("|"))return Y.split("|").map((W)=>W.trim()).includes(Q);try{return new RegExp(Y).test(Q)}catch{return Q===Y}}}class T{static instance=null;config=I;executor=new m;guard=new z;matcher=new h;sessionDisabled=!1;constructor(){}static getInstance(){if(!T.instance)T.instance=new T;return T.instance}loadConfig(Q){let Y=f(I,Q),Z=x();Y=f(Y,Z),this.config=Y}isEnabled(){if(!this.config.enabled)return!1;if(this.sessionDisabled)return!1;return!0}disable(){this.sessionDisabled=!0,console.log("[HookManager] Hooks disabled for this session")}enable(){this.sessionDisabled=!1,console.log("[HookManager] Hooks enabled for this session")}getConfig(){return this.config}registerFunction(Q,Y,Z,W){let $={type:"function",handler:Z,timeout:W?.timeout},X={name:W?.name??`inline-${Q}-${Date.now()}`,matcher:Y,hooks:[$]},G=this.config[Q]??[];return this.config[Q]=[...G,X],()=>{let J=this.config[Q];if(!J)return;this.config[Q]=J.filter((K)=>K!==X)}}async reloadConfig(){let Q=await import("node:fs/promises"),Y=await import("node:path");try{let Z=Y.join(n(),".blade","settings.local.json"),W=await Q.readFile(Z,"utf-8"),$=JSON.parse(W);if($.hooks)this.loadConfig($.hooks)}catch{}}async executePreToolHooks(Q,Y,Z,W){if(!this.isEnabled())return{decision:"allow"};if(W.permissionMode==="plan")return{decision:"allow"};if(!this.guard.canExecute(Y,"PreToolUse"))return{decision:"allow"};let $={hook_event_name:"PreToolUse",hook_execution_id:A(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,project_dir:W.projectDir,session_id:W.sessionId,permission_mode:W.permissionMode},X=this.getMatchingHooks("PreToolUse",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(X.length===0)return{decision:"allow"};let G={projectDir:W.projectDir,sessionId:W.sessionId,permissionMode:W.permissionMode,config:this.config,abortSignal:W.abortSignal};try{let J=await this.executor.executePreToolHooks(X,$,G);if(this.guard.markExecuted(Y,"PreToolUse"),W.permissionMode==="yolo"){if(J.decision==="deny")return J;return{decision:"allow",modifiedInput:J.modifiedInput,warning:J.warning,reason:J.reason}}return J}catch(J){return console.error("[HookManager] Error executing PreToolUse hooks:",J),{decision:"allow",warning:`Hook execution failed: ${J instanceof Error?J.message:String(J)}`}}}async executePostToolHooks(Q,Y,Z,W,$){if(!this.isEnabled())return{};if($.permissionMode==="plan")return{};if(!this.guard.canExecute(Y,"PostToolUse"))return{};let X={hook_event_name:"PostToolUse",hook_execution_id:A(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,tool_response:W,project_dir:$.projectDir,session_id:$.sessionId,permission_mode:$.permissionMode},G=this.getMatchingHooks("PostToolUse",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(G.length===0)return{};let J={projectDir:$.projectDir,sessionId:$.sessionId,permissionMode:$.permissionMode,config:this.config,abortSignal:$.abortSignal};try{let K=await this.executor.executePostToolHooks(G,X,J);return this.guard.markExecuted(Y,"PostToolUse"),K}catch(K){return console.error("[HookManager] Error executing PostToolUse hooks:",K),{warning:`Hook execution failed: ${K instanceof Error?K.message:String(K)}`}}finally{this.guard.cleanup(Y)}}async executeStopHooks(Q){if(!this.isEnabled())return{shouldStop:!0};let Y={hook_event_name:"Stop",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Q.projectDir,session_id:Q.sessionId,permission_mode:Q.permissionMode,reason:Q.reason},Z=this.getMatchingHooks("Stop",{});if(Z.length===0)return{shouldStop:!0};let W={projectDir:Q.projectDir,sessionId:Q.sessionId,permissionMode:Q.permissionMode,config:this.config,abortSignal:Q.abortSignal};try{return await this.executor.executeStopHooks(Z,Y,W)}catch($){return console.error("[HookManager] Error executing Stop hooks:",$),{shouldStop:!0,warning:`Hook execution failed: ${$ instanceof Error?$.message:String($)}`}}}async executeSubagentStopHooks(Q,Y){if(!this.isEnabled())return{shouldStop:!0};let Z={hook_event_name:"SubagentStop",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,agent_type:Q,task_description:Y.taskDescription,success:Y.success,result_summary:Y.resultSummary,error:Y.error},W=this.getMatchingHooks("SubagentStop",{});if(W.length===0)return{shouldStop:!0};let $={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeSubagentStopHooks(W,Z,$)}catch(X){return console.error("[HookManager] Error executing SubagentStop hooks:",X),{shouldStop:!0,warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}async executePermissionRequestHooks(Q,Y,Z,W){if(!this.isEnabled())return{decision:"ask"};let $={hook_event_name:"PermissionRequest",hook_execution_id:A(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,project_dir:W.projectDir,session_id:W.sessionId,permission_mode:W.permissionMode},X=this.getMatchingHooks("PermissionRequest",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(X.length===0)return{decision:"ask"};let G={projectDir:W.projectDir,sessionId:W.sessionId,permissionMode:W.permissionMode,config:this.config,abortSignal:W.abortSignal};try{return await this.executor.executePermissionRequestHooks(X,$,G)}catch(J){return console.error("[HookManager] Error executing PermissionRequest hooks:",J),{decision:"ask",warning:`Hook execution failed: ${J instanceof Error?J.message:String(J)}`}}}async executeUserPromptSubmitHooks(Q,Y){if(!this.isEnabled())return{proceed:!0};let Z={hook_event_name:"UserPromptSubmit",hook_execution_id:A(),timestamp:new Date().toISOString(),user_prompt:Q,has_images:Y.hasImages,image_count:Y.imageCount,project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode},W=this.getMatchingHooks("UserPromptSubmit",{});if(W.length===0)return{proceed:!0};let $={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeUserPromptSubmitHooks(W,Z,$)}catch(X){return console.error("[HookManager] Error executing UserPromptSubmit hooks:",X),{proceed:!0,warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}async executeSessionStartHooks(Q){if(!this.isEnabled())return{proceed:!0};let Y={hook_event_name:"SessionStart",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Q.projectDir,session_id:Q.sessionId,permission_mode:Q.permissionMode,is_resume:Q.isResume,resume_session_id:Q.resumeSessionId},Z=this.getMatchingHooks("SessionStart",{});if(Z.length===0)return{proceed:!0};let W={projectDir:Q.projectDir,sessionId:Q.sessionId,permissionMode:Q.permissionMode,config:this.config,abortSignal:Q.abortSignal};try{return await this.executor.executeSessionStartHooks(Z,Y,W)}catch($){return console.error("[HookManager] Error executing SessionStart hooks:",$),{proceed:!0,warning:`Hook execution failed: ${$ instanceof Error?$.message:String($)}`}}}async executeSessionEndHooks(Q,Y){if(!this.isEnabled())return{};let Z={hook_event_name:"SessionEnd",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,reason:Q},W=this.getMatchingHooks("SessionEnd",{});if(W.length===0)return{};let $={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeSessionEndHooks(W,Z,$),{}}catch(X){return console.error("[HookManager] Error executing SessionEnd hooks:",X),{warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}async executePostToolUseFailureHooks(Q,Y,Z,W,$){if(!this.isEnabled())return{};let X={hook_event_name:"PostToolUseFailure",hook_execution_id:A(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,error:W,error_type:$.errorType,is_interrupt:$.isInterrupt,is_timeout:$.isTimeout,project_dir:$.projectDir,session_id:$.sessionId,permission_mode:$.permissionMode},G=this.getMatchingHooks("PostToolUseFailure",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(G.length===0)return{};let J={projectDir:$.projectDir,sessionId:$.sessionId,permissionMode:$.permissionMode,config:this.config,abortSignal:$.abortSignal};try{return await this.executor.executePostToolUseFailureHooks(G,X,J)}catch(K){return console.error("[HookManager] Error executing PostToolUseFailure hooks:",K),{warning:`Hook execution failed: ${K instanceof Error?K.message:String(K)}`}}}async executeNotificationHooks(Q,Y,Z){if(!this.isEnabled())return{suppress:!1,message:Y};let W={hook_event_name:"Notification",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Z.projectDir,session_id:Z.sessionId,permission_mode:Z.permissionMode,notification_type:Q,title:Z.title,message:Y},$=this.getMatchingHooks("Notification",{});if($.length===0)return{suppress:!1,message:Y};let X={projectDir:Z.projectDir,sessionId:Z.sessionId,permissionMode:Z.permissionMode,config:this.config,abortSignal:Z.abortSignal};try{return await this.executor.executeNotificationHooks($,W,X)}catch(G){return console.error("[HookManager] Error executing Notification hooks:",G),{suppress:!1,message:Y,warning:`Hook execution failed: ${G instanceof Error?G.message:String(G)}`}}}async executeCompactionHooks(Q,Y){if(!this.isEnabled())return{blockCompaction:!1};let Z={hook_event_name:"Compaction",hook_execution_id:A(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,trigger:Q,messages_before:Y.messagesBefore,tokens_before:Y.tokensBefore},W=this.getMatchingHooks("Compaction",{});if(W.length===0)return{blockCompaction:!1};let $={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeCompactionHooks(W,Z,$)}catch(X){return console.error("[HookManager] Error executing Compaction hooks:",X),{blockCompaction:!1,warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}getMatchingHooks(Q,Y){let Z=this.config[Q]||[],W=[];for(let $ of Z)if(this.matcher.matches($.matcher,Y))W.push(...$.hooks);return W}extractFilePath(Q){let Y=["file_path","path","filePath","source","target"];for(let Z of Y){let W=Q[Z];if(typeof W==="string")return W}return}extractCommand(Q,Y){if(Q==="Bash"||Q==="BashTool"){let Z=Y.command;if(typeof Z==="string")return Z}return}cleanup(){this.guard.cleanupAll()}}
41
- export{N as pb,i as qb,T as rb};
41
+ export{N as ob,i as pb,T as qb};
@@ -0,0 +1,6 @@
1
+ import{p as A,q as E,r as N,s as f}from"./chunk-67zx7ymj.js";import{t as D}from"./chunk-qw2wy4jc.js";import{v as L,w as _,x}from"./chunk-m2k4h3bs.js";import{y as B}from"./chunk-cqrkeq20.js";import{K as $,Q as b}from"./chunk-t6y57501.js";import"./chunk-z0qxaky4.js";import"./chunk-t3pyakvc.js";import"./chunk-22gxb9qs.js";import"./chunk-wc2xb6ge.js";import"./chunk-b6mxeey0.js";import"./chunk-k18bq4mn.js";import{Ya as Z}from"./chunk-zcreym5g.js";import"./chunk-wcm0nc84.js";import"./chunk-p6d7k1x6.js";import{vb as Y,wb as T}from"./chunk-cadngkwr.js";import"./chunk-m4a480k7.js";import"./chunk-4hk7wfat.js";import"./chunk-cfrbsnfb.js";import h from"yargs";import{hideBin as R}from"yargs/helpers";var w=new Set(Object.values(Y));function X(k){if(k&&w.has(k))return k;return}function M(k){return k.command("* [message]","Print response and exit (useful for pipes)",(F)=>F.positional("message",{describe:"Message to process",type:"string"}).option("p",{alias:"print",describe:"Print response and exit (useful for pipes)",type:"boolean"}).option("output-format",{describe:'Output format: "text", "json", "stream-json"',type:"string",default:"text"}).option("include-partial-messages",{describe:"Include partial message chunks as they arrive",type:"boolean"}).option("input-format",{describe:'Input format: "text", "stream-json"',type:"string",default:"text"}).option("model",{describe:"Model for the current session",type:"string"}).option("append-system-prompt",{describe:"Append a system prompt to the default system prompt",type:"string"}).option("system-prompt",{describe:"Replace the default system prompt",type:"string"}).option("max-turns",{alias:["maxTurns"],describe:"Maximum conversation turns (-1: unlimited, N>0: limit to N turns)",type:"number"}),async(F)=>{if(!F.p)return;let q=await O(F);process.exit(q)})}async function O(k,F=process){let q;try{await A();let J=await E({message:k.message,_:k._,defaultMessage:"Hello"}),G=await N(J);if(G.mode==="output"){if(G.content)F.stdout.write(`${G.content}
2
+ `);return G.exitCode??0}let Q=G.content,{sessionId:K,messages:U}=await f({sessionId:k.sessionId,continue:k.continue,resume:k.resume,forkSession:k.forkSession,fallbackSessionPrefix:"print"});q=await $.create({sessionId:K,workspaceRoot:T(),modelId:k.model,mcpConfig:k.mcpConfig,strictMcpConfig:k.strictMcpConfig,agents:k.agents?B(k.agents):void 0});let V=await b.createWithRuntime(q,{sessionId:K,systemPrompt:k.systemPrompt,appendSystemPrompt:k.appendSystemPrompt,maxTurns:k.maxTurns,modelId:k.model,permissionMode:X(k.permissionMode),toolWhitelist:k.allowedTools,toolBlacklist:k.disallowedTools,mcpConfig:k.mcpConfig,strictMcpConfig:k.strictMcpConfig}),H=(await Z(V.chatStream(Q,{messages:[...U],userId:"cli-user",sessionId:K,workspaceRoot:T(),permissionMode:X(k.permissionMode)}))).finalMessage||"";if(k.outputFormat==="json")F.stdout.write(`${JSON.stringify({response:H,input:Q,model:k.model,timestamp:new Date().toISOString()},null,2)}
3
+ `);else if(k.outputFormat==="stream-json")F.stdout.write(`${JSON.stringify({type:"response",content:H})}
4
+ `);else F.stdout.write(`${H}
5
+ `);return 0}catch(J){return F.stderr.write(`Error: ${J instanceof Error?J.message:"未知错误"}
6
+ `),1}finally{await q?.dispose()}}async function d(){if(process.argv.slice(2).findIndex((q)=>q==="--print"||q==="-p")===-1)return!1;try{let{print:q,"output-format":J,"include-partial-messages":G,"input-format":Q,"system-prompt":K,"append-system-prompt":U,"max-turns":V,...W}=D,H=h(R(process.argv)).scriptName("blade").strict(!1).options(W).middleware([_,L,x]);return M(H),await H.parse(),!0}catch(q){console.error(`Print mode error: ${q instanceof Error?q.message:"未知错误"}`),process.exit(1)}}export{O as runPrint,d as handlePrintMode};
@@ -1,2 +1,2 @@
1
- import{da as d,fa as h}from"./chunk-31chcjw6.js";var q={debug:{alias:"d",type:"string",describe:'Enable debug mode with optional category filtering (e.g., "agent,ui" or "!chat,!loop")',group:"Debug Options:"},model:{type:"string",describe:"Override the current run model using a model config ID (TUI / print / headless)",group:"AI Options:"},print:{alias:"p",type:"boolean",describe:"Print response and exit (useful for pipes)",group:"Output Options:"},headless:{type:"boolean",describe:"Run full agent loop without Ink UI and print all events to the terminal",group:"Output Options:"},"output-format":{alias:["outputFormat"],type:"string",choices:["text","json","stream-json","jsonl"],default:"text",describe:"Output format: --print supports text|json|stream-json; --headless supports text|jsonl",group:"Output Options:"},"include-partial-messages":{alias:["includePartialMessages"],type:"boolean",describe:"Include partial message chunks as they arrive",group:"Output Options:"},"input-format":{alias:["inputFormat"],type:"string",choices:["text","stream-json"],default:"text",describe:"Input format",group:"Input Options:"},"replay-user-messages":{alias:["replayUserMessages"],type:"boolean",describe:"Re-emit user messages from stdin",group:"Input Options:"},"allowed-tools":{alias:["allowedTools"],type:"array",string:!0,describe:"Comma or space-separated list of tool names to allow",group:"Security Options:"},"disallowed-tools":{alias:["disallowedTools"],type:"array",string:!0,describe:"Comma or space-separated list of tool names to deny",group:"Security Options:"},"mcp-config":{alias:["mcpConfig"],type:"array",string:!0,describe:"Load MCP servers from JSON files or strings",group:"MCP Options:"},"system-prompt":{alias:["systemPrompt"],type:"string",describe:"System prompt to use for the session (replaces default)",group:"AI Options:"},"append-system-prompt":{alias:["appendSystemPrompt"],type:"string",describe:"Append a system prompt to the default system prompt",group:"AI Options:"},"max-turns":{alias:["maxTurns"],type:"number",describe:"Maximum conversation turns (-1: unlimited, 0: disable chat, N>0: limit to N turns)",group:"AI Options:",default:void 0},"permission-mode":{alias:["permissionMode"],type:"string",choices:["default","autoEdit","yolo","plan"],describe:"Permission mode (default: ask for non-read tools, autoEdit: auto-approve edits, yolo: auto-approve all, plan: reserved)",group:"Security Options:"},yolo:{type:"boolean",describe:"Auto-approve all tools (shortcut for --permission-mode=yolo)",group:"Security Options:"},continue:{alias:"c",type:"boolean",describe:"Continue the most recent conversation",group:"Session Options:"},resume:{alias:"r",describe:"Resume a conversation - provide a session ID or interactively select a conversation to resume",group:"Session Options:",coerce:(b)=>{if(b===void 0||b===!0||b==="")return"true";return String(b)}},"fork-session":{alias:["forkSession"],type:"boolean",describe:"Create a new session ID when resuming",group:"Session Options:"},settings:{type:"string",describe:"Path to a settings JSON file or JSON string",group:"Configuration:"},"add-dir":{alias:["addDir"],type:"array",string:!0,describe:"Additional directories to allow tool access to",group:"Security Options:"},ide:{type:"boolean",describe:"Automatically connect to IDE on startup",group:"Integration:"},acp:{type:"boolean",describe:"Run in ACP (Agent Client Protocol) mode for IDE integration",group:"Integration:"},"strict-mcp-config":{alias:["strictMcpConfig"],type:"boolean",describe:"Only use MCP servers from --mcp-config",group:"MCP Options:"},"session-id":{alias:["sessionId"],type:"string",describe:"Use a specific session ID for the conversation",group:"Session Options:"},agents:{type:"string",describe:"JSON object defining custom agents",group:"AI Options:"},"setting-sources":{alias:["settingSources"],type:"string",describe:"Comma-separated list of setting sources to load",group:"Configuration:"},"plugin-dir":{alias:["pluginDir"],type:"array",string:!0,describe:"Load plugins from specified directories",group:"Plugin Options:"}},w={scriptName:"blade",usage:"$0 [command] [options]",description:h(),version:d(),locale:"en",showHelpOnFail:!0,demandCommand:!1,recommendCommands:!0,strict:!1};
1
+ import{Ha as d,Ja as h}from"./chunk-k18bq4mn.js";var q={debug:{alias:"d",type:"string",describe:'Enable debug mode with optional category filtering (e.g., "agent,ui" or "!chat,!loop")',group:"Debug Options:"},model:{type:"string",describe:"Override the current run model using a model config ID (TUI / print / headless)",group:"AI Options:"},print:{alias:"p",type:"boolean",describe:"Print response and exit (useful for pipes)",group:"Output Options:"},headless:{type:"boolean",describe:"Run full agent loop without Ink UI and print all events to the terminal",group:"Output Options:"},"output-format":{alias:["outputFormat"],type:"string",choices:["text","json","stream-json","jsonl"],default:"text",describe:"Output format: --print supports text|json|stream-json; --headless supports text|jsonl",group:"Output Options:"},"include-partial-messages":{alias:["includePartialMessages"],type:"boolean",describe:"Include partial message chunks as they arrive",group:"Output Options:"},"input-format":{alias:["inputFormat"],type:"string",choices:["text","stream-json"],default:"text",describe:"Input format",group:"Input Options:"},"replay-user-messages":{alias:["replayUserMessages"],type:"boolean",describe:"Re-emit user messages from stdin",group:"Input Options:"},"allowed-tools":{alias:["allowedTools"],type:"array",string:!0,describe:"Comma or space-separated list of tool names to allow",group:"Security Options:"},"disallowed-tools":{alias:["disallowedTools"],type:"array",string:!0,describe:"Comma or space-separated list of tool names to deny",group:"Security Options:"},"mcp-config":{alias:["mcpConfig"],type:"array",string:!0,describe:"Load MCP servers from JSON files or strings",group:"MCP Options:"},"system-prompt":{alias:["systemPrompt"],type:"string",describe:"System prompt to use for the session (replaces default)",group:"AI Options:"},"append-system-prompt":{alias:["appendSystemPrompt"],type:"string",describe:"Append a system prompt to the default system prompt",group:"AI Options:"},"max-turns":{alias:["maxTurns"],type:"number",describe:"Maximum conversation turns (-1: unlimited, 0: disable chat, N>0: limit to N turns)",group:"AI Options:",default:void 0},"permission-mode":{alias:["permissionMode"],type:"string",choices:["default","autoEdit","yolo","plan"],describe:"Permission mode (default: ask for non-read tools, autoEdit: auto-approve edits, yolo: auto-approve all, plan: reserved)",group:"Security Options:"},yolo:{type:"boolean",describe:"Auto-approve all tools (shortcut for --permission-mode=yolo)",group:"Security Options:"},continue:{alias:"c",type:"boolean",describe:"Continue the most recent conversation",group:"Session Options:"},resume:{alias:"r",describe:"Resume a conversation - provide a session ID or interactively select a conversation to resume",group:"Session Options:",coerce:(b)=>{if(b===void 0||b===!0||b==="")return"true";return String(b)}},"fork-session":{alias:["forkSession"],type:"boolean",describe:"Create a new session ID when resuming",group:"Session Options:"},settings:{type:"string",describe:"Path to a settings JSON file or JSON string",group:"Configuration:"},"add-dir":{alias:["addDir"],type:"array",string:!0,describe:"Additional directories to allow tool access to",group:"Security Options:"},ide:{type:"boolean",describe:"Automatically connect to IDE on startup",group:"Integration:"},acp:{type:"boolean",describe:"Run in ACP (Agent Client Protocol) mode for IDE integration",group:"Integration:"},"strict-mcp-config":{alias:["strictMcpConfig"],type:"boolean",describe:"Only use MCP servers from --mcp-config",group:"MCP Options:"},"session-id":{alias:["sessionId"],type:"string",describe:"Use a specific session ID for the conversation",group:"Session Options:"},agents:{type:"string",describe:"JSON object defining custom agents",group:"AI Options:"},"setting-sources":{alias:["settingSources"],type:"string",describe:"Comma-separated list of setting sources to load",group:"Configuration:"},"plugin-dir":{alias:["pluginDir"],type:"array",string:!0,describe:"Load plugins from specified directories",group:"Plugin Options:"}},w={scriptName:"blade",usage:"$0 [command] [options]",description:h(),version:d(),locale:"en",showHelpOnFail:!0,demandCommand:!1,recommendCommands:!0,strict:!1};
2
2
  export{q as t,w as u};
@@ -1,4 +1,4 @@
1
- import{_ as O}from"./chunk-pkzy724w.js";import"./chunk-31chcjw6.js";import{Ga as J,Ia as L}from"./chunk-mnw1xcwn.js";import"./chunk-m22zeqdy.js";import"./chunk-f76d6jxg.js";import{Eb as H}from"./chunk-wre40mf1.js";import"./chunk-mt1c5da4.js";import"./chunk-0afbeec0.js";import P from"os";import E from"path";function _(q){return typeof q==="string"?q:void 0}function T(q){if(!Array.isArray(q))return;let z=[];for(let F of q)if(typeof F==="string")z.push(F);else if(F!=null)z.push(String(F));return z}function G(){console.log(`
1
+ import{ba as O}from"./chunk-t3pyakvc.js";import{Ba as L,za as J}from"./chunk-wc2xb6ge.js";import"./chunk-b6mxeey0.js";import"./chunk-k18bq4mn.js";import"./chunk-wcm0nc84.js";import{wb as H}from"./chunk-cadngkwr.js";import"./chunk-4hk7wfat.js";import"./chunk-cfrbsnfb.js";import P from"os";import E from"path";function _(q){return typeof q==="string"?q:void 0}function T(q){if(!Array.isArray(q))return;let z=[];for(let F of q)if(typeof F==="string")z.push(F);else if(F!=null)z.push(String(F));return z}function G(){console.log(`
2
2
  blade mcp
3
3
  `),console.log(`管理 MCP 服务器
4
4
  `),console.log("Commands:"),console.log(" blade mcp add <name> <commandOrUrl> [args...] 添加 MCP 服务器"),console.log(" blade mcp remove <name> 删除 MCP 服务器 [aliases: rm]"),console.log(" blade mcp list 列出所有 MCP 服务器并检查健康状态 [aliases: ls]"),console.log(" blade mcp get <name> 获取服务器详情"),console.log(` blade mcp add-json <name> <json> 从 JSON 字符串添加服务器
@@ -1,2 +1,2 @@
1
- import{sb as Q,tb as W,ub as Y}from"./chunk-vzebrztg.js";import{Ob as X}from"./chunk-mt1c5da4.js";var J=X("Service"),$="https://models.dev/api.json",j=3600000,G=null,Z=async()=>{if(G&&Date.now()-G.timestamp<j)return G.data;try{J.info("Fetching models.dev data...");let q=await fetch($,{headers:{"User-Agent":"Blade-CLI/1.0"}});if(!q.ok)throw Error(`HTTP ${q.status}: ${q.statusText}`);let x=await q.json();return G={data:x,timestamp:Date.now()},J.info(`Loaded ${Object.keys(x).length} providers from models.dev`),x}catch(q){throw J.error("Failed to fetch models.dev data:",q),q}},y=async()=>{let q=await Z(),x=[];for(let[w,k]of Object.entries(q)){let z=Object.keys(k.models||{}).length;if(z===0)continue;x.push({id:w,name:k.name||w,icon:W[w]||W.default,description:`${z} 个模型`,envVars:k.env||[],docUrl:k.doc,defaultBaseUrl:Y[w]})}return x.sort((w,k)=>{let z=Q.indexOf(w.id),K=Q.indexOf(k.id);if(z!==-1&&K!==-1)return z-K;if(z!==-1)return-1;if(K!==-1)return 1;return w.name.localeCompare(k.name)})},f=async(q)=>{let w=(await Z())[q];if(!w?.models)return J.warn(`No models found for provider: ${q}`),[];return Object.values(w.models).map((k)=>({id:k.id,name:k.name||k.id,contextWindow:k.limit?.context,maxOutput:k.limit?.output,inputCost:k.cost?.input,outputCost:k.cost?.output}))};
2
- export{y as e,f};
1
+ import{xb as Q,yb as W,zb as Y}from"./chunk-m4a480k7.js";import{Nb as X}from"./chunk-4hk7wfat.js";var J=X("Service"),$="https://models.dev/api.json",j=3600000,G=null,Z=async()=>{if(G&&Date.now()-G.timestamp<j)return G.data;try{J.info("Fetching models.dev data...");let q=await fetch($,{headers:{"User-Agent":"Blade-CLI/1.0"}});if(!q.ok)throw Error(`HTTP ${q.status}: ${q.statusText}`);let x=await q.json();return G={data:x,timestamp:Date.now()},J.info(`Loaded ${Object.keys(x).length} providers from models.dev`),x}catch(q){throw J.error("Failed to fetch models.dev data:",q),q}},y=async()=>{let q=await Z(),x=[];for(let[w,k]of Object.entries(q)){let z=Object.keys(k.models||{}).length;if(z===0)continue;x.push({id:w,name:k.name||w,icon:W[w]||W.default,description:`${z} 个模型`,envVars:k.env||[],docUrl:k.doc,defaultBaseUrl:Y[w]})}return x.sort((w,k)=>{let z=Q.indexOf(w.id),K=Q.indexOf(k.id);if(z!==-1&&K!==-1)return z-K;if(z!==-1)return-1;if(K!==-1)return 1;return w.name.localeCompare(k.name)})},f=async(q)=>{let w=(await Z())[q];if(!w?.models)return J.warn(`No models found for provider: ${q}`),[];return Object.values(w.models).map((k)=>({id:k.id,name:k.name||k.id,contextWindow:k.limit?.context,maxOutput:k.limit?.output,inputCost:k.cost?.input,outputCost:k.cost?.output}))};
2
+ export{y as j,f as k};
@@ -1,4 +1,4 @@
1
- import{da as S,ea as h}from"./chunk-31chcjw6.js";import{ob as M}from"./chunk-f76d6jxg.js";import{Qb as c}from"./chunk-0afbeec0.js";import{EventEmitter as r}from"events";import{z as L}from"zod";import{isPlainObject as p}from"lodash-es";function J(B){if(B===void 0)return"undefined";if(typeof B==="string")return B;try{let G=JSON.stringify(B);return G===void 0?String(B):G}catch{return String(B)}}class R extends Error{issues;type;constructor(B,G,Q="validation_error"){super(B);this.issues=G;this.type=Q;this.name="ToolValidationError"}}function g(B){let{code:G}=B,Q=B,X=Q.received;switch(G){case"invalid_type":{let Y=Q.expected;return`类型错误:期望 ${J(Y)},实际收到 ${J(X)}`}case"too_small":{let{minimum:Y,inclusive:$}=Q,W=typeof Q.type==="string"?Q.type:void 0;if(W==="string"&&typeof Y==="number")return`长度不能少于 ${Y} 个字符`;if(W==="number"&&typeof Y==="number")return`不能小于${$?"等于":""} ${Y}`;if(W==="array"&&typeof Y==="number")return`数组长度不能少于 ${Y}`;return"值太小"}case"too_big":{let{maximum:Y,inclusive:$}=Q,W=typeof Q.type==="string"?Q.type:void 0;if(W==="string"&&typeof Y==="number")return`长度不能超过 ${Y} 个字符`;if(W==="number"&&typeof Y==="number")return`不能大于${$?"等于":""} ${Y}`;if(W==="array"&&typeof Y==="number")return`数组长度不能超过 ${Y}`;return"值太大"}case"invalid_string":{let Y=Q.validation;if(Y==="email")return"必须是有效的电子邮件地址";if(Y==="url")return"必须是有效的 URL";if(Y==="uuid")return"必须是有效的 UUID";if(p(Y)){let $=Y;if(typeof $.includes==="string")return`必须包含 "${$.includes}"`;if(typeof $.startsWith==="string")return`必须以 "${$.startsWith}" 开头`;if(typeof $.endsWith==="string")return`必须以 "${$.endsWith}" 结尾`}return"字符串格式不正确"}case"invalid_enum_value":{let Y=Q.options;if(Array.isArray(Y))return`必须是以下值之一:${Y.map(($)=>J($)).join(", ")}`;return"必须是枚举允许的值之一"}case"invalid_literal":{let Y=Q.expected;return`必须是字面量值:${J(Y)}`}case"unrecognized_keys":{let Y=Q.keys;if(Array.isArray(Y))return`包含未知的参数:${Y.map(($)=>J($)).join(", ")}`;return"包含未知的参数"}case"invalid_union":return"不符合任何有效的类型定义";case"invalid_date":return"必须是有效的日期";case"custom":return B.message||"自定义验证失败";default:return B.message||"验证失败"}}function u(B){let G=B.issues.map((X)=>{let Y=X.path.join("."),$=g(X),W=X.received;return{field:Y||"root",message:$,value:W}}),Q=G.length===1?`参数验证失败 [${G[0].field}]: ${G[0].message}`:`参数验证失败 (${G.length} 个错误):
1
+ import{Ha as S,Ia as h}from"./chunk-k18bq4mn.js";import{nb as M}from"./chunk-wcm0nc84.js";import{Pb as c}from"./chunk-cfrbsnfb.js";import{EventEmitter as r}from"events";import{z as L}from"zod";import{isPlainObject as p}from"lodash-es";function J(B){if(B===void 0)return"undefined";if(typeof B==="string")return B;try{let G=JSON.stringify(B);return G===void 0?String(B):G}catch{return String(B)}}class R extends Error{issues;type;constructor(B,G,Q="validation_error"){super(B);this.issues=G;this.type=Q;this.name="ToolValidationError"}}function g(B){let{code:G}=B,Q=B,X=Q.received;switch(G){case"invalid_type":{let Y=Q.expected;return`类型错误:期望 ${J(Y)},实际收到 ${J(X)}`}case"too_small":{let{minimum:Y,inclusive:$}=Q,W=typeof Q.type==="string"?Q.type:void 0;if(W==="string"&&typeof Y==="number")return`长度不能少于 ${Y} 个字符`;if(W==="number"&&typeof Y==="number")return`不能小于${$?"等于":""} ${Y}`;if(W==="array"&&typeof Y==="number")return`数组长度不能少于 ${Y}`;return"值太小"}case"too_big":{let{maximum:Y,inclusive:$}=Q,W=typeof Q.type==="string"?Q.type:void 0;if(W==="string"&&typeof Y==="number")return`长度不能超过 ${Y} 个字符`;if(W==="number"&&typeof Y==="number")return`不能大于${$?"等于":""} ${Y}`;if(W==="array"&&typeof Y==="number")return`数组长度不能超过 ${Y}`;return"值太大"}case"invalid_string":{let Y=Q.validation;if(Y==="email")return"必须是有效的电子邮件地址";if(Y==="url")return"必须是有效的 URL";if(Y==="uuid")return"必须是有效的 UUID";if(p(Y)){let $=Y;if(typeof $.includes==="string")return`必须包含 "${$.includes}"`;if(typeof $.startsWith==="string")return`必须以 "${$.startsWith}" 开头`;if(typeof $.endsWith==="string")return`必须以 "${$.endsWith}" 结尾`}return"字符串格式不正确"}case"invalid_enum_value":{let Y=Q.options;if(Array.isArray(Y))return`必须是以下值之一:${Y.map(($)=>J($)).join(", ")}`;return"必须是枚举允许的值之一"}case"invalid_literal":{let Y=Q.expected;return`必须是字面量值:${J(Y)}`}case"unrecognized_keys":{let Y=Q.keys;if(Array.isArray(Y))return`包含未知的参数:${Y.map(($)=>J($)).join(", ")}`;return"包含未知的参数"}case"invalid_union":return"不符合任何有效的类型定义";case"invalid_date":return"必须是有效的日期";case"custom":return B.message||"自定义验证失败";default:return B.message||"验证失败"}}function u(B){let G=B.issues.map((X)=>{let Y=X.path.join("."),$=g(X),W=X.received;return{field:Y||"root",message:$,value:W}}),Q=G.length===1?`参数验证失败 [${G[0].field}]: ${G[0].message}`:`参数验证失败 (${G.length} 个错误):
2
2
  ${G.map((X)=>` - ${X.field}: ${X.message}`).join(`
3
3
  `)}`;return new R(Q,G)}function y(B,G){let Q=B.safeParse(G);if(!Q.success)throw u(Q.error);return Q.data}import{zodToJsonSchema as l}from"zod-to-json-schema";function H(B){let G=l(B,{target:"jsonSchema7",$refStrategy:"none"});if(G.type==="object"&&G.properties&&!Array.isArray(G.required))G.required=[];return G}class w{toolName;params;executeFn;descriptionFn;affectedPathsFn;constructor(B,G,Q,X,Y){this.toolName=B;this.params=G;this.executeFn=Q;this.descriptionFn=X;this.affectedPathsFn=Y}getDescription(){if(this.descriptionFn)return this.descriptionFn(this.params);return`执行工具: ${this.toolName}`}getAffectedPaths(){if(this.affectedPathsFn)return this.affectedPathsFn(this.params);return[]}async execute(B,G,Q){let X={signal:B,updateOutput:G,...Q};return this.executeFn(this.params,X)}}function k(B){return{name:B.name,displayName:B.displayName,kind:B.kind,isReadOnly:B.isReadOnly??M(B.kind),isConcurrencySafe:B.isConcurrencySafe??!1,strict:B.strict??!1,description:B.description,version:B.version||"1.0.0",category:B.category,tags:B.tags||[],getFunctionDeclaration(){let G=H(B.schema),Q=B.description.short;if(B.description.long)Q+=`
4
4
 
@@ -31,4 +31,4 @@ ${B.description.important.map((X)=>`[WARN] ${X}`).join(`
31
31
  `),X.close(),G(q)}catch(W){X.close(),Q(W)}});X.on("error",Q),X.listen(Z,()=>{console.log(`[OAuth] Callback server listening on port ${Z}`)}),setTimeout(()=>{X.close(),Q(Error("OAuth callback timeout"))},300000)})}async exchangeCodeForToken(B,G,Q){let X=B.redirectUri||`http://localhost:${Z}${D}`,Y=new URLSearchParams({grant_type:"authorization_code",code:G,redirect_uri:X,code_verifier:Q,client_id:B.clientId});if(B.clientSecret)Y.append("client_secret",B.clientSecret);let $=await fetch(B.tokenUrl,{method:"POST",headers:{"Content-Type":"application/x-www-form-urlencoded",Accept:"application/json"},body:Y.toString()});if(!$.ok){let W=await $.text();throw Error(`Token exchange failed: ${$.status} - ${W}`)}return await $.json()}async refreshAccessToken(B,G){let Q=new URLSearchParams({grant_type:"refresh_token",refresh_token:G,client_id:B.clientId});if(B.clientSecret)Q.append("client_secret",B.clientSecret);if(B.scopes&&B.scopes.length>0)Q.append("scope",B.scopes.join(" "));let X=await fetch(B.tokenUrl,{method:"POST",headers:{"Content-Type":"application/x-www-form-urlencoded",Accept:"application/json"},body:Q.toString()});if(!X.ok){let Y=await X.text();throw Error(`Token refresh failed: ${X.status} - ${Y}`)}return await X.json()}async authenticate(B,G){if(!G.clientId||!G.authorizationUrl||!G.tokenUrl)throw Error("Missing required OAuth configuration");let Q=this.generatePKCEParams(),X=this.buildAuthorizationUrl(G,Q);console.log(`
32
32
  [OAuth] Opening browser for authentication...`),console.log(`
33
33
  If the browser does not open automatically, copy and paste this URL:`),console.log(X),console.log("");let Y=this.startCallbackServer(Q.state);try{await this.openAuthorizationUrl(X)}catch(F){console.warn("[OAuth] Failed to open browser automatically:",F)}let $=await Y;console.log("[OAuth] Authorization code received, exchanging for tokens...");let W=await this.exchangeCodeForToken(G,$,Q.codeVerifier),q={accessToken:W.access_token,tokenType:W.token_type||"Bearer",refreshToken:W.refresh_token,scope:W.scope};if(W.expires_in)q.expiresAt=Date.now()+W.expires_in*1000;return await this.tokenStorage.saveToken(B,q,G.clientId,G.tokenUrl),console.log("[OAuth] Authentication successful! Token saved."),q}async openAuthorizationUrl(B){let{command:G,args:Q}=this.getBrowserCommand(B);await new Promise((X,Y)=>{let $=m(G,Q,{stdio:"ignore"});$.once("error",Y),$.once("close",(W)=>{if(W===0||W===null)X();else Y(Error(`Failed to open browser (exit code ${W})`))})})}getBrowserCommand(B){if(process.platform==="darwin")return{command:"open",args:[B]};if(process.platform==="win32")return{command:"cmd",args:["/c","start","",B]};return{command:"xdg-open",args:[B]}}async getValidToken(B,G){let Q=await this.tokenStorage.getCredentials(B);if(!Q)return null;let{token:X}=Q;if(!this.tokenStorage.isTokenExpired(X))return X.accessToken;if(X.refreshToken&&G.clientId&&Q.tokenUrl)try{console.log(`[OAuth] Refreshing expired token for server: ${B}`);let Y=await this.refreshAccessToken(G,X.refreshToken),$={accessToken:Y.access_token,tokenType:Y.token_type,refreshToken:Y.refresh_token||X.refreshToken,scope:Y.scope||X.scope};if(Y.expires_in)$.expiresAt=Date.now()+Y.expires_in*1000;return await this.tokenStorage.saveToken(B,$,G.clientId,Q.tokenUrl),$.accessToken}catch(Y){console.error("[OAuth] Failed to refresh token:",Y),await this.tokenStorage.deleteCredentials(B)}return null}}import{EventEmitter as i}from"events";class E extends i{client;config;checkTimer=null;isChecking=!1;consecutiveFailures=0;lastCheckTime=0;currentStatus="healthy";constructor(B,G={}){super();this.client=B,this.config={interval:G.interval??30000,timeout:G.timeout??1e4,enabled:G.enabled??!1,failureThreshold:G.failureThreshold??3}}start(){if(this.checkTimer){console.warn("[HealthMonitor] 健康监控已在运行");return}if(!this.config.enabled){console.log("[HealthMonitor] 健康监控未启用");return}console.log(`[HealthMonitor] 启动健康监控(间隔: ${this.config.interval}ms)`),this.scheduleNextCheck()}stop(){if(this.checkTimer)clearTimeout(this.checkTimer),this.checkTimer=null,console.log("[HealthMonitor] 停止健康监控")}scheduleNextCheck(){this.checkTimer=setTimeout(async()=>{await this.performHealthCheck(),this.scheduleNextCheck()},this.config.interval)}async performHealthCheck(){if(this.isChecking)return console.warn("[HealthMonitor] 上一次检查仍在进行中"),this.getLastResult();this.isChecking=!0,this.lastCheckTime=Date.now(),this.setStatus("checking");try{let B=this.client.connectionStatus;if(B==="connected"){await this.pingServer(),this.consecutiveFailures=0,this.setStatus("healthy");let G={status:"healthy",timestamp:Date.now(),consecutiveFailures:0};return this.emit("healthCheck",G),G}else throw Error(`连接状态异常: ${B}`)}catch(B){this.consecutiveFailures++;let G=B;console.warn(`[HealthMonitor] 健康检查失败(${this.consecutiveFailures}/${this.config.failureThreshold}):`,G.message);let Q;if(this.consecutiveFailures>=this.config.failureThreshold)Q="unhealthy",this.emit("unhealthy",this.consecutiveFailures,G),console.log("[HealthMonitor] 达到失败阈值,触发重连..."),this.triggerReconnection().catch((Y)=>{console.error("[HealthMonitor] 重连失败:",Y)});else Q="degraded";this.setStatus(Q);let X={status:Q,timestamp:Date.now(),consecutiveFailures:this.consecutiveFailures,lastError:G};return this.emit("healthCheck",X),X}finally{this.isChecking=!1}}async pingServer(){let B=new Promise((Q,X)=>{setTimeout(()=>X(Error("Health check timeout")),this.config.timeout)}),G=(async()=>{if(this.client.availableTools.length===0){if(!this.client.server)throw Error("Server info not available")}})();await Promise.race([G,B])}async triggerReconnection(){try{await this.client.disconnect(),await new Promise((B)=>setTimeout(B,1000)),await this.client.connect(),console.log("[HealthMonitor] 重连成功"),this.consecutiveFailures=0,this.setStatus("healthy"),this.emit("reconnected")}catch(B){throw console.error("[HealthMonitor] 重连失败:",B),B}}setStatus(B){if(this.currentStatus!==B){let G=this.currentStatus;this.currentStatus=B,this.emit("statusChanged",B,G)}}getStatus(){return this.currentStatus}getLastResult(){return{status:this.currentStatus,timestamp:this.lastCheckTime,consecutiveFailures:this.consecutiveFailures}}getStatistics(){return{status:this.currentStatus,consecutiveFailures:this.consecutiveFailures,lastCheckTime:this.lastCheckTime,isChecking:this.isChecking,config:this.config}}async checkNow(){return this.performHealthCheck()}resetFailureCount(){if(this.consecutiveFailures=0,this.currentStatus!=="checking")this.setStatus("healthy")}}function d(B){if(!(B instanceof Error))return{type:"unknown",isRetryable:!1,originalError:Error(String(B))};let G=B.message.toLowerCase();if(["command not found","no such file","permission denied","invalid configuration","malformed","syntax error"].some((Y)=>G.includes(Y)))return{type:"config_error",isRetryable:!1,originalError:B};if(G.includes("unauthorized")||G.includes("401")||G.includes("authentication failed"))return{type:"auth_error",isRetryable:!1,originalError:B};if(["timeout","connection refused","network error","temporary","try again","rate limit","too many requests","service unavailable","socket hang up","econnreset","enotfound","econnrefused","etimedout","503","429"].some((Y)=>G.includes(Y)))return{type:"network_temporary",isRetryable:!0,originalError:B};return{type:"unknown",isRetryable:!0,originalError:B}}class N extends n{config;status="disconnected";sdkClient=null;tools=new Map;serverInfo=null;reconnectAttempts=0;MAX_RECONNECT_ATTEMPTS=5;reconnectTimer=null;isManualDisconnect=!1;oauthProvider=null;serverName;healthMonitor=null;constructor(B,G,Q){super();this.config=B;if(this.serverName=G||"default",B.oauth?.enabled)this.oauthProvider=new O;if(Q?.enabled)this.healthMonitor=new E(this,Q),this.healthMonitor.on("unhealthy",(X,Y)=>{this.emit("unhealthy",X,Y)}),this.healthMonitor.on("reconnected",()=>{this.emit("healthMonitorReconnected")})}get connectionStatus(){return this.status}get availableTools(){return Array.from(this.tools.values())}get server(){return this.serverInfo}get healthCheck(){return this.healthMonitor}async connect(){return this.connectWithRetry(3,1000)}async connectWithRetry(B=3,G=1000){if(this.status!=="disconnected"&&this.status!=="error")throw Error("客户端已连接或正在连接中");if(this.status==="error"){if(this.sdkClient){try{await this.sdkClient.close()}catch{}this.sdkClient=null}this.setStatus("disconnected")}let Q=null;for(let X=1;X<=B;X++)try{await this.doConnect(),this.reconnectAttempts=0;return}catch(Y){Q=Y;let $=d(Y);if(!$.isRetryable)throw console.error("[McpClient] 检测到永久性错误,放弃重试:",$.type),Y;if(X<B){let W=G*Math.pow(2,X-1);console.warn(`[McpClient] 连接失败(${X}/${B}),${W}ms 后重试...`),await new Promise((q)=>setTimeout(q,W))}}throw Q||Error("连接失败")}async doConnect(){try{this.setStatus("connecting"),this.sdkClient=new t({name:h(),version:S()},{capabilities:{roots:{listChanged:!0},sampling:{}}}),this.sdkClient.onclose=()=>{this.handleUnexpectedClose()};let B=await this.createTransport();await this.sdkClient.connect(B);let G=this.sdkClient.getServerVersion();if(this.serverInfo={name:G?.name||"Unknown",version:G?.version||"0.0.0"},await this.loadTools(),this.setStatus("connected"),this.emit("connected",this.serverInfo),this.healthMonitor)this.healthMonitor.start()}catch(B){throw this.setStatus("error"),this.emit("error",B),B}}handleUnexpectedClose(){if(this.isManualDisconnect)return;if(this.status==="connected")console.warn("[McpClient] 检测到意外断连,准备重连..."),this.setStatus("error"),this.emit("error",Error("MCP服务器连接意外关闭")),this.scheduleReconnect()}scheduleReconnect(){if(this.reconnectTimer)clearTimeout(this.reconnectTimer),this.reconnectTimer=null;if(this.reconnectAttempts>=this.MAX_RECONNECT_ATTEMPTS){console.error("[McpClient] 达到最大重连次数,放弃重连"),this.emit("reconnectFailed");return}let B=Math.min(1000*Math.pow(2,this.reconnectAttempts),30000);this.reconnectAttempts++,console.log(`[McpClient] 将在 ${B}ms 后进行第 ${this.reconnectAttempts} 次重连...`),this.reconnectTimer=setTimeout(async()=>{try{if(this.sdkClient){try{await this.sdkClient.close()}catch{}this.sdkClient=null}this.setStatus("disconnected"),await this.doConnect(),console.log("[McpClient] 重连成功"),this.reconnectAttempts=0,this.emit("reconnected")}catch(G){if(console.error("[McpClient] 重连失败:",G),d(G).isRetryable)this.scheduleReconnect();else console.error("[McpClient] 检测到永久性错误,停止重连"),this.emit("reconnectFailed")}},B)}async disconnect(){if(this.isManualDisconnect=!0,this.healthMonitor)this.healthMonitor.stop();if(this.reconnectTimer)clearTimeout(this.reconnectTimer),this.reconnectTimer=null;if(this.sdkClient){try{await this.sdkClient.close()}catch(B){console.warn("[McpClient] 断开连接时出错:",B)}this.sdkClient=null}this.tools.clear(),this.serverInfo=null,this.reconnectAttempts=0,this.setStatus("disconnected"),this.emit("disconnected"),this.isManualDisconnect=!1}async callTool(B,G={}){if(!this.sdkClient)throw Error("客户端未连接到服务器");if(!this.tools.has(B))throw Error(`工具 "${B}" 不存在`);try{return await this.sdkClient.callTool({name:B,arguments:G})}catch(Q){throw console.error(`[McpClient] 调用工具 "${B}" 失败:`,Q),Q}}async createTransport(){let{type:B,command:G,args:Q,env:X,url:Y,headers:$,oauth:W}=this.config,q={...$};if(W?.enabled&&this.oauthProvider&&(B==="sse"||B==="http"))try{let F=await this.oauthProvider.getValidToken(this.serverName,W);if(!F){console.log(`[McpClient] 服务器 "${this.serverName}" 需要 OAuth 认证`);let A=await this.oauthProvider.authenticate(this.serverName,W);q.Authorization=`Bearer ${A.accessToken}`}else q.Authorization=`Bearer ${F}`}catch(F){throw console.error("[McpClient] OAuth 认证失败:",F),Error(`OAuth 认证失败: ${F instanceof Error?F.message:String(F)}`)}if(B==="stdio"){if(!G)throw Error("stdio 传输需要 command 参数");let F={};for(let[A,K]of Object.entries(process.env))if(K!==void 0)F[A]=K;return new s({command:G,args:Q||[],env:{...F,...X},stderr:"ignore"})}else if(B==="sse"){if(!Y)throw Error("sse 传输需要 url 参数");return new a(new URL(Y),{requestInit:{headers:q}})}else if(B==="http"){if(!Y)throw Error("http 传输需要 url 参数");let{StreamableHTTPClientTransport:F}=await import("@modelcontextprotocol/sdk/client/streamableHttp.js");return new F(new URL(Y),{requestInit:{headers:q}})}throw Error(`不支持的传输类型: ${B}`)}async loadTools(){if(!this.sdkClient)return;try{let B=await this.sdkClient.listTools();if(this.tools.clear(),B.tools)for(let G of B.tools)this.tools.set(G.name,G);this.emit("toolsUpdated",this.availableTools)}catch(B){throw console.error("[McpClient] 加载工具失败:",B),B}}setStatus(B){let G=this.status;this.status=B,this.emit("statusChanged",B,G)}async initialize(){return this.connect()}async destroy(){return this.disconnect()}async connectToServer(B){return this.connect()}async disconnectFromServer(B){return this.disconnect()}async listResources(B){if(!this.sdkClient)return[];try{return(await this.sdkClient.listResources()).resources||[]}catch{return[]}}async listTools(B){return this.availableTools}async readResource(B,G){if(!this.sdkClient)throw Error("客户端未连接");return(await this.sdkClient.readResource({uri:B})).contents?.[0]||{uri:B,text:""}}}class b extends r{static instance=null;servers=new Map;isDiscovering=!1;constructor(){super()}static getInstance(){if(!b.instance)b.instance=new b;return b.instance}static createIsolated(){return new b}async registerServer(B,G){if(this.servers.has(B))throw Error(`MCP服务器 "${B}" 已经注册`);let Q=new N(G,B,G.healthCheck),X={config:G,client:Q,status:"disconnected",tools:[]};this.setupClientEventHandlers(Q,X,B),this.servers.set(B,X),this.emit("serverRegistered",B,X);try{await this.connectServer(B)}catch(Y){console.warn(`MCP服务器 "${B}" 连接失败:`,Y)}}async unregisterServer(B){let G=this.servers.get(B);if(!G)return;try{await G.client.disconnect()}catch(Q){console.warn(`断开MCP服务器 "${B}" 时出错:`,Q)}this.servers.delete(B),this.emit("serverUnregistered",B)}async connectServer(B){let G=this.servers.get(B);if(!G)throw Error(`MCP服务器 "${B}" 未注册`);if(G.status==="connected")return;try{G.status="connecting",await G.client.connect(),G.connectedAt=new Date,G.lastError=void 0,G.tools=G.client.availableTools}catch(Q){throw G.lastError=Q,G.status="error",Q}}async disconnectServer(B){let G=this.servers.get(B);if(!G)return;await G.client.disconnect(),G.connectedAt=void 0}async reconnectServer(B){let G=this.servers.get(B);if(!G)throw Error(`MCP服务器 "${B}" 未注册`);if(G.status==="connected")await G.client.disconnect();try{G.status="connecting",await G.client.connect(),G.connectedAt=new Date,G.lastError=void 0,G.tools=G.client.availableTools}catch(Q){throw G.lastError=Q,G.status="error",Q}}async getAvailableTools(){let B=[],G=new Map;for(let[Q,X]of this.servers)if(X.status==="connected")for(let Y of X.tools){let $=G.get(Y.name)||0;G.set(Y.name,$+1)}for(let[Q,X]of this.servers)if(X.status==="connected")for(let Y of X.tools){let W=(G.get(Y.name)||0)>1?`${Q}__${Y.name}`:Y.name,q=U(X.client,Q,Y,W);B.push(q)}return B}async findTool(B){for(let[G,Q]of this.servers)if(Q.status==="connected"){let X=Q.tools.find((Y)=>Y.name===B);if(X)return U(Q.client,G,X)}return null}getToolsByServer(B){let G=this.servers.get(B);if(!G||G.status!=="connected")return[];return G.tools.map((Q)=>U(G.client,B,Q))}getServerStatus(B){return this.servers.get(B)||null}getAllServers(){return new Map(this.servers)}async refreshAllTools(){let B=[];for(let[G,Q]of this.servers)if(Q.status==="connected")B.push(this.refreshServerTools(G));await Promise.allSettled(B)}async refreshServerTools(B){let G=this.servers.get(B);if(!G||G.status!=="connected")return;try{let Q=G.client.availableTools,X=G.tools.length;G.tools=Q,this.emit("toolsUpdated",B,Q,X)}catch(Q){console.warn(`刷新服务器 "${B}" 工具列表失败:`,Q)}}setupClientEventHandlers(B,G,Q){B.on("connected",(X)=>{G.status="connected",G.connectedAt=new Date,G.tools=B.availableTools,this.emit("serverConnected",Q,X)}),B.on("disconnected",()=>{G.status="disconnected",G.connectedAt=void 0,G.tools=[],this.emit("serverDisconnected",Q)}),B.on("error",(X)=>{G.status="error",G.lastError=X,this.emit("serverError",Q,X)}),B.on("toolsUpdated",(X)=>{let Y=G.tools.length;G.tools=X,this.emit("toolsUpdated",Q,X,Y)}),B.on("statusChanged",(X,Y)=>{G.status=X,this.emit("serverStatusChanged",Q,X,Y)})}async discoverServers(){if(this.isDiscovering)return Array.from(this.servers.values());this.isDiscovering=!0,this.emit("discoveryStarted");try{return Array.from(this.servers.values())}finally{this.isDiscovering=!1,this.emit("discoveryCompleted")}}async registerServers(B){let G=Object.entries(B).map(([Q,X])=>this.registerServer(Q,X).catch((Y)=>{return console.warn(`注册MCP服务器 "${Q}" 失败:`,Y),Y}));await Promise.allSettled(G)}getStatistics(){let B=0,G=0,Q=0;for(let X of this.servers.values())if(X.status==="connected")B++,G+=X.tools.length;else if(X.status==="error")Q++;return{totalServers:this.servers.size,connectedServers:B,errorServers:Q,totalTools:G,isDiscovering:this.isDiscovering}}async disconnectAll(){let B=[];for(let[G,Q]of this.servers)if(Q.status==="connected")B.push(Q.client.disconnect().catch((X)=>{console.warn(`断开 MCP 服务器 "${G}" 时出错:`,X)}));await Promise.allSettled(B),this.servers.clear()}}
34
- export{k as Z,b as _};
34
+ export{k as aa,b as ba};