grix-connector 3.13.1 → 3.15.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/adapter/cursor/context-window.js +1 -0
- package/dist/adapter/cursor/cursor-adapter.js +11 -5
- package/dist/adapter/cursor/fetch-models.js +2 -0
- package/dist/adapter/cursor/fetch-usage.js +1 -0
- package/dist/adapter/cursor/index.js +1 -1
- package/dist/bridge/bridge.js +9 -9
- package/dist/bridge/raw-event-delivery.js +2 -0
- package/dist/bridge/send-controller.js +3 -3
- package/dist/core/persistence/session-binding-store.js +1 -1
- package/dist/manager.js +1 -1
- package/dist/mcp/stream-http/security.js +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
const r=2e5,o=1e6;function c(s){const t=String(s??"").trim();if(!t)return 2e5;if(/context\s*=\s*1m\b/i.test(t))return 1e6;if(/context\s*=\s*(\d+)k\b/i.test(t)){const n=t.match(/context\s*=\s*(\d+)k\b/i),e=Number(n?.[1]??0);if(Number.isFinite(e)&&e>0)return e*1e3}if(/context\s*=\s*(\d+)\b/i.test(t)){const n=t.match(/context\s*=\s*(\d+)\b/i),e=Number(n?.[1]??0);if(Number.isFinite(e)&&e>=1e4)return e}return 2e5}function u(s,t){const n=Math.max(0,Number.isFinite(s)?s:0),e=Math.max(1,Number.isFinite(t)&&t>0?t:2e5),i=Math.min(100,n/e*100);return{usedPercentage:i,remainingPercentage:Math.max(0,100-i),usedTokens:n,sizeTokens:e}}export{u as buildCursorContextWindow,c as resolveCursorContextWindowSize};
|
|
@@ -1,8 +1,14 @@
|
|
|
1
|
-
import{createInterface as
|
|
2
|
-
`)){const r=o.trim();if(!r||r.toLowerCase().includes("available model"))continue;const d=r.match(/^(\S+)\s+-\s+(.+)$/);d&&n.push({id:d[1],displayName:d[2].trim()})}n.length>0&&(this._availableModels=n,c.info("cursor-adapter",`Loaded ${n.length} available models`))}catch(e){c.warn("cursor-adapter",`Failed to refresh models: ${e instanceof Error?e.message:String(e)}`)}}async stop(){this.stopped=!0,this.alive=!1;for(const e of this.activeBySession.values())S(e.child,"SIGTERM");this.activeBySession.clear(),this.internalApi&&(await this.internalApi.stop(),this.internalApi=null)}isAlive(){return this.alive}async createSession(e){const t=`cursor-${Date.now()}-${++this.sessionSeq}`;return this.sessions.add(t),this.sessionRuntime.set(t,{cwd:typeof e.cwd=="string"?e.cwd:void 0,modelId:typeof e.modelId=="string"?e.modelId:void 0,modeId:typeof e.modeId=="string"?e.modeId:void 0}),t}async resumeSession(e,t){this.sessions.add(e);const i=this.sessionRuntime.get(e)??{};this.sessionRuntime.set(e,{...i,cwd:typeof t?.cwd=="string"?t.cwd:i.cwd,modelId:typeof t?.modelId=="string"?t.modelId:i.modelId,modeId:typeof t?.modeId=="string"?t.modeId:i.modeId})}onAgentProfileChanged(){this.identity.onProfileChanged()}async destroySession(e){this.sessions.delete(e),this.sessionRuntime.delete(e),this.identity.forgetSession(e)}sendPrompt(e){const t=new L(e.adapterSessionId);if(!this.alive||this.stopped)return queueMicrotask(()=>t.emitError(new Error("adapter not running"))),t;const i=this.inboundQueue.shift()??{event_id:`cursor-evt-${Date.now()}`,session_id:e.adapterSessionId,content:e.text},s=e.adapterSessionId,n=this.pendingBySession.get(s)??[];return n.push({event:i,request:e,handle:t}),this.pendingBySession.set(s,n),this.tryStartNext(s),t}tryStartNext(e){if(this.activeBySession.has(e)||!this.alive||this.stopped)return;const t=this.pendingBySession.get(e);if(!t||t.length===0)return;const i=t.shift();t.length===0?this.pendingBySession.delete(e):this.pendingBySession.set(e,t),i&&this.startJob(i,!1,0)}startJob(e,t,i){const{event:s,request:n,handle:o}=e,r=n.adapterSessionId,d=this.sessionRuntime.get(n.adapterSessionId)??{},m=this.config.options??{},u=[...this.config.args??[]];u.push("-p","--output-format","stream-json","--stream-partial-output"),m.trust!==!1&&u.push("--trust");const g=d.cwd||m.workspace;this.internalApi&&g&&(this.ensureWorkspaceMcpAndSkills(g),u.push("--approve-mcps")),g&&u.push("--workspace",g);const k=d.modelId||m.model;k&&u.push("--model",k);const _=d.modeId||m.mode;_&&Q.has(_)&&u.push("--mode",_);const A=d.cursorSessionId??this.bindingStore?.getAcpSessionId(n.adapterSessionId),w=!!(A&&m.use_continue!==!1&&!t);w&&u.push("--resume",A),u.push(this.buildPromptText(n));const y={...process.env,...this.config.env??{}},P=E(this.config.command,typeof y.PATH=="string"?y.PATH:void 0);c.info("cursor-adapter",`job start: event=${s.event_id} session=${s.session_id} retry=${i}`);let p;try{p=$(P,u,{cwd:g||process.cwd(),env:y,stdio:["ignore","pipe","pipe"]}).process}catch(a){this.finishActive(r,"failed",`spawn failed: ${a instanceof Error?a.message:String(a)}`);return}this.callbacks.sendEventAck(s.event_id,s.session_id);const l={event:s,request:n,handle:o,child:p,seq:0,done:!1,stderr:"",timer:null,idleTimer:null,toolCallsInFlight:new Set,retryCount:i,usedContinue:w,workspace:g||process.cwd(),args:u};this.activeBySession.set(r,l),this.armActiveIdleTimer(r);const I=n.timeoutMs&&n.timeoutMs>0?n.timeoutMs:0;I>0&&(l.timer=setTimeout(()=>{l.done||(S(p,"SIGTERM"),this.finishActive(r,"failed",`cursor agent timeout after ${I}ms`))},I)),D({input:p.stdout}).on("line",a=>this.handleStdoutLineForActive(l,a)),p.stderr?.on("data",a=>{const v=l.stderr+String(a??"");l.stderr=v.length>b.STDERR_MAX_CHARS?v.slice(-b.STDERR_MAX_CHARS):v}),p.once("error",a=>{this.finishActive(r,"failed",`spawn failed: ${String(a?.message??a)}`)}),p.once("close",a=>{if(!l.done)if((a??0)===0)this.finishActive(r,"responded");else{if(this.shouldRetryWithoutContinue(l)){l.timer&&clearTimeout(l.timer),l.idleTimer&&clearTimeout(l.idleTimer),this.activeBySession.delete(r),this.startJob(e,!0,l.retryCount+1);return}const v=l.stderr.trim()||`cursor agent exited with code ${a??-1}`;this.finishActive(r,"failed",v)}})}async cancel(e){const t=this.activeBySession.get(e);t&&(S(t.child,"SIGTERM"),this.finishActive(e,"canceled","canceled"))}deliverInboundEvent(e){const t=String(e.session_id??"").trim();if(!t){c.warn("cursor-adapter",`Dropping event ${e.event_id}: missing session_id`),this.callbacks.sendEventResult(e.event_id,"failed","missing session_id");return}if(!this.alive||this.stopped){c.warn("cursor-adapter",`Dropping event ${e.event_id}: adapter not running`),this.callbacks.sendEventAck(e.event_id,t),this.callbacks.sendEventResult(e.event_id,"failed","adapter not running");return}const i={adapterSessionId:t,text:String(e.content??"")},s=new L(t),n=this.pendingBySession.get(t)??[];n.push({event:e,request:i,handle:s}),this.pendingBySession.set(t,n),c.info("cursor-adapter",`inbound queued: event=${e.event_id} session=${t} depth=${n.length}`),this.tryStartNext(t)}killChildWithFallback(e){S(e,"SIGTERM"),setTimeout(()=>{if(e.exitCode===null&&e.signalCode===null)try{S(e,"SIGKILL")}catch{}},5e3).unref()}deliverStopEvent(e,t){if(t){const i=this.activeBySession.get(t);if(i?.event.event_id===e){this.killChildWithFallback(i.child),this.finishActive(t,"canceled","stopped");return}const s=this.pendingBySession.get(t)??[],n=s.findIndex(o=>o.event.event_id===e);if(n>=0){const[o]=s.splice(n,1);s.length===0?this.pendingBySession.delete(t):this.pendingBySession.set(t,s),this.callbacks.sendEventAck(o.event.event_id,o.event.session_id),this.callbacks.sendEventResult(o.event.event_id,"canceled","stopped"),o.handle.emitDone({status:"canceled",error:"stopped"})}return}for(const[i,s]of this.activeBySession.entries())if(s.event.event_id===e){this.killChildWithFallback(s.child),this.finishActive(i,"canceled","stopped");return}for(const[i,s]of this.pendingBySession.entries()){const n=s.findIndex(r=>r.event.event_id===e);if(n<0)continue;const[o]=s.splice(n,1);s.length===0?this.pendingBySession.delete(i):this.pendingBySession.set(i,s),this.callbacks.sendEventAck(o.event.event_id,o.event.session_id),this.callbacks.sendEventResult(o.event.event_id,"canceled","stopped"),o.handle.emitDone({status:"canceled",error:"stopped"});return}}async handleLocalAction(e){const t=String(e.action_type??"").trim().toLowerCase(),i=e.params??{},s=String(i.session_id??"").trim(),n=e.action_id;switch(t){case"set_model":{const o=String(i.model_id??"").trim();if(!o||!s)return this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_params","model_id and session_id are required"),{handled:!0,kind:"set_model"};const r=this.sessionRuntime.get(s)??{};return this.sessionRuntime.set(s,{...r,modelId:o}),this.callbacks.sendLocalActionResult(n,"ok",{outcome:"model_set",session_context:{model_id:o,mode_id:r.modeId??null,modelId:o,modeId:r.modeId??null},model_id:o,mode_id:r.modeId??null,available_models:this.availableModels}),{handled:!0,kind:"set_model"}}case"set_mode":{const o=String(i.mode_id??"").trim();if(!o||!s)return this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_params","mode_id and session_id are required"),{handled:!0,kind:"set_mode"};const r=this.sessionRuntime.get(s)??{};return this.sessionRuntime.set(s,{...r,modeId:o}),this.callbacks.sendLocalActionResult(n,"ok",{outcome:"mode_set",session_context:{model_id:r.modelId??null,mode_id:o,modelId:r.modelId??null,modeId:o},model_id:r.modelId??null,mode_id:o}),{handled:!0,kind:"set_mode"}}case"get_context":{const o=s?this.sessionRuntime.get(s):void 0;return this.callbacks.sendLocalActionResult(n,"ok",{session_context:{model_id:o?.modelId??null,mode_id:o?.modeId??null,cwd:o?.cwd??null},model_id:o?.modelId??null,mode_id:o?.modeId??null,modelId:o?.modelId??null,modeId:o?.modeId??null,cwd:o?.cwd??null,available_models:this.availableModels}),{handled:!0,kind:"get_context"}}case"get_rate_limits":{const o=this.getRateLimitsSnapshot();return this.callbacks.sendLocalActionResult(n,"ok",o),{handled:!0,kind:"get_rate_limits"}}case"get_session_usage":{const o=s?this.getUsageSnapshot(s):null;return o?this.callbacks.sendLocalActionResult(n,"ok",{adapterType:"cursor",available:!0,sampledAt:o.sampledAt,turns:o.turns,tokenUsage:o.total}):this.callbacks.sendLocalActionResult(n,"ok",{adapterType:"cursor",available:!1,sampledAt:null,turns:0,tokenUsage:null}),{handled:!0,kind:"get_session_usage"}}case"session_control":{const o=String(i.verb??"").trim().toLowerCase();if(o==="restart"&&s)this.deliverStopEvent("__all__",s),this.callbacks.sendLocalActionResult(n,"ok",{outcome:"restarted"});else if(o==="status"){const r=this.sessionRuntime.get(s),d=this.activeBySession.has(s);this.callbacks.sendLocalActionResult(n,"ok",{verb:"status",status:d?"running":"idle",session_context:{model_id:r?.modelId??null,mode_id:r?.modeId??null,cwd:r?.cwd??null},model_id:r?.modelId??null,mode_id:r?.modeId??null,modelId:r?.modelId??null,modeId:r?.modeId??null,cwd:r?.cwd??null})}else this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_verb",`Unsupported verb: ${o}`);return{handled:!0,kind:"session_control"}}default:return{handled:!1,kind:"unsupported"}}}setPermissionHandler(e){this.permissionHandler=e}async ping(e){if(!this.alive||this.stopped||(this.config.options??{}).mcp_tools!==!1&&this.internalApi===null)return!1;for(const i of this.activeBySession.values()){const s=i.child?.pid;if(s&&!Y(s))return!1}return!0}getStatus(){let e=0;for(const t of this.pendingBySession.values())e+=t.length;return{alive:this.alive,busy:this.activeBySession.size>0,sessions:this.sessions.size,details:{queueDepth:e+this.inboundQueue.length,activeSessions:this.activeBySession.size}}}getActiveEventIds(){const e=[];for(const t of this.activeBySession.values())t.event.event_id&&e.push(t.event.event_id);return e}clearActiveEventForShutdown(){this.activeBySession.clear()}getMcpConfig(){return null}async probe(e){const t=this.getStatus();return{...await H(this.config.command||"cursor-agent",{alive:t.alive,busy:t.busy,started:this.alive},e),session:this.probeSessionRecord()}}probeSessionRecord(){const e=this.firstKnownCursorSessionId(),t=e?V(e):null;if(!t)return{recordPath:null,lastActivityMs:null,freshMs:null};try{const i=q(t);return{recordPath:t,lastActivityMs:i.mtimeMs,freshMs:Date.now()-i.mtimeMs}}catch{return{recordPath:t,lastActivityMs:null,freshMs:null}}}firstKnownCursorSessionId(){for(const e of this.activeBySession.keys()){const t=this.sessionRuntime.get(e);if(t?.cursorSessionId)return t.cursorSessionId}for(const e of this.sessionRuntime.values())if(e.cursorSessionId)return e.cursorSessionId;return null}getUsageSnapshot(e){return this.lastUsageBySession.get(e)??null}getRateLimitsSnapshot(){return{adapterType:"cursor",available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null}}buildPromptText(e){let t;return!e.contextMessages||e.contextMessages.length===0?t=e.text:t=`${e.contextMessages.map(s=>`[${s.senderId}] ${s.content}`).join(`
|
|
1
|
+
import{createInterface as x}from"node:readline";import{EventEmitter as B}from"node:events";import{mkdirSync as $,readFileSync as E,writeFileSync as L,readdirSync as H,statSync as W}from"node:fs";import{join as b,resolve as z}from"node:path";import{fileURLToPath as J}from"node:url";import{homedir as P}from"node:os";import{GRIX_PATHS as G,log as f}from"../../core/log/index.js";import{IdentityInjector as X}from"../shared/identity-injector.js";import{buildSimpleProbeReport as K}from"../shared/probe-util.js";import{InternalApiServer as Q}from"../../core/mcp/internal-api-server.js";import{syncDefaultSkillsToDir as Y}from"../../default-skills/index.js";import{resolveCommandPath as C,spawnCommand as T,killProcessGroup as w}from"../../core/runtime/spawn.js";import{SessionBindingStore as V}from"../../core/persistence/session-binding-store.js";import{fetchCursorParameterizedModels as Z}from"./fetch-models.js";import{fetchCursorAccountUsage as ee}from"./fetch-usage.js";import{buildCursorContextWindow as D,resolveCursorContextWindowSize as O}from"./context-window.js";const te=12e4,se=6e5,ie=8192,ne="Summarize this conversation for handoff to a fresh chat. Include: goals, key decisions, important file paths, current status, and next steps. Be concise but complete. Output ONLY the summary text.",N=12e4,j=new Set(["completed","failed","error","errored","canceled","cancelled","rejected"]);function re(l){const e=l.replace(/ToolCall$/i,"").trim();return e?e.charAt(0).toUpperCase()+e.slice(1):l||"tool"}function oe(l){const e=String(l.call_id??l.callId??"").trim(),t=l.tool_call??l.toolCall;if(t&&typeof t=="object"&&!Array.isArray(t)){const i=Object.keys(t);if(i.length>0){const s=i[0],n=t[s],r=n&&typeof n=="object"&&!Array.isArray(n)?n:null;return{toolName:re(s),toolInput:r?.args??r?.arguments??{},callId:e}}}return{toolName:String(l.name??l.tool_name??l.toolName??"tool").trim()||"tool",toolInput:l.args??l.tool_input??l.toolInput??{},callId:e}}function ae(l){if(!l||typeof l!="object"||Array.isArray(l))return null;const e=l,t=String(e.type??"").trim().toLowerCase();if(!t)return null;switch(t){case"tool_call":{const i=String(e.subtype??"").trim().toLowerCase(),s=oe(e);return i==="started"||i==="start"?{type:"tool_call",payload:{tool_name:s.toolName,...s.callId?{tool_call_id:s.callId}:{}}}:(j.has(i),null)}case"error":return{type:"error",payload:{message:String(e.message??e.error??e.result??"cursor error").trim()||"cursor error"}};case"result":{const i=String(e.subtype??"").trim().toLowerCase();return e.is_error===!0||i==="error"?{type:"error",payload:{message:String(e.result??e.message??"cursor run failed").trim()||"cursor run failed"}}:null}default:return null}}function le(l,e){if(!l||typeof l!="object"||Array.isArray(l))return"";const t=l;if(String(t.type??"")!=="assistant")return"";const i=t.model_call_id??t.modelCallId;if(i!=null&&String(i).trim()!=="")return"";const s=t.message,n=s&&typeof s=="object"&&!Array.isArray(s)?s.content:void 0;if(!Array.isArray(n))return"";const r=n.map(a=>{if(!a||typeof a!="object"||Array.isArray(a))return"";const o=a;return o.type==="text"?String(o.text??""):""}).filter(Boolean).join("");if(!r)return"";if(e){if(r===e||e.startsWith(r)&&r.length<e.length)return"";if(r.startsWith(e))return r.slice(e.length)}return r}const U=[{id:"approval",displayName:"\u4EBA\u5DE5\u786E\u8BA4"},{id:"full_auto",displayName:"\u81EA\u7531\u6A21\u5F0F"},{id:"plan",displayName:"\u8BA1\u5212\u6A21\u5F0F"}];function R(l){const e=String(l??"").trim().toLowerCase();return U.some(t=>t.id===e)?e:null}function de(l){try{return process.kill(l,0),!0}catch(e){return e.code==="EPERM"}}function q(l,e,t){if(t<0)return null;let i;try{i=H(l,{withFileTypes:!0})}catch{return null}for(const s of i){const n=b(l,s.name);if(s.isDirectory()){const r=q(n,e,t-1);if(r)return r}else if(s.isFile()&&s.name.endsWith(e))return n}return null}function ce(l,e){if(!l)return null;const t=e||b(P(),".cursor","projects");return q(t,`${l}.jsonl`,6)}function ue(l){const e=l.context_messages_json;if(e)try{const t=JSON.parse(e);return!Array.isArray(t)||t.length===0?void 0:t.map(i=>({senderId:String(i.sender_id??"unknown"),content:String(i.content??"")}))}catch{return}}function me(l,e){const t=[String(l??"")],i=String(e.quoted_message_id??"").trim();i&&t.push(`[quoted_message_id] ${i}`);const s=String(e.attachments_json??"").trim();return s&&t.push(`[attachments_json]
|
|
2
|
+
${s}`),t.filter(n=>n.length>0).join(`
|
|
3
|
+
|
|
4
|
+
`)}class F extends B{adapterSessionId;constructor(e){super(),this.adapterSessionId=e}emitDone(e){this.emit("done",e)}emitError(e){if(this.listenerCount("error")===0){f.warn("cursor-adapter",`Prompt handle error (no listeners): ${e.message}`);return}this.emit("error",e)}async cancel(){}}class M extends B{type="cursor";static STDERR_MAX_CHARS=8192;config;callbacks;alive=!1;stopped=!1;permissionHandler=null;internalApi=null;bindingStore=null;identity;inboundQueue=[];pendingBySession=new Map;sessions=new Set;sessionRuntime=new Map;sessionSeq=0;lastUsageBySession=new Map;lastContextBySession=new Map;pendingHandoffBySession=new Map;cachedAccountRateLimits=null;rateLimitsInflight=null;_availableModels=[];activeBySession=new Map;constructor(e,t){super(),this.config=e,this.callbacks=t;const i=e.options??{};this.bindingStore=i.bindingStore instanceof V?i.bindingStore:null,this.identity=new X("cursor-adapter",t.getAgentProfile)}async start(){this.alive=!0,this.stopped=!1,(this.config.options??{}).mcp_tools!==!1&&(this.internalApi=new Q,this.internalApi.setInvokeHandler(async(t,i,s,n)=>this.callbacks.agentInvoke(t,i,n)),await this.internalApi.start()),this.notifyBindingReady(),this.refreshModels().then(()=>this.notifyBindingReady()).catch(()=>{}),this.refreshAccountRateLimits().then(()=>this.notifyBindingReady()).catch(()=>{})}get availableModels(){return this._availableModels}get availableModes(){return U.map(e=>({...e}))}async refreshModels(){try{const e={...process.env,...this.config.env??{}},t=await Z({command:this.config.command,env:e,cwd:process.cwd()});t.length>0&&(this._availableModels=t,f.info("cursor-adapter",`Loaded ${t.length} available models`))}catch(e){f.warn("cursor-adapter",`Failed to refresh models: ${e instanceof Error?e.message:String(e)}`)}}async stop(){this.stopped=!0,this.alive=!1;for(const e of this.activeBySession.values())w(e.child,"SIGTERM");this.activeBySession.clear(),this.internalApi&&(await this.internalApi.stop(),this.internalApi=null)}isAlive(){return this.alive}async createSession(e){const t=`cursor-${Date.now()}-${++this.sessionSeq}`;return this.sessions.add(t),this.sessionRuntime.set(t,{cwd:typeof e.cwd=="string"?e.cwd:void 0,modelId:typeof e.modelId=="string"?e.modelId:void 0,modeId:typeof e.modeId=="string"?e.modeId:void 0}),t}async resumeSession(e,t){this.sessions.add(e);const i=this.sessionRuntime.get(e)??{};this.sessionRuntime.set(e,{...i,cwd:typeof t?.cwd=="string"?t.cwd:i.cwd,modelId:typeof t?.modelId=="string"?t.modelId:i.modelId,modeId:typeof t?.modeId=="string"?t.modeId:i.modeId})}onAgentProfileChanged(){this.identity.onProfileChanged()}async destroySession(e){this.sessions.delete(e),this.sessionRuntime.delete(e),this.identity.forgetSession(e)}sendPrompt(e){const t=new F(e.adapterSessionId);if(!this.alive||this.stopped)return queueMicrotask(()=>t.emitError(new Error("adapter not running"))),t;const i=this.inboundQueue.shift()??{event_id:`cursor-evt-${Date.now()}`,session_id:e.adapterSessionId,content:e.text},s=e.adapterSessionId,n=this.pendingBySession.get(s)??[];return n.push({event:i,request:e,handle:t}),this.pendingBySession.set(s,n),this.tryStartNext(s),t}tryStartNext(e){if(this.activeBySession.has(e)||!this.alive||this.stopped)return;const t=this.pendingBySession.get(e);if(!t||t.length===0)return;const i=t.shift();t.length===0?this.pendingBySession.delete(e):this.pendingBySession.set(e,t),i&&this.startJob(i,!1,0)}startJob(e,t,i){const{event:s,request:n,handle:r}=e,a=n.adapterSessionId,o=this.sessionRuntime.get(n.adapterSessionId)??{},d=this.config.options??{},u=[...this.config.args??[]];u.push("-p","--output-format","stream-json"),d.trust!==!1&&u.push("--trust");const m=o.cwd||d.workspace;this.internalApi&&m&&(this.ensureWorkspaceMcpAndSkills(m),u.push("--approve-mcps")),m&&u.push("--workspace",m);const v=o.modelId||d.model;v&&u.push("--model",v);const _=R(o.modeId||d.mode)??"full_auto";_==="full_auto"?u.push("--yolo"):_==="plan"&&u.push("--plan");const y=o.cursorSessionId??this.bindingStore?.getAcpSessionId(n.adapterSessionId),A=!!(y&&d.use_continue!==!1&&!t);A&&u.push("--resume",y),u.push(this.buildPromptText(n));const c={...process.env,...this.config.env??{}},S=C(this.config.command,typeof c.PATH=="string"?c.PATH:void 0);f.info("cursor-adapter",`job start: event=${s.event_id} session=${s.session_id} retry=${i}`);let p;try{p=T(S,u,{cwd:m||process.cwd(),env:c,stdio:["ignore","pipe","pipe"]}).process}catch(g){this.finishActive(a,"failed",`spawn failed: ${g instanceof Error?g.message:String(g)}`);return}this.callbacks.sendEventAck(s.event_id,s.session_id),this.emit("eventStarted",s.event_id,s.session_id);const h={event:s,request:n,handle:r,child:p,seq:0,done:!1,stderr:"",timer:null,idleTimer:null,toolCallsInFlight:new Set,assistantTextEmitted:"",streamBuffer:"",retryCount:i,usedContinue:A,workspace:m||process.cwd(),args:u};this.activeBySession.set(a,h),this.notifyBindingReady(a),this.armActiveIdleTimer(a);const I=n.timeoutMs&&n.timeoutMs>0?n.timeoutMs:0;I>0&&(h.timer=setTimeout(()=>{h.done||(w(p,"SIGTERM"),this.finishActive(a,"failed",`cursor agent timeout after ${I}ms`))},I)),x({input:p.stdout}).on("line",g=>this.handleStdoutLineForActive(h,g)),p.stderr?.on("data",g=>{const k=h.stderr+String(g??"");h.stderr=k.length>M.STDERR_MAX_CHARS?k.slice(-M.STDERR_MAX_CHARS):k}),p.once("error",g=>{this.finishActive(a,"failed",`spawn failed: ${String(g?.message??g)}`)}),p.once("close",g=>{if(!h.done)if((g??0)===0)this.finishActive(a,"responded");else{if(this.shouldRetryWithoutContinue(h)){h.timer&&clearTimeout(h.timer),h.idleTimer&&clearTimeout(h.idleTimer),this.activeBySession.delete(a),this.startJob(e,!0,h.retryCount+1);return}const k=h.stderr.trim()||`cursor agent exited with code ${g??-1}`;this.finishActive(a,"failed",k)}})}async cancel(e){const t=this.activeBySession.get(e);t&&(w(t.child,"SIGTERM"),this.finishActive(e,"canceled","canceled"))}deliverInboundEvent(e){const t=String(e.session_id??"").trim();if(!t){f.warn("cursor-adapter",`Dropping event ${e.event_id}: missing session_id`),this.callbacks.sendEventResult(e.event_id,"failed","missing session_id");return}if(!this.alive||this.stopped){f.warn("cursor-adapter",`Dropping event ${e.event_id}: adapter not running`),this.callbacks.sendEventAck(e.event_id,t),this.callbacks.sendEventResult(e.event_id,"failed","adapter not running");return}const i={adapterSessionId:t,text:me(String(e.content??""),e),contextMessages:ue(e)},s=new F(t),n=this.pendingBySession.get(t)??[];n.push({event:e,request:i,handle:s}),this.pendingBySession.set(t,n),f.info("cursor-adapter",`inbound queued: event=${e.event_id} session=${t} depth=${n.length}`),this.tryStartNext(t)}cancelSessionWork(e,t){const i=this.pendingBySession.get(e)??[];this.pendingBySession.delete(e);const s=this.activeBySession.get(e);s&&(this.killChildWithFallback(s.child),this.finishActive(e,"canceled",t));for(const n of i)this.callbacks.sendEventAck(n.event.event_id,n.event.session_id),this.callbacks.sendEventResult(n.event.event_id,"canceled",t),n.handle.emitDone({status:"canceled",error:t})}killChildWithFallback(e){w(e,"SIGTERM"),setTimeout(()=>{if(e.exitCode===null&&e.signalCode===null)try{w(e,"SIGKILL")}catch{}},5e3).unref()}deliverStopEvent(e,t){if(t){const i=this.activeBySession.get(t);if(i?.event.event_id===e){this.killChildWithFallback(i.child),this.finishActive(t,"canceled","stopped");return}const s=this.pendingBySession.get(t)??[],n=s.findIndex(r=>r.event.event_id===e);if(n>=0){const[r]=s.splice(n,1);s.length===0?this.pendingBySession.delete(t):this.pendingBySession.set(t,s),this.callbacks.sendEventAck(r.event.event_id,r.event.session_id),this.callbacks.sendEventResult(r.event.event_id,"canceled","stopped"),r.handle.emitDone({status:"canceled",error:"stopped"})}return}for(const[i,s]of this.activeBySession.entries())if(s.event.event_id===e){this.killChildWithFallback(s.child),this.finishActive(i,"canceled","stopped");return}for(const[i,s]of this.pendingBySession.entries()){const n=s.findIndex(a=>a.event.event_id===e);if(n<0)continue;const[r]=s.splice(n,1);s.length===0?this.pendingBySession.delete(i):this.pendingBySession.set(i,s),this.callbacks.sendEventAck(r.event.event_id,r.event.session_id),this.callbacks.sendEventResult(r.event.event_id,"canceled","stopped"),r.handle.emitDone({status:"canceled",error:"stopped"});return}}async handleLocalAction(e){const t=String(e.action_type??"").trim().toLowerCase(),i=e.params??{},s=String(i.session_id??"").trim(),n=e.action_id;switch(t){case"set_model":{const r=String(i.model_id??"").trim();if(!r||!s)return this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_params","model_id and session_id are required"),{handled:!0,kind:"set_model"};const a=this.sessionRuntime.get(s)??{};this.sessionRuntime.set(s,{...a,modelId:r}),this.notifyBindingReady(s);const o=this.getToolbarMeta(s);return this.callbacks.sendLocalActionResult(n,"ok",{outcome:"model_set",session_context:{model_id:r,mode_id:o.mode_id,modelId:r,modeId:o.mode_id},model_id:r,mode_id:o.mode_id,available_models:this.availableModels,available_modes:this.availableModes}),{handled:!0,kind:"set_model"}}case"set_mode":{const r=R(i.mode_id);if(!r||!s)return this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_params","mode_id must be approval, full_auto, or plan and session_id is required"),{handled:!0,kind:"set_mode"};const a=this.sessionRuntime.get(s)??{};this.sessionRuntime.set(s,{...a,modeId:r}),this.notifyBindingReady(s);const o=this.getToolbarMeta(s);return this.callbacks.sendLocalActionResult(n,"ok",{outcome:"mode_set",session_context:{model_id:o.model_id,mode_id:r,modelId:o.model_id,modeId:r},model_id:o.model_id,mode_id:r,available_models:this.availableModels,available_modes:this.availableModes}),{handled:!0,kind:"set_mode"}}case"get_context":{const r=s?this.sessionRuntime.get(s):void 0,a=this.config.options??{},o=this.getToolbarMeta(s);return this.callbacks.sendLocalActionResult(n,"ok",{session_context:{model_id:o.model_id,mode_id:o.mode_id,cwd:r?.cwd??a.workspace??null},model_id:o.model_id,mode_id:o.mode_id,modelId:o.model_id,modeId:o.mode_id,cwd:r?.cwd??a.workspace??null,available_models:this.availableModels,available_modes:this.availableModes}),{handled:!0,kind:"get_context"}}case"get_rate_limits":{await this.refreshAccountRateLimits({force:!0});const r=this.getRateLimitsSnapshot(s||void 0);return this.callbacks.sendLocalActionResult(n,"ok",r),s&&this.notifyBindingReady(s),{handled:!0,kind:"get_rate_limits"}}case"get_session_usage":{const r=s?this.getUsageSnapshot(s):null;return r?this.callbacks.sendLocalActionResult(n,"ok",{adapterType:"cursor",available:!0,sampledAt:r.sampledAt,turns:r.turns,tokenUsage:r.total}):this.callbacks.sendLocalActionResult(n,"ok",{adapterType:"cursor",available:!1,sampledAt:null,turns:0,tokenUsage:null}),{handled:!0,kind:"get_session_usage"}}case"session_control":{const r=String(i.verb??"").trim().toLowerCase();if(r==="restart"&&s)this.cancelSessionWork(s,"restarted"),this.callbacks.sendLocalActionResult(n,"ok",{outcome:"restarted"});else if(r==="status"){const a=this.sessionRuntime.get(s),o=this.config.options??{},d=this.getToolbarMeta(s),u=this.activeBySession.has(s);this.callbacks.sendLocalActionResult(n,"ok",{verb:"status",status:u?"running":"idle",session_context:{model_id:d.model_id,mode_id:d.mode_id,cwd:a?.cwd??o.workspace??null},model_id:d.model_id,mode_id:d.mode_id,modelId:d.model_id,modeId:d.mode_id,cwd:a?.cwd??o.workspace??null,available_models:this.availableModels,available_modes:this.availableModes})}else this.callbacks.sendLocalActionResult(n,"failed",void 0,"invalid_verb",`Unsupported verb: ${r}`);return{handled:!0,kind:"session_control"}}default:return{handled:!1,kind:"unsupported"}}}async execCommand(e,t,i){return e!=="compact"?{status:"unsupported",message:`Unsupported command: ${e}`}:this.runCompactHandoff(i)}async runCompactHandoff(e){if(!e)return{status:"failed",message:"session_id is required"};if(this.activeBySession.has(e))return{status:"failed",message:"Cannot compact while a Cursor turn is active"};const t=this.sessionRuntime.get(e)??{},i=this.config.options??{},s=t.cursorSessionId??this.bindingStore?.getAcpSessionId(e)??null;if(!s)return{status:"failed",message:"No Cursor chat to compact (send a message first)"};const n=String(t.cwd||i.workspace||process.cwd()),r=String(t.modelId||i.model||""),a=R(t.modeId||i.mode)??"full_auto";f.info("cursor-adapter",`[compact] summarize resume=${s} session=${e}`);const o=await this.runPrintOnce({prompt:ne,resumeChatId:s,cwd:n,modelId:r||void 0,modeId:a,timeoutMs:N});if(!o.ok||!o.text.trim())return{status:"failed",message:o.error??"Summarize failed (empty result)"};const d=await this.createEmptyChat();return d?(this.sessionRuntime.set(e,{...t,cwd:n,modelId:r||t.modelId,modeId:a,cursorSessionId:d}),this.bindingStore?.setAcpSessionId(e,d),this.pendingHandoffBySession.set(e,o.text.trim()),this.lastContextBySession.set(e,D(0,O(r))),this.notifyBindingReady(e),f.info("cursor-adapter",`[compact] handoff ready: old=${s} new=${d} summaryChars=${o.text.trim().length}`),{status:"ok",message:"Context compacted via summarize + new chat",data:{previousChatId:s,chatId:d}}):{status:"failed",message:"agent create-chat failed"}}runPrintOnce(e){const t={...process.env,...this.config.env??{}},i=C(this.config.command,typeof t.PATH=="string"?t.PATH:void 0),s=[...this.config.args??[]];s.push("-p","--output-format","stream-json"),this.config.options?.trust!==!1&&s.push("--trust"),e.cwd&&s.push("--workspace",e.cwd),e.modelId&&s.push("--model",e.modelId);const n=R(e.modeId)??"full_auto";return n==="full_auto"?s.push("--yolo"):n==="plan"&&s.push("--plan"),e.resumeChatId&&s.push("--resume",e.resumeChatId),s.push(e.prompt),new Promise(r=>{let a;try{a=T(i,s,{cwd:e.cwd||process.cwd(),env:t,stdio:["ignore","pipe","pipe"]}).process}catch(c){r({ok:!1,text:"",error:`spawn failed: ${c instanceof Error?c.message:String(c)}`});return}let o="",d="",u="",m=!1;const v=e.timeoutMs&&e.timeoutMs>0?e.timeoutMs:N,_=setTimeout(()=>{m||(m=!0,this.killChildWithFallback(a),r({ok:!1,text:o,sessionId:d||void 0,error:`compact timeout after ${v}ms`}))},v);_.unref?.();const y=(c,S)=>{m||(m=!0,clearTimeout(_),r({ok:c,text:o,sessionId:d||void 0,error:S}))};x({input:a.stdout}).on("line",c=>{const S=c.trim();if(!S)return;let p;try{p=JSON.parse(S)}catch{return}if(p?.type!=="result")return;const h=typeof p?.result=="string"?p.result:"";h&&(o=h);const I=String(p?.session_id??"").trim();I&&(d=I)}),a.stderr?.on("data",c=>{u+=typeof c=="string"?c:c.toString("utf8")}),a.on("error",c=>y(!1,c.message)),a.on("close",c=>{if(!m){if(o.trim()){y(!0);return}y(!1,u.trim()||`cursor agent exited with code ${c??"unknown"}`)}})})}createEmptyChat(){const e={...process.env,...this.config.env??{}},t=C(this.config.command,typeof e.PATH=="string"?e.PATH:void 0);return new Promise(i=>{let s;try{s=T(t,["create-chat"],{cwd:process.cwd(),env:e,stdio:["ignore","pipe","pipe"]}).process}catch{i(null);return}let n="",r=!1;const a=setTimeout(()=>{r||(r=!0,this.killChildWithFallback(s),i(null))},3e4);a.unref?.(),s.stdout?.on("data",o=>{n+=typeof o=="string"?o:o.toString("utf8")}),s.on("error",()=>{r||(r=!0,clearTimeout(a),i(null))}),s.on("close",o=>{if(r)return;r=!0,clearTimeout(a);const d=n.trim().split(/\s+/)[0]??"";i(o===0&&/^[0-9a-f-]{16,}$/i.test(d)?d:null)})})}setPermissionHandler(e){this.permissionHandler=e}async ping(e){if(!this.alive||this.stopped||(this.config.options??{}).mcp_tools!==!1&&this.internalApi===null)return!1;for(const i of this.activeBySession.values()){const s=i.child?.pid;if(s&&!de(s))return!1}return!0}getStatus(){let e=0;for(const t of this.pendingBySession.values())e+=t.length;return{alive:this.alive,busy:this.activeBySession.size>0,sessions:this.sessions.size,details:{queueDepth:e+this.inboundQueue.length,activeSessions:this.activeBySession.size}}}getActiveEventIds(){const e=[];for(const t of this.activeBySession.values())t.event.event_id&&e.push(t.event.event_id);return e}clearActiveEventForShutdown(){this.activeBySession.clear()}getMcpConfig(){return null}async probe(e){const t=this.getStatus();return{...await K(this.config.command||"cursor-agent",{alive:t.alive,busy:t.busy,started:this.alive},e),session:this.probeSessionRecord()}}probeSessionRecord(){const e=this.firstKnownCursorSessionId(),t=e?ce(e):null;if(!t)return{recordPath:null,lastActivityMs:null,freshMs:null};try{const i=W(t);return{recordPath:t,lastActivityMs:i.mtimeMs,freshMs:Date.now()-i.mtimeMs}}catch{return{recordPath:t,lastActivityMs:null,freshMs:null}}}firstKnownCursorSessionId(){for(const e of this.activeBySession.keys()){const t=this.sessionRuntime.get(e);if(t?.cursorSessionId)return t.cursorSessionId}for(const e of this.sessionRuntime.values())if(e.cursorSessionId)return e.cursorSessionId;return null}getUsageSnapshot(e){return this.lastUsageBySession.get(e)??null}getRateLimitsSnapshot(e){const t=this.cachedAccountRateLimits,i=e?this.lastContextBySession.get(e)??null:null;if(!t)return{adapterType:"cursor",available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:i,tokenUsage:null};const s=Date.parse(t.sampledAt);return{adapterType:"cursor",available:t.available&&!!t.rateLimits,cached:!0,sampledAt:Number.isFinite(s)?s:Date.now(),rateLimits:t.rateLimits,contextWindow:i,tokenUsage:null,...t.displayMessage?{displayMessage:t.displayMessage}:{},...t.error?{error:t.error}:{}}}async refreshAccountRateLimits(e){const t=e?.force===!0,i=this.cachedAccountRateLimits,s=i?Date.now()-Date.parse(i.sampledAt):Number.POSITIVE_INFINITY;if(!t&&i&&Number.isFinite(s)&&s<6e4||this.rateLimitsInflight&&(await this.rateLimitsInflight,!t))return;const n=(async()=>{const r=await ee();this.cachedAccountRateLimits={available:r.available,sampledAt:r.sampledAt,rateLimits:r.rateLimits,displayMessage:r.displayMessage,error:r.error},r.available&&r.rateLimits?f.info("cursor-adapter",`[rate-limits] monthly=${r.rateLimits.fiveHour.usedPercentage.toFixed(1)}% api=${r.rateLimits.sevenDay.usedPercentage.toFixed(1)}% resetsAt=${r.rateLimits.fiveHour.resetsAt}`+(r.displayMessage?` msg=${r.displayMessage}`:"")):f.warn("cursor-adapter",`[rate-limits] unavailable: ${r.error??"unknown"}`)})();this.rateLimitsInflight=n.finally(()=>{this.rateLimitsInflight=null}),await this.rateLimitsInflight}getToolbarMeta(e){const t=this.config.options??{},i=e||String(t.aibotSessionId??""),s=i?this.sessionRuntime.get(i):void 0,n=String(s?.modelId||t.model||"auto"),r=R(s?.modeId||t.mode)??"full_auto",a={model_id:n,mode_id:r,currentModelId:n,currentModeId:r,available_models:this.availableModels,available_modes:this.availableModes},o=this.cachedAccountRateLimits?.rateLimits;if(o?.fiveHour||o?.sevenDay){const u=Date.parse(this.cachedAccountRateLimits?.sampledAt??"")||Date.now();a.rate_limits={...o.fiveHour?{fiveHour:o.fiveHour}:{},...o.sevenDay?{sevenDay:o.sevenDay}:{},sampledAt:u}}const d=i?this.lastContextBySession.get(i):void 0;return d&&(a.context_window={usedPercentage:d.usedPercentage,remainingPercentage:d.remainingPercentage,usedTokens:d.usedTokens,sizeTokens:d.sizeTokens}),a}notifyBindingReady(e){if(!this.callbacks.sendUpdateBindingCard)return;const t=this.config.options??{},i=e||String(t.aibotSessionId??"");if(!i)return;const s=this.sessionRuntime.get(i),n=String(s?.cwd||t.workspace||"");n&&this.callbacks.sendUpdateBindingCard(i,this.activeBySession.has(i)?"busy":"ready",n,this.getToolbarMeta(i))}buildPromptText(e){let t;!e.contextMessages||e.contextMessages.length===0?t=e.text:t=`${e.contextMessages.map(n=>`[${n.senderId}] ${n.content}`).join(`
|
|
3
5
|
`)}
|
|
4
6
|
|
|
5
7
|
[Current user message]
|
|
6
|
-
${e.text}
|
|
7
|
-
|
|
8
|
-
|
|
8
|
+
${e.text}`;const i=this.pendingHandoffBySession.get(e.adapterSessionId);return i&&(this.pendingHandoffBySession.delete(e.adapterSessionId),t=`[Previous conversation summary \u2014 continue from this handoff]
|
|
9
|
+
${i}
|
|
10
|
+
|
|
11
|
+
[Current user message]
|
|
12
|
+
${t}`),this.identity.injectOnce(e.adapterSessionId,t)}handleStdoutLineForActive(e,t){if(e.done)return;const i=t.trim();if(!i)return;const{event:s}=e;let n;try{n=JSON.parse(i)}catch{this.armActiveIdleTimer(e.request.adapterSessionId);return}if(n?.type==="tool_call"){const o=String(n?.call_id??""),d=String(n?.subtype??"");o&&(d==="started"?(this.flushAssistantStreamBuffer(e),e.toolCallsInFlight.add(o)):j.has(d)&&e.toolCallsInFlight.delete(o))}this.armActiveIdleTimer(e.request.adapterSessionId);const r=ae(n);r&&this.callbacks.sendRawEventEnvelope?.(s.event_id,s.session_id,r);const a=le(n,e.assistantTextEmitted);if(a&&this.enqueueAssistantStreamDelta(e,a),n?.type==="result"){const o=typeof n?.result=="string"?n.result.trim():"";o&&(e.assistantTextEmitted?o.startsWith(e.assistantTextEmitted)&&o.length>e.assistantTextEmitted.length&&this.enqueueAssistantStreamDelta(e,o.slice(e.assistantTextEmitted.length)):this.enqueueAssistantStreamDelta(e,o)),this.flushAssistantStreamBuffer(e);const d=String(n?.session_id??"").trim();if(d){const c=this.sessionRuntime.get(e.request.adapterSessionId)??{};this.sessionRuntime.set(e.request.adapterSessionId,{...c,cursorSessionId:d}),this.bindingStore?.setAcpSessionId(e.request.adapterSessionId,d)}const u=n?.usage??{},m=this.lastUsageBySession.get(e.request.adapterSessionId);this.lastUsageBySession.set(e.request.adapterSessionId,{sampledAt:new Date().toISOString(),turns:(m?.turns??0)+1,total:{input:(m?.total.input??0)+Number(u.inputTokens??0),output:(m?.total.output??0)+Number(u.outputTokens??0),cacheRead:(m?.total.cacheRead??0)+Number(u.cacheReadTokens??0),cacheWrite:(m?.total.cacheWrite??0)+Number(u.cacheWriteTokens??0)}});const v=Number(u.inputTokens??0),_=Number(u.cacheReadTokens??0),y=Number(u.cacheWriteTokens??0),A=[v,_,y].map(c=>Number.isFinite(c)&&c>0?c:0).reduce((c,S)=>c+S,0);if(A>0){const c=String((this.sessionRuntime.get(e.request.adapterSessionId)?.modelId??this.config.options?.model??"")||"");this.lastContextBySession.set(e.request.adapterSessionId,D(A,O(c)))}}}enqueueAssistantStreamDelta(e,t){t&&(e.assistantTextEmitted+=t,e.streamBuffer+=t,e.streamBuffer.length>=ie&&this.flushAssistantStreamBuffer(e))}flushAssistantStreamBuffer(e){if(!e.streamBuffer)return;const t=e.streamBuffer;e.streamBuffer="",e.seq+=1,this.callbacks.sendStreamChunk(e.event.event_id,e.event.session_id,t,e.seq,!1,void 0,void 0)}armActiveIdleTimer(e){const t=this.activeBySession.get(e);if(!t||t.done)return;t.idleTimer&&clearTimeout(t.idleTimer);const i=t.toolCallsInFlight.size>0?se:te;t.idleTimer=setTimeout(()=>{t.done||(f.error("cursor-adapter",`Idle timeout (${i/1e3}s, toolsInFlight=${t.toolCallsInFlight.size}) \u2014 killing cursor child: event=${t.event.event_id} session=${e}`),this.killChildWithFallback(t.child),this.finishActive(e,"failed",`idle timeout after ${i/1e3}s`))},i),t.idleTimer.unref?.()}finishActive(e,t,i){const s=this.activeBySession.get(e);if(!s)return;s.done=!0,s.timer&&clearTimeout(s.timer),s.idleTimer&&clearTimeout(s.idleTimer),this.flushAssistantStreamBuffer(s),s.seq+=1;const n=String(s.event.quoted_message_id??"").trim()||void 0;this.callbacks.sendStreamChunk(s.event.event_id,s.event.session_id,"",s.seq,!0,void 0,n),f.info("cursor-adapter",`job finish: event=${s.event.event_id} session=${s.event.session_id} status=${t}${i?` msg=${i}`:""}`),this.emit("eventDone",s.event.event_id),this.callbacks.sendEventResult(s.event.event_id,t,i),t==="responded"?s.handle.emitDone({status:"completed"}):t==="canceled"?s.handle.emitDone({status:"canceled",error:i}):s.handle.emitDone({status:"failed",error:i}),this.activeBySession.delete(e),this.notifyBindingReady(e),this.tryStartNext(e)}shouldRetryWithoutContinue(e){if(!e.usedContinue||e.retryCount>0)return!1;const t=e.stderr.toLowerCase(),i=t.includes("resume")||t.includes("continue")||t.includes("chat")||t.includes("session"),s=t.includes("not found")||t.includes("no previous session")||t.includes("does not exist")||t.includes("invalid");return i&&s}ensureWorkspaceMcpAndSkills(e){if(!this.internalApi)return;this.ensureCursorMcpConfig(e,this.internalApi.url);const t=b(P(),".cursor","skills"),i=Y(t);i.length>0&&f.info("cursor-adapter",`Synced connector skills to ${t}: [${i.join(", ")}]`)}ensureCursorMcpConfig(e,t){try{const i=b(e,".cursor"),s=b(i,"mcp.json");$(i,{recursive:!0});let n={};try{n=JSON.parse(E(s,"utf8"))}catch{n={}}const r={command:process.execPath,args:[z(J(import.meta.url),"../../../mcp/acp-mcp-server.js"),"--api-url",t]};(!n.mcpServers||typeof n.mcpServers!="object")&&(n.mcpServers={}),n.mcpServers.grix=r,L(s,`${JSON.stringify(n,null,2)}
|
|
13
|
+
`,"utf8"),this.recordCursorMcpRegistry(e,s,r),f.info("cursor-adapter",`MCP config synced: workspace=${e} file=${s}`)}catch(i){f.warn("cursor-adapter",`Failed to ensure .cursor/mcp.json: ${String(i)}`)}}recordCursorMcpRegistry(e,t,i){try{const s=b(G.data,"cursor"),n=b(s,"mcp-registry.json");$(s,{recursive:!0});let r={};try{r=JSON.parse(E(n,"utf8"))}catch{r={}}r[e]={mcp_path:t,server:i,updated_at:new Date().toISOString()},L(n,`${JSON.stringify(r,null,2)}
|
|
14
|
+
`,"utf8")}catch(s){f.warn("cursor-adapter",`Failed to record MCP registry: ${String(s)}`)}}}export{U as CURSOR_MODE_OPTIONS,M as CursorAdapter,le as extractCursorAssistantStreamDelta,ce as findCursorTranscript,ae as mapCursorStreamJsonToRawEnvelope};
|
|
@@ -0,0 +1,2 @@
|
|
|
1
|
+
import{JsonRpcTransport as x}from"../../core/transport/json-rpc.js";import{log as d}from"../../core/log/index.js";import{killProcessGroup as g,resolveCommandPath as h,spawnCommand as w}from"../../core/runtime/spawn.js";const y=3e4,u="none|low|medium|high|xhigh|extra-high|max";function E(o,e){const t=o.trim(),r=e.trim();return t==="default"||t==="default[]"||/^default\[.*\]$/.test(t)?{id:"auto",displayName:!r||r==="default"||r==="Auto"?"Auto (default)":r}:{id:t,displayName:r||t}}function v(o){const e=o.trim();if(!e||e==="auto"||e==="default"||e==="default[]")return"auto";let t=e;return t=t.replace(new RegExp(`-thinking-(?:${u})(?:-fast)?$`),""),t=t.replace(new RegExp(`-(?:${u})-thinking$`),""),t=t.replace(/-thinking(?:-fast)?$/,""),t=t.replace(new RegExp(`-(?:${u})(?:-fast)?$`),""),t=t.replace(/-fast$/,""),t||e}function M(o){let e=o.trim();e=e.replace(/\s*\(default\)\s*$/i,""),e=e.replace(/\s*\(NO ZDR\)\s*$/i,"");const t=/(?:\s+(?:None|Low|Medium|High|Extra High|Max|Fast|Thinking|1M))$/i;for(let r=0;r<8&&t.test(e);r++)e=e.replace(t,"").trim();return e||o.trim()}function T(o){const e=[];for(const t of o.split(`
|
|
2
|
+
`)){const r=t.trim();if(!r||r.toLowerCase().includes("available model")||r.toLowerCase().startsWith("tip:"))continue;const s=r.match(/^(\S+)\s+-\s+(.+)$/);s&&e.push({id:s[1],displayName:s[2].trim()})}return e}function $(o){const e=new Set,t=[];for(const r of o){const s=v(r.id);if(e.has(s))continue;e.add(s);const n=s==="auto"?"Auto (default)":M(r.displayName);t.push({id:s,displayName:n})}return t}async function A(o){return new Promise(e=>{let t="";o.stdout?.on("data",r=>{t+=String(r)}),o.once("close",()=>e(t)),o.once("error",()=>e(t))})}async function I(o){const e={...process.env,...o.env??{}},t=h(o.command,typeof e.PATH=="string"?e.PATH:void 0),{process:r}=w(t,["models"],{cwd:o.cwd??process.cwd(),env:e,stdio:["ignore","pipe","pipe"]}),s=await A(r);return $(T(s))}async function b(o){const e=o.timeoutMs??y,t=o.cwd??process.cwd(),r={...process.env,...o.env??{}},s=h(o.command,typeof r.PATH=="string"?r.PATH:void 0),{process:n}=w(s,["acp"],{cwd:t,env:r,stdio:["pipe","pipe","pipe"]});if(!n.stdout||!n.stdin)throw g(n,"SIGTERM"),new Error("cursor acp spawn missing stdio pipes");const a=new x(n.stdout,n.stdin);a.setHandlers(void 0,(l,m)=>{String(l).startsWith("cursor/")?a.respondSuccess(m,{}):a.respondError(m,-32601,"method not implemented")});const c=new AbortController,C=setTimeout(()=>c.abort(),e),p=l=>{c.abort(l)};n.once("error",p);try{await a.call("initialize",{protocolVersion:1,clientCapabilities:{fs:{readTextFile:!1,writeTextFile:!1}},clientInfo:{name:"grix-connector",version:"0"}},c.signal);const f=((await a.call("session/new",{cwd:t,mcpServers:[]},c.signal)).models?.availableModels??[]).filter(i=>typeof i.modelId=="string"&&i.modelId.length>0).map(i=>E(i.modelId,i.name??i.modelId));if(f.length===0)throw new Error("cursor acp session/new returned no models");return f}finally{clearTimeout(C),n.off("error",p);try{a.close()}catch{}g(n,"SIGTERM")}}async function L(o){try{const e=await b(o);return d.info("cursor-adapter",`Loaded ${e.length} parameterized models via acp`),e}catch(e){const t=e instanceof Error?e.message:String(e);d.warn("cursor-adapter",`ACP model probe failed, falling back to collapsed CLI models: ${t}`)}try{const e=await I(o);return e.length>0&&d.info("cursor-adapter",`Loaded ${e.length} collapsed models via CLI fallback`),e}catch(e){const t=e instanceof Error?e.message:String(e);return d.warn("cursor-adapter",`CLI model fallback failed: ${t}`),[]}}export{M as collapseCursorDisplayName,v as collapseCursorModelId,$ as collapseExplodedCursorModels,I as fetchCollapsedModelsFromCli,L as fetchCursorParameterizedModels,b as fetchCursorParameterizedModelsViaAcp,E as normalizeCursorAvailableModel,T as parseExplodedCursorModels};
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
import{execFile as f}from"node:child_process";import{homedir as p,platform as d}from"node:os";import{join as y}from"node:path";import{promisify as b}from"node:util";const u=b(f),g="https://api2.cursor.sh/aiserver.v1.DashboardService/GetCurrentPeriodUsage";function m(e){return Number.isFinite(e)?Math.max(0,Math.min(100,e)):0}function h(e){if(e==null||e==="")return 0;const t=typeof e=="number"?e:Number(String(e).trim());return!Number.isFinite(t)||t<=0?0:t>1e12?Math.floor(t/1e3):Math.floor(t)}function k(e){const t=e.planUsage;if(!t||typeof t!="object")return null;const r=h(e.billingCycleEnd);if(r<=0)return null;const i=m(Number(t.totalPercentUsed??0)),a=t.apiPercentUsed,s=t.autoPercentUsed,c=m(Number(a!=null&&Number.isFinite(Number(a))?a:s??0));return{fiveHour:{usedPercentage:i,resetsAt:r},sevenDay:{usedPercentage:c,resetsAt:r}}}async function S(){if(d()!=="darwin")return null;try{const{stdout:e}=await u("security",["find-generic-password","-s","cursor-access-token","-a","cursor-user","-w"],{timeout:5e3,encoding:"utf8"});return String(e??"").trim()||null}catch{return null}}async function M(){const e=y(p(),"Library","Application Support","Cursor","User","globalStorage","state.vscdb");try{const{stdout:t}=await u("sqlite3",[e,"SELECT value FROM ItemTable WHERE key = 'cursorAuth/accessToken' LIMIT 1;"],{timeout:5e3,encoding:"utf8"});return String(t??"").trim()||null}catch{return null}}async function v(){const e=await S();return e||M()}async function w(e){const t=new Date().toISOString(),r=e&&Object.prototype.hasOwnProperty.call(e,"token")?e.token??null:await v();if(!r)return{available:!1,rateLimits:null,sampledAt:t,error:"no_cursor_access_token"};const i=e?.fetchImpl??fetch,a=e?.timeoutMs??12e3,s=new AbortController,c=setTimeout(()=>s.abort(),a);try{const n=await i(g,{method:"POST",headers:{Authorization:`Bearer ${r}`,"Content-Type":"application/json","Connect-Protocol-Version":"1","x-cursor-client-type":"cli"},body:"{}",signal:s.signal});if(!n.ok)return{available:!1,rateLimits:null,sampledAt:t,error:`http_${n.status}`};const o=await n.json(),l=k(o);return l?{available:!0,rateLimits:l,sampledAt:t,displayMessage:o.displayMessage}:{available:!1,rateLimits:null,sampledAt:t,displayMessage:o.displayMessage,error:"plan_usage_unavailable"}}catch(n){const o=n instanceof Error?n.message:String(n);return{available:!1,rateLimits:null,sampledAt:t,error:o.includes("abort")?"timeout":o}}finally{clearTimeout(c)}}export{w as fetchCursorAccountUsage,k as mapCursorPeriodUsageToRateLimits,v as readCursorAccessToken};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
import{CursorAdapter as e}from"./cursor-adapter.js";export{e as CursorAdapter};
|
|
1
|
+
import{CursorAdapter as e,CURSOR_MODE_OPTIONS as s,extractCursorAssistantStreamDelta as a,mapCursorStreamJsonToRawEnvelope as t}from"./cursor-adapter.js";import{collapseCursorDisplayName as d,collapseCursorModelId as C,collapseExplodedCursorModels as u,fetchCursorParameterizedModels as p,normalizeCursorAvailableModel as i,parseExplodedCursorModels as m}from"./fetch-models.js";import{fetchCursorAccountUsage as n,mapCursorPeriodUsageToRateLimits as x,readCursorAccessToken as f}from"./fetch-usage.js";import{buildCursorContextWindow as A,resolveCursorContextWindowSize as S}from"./context-window.js";export{s as CURSOR_MODE_OPTIONS,e as CursorAdapter,A as buildCursorContextWindow,d as collapseCursorDisplayName,C as collapseCursorModelId,u as collapseExplodedCursorModels,a as extractCursorAssistantStreamDelta,n as fetchCursorAccountUsage,p as fetchCursorParameterizedModels,x as mapCursorPeriodUsageToRateLimits,t as mapCursorStreamJsonToRawEnvelope,i as normalizeCursorAvailableModel,m as parseExplodedCursorModels,f as readCursorAccessToken,S as resolveCursorContextWindowSize};
|
package/dist/bridge/bridge.js
CHANGED
|
@@ -1,16 +1,16 @@
|
|
|
1
|
-
import x from"node:path";import{realpath as ie,stat as I}from"node:fs/promises";import{tmpdir as ne}from"node:os";import{randomUUID as D}from"node:crypto";import{ConnectionManager as se}from"../core/aibot/index.js";import{ClaudeAdapter as P}from"../adapter/claude/index.js";import{CodexAdapter as oe}from"../adapter/codex/index.js";import{PiAdapter as re}from"../adapter/pi/index.js";import{AcpAdapter as A}from"../adapter/acp/index.js";import{OpenHumanAdapter as ae}from"../adapter/openhuman/index.js";import{CursorAdapter as Q}from"../adapter/cursor/index.js";import{CodeWhaleAdapter as N}from"../adapter/codewhale/index.js";import{OpenCodeAdapter as de}from"../adapter/opencode/index.js";import{AgyAdapter as K}from"../adapter/agy/index.js";import{getCachedAgyModels as ce}from"../adapter/agy/model-list.js";import{getCachedAgyQuotaInfo as le}from"../adapter/agy/quota.js";import{LOCAL_ACTION_ERROR_CODES as T,LOCAL_ACTION_TYPES as b,SESSION_CONTROL_ERROR_CODES as g,SESSION_CONTROL_VERBS as p,SESSION_MODE_IDS as C}from"../adapter/claude/protocol-contract.js";import{parseClaudeSessionUsage as ue}from"../adapter/claude/usage-parser.js";import{resolveSkillScanMode as z}from"../adapter/claude/skill-scanner.js";import{buildReportedSkills as he}from"../default-skills/index.js";import{buildWireSkills as G}from"../core/skill-sync/sync-state.js";import{uploadSkill as ge}from"../core/skill-sync/skill-upload.js";import{fetchAvailableModels as B,getCachedModels as F,readSettingsEnv as me}from"../adapter/claude/model-list.js";import{parseAcpSessionUsage as pe}from"../adapter/acp/usage-parser.js";import{applyProxyEnv as fe,getProxyManager as _e,relayHostsForClientType as ve,buildGatewayRoute as Se}from"../core/proxy/index.js";import{buildProviderEnv as be}from"../core/config/provider-env.js";import{sweepNativeProviderResidue as we}from"./native-provider-sweep.js";import{parseCodexSessionUsage as Ce}from"../adapter/codex/usage-parser.js";import{readCodexProviderSettings as Ae}from"../adapter/codex/codex-trust.js";import{resolveKimiProviderSettings as ke}from"../core/provider-quota/index.js";import{scanCodexSessions as Ee}from"../adapter/codex/session-scanner.js";import{scanClaudeSessions as Re}from"../adapter/claude/session-scanner.js";import{reasonixStampFromSessionId as $e,resolveAcpAgentTypes as ye,scanAcpSessions as Pe,scanReasonixSessionTitles as Te}from"../adapter/acp/session-scanner.js";import{scanPiSessions as xe}from"../adapter/pi/session-scanner.js";import{scanCodeWhaleSessions as He}from"../adapter/codewhale/session-scanner.js";import{scanCursorSessions as Le}from"../adapter/cursor/session-scanner.js";import{scanOpenCodeSessions as Me}from"../adapter/opencode/session-scanner.js";import{parsePiSessionUsage as Ie}from"../adapter/pi/usage-parser.js";import{SessionScanCache as k,resolveCodexLeafDirs as De,resolveClaudeLeafDirs as Qe,resolveAcpLeafDirs as Be,resolvePiLeafDirs as Fe,resolveCodeWhaleLeafDirs as Ue,resolveCursorLeafDirs as qe,resolveOpenCodeLeafDirs as Oe,resolveReasonixLeafDirs as We}from"./session-scan-cache.js";import{log as u,ConversationLog as Ne,AgentApiPacketLog as Ke,BridgeEventLog as ze,GRIX_PATHS as j}from"../core/log/index.js";import{RevokeHandler as Ge}from"./revoke-handler.js";import{AdapterPool as je,PoolFullError as Ve}from"./adapter-pool.js";import{parseSessionControlCommand as Je,handleSessionControlCommand as V,handleSessionControlLocalAction as Ye}from"./session-controller.js";import{isOpenSessionDirectiveMessage as Xe}from"../core/protocol/interaction-parser.js";import{handleAcpSetModel as J,handleAcpSetMode as Y,resolveAcpInitialDefaults as Ze}from"./acp-toolbar-persist.js";import{SessionBindingStore as et}from"../core/persistence/session-binding-store.js";import{handleFileListAction as tt,handleCreateFolderAction as it,serveLocalFile as nt,realHomeDir as X}from"../core/files/index.js";import{getMachineName as st}from"../core/util/index.js";import{uploadReplyFileToAgentMedia as ot}from"../core/protocol/agent-api-media.js";import{ActiveEventStore as rt}from"../core/persistence/active-event-store.js";import{DEFAULT_CONNECTOR_RUNTIME_CONFIG as at,applyConnectorRuntimeConfigPatch as dt,extractConnectorRuntimeConfigPatch as ct}from"./runtime-config.js";import{DEFAULT_QUOTE_TRIGGER_BY_ADAPTER as lt,SendController as ut}from"./send-controller.js";import{sharedProviderQuotaService as U,detectProviderFromModel as H,normalizeProviderId as q,providerQuotaToRateLimits as Z,providerQuotaToCodexRateLimits as ee}from"../core/provider-quota/index.js";import{queryKiroQuota as O}from"../core/provider-quota/kiro.js";import{buildToolUseCard as $,buildToolResultCard as y,buildLocalGrixCardLink as ht}from"./tool-card-utils.js";import{DeferredEventManager as gt}from"./deferred-events.js";import{buildAgentProbeResult as mt,PROBE_CACHE_TTL_STATIC_MS as pt,PROBE_CACHE_TTL_FULL_MS as ft}from"./probe-helper.js";const _t=600*1e3,vt=60*1e3,te=30*1e3,St=10*1e3,L=new Set(["claude","acp","agy","cursor","codex"]),bt=new Set(["claude","codex","cursor","codewhale","opencode","pi","openhuman","agy","acp"]),W=3;class Ei{config;name;aibotHandle;aibotConfig;pool;stopped=!1;revokeHandler=new Ge;sessionBindings=new Map;deferredMgr;sendCtrl=new ut(at);bindingStore;globalConfigStore;upgradeTrigger=null;agentDeletedHandler=null;shareSetHandler=null;skillSyncHandler=null;providerConfigHandler=null;agentProfile={agentName:"",introduction:""};activeEventStore;cachedRateLimits=null;cachedRateLimitsSampledAtMs=null;cachedCodexContextWindow=null;cachedCodexTokenUsage=null;cachedCodexUsageSampledAtMs=null;cachedAcpContextWindow=null;cachedAcpContextWindowSampledAtMs=null;cachedClaudeRateLimitState=null;cachedProviderQuota=null;cachedProviderQuotaSampledAtMs=null;cachedProviderQuotaKey=null;sessionProviderHints=new Map;sessionProviderQuotas=new Map;sessionProviderMeta=new Map;claudeWorkerStatus=new Map;lastReportedSkillsHash="";conversationLog=null;packetLog=null;providerQuotaTimer=null;eventSessionIndex=new Map;inflightEvents=new Map;restartCount=new Map;selfDrivenSessions=new Set;selfDrivenLabels=new Map;probeCache=new Map;sessionScanCache;reasonixTitleScan=null;isRateLimitsCacheFresh(e){if(!Number.isFinite(e))return!1;const o=Number(e);return o>0&&Date.now()-o<=vt}async resolveProviderQuotaSource(e){if(e?.disabled)return null;let o=this.config.providerBaseUrl??this.config.agent.provider?.baseUrl,t=this.config.providerApiKey??this.config.agent.provider?.apiKey;if((!o||!t)&&(this.config.adapterType??"acp")==="claude"){const i=me();o||(o=(i.ANTHROPIC_BASE_URL??"").trim()||void 0),t||(t=(i.ANTHROPIC_API_KEY??"").trim()||(i.ANTHROPIC_AUTH_TOKEN??"").trim()||void 0)}if((!o||!t)&&(this.config.adapterType??"acp")==="codex"){const i=Ae();!o&&i.baseUrl&&(o=i.baseUrl),!t&&i.apiKey&&(t=i.apiKey)}if((!o||!t)&&(this.config.adapterType??"acp")==="acp"&&this.config.aibot.clientType==="kimi"){const i=await ke();!o&&i.baseUrl&&(o=i.baseUrl),!t&&i.apiKey&&(t=i.apiKey)}if(!o||!t)return null;const s=q(e?.providerId)??H(e?.model)?.id??q(this.config.agent.provider?.providerId)??H(this.config.agent.provider?.model)?.id;return{baseUrl:o,...this.config.agent.provider?.quotaBaseUrl?{quotaBaseUrl:this.config.agent.provider.quotaBaseUrl}:{},apiKey:t,...s?{providerId:s}:{}}}async maybeQueryProviderQuota(e=!1,o){if(this.config.aibot.clientType==="kiro"){if(!e&&this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&this.cachedProviderQuota)return this.cachedProviderQuota;try{const i=await O();return this.cachedProviderQuota=i,this.cachedProviderQuotaSampledAtMs=Date.now(),this.cachedProviderQuotaKey="kiro",u.info(this.name,`[provider-quota] kiro queried: success=${i.success}`+(i.balance?` balance=${i.balance.remaining} ${i.balance.unit}`:"")+(i.planName?` plan=${i.planName}`:"")+(i.error?` error=${i.error}`:"")),i}catch(i){return u.warn(this.name,`[provider-quota] kiro query failed: ${i instanceof Error?i.message:String(i)}`),null}}const t=await this.resolveProviderQuotaSource(o);if(!t)return null;const s=U.cacheKey(t);if(!e&&this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&this.cachedProviderQuota&&this.cachedProviderQuotaKey===s)return this.cachedProviderQuota;try{const i=await U.query(t,{fresh:e}),n=i.quota;return this.cachedProviderQuota=n,this.cachedProviderQuotaSampledAtMs=i.sampledAt,this.cachedProviderQuotaKey=i.cacheKey,u.info(this.name,`[provider-quota] queried: provider=${n.provider} success=${n.success}`+(n.tiers.length>0?` tiers=${n.tiers.map(r=>`${r.name}=${r.usedPercent}%`).join(",")}`:"")+(n.balance?` balance=${n.balance.remaining} ${n.balance.unit}`:"")+(n.error?` error=${n.error}`:"")),n}catch(i){return u.warn(this.name,`[provider-quota] query failed: ${i instanceof Error?i.message:String(i)}`),null}}startProviderQuotaTimer(){this.providerQuotaTimer||(this.providerQuotaTimer=setInterval(()=>{if(this.stopped){this.stopProviderQuotaTimer();return}this.refreshAndPushProviderQuota().catch(()=>{})},te),u.info(this.name,`[provider-quota-timer] started (interval=${te}ms)`))}stopProviderQuotaTimer(){this.providerQuotaTimer&&(clearInterval(this.providerQuotaTimer),this.providerQuotaTimer=null,u.info(this.name,"[provider-quota-timer] stopped"))}async refreshAndPushProviderQuota(e=!1){try{let o;if(this.config.aibot.clientType==="kiro")o=await O(),this.cachedProviderQuota=o,this.cachedProviderQuotaSampledAtMs=Date.now(),this.cachedProviderQuotaKey="kiro";else{const t=[...this.sessionBindings.entries()].filter(([s,i])=>!!i&&this.sessionProviderHints.has(s));if(t.length>0&&await Promise.all(t.map(([s,i])=>this.refreshProviderQuotaForSession(s,i,e))),t.length===this.sessionBindings.size&&t.length>0)return;o=await this.maybeQueryProviderQuota(e)}o?.success&&this.pushProviderQuotaToBindings(o,this.sessionProviderHints.size>0)}catch{}}pushProviderQuotaToBindings(e,o=!1){const t=this.providerQuotaToRateLimits(e);for(const[s,i]of this.sessionBindings.entries()){if(!i||o&&this.sessionProviderHints.has(s))continue;const n={provider_quota:e};t&&(n.rate_limits=t);const r=this.cachedAcpContextWindow;r&&("usedPercentage"in r?n.context_window={usedPercentage:r.usedPercentage,remainingPercentage:100-r.usedPercentage}:n.context_window=r),this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:i,meta:n})}}enrichProviderQuotaMeta(e,o="standard",t=this.cachedProviderQuota,s=this.cachedProviderQuotaSampledAtMs){const i={...e??{}};if(!t?.success||(i.provider_quota||(i.provider_quota=t),i.rate_limits))return i;if(o==="codex"){const r=ee(t,s??Date.now());return r&&(i.rate_limits=r.rateLimits,i.rate_limit_primary_percent=r.primaryPercent,i.rate_limit_secondary_percent=r.secondaryPercent,i.rate_limit_primary_window_min=r.primaryWindowMin,i.rate_limit_secondary_window_min=r.secondaryWindowMin),i}const n=Z(t,s??Date.now());return n&&(i.rate_limits=n),i}providerQuotaMetaPayload(e,o,t){let s=this.cachedProviderQuota,i=this.cachedProviderQuotaSampledAtMs;if(o){const r=typeof e?.model_provider=="string"?e.model_provider:void 0,a=typeof e?.model_id=="string"?e.model_id:void 0,d=q(r)??H(r)?.id??H(a)?.id;if(d||a){const l=!!r&&r.trim().toLowerCase()!=="grix"&&!d,h={...d?{providerId:d}:{},...a?{model:a}:{},...l?{disabled:!0}:{}},m=this.sessionProviderHints.get(o),v=m?.providerId!==h.providerId||m?.model!==h.model||m?.disabled!==h.disabled;this.sessionProviderHints.set(o,h),this.sessionProviderMeta.set(o,{...e??{}}),v&&this.sessionProviderQuotas.delete(o),(v||!this.sessionProviderQuotas.has(o))&&t&&this.refreshProviderQuotaForSession(o,t,!1)}const c=this.sessionProviderQuotas.get(o);this.sessionProviderHints.has(o)&&(s=null,i=null),c&&(s=c.quota,i=c.sampledAt)}const n=this.enrichProviderQuotaMeta(e,"standard",s,i);return Object.keys(n).length>0?{meta:n}:{}}async refreshProviderQuotaForSession(e,o,t){const s=this.sessionProviderHints.get(e);if(!s)return;const i=`${s.providerId??""}|${s.model??""}|${s.disabled===!0}`,n=await this.resolveProviderQuotaSource(s);if(n)try{const r=await U.query(n,{fresh:t}),a=this.sessionProviderHints.get(e);if(`${a?.providerId??""}|${a?.model??""}|${a?.disabled===!0}`!==i||(this.sessionProviderQuotas.set(e,{quota:r.quota,sampledAt:r.sampledAt,cacheKey:r.cacheKey}),!r.quota.success))return;const c=this.enrichProviderQuotaMeta(this.sessionProviderMeta.get(e),"standard",r.quota,r.sampledAt);this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:o,meta:c})}catch(r){u.warn(this.name,`[provider-quota] session query failed: session=${e} ${r instanceof Error?r.message:String(r)}`)}}getFreshClaudeRateLimitState(){const e=this.cachedClaudeRateLimitState;return e&&this.isRateLimitsCacheFresh(e.sampledAt)?e:null}getFreshCodexGlobalRateLimitCache(){const e=this.cachedRateLimits,o=this.cachedCodexContextWindow,t=this.cachedCodexTokenUsage;return{sampledAt:Math.max(this.cachedRateLimitsSampledAtMs??0,this.cachedCodexUsageSampledAtMs??0)||null,rateLimits:e,contextWindow:o,tokenUsage:t,hasData:!!(e||o||t)}}getStatus(){const e=this.pool?.getStatus()??{total:0,ready:0,busy:0};return{name:this.name,agentId:this.config.aibot.agentId,alive:!this.stopped,busy:e.busy>0,exhausted:this.pool?[...this.pool.getAllSlots()].some(o=>o.respawn.exhausted):!1,adapterType:this.config.adapterType??"acp",clientType:this.config.aibot.clientType,pool:e}}hasPendingWork(){return this.pool?.hasPendingWork()??!1}hasPendingOrBackgroundWork(){return this.pool?.hasPendingOrBackgroundWork()??!1}relayEnvStale=!1;relayEnvSettledHandler;async recycleAdaptersForRelayChange(){if(!this.pool)return this.relayEnvStale=!1,!0;if(this.hasPendingWork())return this.relayEnvStale=!0,u.info(this.name,"relay changed but agent is busy; adapters keep the old env until the next message (marked stale)"),!1;const e=[...this.pool.getAllSlots()];let o=0,t=!1;for(const s of e){if(this.hasPendingWork())return this.relayEnvStale=!0,u.info(this.name,"new work arrived while recycling; remaining adapters stay stale until the next message"),!1;let i=!1;await this.pool.removeSlot(s.sessionId).catch(n=>{t=!0,i=!0,u.warn(this.name,`failed to recycle adapter slot ${s.sessionId} after relay change: ${n}`)}),i||(o+=1)}return o>0&&u.info(this.name,`recycled ${o} adapter slot(s) so the new Grix relay env takes effect`),t?(this.relayEnvStale=!0,!1):(this.relayEnvStale=!1,this.relayEnvSettledHandler?.(),!0)}setRelayEnvSettledHandler(e){this.relayEnvSettledHandler=e}hasStaleRelayEnv(){return this.relayEnvStale}markRelayEnvStale(){this.relayEnvStale=!0}async probe(e={}){const o=e.conversation?"full":"static",t=e.conversation?ft:pt;if(!e.fresh){const a=this.probeCache.get(o);if(a&&Date.now()-a.sampledAt<t)return{...a.result,cached:!0}}const s=this.config.adapterType??"acp",i=this.createAdapter(s,"__probe__"),n=e.conversation&&s==="acp"?()=>this.runAcpConversationProbe(i,e.timeoutMs??1e4):void 0;let r;try{r=await mt({adapter:i,agentName:this.name,clientType:this.config.aibot.clientType,adapterType:s,providerBaseUrl:this.config.providerBaseUrl??null,opts:e,launchConversationProbe:n})}finally{i.stop().catch(()=>{})}return this.probeCache.set(o,{result:r,sampledAt:r.probed_at}),r}async runAcpConversationProbe(e,o){const t=Date.now(),s="__probe__",i=`probe-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,n=this.config.agent.cwd||ne(),r=()=>Date.now()-t;try{await e.start()}catch(h){return{attempted:!0,ok:!1,latency_ms:r(),error:{code:"conversation_failed",message:`start failed: ${h instanceof Error?h.message:String(h)}`}}}if(!e.isAlive())return{attempted:!0,ok:!1,latency_ms:r(),error:{code:"process_not_started",message:"agent process not alive"}};if(e instanceof A)try{await e.bindSession(s,n)}catch{}let a=null;const d=new Promise(h=>{a=m=>{m===i&&h()},e.on("eventDone",a)});e.deliverInboundEvent({event_id:i,session_id:s,content:"ping",msg_id:i}),e.deliverStopEvent(i,s);let c=null;const l=await Promise.race([d.then(()=>!1),new Promise(h=>{c=setTimeout(()=>h(!0),o)})]);return c&&clearTimeout(c),a&&e.removeListener("eventDone",a),l?{attempted:!0,ok:!1,latency_ms:r(),error:{code:"conversation_timeout",message:`no eventDone within ${o}ms`}}:{attempted:!0,ok:!0,latency_ms:r()}}constructor(e,o){this.config=e,this.name=e.name;const t=e.adapterType??"acp";if(this.sendCtrl.setDefaultQuoteTrigger(lt[t]),this.aibotConfig={...e.aibot,...t==="claude"?{localActions:e.aibot.localActions??["session_control","claude_interaction_reply","get_session_usage","get_rate_limits","set_model","set_mode","thread_compact","get_agent_global_config"]}:{}},e.eventQueue&&(this.aibotConfig.concurrency={max_concurrent:e.eventQueue.maxConcurrent,max_queued:e.eventQueue.maxQueued,queue_timeout_ms:e.eventQueue.queueTimeoutMs,cancelable_queued:e.eventQueue.cancelableQueued,cancelable_running:e.eventQueue.cancelableRunning}),this.conversationLog=e.logDir?new Ne(e.logDir):null,this.packetLog=e.logDir?new Ke(e.logDir):null,this.bindingStore=new et(e.bindingsPath),this.bindingStore.load(),this.globalConfigStore=o??null,this.deferredMgr=new gt(this.name),this.activeEventStore=e.activeEventStorePath?new rt(e.activeEventStorePath):null,t==="codex")this.sessionScanCache=new k(Ee,De);else if(t==="claude")this.sessionScanCache=new k(Re,Qe);else if(t==="pi"){const s=e.agent.env?.PI_CODING_AGENT_DIR;this.sessionScanCache=new k(()=>xe(void 0,s),()=>Fe(void 0,s))}else if(t==="codewhale")this.sessionScanCache=new k(He,Ue);else if(t==="cursor")this.sessionScanCache=new k(Le,qe);else if(t==="opencode")this.sessionScanCache=new k(Me,Oe);else if(t==="agy")this.sessionScanCache=new k(()=>[],()=>[]);else{e.aibot.clientType?.trim().toLowerCase()==="reasonix"&&(this.reasonixTitleScan=new k(Te,We));const s=ye(e.aibot.clientType);this.sessionScanCache=s?new k(()=>Pe(void 0,s),()=>Be(void 0,s)):new k(()=>[],()=>[])}}async start(){await we({agentName:this.name,adapterType:this.config.adapterType??"acp",command:this.config.agent.command,piConfigDir:this.config.agent.env?.PI_CODING_AGENT_DIR,hasProvider:!!this.config.agent.provider,boundCwds:[...this.bindingStore.entries()].map(([,t])=>String(t.cwd??""))}).then(t=>{t.length>0&&u.info(this.name,`restored native provider config after relay disable: ${t.join(", ")}`)}).catch(t=>{u.warn(this.name,`native provider residue sweep failed: ${t instanceof Error?t.message:String(t)}`)}),(this.config.adapterType??"acp")==="claude"&&(await B().catch(()=>{}),this.maybeQueryProviderQuota().catch(()=>{}));const e=!!(this.config.providerBaseUrl&&this.config.providerApiKey||this.config.agent.provider?.baseUrl&&this.config.agent.provider?.apiKey);(this.config.aibot.clientType==="kiro"||this.config.aibot.clientType==="kimi"||e)&&(this.maybeQueryProviderQuota().catch(()=>{}),this.startProviderQuotaTimer()),(this.config.adapterType??"acp")==="codex"&&this.maybeQueryProviderQuota().catch(()=>{}),await this.connectAibot(),this.sendCtrl.bind(this.aibotHandle);const o=this.config.adapterType??"acp";this.pool=new je({maxPoolSize:this.config.poolMaxSize??20,idleTimeoutMs:this.config.poolIdleTimeoutMs??18e5,eventQueue:this.config.eventQueue},t=>{const s=this.createAdapter(o,t);return s instanceof A&&s.on("acpSessionReady",i=>{this.bindingStore.setAcpSessionId(t,i),this.sessionScanCache.invalidate()}),s},(t,s)=>{this.aibotHandle.sendEventAck({event_id:t,session_id:s,received_at:Date.now()})}),this.pool.setEventStateHandler((t,s,i,n)=>{u.info(this.name,`[queue-debug] send event_state session=${s} event=${t} state=${i} queue_pos=${n?.queue_position??""} queue_total=${n?.queue_total??""}`),this.aibotHandle.sendEventState({event_id:t,session_id:s,state:i,content_preview:n?.content_preview,content:n?.content,queue_position:n?.queue_position,queue_total:n?.queue_total,actions:n?.actions,reason:n?.reason,held:n?.held,held_reason:n?.held_reason,updated_at:Date.now()}),this.pushQueueSnapshotForSession(s),(i==="canceled"||i==="failed")&&(this.aibotHandle.sendEventResult({event_id:t,status:i==="canceled"?"canceled":"failed",msg:n?.reason,updated_at:Date.now()}),this.discardEventTrackingState(t))}),this.pool.setQueueComposingHandler((t,s,i)=>{this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:s,...s?{ttl_ms:3e4,ref_event_id:i}:{}})}),this.pool.setInternalErrorHandler(t=>{this.handleSessionInternalError(t).catch(s=>{u.error(this.name,`[recovery] handleSessionInternalError failed event=${t.eventId} session=${t.sessionId}: ${s instanceof Error?s.message:String(s)}`)})}),this.pool.setEventStartedHandler((t,s)=>{if(this.reportSessionSkills(s),this.config.adapterType!=="claude")return;this.claudeWorkerStatus.set(s,"busy");const i=this.bindingStore.get(s);i?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"busy",cwd:i.cwd,meta:this.buildClaudeToolbarMeta(s)})}),this.pool.setEventDoneHandler((t,s)=>{const i=this.config.adapterType??"acp";if(i==="claude"){this.claudeWorkerStatus.set(s,"ready");const n=this.bindingStore.get(s);n?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:n.cwd,meta:this.buildClaudeToolbarMeta(s)})}else if(i==="agy"){const n=this.bindingStore.get(s);if(n?.cwd){const r=this.buildAgyToolbarMeta(s),a=r?.available_models??[];u.info(this.name,`[agy-toolbar-diag] eventDone push binding card session=${s} model_id=${String(r?.model_id??"")} available_models=${a.length} meta_keys=${r?Object.keys(r).join(","):"<none>"}`),this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:n.cwd,meta:r})}else u.info(this.name,`[agy-toolbar-diag] eventDone skip binding card: no binding cwd session=${s}`)}}),this.pool.setSessionActivityHandler((t,s,i)=>{const n=this.selfDrivenSessions.has(t),r=this.selfDrivenLabels.get(t);s?(this.selfDrivenSessions.add(t),i&&this.selfDrivenLabels.set(t,i),this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:!0,ttl_ms:9e4})):(this.selfDrivenSessions.delete(t),this.selfDrivenLabels.delete(t),n&&this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:!1})),(n!==s||s&&i!==void 0&&i!==r)&&this.pushQueueSnapshotForSession(t)}),this.pool.startIdleSweep(),u.info(this.name,`Ready (adapter: ${o}, poolMax: ${this.config.poolMaxSize??20})`)}async stop(){this.stopped=!0,this.pool?.stopIdleSweep(),this.stopProviderQuotaTimer();const e=this.pool?.collectActiveEventIds()??[];e.length>0&&this.activeEventStore&&await this.activeEventStore.save(e);for(const s of e)u.info(this.name,`Canceling active event on shutdown: ${s}`),this.sendEventResultWithCleanup(s,"canceled","process shutting down");e.length>0&&await new Promise(s=>setTimeout(s,100)),this.pool?.clearActiveEventsForShutdown();const o=this.deferredMgr.getAllDeferredEvents();for(const s of o)u.info(this.name,`Failing deferred event on shutdown: ${s.event_id}`),this.sendEventResultWithCleanup(s.event_id,"failed","process shutting down");const t=this.pool?.drainAllQueuedEvents()??[];for(const s of t)u.info(this.name,`Failing queued event on shutdown: ${s.event_id}`),this.sendEventResultWithCleanup(s.event_id,"failed","process shutting down");this.deferredMgr.clearAll(),await this.pool?.stop(),this.aibotHandle?.disconnect(),e.length>0&&this.activeEventStore&&await this.activeEventStore.save([]),this.eventSessionIndex.clear(),this.inflightEvents.clear(),this.restartCount.clear()}resolveSpawnEnv(){const e=be(this.config.agent.clientType,this.config.agent.provider),o=this.config.agent.env,t=Object.keys(e).length>0?{...e,...o}:o;return fe(t,this.name,this.config.agent.clientType)}createAdapter(e,o){switch(e){case"claude":return this.createClaudeAdapter(o);case"codex":return this.createCodexAdapter(o);case"pi":return this.createPiAdapter(o);case"openhuman":return this.createOpenHumanAdapter(o);case"codewhale":return this.createCodeWhaleAdapter(o);case"cursor":return this.createCursorAdapter(o);case"opencode":return this.createOpenCodeAdapter(o);case"agy":return this.createAgyAdapter(o);default:return this.createAcpAdapter(o)}}createCursorAdapter(e){const o={...this.config.adapterOptions??{}};o.bindingStore=this.bindingStore,o.aibotSessionId=e;const t=this.bindingStore.get(e),s=this.resolveSessionModelId(e);s&&(o.model=s),t?.modeId&&(o.mode=t.modeId);const i={sendStreamChunk:(n,r,a,d,c)=>{this.sendStreamChunkByRuntimeConfig(n,r,a,d,c)},sendEventResult:(n,r,a)=>{this.sendEventResultWithCleanup(n,r,a)},sendEventAck:(n,r)=>{this.aibotHandle.sendEventAck({event_id:n,session_id:r,received_at:Date.now()})},sendRawEventEnvelope:(n,r,a)=>{this.aibotHandle.sendMsg({event_id:n,session_id:r,msg_type:1,content:"[cursor] raw_event",extra:{channel_data:{cursor:{raw_event:a}},agent_api_origin:!0}})},agentInvoke:async(n,r,a)=>this.platformInvoke(n,r,a),sendLocalActionResult:(n,r,a,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:n,status:r,...a!==void 0?{result:a}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})}};return new Q({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:o},i)}createClaudeAdapter(e){const o={sendReply:(i,n,r,a,d)=>{this.sendReplyByRuntimeConfig(i,n,r,a,d)},sendStreamChunk:(i,n,r,a,d,c,l)=>{this.sendStreamChunkByRuntimeConfig(i,n,r,a,d,c,l)},sendMedia:(i,n,r,a,d,c,l)=>{this.aibotHandle.sendMedia({event_id:i,session_id:n,content:r,msg_type:2,quoted_message_id:d||void 0,client_msg_id:c||void 0,extra:l?{media_caption:a,...l}:{media_caption:a}})},sendEventResult:(i,n,r,a)=>{this.sendEventResultWithCleanup(i,n,r,a)},sendEventAck:(i,n)=>{this.aibotHandle.sendEventAck({event_id:i,session_id:n,received_at:Date.now()})},agentInvoke:async(i,n,r)=>this.platformInvoke(i,n,r),sendLocalActionResult:(i,n,r,a,d)=>{this.aibotHandle.sendLocalActionResult({action_id:i,status:n,...r!==void 0?{result:r}:{},...a?{error_code:a}:{},...d?{error_msg:d}:{}})},sendToolUse:(i,n,r,a)=>{this.sendToolExecutionCard(i,n,$(r,a))},sendToolResult:(i,n,r,a)=>{this.sendToolExecutionCard(i,n,y(r,a))},getWsUrl:()=>this.config.aibot.url,getAgentId:()=>this.config.aibot.agentId,getAgentProfile:()=>this.agentProfile,getApiKey:()=>this.config.aibot.apiKey,getActiveEventCount:()=>0,getPendingPermissionCount:()=>0,getPendingElicitationCount:()=>0,sendAgentQuestionCard:(i,n,r)=>{const a=r.questions.map(c=>c.header).join(", "),d=ht(`[Agent Question] ${r.request_id}`,"agent_question",r);this.aibotHandle.sendText({event_id:i,session_id:n,content:d,msg_type:1,extra:{card_type:"agent_question",summary_text:a}})},sendPermissionCard:i=>{this.aibotHandle.sendMsg({event_id:i.eventId,session_id:i.sessionId,client_msg_id:`perm_${D()}`,msg_type:1,content:i.toolTitle?`Permission required: ${i.toolTitle}`:"Permission request",extra:{channel_data:{execApproval:{approvalId:i.approvalId,approvalSlug:i.toolName},grix:{execApproval:{approval_command_id:i.approvalId,command:i.toolTitle||i.toolName,host:"claude"}}},agent_api_origin:!0}})},sendDirectMessage:i=>{this.aibotHandle.sendMsg({session_id:i.sessionId,msg_type:1,content:i.content,...i.clientMsgId?{client_msg_id:i.clientMsgId}:{},...i.quotedMessageId?{quoted_message_id:i.quotedMessageId}:{}})},onStatusLineUpdated:i=>{(i.rateLimits?.fiveHour||i.rateLimits?.sevenDay)&&(this.cachedClaudeRateLimitState=i);const n=this.bindingStore.get(e);n?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:this.claudeWorkerStatus.get(e)??"ready",cwd:n.cwd,meta:this.buildClaudeToolbarMeta(e)})},sendMcpFrame:i=>{this.aibotHandle.sendMcpFrame(e,i)}},t=this.config.adapterOptions??{},s={...t,sessionRuntimeResolver:()=>{const i=this.bindingStore.get(e);return{cwd:i?.cwd,modeId:i?.modeId??C.fullAuto,modelId:this.resolveSessionModelId(e),pluginDir:t.pluginDir,claudeSessionId:i?.claudeSessionId,onSessionIdAssigned:n=>{this.bindingStore.setClaudeSessionId(e,n),this.sessionScanCache.invalidate()}}}};return new P({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:s},o)}createCodexAdapter(e){let o=null;const t={sendEventResult:(n,r,a)=>{this.sendEventResultWithCleanup(n,r,a)},sendEventAck:(n,r)=>this.aibotHandle.sendEventAck({event_id:n,session_id:r,received_at:Date.now()}),sendCodexEvent:n=>{this.shouldDropCodexDisplayEvent(n.event_id,n.codex_method)||(this.aibotHandle.sendCodexEvent(this.stampCodexEventQuote(n)),this.logCodexEventToConversation(n))},sendCodexEventReliable:async n=>{if(!this.shouldDropCodexDisplayEvent(n.event_id,n.codex_method)){try{await this.aibotHandle.sendCodexEventReliable(this.stampCodexEventQuote(n))}catch(r){u.warn("bridge",`[codex] sendCodexEventReliable ACK failed event=${n.event_id}: ${r}`)}u.info("bridge",`[codex] sendCodexEventReliable done event=${n.event_id} method=${n.codex_method}`),this.logCodexEventToConversation(n)}},sendRunError:(n,r,a)=>{this.sendStreamChunkByRuntimeConfig(n,r,`
|
|
1
|
+
import x from"node:path";import{realpath as ie,stat as D}from"node:fs/promises";import{tmpdir as ne}from"node:os";import{randomUUID as Q}from"node:crypto";import{ConnectionManager as se}from"../core/aibot/index.js";import{ClaudeAdapter as P}from"../adapter/claude/index.js";import{CodexAdapter as oe}from"../adapter/codex/index.js";import{PiAdapter as re}from"../adapter/pi/index.js";import{AcpAdapter as A}from"../adapter/acp/index.js";import{OpenHumanAdapter as ae}from"../adapter/openhuman/index.js";import{CursorAdapter as M,CURSOR_MODE_OPTIONS as de,fetchCursorAccountUsage as ce}from"../adapter/cursor/index.js";import{CodeWhaleAdapter as N}from"../adapter/codewhale/index.js";import{OpenCodeAdapter as le}from"../adapter/opencode/index.js";import{AgyAdapter as K}from"../adapter/agy/index.js";import{getCachedAgyModels as ue}from"../adapter/agy/model-list.js";import{getCachedAgyQuotaInfo as he}from"../adapter/agy/quota.js";import{LOCAL_ACTION_ERROR_CODES as T,LOCAL_ACTION_TYPES as b,SESSION_CONTROL_ERROR_CODES as g,SESSION_CONTROL_VERBS as p,SESSION_MODE_IDS as C}from"../adapter/claude/protocol-contract.js";import{parseClaudeSessionUsage as ge}from"../adapter/claude/usage-parser.js";import{resolveSkillScanMode as z}from"../adapter/claude/skill-scanner.js";import{buildReportedSkills as me}from"../default-skills/index.js";import{buildWireSkills as G}from"../core/skill-sync/sync-state.js";import{uploadSkill as pe}from"../core/skill-sync/skill-upload.js";import{fetchAvailableModels as B,getCachedModels as F,readSettingsEnv as fe}from"../adapter/claude/model-list.js";import{parseAcpSessionUsage as _e}from"../adapter/acp/usage-parser.js";import{applyProxyEnv as ve,getProxyManager as Se,relayHostsForClientType as be,buildGatewayRoute as we}from"../core/proxy/index.js";import{buildProviderEnv as Ce}from"../core/config/provider-env.js";import{sweepNativeProviderResidue as Ae}from"./native-provider-sweep.js";import{parseCodexSessionUsage as ke}from"../adapter/codex/usage-parser.js";import{readCodexProviderSettings as Re}from"../adapter/codex/codex-trust.js";import{resolveKimiProviderSettings as Ee}from"../core/provider-quota/index.js";import{scanCodexSessions as $e}from"../adapter/codex/session-scanner.js";import{scanClaudeSessions as ye}from"../adapter/claude/session-scanner.js";import{reasonixStampFromSessionId as Pe,resolveAcpAgentTypes as Te,scanAcpSessions as xe,scanReasonixSessionTitles as Me}from"../adapter/acp/session-scanner.js";import{scanPiSessions as He}from"../adapter/pi/session-scanner.js";import{scanCodeWhaleSessions as Le}from"../adapter/codewhale/session-scanner.js";import{scanCursorSessions as Ie}from"../adapter/cursor/session-scanner.js";import{scanOpenCodeSessions as De}from"../adapter/opencode/session-scanner.js";import{parsePiSessionUsage as Qe}from"../adapter/pi/usage-parser.js";import{SessionScanCache as k,resolveCodexLeafDirs as Be,resolveClaudeLeafDirs as Fe,resolveAcpLeafDirs as Ue,resolvePiLeafDirs as qe,resolveCodeWhaleLeafDirs as Oe,resolveCursorLeafDirs as We,resolveOpenCodeLeafDirs as Ne,resolveReasonixLeafDirs as Ke}from"./session-scan-cache.js";import{log as u,ConversationLog as ze,AgentApiPacketLog as Ge,BridgeEventLog as je,GRIX_PATHS as j}from"../core/log/index.js";import{RevokeHandler as Ve}from"./revoke-handler.js";import{AdapterPool as Je,PoolFullError as Ye}from"./adapter-pool.js";import{parseSessionControlCommand as Xe,handleSessionControlCommand as V,handleSessionControlLocalAction as Ze}from"./session-controller.js";import{isOpenSessionDirectiveMessage as et}from"../core/protocol/interaction-parser.js";import{handleAcpSetModel as J,handleAcpSetMode as Y,resolveAcpInitialDefaults as tt}from"./acp-toolbar-persist.js";import{SessionBindingStore as it}from"../core/persistence/session-binding-store.js";import{handleFileListAction as nt,handleCreateFolderAction as st,serveLocalFile as ot,realHomeDir as X}from"../core/files/index.js";import{getMachineName as rt}from"../core/util/index.js";import{uploadReplyFileToAgentMedia as at}from"../core/protocol/agent-api-media.js";import{ActiveEventStore as dt}from"../core/persistence/active-event-store.js";import{DEFAULT_CONNECTOR_RUNTIME_CONFIG as ct,applyConnectorRuntimeConfigPatch as lt,extractConnectorRuntimeConfigPatch as ut}from"./runtime-config.js";import{DEFAULT_QUOTE_TRIGGER_BY_ADAPTER as ht,SendController as gt}from"./send-controller.js";import{sharedProviderQuotaService as U,detectProviderFromModel as H,normalizeProviderId as q,providerQuotaToRateLimits as Z,providerQuotaToCodexRateLimits as ee}from"../core/provider-quota/index.js";import{queryKiroQuota as O}from"../core/provider-quota/kiro.js";import{buildToolUseCard as $,buildToolResultCard as y,buildLocalGrixCardLink as mt}from"./tool-card-utils.js";import{planRawEventDelivery as pt,buildAuxiliaryDetailText as ft}from"./raw-event-delivery.js";import{DeferredEventManager as _t}from"./deferred-events.js";import{buildAgentProbeResult as vt,PROBE_CACHE_TTL_STATIC_MS as St,PROBE_CACHE_TTL_FULL_MS as bt}from"./probe-helper.js";const wt=600*1e3,Ct=60*1e3,te=30*1e3,At=10*1e3,L=new Set(["claude","acp","agy","cursor","codex"]),kt=new Set(["claude","codex","cursor","codewhale","opencode","pi","openhuman","agy","acp"]),W=3;class Ti{config;name;aibotHandle;aibotConfig;pool;stopped=!1;revokeHandler=new Ve;sessionBindings=new Map;deferredMgr;sendCtrl=new gt(ct);bindingStore;globalConfigStore;upgradeTrigger=null;agentDeletedHandler=null;shareSetHandler=null;skillSyncHandler=null;providerConfigHandler=null;agentProfile={agentName:"",introduction:""};activeEventStore;cachedRateLimits=null;cachedRateLimitsSampledAtMs=null;cachedCodexContextWindow=null;cachedCodexTokenUsage=null;cachedCodexUsageSampledAtMs=null;cachedAcpContextWindow=null;cachedAcpContextWindowSampledAtMs=null;cachedClaudeRateLimitState=null;cachedProviderQuota=null;cachedProviderQuotaSampledAtMs=null;cachedProviderQuotaKey=null;sessionProviderHints=new Map;sessionProviderQuotas=new Map;sessionProviderMeta=new Map;claudeWorkerStatus=new Map;lastReportedSkillsHash="";conversationLog=null;packetLog=null;providerQuotaTimer=null;eventSessionIndex=new Map;inflightEvents=new Map;restartCount=new Map;selfDrivenSessions=new Set;selfDrivenLabels=new Map;probeCache=new Map;sessionScanCache;reasonixTitleScan=null;isRateLimitsCacheFresh(e){if(!Number.isFinite(e))return!1;const o=Number(e);return o>0&&Date.now()-o<=Ct}async resolveProviderQuotaSource(e){if(e?.disabled)return null;let o=this.config.providerBaseUrl??this.config.agent.provider?.baseUrl,t=this.config.providerApiKey??this.config.agent.provider?.apiKey;if((!o||!t)&&(this.config.adapterType??"acp")==="claude"){const i=fe();o||(o=(i.ANTHROPIC_BASE_URL??"").trim()||void 0),t||(t=(i.ANTHROPIC_API_KEY??"").trim()||(i.ANTHROPIC_AUTH_TOKEN??"").trim()||void 0)}if((!o||!t)&&(this.config.adapterType??"acp")==="codex"){const i=Re();!o&&i.baseUrl&&(o=i.baseUrl),!t&&i.apiKey&&(t=i.apiKey)}if((!o||!t)&&(this.config.adapterType??"acp")==="acp"&&this.config.aibot.clientType==="kimi"){const i=await Ee();!o&&i.baseUrl&&(o=i.baseUrl),!t&&i.apiKey&&(t=i.apiKey)}if(!o||!t)return null;const s=q(e?.providerId)??H(e?.model)?.id??q(this.config.agent.provider?.providerId)??H(this.config.agent.provider?.model)?.id;return{baseUrl:o,...this.config.agent.provider?.quotaBaseUrl?{quotaBaseUrl:this.config.agent.provider.quotaBaseUrl}:{},apiKey:t,...s?{providerId:s}:{}}}async maybeQueryProviderQuota(e=!1,o){if(this.config.aibot.clientType==="kiro"){if(!e&&this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&this.cachedProviderQuota)return this.cachedProviderQuota;try{const i=await O();return this.cachedProviderQuota=i,this.cachedProviderQuotaSampledAtMs=Date.now(),this.cachedProviderQuotaKey="kiro",u.info(this.name,`[provider-quota] kiro queried: success=${i.success}`+(i.balance?` balance=${i.balance.remaining} ${i.balance.unit}`:"")+(i.planName?` plan=${i.planName}`:"")+(i.error?` error=${i.error}`:"")),i}catch(i){return u.warn(this.name,`[provider-quota] kiro query failed: ${i instanceof Error?i.message:String(i)}`),null}}const t=await this.resolveProviderQuotaSource(o);if(!t)return null;const s=U.cacheKey(t);if(!e&&this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&this.cachedProviderQuota&&this.cachedProviderQuotaKey===s)return this.cachedProviderQuota;try{const i=await U.query(t,{fresh:e}),n=i.quota;return this.cachedProviderQuota=n,this.cachedProviderQuotaSampledAtMs=i.sampledAt,this.cachedProviderQuotaKey=i.cacheKey,u.info(this.name,`[provider-quota] queried: provider=${n.provider} success=${n.success}`+(n.tiers.length>0?` tiers=${n.tiers.map(a=>`${a.name}=${a.usedPercent}%`).join(",")}`:"")+(n.balance?` balance=${n.balance.remaining} ${n.balance.unit}`:"")+(n.error?` error=${n.error}`:"")),n}catch(i){return u.warn(this.name,`[provider-quota] query failed: ${i instanceof Error?i.message:String(i)}`),null}}startProviderQuotaTimer(){this.providerQuotaTimer||(this.providerQuotaTimer=setInterval(()=>{if(this.stopped){this.stopProviderQuotaTimer();return}this.refreshAndPushProviderQuota().catch(()=>{})},te),u.info(this.name,`[provider-quota-timer] started (interval=${te}ms)`))}stopProviderQuotaTimer(){this.providerQuotaTimer&&(clearInterval(this.providerQuotaTimer),this.providerQuotaTimer=null,u.info(this.name,"[provider-quota-timer] stopped"))}async refreshAndPushProviderQuota(e=!1){try{let o;if(this.config.aibot.clientType==="kiro")o=await O(),this.cachedProviderQuota=o,this.cachedProviderQuotaSampledAtMs=Date.now(),this.cachedProviderQuotaKey="kiro";else{const t=[...this.sessionBindings.entries()].filter(([s,i])=>!!i&&this.sessionProviderHints.has(s));if(t.length>0&&await Promise.all(t.map(([s,i])=>this.refreshProviderQuotaForSession(s,i,e))),t.length===this.sessionBindings.size&&t.length>0)return;o=await this.maybeQueryProviderQuota(e)}o?.success&&this.pushProviderQuotaToBindings(o,this.sessionProviderHints.size>0)}catch{}}pushProviderQuotaToBindings(e,o=!1){const t=this.providerQuotaToRateLimits(e);for(const[s,i]of this.sessionBindings.entries()){if(!i||o&&this.sessionProviderHints.has(s))continue;const n={provider_quota:e};t&&(n.rate_limits=t);const a=this.cachedAcpContextWindow;a&&("usedPercentage"in a?n.context_window={usedPercentage:a.usedPercentage,remainingPercentage:100-a.usedPercentage}:n.context_window=a),this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:i,meta:n})}}enrichProviderQuotaMeta(e,o="standard",t=this.cachedProviderQuota,s=this.cachedProviderQuotaSampledAtMs){const i={...e??{}};if(!t?.success||(i.provider_quota||(i.provider_quota=t),i.rate_limits))return i;if(o==="codex"){const a=ee(t,s??Date.now());return a&&(i.rate_limits=a.rateLimits,i.rate_limit_primary_percent=a.primaryPercent,i.rate_limit_secondary_percent=a.secondaryPercent,i.rate_limit_primary_window_min=a.primaryWindowMin,i.rate_limit_secondary_window_min=a.secondaryWindowMin),i}const n=Z(t,s??Date.now());return n&&(i.rate_limits=n),i}providerQuotaMetaPayload(e,o,t){let s=this.cachedProviderQuota,i=this.cachedProviderQuotaSampledAtMs;if(o){const a=typeof e?.model_provider=="string"?e.model_provider:void 0,r=typeof e?.model_id=="string"?e.model_id:void 0,d=q(a)??H(a)?.id??H(r)?.id;if(d||r){const l=!!a&&a.trim().toLowerCase()!=="grix"&&!d,h={...d?{providerId:d}:{},...r?{model:r}:{},...l?{disabled:!0}:{}},m=this.sessionProviderHints.get(o),v=m?.providerId!==h.providerId||m?.model!==h.model||m?.disabled!==h.disabled;this.sessionProviderHints.set(o,h),this.sessionProviderMeta.set(o,{...e??{}}),v&&this.sessionProviderQuotas.delete(o),(v||!this.sessionProviderQuotas.has(o))&&t&&this.refreshProviderQuotaForSession(o,t,!1)}const c=this.sessionProviderQuotas.get(o);this.sessionProviderHints.has(o)&&(s=null,i=null),c&&(s=c.quota,i=c.sampledAt)}const n=this.enrichProviderQuotaMeta(e,"standard",s,i);return Object.keys(n).length>0?{meta:n}:{}}async refreshProviderQuotaForSession(e,o,t){const s=this.sessionProviderHints.get(e);if(!s)return;const i=`${s.providerId??""}|${s.model??""}|${s.disabled===!0}`,n=await this.resolveProviderQuotaSource(s);if(n)try{const a=await U.query(n,{fresh:t}),r=this.sessionProviderHints.get(e);if(`${r?.providerId??""}|${r?.model??""}|${r?.disabled===!0}`!==i||(this.sessionProviderQuotas.set(e,{quota:a.quota,sampledAt:a.sampledAt,cacheKey:a.cacheKey}),!a.quota.success))return;const c=this.enrichProviderQuotaMeta(this.sessionProviderMeta.get(e),"standard",a.quota,a.sampledAt);this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:o,meta:c})}catch(a){u.warn(this.name,`[provider-quota] session query failed: session=${e} ${a instanceof Error?a.message:String(a)}`)}}getFreshClaudeRateLimitState(){const e=this.cachedClaudeRateLimitState;return e&&this.isRateLimitsCacheFresh(e.sampledAt)?e:null}getFreshCodexGlobalRateLimitCache(){const e=this.cachedRateLimits,o=this.cachedCodexContextWindow,t=this.cachedCodexTokenUsage;return{sampledAt:Math.max(this.cachedRateLimitsSampledAtMs??0,this.cachedCodexUsageSampledAtMs??0)||null,rateLimits:e,contextWindow:o,tokenUsage:t,hasData:!!(e||o||t)}}getStatus(){const e=this.pool?.getStatus()??{total:0,ready:0,busy:0};return{name:this.name,agentId:this.config.aibot.agentId,alive:!this.stopped,busy:e.busy>0,exhausted:this.pool?[...this.pool.getAllSlots()].some(o=>o.respawn.exhausted):!1,adapterType:this.config.adapterType??"acp",clientType:this.config.aibot.clientType,pool:e}}hasPendingWork(){return this.pool?.hasPendingWork()??!1}hasPendingOrBackgroundWork(){return this.pool?.hasPendingOrBackgroundWork()??!1}relayEnvStale=!1;relayEnvSettledHandler;async recycleAdaptersForRelayChange(){if(!this.pool)return this.relayEnvStale=!1,!0;if(this.hasPendingWork())return this.relayEnvStale=!0,u.info(this.name,"relay changed but agent is busy; adapters keep the old env until the next message (marked stale)"),!1;const e=[...this.pool.getAllSlots()];let o=0,t=!1;for(const s of e){if(this.hasPendingWork())return this.relayEnvStale=!0,u.info(this.name,"new work arrived while recycling; remaining adapters stay stale until the next message"),!1;let i=!1;await this.pool.removeSlot(s.sessionId).catch(n=>{t=!0,i=!0,u.warn(this.name,`failed to recycle adapter slot ${s.sessionId} after relay change: ${n}`)}),i||(o+=1)}return o>0&&u.info(this.name,`recycled ${o} adapter slot(s) so the new Grix relay env takes effect`),t?(this.relayEnvStale=!0,!1):(this.relayEnvStale=!1,this.relayEnvSettledHandler?.(),!0)}setRelayEnvSettledHandler(e){this.relayEnvSettledHandler=e}hasStaleRelayEnv(){return this.relayEnvStale}markRelayEnvStale(){this.relayEnvStale=!0}async probe(e={}){const o=e.conversation?"full":"static",t=e.conversation?bt:St;if(!e.fresh){const r=this.probeCache.get(o);if(r&&Date.now()-r.sampledAt<t)return{...r.result,cached:!0}}const s=this.config.adapterType??"acp",i=this.createAdapter(s,"__probe__"),n=e.conversation&&s==="acp"?()=>this.runAcpConversationProbe(i,e.timeoutMs??1e4):void 0;let a;try{a=await vt({adapter:i,agentName:this.name,clientType:this.config.aibot.clientType,adapterType:s,providerBaseUrl:this.config.providerBaseUrl??null,opts:e,launchConversationProbe:n})}finally{i.stop().catch(()=>{})}return this.probeCache.set(o,{result:a,sampledAt:a.probed_at}),a}async runAcpConversationProbe(e,o){const t=Date.now(),s="__probe__",i=`probe-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,n=this.config.agent.cwd||ne(),a=()=>Date.now()-t;try{await e.start()}catch(h){return{attempted:!0,ok:!1,latency_ms:a(),error:{code:"conversation_failed",message:`start failed: ${h instanceof Error?h.message:String(h)}`}}}if(!e.isAlive())return{attempted:!0,ok:!1,latency_ms:a(),error:{code:"process_not_started",message:"agent process not alive"}};if(e instanceof A)try{await e.bindSession(s,n)}catch{}let r=null;const d=new Promise(h=>{r=m=>{m===i&&h()},e.on("eventDone",r)});e.deliverInboundEvent({event_id:i,session_id:s,content:"ping",msg_id:i}),e.deliverStopEvent(i,s);let c=null;const l=await Promise.race([d.then(()=>!1),new Promise(h=>{c=setTimeout(()=>h(!0),o)})]);return c&&clearTimeout(c),r&&e.removeListener("eventDone",r),l?{attempted:!0,ok:!1,latency_ms:a(),error:{code:"conversation_timeout",message:`no eventDone within ${o}ms`}}:{attempted:!0,ok:!0,latency_ms:a()}}constructor(e,o){this.config=e,this.name=e.name;const t=e.adapterType??"acp";if(this.sendCtrl.setDefaultQuoteTrigger(ht[t]),this.aibotConfig={...e.aibot,...t==="claude"?{localActions:e.aibot.localActions??["session_control","claude_interaction_reply","get_session_usage","get_rate_limits","set_model","set_mode","thread_compact","get_agent_global_config"]}:{}},e.eventQueue&&(this.aibotConfig.concurrency={max_concurrent:e.eventQueue.maxConcurrent,max_queued:e.eventQueue.maxQueued,queue_timeout_ms:e.eventQueue.queueTimeoutMs,cancelable_queued:e.eventQueue.cancelableQueued,cancelable_running:e.eventQueue.cancelableRunning}),this.conversationLog=e.logDir?new ze(e.logDir):null,this.packetLog=e.logDir?new Ge(e.logDir):null,this.bindingStore=new it(e.bindingsPath),this.bindingStore.load(),this.globalConfigStore=o??null,this.deferredMgr=new _t(this.name),this.activeEventStore=e.activeEventStorePath?new dt(e.activeEventStorePath):null,t==="codex")this.sessionScanCache=new k($e,Be);else if(t==="claude")this.sessionScanCache=new k(ye,Fe);else if(t==="pi"){const s=e.agent.env?.PI_CODING_AGENT_DIR;this.sessionScanCache=new k(()=>He(void 0,s),()=>qe(void 0,s))}else if(t==="codewhale")this.sessionScanCache=new k(Le,Oe);else if(t==="cursor")this.sessionScanCache=new k(Ie,We);else if(t==="opencode")this.sessionScanCache=new k(De,Ne);else if(t==="agy")this.sessionScanCache=new k(()=>[],()=>[]);else{e.aibot.clientType?.trim().toLowerCase()==="reasonix"&&(this.reasonixTitleScan=new k(Me,Ke));const s=Te(e.aibot.clientType);this.sessionScanCache=s?new k(()=>xe(void 0,s),()=>Ue(void 0,s)):new k(()=>[],()=>[])}}async start(){await Ae({agentName:this.name,adapterType:this.config.adapterType??"acp",command:this.config.agent.command,piConfigDir:this.config.agent.env?.PI_CODING_AGENT_DIR,hasProvider:!!this.config.agent.provider,boundCwds:[...this.bindingStore.entries()].map(([,t])=>String(t.cwd??""))}).then(t=>{t.length>0&&u.info(this.name,`restored native provider config after relay disable: ${t.join(", ")}`)}).catch(t=>{u.warn(this.name,`native provider residue sweep failed: ${t instanceof Error?t.message:String(t)}`)}),(this.config.adapterType??"acp")==="claude"&&(await B().catch(()=>{}),this.maybeQueryProviderQuota().catch(()=>{}));const e=!!(this.config.providerBaseUrl&&this.config.providerApiKey||this.config.agent.provider?.baseUrl&&this.config.agent.provider?.apiKey);(this.config.aibot.clientType==="kiro"||this.config.aibot.clientType==="kimi"||e)&&(this.maybeQueryProviderQuota().catch(()=>{}),this.startProviderQuotaTimer()),(this.config.adapterType??"acp")==="codex"&&this.maybeQueryProviderQuota().catch(()=>{}),await this.connectAibot(),this.sendCtrl.bind(this.aibotHandle);const o=this.config.adapterType??"acp";this.pool=new Je({maxPoolSize:this.config.poolMaxSize??20,idleTimeoutMs:this.config.poolIdleTimeoutMs??18e5,eventQueue:this.config.eventQueue},t=>{const s=this.createAdapter(o,t);return s instanceof A&&s.on("acpSessionReady",i=>{this.bindingStore.setAcpSessionId(t,i),this.sessionScanCache.invalidate()}),s},(t,s)=>{this.aibotHandle.sendEventAck({event_id:t,session_id:s,received_at:Date.now()})}),this.pool.setEventStateHandler((t,s,i,n)=>{u.info(this.name,`[queue-debug] send event_state session=${s} event=${t} state=${i} queue_pos=${n?.queue_position??""} queue_total=${n?.queue_total??""}`),this.aibotHandle.sendEventState({event_id:t,session_id:s,state:i,content_preview:n?.content_preview,content:n?.content,queue_position:n?.queue_position,queue_total:n?.queue_total,actions:n?.actions,reason:n?.reason,held:n?.held,held_reason:n?.held_reason,updated_at:Date.now()}),this.pushQueueSnapshotForSession(s),(i==="canceled"||i==="failed")&&(this.aibotHandle.sendEventResult({event_id:t,status:i==="canceled"?"canceled":"failed",msg:n?.reason,updated_at:Date.now()}),this.discardEventTrackingState(t))}),this.pool.setQueueComposingHandler((t,s,i)=>{this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:s,...s?{ttl_ms:3e4,ref_event_id:i}:{}})}),this.pool.setInternalErrorHandler(t=>{this.handleSessionInternalError(t).catch(s=>{u.error(this.name,`[recovery] handleSessionInternalError failed event=${t.eventId} session=${t.sessionId}: ${s instanceof Error?s.message:String(s)}`)})}),this.pool.setEventStartedHandler((t,s)=>{if(this.reportSessionSkills(s),this.config.adapterType!=="claude")return;this.claudeWorkerStatus.set(s,"busy");const i=this.bindingStore.get(s);i?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"busy",cwd:i.cwd,meta:this.buildClaudeToolbarMeta(s)})}),this.pool.setEventDoneHandler((t,s)=>{const i=this.config.adapterType??"acp";if(i==="claude"){this.claudeWorkerStatus.set(s,"ready");const n=this.bindingStore.get(s);n?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:n.cwd,meta:this.buildClaudeToolbarMeta(s)})}else if(i==="agy"){const n=this.bindingStore.get(s);if(n?.cwd){const a=this.buildAgyToolbarMeta(s),r=a?.available_models??[];u.info(this.name,`[agy-toolbar-diag] eventDone push binding card session=${s} model_id=${String(a?.model_id??"")} available_models=${r.length} meta_keys=${a?Object.keys(a).join(","):"<none>"}`),this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:"ready",cwd:n.cwd,meta:a})}else u.info(this.name,`[agy-toolbar-diag] eventDone skip binding card: no binding cwd session=${s}`)}}),this.pool.setSessionActivityHandler((t,s,i)=>{const n=this.selfDrivenSessions.has(t),a=this.selfDrivenLabels.get(t);s?(this.selfDrivenSessions.add(t),i&&this.selfDrivenLabels.set(t,i),this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:!0,ttl_ms:9e4})):(this.selfDrivenSessions.delete(t),this.selfDrivenLabels.delete(t),n&&this.aibotHandle.sendSessionActivitySet({session_id:t,kind:"composing",active:!1})),(n!==s||s&&i!==void 0&&i!==a)&&this.pushQueueSnapshotForSession(t)}),this.pool.startIdleSweep(),u.info(this.name,`Ready (adapter: ${o}, poolMax: ${this.config.poolMaxSize??20})`)}async stop(){this.stopped=!0,this.pool?.stopIdleSweep(),this.stopProviderQuotaTimer();const e=this.pool?.collectActiveEventIds()??[];e.length>0&&this.activeEventStore&&await this.activeEventStore.save(e);for(const s of e)u.info(this.name,`Canceling active event on shutdown: ${s}`),this.sendEventResultWithCleanup(s,"canceled","process shutting down");e.length>0&&await new Promise(s=>setTimeout(s,100)),this.pool?.clearActiveEventsForShutdown();const o=this.deferredMgr.getAllDeferredEvents();for(const s of o)u.info(this.name,`Failing deferred event on shutdown: ${s.event_id}`),this.sendEventResultWithCleanup(s.event_id,"failed","process shutting down");const t=this.pool?.drainAllQueuedEvents()??[];for(const s of t)u.info(this.name,`Failing queued event on shutdown: ${s.event_id}`),this.sendEventResultWithCleanup(s.event_id,"failed","process shutting down");this.deferredMgr.clearAll(),await this.pool?.stop(),this.aibotHandle?.disconnect(),e.length>0&&this.activeEventStore&&await this.activeEventStore.save([]),this.eventSessionIndex.clear(),this.inflightEvents.clear(),this.restartCount.clear()}resolveSpawnEnv(){const e=Ce(this.config.agent.clientType,this.config.agent.provider),o=this.config.agent.env,t=Object.keys(e).length>0?{...e,...o}:o;return ve(t,this.name,this.config.agent.clientType)}createAdapter(e,o){switch(e){case"claude":return this.createClaudeAdapter(o);case"codex":return this.createCodexAdapter(o);case"pi":return this.createPiAdapter(o);case"openhuman":return this.createOpenHumanAdapter(o);case"codewhale":return this.createCodeWhaleAdapter(o);case"cursor":return this.createCursorAdapter(o);case"opencode":return this.createOpenCodeAdapter(o);case"agy":return this.createAgyAdapter(o);default:return this.createAcpAdapter(o)}}createCursorAdapter(e){const o={...this.config.adapterOptions??{}};o.bindingStore=this.bindingStore,o.aibotSessionId=e;const t=this.bindingStore.get(e),s=this.resolveSessionModelId(e);s&&(o.model=s);const i=this.bindingStore.getCursorModeId(e),n=i??"full_auto";i||this.bindingStore.setCursorModeId(e,n),o.mode=n,t?.cwd&&(o.workspace=t.cwd);const a={sendStreamChunk:(r,d,c,l,h,m,v)=>{this.sendStreamChunkByRuntimeConfig(r,d,c,l,h,m,v)},sendEventResult:(r,d,c)=>{this.sendEventResultWithCleanup(r,d,c)},sendEventAck:(r,d)=>{this.aibotHandle.sendEventAck({event_id:r,session_id:d,received_at:Date.now()})},sendRawEventEnvelope:(r,d,c)=>{this.deliverRawEventEnvelope(r,d,c,"cursor",this.buildCursorRawEventFallbackText(c))},agentInvoke:async(r,d,c)=>this.platformInvoke(r,d,c),sendLocalActionResult:(r,d,c,l,h)=>{this.aibotHandle.sendLocalActionResult({action_id:r,status:d,...c!==void 0?{result:c}:{},...l?{error_code:l}:{},...h?{error_msg:h}:{}})},sendUpdateBindingCard:(r,d,c,l)=>{this.aibotHandle.sendUpdateBindingCard({session_id:r,worker_status:d,cwd:c,...l?{meta:l}:{}})},getAgentProfile:()=>this.agentProfile};return new M({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:o},a)}createClaudeAdapter(e){const o={sendReply:(i,n,a,r,d)=>{this.sendReplyByRuntimeConfig(i,n,a,r,d)},sendStreamChunk:(i,n,a,r,d,c,l)=>{this.sendStreamChunkByRuntimeConfig(i,n,a,r,d,c,l)},sendMedia:(i,n,a,r,d,c,l)=>{this.aibotHandle.sendMedia({event_id:i,session_id:n,content:a,msg_type:2,quoted_message_id:d||void 0,client_msg_id:c||void 0,extra:l?{media_caption:r,...l}:{media_caption:r}})},sendEventResult:(i,n,a,r)=>{this.sendEventResultWithCleanup(i,n,a,r)},sendEventAck:(i,n)=>{this.aibotHandle.sendEventAck({event_id:i,session_id:n,received_at:Date.now()})},agentInvoke:async(i,n,a)=>this.platformInvoke(i,n,a),sendLocalActionResult:(i,n,a,r,d)=>{this.aibotHandle.sendLocalActionResult({action_id:i,status:n,...a!==void 0?{result:a}:{},...r?{error_code:r}:{},...d?{error_msg:d}:{}})},sendToolUse:(i,n,a,r)=>{this.sendToolExecutionCard(i,n,$(a,r))},sendToolResult:(i,n,a,r)=>{this.sendToolExecutionCard(i,n,y(a,r))},getWsUrl:()=>this.config.aibot.url,getAgentId:()=>this.config.aibot.agentId,getAgentProfile:()=>this.agentProfile,getApiKey:()=>this.config.aibot.apiKey,getActiveEventCount:()=>0,getPendingPermissionCount:()=>0,getPendingElicitationCount:()=>0,sendAgentQuestionCard:(i,n,a)=>{const r=a.questions.map(c=>c.header).join(", "),d=mt(`[Agent Question] ${a.request_id}`,"agent_question",a);this.aibotHandle.sendText({event_id:i,session_id:n,content:d,msg_type:1,extra:{card_type:"agent_question",summary_text:r}})},sendPermissionCard:i=>{this.aibotHandle.sendMsg({event_id:i.eventId,session_id:i.sessionId,client_msg_id:`perm_${Q()}`,msg_type:1,content:i.toolTitle?`Permission required: ${i.toolTitle}`:"Permission request",extra:{channel_data:{execApproval:{approvalId:i.approvalId,approvalSlug:i.toolName},grix:{execApproval:{approval_command_id:i.approvalId,command:i.toolTitle||i.toolName,host:"claude"}}},agent_api_origin:!0}})},sendDirectMessage:i=>{this.aibotHandle.sendMsg({session_id:i.sessionId,msg_type:1,content:i.content,...i.clientMsgId?{client_msg_id:i.clientMsgId}:{},...i.quotedMessageId?{quoted_message_id:i.quotedMessageId}:{}})},onStatusLineUpdated:i=>{(i.rateLimits?.fiveHour||i.rateLimits?.sevenDay)&&(this.cachedClaudeRateLimitState=i);const n=this.bindingStore.get(e);n?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:this.claudeWorkerStatus.get(e)??"ready",cwd:n.cwd,meta:this.buildClaudeToolbarMeta(e)})},sendMcpFrame:i=>{this.aibotHandle.sendMcpFrame(e,i)}},t=this.config.adapterOptions??{},s={...t,sessionRuntimeResolver:()=>{const i=this.bindingStore.get(e);return{cwd:i?.cwd,modeId:i?.modeId??C.fullAuto,modelId:this.resolveSessionModelId(e),pluginDir:t.pluginDir,claudeSessionId:i?.claudeSessionId,onSessionIdAssigned:n=>{this.bindingStore.setClaudeSessionId(e,n),this.sessionScanCache.invalidate()}}}};return new P({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:s},o)}createCodexAdapter(e){let o=null;const t={sendEventResult:(n,a,r)=>{this.sendEventResultWithCleanup(n,a,r)},sendEventAck:(n,a)=>this.aibotHandle.sendEventAck({event_id:n,session_id:a,received_at:Date.now()}),sendCodexEvent:n=>{this.shouldDropCodexDisplayEvent(n.event_id,n.codex_method)||(this.aibotHandle.sendCodexEvent(this.stampCodexEventQuote(n)),this.logCodexEventToConversation(n))},sendCodexEventReliable:async n=>{if(!this.shouldDropCodexDisplayEvent(n.event_id,n.codex_method)){try{await this.aibotHandle.sendCodexEventReliable(this.stampCodexEventQuote(n))}catch(a){u.warn("bridge",`[codex] sendCodexEventReliable ACK failed event=${n.event_id}: ${a}`)}u.info("bridge",`[codex] sendCodexEventReliable done event=${n.event_id} method=${n.codex_method}`),this.logCodexEventToConversation(n)}},sendRunError:(n,a,r)=>{this.sendStreamChunkByRuntimeConfig(n,a,`
|
|
2
2
|
|
|
3
|
-
Error: ${
|
|
3
|
+
Error: ${r}`,1,!1)},sendUpdateBindingCard:(n,a,r,d)=>{const c={...d??{}};if(!c.rate_limits&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const l=this.providerQuotaToCodexRateLimits(this.cachedProviderQuota);l&&(c.rate_limits=l.rateLimits,c.rate_limit_primary_percent=l.primaryPercent,c.rate_limit_secondary_percent=l.secondaryPercent,c.rate_limit_primary_window_min=l.primaryWindowMin,c.rate_limit_secondary_window_min=l.secondaryWindowMin)}!c.provider_quota&&this.cachedProviderQuota?.success&&(c.provider_quota=this.cachedProviderQuota),this.aibotHandle.sendUpdateBindingCard({session_id:n,worker_status:a,cwd:r,...Object.keys(c).length>0?{meta:c}:{}})},agentInvoke:async(n,a,r)=>this.platformInvoke(n,a,r),sendLocalActionResult:(n,a,r,d,c)=>this.aibotHandle.sendLocalActionResult({action_id:n,status:a,...r!==void 0?{result:r}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}}),sendSessionActivitySet:(n,a,r,d)=>{this.aibotHandle.sendSessionActivitySet({session_id:n,kind:a,active:r,...d??{}})},getConversationLog:()=>this.conversationLog,getAgentProfile:()=>this.agentProfile,onRateLimitsUpdated:n=>{this.cachedRateLimits=n,this.cachedRateLimitsSampledAtMs=Date.now(),this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const a=this.bindingStore.get(e);if(a?.cwd){const r=this.cachedRateLimitsSampledAtMs,d={rate_limits:{...n.primary.windowMinutes>0?{primary:n.primary}:{},...n.secondary.windowMinutes>0?{secondary:n.secondary}:{},sampledAt:r},rate_limit_primary_percent:n.primary.usedPercent,rate_limit_secondary_percent:n.secondary.usedPercent,rate_limit_primary_window_min:n.primary.windowMinutes,rate_limit_secondary_window_min:n.secondary.windowMinutes,...o?.getEffortMeta()};this.cachedProviderQuota?.success&&(d.provider_quota=this.cachedProviderQuota),this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:a.cwd,meta:d})}},onContextWindowUpdated:n=>{if(!n)return;this.cachedCodexContextWindow=n,this.cachedCodexUsageSampledAtMs=Date.now();const a=this.bindingStore.get(e);a?.cwd&&this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:a.cwd,meta:{context_window:n,...o?.getEffortMeta()}})},onTokenUsageUpdated:n=>{n&&(this.cachedCodexTokenUsage=n,this.cachedCodexUsageSampledAtMs=Date.now())}},s=this.config.adapterOptions??{},i=this.globalConfigStore?.get(this.name);return o=new oe({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:{...s,model:this.resolveCodexSessionModelId(e)??s.model,collaborationMode:this.bindingStore.getCodexModeId(e)??s.collaborationMode,reasoningEffort:i?.codexReasoningEffort??s.reasoningEffort,serviceTier:this.bindingStore.getCodexServiceTier(e)??s.serviceTier,sandboxMode:i?.codexSandboxMode??s.sandboxMode,aibotSessionId:e,bindingStore:this.bindingStore}},t),o}createCodeWhaleAdapter(e){const o={sendEventResult:(s,i,n)=>{this.sendEventResultWithCleanup(s,i,n)},sendEventAck:(s,i)=>this.aibotHandle.sendEventAck({event_id:s,session_id:i,received_at:Date.now()}),sendStreamChunk:(s,i,n,a,r,d)=>{this.sendStreamChunkByRuntimeConfig(s,i,n,a,r,d)},sendUpdateBindingCard:(s,i,n,a)=>this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:i,cwd:n,...this.providerQuotaMetaPayload(a,s,n)}),sendLocalActionResult:(s,i,n,a,r)=>this.aibotHandle.sendLocalActionResult({action_id:s,status:i,...n!==void 0?{result:n}:{},...a?{error_code:a}:{},...r?{error_msg:r}:{}}),sendSessionActivitySet:(s,i,n,a)=>{this.aibotHandle.sendSessionActivitySet({session_id:s,kind:i,active:n,...a??{}})},sendToolUse:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,$(n,a))},sendToolResult:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,y(n,a))},agentInvoke:async(s,i,n)=>this.platformInvoke(s,i,n),getConversationLog:()=>this.conversationLog,getAgentProfile:()=>this.agentProfile},t=this.config.adapterOptions??{};return new N({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:{...t,aibotSessionId:e,bindingStore:this.bindingStore,agentName:this.name,...this.config.agent.provider?{provider:this.config.agent.provider}:{}}},o)}createPiAdapter(e){const o={sendEventResult:(s,i,n)=>{this.sendEventResultWithCleanup(s,i,n),u.info("bridge",`[pi] sendEventResult event=${s} status=${i}`)},sendEventAck:(s,i)=>this.aibotHandle.sendEventAck({event_id:s,session_id:i,received_at:Date.now()}),sendUpdateBindingCard:(s,i,n,a)=>this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:i,cwd:n,...this.providerQuotaMetaPayload(a,s,n)}),agentInvoke:async(s,i,n)=>this.platformInvoke(s,i,n),sendLocalActionResult:(s,i,n,a,r)=>{this.aibotHandle.sendLocalActionResult({action_id:s,status:i,...n!==void 0?{result:n}:{},...a?{error_code:a}:{},...r?{error_msg:r}:{}})},sendSessionActivitySet:(s,i,n,a)=>{this.aibotHandle.sendSessionActivitySet({session_id:s,kind:i,active:n,...a??{}})},sendToolUse:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,$(n,a))},sendToolResult:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,y(n,a))},sendStreamChunk:(s,i,n,a,r,d)=>{this.sendStreamChunkByRuntimeConfig(s,i,n,a,r,d),r&&u.info("bridge",`[pi] sendFinalStreamChunk event=${s} seq=${a}`)},sendFinalStreamChunkReliable:this.reliableFinalWiring("pi"),sendThinking:(s,i,n)=>{this.sendThinkingByRuntimeConfig(s,i,n)},sendRunError:(s,i,n,a,r)=>{this.sendStreamChunkByRuntimeConfig(s,i,`
|
|
4
4
|
|
|
5
|
-
Error: ${n}`,
|
|
5
|
+
Error: ${n}`,a,!1,r)},getAgentProfile:()=>this.agentProfile},t=this.config.adapterOptions??{};return new re({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:{...t,aibotSessionId:e,bindingStore:this.bindingStore,agentName:this.name,...this.config.agent.provider?{provider:this.config.agent.provider}:{}}},o)}createOpenHumanAdapter(e){const o={sendStreamChunk:(s,i,n,a,r,d)=>{this.finalizeThinking(s,i),this.sendStreamChunkByRuntimeConfig(s,i,n,a,r,d)},sendFinalStreamChunkReliable:async(s,i,n,a)=>{if(this.finalizeThinking(s,i),this.resolveEventRuntimeConfig(s).responseDelivery==="single_message"&&s){this.bufferStreamChunk(s,i,"",!0);return}try{await this.aibotHandle.sendStreamChunkRequest({event_id:s,session_id:i,delta_content:"",chunk_seq:n,is_finish:!0,...a?{client_msg_id:a}:{}})}catch(d){u.warn("bridge",`[openhuman] sendFinalStreamChunkReliable ACK failed event=${s}: ${d}`)}},sendEventResult:(s,i,n)=>{this.sendEventResultWithCleanup(s,i,n),u.info("bridge",`[openhuman] sendEventResult event=${s} status=${i}`)},sendEventAck:(s,i)=>this.aibotHandle.sendEventAck({event_id:s,session_id:i,received_at:Date.now()}),sendToolUse:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,$(n,a))},sendToolResult:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,y(n,a))},sendThinking:(s,i,n)=>{this.sendThinkingByRuntimeConfig(s,i,n)},sendRunError:(s,i,n)=>{this.sendStreamChunkByRuntimeConfig(s,i,`
|
|
6
6
|
|
|
7
|
-
Error: ${n}`,0,!1)},sendUpdateBindingCard:(s,i,n,
|
|
7
|
+
Error: ${n}`,0,!1)},sendUpdateBindingCard:(s,i,n,a)=>this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:i,cwd:n,...this.providerQuotaMetaPayload(a,s,n)}),agentInvoke:async(s,i,n)=>this.platformInvoke(s,i,n),sendLocalActionResult:(s,i,n,a,r)=>{this.aibotHandle.sendLocalActionResult({action_id:s,status:i,...n!==void 0?{result:n}:{},...a?{error_code:a}:{},...r?{error_msg:r}:{}})},getAgentProfile:()=>this.agentProfile},t=this.config.adapterOptions??{};return new ae({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:t},o,{port:t.port,host:t.host,workspaceDir:t.workspace_dir,sessionToken:t.session_token,enableSessionBinding:!0,aibotSessionId:e})}createOpenCodeAdapter(e){const o={sendStreamChunk:(s,i,n,a,r,d)=>{this.sendStreamChunkByRuntimeConfig(s,i,n,a,r,d)},sendFinalStreamChunkReliable:this.reliableFinalWiring("opencode"),sendEventResult:(s,i,n)=>{this.sendEventResultWithCleanup(s,i,n),u.info("bridge",`[opencode] sendEventResult event=${s} status=${i}`)},sendEventAck:(s,i)=>this.aibotHandle.sendEventAck({event_id:s,session_id:i,received_at:Date.now()}),sendToolUse:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,$(n,a))},sendToolResult:(s,i,n,a)=>{this.sendToolExecutionCard(s,i,y(n,a))},sendThinking:(s,i,n)=>{this.sendThinkingByRuntimeConfig(s,i,n)},sendRunError:(s,i,n)=>{this.sendStreamChunkByRuntimeConfig(s,i,`
|
|
8
8
|
|
|
9
|
-
Error: ${n}`,0,!1)},sendUpdateBindingCard:(s,i,n,
|
|
9
|
+
Error: ${n}`,0,!1)},sendUpdateBindingCard:(s,i,n,a)=>this.aibotHandle.sendUpdateBindingCard({session_id:s,worker_status:i,cwd:n,...this.providerQuotaMetaPayload(a,s,n)}),agentInvoke:async(s,i,n)=>this.platformInvoke(s,i,n),sendLocalActionResult:(s,i,n,a,r)=>{this.aibotHandle.sendLocalActionResult({action_id:s,status:i,...n!==void 0?{result:n}:{},...a?{error_code:a}:{},...r?{error_msg:r}:{}})},getAgentProfile:()=>this.agentProfile},t=this.config.adapterOptions??{};return new le({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:t},o,{port:t.port,hostname:t.hostname,model:t.model,agent:t.agent,permissionPolicy:t.permission_policy,enableSessionBinding:!0,aibotSessionId:e,bindingStore:this.bindingStore,...this.config.agent.provider?{provider:this.config.agent.provider}:{}})}createAgyAdapter(e){const o={sendStreamChunk:(s,i,n,a,r,d)=>{this.sendStreamChunkByRuntimeConfig(s,i,n,a,r,d)},sendEventResult:(s,i,n)=>{this.sendEventResultWithCleanup(s,i,n)},sendEventAck:(s,i)=>{this.aibotHandle.sendEventAck({event_id:s,session_id:i,received_at:Date.now()})},agentInvoke:async(s,i,n)=>this.platformInvoke(s,i,n),forceCompleteInternalEvent:(s,i)=>{this.pool.eventComplete(s,i),this.pushQueueSnapshotForSession(i)},persistConversationId:(s,i)=>{this.bindingStore.setAgyConversationId(s,i)},getAgentProfile:()=>this.agentProfile},t=s=>{const i=this.bindingStore.get(s);return{cwd:i?.cwd,modelId:this.resolveSessionModelId(s),conversationId:i?.agyConversationId}};return new K({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv(),options:this.config.adapterOptions??{}},o,t)}createAcpAdapter(e){const o=this.isAcpRawTransportEnabled(),t={sendStreamChunk:(r,d,c,l,h,m,v)=>{this.finalizeThinking(r,d),this.sendStreamChunkByRuntimeConfig(r,d,c,l,h,m,v)},sendFinalStreamChunkReliable:this.reliableFinalWiring("acp",4e3),sendEventResult:(r,d,c)=>{this.sendEventResultWithCleanup(r,d,c)},sendEventAck:(r,d)=>{this.aibotHandle.sendEventAck({event_id:r,session_id:d,received_at:Date.now()})},agentInvoke:async(r,d,c)=>this.platformInvoke(r,d,c),sendLocalActionResult:(r,d,c,l,h)=>{this.aibotHandle.sendLocalActionResult({action_id:r,status:d,...c!==void 0?{result:c}:{},...l?{error_code:l}:{},...h?{error_msg:h}:{}})},sendRawEventEnvelope:(r,d,c)=>{this.sendAcpRawEventEnvelope(r,d,c)},sendToolUse:(r,d,c,l)=>{if(o){this.sendAcpRawEventEnvelope(r,d,{type:"tool_use",payload:{tool_name:c,tool_input:l??""}});return}this.sendToolExecutionCard(r,d,$(c,l))},sendToolResult:(r,d,c,l)=>{this.sendToolExecutionCard(r,d,y(c,l))},sendThinking:(r,d,c)=>{this.sendThinkingByRuntimeConfig(r,d,c)},sendRunError:(r,d,c)=>{this.sendStreamChunkByRuntimeConfig(r,d,`
|
|
10
10
|
|
|
11
|
-
Error: ${c}`,1,!1)},sendPermissionCard:a=>{if(u.info("bridge","sendPermissionCard callback entered",{rawTransport:o,eventId:a.eventId,sessionId:a.sessionId,toolCallId:a.toolCallId,toolName:a.toolName,toolTitle:a.toolTitle}),o){this.sendAcpRawEventEnvelope(a.eventId,a.sessionId,{type:"permission_request",payload:{tool_call_id:a.toolCallId,tool_name:a.toolName,tool_title:a.toolTitle,options:a.options}}),u.info("bridge","sendPermissionCard: sent via rawEventEnvelope",{eventId:a.eventId,toolCallId:a.toolCallId});return}const d=`perm_${D()}`,c={event_id:a.eventId,session_id:a.sessionId,client_msg_id:d,msg_type:1,content:a.toolTitle?`Permission required: ${a.toolTitle}`:"Permission request",extra:{channel_data:{execApproval:{approvalId:a.toolCallId,approvalSlug:a.toolName},grix:{execApproval:{approval_command_id:a.toolCallId,command:a.toolTitle||a.toolName,host:"acp"}}},agent_api_origin:!0}};u.info("bridge","sendPermissionCard: about to invoke aibotHandle.sendMsg",{eventId:a.eventId,sessionId:a.sessionId,clientMsgId:d,toolCallId:a.toolCallId,contentLength:c.content.length});try{this.aibotHandle.sendMsg(c),u.info("bridge","sendPermissionCard: aibotHandle.sendMsg returned",{eventId:a.eventId,clientMsgId:d})}catch(l){u.error("bridge","sendPermissionCard: aibotHandle.sendMsg threw",{eventId:a.eventId,clientMsgId:d,error:l instanceof Error?l.message:String(l)})}},sendAuthNotification:(a,d)=>{a&&this.aibotHandle.sendMsg({session_id:a,msg_type:1,content:d,extra:{biz_card:{version:1,type:"agent_error",payload:{error:{name:"AuthRequired",message:d}}}}})},sendAgentMessage:(a,d)=>{a&&d&&this.aibotHandle.sendMsg({session_id:a,msg_type:1,content:d})},sendUpdateBindingCard:(a,d,c,l)=>{const h={...l??{}};if(!h.rate_limits&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const m=this.providerQuotaToRateLimits(this.cachedProviderQuota);m&&(h.rate_limits=m)}!h.provider_quota&&this.cachedProviderQuota?.success&&(h.provider_quota=this.cachedProviderQuota),this.aibotHandle.sendUpdateBindingCard({session_id:a,worker_status:d,cwd:c,...Object.keys(h).length>0?{meta:h}:{}})},onSkillsUpdate:a=>{try{this.aibotHandle.sendSkillsUpdate({skills:G(a,j.skills)})}catch(d){}},onContextWindowUpdated:a=>{this.cachedAcpContextWindow=a,this.cachedAcpContextWindowSampledAtMs=Date.now();const d="usedPercentage"in a?a.usedPercentage.toFixed(1):(a.used/a.size*100).toFixed(1);u.info(this.name,`[acp] context_window updated: ${d}%`);const c=this.bindingStore.get(e);if(c?.cwd){const l="usedPercentage"in a?a.usedPercentage:Math.min(100,a.used/a.size*100),h={context_window:{..."usedPercentage"in a?{}:a,usedPercentage:l,remainingPercentage:100-l}};if((this.config.aibot.clientType==="kiro"||this.config.aibot.clientType==="kimi")&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{}),h.provider_quota=this.cachedProviderQuota;const m=this.providerQuotaToRateLimits(this.cachedProviderQuota);m&&(h.rate_limits=m)}this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:c.cwd,meta:h})}},sendMcpFrame:a=>{this.aibotHandle.sendMcpFrame(e,a)},getAgentProfile:()=>this.agentProfile},s=e?this.bindingStore.get(e):void 0,i=this.globalConfigStore?.get(this.name),{initialModel:n,initialMode:r}=Ze({sessionBinding:s,globalDefaults:i,configInitialMode:this.config.acpInitialMode});return e&&n&&!s?.acpModelId&&this.bindingStore.setAcpModelId(e,n),(n||r)&&u.info(this.name,`[toolbar] hydrate from binding: session=${e} model=${n??"<none>"} mode=${r??"<none>"}`),new A({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv()},t,{acpAuthMethod:this.config.acpAuthMethod,acpInitialMode:r,acpInitialModel:n,acpMcpTools:this.config.acpMcpTools,rawTransport:o,eventResultsPath:this.config.eventResultsPath,approvalMode:this.config.approvalMode,bindingStore:this.config.enableSessionBinding?this.bindingStore:void 0,aibotSessionId:e,autoInjectArgs:this.config.autoInjectArgs,nativeProviderScope:this.name,bridgeLog:this.config.logDir?new ze(this.config.logDir,`${this.name}-${e}`):null})}async connectAibot(){const e=new se;this.aibotHandle=await e.connect(this.aibotConfig,{aborted:()=>this.stopped,label:this.name,packetLog:this.packetLog,maxRetries:this.config.connectMaxRetries});const o=this.aibotHandle.authAck;if(this.applyAgentProfile(o?.agent_name,o?.introduction,{source:"auth_ack",respawnOnChange:!1}),this.aibotHandle.onEvent(t=>{this.handleAibotEvent(t).catch(s=>{u.error(this.name,`handleAibotEvent failed: ${s}`),this.aibotHandle.sendEventAck({event_id:t.event_id,session_id:t.session_id,received_at:Date.now()});const i=s instanceof Error?s.message:String(s);if(/CWD must be|Bound directory does not exist|Bound path is not a directory/i.test(i)&&t.session_id){this.bindingStore.delete(t.session_id),this.sessionBindings.delete(t.session_id),this.sessionProviderHints.delete(t.session_id),this.sessionProviderQuotas.delete(t.session_id),this.sessionProviderMeta.delete(t.session_id);const r=this.pool.getSlot(t.session_id);r?.adapter instanceof A&&r.adapter.getSessionBindings().delete(t.session_id);const a=this.config.adapterType??"acp",d=this.resolveBindingChannelKey(a);this.aibotHandle.sendMsg({event_id:t.event_id,session_id:t.session_id,msg_type:1,content:i,extra:{channel_data:{[d]:{sessionBinding:{status:"missing",reason:"binding_stale",error_code:g.invalidCwd}}}},quoted_message_id:t.msg_id});return}this.aibotHandle.sendEventResult({event_id:t.event_id,status:"failed",msg:i,updated_at:Date.now()})})}),this.aibotHandle.onStop(t=>{try{this.handleAibotStop(t)}catch(s){u.error(this.name,`handleAibotStop failed: ${s}`)}}),this.aibotHandle.onAgentDeleted(t=>{u.warn(this.name,`agent deleted on platform (source=${t?.source??"unknown"}${t?.reason?` reason=${t.reason}`:""}), notifying manager to clean up`);try{this.agentDeletedHandler?.()}catch(s){u.error(this.name,`agentDeletedHandler failed: ${s}`)}}),this.aibotHandle.onShareSet(t=>{try{this.shareSetHandler?.(Array.isArray(t?.shared_to)?t.shared_to:[])}catch(s){u.error(this.name,`onShareSet failed: ${s}`)}}),this.aibotHandle.onProfilePush(t=>{try{this.applyAgentProfile(t?.agent_name,t?.introduction,{source:"profile_push",respawnOnChange:!0})}catch(s){u.error(this.name,`onProfilePush failed: ${s}`)}}),this.aibotHandle.onSkillSync(t=>{try{u.info(this.name,`skill_sync received owner=${t?.owner_id??""} name=${t?.name??""}`),this.skillSyncHandler?.()}catch(s){u.error(this.name,`onSkillSync failed: ${s}`)}}),this.aibotHandle.onRevoke(t=>{try{this.handleAibotRevoke(t)}catch(s){u.error(this.name,`handleAibotRevoke failed: ${s}`)}}),this.aibotHandle.onLocalAction(t=>{this.handleAibotLocalAction(t).catch(s=>{u.error(this.name,`handleAibotLocalAction failed: ${s}`)})}),this.aibotHandle.onEventCancel(t=>{u.info(this.name,`recv event_cancel event_id=${t.event_id} session_id=${t.session_id}`),this.handleEventCancel(t).catch(s=>{u.error(this.name,`handleEventCancel failed: ${s}`)})}),this.aibotHandle.onMcpFrame((t,s)=>{const i=this.pool.getSlot(t)?.adapter;i?.deliverMcpFrameToAgent?i.deliverMcpFrameToAgent(s):u.warn(this.name,`mcp_frame: no adapter for session=${t}`)}),this.aibotHandle.onQueueClear(t=>{const s=this.pool.clearQueue(t.session_id);this.aibotHandle.sendQueueClearResult({session_id:t.session_id,canceled_event_ids:s}),this.pushQueueSnapshotForSession(t.session_id)}),this.aibotHandle.onQueueReorder(t=>{const s=Array.isArray(t.ordered_event_ids)?t.ordered_event_ids.filter(n=>typeof n=="string"&&n.length>0):[],i=this.pool.reorderQueue(t.session_id,s);this.aibotHandle.sendQueueReorderResult({session_id:t.session_id,applied_event_ids:i}),this.pushQueueSnapshotForSession(t.session_id)}),this.aibotHandle.onEventHold(t=>{const s=typeof t.session_id=="string"?t.session_id:"",i=typeof t.event_id=="string"?t.event_id:"",n=t.hold!==!1;if(u.info(this.name,`recv event_hold event_id=${i} session_id=${s} hold=${n} reason=${t.reason??""}`),!s||!i){this.aibotHandle.sendEventHoldResult({session_id:s,event_id:i,ok:!1,held:!1,error:"bad_request"});return}const r=typeof t.reason=="string"?t.reason:"",a=typeof t.ttl_ms=="number"?t.ttl_ms:void 0,d=this.pool.holdEvent(s,i,n,r,a),c=d==="ok";this.aibotHandle.sendEventHoldResult({session_id:s,event_id:i,ok:c,held:c?n:!1,...c?{}:{error:d}}),c&&this.pushQueueSnapshotForSession(s)}),this.aibotHandle.onQueueEdit(t=>{const s=typeof t.session_id=="string"?t.session_id:"",i=typeof t.event_id=="string"?t.event_id:"";if(u.info(this.name,`recv queue_edit event_id=${i} session_id=${s}`),!s||!i){this.aibotHandle.sendQueueEditResult({session_id:s,event_id:i,ok:!1,error:"bad_request"});return}const n=typeof t.content=="string"?t.content:"",r=this.pool.editQueuedEvent(s,i,n),a=r==="ok";this.aibotHandle.sendQueueEditResult({session_id:s,event_id:i,ok:a,...a?{}:{error:r}}),a&&this.pushQueueSnapshotForSession(s)}),this.aibotHandle.onQueueSnapshotQuery(t=>{this.replyQueueSnapshotForSession(t.session_id)}),u.info(this.name,"Connected to aibot"),this.activeEventStore){const t=await this.activeEventStore.drain();if(t.length>0){u.warn(this.name,`Recovering ${t.length} stale event(s) from previous run`);for(const s of t)u.info(this.name,`Failing stale event on startup: ${s}`),this.aibotHandle.sendEventResult({event_id:s,status:"failed",msg:"process restarted, event lost",updated_at:Date.now()})}}this.pushQueueSnapshots(),this.aibotHandle.onReconnected(()=>{this.pushQueueSnapshots();const t=this.aibotHandle.authAck;this.applyAgentProfile(t?.agent_name,t?.introduction,{source:"reconnect",respawnOnChange:!0})}),this.aibotHandle.onStreamRejected((t,s)=>{this.sendCtrl.markEventRejected(t)})}applyAgentProfile(e,o,t){const s=String(e??"").trim(),i=String(o??"").trim(),n=s!==this.agentProfile.agentName||i!==this.agentProfile.introduction;this.agentProfile={agentName:s,introduction:i},s||i?u.info(this.name,`agent profile (${t.source}): GOT name="${s}" intro_len=${i.length} changed=${n}`):u.warn(this.name,`agent profile (${t.source}): EMPTY \u2014 \u670D\u52A1\u7AEF\u672A\u4E0B\u53D1 agent_name/introduction\uFF08auth_ack \u5B57\u6BB5\u7F3A\u5931\u6216\u503C\u4E3A\u7A7A\uFF09`),n&&t.respawnOnChange&&this.applyProfileChangeToAdapters("agent_profile_changed")}applyProfileChangeToAdapters(e){if(!this.pool)return;const o=this.pool.getAllSlots();let t=0;for(const r of o)if(r.adapter.onAgentProfileChanged)try{r.adapter.onAgentProfileChanged(),t++}catch(a){u.warn(this.name,`onAgentProfileChanged failed for session=${r.sessionId}: ${a}`)}const s=o.filter(r=>r.adapter instanceof P),i=s.filter(r=>{if(r.state!=="ready")return!1;const a=r.adapter.getStatus();return!a.busy&&!a.backgroundBusy}),n=s.length-i.length;u.info(this.name,`${e}: notified ${t} adapter(s) via hook; respawning ${i.length} idle Claude slot(s), skipping ${n} busy`);for(const r of i)this.pool.removeSlot(r.sessionId).catch(a=>{u.warn(this.name,`removeSlot failed during ${e}: ${a}`)})}pushQueueSnapshots(){if(!(!this.config.eventQueue||!this.pool))for(const e of this.pool.getAllSlots())this.pushQueueSnapshotForSession(e.sessionId)}buildQueueSnapshotPayload(e){const o=this.pool?.getQueueSnapshot(e)??null,t=o?[...o.running]:[],s=o?o.running_items.map(n=>({event_id:n.event_id,...n.content_preview?{content_preview:n.content_preview}:{},...n.title?{title:n.title}:{},...n.summary?{summary:n.summary}:{},actions:[{type:"stop"}]})):[],i=o?o.queued.map(n=>({event_id:n.event_id,position:n.position,...n.content_preview?{content_preview:n.content_preview}:{},...typeof n.content=="string"?{content:n.content}:{},...n.title?{title:n.title}:{},...n.summary?{summary:n.summary}:{},held:n.held===!0,held_reason:n.held_reason??"",actions:[{type:"cancel"}]})):[];if(t.length===0&&this.selfDrivenSessions.has(e)&&this.pool?.getSlot(e)){const n=`selfdrive_${e}`,r=this.selfDrivenLabels.get(e)??"Background task in progress";t.push(n),s.push({event_id:n,content_preview:r,title:r,summary:r,actions:[]})}return{session_id:e,running:t,running_items:s,queued:i}}pushQueueSnapshotForSession(e){if(!this.config.eventQueue||!this.pool)return;const o=this.buildQueueSnapshotPayload(e);u.info(this.name,`[queue-debug] push snapshot session=${e} running=${o.running.length} queued=${o.queued.length} running_ids=[${o.running.join(",")}]`),this.aibotHandle.sendQueueSnapshot(o)}replyQueueSnapshotForSession(e){!this.config.eventQueue||!this.pool||this.aibotHandle.sendQueueSnapshot(this.buildQueueSnapshotPayload(e))}async platformInvoke(e,o,t){return e==="file_link"?nt(o):e==="file_upload"?this.uploadFileAndSendMedia(o):this.aibotHandle.agentInvoke(e,o,t)}async uploadFileAndSendMedia(e){const o=String(e.file_path??"").trim(),t=String(e.session_id??"").trim(),s=String(e.caption??"").trim(),i=String(e.reply_to_message_id??"").trim();if(!o)throw new Error("file_path is required");if(!t)throw new Error("session_id is required");const n=await ot({wsURL:this.config.aibot.url,apiKey:this.config.aibot.apiKey,sessionID:t,filePath:o}),r=await this.aibotHandle.sendMedia({session_id:t,msg_type:2,content:s||`[${n.attachment_type}]`,client_msg_id:`file_upload_${D()}`,...i?{quoted_message_id:i}:{},extra:n.extra});if(r.cmd!=="send_ack"){const d=r.payload??{},c=String(d.msg??r.cmd);throw new Error(`media message send failed: ${c}`)}const a=r.payload??{};return{ok:!0,file_name:n.file_name,attachment_type:n.attachment_type,access_url:n.access_url,message_id:a.msg_id!=null?String(a.msg_id):null}}sendReplyByRuntimeConfig(e,o,t,s,i){this.indexEventSession(e,o),this.sendCtrl.sendReply(e,o,t,s,i),t&&this.conversationLog?.logOutbound?.(o,e,"reply",t)}stampCodexEventQuote(e){if(e.quoted_message_id!=null){if(e.quoted_message_id)return e;const{quoted_message_id:t,...s}=e;return s}const o=this.sendCtrl.getDefaultQuotedMessageId(e.event_id);return o?{...e,quoted_message_id:o}:e}discardEventTrackingState(e){this.inflightEvents.delete(e),this.restartCount.delete(e),this.eventSessionIndex.delete(e),this.sendCtrl.discardEventState(e)}reliableFinalWiring(e,o){return async(t,s,i)=>{try{await this.sendCtrl.sendFinalStreamChunkReliable(t,s,i,o)}catch(n){u.warn("bridge",`[${e}] sendFinalStreamChunkReliable ACK failed event=${t}: ${n}`)}u.info("bridge",`[${e}] sendFinalStreamChunkReliable done event=${t}`)}}sendStreamChunkByRuntimeConfig(e,o,t,s,i,n,r){this.indexEventSession(e,o),this.sendCtrl.sendStreamChunk(e,o,t,s,i,n,r),(t||i)&&this.conversationLog?.logOutbound?.(o,e,i?"stream_chunk_finish":"stream_chunk",t)}sendRunErrorAsChunk(e,o,t){this.sendStreamChunkByRuntimeConfig(e,o,`
|
|
11
|
+
Error: ${c}`,1,!1)},sendPermissionCard:r=>{if(u.info("bridge","sendPermissionCard callback entered",{rawTransport:o,eventId:r.eventId,sessionId:r.sessionId,toolCallId:r.toolCallId,toolName:r.toolName,toolTitle:r.toolTitle}),o){this.sendAcpRawEventEnvelope(r.eventId,r.sessionId,{type:"permission_request",payload:{tool_call_id:r.toolCallId,tool_name:r.toolName,tool_title:r.toolTitle,options:r.options}}),u.info("bridge","sendPermissionCard: sent via rawEventEnvelope",{eventId:r.eventId,toolCallId:r.toolCallId});return}const d=`perm_${Q()}`,c={event_id:r.eventId,session_id:r.sessionId,client_msg_id:d,msg_type:1,content:r.toolTitle?`Permission required: ${r.toolTitle}`:"Permission request",extra:{channel_data:{execApproval:{approvalId:r.toolCallId,approvalSlug:r.toolName},grix:{execApproval:{approval_command_id:r.toolCallId,command:r.toolTitle||r.toolName,host:"acp"}}},agent_api_origin:!0}};u.info("bridge","sendPermissionCard: about to invoke aibotHandle.sendMsg",{eventId:r.eventId,sessionId:r.sessionId,clientMsgId:d,toolCallId:r.toolCallId,contentLength:c.content.length});try{this.aibotHandle.sendMsg(c),u.info("bridge","sendPermissionCard: aibotHandle.sendMsg returned",{eventId:r.eventId,clientMsgId:d})}catch(l){u.error("bridge","sendPermissionCard: aibotHandle.sendMsg threw",{eventId:r.eventId,clientMsgId:d,error:l instanceof Error?l.message:String(l)})}},sendAuthNotification:(r,d)=>{r&&this.aibotHandle.sendMsg({session_id:r,msg_type:1,content:d,extra:{biz_card:{version:1,type:"agent_error",payload:{error:{name:"AuthRequired",message:d}}}}})},sendAgentMessage:(r,d)=>{r&&d&&this.aibotHandle.sendMsg({session_id:r,msg_type:1,content:d})},sendUpdateBindingCard:(r,d,c,l)=>{const h={...l??{}};if(!h.rate_limits&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const m=this.providerQuotaToRateLimits(this.cachedProviderQuota);m&&(h.rate_limits=m)}!h.provider_quota&&this.cachedProviderQuota?.success&&(h.provider_quota=this.cachedProviderQuota),this.aibotHandle.sendUpdateBindingCard({session_id:r,worker_status:d,cwd:c,...Object.keys(h).length>0?{meta:h}:{}})},onSkillsUpdate:r=>{try{this.aibotHandle.sendSkillsUpdate({skills:G(r,j.skills)})}catch(d){}},onContextWindowUpdated:r=>{this.cachedAcpContextWindow=r,this.cachedAcpContextWindowSampledAtMs=Date.now();const d="usedPercentage"in r?r.usedPercentage.toFixed(1):(r.used/r.size*100).toFixed(1);u.info(this.name,`[acp] context_window updated: ${d}%`);const c=this.bindingStore.get(e);if(c?.cwd){const l="usedPercentage"in r?r.usedPercentage:Math.min(100,r.used/r.size*100),h={context_window:{..."usedPercentage"in r?{}:r,usedPercentage:l,remainingPercentage:100-l}};if((this.config.aibot.clientType==="kiro"||this.config.aibot.clientType==="kimi")&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{}),h.provider_quota=this.cachedProviderQuota;const m=this.providerQuotaToRateLimits(this.cachedProviderQuota);m&&(h.rate_limits=m)}this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:c.cwd,meta:h})}},sendMcpFrame:r=>{this.aibotHandle.sendMcpFrame(e,r)},getAgentProfile:()=>this.agentProfile},s=e?this.bindingStore.get(e):void 0,i=this.globalConfigStore?.get(this.name),{initialModel:n,initialMode:a}=tt({sessionBinding:s,globalDefaults:i,configInitialMode:this.config.acpInitialMode});return e&&n&&!s?.acpModelId&&this.bindingStore.setAcpModelId(e,n),(n||a)&&u.info(this.name,`[toolbar] hydrate from binding: session=${e} model=${n??"<none>"} mode=${a??"<none>"}`),new A({command:this.config.agent.command,args:this.config.agent.args,env:this.resolveSpawnEnv()},t,{acpAuthMethod:this.config.acpAuthMethod,acpInitialMode:a,acpInitialModel:n,acpMcpTools:this.config.acpMcpTools,rawTransport:o,eventResultsPath:this.config.eventResultsPath,approvalMode:this.config.approvalMode,bindingStore:this.config.enableSessionBinding?this.bindingStore:void 0,aibotSessionId:e,autoInjectArgs:this.config.autoInjectArgs,nativeProviderScope:this.name,bridgeLog:this.config.logDir?new je(this.config.logDir,`${this.name}-${e}`):null})}async connectAibot(){const e=new se;this.aibotHandle=await e.connect(this.aibotConfig,{aborted:()=>this.stopped,label:this.name,packetLog:this.packetLog,maxRetries:this.config.connectMaxRetries});const o=this.aibotHandle.authAck;if(this.applyAgentProfile(o?.agent_name,o?.introduction,{source:"auth_ack",respawnOnChange:!1}),this.aibotHandle.onEvent(t=>{this.handleAibotEvent(t).catch(s=>{u.error(this.name,`handleAibotEvent failed: ${s}`),this.aibotHandle.sendEventAck({event_id:t.event_id,session_id:t.session_id,received_at:Date.now()});const i=s instanceof Error?s.message:String(s);if(/CWD must be|Bound directory does not exist|Bound path is not a directory/i.test(i)&&t.session_id){this.bindingStore.delete(t.session_id),this.sessionBindings.delete(t.session_id),this.sessionProviderHints.delete(t.session_id),this.sessionProviderQuotas.delete(t.session_id),this.sessionProviderMeta.delete(t.session_id);const a=this.pool.getSlot(t.session_id);a?.adapter instanceof A&&a.adapter.getSessionBindings().delete(t.session_id);const r=this.config.adapterType??"acp",d=this.resolveBindingChannelKey(r);this.aibotHandle.sendMsg({event_id:t.event_id,session_id:t.session_id,msg_type:1,content:i,extra:{channel_data:{[d]:{sessionBinding:{status:"missing",reason:"binding_stale",error_code:g.invalidCwd}}}},quoted_message_id:t.msg_id});return}this.aibotHandle.sendEventResult({event_id:t.event_id,status:"failed",msg:i,updated_at:Date.now()})})}),this.aibotHandle.onStop(t=>{try{this.handleAibotStop(t)}catch(s){u.error(this.name,`handleAibotStop failed: ${s}`)}}),this.aibotHandle.onAgentDeleted(t=>{u.warn(this.name,`agent deleted on platform (source=${t?.source??"unknown"}${t?.reason?` reason=${t.reason}`:""}), notifying manager to clean up`);try{this.agentDeletedHandler?.()}catch(s){u.error(this.name,`agentDeletedHandler failed: ${s}`)}}),this.aibotHandle.onShareSet(t=>{try{this.shareSetHandler?.(Array.isArray(t?.shared_to)?t.shared_to:[])}catch(s){u.error(this.name,`onShareSet failed: ${s}`)}}),this.aibotHandle.onProfilePush(t=>{try{this.applyAgentProfile(t?.agent_name,t?.introduction,{source:"profile_push",respawnOnChange:!0})}catch(s){u.error(this.name,`onProfilePush failed: ${s}`)}}),this.aibotHandle.onSkillSync(t=>{try{u.info(this.name,`skill_sync received owner=${t?.owner_id??""} name=${t?.name??""}`),this.skillSyncHandler?.()}catch(s){u.error(this.name,`onSkillSync failed: ${s}`)}}),this.aibotHandle.onRevoke(t=>{try{this.handleAibotRevoke(t)}catch(s){u.error(this.name,`handleAibotRevoke failed: ${s}`)}}),this.aibotHandle.onLocalAction(t=>{this.handleAibotLocalAction(t).catch(s=>{u.error(this.name,`handleAibotLocalAction failed: ${s}`)})}),this.aibotHandle.onEventCancel(t=>{u.info(this.name,`recv event_cancel event_id=${t.event_id} session_id=${t.session_id}`),this.handleEventCancel(t).catch(s=>{u.error(this.name,`handleEventCancel failed: ${s}`)})}),this.aibotHandle.onMcpFrame((t,s)=>{const i=this.pool.getSlot(t)?.adapter;i?.deliverMcpFrameToAgent?i.deliverMcpFrameToAgent(s):u.warn(this.name,`mcp_frame: no adapter for session=${t}`)}),this.aibotHandle.onQueueClear(t=>{const s=this.pool.clearQueue(t.session_id);this.aibotHandle.sendQueueClearResult({session_id:t.session_id,canceled_event_ids:s}),this.pushQueueSnapshotForSession(t.session_id)}),this.aibotHandle.onQueueReorder(t=>{const s=Array.isArray(t.ordered_event_ids)?t.ordered_event_ids.filter(n=>typeof n=="string"&&n.length>0):[],i=this.pool.reorderQueue(t.session_id,s);this.aibotHandle.sendQueueReorderResult({session_id:t.session_id,applied_event_ids:i}),this.pushQueueSnapshotForSession(t.session_id)}),this.aibotHandle.onEventHold(t=>{const s=typeof t.session_id=="string"?t.session_id:"",i=typeof t.event_id=="string"?t.event_id:"",n=t.hold!==!1;if(u.info(this.name,`recv event_hold event_id=${i} session_id=${s} hold=${n} reason=${t.reason??""}`),!s||!i){this.aibotHandle.sendEventHoldResult({session_id:s,event_id:i,ok:!1,held:!1,error:"bad_request"});return}const a=typeof t.reason=="string"?t.reason:"",r=typeof t.ttl_ms=="number"?t.ttl_ms:void 0,d=this.pool.holdEvent(s,i,n,a,r),c=d==="ok";this.aibotHandle.sendEventHoldResult({session_id:s,event_id:i,ok:c,held:c?n:!1,...c?{}:{error:d}}),c&&this.pushQueueSnapshotForSession(s)}),this.aibotHandle.onQueueEdit(t=>{const s=typeof t.session_id=="string"?t.session_id:"",i=typeof t.event_id=="string"?t.event_id:"";if(u.info(this.name,`recv queue_edit event_id=${i} session_id=${s}`),!s||!i){this.aibotHandle.sendQueueEditResult({session_id:s,event_id:i,ok:!1,error:"bad_request"});return}const n=typeof t.content=="string"?t.content:"",a=this.pool.editQueuedEvent(s,i,n),r=a==="ok";this.aibotHandle.sendQueueEditResult({session_id:s,event_id:i,ok:r,...r?{}:{error:a}}),r&&this.pushQueueSnapshotForSession(s)}),this.aibotHandle.onQueueSnapshotQuery(t=>{this.replyQueueSnapshotForSession(t.session_id)}),u.info(this.name,"Connected to aibot"),this.activeEventStore){const t=await this.activeEventStore.drain();if(t.length>0){u.warn(this.name,`Recovering ${t.length} stale event(s) from previous run`);for(const s of t)u.info(this.name,`Failing stale event on startup: ${s}`),this.aibotHandle.sendEventResult({event_id:s,status:"failed",msg:"process restarted, event lost",updated_at:Date.now()})}}this.pushQueueSnapshots(),this.aibotHandle.onReconnected(()=>{this.pushQueueSnapshots();const t=this.aibotHandle.authAck;this.applyAgentProfile(t?.agent_name,t?.introduction,{source:"reconnect",respawnOnChange:!0})}),this.aibotHandle.onStreamRejected((t,s)=>{this.sendCtrl.markEventRejected(t)})}applyAgentProfile(e,o,t){const s=String(e??"").trim(),i=String(o??"").trim(),n=s!==this.agentProfile.agentName||i!==this.agentProfile.introduction;this.agentProfile={agentName:s,introduction:i},s||i?u.info(this.name,`agent profile (${t.source}): GOT name="${s}" intro_len=${i.length} changed=${n}`):u.warn(this.name,`agent profile (${t.source}): EMPTY \u2014 \u670D\u52A1\u7AEF\u672A\u4E0B\u53D1 agent_name/introduction\uFF08auth_ack \u5B57\u6BB5\u7F3A\u5931\u6216\u503C\u4E3A\u7A7A\uFF09`),n&&t.respawnOnChange&&this.applyProfileChangeToAdapters("agent_profile_changed")}applyProfileChangeToAdapters(e){if(!this.pool)return;const o=this.pool.getAllSlots();let t=0;for(const a of o)if(a.adapter.onAgentProfileChanged)try{a.adapter.onAgentProfileChanged(),t++}catch(r){u.warn(this.name,`onAgentProfileChanged failed for session=${a.sessionId}: ${r}`)}const s=o.filter(a=>a.adapter instanceof P),i=s.filter(a=>{if(a.state!=="ready")return!1;const r=a.adapter.getStatus();return!r.busy&&!r.backgroundBusy}),n=s.length-i.length;u.info(this.name,`${e}: notified ${t} adapter(s) via hook; respawning ${i.length} idle Claude slot(s), skipping ${n} busy`);for(const a of i)this.pool.removeSlot(a.sessionId).catch(r=>{u.warn(this.name,`removeSlot failed during ${e}: ${r}`)})}pushQueueSnapshots(){if(!(!this.config.eventQueue||!this.pool))for(const e of this.pool.getAllSlots())this.pushQueueSnapshotForSession(e.sessionId)}buildQueueSnapshotPayload(e){const o=this.pool?.getQueueSnapshot(e)??null,t=o?[...o.running]:[],s=o?o.running_items.map(n=>({event_id:n.event_id,...n.content_preview?{content_preview:n.content_preview}:{},...n.title?{title:n.title}:{},...n.summary?{summary:n.summary}:{},actions:[{type:"stop"}]})):[],i=o?o.queued.map(n=>({event_id:n.event_id,position:n.position,...n.content_preview?{content_preview:n.content_preview}:{},...typeof n.content=="string"?{content:n.content}:{},...n.title?{title:n.title}:{},...n.summary?{summary:n.summary}:{},held:n.held===!0,held_reason:n.held_reason??"",actions:[{type:"cancel"}]})):[];if(t.length===0&&this.selfDrivenSessions.has(e)&&this.pool?.getSlot(e)){const n=`selfdrive_${e}`,a=this.selfDrivenLabels.get(e)??"Background task in progress";t.push(n),s.push({event_id:n,content_preview:a,title:a,summary:a,actions:[]})}return{session_id:e,running:t,running_items:s,queued:i}}pushQueueSnapshotForSession(e){if(!this.config.eventQueue||!this.pool)return;const o=this.buildQueueSnapshotPayload(e);u.info(this.name,`[queue-debug] push snapshot session=${e} running=${o.running.length} queued=${o.queued.length} running_ids=[${o.running.join(",")}]`),this.aibotHandle.sendQueueSnapshot(o)}replyQueueSnapshotForSession(e){!this.config.eventQueue||!this.pool||this.aibotHandle.sendQueueSnapshot(this.buildQueueSnapshotPayload(e))}async platformInvoke(e,o,t){return e==="file_link"?ot(o):e==="file_upload"?this.uploadFileAndSendMedia(o):this.aibotHandle.agentInvoke(e,o,t)}async uploadFileAndSendMedia(e){const o=String(e.file_path??"").trim(),t=String(e.session_id??"").trim(),s=String(e.caption??"").trim(),i=String(e.reply_to_message_id??"").trim();if(!o)throw new Error("file_path is required");if(!t)throw new Error("session_id is required");const n=await at({wsURL:this.config.aibot.url,apiKey:this.config.aibot.apiKey,sessionID:t,filePath:o}),a=await this.aibotHandle.sendMedia({session_id:t,msg_type:2,content:s||`[${n.attachment_type}]`,client_msg_id:`file_upload_${Q()}`,...i?{quoted_message_id:i}:{},extra:n.extra});if(a.cmd!=="send_ack"){const d=a.payload??{},c=String(d.msg??a.cmd);throw new Error(`media message send failed: ${c}`)}const r=a.payload??{};return{ok:!0,file_name:n.file_name,attachment_type:n.attachment_type,access_url:n.access_url,message_id:r.msg_id!=null?String(r.msg_id):null}}sendReplyByRuntimeConfig(e,o,t,s,i){this.indexEventSession(e,o),this.sendCtrl.sendReply(e,o,t,s,i),t&&this.conversationLog?.logOutbound?.(o,e,"reply",t)}stampCodexEventQuote(e){if(e.quoted_message_id!=null){if(e.quoted_message_id)return e;const{quoted_message_id:t,...s}=e;return s}const o=this.sendCtrl.getDefaultQuotedMessageId(e.event_id);return o?{...e,quoted_message_id:o}:e}discardEventTrackingState(e){this.inflightEvents.delete(e),this.restartCount.delete(e),this.eventSessionIndex.delete(e),this.sendCtrl.discardEventState(e)}reliableFinalWiring(e,o){return async(t,s,i)=>{try{await this.sendCtrl.sendFinalStreamChunkReliable(t,s,i,o)}catch(n){u.warn("bridge",`[${e}] sendFinalStreamChunkReliable ACK failed event=${t}: ${n}`)}u.info("bridge",`[${e}] sendFinalStreamChunkReliable done event=${t}`)}}sendStreamChunkByRuntimeConfig(e,o,t,s,i,n,a){this.indexEventSession(e,o),this.sendCtrl.sendStreamChunk(e,o,t,s,i,n,a),(t||i)&&this.conversationLog?.logOutbound?.(o,e,i?"stream_chunk_finish":"stream_chunk",t)}sendRunErrorAsChunk(e,o,t){this.sendStreamChunkByRuntimeConfig(e,o,`
|
|
12
12
|
|
|
13
|
-
Error: ${t}`,1,!1)}sendEventResultWithCleanup(e,o,t,s){this.sendCtrl.sendEventResult(e,o,t,s);const i=this.eventSessionIndex.get(e);i&&(this.pool.eventComplete(e,i),this.pushQueueSnapshotForSession(i),this.conversationLog?.logResult?.(i,e,o,t),this.eventSessionIndex.delete(e)),this.inflightEvents.delete(e),this.restartCount.delete(e),o==="responded"&&(this.cachedProviderQuotaSampledAtMs=null,this.refreshAndPushProviderQuota(!0).catch(()=>{}))}async handleSessionInternalError(e){const{eventId:o,sessionId:t,errorMsg:s}=e;if(this.stopped)return;const i=this.inflightEvents.get(o);if(!i){u.warn(this.name,`[recovery] no inflight event for internalError event=${o} session=${t}; surface failure directly`),this.sendRunErrorAsChunk(o,t,s),this.sendEventResultWithCleanup(o,"failed",s,"agent_stop_failure");return}const n=(this.restartCount.get(o)??0)+1;this.restartCount.set(o,n);const r=this.config.adapterType??"acp";if(n>W){u.error(this.name,`[recovery] adapter=${r} session=${t} event=${o} restart=${n}/${W} outcome=give-up err=${s}`),this.sendRunErrorAsChunk(o,t,s),this.sendEventResultWithCleanup(o,"failed",s,"agent_stop_failure");return}u.info(this.name,`[recovery] adapter=${r} session=${t} event=${o} restart=${n}/${W} outcome=restarting err=${s}`);const a=this.pool.drainQueuedForSession(t);a.length>0&&u.info(this.name,`[recovery] session=${t} preserved ${a.length} queued sibling event(s) across restart`);try{await this.pool.removeSlot(t)}catch(l){u.warn(this.name,`[recovery] removeSlot failed session=${t}: ${l instanceof Error?l.message:String(l)}`)}if(this.stopped)return;const d=this.resolveRecoveryPrompt(r,i),c={...i,content:d};try{await this.pool.deliverInboundEvent(c)}catch(l){u.error(this.name,`[recovery] redeliver failed event=${o} session=${t}: ${l instanceof Error?l.message:String(l)}`),this.sendEventResultWithCleanup(o,"failed",l instanceof Error?l.message:String(l));return}for(const l of a){if(this.stopped)break;try{await this.pool.deliverInboundEvent(l)}catch(h){u.error(this.name,`[recovery] sibling redeliver failed event=${l.event_id} session=${t}: ${h instanceof Error?h.message:String(h)}`),this.sendEventResultWithCleanup(l.event_id,"failed",h instanceof Error?h.message:String(h))}}}resolveRecoveryPrompt(e,o){return e==="acp"?"continue":o.content}sendThinkingByRuntimeConfig(e,o,t){this.sendCtrl.sendThinking(e,o,t)}bufferStreamChunk(e,o,t,s,i){this.sendCtrl.bufferOnly(e,o,t,s,i)}flushBufferedStreamText(e){}resolveEventRuntimeConfig(e){return this.sendCtrl.resolveEventRuntimeConfig(e)}captureEventRuntimeConfig(e){this.sendCtrl.captureEventRuntimeConfig(e),this.indexEventSession(e.event_id,e.session_id),e.event_id&&!this.inflightEvents.has(e.event_id)&&this.inflightEvents.set(e.event_id,e)}indexEventSession(e,o){!e||!o||this.eventSessionIndex.set(e,o)}shouldDropToolDisplayEvent(e){return this.sendCtrl.shouldDropToolDisplayEvent(e)}shouldDropThinkingDisplayEvent(e){return this.sendCtrl.shouldDropThinkingDisplayEvent(e)}shouldDropCodexDisplayEvent(e,o){return this.sendCtrl.shouldDropCodexDisplayEvent(e,o)}logCodexEventToConversation(e){if(!this.conversationLog||e.codex_method!=="item/agentMessage/delta")return;const t=e.codex_payload?.params?.delta;if(!t)return;const s=this.eventSessionIndex.get(e.event_id)??e.session_id;this.conversationLog.append(s,{ts:Date.now(),dir:"outbound",event_id:e.event_id,kind:"codex_delta",text_len:t.length,content:t})}isAcpRawTransportEnabled(){return(this.config.adapterOptions??{}).raw_transport===!0}shouldDropAcpRawDisplayEvent(e,o){return this.sendCtrl.shouldDropAcpRawDisplayEvent(e,o)}sendAcpRawEventEnvelope(e,o,t){this.shouldDropAcpRawDisplayEvent(e,t.type)||this.aibotHandle.sendMsg({event_id:e,session_id:o,msg_type:1,content:this.buildAcpRawEventFallbackText(t),extra:{channel_data:{acp:{raw_event:t}},agent_api_origin:!0}})}buildAcpRawEventFallbackText(e){const o=String(e.type??"").trim();if(!o)return"[acp] event";switch(o){case"permission_request":return`Permission required: ${String(e.payload?.tool_title??e.payload?.tool_name??"permission request")}`;case"tool_use":return`[tool] ${String(e.payload?.tool_name??"tool")}`;case"tool_result":return"[tool result]";case"thinking":return"[thinking]";case"error":return`[error] ${String(e.payload?.message??"agent error")}`;case"result":return"[result]";default:return`[acp] ${o}`}}sendToolExecutionCard(e,o,t,s){this.sendCtrl.sendToolExecutionCard(e,o,t,s)}async handleAibotEvent(e){if(this.relayEnvStale&&!this.hasPendingWork()&&await this.recycleAdaptersForRelayChange(),this.stopped){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:"agent shutting down",updated_at:Date.now()});return}this.logInboundConversation(e);const o=this.config.adapterType??"acp",t=ct(e.extra);if(t.error){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:"connector_config_invalid",msg:t.error,updated_at:Date.now()});return}const s=t.patch,i=Je(e);if(i){if(i.verb===p.exec){await this.handleSessionControlCommand(i,e);return}if(i.verb===p.listSessions){await this.handleListSessionsTextCommand(e);return}if(o==="claude"){await this.handleSessionControlCommand(i,e);return}if(o==="codex"&&i.verb===p.open){await this.handleCodexSessionControlOpen(i,e);return}if(o==="pi"&&i.verb===p.open){await this.handlePiSessionControlOpen(i,e);return}if(o==="pi"&&i.verb===p.restart){await this.handlePiSessionControlRestart(e);return}if((o==="openhuman"||o==="opencode")&&i.verb===p.open){await this.handleOpenHumanSessionControlOpen(i,e);return}if(o==="codewhale"&&i.verb===p.open){await this.handleCodeWhaleSessionControlOpen(i,e);return}if(this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),i.verb===p.open){const a=i.args.trim();if(!a){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.cwdRequired,msg:"cwd is required",updated_at:Date.now()});return}try{const d=x.resolve(a);if(!(await I(d)).isDirectory()){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.invalidCwd,msg:`Path is not a directory: ${d}`,updated_at:Date.now()});return}}catch(d){const c=String(d?.code??""),l=c==="ENOENT"?`Directory does not exist: ${x.resolve(a)}`:c==="EACCES"||c==="EPERM"?"Directory is not accessible":`Invalid path: ${a}`;this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.invalidCwd,msg:l,updated_at:Date.now()});return}}if(i.verb===p.open){const a=this.bindingStore.get(e.session_id);if(a?.cwd)try{await I(a.cwd)}catch{u.info("bridge",`Stale binding detected for session ${e.session_id}: ${a.cwd} no longer exists, clearing`),this.bindingStore.delete(e.session_id),this.sessionBindings.delete(e.session_id),this.sessionProviderHints.delete(e.session_id),this.sessionProviderQuotas.delete(e.session_id),this.sessionProviderMeta.delete(e.session_id);const d=this.pool.getSlot(e.session_id);d?.adapter instanceof A&&d.adapter.getSessionBindings().delete(e.session_id)}}if(await this.handleSessionControlForPool(i,e),o==="acp"&&i.verb===p.stop){const d=this.bindingStore.get(e.session_id)?.cwd??"";await this.pool.removeSlot(e.session_id).catch(()=>{}),this.sessionBindings.delete(e.session_id),this.sessionProviderHints.delete(e.session_id),this.sessionProviderQuotas.delete(e.session_id),this.sessionProviderMeta.delete(e.session_id),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:`Session worker stopped for ${d}`,updated_at:Date.now()});return}if(V(i,e,this.sessionControlCtx(e.session_id),{...this.sessionControlSenders(),sendEventAck:()=>{}}),i.verb===p.open&&(await this.deferredMgr.release(e.session_id,this.deferredCallbacks()),o==="agy")){const a=this.bindingStore.get(e.session_id)?.cwd??"";a&&this.aibotHandle.sendUpdateBindingCard({session_id:e.session_id,worker_status:"ready",cwd:a,meta:this.buildAgyToolbarMeta(e.session_id)})}return}if(e.mirror_mode==="record_only"){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",updated_at:Date.now()});return}if(this.isStaleEvent(e)){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:"event_stale",msg:"event is stale and will not be processed",updated_at:Date.now()});return}if(bt.has(o)&&!this.bindingStore.get(e.session_id)?.cwd){const d=o;u.info(this.name,`[${d}] binding missing session_id=${e.session_id} event_id=${e.event_id}`),this.deferredMgr.defer(d,String(e.session_id??"").trim(),this.buildInboundEvent(e,s));const c=this.resolveBindingChannelKey(o);this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendMsg({event_id:e.event_id,session_id:e.session_id,msg_type:1,content:"Session binding missing.",extra:{channel_data:{[c]:{sessionBinding:{status:"missing",reason:"binding_missing",error_code:g.bindingMissing}}}},quoted_message_id:e.msg_id});return}if((this.config.adapterType??"acp")==="acp"){const d=String(e.content??"").trim().match(/^\/(\S+)\s*(.*)/);if(d){const[,c,l]=d,m=this.pool.getSlot(e.session_id)?.adapter;if(m?.execCommand&&(m.getSupportedCommands?.()??[]).some(f=>f.name===c||f.name===`/${c}`)){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()});try{const f=await m.execCommand(c,l.trim(),e.session_id);f.status==="options"&&f.data&&this.handleExecCommandOptions(e.session_id,c,f.data),this.aibotHandle.sendEventResult({event_id:e.event_id,status:f.status==="failed"?"failed":"responded",msg:f.message,updated_at:Date.now()})}catch(f){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:f instanceof Error?f.message:String(f),updated_at:Date.now()})}return}}}const r=this.buildInboundEvent(e,s);try{this.captureEventRuntimeConfig(r),await this.pool.deliverInboundEvent(r)}catch(a){if(this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.discardEventTrackingState(e.event_id),a instanceof Ve)this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:a.message,updated_at:Date.now()});else throw a}}async handleCodexSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(r,a)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:r,...a,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const r=await this.resolveCwdForBinding(s),a=this.bindingStore.get(t);if(a?.cwd){let d=!1;try{if(await this.resolveCwdForBinding(a.cwd)===r){i(),n("responded",{msg:`Session already bound to ${a.cwd}`});return}}catch{d=!0,u.info("bridge",`Stale codex binding for session ${t}: ${a.cwd} no longer exists, allowing rebind`)}if(!d){i(),n("failed",{msg:`Session already bound to ${a.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}}this.bindingStore.set(t,r),this.sessionBindings.set(t,r),this.deferredMgr.sendCodexDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,r),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:r}),i(),n("responded",{msg:`Session bound to ${r}`})}catch(r){i(),n("failed",{code:g.invalidCwd,msg:r instanceof Error?r.message:String(r)})}}async handleSessionControlForPool(e,o){e.verb===p.open&&await this.bindSessionForPool(o.session_id,e.args.trim())}async replayDeferredEventsForSession(e){await this.deferredMgr.release(e,this.deferredCallbacks())}async handleSessionControlLocalActionForPool(e){if(String(e.action_type??"")!==b.sessionControl)return;const o=e.params??{};if(String(o.verb??"").trim().toLowerCase()!==p.open)return;const s=String(o.session_id??"").trim(),i=String(o.cwd??"").trim();!s||!i||await this.bindSessionForPool(s,i)}async handleCodexSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(r,a,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:r,...a!==void 0?{result:a}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const r=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,r);const a=this.bindingStore.get(t);if(a?.cwd){let d=!1;try{const c=await this.resolveCwdForBinding(a.cwd);if(c===r){this.setResolvedAgentSessionId(t,i),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,c)});return}}catch{d=!0,u.info("bridge",`Stale codex binding for session ${t}: ${a.cwd} no longer exists, allowing rebind`)}if(!d){n("failed",void 0,g.rebindForbidden,`Session already bound to ${a.cwd}`);return}}this.bindingStore.set(t,r),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,r),this.deferredMgr.sendCodexDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,r),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,r)})}catch(r){n("failed",void 0,r?.sessionControlErrorCode??g.invalidCwd,r instanceof Error?r.message:String(r))}}async handleCursorSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(r,a,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:r,...a!==void 0?{result:a}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const r=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,r);const a=this.bindingStore.get(t);if(a?.cwd){let d=!1;try{const c=await this.resolveCwdForBinding(a.cwd);if(c===r){this.setResolvedAgentSessionId(t,i),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,c)});return}}catch{d=!0,u.info("bridge",`Stale cursor binding for session ${t}: ${a.cwd} no longer exists, allowing rebind`)}if(!d){n("failed",void 0,g.rebindForbidden,`Session already bound to ${a.cwd}`);return}}this.bindingStore.set(t,r),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,r),this.deferredMgr.sendCursorDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,r),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:r}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,r)})}catch(r){n("failed",void 0,r?.sessionControlErrorCode??g.invalidCwd,r instanceof Error?r.message:String(r))}}async handlePiSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(r,a)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:r,...a,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const r=await this.resolveCwdForBinding(s),a=this.bindingStore.get(t);if(a?.cwd){let d=!1;try{if(await this.resolveCwdForBinding(a.cwd)===r){i(),n("responded",{msg:`Session already bound to ${a.cwd}`}),this.aibotHandle.sendMsg({event_id:o.event_id,session_id:t,msg_type:1,content:`\u2705 Session already bound to \`${a.cwd}\``,quoted_message_id:o.msg_id});return}}catch{d=!0,u.info("bridge",`Stale pi binding for session ${t}: ${a.cwd} no longer exists, allowing rebind`)}if(!d){i(),n("failed",{msg:`Session already bound to ${a.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}}this.bindingStore.set(t,r),this.sessionBindings.set(t,r),await this.ensureSlotStarted(t).catch(d=>{u.warn("bridge",`pi ensureSlotStarted on bind failed (non-fatal): ${d instanceof Error?d.message:String(d)}`)}),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:r}),i(),n("responded",{msg:`Session bound to ${r}`}),this.aibotHandle.sendMsg({event_id:o.event_id,session_id:t,msg_type:1,content:`\u2705 Working directory bound: \`${r}\``,quoted_message_id:o.msg_id})}catch(r){i(),n("failed",{code:g.invalidCwd,msg:r instanceof Error?r.message:String(r)})}}async handlePiSessionControlRestart(e){const o=e.session_id,t=()=>this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:o,received_at:Date.now()}),s=(r,a)=>this.aibotHandle.sendEventResult({event_id:e.event_id,status:r,...a,updated_at:Date.now()}),n=this.bindingStore.get(o)?.cwd??"";if(!n){t(),s("failed",{msg:"session binding was not found",code:g.bindingMissing});return}t(),await this.pool.removeSlot(o).catch(()=>{}),this.aibotHandle.sendUpdateBindingCard({session_id:o,worker_status:"ready",cwd:n}),s("responded",{msg:`Session worker restarted for ${n}`})}async handlePiSessionControlRestartLocalAction(e){const o=e.params??{},t=String(o.session_id??"").trim(),i=(t?this.bindingStore.get(t):void 0)?.cwd??"",n=(r,a,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:r,...a!==void 0?{result:a}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t){n("failed",void 0,"session_id_required","session_id is required for restart");return}if(!i){n("failed",void 0,g.bindingMissing,"Session binding missing. Open a workspace first.");return}await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:i}),n("ok",{outcome:"restarted",binding:{aibotSessionId:t,cwd:i,workerStatus:"ready"}})}async bindSessionForPool(e,o){const t=String(o??"").trim();if(!e||!t)return;const s=this.pool.getOrCreateSlot(e);if(!s||(s.startPromise&&await s.startPromise,!s.adapter))return;const i=s.adapter instanceof A?s.adapter:null;if(!i?.hasSessionBinding)return;const n=await this.resolveCwdForBinding(t);i.announceDeferredComposing(e),await i.bindSession(e,n),this.sessionBindings.set(e,n),this.sessionScanCache.invalidate(),i.replayDeferredEvents(e)}deferredCallbacks(){return{captureEventRuntimeConfig:e=>this.captureEventRuntimeConfig(e),deliverInboundEvent:e=>this.pool.deliverInboundEvent(e),sendEventResult:(e,o,t)=>this.sendEventResultWithCleanup(e,o,t),sendSessionComposing:(e,o,t)=>{const s={};o&&(s.ttl_ms=t?.ttlMs??3e4,t?.activity&&(s.activity=t.activity)),this.aibotHandle.sendSessionActivitySet({session_id:e,kind:"composing",active:o,...s})}}}async handleOpenHumanSessionControlOpen(e,o){const t=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:o.session_id,received_at:Date.now()});try{await this.resolveCwdForBinding(e.args.trim()),await this.handleSessionControlForPool(e,o),V(e,o,this.sessionControlCtx(o.session_id),this.sessionControlSenders()),await this.deferredMgr.release(o.session_id,this.deferredCallbacks())}catch(s){t(),this.aibotHandle.sendEventResult({event_id:o.event_id,status:"failed",code:s?.cwdErrorCode,msg:s instanceof Error?s.message:String(s),updated_at:Date.now()})}}async handleCodeWhaleSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(r,a)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:r,...a,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const r=await this.resolveCwdForBinding(s),a=this.bindingStore.get(t);if(a?.cwd){if(await this.resolveCwdForBinding(a.cwd)!==r){i(),n("failed",{msg:`Session already bound to ${a.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}i(),n("responded",{msg:`Session already bound to ${a.cwd}`});return}this.bindingStore.set(t,r),this.sessionBindings.set(t,r),await this.bindSessionForPool(t,r),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:r}),i(),n("responded",{msg:`Session bound to ${r}`})}catch(r){i(),n("failed",{code:g.invalidCwd,msg:r instanceof Error?r.message:String(r)})}}async handleCodeWhaleSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(r,a,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:r,...a!==void 0?{result:a}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const r=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,r);const a=this.bindingStore.get(t);if(a?.cwd){const d=await this.resolveCwdForBinding(a.cwd);if(d!==r){n("failed",void 0,g.rebindForbidden,`Session already bound to ${a.cwd}`);return}this.setResolvedAgentSessionId(t,i),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,d)});return}this.bindingStore.set(t,r),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,r),await this.bindSessionForPool(t,r),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:r}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,r)})}catch(r){n("failed",void 0,r?.sessionControlErrorCode??g.invalidCwd,r instanceof Error?r.message:String(r))}}normalizeClaudeModeId(e){return String(e??"").trim().toLowerCase()===C.approval?C.approval:C.fullAuto}handleExecCommandOptions(e,o,t){const s=Array.isArray(t?.options)?t.options:[];if(s.length===0)return;const n=this.bindingStore.get(e)?.cwd??"",r={};if(o==="model"){r.available_models=s.map(d=>({id:d.id,displayName:d.label}));const a=s.find(d=>d.current);a&&(r.model_id=a.id)}else if(o==="mode"){r.available_modes=s.map(d=>({id:d.id,displayName:d.label}));const a=s.find(d=>d.current);a&&(r.mode_id=a.id)}else r[`${o}_options`]=s;this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:n,meta:r})}resolveSessionModelId(e){const o=this.bindingStore.getModelId(e);if(o)return o;const t=String(this.globalConfigStore?.get(this.name)?.modelId??"").trim();if(t)return this.bindingStore.setModelId(e,t),t}resolveCodexSessionModelId(e){const o=this.bindingStore.getCodexModelId(e);if(o)return o;const t=String(this.globalConfigStore?.get(this.name)?.codexModelId??"").trim();if(t)return this.bindingStore.setCodexModelId(e,t),t}buildAgyToolbarMeta(e){if(!this.bindingStore.get(e))return;const t=ce(this.config.agent.command),s=t.length>0?t[0].id:"";let i=this.resolveSessionModelId(e)||s;t.length>0&&!t.some(r=>r.id===i)&&(i=s);const n=le();return{model_id:i,currentModelId:i,available_models:t.map(r=>({id:r.id,displayName:r.displayName})),...n.plan!==void 0&&{plan:n.plan},...n.quota_exhausted!==void 0&&{quota_exhausted:n.quota_exhausted},...n.quota_reset_at!==void 0&&{quota_reset_at:n.quota_reset_at},...n.available_credits!==void 0&&{available_credits:n.available_credits}}}async handleAgySetModel(e,o){const t=e.params??{},s=String(t.model_id??t.modelId??"").trim();if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for set_model"});return}if(!s){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"set_model_invalid",error_msg:"model_id is required"});return}const i=this.bindingStore.get(o);if(!i?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"session binding was not found"});return}i.modelId!==s&&(this.bindingStore.setModelId(o,s),this.globalConfigStore?.set(this.name,{modelId:s})),this.aibotHandle.sendUpdateBindingCard({session_id:o,worker_status:"ready",cwd:i.cwd,meta:this.buildAgyToolbarMeta(o)}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"model_set",model_id:s,binding:{cwd:i.cwd,model_id:s}}})}buildClaudeToolbarMeta(e){const o=this.bindingStore.get(e);if(!o)return;const t=F(),s=t.length>0?t[0].id:"";let i=this.resolveSessionModelId(e)||s;t.length>0&&!t.some(l=>l.id===i)&&(i=s);const n=this.normalizeClaudeModeId(o.modeId),r={model_id:i,mode_id:n,currentModelId:i,currentModeId:n,available_models:t.map(l=>({id:l.id,displayName:l.displayName}))},a=this.pool.getSlot(e),d=a?.adapter instanceof P?a.adapter.getSessionState():null;(d?.rateLimits?.fiveHour||d?.rateLimits?.sevenDay)&&(this.cachedClaudeRateLimitState=d);const c=d??this.getFreshClaudeRateLimitState();if(c){const l=c.rateLimits;l&&(l.fiveHour||l.sevenDay)&&(r.rate_limits={...l.fiveHour?{fiveHour:l.fiveHour}:{},...l.sevenDay?{sevenDay:l.sevenDay}:{},sampledAt:c.sampledAt||Date.now()})}if(d){const l=d.contextWindow;l.usedPercentage!=null&&(r.context_window={usedPercentage:l.usedPercentage,remainingPercentage:l.remainingPercentage})}if(!r.rate_limits&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const l=this.providerQuotaToRateLimits(this.cachedProviderQuota);l&&(r.rate_limits=l),u.info(this.name,`[toolbar-meta] provider quota fallback: hasCached=${!!this.cachedProviderQuota} fromProvider=${JSON.stringify(l)}`)}return r.rate_limits&&u.info(this.name,`[toolbar-meta] rate_limits included: ${JSON.stringify(r.rate_limits)}`),r}providerQuotaToCodexRateLimits(e){return ee(e,this.cachedProviderQuotaSampledAtMs??Date.now())}providerQuotaToRateLimits(e){return Z(e,this.cachedProviderQuotaSampledAtMs??Date.now())}async resolveCwdForBinding(e){const o=String(e??"").trim();if(process.platform!=="win32"&&(/^[a-zA-Z]:[\\/]/.test(o)||/^\\\\/.test(o))){const i=new Error(`Specified path is not valid on this host: ${o}`);throw i.cwdErrorCode=g.invalidCwd,i}const t=x.resolve(o);let s;try{s=await I(t)}catch(i){const n=String(i?.code??"");if(n==="ENOENT"){const r=new Error(`Specified path does not exist: ${t}`);throw r.cwdErrorCode=g.invalidCwd,r}if(n==="EACCES"||n==="EPERM"){const r=new Error("Specified path is not accessible.");throw r.cwdErrorCode=g.invalidCwd,r}throw i}if(!s.isDirectory()){const i=new Error("Specified path is not a directory.");throw i.cwdErrorCode=g.invalidCwd,i}try{return await ie(t)}catch{return t}}getClaudeWorkerStatus(e){const o=this.pool.getSlot(e);return o?o.state==="starting"?"starting":o.state==="stopped"?"stopped":o.adapter.getStatus().busy?"busy":"ready":"stopped"}refreshClaudeWorkerStatusCard(e,o){const t=this.getClaudeWorkerStatus(e);return this.claudeWorkerStatus.set(e,t),this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:t,cwd:o,meta:this.buildClaudeToolbarMeta(e)}),t}async handleSkillUploadLocalAction(e){const o=e.params??{},t=String(o.name??"").trim(),s=String(o.session_id??"").trim();if(!t){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_SKILL_NAME",error_msg:"name is required"});return}const i=z(this.config.adapterType,this.config.aibot.clientType),n=this.bindingStore.get(s)?.cwd||this.sessionBindings.get(s)||void 0,r=i==="kiro"?n:n||process.cwd();try{await ge(t,{mode:i,projectDir:r},{apiKey:this.config.aibot.apiKey,wsUrl:this.config.aibot.url}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{name:t}})}catch(a){u.warn(this.name,`skill_upload failed name=${t}: ${a instanceof Error?a.message:String(a)}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"SKILL_UPLOAD_FAILED",error_msg:a instanceof Error?a.message:String(a)})}}reportSessionSkills(e){const o=z(this.config.adapterType,this.config.aibot.clientType),t=this.bindingStore.get(e)?.cwd||this.sessionBindings.get(e)||void 0,s=o==="kiro"?t:t||process.cwd();let i;try{i=he({mode:o,projectDir:s})}catch{return}if(i.length===0)return;const n=JSON.stringify(i.map(r=>`${r.source}:${r.name}`));if(n!==this.lastReportedSkillsHash){this.lastReportedSkillsHash=n;try{this.aibotHandle.sendSkillsUpdate({skills:G(i,j.skills)})}catch{}}}async ensureSlotStarted(e,o=6e4){const t=this.pool.getOrCreateSlot(e);if(!t)throw new Error("Failed to allocate session slot");t.startPromise&&(u.info(this.name,`ensureSlotStarted: awaiting startPromise for session=${e}`),await Promise.race([t.startPromise,new Promise((s,i)=>setTimeout(()=>i(new Error(`ensureSlotStarted timeout (${o}ms) session=${e}`)),o))]),u.info(this.name,`ensureSlotStarted: startPromise resolved for session=${e}`))}resolveOrphanTitle(e){if(!e||!this.reasonixTitleScan)return;const o=$e(e);if(!o)return;const t=this.reasonixTitleScan.get().filter(s=>s.stamp===o);return t.length===1?t[0].title:void 0}resolveAgentSessionId(e){switch(this.config.adapterType??"acp"){case"claude":return e.claudeSessionId;case"codex":return e.codexThreadId;case"pi":return e.piSessionPath;case"codewhale":return e.codewhaleThreadId;case"agy":return e.agyConversationId;default:return e.acpSessionId}}providerKeyForAdapter(){const e=this.config.adapterType??"acp";switch(e){case"claude":case"codex":case"pi":case"codewhale":return e;default:return"acp"}}setResolvedAgentSessionId(e,o){const t=String(e??"").trim(),s=String(o??"").trim();if(!t||!s)return;switch(this.config.adapterType??"acp"){case"claude":this.bindingStore.setClaudeSessionId(t,s);break;case"codex":this.bindingStore.setCodexThreadId(t,s);break;case"pi":this.bindingStore.setPiSessionPath(t,s);break;case"codewhale":this.bindingStore.setCodeWhaleThreadId(t,s);break;case"agy":this.bindingStore.setAgyConversationId(t,s);break;default:this.bindingStore.setAcpSessionId(t,s);break}this.sessionScanCache.invalidate()}normalizePathForCompare(e){const o=String(e??"").trim();if(!o)return"";const t=x.resolve(o);return process.platform==="win32"?t.toLowerCase():t}ensureImportedAgentSession(e,o){const t=String(e??"").trim();if(!t)return;const s=this.normalizePathForCompare(o);let i="";const n=this.config.adapterType??"acp";if(n==="codex"?i=this.sessionScanCache.get().find(d=>d.threadId===t)?.cwd??"":n==="claude"?i=this.sessionScanCache.get().find(d=>d.sessionId===t)?.cwd??"":n==="acp"&&(i=this.sessionScanCache.get().find(d=>d.sessionId===t)?.cwd??""),!i){for(const[,a]of this.bindingStore.entries())if(this.resolveAgentSessionId(a)===t){i=a.cwd??"";break}}if(!i){const a=new Error(`agent session not found: ${t}`);throw a.sessionControlErrorCode=g.invalidAgentSession,a}const r=this.normalizePathForCompare(i);if(r&&s&&r!==s){const a=new Error(`agent session cwd mismatch: expected ${o}, got ${i}`);throw a.sessionControlErrorCode=g.invalidAgentSession,a}}buildOpenedBindingResult(e,o,t="ready"){const s=this.bindingStore.get(e),i=s?String(this.resolveAgentSessionId(s)??"").trim():"",n={aibotSessionId:e,providerKey:this.providerKeyForAdapter(),cwd:o,workerStatus:t};return i&&(n.bindingId=i,n.agentSessionId=i),n}hasDiskScanner(){const e=this.config.adapterType??"acp";return e==="codex"||e==="claude"||e==="acp"}async handleListSessionsTextCommand(e){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()});const o=this.config.adapterType??"acp",t=new Map;for(const a of this.pool.getAllSlots())t.set(a.sessionId,a);const s=Array.from(this.bindingStore.entries()),i=new Map;for(const[a,d]of s){const c=this.resolveAgentSessionId(d);if(c){const l=t.get(a),h=l?l.adapter.getStatus().busy?"busy":"ready":"inactive";i.set(c,{aibotSessionId:a,workerStatus:h})}}const n=[],r=new Set;if(o==="codex"){const a=this.sessionScanCache.get();for(const d of a){r.add(d.threadId);const c=i.get(d.threadId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.threadId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??(d.archived?"archived":"inactive")}`),n.push(` Created: ${new Date(d.createdAt).toISOString()}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="claude"){const a=this.sessionScanCache.get();for(const d of a){r.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="acp"){const a=this.sessionScanCache.get();for(const d of a){r.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="pi"){const a=this.sessionScanCache.get();for(const d of a){r.add(d.sessionPath);const c=i.get(d.sessionPath);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionPath}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="codewhale"||o==="cursor"||o==="opencode"){const a=this.sessionScanCache.get();for(const d of a){r.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}for(const[a,d]of s){const c=this.resolveAgentSessionId(d);if(c&&r.has(c))continue;const l=t.get(a),h=l?l.adapter.getStatus().busy?"busy":"ready":"closed",m=this.resolveOrphanTitle(c)??(c?c.slice(0,8)+"\u2026":a.slice(0,8)+"\u2026");n.push(` Title: ${m}`),n.push(` AIBot: ${a}`),c&&n.push(` Agent: ${c}`),n.push(` CWD: ${d.cwd??"-"}`),n.push(` State: ${h}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}if(n.length===0){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:"No sessions found.",updated_at:Date.now()});return}this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:`Sessions (${n.filter(a=>a==="---").length}):
|
|
13
|
+
Error: ${t}`,1,!1)}sendEventResultWithCleanup(e,o,t,s){this.sendCtrl.sendEventResult(e,o,t,s);const i=this.eventSessionIndex.get(e);i&&(this.pool.eventComplete(e,i),this.pushQueueSnapshotForSession(i),this.conversationLog?.logResult?.(i,e,o,t),this.eventSessionIndex.delete(e)),this.inflightEvents.delete(e),this.restartCount.delete(e),o==="responded"&&(this.cachedProviderQuotaSampledAtMs=null,this.refreshAndPushProviderQuota(!0).catch(()=>{}))}async handleSessionInternalError(e){const{eventId:o,sessionId:t,errorMsg:s}=e;if(this.stopped)return;const i=this.inflightEvents.get(o);if(!i){u.warn(this.name,`[recovery] no inflight event for internalError event=${o} session=${t}; surface failure directly`),this.sendRunErrorAsChunk(o,t,s),this.sendEventResultWithCleanup(o,"failed",s,"agent_stop_failure");return}const n=(this.restartCount.get(o)??0)+1;this.restartCount.set(o,n);const a=this.config.adapterType??"acp";if(n>W){u.error(this.name,`[recovery] adapter=${a} session=${t} event=${o} restart=${n}/${W} outcome=give-up err=${s}`),this.sendRunErrorAsChunk(o,t,s),this.sendEventResultWithCleanup(o,"failed",s,"agent_stop_failure");return}u.info(this.name,`[recovery] adapter=${a} session=${t} event=${o} restart=${n}/${W} outcome=restarting err=${s}`);const r=this.pool.drainQueuedForSession(t);r.length>0&&u.info(this.name,`[recovery] session=${t} preserved ${r.length} queued sibling event(s) across restart`);try{await this.pool.removeSlot(t)}catch(l){u.warn(this.name,`[recovery] removeSlot failed session=${t}: ${l instanceof Error?l.message:String(l)}`)}if(this.stopped)return;const d=this.resolveRecoveryPrompt(a,i),c={...i,content:d};try{await this.pool.deliverInboundEvent(c)}catch(l){u.error(this.name,`[recovery] redeliver failed event=${o} session=${t}: ${l instanceof Error?l.message:String(l)}`),this.sendEventResultWithCleanup(o,"failed",l instanceof Error?l.message:String(l));return}for(const l of r){if(this.stopped)break;try{await this.pool.deliverInboundEvent(l)}catch(h){u.error(this.name,`[recovery] sibling redeliver failed event=${l.event_id} session=${t}: ${h instanceof Error?h.message:String(h)}`),this.sendEventResultWithCleanup(l.event_id,"failed",h instanceof Error?h.message:String(h))}}}resolveRecoveryPrompt(e,o){return e==="acp"?"continue":o.content}sendThinkingByRuntimeConfig(e,o,t){this.sendCtrl.sendThinking(e,o,t)}bufferStreamChunk(e,o,t,s,i){this.sendCtrl.bufferOnly(e,o,t,s,i)}flushBufferedStreamText(e){}resolveEventRuntimeConfig(e){return this.sendCtrl.resolveEventRuntimeConfig(e)}captureEventRuntimeConfig(e){this.sendCtrl.captureEventRuntimeConfig(e),this.indexEventSession(e.event_id,e.session_id),e.event_id&&!this.inflightEvents.has(e.event_id)&&this.inflightEvents.set(e.event_id,e)}indexEventSession(e,o){!e||!o||this.eventSessionIndex.set(e,o)}shouldDropToolDisplayEvent(e){return this.sendCtrl.shouldDropToolDisplayEvent(e)}shouldDropThinkingDisplayEvent(e){return this.sendCtrl.shouldDropThinkingDisplayEvent(e)}shouldDropCodexDisplayEvent(e,o){return this.sendCtrl.shouldDropCodexDisplayEvent(e,o)}logCodexEventToConversation(e){if(!this.conversationLog||e.codex_method!=="item/agentMessage/delta")return;const t=e.codex_payload?.params?.delta;if(!t)return;const s=this.eventSessionIndex.get(e.event_id)??e.session_id;this.conversationLog.append(s,{ts:Date.now(),dir:"outbound",event_id:e.event_id,kind:"codex_delta",text_len:t.length,content:t})}isAcpRawTransportEnabled(){return(this.config.adapterOptions??{}).raw_transport===!0}shouldDropAcpRawDisplayEvent(e,o){return this.sendCtrl.shouldDropAcpRawDisplayEvent(e,o)}sendAcpRawEventEnvelope(e,o,t){this.shouldDropAcpRawDisplayEvent(e,t.type)||this.deliverRawEventEnvelope(e,o,t,"acp",this.buildAcpRawEventFallbackText(t))}buildAcpRawEventFallbackText(e){const o=String(e.type??"").trim();if(!o)return"[acp] event";switch(o){case"permission_request":return`Permission required: ${String(e.payload?.tool_title??e.payload?.tool_name??"permission request")}`;case"tool_use":return`[tool] ${String(e.payload?.tool_name??"tool")}`;case"tool_result":return"[tool result]";case"thinking":return"[thinking]";case"error":return`[error] ${String(e.payload?.message??"agent error")}`;case"result":return"[result]";default:return`[acp] ${o}`}}rawDetailSeq=0;deliverRawEventEnvelope(e,o,t,s,i){const n=pt({envelope:t,fallbackText:i,channelKey:s,allocateRefId:()=>`${e}_rawd_${++this.rawDetailSeq}`}),a=()=>{this.aibotHandle.sendMsg({event_id:e,session_id:o,msg_type:1,content:n.fallbackText,extra:{channel_data:{[s]:{raw_event:n.envelope}},agent_api_origin:!0}})};if(!n.sharded){a();return}u.info("bridge",`${s} raw_event oversized, sharded delivery: event=${e} fields=${n.oversizedFields.map(r=>r.field).join(",")}`),(async()=>{for(const r of n.oversizedFields)await this.sendCtrl.deliverAuxiliaryLargeText(e,o,ft({envelopeType:t.type,field:r.field,fullText:r.fullText}),r.refClientMsgId);a()})().catch(r=>{u.warn("bridge",`${s} raw_event sharded delivery failed event=${e}: ${r}`)})}buildCursorRawEventFallbackText(e){const o=String(e?.type??"").trim(),t=e?.payload&&typeof e.payload=="object"?e.payload:{};switch(o){case"permission_request":return`Permission required: ${String(t.tool_title??t.tool_name??"permission request")}`;case"tool_use":case"tool_call":case"tool_execution_start":return`[tool] ${String(t.tool_name??t.toolName??"tool")}`;case"tool_result":case"tool_execution_end":case"tool_execution_update":return"[tool result]";case"error":return`[error] ${String(t.message??"agent error")}`;default:return o?`[cursor] ${o}`:"[cursor] event"}}sendToolExecutionCard(e,o,t,s){this.sendCtrl.sendToolExecutionCard(e,o,t,s)}async handleAibotEvent(e){if(this.relayEnvStale&&!this.hasPendingWork()&&await this.recycleAdaptersForRelayChange(),this.stopped){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:"agent shutting down",updated_at:Date.now()});return}this.logInboundConversation(e);const o=this.config.adapterType??"acp",t=ut(e.extra);if(t.error){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:"connector_config_invalid",msg:t.error,updated_at:Date.now()});return}const s=t.patch,i=Xe(e);if(i){if(i.verb===p.exec){await this.handleSessionControlCommand(i,e);return}if(i.verb===p.listSessions){await this.handleListSessionsTextCommand(e);return}if(o==="claude"){await this.handleSessionControlCommand(i,e);return}if(o==="codex"&&i.verb===p.open){await this.handleCodexSessionControlOpen(i,e);return}if(o==="pi"&&i.verb===p.open){await this.handlePiSessionControlOpen(i,e);return}if(o==="pi"&&i.verb===p.restart){await this.handlePiSessionControlRestart(e);return}if((o==="openhuman"||o==="opencode")&&i.verb===p.open){await this.handleOpenHumanSessionControlOpen(i,e);return}if(o==="codewhale"&&i.verb===p.open){await this.handleCodeWhaleSessionControlOpen(i,e);return}if(this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),i.verb===p.open){const r=i.args.trim();if(!r){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.cwdRequired,msg:"cwd is required",updated_at:Date.now()});return}try{const d=x.resolve(r);if(!(await D(d)).isDirectory()){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.invalidCwd,msg:`Path is not a directory: ${d}`,updated_at:Date.now()});return}}catch(d){const c=String(d?.code??""),l=c==="ENOENT"?`Directory does not exist: ${x.resolve(r)}`:c==="EACCES"||c==="EPERM"?"Directory is not accessible":`Invalid path: ${r}`;this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:g.invalidCwd,msg:l,updated_at:Date.now()});return}}if(i.verb===p.open){const r=this.bindingStore.get(e.session_id);if(r?.cwd)try{await D(r.cwd)}catch{u.info("bridge",`Stale binding detected for session ${e.session_id}: ${r.cwd} no longer exists, clearing`),this.bindingStore.delete(e.session_id),this.sessionBindings.delete(e.session_id),this.sessionProviderHints.delete(e.session_id),this.sessionProviderQuotas.delete(e.session_id),this.sessionProviderMeta.delete(e.session_id);const d=this.pool.getSlot(e.session_id);d?.adapter instanceof A&&d.adapter.getSessionBindings().delete(e.session_id)}}if(await this.handleSessionControlForPool(i,e),o==="acp"&&i.verb===p.stop){const d=this.bindingStore.get(e.session_id)?.cwd??"";await this.pool.removeSlot(e.session_id).catch(()=>{}),this.sessionBindings.delete(e.session_id),this.sessionProviderHints.delete(e.session_id),this.sessionProviderQuotas.delete(e.session_id),this.sessionProviderMeta.delete(e.session_id),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:`Session worker stopped for ${d}`,updated_at:Date.now()});return}if(V(i,e,this.sessionControlCtx(e.session_id),{...this.sessionControlSenders(),sendEventAck:()=>{}}),i.verb===p.open&&(await this.deferredMgr.release(e.session_id,this.deferredCallbacks()),o==="agy")){const r=this.bindingStore.get(e.session_id)?.cwd??"";r&&this.aibotHandle.sendUpdateBindingCard({session_id:e.session_id,worker_status:"ready",cwd:r,meta:this.buildAgyToolbarMeta(e.session_id)})}return}if(e.mirror_mode==="record_only"){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",updated_at:Date.now()});return}if(this.isStaleEvent(e)){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",code:"event_stale",msg:"event is stale and will not be processed",updated_at:Date.now()});return}if(kt.has(o)&&!this.bindingStore.get(e.session_id)?.cwd){const d=o;u.info(this.name,`[${d}] binding missing session_id=${e.session_id} event_id=${e.event_id}`),this.deferredMgr.defer(d,String(e.session_id??"").trim(),this.buildInboundEvent(e,s));const c=this.resolveBindingChannelKey(o);this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.aibotHandle.sendMsg({event_id:e.event_id,session_id:e.session_id,msg_type:1,content:"Session binding missing.",extra:{channel_data:{[c]:{sessionBinding:{status:"missing",reason:"binding_missing",error_code:g.bindingMissing}}}},quoted_message_id:e.msg_id});return}if((this.config.adapterType??"acp")==="acp"){const d=String(e.content??"").trim().match(/^\/(\S+)\s*(.*)/);if(d){const[,c,l]=d,m=this.pool.getSlot(e.session_id)?.adapter;if(m?.execCommand&&(m.getSupportedCommands?.()??[]).some(f=>f.name===c||f.name===`/${c}`)){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()});try{const f=await m.execCommand(c,l.trim(),e.session_id);f.status==="options"&&f.data&&this.handleExecCommandOptions(e.session_id,c,f.data),this.aibotHandle.sendEventResult({event_id:e.event_id,status:f.status==="failed"?"failed":"responded",msg:f.message,updated_at:Date.now()})}catch(f){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:f instanceof Error?f.message:String(f),updated_at:Date.now()})}return}}}const a=this.buildInboundEvent(e,s);try{this.captureEventRuntimeConfig(a),await this.pool.deliverInboundEvent(a)}catch(r){if(this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()}),this.discardEventTrackingState(e.event_id),r instanceof Ye)this.aibotHandle.sendEventResult({event_id:e.event_id,status:"failed",msg:r.message,updated_at:Date.now()});else throw r}}async handleCodexSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(a,r)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:a,...r,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const a=await this.resolveCwdForBinding(s),r=this.bindingStore.get(t);if(r?.cwd){let d=!1;try{if(await this.resolveCwdForBinding(r.cwd)===a){i(),n("responded",{msg:`Session already bound to ${r.cwd}`});return}}catch{d=!0,u.info("bridge",`Stale codex binding for session ${t}: ${r.cwd} no longer exists, allowing rebind`)}if(!d){i(),n("failed",{msg:`Session already bound to ${r.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}}this.bindingStore.set(t,a),this.sessionBindings.set(t,a),this.deferredMgr.sendCodexDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,a),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:a}),i(),n("responded",{msg:`Session bound to ${a}`})}catch(a){i(),n("failed",{code:g.invalidCwd,msg:a instanceof Error?a.message:String(a)})}}async handleSessionControlForPool(e,o){e.verb===p.open&&await this.bindSessionForPool(o.session_id,e.args.trim())}async replayDeferredEventsForSession(e){await this.deferredMgr.release(e,this.deferredCallbacks())}async handleSessionControlLocalActionForPool(e){if(String(e.action_type??"")!==b.sessionControl)return;const o=e.params??{};if(String(o.verb??"").trim().toLowerCase()!==p.open)return;const s=String(o.session_id??"").trim(),i=String(o.cwd??"").trim();!s||!i||await this.bindSessionForPool(s,i)}async handleCodexSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(a,r,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:a,...r!==void 0?{result:r}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const a=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,a);const r=this.bindingStore.get(t);if(r?.cwd){let d=!1;try{const c=await this.resolveCwdForBinding(r.cwd);if(c===a){this.setResolvedAgentSessionId(t,i),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,c)});return}}catch{d=!0,u.info("bridge",`Stale codex binding for session ${t}: ${r.cwd} no longer exists, allowing rebind`)}if(!d){n("failed",void 0,g.rebindForbidden,`Session already bound to ${r.cwd}`);return}}this.bindingStore.set(t,a),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,a),this.deferredMgr.sendCodexDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,a),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,a)})}catch(a){n("failed",void 0,a?.sessionControlErrorCode??g.invalidCwd,a instanceof Error?a.message:String(a))}}async handleCursorSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(a,r,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:a,...r!==void 0?{result:r}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const a=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,a);const r=this.bindingStore.get(t);if(r?.cwd){let d=!1;try{const c=await this.resolveCwdForBinding(r.cwd);if(c===a){this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,c),await this.bindSessionForPool(t,c),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:c,meta:this.buildCursorToolbarMeta(t)}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,c)});return}}catch{d=!0,u.info("bridge",`Stale cursor binding for session ${t}: ${r.cwd} no longer exists, allowing rebind`)}if(!d){n("failed",void 0,g.rebindForbidden,`Session already bound to ${r.cwd}`);return}}this.bindingStore.set(t,a),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,a),this.deferredMgr.sendCursorDeferredReplayComposing(t,this.deferredCallbacks()),await this.bindSessionForPool(t,a),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:a,meta:this.buildCursorToolbarMeta(t)}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,a)})}catch(a){n("failed",void 0,a?.sessionControlErrorCode??g.invalidCwd,a instanceof Error?a.message:String(a))}}async handlePiSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(a,r)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:a,...r,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const a=await this.resolveCwdForBinding(s),r=this.bindingStore.get(t);if(r?.cwd){let d=!1;try{if(await this.resolveCwdForBinding(r.cwd)===a){i(),n("responded",{msg:`Session already bound to ${r.cwd}`}),this.aibotHandle.sendMsg({event_id:o.event_id,session_id:t,msg_type:1,content:`\u2705 Session already bound to \`${r.cwd}\``,quoted_message_id:o.msg_id});return}}catch{d=!0,u.info("bridge",`Stale pi binding for session ${t}: ${r.cwd} no longer exists, allowing rebind`)}if(!d){i(),n("failed",{msg:`Session already bound to ${r.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}}this.bindingStore.set(t,a),this.sessionBindings.set(t,a),await this.ensureSlotStarted(t).catch(d=>{u.warn("bridge",`pi ensureSlotStarted on bind failed (non-fatal): ${d instanceof Error?d.message:String(d)}`)}),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:a}),i(),n("responded",{msg:`Session bound to ${a}`}),this.aibotHandle.sendMsg({event_id:o.event_id,session_id:t,msg_type:1,content:`\u2705 Working directory bound: \`${a}\``,quoted_message_id:o.msg_id})}catch(a){i(),n("failed",{code:g.invalidCwd,msg:a instanceof Error?a.message:String(a)})}}async handlePiSessionControlRestart(e){const o=e.session_id,t=()=>this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:o,received_at:Date.now()}),s=(a,r)=>this.aibotHandle.sendEventResult({event_id:e.event_id,status:a,...r,updated_at:Date.now()}),n=this.bindingStore.get(o)?.cwd??"";if(!n){t(),s("failed",{msg:"session binding was not found",code:g.bindingMissing});return}t(),await this.pool.removeSlot(o).catch(()=>{}),this.aibotHandle.sendUpdateBindingCard({session_id:o,worker_status:"ready",cwd:n}),s("responded",{msg:`Session worker restarted for ${n}`})}async handlePiSessionControlRestartLocalAction(e){const o=e.params??{},t=String(o.session_id??"").trim(),i=(t?this.bindingStore.get(t):void 0)?.cwd??"",n=(a,r,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:a,...r!==void 0?{result:r}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t){n("failed",void 0,"session_id_required","session_id is required for restart");return}if(!i){n("failed",void 0,g.bindingMissing,"Session binding missing. Open a workspace first.");return}await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:i}),n("ok",{outcome:"restarted",binding:{aibotSessionId:t,cwd:i,workerStatus:"ready"}})}async bindSessionForPool(e,o){const t=String(o??"").trim();if(!e||!t)return;const s=this.pool.getOrCreateSlot(e);if(!s||(s.startPromise&&await s.startPromise,!s.adapter))return;const i=s.adapter instanceof A?s.adapter:null;if(!i?.hasSessionBinding)return;const n=await this.resolveCwdForBinding(t);i.announceDeferredComposing(e),await i.bindSession(e,n),this.sessionBindings.set(e,n),this.sessionScanCache.invalidate(),i.replayDeferredEvents(e)}deferredCallbacks(){return{captureEventRuntimeConfig:e=>this.captureEventRuntimeConfig(e),deliverInboundEvent:e=>this.pool.deliverInboundEvent(e),sendEventResult:(e,o,t)=>this.sendEventResultWithCleanup(e,o,t),sendSessionComposing:(e,o,t)=>{const s={};o&&(s.ttl_ms=t?.ttlMs??3e4,t?.activity&&(s.activity=t.activity)),this.aibotHandle.sendSessionActivitySet({session_id:e,kind:"composing",active:o,...s})}}}async handleOpenHumanSessionControlOpen(e,o){const t=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:o.session_id,received_at:Date.now()});try{await this.resolveCwdForBinding(e.args.trim()),await this.handleSessionControlForPool(e,o),V(e,o,this.sessionControlCtx(o.session_id),this.sessionControlSenders()),await this.deferredMgr.release(o.session_id,this.deferredCallbacks())}catch(s){t(),this.aibotHandle.sendEventResult({event_id:o.event_id,status:"failed",code:s?.cwdErrorCode,msg:s instanceof Error?s.message:String(s),updated_at:Date.now()})}}async handleCodeWhaleSessionControlOpen(e,o){const t=o.session_id,s=e.args.trim(),i=()=>this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()}),n=(a,r)=>this.aibotHandle.sendEventResult({event_id:o.event_id,status:a,...r,updated_at:Date.now()});if(!s){i(),n("failed",{msg:"Usage: /grix open <working-directory>",code:g.cwdRequired});return}try{const a=await this.resolveCwdForBinding(s),r=this.bindingStore.get(t);if(r?.cwd){if(await this.resolveCwdForBinding(r.cwd)!==a){i(),n("failed",{msg:`Session already bound to ${r.cwd}. Rebinding is not allowed.`,code:g.rebindForbidden});return}i(),n("responded",{msg:`Session already bound to ${r.cwd}`});return}this.bindingStore.set(t,a),this.sessionBindings.set(t,a),await this.bindSessionForPool(t,a),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:a}),i(),n("responded",{msg:`Session bound to ${a}`})}catch(a){i(),n("failed",{code:g.invalidCwd,msg:a instanceof Error?a.message:String(a)})}}async handleCodeWhaleSessionControlLocalActionOpen(e){const o=e.params??{},t=String(o.session_id??"").trim(),s=String(o.cwd??"").trim(),i=String(o.agent_session_id??"").trim(),n=(a,r,d,c)=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:a,...r!==void 0?{result:r}:{},...d?{error_code:d}:{},...c?{error_msg:c}:{}})};if(!t||!s){n("failed",void 0,g.cwdRequired,"session cwd is required");return}try{const a=await this.resolveCwdForBinding(s);this.ensureImportedAgentSession(i,a);const r=this.bindingStore.get(t);if(r?.cwd){const d=await this.resolveCwdForBinding(r.cwd);if(d!==a){n("failed",void 0,g.rebindForbidden,`Session already bound to ${r.cwd}`);return}this.setResolvedAgentSessionId(t,i),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,d)});return}this.bindingStore.set(t,a),this.setResolvedAgentSessionId(t,i),this.sessionBindings.set(t,a),await this.bindSessionForPool(t,a),await this.deferredMgr.release(t,this.deferredCallbacks(),{announceComposing:!1}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:a}),n("ok",{outcome:"opened",binding:this.buildOpenedBindingResult(t,a)})}catch(a){n("failed",void 0,a?.sessionControlErrorCode??g.invalidCwd,a instanceof Error?a.message:String(a))}}normalizeClaudeModeId(e){return String(e??"").trim().toLowerCase()===C.approval?C.approval:C.fullAuto}handleExecCommandOptions(e,o,t){const s=Array.isArray(t?.options)?t.options:[];if(s.length===0)return;const n=this.bindingStore.get(e)?.cwd??"",a={};if(o==="model"){a.available_models=s.map(d=>({id:d.id,displayName:d.label}));const r=s.find(d=>d.current);r&&(a.model_id=r.id)}else if(o==="mode"){a.available_modes=s.map(d=>({id:d.id,displayName:d.label}));const r=s.find(d=>d.current);r&&(a.mode_id=r.id)}else a[`${o}_options`]=s;this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:"ready",cwd:n,meta:a})}resolveSessionModelId(e){const o=this.bindingStore.getModelId(e);if(o)return o;const t=String(this.globalConfigStore?.get(this.name)?.modelId??"").trim();if(t)return this.bindingStore.setModelId(e,t),t}resolveCodexSessionModelId(e){const o=this.bindingStore.getCodexModelId(e);if(o)return o;const t=String(this.globalConfigStore?.get(this.name)?.codexModelId??"").trim();if(t)return this.bindingStore.setCodexModelId(e,t),t}buildCursorToolbarMeta(e){const o=this.pool.getSlot(e);if(o?.adapter instanceof M)return o.adapter.getToolbarMeta(e);const t=this.bindingStore.getModelId(e)??"auto",s=this.bindingStore.getCursorModeId(e)??"full_auto";return{model_id:t,mode_id:s,currentModelId:t,currentModeId:s,available_models:[],available_modes:de.map(i=>({...i}))}}buildAgyToolbarMeta(e){if(!this.bindingStore.get(e))return;const t=ue(this.config.agent.command),s=t.length>0?t[0].id:"";let i=this.resolveSessionModelId(e)||s;t.length>0&&!t.some(a=>a.id===i)&&(i=s);const n=he();return{model_id:i,currentModelId:i,available_models:t.map(a=>({id:a.id,displayName:a.displayName})),...n.plan!==void 0&&{plan:n.plan},...n.quota_exhausted!==void 0&&{quota_exhausted:n.quota_exhausted},...n.quota_reset_at!==void 0&&{quota_reset_at:n.quota_reset_at},...n.available_credits!==void 0&&{available_credits:n.available_credits}}}async handleAgySetModel(e,o){const t=e.params??{},s=String(t.model_id??t.modelId??"").trim();if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for set_model"});return}if(!s){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"set_model_invalid",error_msg:"model_id is required"});return}const i=this.bindingStore.get(o);if(!i?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"session binding was not found"});return}i.modelId!==s&&(this.bindingStore.setModelId(o,s),this.globalConfigStore?.set(this.name,{modelId:s})),this.aibotHandle.sendUpdateBindingCard({session_id:o,worker_status:"ready",cwd:i.cwd,meta:this.buildAgyToolbarMeta(o)}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"model_set",model_id:s,binding:{cwd:i.cwd,model_id:s}}})}buildClaudeToolbarMeta(e){const o=this.bindingStore.get(e);if(!o)return;const t=F(),s=t.length>0?t[0].id:"";let i=this.resolveSessionModelId(e)||s;t.length>0&&!t.some(l=>l.id===i)&&(i=s);const n=this.normalizeClaudeModeId(o.modeId),a={model_id:i,mode_id:n,currentModelId:i,currentModeId:n,available_models:t.map(l=>({id:l.id,displayName:l.displayName}))},r=this.pool.getSlot(e),d=r?.adapter instanceof P?r.adapter.getSessionState():null;(d?.rateLimits?.fiveHour||d?.rateLimits?.sevenDay)&&(this.cachedClaudeRateLimitState=d);const c=d??this.getFreshClaudeRateLimitState();if(c){const l=c.rateLimits;l&&(l.fiveHour||l.sevenDay)&&(a.rate_limits={...l.fiveHour?{fiveHour:l.fiveHour}:{},...l.sevenDay?{sevenDay:l.sevenDay}:{},sampledAt:c.sampledAt||Date.now()})}if(d){const l=d.contextWindow;l.usedPercentage!=null&&(a.context_window={usedPercentage:l.usedPercentage,remainingPercentage:l.remainingPercentage})}if(!a.rate_limits&&this.cachedProviderQuota?.success){this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)||this.maybeQueryProviderQuota().catch(()=>{});const l=this.providerQuotaToRateLimits(this.cachedProviderQuota);l&&(a.rate_limits=l),u.info(this.name,`[toolbar-meta] provider quota fallback: hasCached=${!!this.cachedProviderQuota} fromProvider=${JSON.stringify(l)}`)}return a.rate_limits&&u.info(this.name,`[toolbar-meta] rate_limits included: ${JSON.stringify(a.rate_limits)}`),a}providerQuotaToCodexRateLimits(e){return ee(e,this.cachedProviderQuotaSampledAtMs??Date.now())}providerQuotaToRateLimits(e){return Z(e,this.cachedProviderQuotaSampledAtMs??Date.now())}async resolveCwdForBinding(e){const o=String(e??"").trim();if(process.platform!=="win32"&&(/^[a-zA-Z]:[\\/]/.test(o)||/^\\\\/.test(o))){const i=new Error(`Specified path is not valid on this host: ${o}`);throw i.cwdErrorCode=g.invalidCwd,i}const t=x.resolve(o);let s;try{s=await D(t)}catch(i){const n=String(i?.code??"");if(n==="ENOENT"){const a=new Error(`Specified path does not exist: ${t}`);throw a.cwdErrorCode=g.invalidCwd,a}if(n==="EACCES"||n==="EPERM"){const a=new Error("Specified path is not accessible.");throw a.cwdErrorCode=g.invalidCwd,a}throw i}if(!s.isDirectory()){const i=new Error("Specified path is not a directory.");throw i.cwdErrorCode=g.invalidCwd,i}try{return await ie(t)}catch{return t}}getClaudeWorkerStatus(e){const o=this.pool.getSlot(e);return o?o.state==="starting"?"starting":o.state==="stopped"?"stopped":o.adapter.getStatus().busy?"busy":"ready":"stopped"}refreshClaudeWorkerStatusCard(e,o){const t=this.getClaudeWorkerStatus(e);return this.claudeWorkerStatus.set(e,t),this.aibotHandle.sendUpdateBindingCard({session_id:e,worker_status:t,cwd:o,meta:this.buildClaudeToolbarMeta(e)}),t}async handleSkillUploadLocalAction(e){const o=e.params??{},t=String(o.name??"").trim(),s=String(o.session_id??"").trim();if(!t){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_SKILL_NAME",error_msg:"name is required"});return}const i=z(this.config.adapterType,this.config.aibot.clientType),n=this.bindingStore.get(s)?.cwd||this.sessionBindings.get(s)||void 0,a=i==="kiro"?n:n||process.cwd();try{await pe(t,{mode:i,projectDir:a},{apiKey:this.config.aibot.apiKey,wsUrl:this.config.aibot.url}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{name:t}})}catch(r){u.warn(this.name,`skill_upload failed name=${t}: ${r instanceof Error?r.message:String(r)}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"SKILL_UPLOAD_FAILED",error_msg:r instanceof Error?r.message:String(r)})}}reportSessionSkills(e){const o=z(this.config.adapterType,this.config.aibot.clientType),t=this.bindingStore.get(e)?.cwd||this.sessionBindings.get(e)||void 0,s=o==="kiro"?t:t||process.cwd();let i;try{i=me({mode:o,projectDir:s})}catch{return}if(i.length===0)return;const n=JSON.stringify(i.map(a=>`${a.source}:${a.name}`));if(n!==this.lastReportedSkillsHash){this.lastReportedSkillsHash=n;try{this.aibotHandle.sendSkillsUpdate({skills:G(i,j.skills)})}catch{}}}async ensureSlotStarted(e,o=6e4){const t=this.pool.getOrCreateSlot(e);if(!t)throw new Error("Failed to allocate session slot");t.startPromise&&(u.info(this.name,`ensureSlotStarted: awaiting startPromise for session=${e}`),await Promise.race([t.startPromise,new Promise((s,i)=>setTimeout(()=>i(new Error(`ensureSlotStarted timeout (${o}ms) session=${e}`)),o))]),u.info(this.name,`ensureSlotStarted: startPromise resolved for session=${e}`))}resolveOrphanTitle(e){if(!e||!this.reasonixTitleScan)return;const o=Pe(e);if(!o)return;const t=this.reasonixTitleScan.get().filter(s=>s.stamp===o);return t.length===1?t[0].title:void 0}resolveAgentSessionId(e){switch(this.config.adapterType??"acp"){case"claude":return e.claudeSessionId;case"codex":return e.codexThreadId;case"pi":return e.piSessionPath;case"codewhale":return e.codewhaleThreadId;case"agy":return e.agyConversationId;default:return e.acpSessionId}}providerKeyForAdapter(){const e=this.config.adapterType??"acp";switch(e){case"claude":case"codex":case"pi":case"codewhale":return e;default:return"acp"}}setResolvedAgentSessionId(e,o){const t=String(e??"").trim(),s=String(o??"").trim();if(!t||!s)return;switch(this.config.adapterType??"acp"){case"claude":this.bindingStore.setClaudeSessionId(t,s);break;case"codex":this.bindingStore.setCodexThreadId(t,s);break;case"pi":this.bindingStore.setPiSessionPath(t,s);break;case"codewhale":this.bindingStore.setCodeWhaleThreadId(t,s);break;case"agy":this.bindingStore.setAgyConversationId(t,s);break;default:this.bindingStore.setAcpSessionId(t,s);break}this.sessionScanCache.invalidate()}normalizePathForCompare(e){const o=String(e??"").trim();if(!o)return"";const t=x.resolve(o);return process.platform==="win32"?t.toLowerCase():t}ensureImportedAgentSession(e,o){const t=String(e??"").trim();if(!t)return;const s=this.normalizePathForCompare(o);let i="";const n=this.config.adapterType??"acp";if(n==="codex"?i=this.sessionScanCache.get().find(d=>d.threadId===t)?.cwd??"":n==="claude"?i=this.sessionScanCache.get().find(d=>d.sessionId===t)?.cwd??"":n==="acp"&&(i=this.sessionScanCache.get().find(d=>d.sessionId===t)?.cwd??""),!i){for(const[,r]of this.bindingStore.entries())if(this.resolveAgentSessionId(r)===t){i=r.cwd??"";break}}if(!i){const r=new Error(`agent session not found: ${t}`);throw r.sessionControlErrorCode=g.invalidAgentSession,r}const a=this.normalizePathForCompare(i);if(a&&s&&a!==s){const r=new Error(`agent session cwd mismatch: expected ${o}, got ${i}`);throw r.sessionControlErrorCode=g.invalidAgentSession,r}}buildOpenedBindingResult(e,o,t="ready"){const s=this.bindingStore.get(e),i=s?String(this.resolveAgentSessionId(s)??"").trim():"",n={aibotSessionId:e,providerKey:this.providerKeyForAdapter(),cwd:o,workerStatus:t};return i&&(n.bindingId=i,n.agentSessionId=i),n}hasDiskScanner(){const e=this.config.adapterType??"acp";return e==="codex"||e==="claude"||e==="acp"}async handleListSessionsTextCommand(e){this.aibotHandle.sendEventAck({event_id:e.event_id,session_id:e.session_id,received_at:Date.now()});const o=this.config.adapterType??"acp",t=new Map;for(const r of this.pool.getAllSlots())t.set(r.sessionId,r);const s=Array.from(this.bindingStore.entries()),i=new Map;for(const[r,d]of s){const c=this.resolveAgentSessionId(d);if(c){const l=t.get(r),h=l?l.adapter.getStatus().busy?"busy":"ready":"inactive";i.set(c,{aibotSessionId:r,workerStatus:h})}}const n=[],a=new Set;if(o==="codex"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.threadId);const c=i.get(d.threadId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.threadId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??(d.archived?"archived":"inactive")}`),n.push(` Created: ${new Date(d.createdAt).toISOString()}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="claude"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="acp"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="pi"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionPath);const c=i.get(d.sessionPath);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionPath}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}else if(o==="codewhale"||o==="cursor"||o==="opencode"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionId);const c=i.get(d.sessionId);d.title&&n.push(` Title: ${d.title}`),n.push(` Agent: ${d.sessionId}`),c&&n.push(` AIBot: ${c.aibotSessionId}`),n.push(` CWD: ${d.cwd||"-"}`),n.push(` State: ${c?.workerStatus??"inactive"}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}}for(const[r,d]of s){const c=this.resolveAgentSessionId(d);if(c&&a.has(c))continue;const l=t.get(r),h=l?l.adapter.getStatus().busy?"busy":"ready":"closed",m=this.resolveOrphanTitle(c)??(c?c.slice(0,8)+"\u2026":r.slice(0,8)+"\u2026");n.push(` Title: ${m}`),n.push(` AIBot: ${r}`),c&&n.push(` Agent: ${c}`),n.push(` CWD: ${d.cwd??"-"}`),n.push(` State: ${h}`),n.push(` Updated: ${new Date(d.updatedAt).toISOString()}`),n.push("---")}if(n.length===0){this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:"No sessions found.",updated_at:Date.now()});return}this.aibotHandle.sendEventResult({event_id:e.event_id,status:"responded",msg:`Sessions (${n.filter(r=>r==="---").length}):
|
|
14
14
|
${n.join(`
|
|
15
|
-
`)}`,updated_at:Date.now()})}async handleListSessionsLocalAction(e){const o=this.config.adapterType??"acp",t=new Map;for(const
|
|
16
|
-
Models (current: ${n?.modelId??"default"}): ${r}`);return}case p.exec:{const[i,...n]=e.args.trim().split(/\s+/);if(!i){s("failed","Usage: /grix exec <command> [args]",g.verbInvalid);return}const a=this.pool.getSlot(t)?.adapter;if(!a?.execCommand){s("failed","Agent does not support command execution",g.verbInvalid);return}const d=a.getSupportedCommands?.()??[];if(!d.some(c=>c.name===i)){s("failed",`Unknown command: ${i}. Supported: ${d.map(c=>c.name).join(", ")}`,g.verbInvalid);return}try{const c=await a.execCommand(i,n.join(" "),t);s(c.status==="ok"?"responded":"failed",c.message??`${i} ${c.status}`,c.status==="ok"?void 0:g.runtimeError)}catch(c){s("failed",`exec error: ${c instanceof Error?c.message:c}`,g.runtimeError)}return}default:s("failed",`Unsupported command for Claude: /grix ${e.verb}`,g.verbInvalid)}}catch(i){s("failed",i instanceof Error?i.message:String(i),g.runtimeError)}}async handleSessionControlLocalAction(e){const o=String(e.action_type??"").trim();if(o!==b.sessionControl&&o!==b.setMode&&o!=="set_mode"&&o!==b.setModel&&o!=="set_model")return!1;const t=e.params??{},s=String(t.session_id??"").trim(),i=o===b.setMode?p.setMode:o===b.setModel?p.setModel:String(t.verb??"").trim().toLowerCase();if(u.info(this.name,`handleSessionControlLocalAction verb=${i} action_id=${e.action_id} session_id=${s}`),!s)return this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:T.localActionRouteMissing,error_msg:"local action session_id is required"}),!0;const n=a=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:a})},r=(a,d)=>{u.warn(this.name,`session_control local_action failed action_id=${e.action_id} session_id=${s} verb=${i} code=${a} msg=${d}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:a,error_msg:d})};try{switch(i){case p.open:{await B().catch(()=>{});const a=String(t.cwd??"").trim(),d=String(t.agent_session_id??"").trim();if(!a)return r(g.cwdRequired,"session control cwd is required"),!0;u.info(this.name,`handleSessionControlLocalAction open cwd=${a} session_id=${s}`);const c=await this.resolveCwdForBinding(a);this.ensureImportedAgentSession(d,c);const l=this.bindingStore.get(s);if(l?.cwd){const h=await this.resolveCwdForBinding(l.cwd);return h!==c?(r(g.rebindForbidden,"session binding cannot be changed to another working directory"),!0):(this.bindingStore.ensureModeId(s,C.fullAuto),this.setResolvedAgentSessionId(s,d),this.sessionBindings.set(s,h),await this.ensureSlotStarted(s),await this.replayDeferredEventsForSession(s),this.refreshClaudeWorkerStatusCard(s,h),n({outcome:"opened",binding:{...this.buildOpenedBindingResult(s,h),mode_id:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId)}}),!0)}return this.bindingStore.set(s,c,{modeId:C.fullAuto}),this.setResolvedAgentSessionId(s,d),this.sessionBindings.set(s,c),await this.ensureSlotStarted(s),await this.replayDeferredEventsForSession(s),this.refreshClaudeWorkerStatusCard(s,c),n({outcome:"opened",binding:{...this.buildOpenedBindingResult(s,c),mode_id:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId)}}),!0}case p.status:case p.where:{const a=this.bindingStore.get(s);return a?.cwd?(n({outcome:i,binding:{cwd:a.cwd,mode_id:this.normalizeClaudeModeId(a.modeId),worker_status:this.getClaudeWorkerStatus(s)}}),!0):(r(g.bindingMissing,"session binding was not found"),!0)}case p.stop:{const a=this.bindingStore.get(s);return a?.cwd?(await this.pool.removeSlot(s),n({outcome:"stopped",binding:{cwd:a.cwd,mode_id:this.normalizeClaudeModeId(a.modeId),worker_status:"stopped"}}),!0):(r(g.bindingMissing,"session binding was not found"),!0)}case p.restart:{const a=this.bindingStore.get(s);return a?.cwd?(await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,a.cwd),n({outcome:"restarted",binding:{cwd:a.cwd,mode_id:this.normalizeClaudeModeId(a.modeId)}}),!0):(r(g.bindingMissing,"session binding was not found"),!0)}case p.setMode:{const a=String(t.mode_id??t.modeId??"").trim();if(!a)return r(T.modeInvalid,"set mode_id is invalid"),!0;const d=this.normalizeClaudeModeId(a);if(d!==a.toLowerCase())return r(T.modeInvalid,"set mode_id is invalid"),!0;const c=this.bindingStore.get(s);if(!c?.cwd)return r(g.bindingMissing,"session binding was not found"),!0;if(this.normalizeClaudeModeId(c.modeId)!==d){if(this.getClaudeWorkerStatus(s)==="busy")return r(g.workerBusy,"\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148\u505C\u6B62\u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u5F0F"),!0;this.bindingStore.setModeId(s,d),await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,c.cwd)}return n({outcome:"mode_set",mode_id:d,binding:{cwd:c.cwd,mode_id:d}}),!0}case p.setModel:{const a=String(t.model_id??t.modelId??"").trim();if(!a)return r("set_model_invalid","model_id is required"),!0;const d=this.bindingStore.get(s);if(!d?.cwd)return r(g.bindingMissing,"session binding was not found"),!0;if((d.modelId??"")!==a){if(this.getClaudeWorkerStatus(s)==="busy")return r(g.workerBusy,"\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148\u505C\u6B62\u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u578B"),!0;this.bindingStore.setModelId(s,a),this.globalConfigStore?.set(this.name,{modelId:a}),await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,d.cwd)}return n({outcome:"model_set",model_id:a,binding:{cwd:d.cwd,model_id:a}}),!0}case p.listOptions:{const a=F(),d=this.pool.getSlot(s);return n({modes:[C.fullAuto,C.approval],currentModeId:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId),models:a.map(c=>({modelId:c.id,name:c.displayName})),currentModelId:this.bindingStore.get(s)?.modelId??"",available_models:a.map(c=>({id:c.id,displayName:c.displayName})),agent_commands:d?.adapter?.getSupportedCommands?.()??[]}),!0}case p.exec:{const[a,...d]=String(t.args??"").trim().split(/\s+/);if(!a)return r(g.verbInvalid,"Usage: exec <command> [args]"),!0;await this.ensureSlotStarted(s);const l=this.pool.getSlot(s)?.adapter;if(!l?.execCommand)return r(g.verbInvalid,"Agent does not support command execution"),!0;const h=l.getSupportedCommands?.()??[];if(!h.some(m=>m.name===a))return r(g.verbInvalid,`Unknown command: ${a}. Supported: ${h.map(m=>m.name).join(", ")}`),!0;try{const m=await l.execCommand(a,d.join(" "),s);m.status==="ok"?n({outcome:"exec",command:a,message:m.message,data:m.data}):r(g.runtimeError,m.message??`${a} failed`)}catch(m){r(g.runtimeError,`exec error: ${m instanceof Error?m.message:m}`)}return!0}default:return r(g.verbInvalid,`session control verb ${i} is not supported`),!0}}catch(a){const d=a instanceof Error&&a.cwdErrorCode?a.cwdErrorCode:a instanceof Error&&a.sessionControlErrorCode?a.sessionControlErrorCode:g.runtimeError;return u.error(this.name,`handleSessionControlLocalAction error verb=${i} session_id=${s}: ${a instanceof Error?a.message:a}`),r(d,a instanceof Error?a.message:String(a)),!0}}async handleEventCancel(e){const{event_id:o,session_id:t}=e;if(u.info(this.name,`handleEventCancel start event_id=${o} session_id=${t}`),this.pool.cancelEvent(o,t)){if(!(this.pool.getSlot(t)?.adapter?.getActiveEventIds().includes(o)??!1)){this.sendEventResultWithCleanup(o,"canceled","canceled"),this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!0,final_state:"canceled"});return}await this.waitForEventDone(o,t,15e3),this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!0,final_state:"canceled"}),this.pushQueueSnapshotForSession(t);return}this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!1,reason:"event not found or not cancelable"})}waitForEventDone(e,o,t){return new Promise(s=>{const i=this.pool.getSlot(o);if(!i?.adapter){s();return}const n=setTimeout(()=>{i.adapter.removeListener("eventDone",r),s()},t),r=a=>{a===e&&(clearTimeout(n),i.adapter.removeListener("eventDone",r),s())};i.adapter.on("eventDone",r)})}handleAibotStop(e){if(u.info(this.name,`[stop-trace] handleAibotStop begin session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"} adapterType=${this.config.adapterType??"acp"}`),this.aibotHandle.sendEventStopAck({stop_id:e.stop_id,event_id:e.event_id,accepted:!0,updated_at:Date.now()}),this.pool.removeQueuedEvent(e.session_id,e.event_id)){u.info(this.name,`[stop-trace] handleAibotStop removed queued(not-running) event -> stopResult(stopped) session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"}`),this.discardEventTrackingState(e.event_id),this.pushQueueSnapshotForSession(e.session_id),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()});return}const o=this.pool.getSlot(e.session_id),t=o?.adapter?.getStatus().busy??!1,s=(this.config.adapterType??"acp")==="acp",i=o?.adapter?.getActiveEventIds,n=typeof i=="function"?i.call(o.adapter):[],r=n.length>0?n.includes(e.event_id):t,a=(s||(this.config.adapterType??"acp")==="codex"||(this.config.adapterType??"acp")==="claude")&&r;if(u.info(this.name,`[stop-trace] handleAibotStop decision session=${e.session_id} event=${e.event_id} slotExists=${!!o?.adapter} busy=${t} activeIds=[${n.join(",")}] stoppingActiveEvent=${r} killOnStop=${a}`),r&&this.sendCtrl.markEventStopped(e.event_id),o?.adapter&&t){const d=a?this.pool.drainQueuedForSession(e.session_id):[];let c=!1,l=null;const h=v=>{c||(c=!0,l&&(clearTimeout(l),l=null),o.adapter.removeListener("eventDone",m),u.info(this.name,`[stop-trace] handleAibotStop ${v} -> stopResult(stopped) session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"} killOnStop=${a}`),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()}),a&&this.killAndResumeStopSlot(e.session_id,d))},m=v=>{v===e.event_id&&h("eventDone")};o.adapter.on("eventDone",m),l=setTimeout(()=>h("timeout"),St),this.pool.deliverStopEvent(e.event_id,e.session_id)}else u.info(this.name,`[stop-trace] handleAibotStop slot-not-busy -> immediate stopResult(stopped) session=${e.session_id} event=${e.event_id} slotExists=${!!o?.adapter}`),this.pool.deliverStopEvent(e.event_id,e.session_id),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()});(this.config.adapterType??"acp")==="pi"&&this.aibotHandle.sendText({event_id:e.event_id,session_id:e.session_id,content:"/stop",msg_type:0})}async killAndResumeStopSlot(e,o){if(!this.stopped){u.info(this.name,`[stop-trace] killAndResumeStopSlot begin session=${e} siblings=${o.length} -> removeSlot (kill process group)`);try{await this.pool.removeSlot(e),u.info(this.name,`[stop-trace] killAndResumeStopSlot removeSlot done session=${e} (process killed) -> redeliver ${o.length} sibling(s)`)}catch(t){u.warn(this.name,`[acp-stop] removeSlot failed session=${e}: ${t instanceof Error?t.message:String(t)}`)}if(!this.stopped)for(const t of o){if(this.stopped)break;try{await this.pool.deliverInboundEvent(t)}catch(s){u.error(this.name,`[acp-stop] sibling redeliver failed event=${t.event_id} session=${e}: ${s instanceof Error?s.message:String(s)}`),this.sendEventResultWithCleanup(t.event_id,"failed",s instanceof Error?s.message:String(s))}}}}handleAibotRevoke(e){if(!e.event_id||!this.revokeHandler.checkAndTrack(e.event_id))return;const o=e.event_id,t=e.session_id;if(t&&this.pool.cancelEvent(o,t)){(this.pool.getSlot(t)?.adapter?.getActiveEventIds().includes(o)??!1)||this.sendEventResultWithCleanup(o,"canceled","revoked");return}if(this.deferredMgr.removeEvent(o)){this.aibotHandle.sendEventResult({event_id:o,status:"canceled",msg:"revoked",updated_at:Date.now()});return}this.pool.deliverStopEvent(o,t||void 0)}async handleConfigureGatewayProvider(e){const o=e.params??{},t=String(o.api_key??"").trim(),s=String(o.anthropic_base_url??"").trim(),i=String(o.openai_base_url??"").trim(),n=typeof o.provider_id=="string"&&o.provider_id.trim().toLowerCase()||void 0,r=typeof o.quota_base_url=="string"&&o.quota_base_url.trim()||void 0,a=typeof o.model=="string"&&o.model.trim()||void 0;let d=!1;if(!t){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_API_KEY",error_msg:"api_key is required"});return}const c=this.config.adapterType??"acp",l=c==="claude"||c==="codex";try{if(l){const h=c==="claude"?s:i;if(!h){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_TARGET_BASE_URL",error_msg:`${c==="claude"?"anthropic_base_url":"openai_base_url"} is required`});return}const m=_e();if(!m){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"PROXY_UNAVAILABLE",error_msg:"MITM proxy manager not initialized"});return}const v=ve(c),f=Se(c,h,t,{model:a});m.setRoute(f);for(const E of v)m.setHostDefaultRoute(E,f.routeKey);await m.setAgentRelayEnabled(this.name,!0,{relayHosts:v}),d=!await this.recycleAdaptersForRelayChange()}else{if(!this.providerConfigHandler){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"unsupported",error_code:"UNSUPPORTED_CLIENT_TYPE",error_msg:`client type "${c}" does not support Grix relay provider config`});return}await this.providerConfigHandler({...n?{provider_id:n}:{},base_url:i||void 0,...r?{quota_base_url:r}:{},api_key:t,model:a})}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{accepted:!0,restart_deferred:d}})}catch(h){u.error(this.name,`handleConfigureGatewayProvider failed: ${h}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"CONFIGURE_FAILED",error_msg:h instanceof Error?h.message:String(h)})}}async handleAibotLocalAction(e){const o=e.action_type??"",t=String((e.params??{}).session_id??""),s=String((e.params??{}).verb??"").trim().toLowerCase();u.debug(this.name,`local_action received action_type=${o} verb=${s||"-"} action_id=${e.action_id} session_id=${t}`);const i=(this.config.adapterType??"acp")==="claude";if(o===b.sessionControl&&s===p.exec&&await this.handleSessionControlLocalAction(e))return;if(o===b.sessionControl&&s===p.listSessions){await this.handleListSessionsLocalAction(e);return}if(o==="skill_upload"){await this.handleSkillUploadLocalAction(e);return}if(i&&(o===b.interactionReply||o==="exec_approve"||o==="exec_reject")&&(await this.pool.deliverLocalAction(e)).handled||i&&await this.handleSessionControlLocalAction(e))return;if(o===b.sessionControl){const c=this.config.adapterType??"acp",l=c==="codex",h=c==="pi",m=String((e.params??{}).verb??"").trim().toLowerCase();if(l&&m===p.open){await this.handleCodexSessionControlLocalActionOpen(e);return}if(c==="cursor"&&m===p.open){await this.handleCursorSessionControlLocalActionOpen(e);return}if(l&&m==="restart"){const S=this.bindingStore.get(t)?.cwd??"";await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"restarted",binding:{aibotSessionId:t,cwd:S,workerStatus:"ready"}}});return}if(h&&m===p.open){try{const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=await this.resolveCwdForBinding(S),R=String(_.agent_session_id??"").trim();this.ensureImportedAgentSession(R,w),this.bindingStore.set(t,w),this.setResolvedAgentSessionId(t,R),this.sessionBindings.set(t,w),await this.ensureSlotStarted(t).catch(M=>{u.warn("bridge",`pi ensureSlotStarted on local-action bind failed (non-fatal): ${M instanceof Error?M.message:String(M)}`)}),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:w}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,w)}})}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.invalidCwd,error_msg:_ instanceof Error?_.message:String(_)})}return}if(h&&m===p.restart){await this.handlePiSessionControlRestartLocalAction(e);return}if((c==="openhuman"||c==="opencode")&&m===p.open){try{const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=await this.resolveCwdForBinding(S),R=String(_.agent_session_id??"").trim();this.ensureImportedAgentSession(R,w),this.bindingStore.set(t,w),this.setResolvedAgentSessionId(t,R),this.sessionBindings.set(t,w),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,w)}})}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.invalidCwd,error_msg:_ instanceof Error?_.message:String(_)})}return}if(c==="codewhale"&&m===p.open){await this.handleCodeWhaleSessionControlLocalActionOpen(e);return}if(c==="acp"&&m===p.stop){const S=this.bindingStore.get(t)?.cwd??"";await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"stopped",binding:{aibotSessionId:t,cwd:S,workerStatus:"stopped"}}});return}try{if(m===p.open){const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=String(_.agent_session_id??"").trim();if(w){const R=await this.resolveCwdForBinding(S);this.ensureImportedAgentSession(w,R)}}await this.handleSessionControlLocalActionForPool(e)}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.runtimeError,error_msg:_ instanceof Error?_.message:String(_)});return}if(m===p.open){const _=e.params??{},S=await this.resolveCwdForBinding(String(_.cwd??"").trim());this.setResolvedAgentSessionId(t,String(_.agent_session_id??"").trim()),c==="agy"&&(this.bindingStore.set(t,S),this.sessionBindings.set(t,S)),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,S)}}),(this.config.adapterType??"acp")==="agy"&&this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:S,meta:this.buildAgyToolbarMeta(t)})}else Ye(e,this.sessionControlCtx(t),this.sessionControlSenders());return}if(o==="file_list"){const c=Date.now(),l=t?this.bindingStore.get(t)?.cwd:void 0,h=e.params??{},m=String(h.parent_id??"").trim(),v=Array.isArray(h.allowed_extensions)?h.allowed_extensions.filter(S=>typeof S=="string").map(S=>S.trim()).filter(S=>S.length>0):[];u.info("file-list-diag",`plugin << recv action_id=${e.action_id} session_id=${t} parent_id=${m||"<root>"} show_hidden=${!!h.show_hidden} ext_count=${v.length} bound_cwd=${l??"<none>"}`);const f=await tt({parent_id:m||null,session_id:t,show_hidden:!!h.show_hidden,allowed_extensions:v},{resolveCwd:()=>l??this.config.agent.cwd??process.cwd(),fallbackDir:X()}),E=Date.now()-c,_=f.result?.files?.length??0;u.info("file-list-diag",`plugin -> reply action_id=${e.action_id} status=${f.status} elapsed=${E}ms count=${_} current_path=${f.result?.current_path??""} error_code=${f.error_code??""} error_msg=${f.error_msg??""}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:f.status,...f.result?{result:{...f.result,machine_name:st()}}:{},...f.error_code?{error_code:f.error_code}:{},...f.error_msg?{error_msg:f.error_msg}:{}});return}if(o==="create_folder"){const c=t?this.bindingStore.get(t)?.cwd:void 0,l=String((e.params??{}).parent_id??"").trim(),h=String((e.params??{}).name??"").trim(),m=await it({parent_id:l||null,name:h,session_id:t},{resolveCwd:()=>c??this.config.agent.cwd??process.cwd(),fallbackDir:X()});this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:m.status,...m.result?{result:m.result}:{},...m.error_code?{error_code:m.error_code}:{},...m.error_msg?{error_msg:m.error_msg}:{}});return}if(o===b.setModel&&(this.config.adapterType??"acp")==="agy"){await this.handleAgySetModel(e,t);return}if(o===b.getSessionUsage){await this.handleGetSessionUsage(e,t);return}if(o===b.getRateLimits){await this.handleGetRateLimits(e,t);return}const n=(this.config.adapterType??"acp")==="acp";if((i||n)&&o===b.threadCompact){await this.handleThreadCompact(e,t);return}if(o==="connector_rollback"){await this.handleConnectorRollback(e);return}if(o==="connector_upgrade_push"){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{accepted:!0}}),this.upgradeTrigger?.();return}if(o===b.getAgentGlobalConfig){const c=this.globalConfigStore?.get(this.name);this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{agentName:this.name,...c?{config:c}:{config:null}}});return}if(o==="configure_gateway_provider"){await this.handleConfigureGatewayProvider(e);return}const r=this.config.adapterType??"acp",a=(r==="codex"||r==="cursor"||r==="pi"||r==="openhuman"||r==="opencode"||r==="acp")&&!!t&&!!this.bindingStore.get(t)?.cwd,d=await this.pool.deliverLocalAction(e,{autoCreateSlot:a});if(d.handled){if(d.kind==="set_mode"){const c=String((e.params??{}).mode_id??"");c&&(r==="cursor"||r==="claude"?this.bindingStore.setModeId(t,c):L.has(r)||this.globalConfigStore?.set(this.name,{acpInitialMode:c}))}else if(d.kind==="set_model"){const c=String((e.params??{}).model_id??"");c&&(r==="cursor"?(this.bindingStore.setModelId(t,c),this.globalConfigStore?.set(this.name,{modelId:c})):r==="codex"?this.globalConfigStore?.set(this.name,{codexModelId:c}):L.has(r)||this.globalConfigStore?.set(this.name,{modelId:c}))}else if(d.kind==="set_reasoning_effort"){const c=String((e.params??{}).reasoning_effort??(e.params??{}).reasoning_eff??(e.params??{}).effort??"");c&&this.globalConfigStore?.set(this.name,{codexReasoningEffort:c})}else if(d.kind==="set_sandbox_mode"){const c=String((e.params??{}).sandbox_mode??(e.params??{}).sandboxMode??"");if(c){const l=c==="default"?void 0:c;this.globalConfigStore?.set(this.name,{codexSandboxMode:l})}}return}if((r==="codex"||r==="cursor"||r==="pi"||r==="openhuman"||r==="opencode")&&t&&!this.bindingStore.get(t)?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"Session binding missing. Open a workspace first."});return}if(r==="acp"&&(o==="set_mode"||o==="set_model")){const c=this.sessionControlSenders(),l=this.pool.getSlot(t)?.adapter,h={bindingStore:this.bindingStore,acpAdapter:l instanceof A?l:null,globalConfigStore:this.globalConfigStore,agentName:this.name,log:u};if(o==="set_mode"){const m=String((e.params??{}).mode_id??""),v=await Y(h,t,m);if(v.status==="failed")c.sendLocalActionResult(e.action_id,"failed",void 0,v.errorCode,v.errorMsg);else{const f=l instanceof A?l.buildToolbarContext(v.result?.outcome==="mode_set"?"mode_set":"mode_set_failed"):null,E=f?{...f,...v.result}:v.result;c.sendLocalActionResult(e.action_id,"ok",E)}}else{const m=String((e.params??{}).model_id??""),v=await J(h,t,m);if(v.status==="failed")c.sendLocalActionResult(e.action_id,"failed",void 0,v.errorCode,v.errorMsg);else{const f=l instanceof A?l.buildToolbarContext(v.result?.outcome==="model_set"?"model_set":"model_set_failed"):null,E=f?{...f,...v.result}:v.result;c.sendLocalActionResult(e.action_id,"ok",E)}}return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"unsupported_local_action",error_msg:`action type ${o} is not supported`})}async handleGetSessionUsage(e,o){if(!o){u.warn(this.name,`[usage] get_session_usage rejected: no session_id in params, action_id=${e.action_id}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for get_session_usage"});return}const t=this.config.adapterType??"acp",s=this.bindingStore.get(o),i=s?.cwd??this.config.agent.cwd??process.cwd();u.info(this.name,`[usage] get_session_usage action_id=${e.action_id} session_id=${o} adapterType=${t} hasBinding=${!!s} cwd=${i}`);let n=null,r,a;switch(t){case"claude":{if(r=s?.claudeSessionId,!r){u.warn(this.name,`[usage] no claude binding for session_id=${o}, action_id=${e.action_id}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Claude session binding found"});return}u.info(this.name,`[usage] parsing claude usage: claudeSessionId=${r} cwd=${i}`),n=await ue(r,i),a="claude";break}case"agy":{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"agy",available:!1,reason:"agy print-mode adapter does not track token usage"}});return}case"acp":default:{if(r=s?.acpSessionId,!r){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No ACP session binding found"});return}n=await pe(r,i,this.config.aibot.clientType),a=t,(!a||a==="acp")&&(a="acp");break}case"codex":{if(r=s?.codexThreadId,!r){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Codex thread binding found"});return}n=await Ce(r),a="codex";break}case"pi":{if(r=s?.piSessionPath,!r){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Pi session path binding found"});return}n=await Ie(r),a="pi";break}case"cursor":{const c=this.pool.getSlot(o)?.adapter;if(!(c instanceof Q)){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Cursor session found"});return}const l=c.getUsageSnapshot(o);if(!l){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"cursor",models:[{modelId:this.bindingStore.get(o)?.modelId??"auto",turns:l.turns,input:l.total.input,output:l.total.output,cacheRead:l.total.cacheRead,cacheWrite:l.total.cacheWrite}],total:{input:l.total.input,output:l.total.output,cacheRead:l.total.cacheRead,cacheWrite:l.total.cacheWrite},turns:l.turns,sampledAt:l.sampledAt}});return}case"codewhale":{const c=this.pool.getSlot(o)?.adapter;if(!(c instanceof N)){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No CodeWhale session found"});return}const l=c.getUsageSnapshot();if(!l){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"codewhale",models:[{modelId:this.bindingStore.get(o)?.modelId??"codewhale",turns:l.turns,input:l.total.input,output:l.total.output}],total:{input:l.total.input,output:l.total.output},turns:l.turns,sampledAt:l.sampledAt}});return}}if(!n){u.info(this.name,`[usage] no usage data found: session_id=${o} adapterSessionId=${r} adapterType=${a}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}u.info(this.name,`[usage] result ok: session_id=${o} adapterSessionId=${r} turns=${n.turns} models=${n.models.length}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:r,adapterType:a,models:n.models,total:n.total,turns:n.turns,sampledAt:new Date().toISOString()}})}async handleThreadCompact(e,o){if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for thread_compact"});return}if(!this.bindingStore.get(o)?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"session binding was not found"});return}try{await this.ensureSlotStarted(o);const i=this.pool.getSlot(o)?.adapter;if(!i?.execCommand){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:"Agent does not support command execution"});return}u.info(this.name,`thread_compact session_id=${o} action_id=${e.action_id}`);const n=await i.execCommand("compact","",o);n.status==="ok"?this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"compacted",message:n.message,data:n.data}}):this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:n.message??"compact failed"})}catch(s){u.warn(this.name,`thread_compact error session_id=${o}: ${s instanceof Error?s.message:s}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:s instanceof Error?s.message:String(s)})}}async handleGetRateLimits(e,o){const t=this.config.adapterType??"acp";if(this.config.aibot.clientType==="kiro"){const s=this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs),i=this.cachedAcpContextWindow;if(s&&this.cachedProviderQuota){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!0,cached:!0,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:i,tokenUsage:null,providerQuota:this.cachedProviderQuota}});return}try{const n=await O();this.cachedProviderQuota=n,this.cachedProviderQuotaSampledAtMs=Date.now(),u.info(this.name,`[rate-limits] kiro quota queried: success=${n.success}`+(n.balance?` balance=${n.balance.remaining} ${n.balance.unit}`:"")+(n.error?` error=${n.error}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!0,cached:!1,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:i,tokenUsage:null,providerQuota:n}})}catch(n){u.warn(this.name,`[rate-limits] kiro quota query failed: ${n instanceof Error?n.message:String(n)}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:i,tokenUsage:null}})}return}switch(t){case"codex":{const s=this.maybeQueryProviderQuota(),i=this.getFreshCodexGlobalRateLimitCache();if(i.hasData&&!this.isRateLimitsCacheFresh(this.cachedRateLimitsSampledAtMs)&&!this.isRateLimitsCacheFresh(this.cachedCodexUsageSampledAtMs)){const l=this.pool.getAllSlots().find(h=>h.state==="ready"&&h.adapter);if(l&&(u.info(this.name,`[rate-limits] codex cache stale, refreshing from slot: session=${l.sessionId}`),(await l.adapter.handleLocalAction?.(e))?.handled))return}if(i.hasData){i.rateLimits?u.info(this.name,`[rate-limits] codex cached: primary=${i.rateLimits.primary.usedPercent.toFixed(1)}% resetsAt=${i.rateLimits.primary.resetsAt} secondary=${i.rateLimits.secondary.usedPercent.toFixed(1)}% resetsAt=${i.rateLimits.secondary.resetsAt}`):u.info(this.name,`[rate-limits] codex cached context/token only: hasContext=${!!i.contextWindow} hasToken=${!!i.tokenUsage}`);const l=await s;this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"codex",available:i.hasData||!!l,cached:!0,sampledAt:i.sampledAt,rateLimits:i.rateLimits,contextWindow:i.contextWindow,tokenUsage:i.tokenUsage,providerQuota:l}});return}const r=this.pool.getAllSlots().find(l=>l.state==="ready"&&l.adapter);if(r&&(u.info(this.name,`[rate-limits] codex reuse existing slot: session=${r.sessionId}`),(await r.adapter.handleLocalAction?.(e))?.handled))return;const a=this.resolveRateLimitWakeSessionId(o,t);if(a){const l=await this.wakeRateLimitSlot(a,t);if(l?.adapter&&(await l.adapter.handleLocalAction?.(e))?.handled)return}const d=await s,c=!!d;u.info(this.name,`[rate-limits] codex no native data, providerQuota=${d?d.provider:"none"} available=${c}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"codex",available:c,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null,providerQuota:d}});return}case"claude":{const s=this.maybeQueryProviderQuota(),i=this.getFreshClaudeRateLimitState();if(i){const c=i.rateLimits,l=await s;u.info(this.name,`[rate-limits] claude global cached: sampledAt=${i.sampledAt} hasRateLimits=${!!c}`+(l?` providerQuota=${l.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!0,cached:!0,sampledAt:i.sampledAt,rateLimits:i.rateLimits??null,contextWindow:i.contextWindow,tokenUsage:null,providerQuota:l}});return}let n=this.pool.getAllSlots().find(c=>c.state==="ready"&&c.adapter instanceof P)??null;if(!n){const c=this.resolveRateLimitWakeSessionId(o,t);c&&(n=await this.wakeRateLimitSlot(c,t))}u.info(this.name,`[rate-limits] handleGetRateLimits: session_id=${o} adapterType=claude hasSlot=${!!n} hasAdapter=${!!n?.adapter}`);const r=n?.adapter,a=r instanceof P?r.getSessionState():null,d=await s;if(a){(a.rateLimits?.fiveHour||a.rateLimits?.sevenDay||a.contextWindow?.usedPercentage!=null)&&(this.cachedClaudeRateLimitState=a);const c=this.getFreshClaudeRateLimitState(),l=c?.rateLimits&&!a.rateLimits?{...a,rateLimits:c.rateLimits}:a,h=l.rateLimits;u.info(this.name,`[rate-limits] claude global state: sampledAt=${l.sampledAt} hasRateLimits=${!!h}`+(h?` fiveHour=${h.fiveHour?.usedPercentage??"n/a"}% resetsAt=${h.fiveHour?.resetsAt??"n/a"} sevenDay=${h.sevenDay?.usedPercentage??"n/a"}% resetsAt=${h.sevenDay?.resetsAt??"n/a"}`:"")+(c?.rateLimits&&!a.rateLimits?" source=live+cached-fallback":" source=live")+(d?` providerQuota=${d.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!0,cached:!1,sampledAt:l.sampledAt,rateLimits:l.rateLimits??null,contextWindow:l.contextWindow,tokenUsage:null,providerQuota:d}})}else u.info(this.name,`[rate-limits] claude no global state: hasAdapter=${!!r} adapterType=${r?.constructor?.name??"n/a"}`+(d?` providerQuota=${d.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!!d,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null,providerQuota:d}});return}case"cursor":{const i=(o?this.pool.getSlot(o):null)?.adapter,n=i instanceof Q?i.getRateLimitsSnapshot():{adapterType:"cursor",available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null};this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:n});return}default:{const s=this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&!!this.cachedProviderQuota,i=await this.maybeQueryProviderQuota();if(!i){u.info(this.name,`[rate-limits] no provider quota available for adapterType=${t}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:t,available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:this.cachedAcpContextWindow??null,tokenUsage:null}});return}u.info(this.name,`[rate-limits] provider quota ${s?"cached":"queried"}: provider=${i.provider} success=${i.success}`+(i.tiers.length>0?` tiers=${i.tiers.map(n=>`${n.name}=${n.usedPercent}%`).join(",")}`:"")+(i.balance?` balance=${i.balance.remaining} ${i.balance.unit}`:"")+(i.error?` error=${i.error}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:t,available:!0,cached:s,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:this.cachedAcpContextWindow??null,tokenUsage:null,providerQuota:i}});return}}}resolveRateLimitWakeSessionId(e,o){const t=String(e??"").trim(),s=this.bindingStore.getMostRecentlyUpdatedSessionId({requireCwd:!0});return s?t&&t===s?t:o==="codex"||o==="claude"?(t&&t!==s&&u.info(this.name,`[rate-limits] ${o} remap wake session: requested=${t} use_recent=${s}`),s):t&&t!==s?(u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=not_recent_session recent=${s}`),null):t||null:(u.info(this.name,`[rate-limits] skip wake slot: adapterType=${o} reason=no_recent_binding`),null)}async wakeRateLimitSlot(e,o){const t=String(e??"").trim();if(!t)return null;if(!this.bindingStore.get(t)?.cwd)return u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=binding_cwd_missing`),null;try{const i=this.pool.getOrCreateSlot(t);if(!i)return u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=slot_unavailable`),null;if(i.startPromise){const n=o==="claude"?6e4:2e4;await Promise.race([i.startPromise,new Promise((r,a)=>setTimeout(()=>a(new Error(`wake rate-limits slot timeout (${n}ms)`)),n))])}return u.info(this.name,`[rate-limits] wake slot success: session=${t} adapterType=${o}`),this.pool.getSlot(t)??i}catch(i){return u.warn(this.name,`[rate-limits] wake slot failed: session=${t} adapterType=${o} err=${i instanceof Error?i.message:String(i)}`),null}}sessionControlCtx(e){const o=this.pool.getSlot(e),t=o?.adapter instanceof A?o.adapter:null,s=(this.config.adapterType??"acp")==="acp",i={bindingStore:this.bindingStore,acpAdapter:t,globalConfigStore:this.globalConfigStore,agentName:this.name,log:u};return{getCwd:()=>this.bindingStore.get(e)?.cwd??this.config.agent.cwd??process.cwd(),getSessionBindings:()=>{if(t)return t.getSessionBindings();const n=this.sessionBindings;if(e&&!n.has(e)){const r=this.bindingStore.get(e);r?.cwd&&n.set(e,r.cwd)}return n},getStatus:()=>this.getStatus(),isAcpAlive:!!t?.isAlive(),getAcpSessionOptions:()=>t?.acpSessionOptions??null,setMode:n=>t?t.setMode(n):Promise.resolve(!1),setModel:n=>t?t.setModel(n):Promise.resolve(!1),acpSetMode:s?(n,r)=>Y(i,n,r):void 0,acpSetModel:s?(n,r)=>J(i,n,r):void 0,getPendingApproval:n=>{const r=t?.pendingApprovalEntries.get(n);return r?{requestId:r}:void 0},deletePendingApproval:n=>t?.pendingApprovalEntries.delete(n)??!1,respondPermission:(n,r)=>(t&&t.respondToPermission(n,r),Promise.resolve()),onSessionBound:(n,r)=>{this.bindingStore.set(n,r)},onSessionUnbound:n=>{const a=this.bindingStore.get(n)?.cwd??"";this.bindingStore.delete(n),this.sessionBindings.delete(n),this.sessionProviderHints.delete(n),this.sessionProviderQuotas.delete(n),this.sessionProviderMeta.delete(n),this.claudeWorkerStatus.delete(n),this.deferredMgr.clearSession(n);for(const d of this.pool.drainQueuedForSession(n))this.discardEventTrackingState(d.event_id);a&&this.aibotHandle.sendUpdateBindingCard({session_id:n,worker_status:"stopped",cwd:a})},cancelActiveRun:()=>(this.config.adapterType??"acp")==="agy"&&o?.adapter instanceof K?(o.adapter.cancelCurrentRun(),Promise.resolve()):o?.adapter?.cancel("")??Promise.resolve(),onModeSet:n=>{const r=this.config.adapterType??"acp";L.has(r)||this.globalConfigStore?.set(this.name,{acpInitialMode:n})},onModelSet:n=>{const r=this.config.adapterType??"acp";L.has(r)||this.globalConfigStore?.set(this.name,{modelId:n})}}}sessionControlSenders(){return{sendEventAck:(e,o)=>this.aibotHandle.sendEventAck({event_id:e,session_id:o,received_at:Date.now()}),sendEventResult:(e,o,t)=>this.aibotHandle.sendEventResult({event_id:e,status:o,...t?.msg?{msg:t.msg}:{},...t?.code?{code:t.code}:{},updated_at:Date.now()}),sendLocalActionResult:(e,o,t,s,i)=>this.aibotHandle.sendLocalActionResult({action_id:e,status:o,...t?{result:t}:{},...s?{error_code:s}:{},...i?{error_msg:i}:{}})}}resolveBindingChannelKey(e){switch(e){case"claude":return"grix-claude";case"codex":return"codex";case"cursor":return"cursor";case"pi":return"pi";case"openhuman":return"openhuman";case"codewhale":return"codewhale";case"opencode":return"opencode";case"agy":return"acp";case"acp":return this.config.aibot.clientType==="qwen"?"qwen":"acp";default:return e}}finalizeThinking(e,o){this.sendCtrl.finalizeThinking(e,o)}logInboundConversation(e){const o=String(e.session_id??"").trim();o&&this.conversationLog?.logInbound(o,{event_id:e.event_id,msg_id:e.msg_id,sender_id:e.sender_id,msg_type:e.msg_type,content:e.content??""})}buildInboundEvent(e,o){const t=dt(this.sendCtrl.getGlobalRuntimeConfig(),o);return{event_id:e.event_id,session_id:e.session_id,thread_id:e.thread_id,sender_id:e.sender_id,msg_id:e.msg_id,msg_type:e.msg_type,content:e.content??"",quoted_message_id:e.quoted_message_id,context_messages_json:this.buildContextMessagesJson(e.context_messages),extra_json:e.extra?JSON.stringify(e.extra):void 0,connector_runtime_config:{response_delivery:t.responseDelivery,tool_events:t.toolEvents,thinking_events:t.thinkingEvents},session_type:e.session_type,created_at:e.created_at}}buildContextMessagesJson(e){if(!e)return;const o=e.filter(t=>!Xe(String(t?.content??"")));if(o.length!==0)return JSON.stringify(o)}isStaleEvent(e){const o=Number(e.created_at);return!Number.isFinite(o)||o<=0?!1:Date.now()-o>_t}async handleConnectorRollback(e){const o=String((e.params??{}).target_version??"").trim(),t=String((e.params??{}).reason??"server_initiated");if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_TARGET_VERSION",error_msg:"target_version is required for connector_rollback"});return}u.info(this.name,`connector_rollback: target=${o} reason=${t}`);try{const{npmInstall:s,writePending:i,removePending:n,upgradeLog:r}=await import("../core/upgrade/npm-upgrader.js"),{resolveClientVersion:a}=await import("../core/util/client-version.js"),d=a();r(`server rollback: ${d} -> ${o} reason=${t}`),i(d,o),await s("grix-connector",o),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{rolled_back_to:o}}),process.kill(process.pid,"SIGTERM")}catch(s){try{const{removePending:n}=await import("../core/upgrade/npm-upgrader.js");n()}catch{}const i=s instanceof Error?s.message:String(s);u.error(this.name,`connector_rollback failed: ${i}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"ROLLBACK_FAILED",error_msg:i})}}setUpgradeTrigger(e){this.upgradeTrigger=e}setAgentDeletedHandler(e){this.agentDeletedHandler=e}setSkillSyncHandler(e){this.skillSyncHandler=e}setShareSetHandler(e){this.shareSetHandler=e}setProviderConfigHandler(e){this.providerConfigHandler=e}isSharedInstance(){return!!this.config.aibot.sharedOwnerId}}export{Ei as AgentInstance};
|
|
15
|
+
`)}`,updated_at:Date.now()})}async handleListSessionsLocalAction(e){const o=this.config.adapterType??"acp",t=new Map;for(const r of this.pool.getAllSlots())t.set(r.sessionId,r);const s=Array.from(this.bindingStore.entries()),i=new Map;for(const[r,d]of s){const c=this.resolveAgentSessionId(d);if(c){const l=t.get(r),h=l?l.adapter.getStatus().busy?"busy":"ready":"inactive";i.set(c,{aibotSessionId:r,workerStatus:h,bindingUpdatedAt:d.updatedAt??0})}}const n=[],a=new Set;if(o==="codex"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.threadId);const c=i.get(d.threadId),l={agentSessionId:d.threadId,cwd:d.cwd||null,workerStatus:c?.workerStatus??(d.archived?"archived":"inactive"),createdAt:d.createdAt,updatedAt:d.updatedAt,archived:d.archived};c&&(l.aibotSessionId=c.aibotSessionId),d.title&&(l.title=d.title),n.push(l)}}else if(o==="claude"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionId);const c=i.get(d.sessionId),l={agentSessionId:d.sessionId,cwd:d.cwd||null,workerStatus:c?.workerStatus??"inactive",updatedAt:d.updatedAt};c&&(l.aibotSessionId=c.aibotSessionId),d.title&&(l.title=d.title),n.push(l)}}else if(o==="acp"){const r=this.sessionScanCache.get();for(const d of r){a.add(d.sessionId);const c=i.get(d.sessionId),l=c&&c.bindingUpdatedAt>0?c.bindingUpdatedAt:d.updatedAt,h={agentSessionId:d.sessionId,cwd:d.cwd||null,agentType:d.agentType,workerStatus:c?.workerStatus??"inactive",updatedAt:l};c&&(h.aibotSessionId=c.aibotSessionId),d.title&&(h.title=d.title),d.createdAt&&(h.createdAt=d.createdAt),n.push(h)}}else if(o==="pi"||o==="codewhale"||o==="cursor"||o==="opencode"){const r=this.sessionScanCache.get();for(const d of r){const c="sessionPath"in d?d.sessionPath:d.sessionId;a.add(c);const l=i.get(c),h={agentSessionId:c,cwd:d.cwd||null,workerStatus:l?.workerStatus??("archived"in d&&d.archived?"archived":"inactive"),updatedAt:l&&l.bindingUpdatedAt>0?l.bindingUpdatedAt:d.updatedAt};l&&(h.aibotSessionId=l.aibotSessionId),d.title&&(h.title=d.title),d.createdAt&&(h.createdAt=d.createdAt),"archived"in d&&(h.archived=d.archived),n.push(h)}}for(const[r,d]of s){const c=this.resolveAgentSessionId(d);if(c&&a.has(c))continue;const l=t.get(r),h=l?l.adapter.getStatus().busy?"busy":"ready":"closed",m={aibotSessionId:r,cwd:d.cwd??null,workerStatus:h,updatedAt:d.updatedAt,title:this.resolveOrphanTitle(c)??(c?`${c.slice(0,8)}\u2026`:`${r.slice(0,8)}\u2026`)};c&&(m.agentSessionId=c),n.push(m)}n.sort((r,d)=>d.updatedAt-r.updatedAt),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"sessions_listed",sessions:n,total:n.length}})}async handleSessionControlCommand(e,o){const t=o.session_id,s=(i,n,a)=>{this.aibotHandle.sendEventResult({event_id:o.event_id,status:i,...n?{msg:n}:{},...a?{code:a}:{},updated_at:Date.now()})};this.aibotHandle.sendEventAck({event_id:o.event_id,session_id:t,received_at:Date.now()});try{switch(e.verb){case p.open:{await B().catch(()=>{});const i=e.args.trim();if(!i){s("failed","Usage: /grix open <working-directory>",g.cwdRequired);return}let n="";try{n=await this.resolveCwdForBinding(i)}catch(r){s("failed",r instanceof Error?r.message:String(r),g.invalidCwd);return}const a=this.bindingStore.get(t);if(a?.cwd){const r=await this.resolveCwdForBinding(a.cwd);if(r!==n){s("failed","session binding cannot be changed to another working directory",g.rebindForbidden);return}this.bindingStore.ensureModeId(t,C.fullAuto),this.sessionBindings.set(t,r),await this.ensureSlotStarted(t),await this.deferredMgr.release(t,this.deferredCallbacks()),this.refreshClaudeWorkerStatusCard(t,r),s("responded",`Working directory already bound: ${r}`);return}this.bindingStore.set(t,n,{modeId:C.fullAuto}),this.sessionBindings.set(t,n),await this.ensureSlotStarted(t),await this.deferredMgr.release(t,this.deferredCallbacks()),this.refreshClaudeWorkerStatusCard(t,n),s("responded",`Session bound to ${n}`);return}case p.where:{const i=this.bindingStore.get(t);if(!i?.cwd){s("failed","session binding was not found",g.bindingMissing);return}s("responded",`Working directory: ${i.cwd}`);return}case p.status:{const i=this.bindingStore.get(t);if(!i?.cwd){s("failed","session binding was not found",g.bindingMissing);return}const n=this.normalizeClaudeModeId(i.modeId),a=this.getClaudeWorkerStatus(t);s("responded",`Status: worker=${a} mode=${n} cwd=${i.cwd}`);return}case p.stop:{const i=this.bindingStore.get(t);if(!i?.cwd){s("failed","session binding was not found",g.bindingMissing);return}await this.pool.removeSlot(t),this.claudeWorkerStatus.set(t,"stopped"),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"stopped",cwd:i.cwd}),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"stopped",cwd:i.cwd,meta:this.buildClaudeToolbarMeta(t)}),s("responded",`Session worker stopped for ${i.cwd}`);return}case p.restart:{const i=this.bindingStore.get(t);if(!i?.cwd){s("failed","session binding was not found",g.bindingMissing);return}await this.pool.removeSlot(t),await this.ensureSlotStarted(t),this.refreshClaudeWorkerStatusCard(t,i.cwd),s("responded",`Session worker restarted for ${i.cwd}`);return}case p.setMode:{const i=e.args.trim();if(!i){s("failed","Usage: /grix set_mode <mode-id>",T.modeInvalid);return}const n=this.normalizeClaudeModeId(i);if(n!==i.toLowerCase()){s("failed","set mode_id is invalid",T.modeInvalid);return}const a=this.bindingStore.get(t);if(!a?.cwd){s("failed","session binding was not found",g.bindingMissing);return}if(this.normalizeClaudeModeId(a.modeId)===n){s("responded",`Mode unchanged: ${n}`);return}if(this.getClaudeWorkerStatus(t)==="busy"){s("failed","\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148 /grix stop \u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u5F0F",g.workerBusy);return}this.bindingStore.setModeId(t,n),await this.pool.removeSlot(t),await this.ensureSlotStarted(t),this.refreshClaudeWorkerStatusCard(t,a.cwd),s("responded",`Mode set to ${n}`);return}case p.setModel:{const i=e.args.trim();if(!i){s("failed","Usage: /grix set_model <model-id>");return}const n=this.bindingStore.get(t);if(!n?.cwd){s("failed","session binding was not found",g.bindingMissing);return}if((n.modelId??"")===i){s("responded",`Model unchanged: ${i}`);return}if(this.getClaudeWorkerStatus(t)==="busy"){s("failed","\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148 /grix stop \u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u578B",g.workerBusy);return}this.bindingStore.setModelId(t,i),this.globalConfigStore?.set(this.name,{modelId:i}),await this.pool.removeSlot(t),await this.ensureSlotStarted(t),this.refreshClaudeWorkerStatusCard(t,n.cwd),s("responded",`Model set to ${i}`);return}case p.listOptions:{const i=F(),n=this.bindingStore.get(t),a=i.map(d=>d.id).join(", "),r=`${C.fullAuto}, ${C.approval}`;s("responded",`Modes (current: ${this.normalizeClaudeModeId(n?.modeId)}): ${r}
|
|
16
|
+
Models (current: ${n?.modelId??"default"}): ${a}`);return}case p.exec:{const[i,...n]=e.args.trim().split(/\s+/);if(!i){s("failed","Usage: /grix exec <command> [args]",g.verbInvalid);return}const r=this.pool.getSlot(t)?.adapter;if(!r?.execCommand){s("failed","Agent does not support command execution",g.verbInvalid);return}const d=r.getSupportedCommands?.()??[];if(!d.some(c=>c.name===i)){s("failed",`Unknown command: ${i}. Supported: ${d.map(c=>c.name).join(", ")}`,g.verbInvalid);return}try{const c=await r.execCommand(i,n.join(" "),t);s(c.status==="ok"?"responded":"failed",c.message??`${i} ${c.status}`,c.status==="ok"?void 0:g.runtimeError)}catch(c){s("failed",`exec error: ${c instanceof Error?c.message:c}`,g.runtimeError)}return}default:s("failed",`Unsupported command for Claude: /grix ${e.verb}`,g.verbInvalid)}}catch(i){s("failed",i instanceof Error?i.message:String(i),g.runtimeError)}}async handleSessionControlLocalAction(e){const o=String(e.action_type??"").trim();if(o!==b.sessionControl&&o!==b.setMode&&o!=="set_mode"&&o!==b.setModel&&o!=="set_model")return!1;const t=e.params??{},s=String(t.session_id??"").trim(),i=o===b.setMode?p.setMode:o===b.setModel?p.setModel:String(t.verb??"").trim().toLowerCase();if(u.info(this.name,`handleSessionControlLocalAction verb=${i} action_id=${e.action_id} session_id=${s}`),!s)return this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:T.localActionRouteMissing,error_msg:"local action session_id is required"}),!0;const n=r=>{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:r})},a=(r,d)=>{u.warn(this.name,`session_control local_action failed action_id=${e.action_id} session_id=${s} verb=${i} code=${r} msg=${d}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:r,error_msg:d})};try{switch(i){case p.open:{await B().catch(()=>{});const r=String(t.cwd??"").trim(),d=String(t.agent_session_id??"").trim();if(!r)return a(g.cwdRequired,"session control cwd is required"),!0;u.info(this.name,`handleSessionControlLocalAction open cwd=${r} session_id=${s}`);const c=await this.resolveCwdForBinding(r);this.ensureImportedAgentSession(d,c);const l=this.bindingStore.get(s);if(l?.cwd){const h=await this.resolveCwdForBinding(l.cwd);return h!==c?(a(g.rebindForbidden,"session binding cannot be changed to another working directory"),!0):(this.bindingStore.ensureModeId(s,C.fullAuto),this.setResolvedAgentSessionId(s,d),this.sessionBindings.set(s,h),await this.ensureSlotStarted(s),await this.replayDeferredEventsForSession(s),this.refreshClaudeWorkerStatusCard(s,h),n({outcome:"opened",binding:{...this.buildOpenedBindingResult(s,h),mode_id:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId)}}),!0)}return this.bindingStore.set(s,c,{modeId:C.fullAuto}),this.setResolvedAgentSessionId(s,d),this.sessionBindings.set(s,c),await this.ensureSlotStarted(s),await this.replayDeferredEventsForSession(s),this.refreshClaudeWorkerStatusCard(s,c),n({outcome:"opened",binding:{...this.buildOpenedBindingResult(s,c),mode_id:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId)}}),!0}case p.status:case p.where:{const r=this.bindingStore.get(s);return r?.cwd?(n({outcome:i,binding:{cwd:r.cwd,mode_id:this.normalizeClaudeModeId(r.modeId),worker_status:this.getClaudeWorkerStatus(s)}}),!0):(a(g.bindingMissing,"session binding was not found"),!0)}case p.stop:{const r=this.bindingStore.get(s);return r?.cwd?(await this.pool.removeSlot(s),n({outcome:"stopped",binding:{cwd:r.cwd,mode_id:this.normalizeClaudeModeId(r.modeId),worker_status:"stopped"}}),!0):(a(g.bindingMissing,"session binding was not found"),!0)}case p.restart:{const r=this.bindingStore.get(s);return r?.cwd?(await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,r.cwd),n({outcome:"restarted",binding:{cwd:r.cwd,mode_id:this.normalizeClaudeModeId(r.modeId)}}),!0):(a(g.bindingMissing,"session binding was not found"),!0)}case p.setMode:{const r=String(t.mode_id??t.modeId??"").trim();if(!r)return a(T.modeInvalid,"set mode_id is invalid"),!0;const d=this.normalizeClaudeModeId(r);if(d!==r.toLowerCase())return a(T.modeInvalid,"set mode_id is invalid"),!0;const c=this.bindingStore.get(s);if(!c?.cwd)return a(g.bindingMissing,"session binding was not found"),!0;if(this.normalizeClaudeModeId(c.modeId)!==d){if(this.getClaudeWorkerStatus(s)==="busy")return a(g.workerBusy,"\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148\u505C\u6B62\u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u5F0F"),!0;this.bindingStore.setModeId(s,d),await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,c.cwd)}return n({outcome:"mode_set",mode_id:d,binding:{cwd:c.cwd,mode_id:d}}),!0}case p.setModel:{const r=String(t.model_id??t.modelId??"").trim();if(!r)return a("set_model_invalid","model_id is required"),!0;const d=this.bindingStore.get(s);if(!d?.cwd)return a(g.bindingMissing,"session binding was not found"),!0;if((d.modelId??"")!==r){if(this.getClaudeWorkerStatus(s)==="busy")return a(g.workerBusy,"\u5F53\u524D\u6709\u6D88\u606F\u6B63\u5728\u8FD0\u884C\uFF0C\u8BF7\u5148\u505C\u6B62\u6216\u7B49\u5F85\u5176\u5B8C\u6210\u540E\u518D\u5207\u6362\u6A21\u578B"),!0;this.bindingStore.setModelId(s,r),this.globalConfigStore?.set(this.name,{modelId:r}),await this.pool.removeSlot(s),await this.ensureSlotStarted(s),this.refreshClaudeWorkerStatusCard(s,d.cwd)}return n({outcome:"model_set",model_id:r,binding:{cwd:d.cwd,model_id:r}}),!0}case p.listOptions:{const r=F(),d=this.pool.getSlot(s);return n({modes:[C.fullAuto,C.approval],currentModeId:this.normalizeClaudeModeId(this.bindingStore.get(s)?.modeId),models:r.map(c=>({modelId:c.id,name:c.displayName})),currentModelId:this.bindingStore.get(s)?.modelId??"",available_models:r.map(c=>({id:c.id,displayName:c.displayName})),agent_commands:d?.adapter?.getSupportedCommands?.()??[]}),!0}case p.exec:{const[r,...d]=String(t.args??"").trim().split(/\s+/);if(!r)return a(g.verbInvalid,"Usage: exec <command> [args]"),!0;await this.ensureSlotStarted(s);const l=this.pool.getSlot(s)?.adapter;if(!l?.execCommand)return a(g.verbInvalid,"Agent does not support command execution"),!0;const h=l.getSupportedCommands?.()??[];if(!h.some(m=>m.name===r))return a(g.verbInvalid,`Unknown command: ${r}. Supported: ${h.map(m=>m.name).join(", ")}`),!0;try{const m=await l.execCommand(r,d.join(" "),s);m.status==="ok"?n({outcome:"exec",command:r,message:m.message,data:m.data}):a(g.runtimeError,m.message??`${r} failed`)}catch(m){a(g.runtimeError,`exec error: ${m instanceof Error?m.message:m}`)}return!0}default:return a(g.verbInvalid,`session control verb ${i} is not supported`),!0}}catch(r){const d=r instanceof Error&&r.cwdErrorCode?r.cwdErrorCode:r instanceof Error&&r.sessionControlErrorCode?r.sessionControlErrorCode:g.runtimeError;return u.error(this.name,`handleSessionControlLocalAction error verb=${i} session_id=${s}: ${r instanceof Error?r.message:r}`),a(d,r instanceof Error?r.message:String(r)),!0}}async handleEventCancel(e){const{event_id:o,session_id:t}=e;if(u.info(this.name,`handleEventCancel start event_id=${o} session_id=${t}`),this.pool.cancelEvent(o,t)){if(!(this.pool.getSlot(t)?.adapter?.getActiveEventIds().includes(o)??!1)){this.sendEventResultWithCleanup(o,"canceled","canceled"),this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!0,final_state:"canceled"});return}await this.waitForEventDone(o,t,15e3),this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!0,final_state:"canceled"}),this.pushQueueSnapshotForSession(t);return}this.aibotHandle.sendEventCancelResult({event_id:o,accepted:!1,reason:"event not found or not cancelable"})}waitForEventDone(e,o,t){return new Promise(s=>{const i=this.pool.getSlot(o);if(!i?.adapter){s();return}const n=setTimeout(()=>{i.adapter.removeListener("eventDone",a),s()},t),a=r=>{r===e&&(clearTimeout(n),i.adapter.removeListener("eventDone",a),s())};i.adapter.on("eventDone",a)})}handleAibotStop(e){if(u.info(this.name,`[stop-trace] handleAibotStop begin session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"} adapterType=${this.config.adapterType??"acp"}`),this.aibotHandle.sendEventStopAck({stop_id:e.stop_id,event_id:e.event_id,accepted:!0,updated_at:Date.now()}),this.pool.removeQueuedEvent(e.session_id,e.event_id)){u.info(this.name,`[stop-trace] handleAibotStop removed queued(not-running) event -> stopResult(stopped) session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"}`),this.discardEventTrackingState(e.event_id),this.pushQueueSnapshotForSession(e.session_id),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()});return}const o=this.pool.getSlot(e.session_id),t=o?.adapter?.getStatus().busy??!1,s=(this.config.adapterType??"acp")==="acp",i=o?.adapter?.getActiveEventIds,n=typeof i=="function"?i.call(o.adapter):[],a=n.length>0?n.includes(e.event_id):t,r=(s||(this.config.adapterType??"acp")==="codex"||(this.config.adapterType??"acp")==="claude")&&a;if(u.info(this.name,`[stop-trace] handleAibotStop decision session=${e.session_id} event=${e.event_id} slotExists=${!!o?.adapter} busy=${t} activeIds=[${n.join(",")}] stoppingActiveEvent=${a} killOnStop=${r}`),a&&this.sendCtrl.markEventStopped(e.event_id),o?.adapter&&t){const d=r?this.pool.drainQueuedForSession(e.session_id):[];let c=!1,l=null;const h=v=>{c||(c=!0,l&&(clearTimeout(l),l=null),o.adapter.removeListener("eventDone",m),u.info(this.name,`[stop-trace] handleAibotStop ${v} -> stopResult(stopped) session=${e.session_id} event=${e.event_id} stop_id=${e.stop_id||"-"} killOnStop=${r}`),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()}),r&&this.killAndResumeStopSlot(e.session_id,d))},m=v=>{v===e.event_id&&h("eventDone")};o.adapter.on("eventDone",m),l=setTimeout(()=>h("timeout"),At),this.pool.deliverStopEvent(e.event_id,e.session_id)}else u.info(this.name,`[stop-trace] handleAibotStop slot-not-busy -> immediate stopResult(stopped) session=${e.session_id} event=${e.event_id} slotExists=${!!o?.adapter}`),this.pool.deliverStopEvent(e.event_id,e.session_id),this.aibotHandle.sendEventStopResult({stop_id:e.stop_id,event_id:e.event_id,status:"stopped",updated_at:Date.now()});(this.config.adapterType??"acp")==="pi"&&this.aibotHandle.sendText({event_id:e.event_id,session_id:e.session_id,content:"/stop",msg_type:0})}async killAndResumeStopSlot(e,o){if(!this.stopped){u.info(this.name,`[stop-trace] killAndResumeStopSlot begin session=${e} siblings=${o.length} -> removeSlot (kill process group)`);try{await this.pool.removeSlot(e),u.info(this.name,`[stop-trace] killAndResumeStopSlot removeSlot done session=${e} (process killed) -> redeliver ${o.length} sibling(s)`)}catch(t){u.warn(this.name,`[acp-stop] removeSlot failed session=${e}: ${t instanceof Error?t.message:String(t)}`)}if(!this.stopped)for(const t of o){if(this.stopped)break;try{await this.pool.deliverInboundEvent(t)}catch(s){u.error(this.name,`[acp-stop] sibling redeliver failed event=${t.event_id} session=${e}: ${s instanceof Error?s.message:String(s)}`),this.sendEventResultWithCleanup(t.event_id,"failed",s instanceof Error?s.message:String(s))}}}}handleAibotRevoke(e){if(!e.event_id||!this.revokeHandler.checkAndTrack(e.event_id))return;const o=e.event_id,t=e.session_id;if(t&&this.pool.cancelEvent(o,t)){(this.pool.getSlot(t)?.adapter?.getActiveEventIds().includes(o)??!1)||this.sendEventResultWithCleanup(o,"canceled","revoked");return}if(this.deferredMgr.removeEvent(o)){this.aibotHandle.sendEventResult({event_id:o,status:"canceled",msg:"revoked",updated_at:Date.now()});return}this.pool.deliverStopEvent(o,t||void 0)}async handleConfigureGatewayProvider(e){const o=e.params??{},t=String(o.api_key??"").trim(),s=String(o.anthropic_base_url??"").trim(),i=String(o.openai_base_url??"").trim(),n=typeof o.provider_id=="string"&&o.provider_id.trim().toLowerCase()||void 0,a=typeof o.quota_base_url=="string"&&o.quota_base_url.trim()||void 0,r=typeof o.model=="string"&&o.model.trim()||void 0;let d=!1;if(!t){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_API_KEY",error_msg:"api_key is required"});return}const c=this.config.adapterType??"acp",l=c==="claude"||c==="codex";try{if(l){const h=c==="claude"?s:i;if(!h){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_TARGET_BASE_URL",error_msg:`${c==="claude"?"anthropic_base_url":"openai_base_url"} is required`});return}const m=Se();if(!m){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"PROXY_UNAVAILABLE",error_msg:"MITM proxy manager not initialized"});return}const v=be(c),f=we(c,h,t,{model:r});m.setRoute(f);for(const R of v)m.setHostDefaultRoute(R,f.routeKey);await m.setAgentRelayEnabled(this.name,!0,{relayHosts:v}),d=!await this.recycleAdaptersForRelayChange()}else{if(!this.providerConfigHandler){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"unsupported",error_code:"UNSUPPORTED_CLIENT_TYPE",error_msg:`client type "${c}" does not support Grix relay provider config`});return}await this.providerConfigHandler({...n?{provider_id:n}:{},base_url:i||void 0,...a?{quota_base_url:a}:{},api_key:t,model:r})}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{accepted:!0,restart_deferred:d}})}catch(h){u.error(this.name,`handleConfigureGatewayProvider failed: ${h}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"CONFIGURE_FAILED",error_msg:h instanceof Error?h.message:String(h)})}}async handleAibotLocalAction(e){const o=e.action_type??"",t=String((e.params??{}).session_id??""),s=String((e.params??{}).verb??"").trim().toLowerCase();u.debug(this.name,`local_action received action_type=${o} verb=${s||"-"} action_id=${e.action_id} session_id=${t}`);const i=(this.config.adapterType??"acp")==="claude";if(o===b.sessionControl&&s===p.exec&&await this.handleSessionControlLocalAction(e))return;if(o===b.sessionControl&&s===p.listSessions){await this.handleListSessionsLocalAction(e);return}if(o==="skill_upload"){await this.handleSkillUploadLocalAction(e);return}if(i&&(o===b.interactionReply||o==="exec_approve"||o==="exec_reject")&&(await this.pool.deliverLocalAction(e)).handled||i&&await this.handleSessionControlLocalAction(e))return;if(o===b.sessionControl){const c=this.config.adapterType??"acp",l=c==="codex",h=c==="pi",m=String((e.params??{}).verb??"").trim().toLowerCase();if(l&&m===p.open){await this.handleCodexSessionControlLocalActionOpen(e);return}if(c==="cursor"&&m===p.open){await this.handleCursorSessionControlLocalActionOpen(e);return}if(l&&m==="restart"){const S=this.bindingStore.get(t)?.cwd??"";await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"restarted",binding:{aibotSessionId:t,cwd:S,workerStatus:"ready"}}});return}if(h&&m===p.open){try{const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=await this.resolveCwdForBinding(S),E=String(_.agent_session_id??"").trim();this.ensureImportedAgentSession(E,w),this.bindingStore.set(t,w),this.setResolvedAgentSessionId(t,E),this.sessionBindings.set(t,w),await this.ensureSlotStarted(t).catch(I=>{u.warn("bridge",`pi ensureSlotStarted on local-action bind failed (non-fatal): ${I instanceof Error?I.message:String(I)}`)}),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:w}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,w)}})}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.invalidCwd,error_msg:_ instanceof Error?_.message:String(_)})}return}if(h&&m===p.restart){await this.handlePiSessionControlRestartLocalAction(e);return}if((c==="openhuman"||c==="opencode")&&m===p.open){try{const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=await this.resolveCwdForBinding(S),E=String(_.agent_session_id??"").trim();this.ensureImportedAgentSession(E,w),this.bindingStore.set(t,w),this.setResolvedAgentSessionId(t,E),this.sessionBindings.set(t,w),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,w)}})}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.invalidCwd,error_msg:_ instanceof Error?_.message:String(_)})}return}if(c==="codewhale"&&m===p.open){await this.handleCodeWhaleSessionControlLocalActionOpen(e);return}if(c==="acp"&&m===p.stop){const S=this.bindingStore.get(t)?.cwd??"";await this.pool.removeSlot(t).catch(()=>{}),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"stopped",binding:{aibotSessionId:t,cwd:S,workerStatus:"stopped"}}});return}try{if(m===p.open){const _=e.params??{},S=String(_.cwd??"").trim();if(!S){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.cwdRequired,error_msg:"session cwd is required"});return}const w=String(_.agent_session_id??"").trim();if(w){const E=await this.resolveCwdForBinding(S);this.ensureImportedAgentSession(w,E)}}await this.handleSessionControlLocalActionForPool(e)}catch(_){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:_?.sessionControlErrorCode??_?.cwdErrorCode??g.runtimeError,error_msg:_ instanceof Error?_.message:String(_)});return}if(m===p.open){const _=e.params??{},S=await this.resolveCwdForBinding(String(_.cwd??"").trim());this.setResolvedAgentSessionId(t,String(_.agent_session_id??"").trim()),c==="agy"&&(this.bindingStore.set(t,S),this.sessionBindings.set(t,S)),await this.deferredMgr.release(t,this.deferredCallbacks()),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"opened",binding:this.buildOpenedBindingResult(t,S)}}),(this.config.adapterType??"acp")==="agy"&&this.aibotHandle.sendUpdateBindingCard({session_id:t,worker_status:"ready",cwd:S,meta:this.buildAgyToolbarMeta(t)})}else Ze(e,this.sessionControlCtx(t),this.sessionControlSenders());return}if(o==="file_list"){const c=Date.now(),l=t?this.bindingStore.get(t)?.cwd:void 0,h=e.params??{},m=String(h.parent_id??"").trim(),v=Array.isArray(h.allowed_extensions)?h.allowed_extensions.filter(S=>typeof S=="string").map(S=>S.trim()).filter(S=>S.length>0):[];u.info("file-list-diag",`plugin << recv action_id=${e.action_id} session_id=${t} parent_id=${m||"<root>"} show_hidden=${!!h.show_hidden} ext_count=${v.length} bound_cwd=${l??"<none>"}`);const f=await nt({parent_id:m||null,session_id:t,show_hidden:!!h.show_hidden,allowed_extensions:v},{resolveCwd:()=>l??this.config.agent.cwd??process.cwd(),fallbackDir:X()}),R=Date.now()-c,_=f.result?.files?.length??0;u.info("file-list-diag",`plugin -> reply action_id=${e.action_id} status=${f.status} elapsed=${R}ms count=${_} current_path=${f.result?.current_path??""} error_code=${f.error_code??""} error_msg=${f.error_msg??""}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:f.status,...f.result?{result:{...f.result,machine_name:rt()}}:{},...f.error_code?{error_code:f.error_code}:{},...f.error_msg?{error_msg:f.error_msg}:{}});return}if(o==="create_folder"){const c=t?this.bindingStore.get(t)?.cwd:void 0,l=String((e.params??{}).parent_id??"").trim(),h=String((e.params??{}).name??"").trim(),m=await st({parent_id:l||null,name:h,session_id:t},{resolveCwd:()=>c??this.config.agent.cwd??process.cwd(),fallbackDir:X()});this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:m.status,...m.result?{result:m.result}:{},...m.error_code?{error_code:m.error_code}:{},...m.error_msg?{error_msg:m.error_msg}:{}});return}if(o===b.setModel&&(this.config.adapterType??"acp")==="agy"){await this.handleAgySetModel(e,t);return}if(o===b.getSessionUsage){await this.handleGetSessionUsage(e,t);return}if(o===b.getRateLimits){await this.handleGetRateLimits(e,t);return}const n=(this.config.adapterType??"acp")==="acp";if((i||n)&&o===b.threadCompact){await this.handleThreadCompact(e,t);return}if(o==="connector_rollback"){await this.handleConnectorRollback(e);return}if(o==="connector_upgrade_push"){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{accepted:!0}}),this.upgradeTrigger?.();return}if(o===b.getAgentGlobalConfig){const c=this.globalConfigStore?.get(this.name);this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{agentName:this.name,...c?{config:c}:{config:null}}});return}if(o==="configure_gateway_provider"){await this.handleConfigureGatewayProvider(e);return}const a=this.config.adapterType??"acp",r=(a==="codex"||a==="cursor"||a==="pi"||a==="openhuman"||a==="opencode"||a==="acp")&&!!t&&!!this.bindingStore.get(t)?.cwd,d=await this.pool.deliverLocalAction(e,{autoCreateSlot:r});if(d.handled){if(d.kind==="set_mode"){const c=String((e.params??{}).mode_id??"");c&&(a==="cursor"?this.bindingStore.setCursorModeId(t,c):a==="claude"?this.bindingStore.setModeId(t,c):L.has(a)||this.globalConfigStore?.set(this.name,{acpInitialMode:c}))}else if(d.kind==="set_model"){const c=String((e.params??{}).model_id??"");c&&(a==="cursor"?(this.bindingStore.setModelId(t,c),this.globalConfigStore?.set(this.name,{modelId:c})):a==="codex"?this.globalConfigStore?.set(this.name,{codexModelId:c}):L.has(a)||this.globalConfigStore?.set(this.name,{modelId:c}))}else if(d.kind==="set_reasoning_effort"){const c=String((e.params??{}).reasoning_effort??(e.params??{}).reasoning_eff??(e.params??{}).effort??"");c&&this.globalConfigStore?.set(this.name,{codexReasoningEffort:c})}else if(d.kind==="set_sandbox_mode"){const c=String((e.params??{}).sandbox_mode??(e.params??{}).sandboxMode??"");if(c){const l=c==="default"?void 0:c;this.globalConfigStore?.set(this.name,{codexSandboxMode:l})}}return}if((a==="codex"||a==="cursor"||a==="pi"||a==="openhuman"||a==="opencode")&&t&&!this.bindingStore.get(t)?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"Session binding missing. Open a workspace first."});return}if(a==="acp"&&(o==="set_mode"||o==="set_model")){const c=this.sessionControlSenders(),l=this.pool.getSlot(t)?.adapter,h={bindingStore:this.bindingStore,acpAdapter:l instanceof A?l:null,globalConfigStore:this.globalConfigStore,agentName:this.name,log:u};if(o==="set_mode"){const m=String((e.params??{}).mode_id??""),v=await Y(h,t,m);if(v.status==="failed")c.sendLocalActionResult(e.action_id,"failed",void 0,v.errorCode,v.errorMsg);else{const f=l instanceof A?l.buildToolbarContext(v.result?.outcome==="mode_set"?"mode_set":"mode_set_failed"):null,R=f?{...f,...v.result}:v.result;c.sendLocalActionResult(e.action_id,"ok",R)}}else{const m=String((e.params??{}).model_id??""),v=await J(h,t,m);if(v.status==="failed")c.sendLocalActionResult(e.action_id,"failed",void 0,v.errorCode,v.errorMsg);else{const f=l instanceof A?l.buildToolbarContext(v.result?.outcome==="model_set"?"model_set":"model_set_failed"):null,R=f?{...f,...v.result}:v.result;c.sendLocalActionResult(e.action_id,"ok",R)}}return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"unsupported_local_action",error_msg:`action type ${o} is not supported`})}async handleGetSessionUsage(e,o){if(!o){u.warn(this.name,`[usage] get_session_usage rejected: no session_id in params, action_id=${e.action_id}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for get_session_usage"});return}const t=this.config.adapterType??"acp",s=this.bindingStore.get(o),i=s?.cwd??this.config.agent.cwd??process.cwd();u.info(this.name,`[usage] get_session_usage action_id=${e.action_id} session_id=${o} adapterType=${t} hasBinding=${!!s} cwd=${i}`);let n=null,a,r;switch(t){case"claude":{if(a=s?.claudeSessionId,!a){u.warn(this.name,`[usage] no claude binding for session_id=${o}, action_id=${e.action_id}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Claude session binding found"});return}u.info(this.name,`[usage] parsing claude usage: claudeSessionId=${a} cwd=${i}`),n=await ge(a,i),r="claude";break}case"agy":{this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"agy",available:!1,reason:"agy print-mode adapter does not track token usage"}});return}case"acp":default:{if(a=s?.acpSessionId,!a){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No ACP session binding found"});return}n=await _e(a,i,this.config.aibot.clientType),r=t,(!r||r==="acp")&&(r="acp");break}case"codex":{if(a=s?.codexThreadId,!a){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Codex thread binding found"});return}n=await ke(a),r="codex";break}case"pi":{if(a=s?.piSessionPath,!a){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Pi session path binding found"});return}n=await Qe(a),r="pi";break}case"cursor":{const c=this.pool.getSlot(o)?.adapter;if(!(c instanceof M)){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No Cursor session found"});return}const l=c.getUsageSnapshot(o);if(!l){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"cursor",models:[{modelId:this.bindingStore.get(o)?.modelId??"auto",turns:l.turns,input:l.total.input,output:l.total.output,cacheRead:l.total.cacheRead,cacheWrite:l.total.cacheWrite}],total:{input:l.total.input,output:l.total.output,cacheRead:l.total.cacheRead,cacheWrite:l.total.cacheWrite},turns:l.turns,sampledAt:l.sampledAt}});return}case"codewhale":{const c=this.pool.getSlot(o)?.adapter;if(!(c instanceof N)){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"no_binding",error_msg:"No CodeWhale session found"});return}const l=c.getUsageSnapshot();if(!l){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:o,adapterType:"codewhale",models:[{modelId:this.bindingStore.get(o)?.modelId??"codewhale",turns:l.turns,input:l.total.input,output:l.total.output}],total:{input:l.total.input,output:l.total.output},turns:l.turns,sampledAt:l.sampledAt}});return}}if(!n){u.info(this.name,`[usage] no usage data found: session_id=${o} adapterSessionId=${a} adapterType=${r}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"usage_not_found",error_msg:"No usage data found for this session"});return}u.info(this.name,`[usage] result ok: session_id=${o} adapterSessionId=${a} turns=${n.turns} models=${n.models.length}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{sessionId:a,adapterType:r,models:n.models,total:n.total,turns:n.turns,sampledAt:new Date().toISOString()}})}async handleThreadCompact(e,o){if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"session_id_required",error_msg:"session_id is required for thread_compact"});return}if(!this.bindingStore.get(o)?.cwd){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.bindingMissing,error_msg:"session binding was not found"});return}try{await this.ensureSlotStarted(o);const i=this.pool.getSlot(o)?.adapter;if(!i?.execCommand){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:"Agent does not support command execution"});return}u.info(this.name,`thread_compact session_id=${o} action_id=${e.action_id}`);const n=await i.execCommand("compact","",o);n.status==="ok"?this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{outcome:"compacted",message:n.message,data:n.data}}):this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:n.message??"compact failed"})}catch(s){u.warn(this.name,`thread_compact error session_id=${o}: ${s instanceof Error?s.message:s}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:g.runtimeError,error_msg:s instanceof Error?s.message:String(s)})}}async handleGetRateLimits(e,o){const t=this.config.adapterType??"acp";if(this.config.aibot.clientType==="kiro"){const s=this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs),i=this.cachedAcpContextWindow;if(s&&this.cachedProviderQuota){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!0,cached:!0,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:i,tokenUsage:null,providerQuota:this.cachedProviderQuota}});return}try{const n=await O();this.cachedProviderQuota=n,this.cachedProviderQuotaSampledAtMs=Date.now(),u.info(this.name,`[rate-limits] kiro quota queried: success=${n.success}`+(n.balance?` balance=${n.balance.remaining} ${n.balance.unit}`:"")+(n.error?` error=${n.error}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!0,cached:!1,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:i,tokenUsage:null,providerQuota:n}})}catch(n){u.warn(this.name,`[rate-limits] kiro quota query failed: ${n instanceof Error?n.message:String(n)}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"acp",available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:i,tokenUsage:null}})}return}switch(t){case"codex":{const s=this.maybeQueryProviderQuota(),i=this.getFreshCodexGlobalRateLimitCache();if(i.hasData&&!this.isRateLimitsCacheFresh(this.cachedRateLimitsSampledAtMs)&&!this.isRateLimitsCacheFresh(this.cachedCodexUsageSampledAtMs)){const l=this.pool.getAllSlots().find(h=>h.state==="ready"&&h.adapter);if(l&&(u.info(this.name,`[rate-limits] codex cache stale, refreshing from slot: session=${l.sessionId}`),(await l.adapter.handleLocalAction?.(e))?.handled))return}if(i.hasData){i.rateLimits?u.info(this.name,`[rate-limits] codex cached: primary=${i.rateLimits.primary.usedPercent.toFixed(1)}% resetsAt=${i.rateLimits.primary.resetsAt} secondary=${i.rateLimits.secondary.usedPercent.toFixed(1)}% resetsAt=${i.rateLimits.secondary.resetsAt}`):u.info(this.name,`[rate-limits] codex cached context/token only: hasContext=${!!i.contextWindow} hasToken=${!!i.tokenUsage}`);const l=await s;this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"codex",available:i.hasData||!!l,cached:!0,sampledAt:i.sampledAt,rateLimits:i.rateLimits,contextWindow:i.contextWindow,tokenUsage:i.tokenUsage,providerQuota:l}});return}const a=this.pool.getAllSlots().find(l=>l.state==="ready"&&l.adapter);if(a&&(u.info(this.name,`[rate-limits] codex reuse existing slot: session=${a.sessionId}`),(await a.adapter.handleLocalAction?.(e))?.handled))return;const r=this.resolveRateLimitWakeSessionId(o,t);if(r){const l=await this.wakeRateLimitSlot(r,t);if(l?.adapter&&(await l.adapter.handleLocalAction?.(e))?.handled)return}const d=await s,c=!!d;u.info(this.name,`[rate-limits] codex no native data, providerQuota=${d?d.provider:"none"} available=${c}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"codex",available:c,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null,providerQuota:d}});return}case"claude":{const s=this.maybeQueryProviderQuota(),i=this.getFreshClaudeRateLimitState();if(i){const c=i.rateLimits,l=await s;u.info(this.name,`[rate-limits] claude global cached: sampledAt=${i.sampledAt} hasRateLimits=${!!c}`+(l?` providerQuota=${l.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!0,cached:!0,sampledAt:i.sampledAt,rateLimits:i.rateLimits??null,contextWindow:i.contextWindow,tokenUsage:null,providerQuota:l}});return}let n=this.pool.getAllSlots().find(c=>c.state==="ready"&&c.adapter instanceof P)??null;if(!n){const c=this.resolveRateLimitWakeSessionId(o,t);c&&(n=await this.wakeRateLimitSlot(c,t))}u.info(this.name,`[rate-limits] handleGetRateLimits: session_id=${o} adapterType=claude hasSlot=${!!n} hasAdapter=${!!n?.adapter}`);const a=n?.adapter,r=a instanceof P?a.getSessionState():null,d=await s;if(r){(r.rateLimits?.fiveHour||r.rateLimits?.sevenDay||r.contextWindow?.usedPercentage!=null)&&(this.cachedClaudeRateLimitState=r);const c=this.getFreshClaudeRateLimitState(),l=c?.rateLimits&&!r.rateLimits?{...r,rateLimits:c.rateLimits}:r,h=l.rateLimits;u.info(this.name,`[rate-limits] claude global state: sampledAt=${l.sampledAt} hasRateLimits=${!!h}`+(h?` fiveHour=${h.fiveHour?.usedPercentage??"n/a"}% resetsAt=${h.fiveHour?.resetsAt??"n/a"} sevenDay=${h.sevenDay?.usedPercentage??"n/a"}% resetsAt=${h.sevenDay?.resetsAt??"n/a"}`:"")+(c?.rateLimits&&!r.rateLimits?" source=live+cached-fallback":" source=live")+(d?` providerQuota=${d.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!0,cached:!1,sampledAt:l.sampledAt,rateLimits:l.rateLimits??null,contextWindow:l.contextWindow,tokenUsage:null,providerQuota:d}})}else u.info(this.name,`[rate-limits] claude no global state: hasAdapter=${!!a} adapterType=${a?.constructor?.name??"n/a"}`+(d?` providerQuota=${d.provider}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"claude",available:!!d,cached:!1,sampledAt:null,rateLimits:null,contextWindow:null,tokenUsage:null,providerQuota:d}});return}case"cursor":{const i=(o?this.pool.getSlot(o):null)?.adapter;if(i instanceof M){await i.refreshAccountRateLimits({force:!0});const a=i.getRateLimitsSnapshot(o);u.info(this.name,`[rate-limits] cursor: available=${a.available} monthly=${a.rateLimits?.fiveHour?.usedPercentage??"n/a"}% api=${a.rateLimits?.sevenDay?.usedPercentage??"n/a"}%`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:a});return}const n=await ce();u.info(this.name,`[rate-limits] cursor(no-slot): available=${n.available} monthly=${n.rateLimits?.fiveHour?.usedPercentage??"n/a"}% api=${n.rateLimits?.sevenDay?.usedPercentage??"n/a"}%`+(n.error?` error=${n.error}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:"cursor",available:n.available,cached:!1,sampledAt:Number.isFinite(Date.parse(n.sampledAt))?Date.parse(n.sampledAt):Date.now(),rateLimits:n.rateLimits,contextWindow:null,tokenUsage:null,...n.displayMessage?{displayMessage:n.displayMessage}:{},...n.error?{error:n.error}:{}}});return}default:{const s=this.isRateLimitsCacheFresh(this.cachedProviderQuotaSampledAtMs)&&!!this.cachedProviderQuota,i=await this.maybeQueryProviderQuota();if(!i){u.info(this.name,`[rate-limits] no provider quota available for adapterType=${t}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:t,available:!1,cached:!1,sampledAt:null,rateLimits:null,contextWindow:this.cachedAcpContextWindow??null,tokenUsage:null}});return}u.info(this.name,`[rate-limits] provider quota ${s?"cached":"queried"}: provider=${i.provider} success=${i.success}`+(i.tiers.length>0?` tiers=${i.tiers.map(n=>`${n.name}=${n.usedPercent}%`).join(",")}`:"")+(i.balance?` balance=${i.balance.remaining} ${i.balance.unit}`:"")+(i.error?` error=${i.error}`:"")),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{adapterType:t,available:!0,cached:s,sampledAt:this.cachedProviderQuotaSampledAtMs,rateLimits:null,contextWindow:this.cachedAcpContextWindow??null,tokenUsage:null,providerQuota:i}});return}}}resolveRateLimitWakeSessionId(e,o){const t=String(e??"").trim(),s=this.bindingStore.getMostRecentlyUpdatedSessionId({requireCwd:!0});return s?t&&t===s?t:o==="codex"||o==="claude"?(t&&t!==s&&u.info(this.name,`[rate-limits] ${o} remap wake session: requested=${t} use_recent=${s}`),s):t&&t!==s?(u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=not_recent_session recent=${s}`),null):t||null:(u.info(this.name,`[rate-limits] skip wake slot: adapterType=${o} reason=no_recent_binding`),null)}async wakeRateLimitSlot(e,o){const t=String(e??"").trim();if(!t)return null;if(!this.bindingStore.get(t)?.cwd)return u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=binding_cwd_missing`),null;try{const i=this.pool.getOrCreateSlot(t);if(!i)return u.info(this.name,`[rate-limits] skip wake slot: session=${t} adapterType=${o} reason=slot_unavailable`),null;if(i.startPromise){const n=o==="claude"?6e4:2e4;await Promise.race([i.startPromise,new Promise((a,r)=>setTimeout(()=>r(new Error(`wake rate-limits slot timeout (${n}ms)`)),n))])}return u.info(this.name,`[rate-limits] wake slot success: session=${t} adapterType=${o}`),this.pool.getSlot(t)??i}catch(i){return u.warn(this.name,`[rate-limits] wake slot failed: session=${t} adapterType=${o} err=${i instanceof Error?i.message:String(i)}`),null}}sessionControlCtx(e){const o=this.pool.getSlot(e),t=o?.adapter instanceof A?o.adapter:null,s=(this.config.adapterType??"acp")==="acp",i={bindingStore:this.bindingStore,acpAdapter:t,globalConfigStore:this.globalConfigStore,agentName:this.name,log:u};return{getCwd:()=>this.bindingStore.get(e)?.cwd??this.config.agent.cwd??process.cwd(),getSessionBindings:()=>{if(t)return t.getSessionBindings();const n=this.sessionBindings;if(e&&!n.has(e)){const a=this.bindingStore.get(e);a?.cwd&&n.set(e,a.cwd)}return n},getStatus:()=>this.getStatus(),isAcpAlive:!!t?.isAlive(),getAcpSessionOptions:()=>t?.acpSessionOptions??null,setMode:n=>t?t.setMode(n):Promise.resolve(!1),setModel:n=>t?t.setModel(n):Promise.resolve(!1),acpSetMode:s?(n,a)=>Y(i,n,a):void 0,acpSetModel:s?(n,a)=>J(i,n,a):void 0,getPendingApproval:n=>{const a=t?.pendingApprovalEntries.get(n);return a?{requestId:a}:void 0},deletePendingApproval:n=>t?.pendingApprovalEntries.delete(n)??!1,respondPermission:(n,a)=>(t&&t.respondToPermission(n,a),Promise.resolve()),onSessionBound:(n,a)=>{this.bindingStore.set(n,a)},onSessionUnbound:n=>{const r=this.bindingStore.get(n)?.cwd??"";this.bindingStore.delete(n),this.sessionBindings.delete(n),this.sessionProviderHints.delete(n),this.sessionProviderQuotas.delete(n),this.sessionProviderMeta.delete(n),this.claudeWorkerStatus.delete(n),this.deferredMgr.clearSession(n);for(const d of this.pool.drainQueuedForSession(n))this.discardEventTrackingState(d.event_id);r&&this.aibotHandle.sendUpdateBindingCard({session_id:n,worker_status:"stopped",cwd:r})},cancelActiveRun:()=>(this.config.adapterType??"acp")==="agy"&&o?.adapter instanceof K?(o.adapter.cancelCurrentRun(),Promise.resolve()):o?.adapter?.cancel("")??Promise.resolve(),onModeSet:n=>{const a=this.config.adapterType??"acp";L.has(a)||this.globalConfigStore?.set(this.name,{acpInitialMode:n})},onModelSet:n=>{const a=this.config.adapterType??"acp";L.has(a)||this.globalConfigStore?.set(this.name,{modelId:n})}}}sessionControlSenders(){return{sendEventAck:(e,o)=>this.aibotHandle.sendEventAck({event_id:e,session_id:o,received_at:Date.now()}),sendEventResult:(e,o,t)=>this.aibotHandle.sendEventResult({event_id:e,status:o,...t?.msg?{msg:t.msg}:{},...t?.code?{code:t.code}:{},updated_at:Date.now()}),sendLocalActionResult:(e,o,t,s,i)=>this.aibotHandle.sendLocalActionResult({action_id:e,status:o,...t?{result:t}:{},...s?{error_code:s}:{},...i?{error_msg:i}:{}})}}resolveBindingChannelKey(e){switch(e){case"claude":return"grix-claude";case"codex":return"codex";case"cursor":return"cursor";case"pi":return"pi";case"openhuman":return"openhuman";case"codewhale":return"codewhale";case"opencode":return"opencode";case"agy":return"acp";case"acp":return this.config.aibot.clientType==="qwen"?"qwen":"acp";default:return e}}finalizeThinking(e,o){this.sendCtrl.finalizeThinking(e,o)}logInboundConversation(e){const o=String(e.session_id??"").trim();o&&this.conversationLog?.logInbound(o,{event_id:e.event_id,msg_id:e.msg_id,sender_id:e.sender_id,msg_type:e.msg_type,content:e.content??""})}buildInboundEvent(e,o){const t=lt(this.sendCtrl.getGlobalRuntimeConfig(),o);return{event_id:e.event_id,session_id:e.session_id,thread_id:e.thread_id,sender_id:e.sender_id,msg_id:e.msg_id,msg_type:e.msg_type,content:e.content??"",quoted_message_id:e.quoted_message_id,context_messages_json:this.buildContextMessagesJson(e.context_messages),extra_json:e.extra?JSON.stringify(e.extra):void 0,connector_runtime_config:{response_delivery:t.responseDelivery,tool_events:t.toolEvents,thinking_events:t.thinkingEvents},session_type:e.session_type,created_at:e.created_at}}buildContextMessagesJson(e){if(!e)return;const o=e.filter(t=>!et(String(t?.content??"")));if(o.length!==0)return JSON.stringify(o)}isStaleEvent(e){const o=Number(e.created_at);return!Number.isFinite(o)||o<=0?!1:Date.now()-o>wt}async handleConnectorRollback(e){const o=String((e.params??{}).target_version??"").trim(),t=String((e.params??{}).reason??"server_initiated");if(!o){this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"MISSING_TARGET_VERSION",error_msg:"target_version is required for connector_rollback"});return}u.info(this.name,`connector_rollback: target=${o} reason=${t}`);try{const{npmInstall:s,writePending:i,removePending:n,upgradeLog:a}=await import("../core/upgrade/npm-upgrader.js"),{resolveClientVersion:r}=await import("../core/util/client-version.js"),d=r();a(`server rollback: ${d} -> ${o} reason=${t}`),i(d,o),await s("grix-connector",o),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"ok",result:{rolled_back_to:o}}),process.kill(process.pid,"SIGTERM")}catch(s){try{const{removePending:n}=await import("../core/upgrade/npm-upgrader.js");n()}catch{}const i=s instanceof Error?s.message:String(s);u.error(this.name,`connector_rollback failed: ${i}`),this.aibotHandle.sendLocalActionResult({action_id:e.action_id,status:"failed",error_code:"ROLLBACK_FAILED",error_msg:i})}}setUpgradeTrigger(e){this.upgradeTrigger=e}setAgentDeletedHandler(e){this.agentDeletedHandler=e}setSkillSyncHandler(e){this.skillSyncHandler=e}setShareSetHandler(e){this.shareSetHandler=e}setProviderConfigHandler(e){this.providerConfigHandler=e}isSharedInstance(){return!!this.config.aibot.sharedOwnerId}}export{Ti as AgentInstance};
|
|
@@ -0,0 +1,2 @@
|
|
|
1
|
+
import{splitTextForAibotProtocol as R}from"../core/protocol/protocol-text.js";const g=4*1024,L=8*1024,B=500,y=512;function F(){return T("GRIX_RAW_EVENT_EXTRA_MAX",g)}function b(){return T("GRIX_RAW_EVENT_FALLBACK_MAX",L)}function T(e,r){const l=process.env[e];if(l===void 0||l==="")return r;const t=Number(l);return Number.isFinite(t)&&t>0?Math.floor(t):r}function s(e){return Buffer.byteLength(e,"utf8")}const h=["tool_input","content","message","result"],v="fallback_text",I="__payload__";function u(e){if(typeof e=="string")return e;try{return JSON.stringify(e,null,2)??String(e)}catch{return String(e)}}function M(e,r){const l=e.slice(0,B),t=s(e)-s(l);return`${l}\u2026 [+${t}B, full in ${r}]`}function E(e,r){return s(JSON.stringify({channel_data:{[r]:{raw_event:e}},agent_api_origin:!0}))}function D(e){const r=e.extraMaxBytes??F(),l=e.fallbackMaxBytes??b(),{envelope:t,fallbackText:f,channelKey:_,allocateRefId:d}=e,a=[];let o=t.payload,x=f;if(s(f)>l){const c=d();a.push({field:v,fullText:f,refClientMsgId:c}),x=M(f,c)}if(E(t,_)>r){o={...t.payload},o.result!==void 0&&o.content!==void 0&&s(u(o.result))>y&&(delete o.result,o.result_omitted="see content");const c=h.filter(n=>o[n]!==void 0).map(n=>({field:n,size:s(u(o[n]))})).filter(n=>n.size>y).sort((n,i)=>i.size-n.size);for(const{field:n}of c){if(E({...t,payload:o},_)<=r)break;const i=u(o[n]),p=d();a.push({field:n,fullText:i,refClientMsgId:p}),o[n]=M(i,p),o[`${n}_ref`]=p,o[`${n}_full_len`]=i.length}if(E({...t,payload:o},_)>r){const n=u(t.payload),i=d();a.push({field:I,fullText:n,refClientMsgId:i}),o={payload_ref:i,payload_full_len:n.length,payload_oversized:!0}}}const A=a.length>0;return{sharded:A,envelope:A?{...t,payload:o}:t,fallbackText:x,oversizedFields:a}}function N(e){const{envelopeType:r,field:l,fullText:t}=e;return`[${r} detail] ${l} (${s(t)}B)
|
|
2
|
+
${t}`}export{g as DEFAULT_RAW_EVENT_EXTRA_MAX_BYTES,L as DEFAULT_RAW_EVENT_FALLBACK_MAX_BYTES,v as FALLBACK_FIELD_NAME,I as PAYLOAD_FIELD_NAME,B as RAW_EVENT_PREVIEW_CHARS,N as buildAuxiliaryDetailText,D as planRawEventDelivery,F as resolveRawEventExtraMaxBytes,b as resolveRawEventFallbackMaxBytes,R as splitTextForAibotProtocol,s as utf8Size};
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import{isRequestTimeoutError as C}from"../core/aibot/errors.js";import{log as h}from"../core/log/index.js";import{splitTextForAibotProtocol as
|
|
2
|
-
`:"");if(!r
|
|
3
|
-
`);const
|
|
1
|
+
import{isRequestTimeoutError as C}from"../core/aibot/errors.js";import{log as h}from"../core/log/index.js";import{splitTextForAibotProtocol as m,AIBOT_PROTOCOL_MAX_RUNES as k}from"../core/protocol/protocol-text.js";import{isThinkingEventDropped as E,isToolEventDropped as S,normalizeRuntimeConfigSnapshot as T}from"./runtime-config.js";const q={claude:!1,acp:!1,codex:!0,codewhale:!0,pi:!0,openhuman:!1,cursor:!1,opencode:!0,agy:!0};class d{handle;connectorRuntimeConfig;eventRuntimeConfigs=new Map;eventQuotedMessageIds=new Map;defaultQuoteTrigger=!1;thinkingSeqMap=new Map;toolCardSeq=0;bufferedEventTexts=new Map;chunkSeqMap=new Map;rejectedEvents=new Set;stoppedEvents=new Set;streamCoalesce=new Map;streamCoalesceMs=y();static STREAM_COALESCE_MAX_RUNES=2048;recentToolCards=new Map;static TOOL_CARD_DEDUP_MS=100;constructor(e){this.connectorRuntimeConfig={...e}}bind(e){this.handle=e}markEventRejected(e){if(!e)return;this.rejectedEvents.add(e);const t=`${e}:`;for(const s of[...this.streamCoalesce.keys()])if(s.startsWith(t)){const i=this.streamCoalesce.get(s);i.timer&&clearTimeout(i.timer),this.streamCoalesce.delete(s)}h.info("send-ctrl",`event marked as stream-rejected, future chunks will be dropped event=${e}`)}markEventStopped(e){e&&(this.stoppedEvents.add(e),h.info("send-ctrl",`[stop-trace] event marked stopped; subsequent outbound will be flagged event=${e}`))}flagOutboundAfterStop(e,t,s,i,r,o){!e||!this.stoppedEvents.has(e)||h.warn("send-ctrl",`[stop-trace] OUTBOUND AFTER STOP ${s} event=${e} session=${t} seq=${i} is_finish=${r} len=${o}`)}getGlobalRuntimeConfig(){return this.connectorRuntimeConfig}setStreamCoalesceMs(e){this.streamCoalesceMs=Number.isFinite(e)&&e>0?Math.floor(e):0}captureEventRuntimeConfig(e){const t=T(e.connector_runtime_config);if(t||!this.eventRuntimeConfigs.has(e.event_id)){const s=t??this.connectorRuntimeConfig;this.eventRuntimeConfigs.set(e.event_id,{responseDelivery:s.responseDelivery,toolEvents:s.toolEvents,thinkingEvents:s.thinkingEvents})}e.msg_id&&!this.eventQuotedMessageIds.has(e.event_id)&&this.eventQuotedMessageIds.set(e.event_id,e.msg_id)}discardEventState(e){e&&this.clearEventTrackingState(e)}clearEventTrackingState(e){this.eventRuntimeConfigs.delete(e),this.eventQuotedMessageIds.delete(e),this.rejectedEvents.delete(e),this.bufferedEventTexts.delete(e);for(const t of[...this.streamCoalesce.keys()])if(t.startsWith(`${e}:`)){const s=this.streamCoalesce.get(t);s.timer&&clearTimeout(s.timer),this.streamCoalesce.delete(t)}this.stoppedEvents.delete(e)&&h.info("send-ctrl",`[stop-trace] event tracking cleared stop flag event=${e}`);for(const t of this.chunkSeqMap.keys())t.startsWith(`${e}:`)&&this.chunkSeqMap.delete(t);for(const t of this.recentToolCards.keys())t.startsWith(`${e}:`)&&this.recentToolCards.delete(t)}setDefaultQuoteTrigger(e){this.defaultQuoteTrigger=e}getDefaultQuotedMessageId(e){if(!(!this.defaultQuoteTrigger||!e))return this.eventQuotedMessageIds.get(e)}resolveEventRuntimeConfig(e){if(e){const t=this.eventRuntimeConfigs.get(e);if(t)return t}return this.connectorRuntimeConfig}shouldDropToolDisplayEvent(e){return S(this.resolveEventRuntimeConfig(e))}shouldDropThinkingDisplayEvent(e){return E(this.resolveEventRuntimeConfig(e))}shouldDropCodexDisplayEvent(e,t){return this.shouldDropToolDisplayEvent(e)?t==="item/completed":!1}shouldDropAcpRawDisplayEvent(e,t){return!!(this.shouldDropToolDisplayEvent(e)&&(t==="tool_use"||t==="tool_result")||this.shouldDropThinkingDisplayEvent(e)&&t==="thinking")}sendReply(e,t,s,i,r,o){if(!o?.skipToolFilter&&this.shouldDropToolDisplayEvent(e)&&D(s,r)||e&&this.rejectedEvents.has(e))return;if(this.flagOutboundAfterStop(e,t,"reply",0,!1,(s||"").length),this.resolveEventRuntimeConfig(e).responseDelivery==="stream"&&!r){const n=m(s),u=e?`reply_stream_${e}`:`reply_stream_${Date.now()}`;if(n.length===0){this.handle.sendMsg({event_id:e,session_id:t,msg_type:1,content:s||" ",quoted_message_id:i||void 0});return}for(let l=0;l<n.length;l++){const f=l===n.length-1;this.sendStreamChunk(e,t,n[l],l+1,f,u,i)}return}this.handle.sendMsg({event_id:e,session_id:t,msg_type:1,content:s,quoted_message_id:i||void 0,...r?{extra:r}:{}})}sendStreamChunk(e,t,s,i,r,o,a){if(e&&this.rejectedEvents.has(e))return;if(a=a??this.getDefaultQuotedMessageId(e),this.flagOutboundAfterStop(e,t,"stream_chunk",i,r,(s||"").length),this.finalizeThinking(e,t),this.resolveEventRuntimeConfig(e).responseDelivery==="single_message"&&e){this.bufferStreamChunk(e,t,s,r,a);return}if(this.streamCoalesceMs<=0){this.emitStreamChunkParts(e,t,s,i,r,o,a);return}this.coalesceStreamChunk(e,t,s,r,o,a)}async sendFinalStreamChunkReliable(e,t,s,i){if(e&&this.rejectedEvents.has(e))return;if(this.finalizeThinking(e,t),this.resolveEventRuntimeConfig(e).responseDelivery==="single_message"&&e){this.flushCoalesceForEvent(e),this.bufferOnly(e,t,"",!0);return}const o=`${e}:${s??"default"}`;this.flushCoalesce(o,!1),this.streamCoalesce.delete(o);const a=(this.chunkSeqMap.get(o)??0)+1;this.chunkSeqMap.set(o,a);const n=this.getDefaultQuotedMessageId(e),u={event_id:e,session_id:t,delta_content:"",chunk_seq:a,is_finish:!0,...s?{client_msg_id:s}:{},...n?{quoted_message_id:n}:{}};this.flagOutboundAfterStop(e,t,"stream_chunk_final",a,!0,0);try{await this.handle.sendStreamChunkRequest(u,i)}catch(l){C(l)?(h.warn("send-ctrl",`final chunk ACK timed out, resend same seq=${a} event=${e}`),this.handle.sendStreamChunk(u)):h.warn("send-ctrl",`final chunk send failed (no resend) event=${e}: ${l}`)}}async deliverAuxiliaryLargeText(e,t,s,i){if(!s||e&&this.rejectedEvents.has(e))return!1;if(this.resolveEventRuntimeConfig(e).responseDelivery==="single_message"&&e){const a=this.getDefaultQuotedMessageId(e),n=m(s);for(let u=0;u<n.length;u++)this.flagOutboundAfterStop(e,t,"aux_detail_msg",u+1,!1,n[u].length),this.handle.sendMsg({event_id:e,session_id:t,msg_type:1,content:n[u],...u===0&&a?{quoted_message_id:a}:{},client_msg_id:n.length>1?`${i}_p${u+1}`:i});return!0}const o=m(s);for(const a of o)this.sendStreamChunk(e,t,a,0,!1,i);return await this.sendFinalStreamChunkReliable(e,t,i),!0}coalesceStreamChunk(e,t,s,i,r,o){if(!s&&!i)return;const a=`${e}:${r??"default"}`;let n=this.streamCoalesce.get(a);if(n||(n={eventId:e,sessionId:t,clientMsgId:r,quotedMessageId:o,text:"",flushedOnce:!1},this.streamCoalesce.set(a,n)),s&&(n.text+=s),n.sessionId=t,r!==void 0&&(n.clientMsgId=r),!n.quotedMessageId&&o&&(n.quotedMessageId=o),i){this.flushCoalesce(a,!0);return}if(!n.flushedOnce||n.text.length>=d.STREAM_COALESCE_MAX_RUNES){this.flushCoalesce(a,!1);return}n.timer||(n.timer=setTimeout(()=>this.flushCoalesce(a,!1),this.streamCoalesceMs))}flushCoalesce(e,t){const s=this.streamCoalesce.get(e);if(!s)return;s.timer&&(clearTimeout(s.timer),s.timer=void 0);const i=s.text;s.text="",s.flushedOnce=!0,t&&this.streamCoalesce.delete(e),!(!i&&!t)&&this.emitStreamChunkParts(s.eventId,s.sessionId,i,0,t,s.clientMsgId,s.quotedMessageId)}flushCoalesceForEvent(e){const t=`${e}:`;for(const s of[...this.streamCoalesce.keys()])s.startsWith(t)&&(this.flushCoalesce(s,!1),this.streamCoalesce.delete(s))}emitStreamChunkParts(e,t,s,i,r,o,a){const n=s||(r?`
|
|
2
|
+
`:"");if(!n&&!r)return;const u=m(n,k);u.length===0&&u.push(n||`
|
|
3
|
+
`);const l=`${e}:${o??"default"}`;for(let f=0;f<u.length;f++){const p=f===u.length-1,g=this.chunkSeqMap.get(l)??0,_=f===0&&i>g?i:g+1;this.chunkSeqMap.set(l,_),this.handle.sendStreamChunk({event_id:e,session_id:t,delta_content:u[f],chunk_seq:_,is_finish:p&&r,...o?{client_msg_id:o}:{},...a?{quoted_message_id:a}:{}})}}sendEventResult(e,t,s,i){this.finalizeThinking(e,""),this.flushCoalesceForEvent(e),this.flushBufferedStreamText(e),this.handle.sendEventResult({event_id:e,status:t,...s?{msg:s}:{},...i?{code:i}:{},updated_at:Date.now()}),this.clearEventTrackingState(e)}sendThinking(e,t,s){if(this.shouldDropThinkingDisplayEvent(e)||e&&this.rejectedEvents.has(e))return;this.flagOutboundAfterStop(e,t,"thinking",this.thinkingSeqMap.get(e)??0,!1,(s||"").length);const i=(this.thinkingSeqMap.get(e)??0)+1;this.thinkingSeqMap.set(e,i),this.handle.sendStreamChunk({event_id:e,session_id:t,delta_content:s,chunk_seq:i,is_finish:!1,client_msg_id:`${e}_thinking`,is_thinking:!0})}sendToolExecutionCard(e,t,s,i){if(this.shouldDropToolDisplayEvent(e)){h.info("send-ctrl",`[tool-card] DROP event=${e} session=${t} summary=${s.summaryText}`);return}const o=Date.now(),a=`${e}:${s.summaryText}:${s.detailText??""}`,n=this.recentToolCards.get(a);if(n!==void 0&&o-n<d.TOOL_CARD_DEDUP_MS){h.info("send-ctrl",`[tool-card] DEDUP event=${e} session=${t} summary=${s.summaryText}`);return}this.recentToolCards.set(a,o);const u={summary_text:s.summaryText};s.detailText&&(u.detail_text=s.detailText),h.info("send-ctrl",`[tool-card] SEND event=${e} session=${t} summary=${s.summaryText}`),this.handle.sendMsg({event_id:e,session_id:t,client_msg_id:i??`tool_${++this.toolCardSeq}_${Date.now()}`,msg_type:1,content:"",extra:{channel_data:{grix:{toolExecution:u}},agent_api_origin:!0}})}finalizeThinking(e,t){const s=this.thinkingSeqMap.get(e);s!==void 0&&(this.thinkingSeqMap.delete(e),t&&(e&&this.rejectedEvents.has(e)||this.handle.sendStreamChunk({event_id:e,session_id:t,delta_content:"",chunk_seq:s+1,is_finish:!0,client_msg_id:`${e}_thinking`,is_thinking:!0})))}bufferOnly(e,t,s,i,r){r=r??this.getDefaultQuotedMessageId(e),this.bufferStreamChunk(e,t,s,i,r)}bufferStreamChunk(e,t,s,i,r){const o=this.bufferedEventTexts.get(e);o?(s&&(o.text+=s),!o.quotedMessageId&&r&&(o.quotedMessageId=r)):this.bufferedEventTexts.set(e,{sessionId:t,text:s??"",quotedMessageId:r||void 0}),i&&this.flushBufferedStreamText(e)}flushBufferedStreamText(e){const t=this.bufferedEventTexts.get(e);t&&(this.bufferedEventTexts.delete(e),t.text&&this.handle.sendMsg({event_id:e,session_id:t.sessionId,msg_type:1,content:t.text,...t.quotedMessageId?{quoted_message_id:t.quotedMessageId}:{}}))}}function y(){const c=process.env.GRIX_STREAM_COALESCE_MS;if(c===void 0||c==="")return 500;const e=Number(c);return Number.isFinite(e)&&e>=0?Math.floor(e):500}function D(c,e){return e&&typeof e=="object"&&e.channel_data?.grix?.toolExecution?!0:/^\[tool\]/.test(c)||/^\[tool result\]/.test(c)}export{q as DEFAULT_QUOTE_TRIGGER_BY_ADAPTER,d as SendController};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
import{readJSONFile as c,writeJSONFileAtomic as g}from"../util/json-file.js";import{log as
|
|
1
|
+
import{readJSONFile as c,writeJSONFileAtomic as g}from"../util/json-file.js";import{log as I}from"../log/index.js";import{SESSION_MODE_IDS as s}from"../../adapter/claude/protocol-contract.js";const a=s.fullAuto;function n(r){const e=String(r??"").trim().toLowerCase();return e===s.approval||e===s.fullAuto?e:a}class x{bindings=new Map;filePath;writePromise=Promise.resolve();constructor(e){this.filePath=e??null}load(){if(!this.filePath)return;const e=c(this.filePath);if(Array.isArray(e)){this.bindings.clear();for(const d of e)d.aibotSessionId&&this.bindings.set(d.aibotSessionId,{...d,modeId:n(d.modeId)})}}set(e,d,t){const i=this.bindings.get(e),o=n(t?.modeId??i?.modeId);this.bindings.set(e,{aibotSessionId:e,cwd:d,acpSessionId:i?.acpSessionId,claudeSessionId:i?.claudeSessionId,codexThreadId:i?.codexThreadId,codexModelId:i?.codexModelId,codexModeId:i?.codexModeId,codewhaleThreadId:i?.codewhaleThreadId,agyConversationId:i?.agyConversationId,codexReasoningEffort:i?.codexReasoningEffort,codexServiceTier:i?.codexServiceTier,codexSandboxMode:i?.codexSandboxMode,piSessionPath:i?.piSessionPath,modeId:o,modelId:i?.modelId,cursorModeId:i?.cursorModeId,acpModelId:i?.acpModelId,acpModeId:i?.acpModeId,updatedAt:Date.now()}),this.scheduleWrite()}setAcpSessionId(e,d){this.updateBinding(e,{acpSessionId:d})}setClaudeSessionId(e,d){this.updateBinding(e,{claudeSessionId:d})}getClaudeSessionId(e){return this.bindings.get(e)?.claudeSessionId}setCodexThreadId(e,d){this.updateBinding(e,{codexThreadId:d})}getCodexThreadId(e){return this.bindings.get(e)?.codexThreadId}setAgyConversationId(e,d){this.updateBinding(e,{agyConversationId:d})}getAgyConversationId(e){return this.bindings.get(e)?.agyConversationId}setCodexContext(e,d){this.updateBinding(e,{codexModelId:d.modelId,codexModeId:d.modeId,codexReasoningEffort:d.reasoningEffort,codexServiceTier:d.serviceTier,codexSandboxMode:d.sandboxMode})}getCodexModelId(e){return this.bindings.get(e)?.codexModelId}setCodexModelId(e,d){const t=this.bindings.get(e);if(!t)return;const i=String(d??"").trim();!i||t.codexModelId===i||(this.bindings.set(e,{...t,codexModelId:i,updatedAt:Date.now()}),this.scheduleWrite())}getCodexModeId(e){return this.bindings.get(e)?.codexModeId}getCodexReasoningEffort(e){return this.bindings.get(e)?.codexReasoningEffort}getCodexServiceTier(e){return this.bindings.get(e)?.codexServiceTier}getCodexSandboxMode(e){return this.bindings.get(e)?.codexSandboxMode}setPiSessionPath(e,d){this.updateBinding(e,{piSessionPath:d})}getPiSessionPath(e){return this.bindings.get(e)?.piSessionPath}setCodeWhaleThreadId(e,d){this.updateBinding(e,{codewhaleThreadId:d})}getCodeWhaleThreadId(e){return this.bindings.get(e)?.codewhaleThreadId}setModeId(e,d){const t=this.bindings.get(e);if(!t)return;const i=n(d);t.modeId!==i&&(this.bindings.set(e,{...t,modeId:i,updatedAt:Date.now()}),this.scheduleWrite())}ensureModeId(e,d=a){const t=this.bindings.get(e);t&&(t.modeId||(this.bindings.set(e,{...t,modeId:n(d),updatedAt:Date.now()}),this.scheduleWrite()))}getAcpSessionId(e){return this.bindings.get(e)?.acpSessionId}getModeId(e){const d=this.bindings.get(e)?.modeId;return d?n(d):void 0}setModelId(e,d){const t=this.bindings.get(e);if(!t)return;const i=d.trim();t.modelId!==i&&(this.bindings.set(e,{...t,modelId:i,updatedAt:Date.now()}),this.scheduleWrite())}getModelId(e){return this.bindings.get(e)?.modelId}setCursorModeId(e,d){const t=this.bindings.get(e);if(!t)return;const i=String(d??"").trim().toLowerCase(),o=i===""?void 0:i;t.cursorModeId!==o&&(this.bindings.set(e,{...t,cursorModeId:o,updatedAt:Date.now()}),this.scheduleWrite())}getCursorModeId(e){return this.bindings.get(e)?.cursorModeId}setAcpModelId(e,d){const t=this.bindings.get(e);if(!t)return;const i=String(d??"").trim(),o=i===""?void 0:i;t.acpModelId!==o&&(this.bindings.set(e,{...t,acpModelId:o,updatedAt:Date.now()}),this.scheduleWrite())}getAcpModelId(e){return this.bindings.get(e)?.acpModelId}setAcpModeId(e,d){const t=this.bindings.get(e);if(!t)return;const i=String(d??"").trim(),o=i===""?void 0:i;t.acpModeId!==o&&(this.bindings.set(e,{...t,acpModeId:o,updatedAt:Date.now()}),this.scheduleWrite())}getAcpModeId(e){return this.bindings.get(e)?.acpModeId}get(e){return this.bindings.get(e)}getMostRecentlyUpdatedSessionId(e){const d=e?.requireCwd!==!1;let t,i=-1;for(const o of this.bindings.values())d&&!String(o.cwd??"").trim()||o.updatedAt>i&&(i=o.updatedAt,t=o.aibotSessionId);return t}delete(e){this.bindings.delete(e),this.scheduleWrite()}entries(){return this.bindings.entries()}async flush(){await this.writePromise}scheduleWrite(){this.filePath&&(this.writePromise=this.writePromise.then(()=>this.write()).catch(e=>{I.error("session-binding-store",`Persist failed: ${e instanceof Error?e.message:e}`)}))}async write(){if(!this.filePath)return;const e=Array.from(this.bindings.values());await g(this.filePath,e)}updateBinding(e,d){const t=this.bindings.get(e),i={aibotSessionId:e,acpSessionId:Object.prototype.hasOwnProperty.call(d,"acpSessionId")?d.acpSessionId:t?.acpSessionId,claudeSessionId:Object.prototype.hasOwnProperty.call(d,"claudeSessionId")?d.claudeSessionId:t?.claudeSessionId,codexThreadId:Object.prototype.hasOwnProperty.call(d,"codexThreadId")?d.codexThreadId:t?.codexThreadId,codexModelId:Object.prototype.hasOwnProperty.call(d,"codexModelId")?d.codexModelId:t?.codexModelId,codexModeId:Object.prototype.hasOwnProperty.call(d,"codexModeId")?d.codexModeId:t?.codexModeId,codewhaleThreadId:Object.prototype.hasOwnProperty.call(d,"codewhaleThreadId")?d.codewhaleThreadId:t?.codewhaleThreadId,agyConversationId:Object.prototype.hasOwnProperty.call(d,"agyConversationId")?d.agyConversationId:t?.agyConversationId,codexReasoningEffort:Object.prototype.hasOwnProperty.call(d,"codexReasoningEffort")?d.codexReasoningEffort:t?.codexReasoningEffort,codexServiceTier:Object.prototype.hasOwnProperty.call(d,"codexServiceTier")?d.codexServiceTier:t?.codexServiceTier,codexSandboxMode:Object.prototype.hasOwnProperty.call(d,"codexSandboxMode")?d.codexSandboxMode:t?.codexSandboxMode,piSessionPath:Object.prototype.hasOwnProperty.call(d,"piSessionPath")?d.piSessionPath:t?.piSessionPath,cwd:t?.cwd,modeId:n(t?.modeId),modelId:t?.modelId,cursorModeId:t?.cursorModeId,acpModelId:t?.acpModelId,acpModeId:t?.acpModeId,updatedAt:Date.now()};this.bindings.set(e,i),this.scheduleWrite()}}export{x as SessionBindingStore};
|
package/dist/manager.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
import{readFileSync as K,readdirSync as G,writeFileSync as z}from"node:fs";import{join as v}from"node:path";import{AgentInstance as T}from"./bridge/bridge.js";import{buildSharedInstanceConfig as J,diffSharedOwners as V,sharedInstanceKey as X,SharedOwnersCache as Y}from"./manager-share-config.js";import{isAgentDeletedError as D}from"./core/aibot/errors.js";import{GRIX_PATHS as w,log as l}from"./core/log/index.js";import{resolveClientVersion as Z}from"./core/util/client-version.js";import{UpgradeChecker as ee}from"./core/upgrade/upgrade-checker.js";import{SkillSyncer as te}from"./core/skill-sync/skill-syncer.js";import{AgentGlobalConfigStore as ne}from"./core/persistence/agent-global-config-store.js";import{resolveSkillScanMode as ae}from"./adapter/claude/skill-scanner.js";import{buildWireSkills as re}from"./core/skill-sync/sync-state.js";import{buildReportedSkills as se,logDefaultSkillsCheck as oe,cleanupProjectedSkills as P}from"./default-skills/index.js";import{supportsDirectProviderConfig as ie}from"./core/config/provider-env.js";import{resolveCopilotCommand as O}from"./core/runtime/copilot-resolve.js";import{resolveCursorCommand as q}from"./core/runtime/cursor-resolve.js";import{getCliVersion as ce,resolveCliPath as le}from"./core/util/cli-probe.js";import{AgentInstaller as de}from"./core/installer/installer.js";import{reportInstallFailure as pe}from"./core/observability/sentry.js";const me=8e3;function ue(){const t=O();return[{clientType:"openclaw",command:"openclaw"},{clientType:"claude",command:"claude"},{clientType:"codex",command:"codex"},{clientType:"qwen",command:"qwen"},{clientType:"hermes",command:"hermes"},{clientType:"reasonix",command:"reasonix"},{clientType:"codewhale",command:"codewhale"},{clientType:"opencode",command:"opencode"},{clientType:"cursor",command:q()},{clientType:"pi",command:"pi"},{clientType:"openhuman",command:"openhuman-core"},{clientType:"kiro",command:"kiro-cli"},{clientType:"kimi",command:"kimi"},{clientType:"copilot",command:t.command},{clientType:"agy",command:"agy"}]}async function ge(){return Promise.all(ue().map(async t=>{const e=await le(t.command);if(!e)return{client_type:t.clientType,command:t.command,installed:!1,path:null,version:null};const n=await ce(e,t.versionArgs??["--version"]);return{client_type:t.clientType,command:t.command,installed:!0,path:e,version:n.version,error:n.error}}))}function he(t){switch(t){case"claude":return{adapterType:"claude",command:"claude"};case"codex":return{adapterType:"codex",command:"codex",options:{sandboxMode:"danger-full-access"}};case"gemini":return{adapterType:"acp",command:"gemini",autoInjectArgs:{acp:!0},enableSessionBinding:!0};case"qwen":return{adapterType:"acp",command:"qwen",adapterHint:"qwen/base",autoInjectArgs:{acp:!0},enableSessionBinding:!0};case"pi":return{adapterType:"pi",command:"pi"};case"cursor":return{adapterType:"cursor",command:q()};case"reasonix":return{adapterType:"acp",command:"reasonix",args:["acp"],enableSessionBinding:!0};case"codewhale":return{adapterType:"codewhale",command:"codewhale",enableSessionBinding:!0};case"openhuman":return{adapterType:"openhuman",command:"openhuman-core",enableSessionBinding:!0};case"kiro":return{adapterType:"acp",command:"kiro-cli",args:["acp"],enableSessionBinding:!0};case"kimi":return{adapterType:"acp",command:"kimi",args:["acp"],options:{initial_mode:"yolo"},enableSessionBinding:!0};case"opencode":return{adapterType:"opencode",command:"opencode",args:["serve"],enableSessionBinding:!0};case"copilot":{const e=O();return{adapterType:"acp",command:e.command,args:[...e.prefixArgs,"--acp"],enableSessionBinding:!0}}case"agy":return{adapterType:"agy",command:"agy",enableSessionBinding:!0};case"hermes":return{adapterType:"acp",command:"hermes",args:["acp"],enableSessionBinding:!0};default:throw new Error(`Unsupported client_type: ${t}`)}}function H(t,e){return String(t??"").trim().toLowerCase().replace(/[^a-z0-9._-]+/g,"-").replace(/-+/g,"-").replace(/^-|-$/g,"")||String(e??"").trim().toLowerCase().replace(/[^a-z0-9._-]+/g,"-").replace(/-+/g,"-").replace(/^-|-$/g,"")||"default"}function fe(t,e){return v(w.data,`session-bindings-${H(t,e)}.json`)}function _e(t,e){return v(w.data,`active-events-${H(t,e)}.json`)}function ye(...t){const e=[],n=new Set;for(const r of t)for(const s of r??[]){const i=String(s??"").trim(),o=i.toLowerCase();!i||n.has(o)||(n.add(o),e.push(i))}return e.length>0?e:void 0}function we(t,e){const n={claude:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},codex:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},cursor:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},acp:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},pi:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},codewhale:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},openhuman:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},opencode:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},agy:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0}},r=n[t]??n.acp;return{maxConcurrent:e?.max_concurrent??r.maxConcurrent??1,maxQueued:e?.max_queued??r.maxQueued??3,queueTimeoutMs:e?.queue_timeout_ms??r.queueTimeoutMs??0,cancelableQueued:!0,cancelableRunning:!0}}function b(t,e={}){const n=Z(),r=String(t.client_type??"").trim().toLowerCase(),s=he(r),i=String(t.ws_url??"").trim(),o="get_session_usage",p="get_rate_limits",c="get_agent_global_config";if(!t.name?.trim())throw new Error("agent name is required");if(!i)throw new Error(`agent ${t.name}: ws_url is required`);if(!t.agent_id?.trim())throw new Error(`agent ${t.name}: agent_id is required`);if(!t.api_key?.trim())throw new Error(`agent ${t.name}: api_key is required`);const a=s.adapterType,g=a==="acp",m=r==="qwen",_={...s.options??{}},S=a==="codex"?{capabilities:["local_action_v1","agent_invoke"],localActions:["session_control","get_context","set_model","set_mode","set_reasoning_effort","set_service_tier","set_sandbox_mode","exec_approve","exec_reject","file_list","create_folder","turn_interrupt","permission_approve","permission_reject","thread_compact",o,p]}:null,d=a==="claude"?{localActions:["session_control","set_mode","set_model","claude_interaction_reply","exec_approve","exec_reject","file_list","create_folder","thread_compact",o,p]}:null,f=m?{capabilities:["stream_chunk","local_action_v1"],localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o],adapterHint:"qwen/base"}:null,h=a==="pi"?{adapterHint:"pi/base",capabilities:["local_action_v1"],localActions:["session_control","set_model","get_context","file_list","create_folder",o]}:null,u=a==="openhuman"?{adapterHint:"openhuman/base",capabilities:["local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,y=a==="cursor"?{adapterHint:"cursor/base",capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","set_mode","get_context","file_list","create_folder",o,p]}:null,A=a==="codewhale"?{capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,C=a==="opencode"?{adapterHint:"opencode/base",capabilities:["stream_chunk","local_action_v1"],localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o]}:null,$=a==="agy"?{adapterHint:"agy/base",capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,R=g&&!m?{localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o]}:null,B=r==="kiro"?{localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder","thread_compact",o,p]}:null,U=a==="codex"||a==="claude"||r==="gemini"?["session_control","set_model","set_mode"]:void 0,F=[o,p];g&&_.raw_transport===void 0&&(_.raw_transport=r==="gemini");const L=`${r}/base`,I=ae(a,r),Q=I==="kiro"?void 0:process.cwd(),E=se({mode:I,projectDir:Q}),M=E.filter(x=>x.source==="connector").map(x=>x.name);M.length>0&&!e.quiet&&l.info("manager",`[${t.name}] injecting connector skills: [${M.join(", ")}]`);const j=re(E,w.skills);let W=j.length>0?j:void 0;return{name:t.name,adapterType:a,aibot:{url:i,agentId:t.agent_id,apiKey:t.api_key,clientType:r,clientVersion:n,adapterHint:s.adapterHint??f?.adapterHint??h?.adapterHint??u?.adapterHint??y?.adapterHint??C?.adapterHint??$?.adapterHint??L,capabilities:S?.capabilities??A?.capabilities??h?.capabilities??u?.capabilities??y?.capabilities??C?.capabilities??$?.capabilities??f?.capabilities??["stream_chunk","local_action_v1","connector_upgrade"],localActions:ye(S?.localActions??A?.localActions??d?.localActions??h?.localActions??u?.localActions??y?.localActions??C?.localActions??$?.localActions??f?.localActions??B?.localActions??R?.localActions??["exec_approve","exec_reject"],U,F,["connector_rollback","connector_upgrade_push",c,"configure_gateway_provider","skill_upload"]),skills:W},agent:{command:s.command,args:[...s.args??[],...t.args??[]],env:t.env,clientType:r,...t.provider?{provider:{...t.provider.provider_id?.trim()?{providerId:t.provider.provider_id.trim()}:{},...t.provider.base_url?.trim()?{baseUrl:t.provider.base_url.trim()}:{},...t.provider.quota_base_url?.trim()?{quotaBaseUrl:t.provider.quota_base_url.trim()}:{},...t.provider.api_key?.trim()?{apiKey:t.provider.api_key.trim()}:{},...t.provider.model?.trim()?{model:t.provider.model.trim()}:{}}}:{}},adapterOptions:_,acpAuthMethod:_.auth_method,acpInitialMode:_.initial_mode,acpMcpTools:_.acp_mcp_tools,promptTimeoutMs:t.prompt_timeout_ms,bindingsPath:fe(t.name,t.agent_id),activeEventStorePath:_e(t.name,t.agent_id),...s.enableSessionBinding||g?{enableSessionBinding:!0}:{},...s.autoInjectArgs||r==="reasonix"&&t.provider?.model?.trim()?{autoInjectArgs:{...s.autoInjectArgs??{},...r==="reasonix"&&t.provider?.model?.trim()?{model:t.provider.model.trim()}:{}}}:{},poolMaxSize:t.pool?.maxSize,poolIdleTimeoutMs:t.pool?.idleTimeoutMs,eventQueue:we(a,t.event_queue),logDir:w.log,providerBaseUrl:t.provider_base_url?.trim()||void 0,providerApiKey:t.provider_api_key?.trim()||void 0}}function Se(){const t=process.env.GRIX_AGENT_STARTUP_WAIT_MS,e=Number(t);return Number.isFinite(e)&&e>=500?Math.floor(e):me}function N(t){const e=[],n=[],r=[],s=G(t).filter(i=>i.endsWith(".json")).sort();for(const i of s)try{const o=JSON.parse(K(v(t,i),"utf-8"));if(o===null||typeof o!="object"||Array.isArray(o)||!("agents"in o)){r.push(i);continue}if(!Array.isArray(o.agents)){n.push({file:i,error:"agents is not an array"});continue}if(o.agents.length===0){n.push({file:i,error:"agents array is empty"});continue}for(const p of o.agents)e.push({entry:p,file:i})}catch(o){n.push({file:i,error:o instanceof Error?o.message:String(o)})}return{entries:e,fileErrors:n,skipped:r}}function k(t){if(t===null||typeof t!="object")return JSON.stringify(t)??"null";if(Array.isArray(t))return`[${t.map(k).join(",")}]`;const e=t;return`{${Object.keys(e).sort().map(r=>`${JSON.stringify(r)}:${k(e[r])}`).join(",")}}`}function ve(t,e){return k(t)===k(e)}class Le{instances=[];configMap=new Map;rawConfigMap=new Map;reloadChain=Promise.resolve();sharedInstances=new Map;shareSyncChains=new Map;stopping=!1;deletedAgentCleanups=new Set;upgradeChecker=null;skillSyncer=null;globalConfigStore;configDir=w.config;sharedOwnersCache=new Y(w.data);installer=new de;async start(e){const n=e??w.config;this.configDir=n,l.info("manager",`Loading configs from ${n}`),oe(),P(),this.globalConfigStore=new ne(v(w.data,"agent-global-configs.json")),this.globalConfigStore.load();const{entries:r,fileErrors:s,skipped:i}=N(n);for(const a of s)l.error("manager",`Failed to load ${a.file}: ${a.error}`);for(const a of i)l.info("manager",`Skipped ${a}: not an agent config (no "agents" key)`);if(r.length===0&&s.length===0&&i.length===0)throw new Error(`No config files found in ${n}`);const o=[];let p=s.length;for(const{entry:a,file:g}of r)try{const m=b(a);o.push({config:m,entry:a,file:g}),l.info("manager",`Loaded ${m.name} (${m.adapterType??"acp"}) from ${g}`)}catch(m){const _=typeof a?.name=="string"?a.name:"<unknown>";l.error("manager",`Invalid agent config in ${g} (name=${_}): ${m}`),p++}let c=0;if(o.length>0){const a=Se();l.info("manager",`Starting ${o.length} agent(s), startup wait=${a}ms`);const g=()=>this.upgradeChecker?.triggerCheck(),m=d=>{this.instances=this.instances.filter(f=>f!==d)},_=o.map(({config:d,entry:f,file:h})=>{const u=new T(d,this.globalConfigStore);return u.setUpgradeTrigger(g),u.setShareSetHandler(y=>this.onShareSet(d,y)),u.setAgentDeletedHandler(()=>this.onAgentDeletedByPlatform(d.name)),u.setSkillSyncHandler(()=>this.skillSyncer?.triggerSync()),this.instances.push(u),this.configMap.set(d.name,d),this.rawConfigMap.set(d.name,{entry:f,file:h}),{config:d,instance:u,startPromise:u.start()}}),S=await Promise.all(_.map(async d=>{const f=await new Promise(h=>{let u=!1;const y=setTimeout(()=>{u||(u=!0,h({kind:"timeout"}))},a);d.startPromise.then(()=>{u||(u=!0,clearTimeout(y),h({kind:"started"}))}).catch(A=>{u||(u=!0,clearTimeout(y),h({kind:"failed",error:A}))})});return{task:d,outcome:f}}));for(const{task:d,outcome:f}of S){if(f.kind==="started"){this.restoreCachedSharedInstances(d.config);continue}if(f.kind==="failed"){if(D(f.error)){this.onAgentDeletedByPlatform(d.config.name);continue}m(d.instance),l.error("manager",`Failed to start ${d.config.name}: ${f.error}`);continue}c++,l.warn("manager",`Startup pending for ${d.config.name}, continue retrying in background`),d.startPromise.then(()=>{l.info("manager",`Delayed start succeeded: ${d.config.name}`),this.restoreCachedSharedInstances(d.config)}).catch(h=>{if(D(h)){this.onAgentDeletedByPlatform(d.config.name);return}m(d.instance),l.error("manager",`Delayed start failed: ${d.config.name}: ${h}`)})}if(this.instances.length>0){const d=Math.max(0,this.instances.length-c);l.info("manager",`${d}/${o.length} agent(s) running now`)}c>0&&l.warn("manager",`${c} agent(s) still connecting in background`)}if(this.instances.length===0&&o.length>0)throw new Error("All agent configurations failed to start");o.length>0&&(this.upgradeChecker=new ee(o.map(({config:a})=>({apiKey:a.aibot.apiKey,wsUrl:a.aibot.url})),()=>this.instances.some(a=>a.hasPendingOrBackgroundWork())||[...this.sharedInstances.values()].some(a=>a.hasPendingOrBackgroundWork())),await this.upgradeChecker.start()),o.length>0&&(this.skillSyncer=new te(o.map(({config:a})=>({apiKey:a.aibot.apiKey,wsUrl:a.aibot.url})),w.skills),await this.skillSyncer.start())}async stop(){l.info("manager","Stopping all agents..."),this.stopping=!0,this.upgradeChecker?.stop(),this.skillSyncer?.stop(),await Promise.allSettled([...this.shareSyncChains.values()]),this.shareSyncChains.clear();const e=[...this.sharedInstances.values()];this.sharedInstances.clear(),await Promise.allSettled([...this.instances.map(n=>n.stop()),...e.map(n=>n.stop())]),await this.globalConfigStore?.flush(),this.instances=[],P(),l.info("manager","All stopped")}onShareSet(e,n){this.sharedOwnersCache.save(e.name,n);const s=(this.shareSyncChains.get(e.name)??Promise.resolve()).catch(()=>{}).then(()=>this.syncSharedInstances(e,n));this.shareSyncChains.set(e.name,s)}async syncSharedInstances(e,n){if(this.stopping)return;const{toAdd:r,toRemove:s}=V(e.name,this.sharedInstances.keys(),n);for(const i of r){if(this.stopping)break;const o=X(e.name,i);try{const p=J(e,i),c=new T(p,this.globalConfigStore);this.sharedInstances.set(o,c),await c.start(),l.info("manager",`shared instance started: ${o}`)}catch(p){this.sharedInstances.delete(o);const c=p instanceof Error?p.message:String(p);/Auth failed/i.test(c)?(this.sharedOwnersCache.remove(e.name,i),l.warn("manager",`shared instance auth rejected, removed from cache: ${o}`)):l.error("manager",`start shared instance failed ${o}: ${p}`)}}for(const{key:i}of s){const o=this.sharedInstances.get(i);o&&(this.sharedInstances.delete(i),o.stop().catch(p=>l.error("manager",`stop shared instance failed ${i}: ${p}`)),l.info("manager",`shared instance stopped: ${i}`))}}restoreCachedSharedInstances(e){const n=this.sharedOwnersCache.load(e.name);n.length!==0&&(l.info("manager",`restoring ${n.length} cached shared instance(s) for ${e.name}`),this.onShareSet(e,n))}getAgentsStatus(){return this.instances.map(e=>e.getStatus())}async addAgent(e){await this.addAgentInternal(e,"agents.json"),this.persistAgentsConfig(),l.info("manager",`Added agent: ${e.name}`)}async addAgentInternal(e,n){const r=b(e);if(this.instances.some(s=>s.name===r.name))throw new Error(`Agent "${r.name}" already exists`);await this.startInstanceFromConfig(r,e,n)}async startInstanceFromConfig(e,n,r){const s=new T(e,this.globalConfigStore);s.setUpgradeTrigger(()=>this.upgradeChecker?.triggerCheck()),s.setShareSetHandler(i=>this.onShareSet(e,i)),s.setAgentDeletedHandler(()=>this.onAgentDeletedByPlatform(e.name)),s.setSkillSyncHandler(()=>this.skillSyncer?.triggerSync()),s.setProviderConfigHandler(i=>this.setAgentProvider(e.name,i)),s.setRelayEnvSettledHandler(()=>this.relayEnvSettledHandler?.()),await s.start(),this.instances.push(s),this.configMap.set(e.name,e),this.rawConfigMap.set(e.name,{entry:n,file:r}),this.relayEnvSettledHandler?.()}async removeAgent(e){if(!this.instances.some(n=>n.name===e))throw Object.assign(new Error(`Agent "${e}" not found`),{code:"NOT_FOUND"});await this.removeAgentInternal(e),this.persistAgentsConfig(),l.info("manager",`Removed agent: ${e}`),this.relayEnvSettledHandler?.()}async removeAgentInternal(e,n){const r=this.instances.findIndex(p=>p.name===e),s=r===-1?void 0:this.instances[r];r!==-1&&this.instances.splice(r,1),this.configMap.delete(e),this.rawConfigMap.delete(e);const i=this.shareSyncChains.get(e);this.shareSyncChains.delete(e),i&&await i.catch(()=>{});const o=`${e}#shared:`;for(const[p,c]of[...this.sharedInstances.entries()])p.startsWith(o)&&(this.sharedInstances.delete(p),c.stop().catch(a=>l.error("manager",`stop shared instance failed ${p}: ${a}`)));n?.keepShareCache||this.sharedOwnersCache.delete(e),s&&await s.stop()}onAgentDeletedByPlatform(e){this.stopping||this.deletedAgentCleanups.has(e)||(this.deletedAgentCleanups.add(e),(async()=>{try{if(!(this.rawConfigMap.has(e)||this.instances.some(r=>r.name===e)))return;l.warn("manager",`Agent "${e}" \u5DF2\u5728\u5E73\u53F0\u5220\u9664\uFF0C\u505C\u6B62\u5B9E\u4F8B\u5E76\u6E05\u7406\u672C\u5730\u914D\u7F6E`),await this.removeAgentInternal(e),this.persistAgentsConfig(),l.info("manager",`Agent "${e}" \u5DF2\u5728\u5E73\u53F0\u5220\u9664\uFF0C\u672C\u5730\u914D\u7F6E\u5DF2\u6E05\u7406\uFF08agents.json \u5DF2\u79FB\u9664\u8BE5\u6761\u76EE\uFF09`)}catch(n){l.error("manager",`cleanup for deleted agent "${e}" failed: ${n}`)}finally{this.deletedAgentCleanups.delete(e)}})())}persistAgentsConfig(){const e=v(this.configDir,"agents.json");try{const n=[];for(const[,s]of this.configMap){const i=this.rawConfigMap.get(s.name)?.entry.provider;n.push({name:s.name,ws_url:s.aibot.url,agent_id:s.aibot.agentId,api_key:s.aibot.apiKey,client_type:s.aibot.clientType,...i?{provider:i}:{}})}z(e,JSON.stringify({agents:n},null,4)+`
|
|
1
|
+
import{readFileSync as K,readdirSync as G,writeFileSync as z}from"node:fs";import{join as v}from"node:path";import{AgentInstance as T}from"./bridge/bridge.js";import{buildSharedInstanceConfig as J,diffSharedOwners as V,sharedInstanceKey as X,SharedOwnersCache as Y}from"./manager-share-config.js";import{isAgentDeletedError as D}from"./core/aibot/errors.js";import{GRIX_PATHS as w,log as l}from"./core/log/index.js";import{resolveClientVersion as Z}from"./core/util/client-version.js";import{UpgradeChecker as ee}from"./core/upgrade/upgrade-checker.js";import{SkillSyncer as te}from"./core/skill-sync/skill-syncer.js";import{AgentGlobalConfigStore as ne}from"./core/persistence/agent-global-config-store.js";import{resolveSkillScanMode as ae}from"./adapter/claude/skill-scanner.js";import{buildWireSkills as re}from"./core/skill-sync/sync-state.js";import{buildReportedSkills as se,logDefaultSkillsCheck as oe,cleanupProjectedSkills as P}from"./default-skills/index.js";import{supportsDirectProviderConfig as ie}from"./core/config/provider-env.js";import{resolveCopilotCommand as O}from"./core/runtime/copilot-resolve.js";import{resolveCursorCommand as q}from"./core/runtime/cursor-resolve.js";import{getCliVersion as ce,resolveCliPath as le}from"./core/util/cli-probe.js";import{AgentInstaller as de}from"./core/installer/installer.js";import{reportInstallFailure as pe}from"./core/observability/sentry.js";const me=8e3;function ue(){const t=O();return[{clientType:"openclaw",command:"openclaw"},{clientType:"claude",command:"claude"},{clientType:"codex",command:"codex"},{clientType:"qwen",command:"qwen"},{clientType:"hermes",command:"hermes"},{clientType:"reasonix",command:"reasonix"},{clientType:"codewhale",command:"codewhale"},{clientType:"opencode",command:"opencode"},{clientType:"cursor",command:q()},{clientType:"pi",command:"pi"},{clientType:"openhuman",command:"openhuman-core"},{clientType:"kiro",command:"kiro-cli"},{clientType:"kimi",command:"kimi"},{clientType:"copilot",command:t.command},{clientType:"agy",command:"agy"}]}async function ge(){return Promise.all(ue().map(async t=>{const e=await le(t.command);if(!e)return{client_type:t.clientType,command:t.command,installed:!1,path:null,version:null};const n=await ce(e,t.versionArgs??["--version"]);return{client_type:t.clientType,command:t.command,installed:!0,path:e,version:n.version,error:n.error}}))}function he(t){switch(t){case"claude":return{adapterType:"claude",command:"claude"};case"codex":return{adapterType:"codex",command:"codex",options:{sandboxMode:"danger-full-access"}};case"gemini":return{adapterType:"acp",command:"gemini",autoInjectArgs:{acp:!0},enableSessionBinding:!0};case"qwen":return{adapterType:"acp",command:"qwen",adapterHint:"qwen/base",autoInjectArgs:{acp:!0},enableSessionBinding:!0};case"pi":return{adapterType:"pi",command:"pi"};case"cursor":return{adapterType:"cursor",command:q()};case"reasonix":return{adapterType:"acp",command:"reasonix",args:["acp"],enableSessionBinding:!0};case"codewhale":return{adapterType:"codewhale",command:"codewhale",enableSessionBinding:!0};case"openhuman":return{adapterType:"openhuman",command:"openhuman-core",enableSessionBinding:!0};case"kiro":return{adapterType:"acp",command:"kiro-cli",args:["acp"],enableSessionBinding:!0};case"kimi":return{adapterType:"acp",command:"kimi",args:["acp"],options:{initial_mode:"yolo"},enableSessionBinding:!0};case"opencode":return{adapterType:"opencode",command:"opencode",args:["serve"],enableSessionBinding:!0};case"copilot":{const e=O();return{adapterType:"acp",command:e.command,args:[...e.prefixArgs,"--acp"],enableSessionBinding:!0}}case"agy":return{adapterType:"agy",command:"agy",enableSessionBinding:!0};case"hermes":return{adapterType:"acp",command:"hermes",args:["acp"],enableSessionBinding:!0};default:throw new Error(`Unsupported client_type: ${t}`)}}function H(t,e){return String(t??"").trim().toLowerCase().replace(/[^a-z0-9._-]+/g,"-").replace(/-+/g,"-").replace(/^-|-$/g,"")||String(e??"").trim().toLowerCase().replace(/[^a-z0-9._-]+/g,"-").replace(/-+/g,"-").replace(/^-|-$/g,"")||"default"}function fe(t,e){return v(w.data,`session-bindings-${H(t,e)}.json`)}function _e(t,e){return v(w.data,`active-events-${H(t,e)}.json`)}function ye(...t){const e=[],n=new Set;for(const r of t)for(const s of r??[]){const i=String(s??"").trim(),o=i.toLowerCase();!i||n.has(o)||(n.add(o),e.push(i))}return e.length>0?e:void 0}function we(t,e){const n={claude:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},codex:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},cursor:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},acp:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},pi:{maxConcurrent:1,maxQueued:5,queueTimeoutMs:0},codewhale:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},openhuman:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},opencode:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0},agy:{maxConcurrent:1,maxQueued:3,queueTimeoutMs:0}},r=n[t]??n.acp;return{maxConcurrent:e?.max_concurrent??r.maxConcurrent??1,maxQueued:e?.max_queued??r.maxQueued??3,queueTimeoutMs:e?.queue_timeout_ms??r.queueTimeoutMs??0,cancelableQueued:!0,cancelableRunning:!0}}function b(t,e={}){const n=Z(),r=String(t.client_type??"").trim().toLowerCase(),s=he(r),i=String(t.ws_url??"").trim(),o="get_session_usage",p="get_rate_limits",c="get_agent_global_config";if(!t.name?.trim())throw new Error("agent name is required");if(!i)throw new Error(`agent ${t.name}: ws_url is required`);if(!t.agent_id?.trim())throw new Error(`agent ${t.name}: agent_id is required`);if(!t.api_key?.trim())throw new Error(`agent ${t.name}: api_key is required`);const a=s.adapterType,g=a==="acp",m=r==="qwen",_={...s.options??{}},S=a==="codex"?{capabilities:["local_action_v1","agent_invoke"],localActions:["session_control","get_context","set_model","set_mode","set_reasoning_effort","set_service_tier","set_sandbox_mode","exec_approve","exec_reject","file_list","create_folder","turn_interrupt","permission_approve","permission_reject","thread_compact",o,p]}:null,d=a==="claude"?{localActions:["session_control","set_mode","set_model","claude_interaction_reply","exec_approve","exec_reject","file_list","create_folder","thread_compact",o,p]}:null,f=m?{capabilities:["stream_chunk","local_action_v1"],localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o],adapterHint:"qwen/base"}:null,h=a==="pi"?{adapterHint:"pi/base",capabilities:["local_action_v1"],localActions:["session_control","set_model","get_context","file_list","create_folder",o]}:null,u=a==="openhuman"?{adapterHint:"openhuman/base",capabilities:["local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,y=a==="cursor"?{adapterHint:"cursor/base",capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","set_mode","get_context","file_list","create_folder","thread_compact",p]}:null,A=a==="codewhale"?{capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,C=a==="opencode"?{adapterHint:"opencode/base",capabilities:["stream_chunk","local_action_v1"],localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o]}:null,$=a==="agy"?{adapterHint:"agy/base",capabilities:["stream_chunk","local_action_v1"],localActions:["session_control","set_model","file_list","create_folder",o]}:null,R=g&&!m?{localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder",o]}:null,B=r==="kiro"?{localActions:["exec_approve","exec_reject","permission_approve","permission_reject","session_control","set_model","set_mode","file_list","create_folder","thread_compact",o,p]}:null,U=a==="codex"||a==="claude"||r==="gemini"?["session_control","set_model","set_mode"]:void 0,F=a==="cursor"?[p]:[o,p];g&&_.raw_transport===void 0&&(_.raw_transport=r==="gemini");const L=`${r}/base`,I=ae(a,r),Q=I==="kiro"?void 0:process.cwd(),E=se({mode:I,projectDir:Q}),M=E.filter(x=>x.source==="connector").map(x=>x.name);M.length>0&&!e.quiet&&l.info("manager",`[${t.name}] injecting connector skills: [${M.join(", ")}]`);const j=re(E,w.skills);let W=j.length>0?j:void 0;return{name:t.name,adapterType:a,aibot:{url:i,agentId:t.agent_id,apiKey:t.api_key,clientType:r,clientVersion:n,adapterHint:s.adapterHint??f?.adapterHint??h?.adapterHint??u?.adapterHint??y?.adapterHint??C?.adapterHint??$?.adapterHint??L,capabilities:S?.capabilities??A?.capabilities??h?.capabilities??u?.capabilities??y?.capabilities??C?.capabilities??$?.capabilities??f?.capabilities??["stream_chunk","local_action_v1","connector_upgrade"],localActions:ye(S?.localActions??A?.localActions??d?.localActions??h?.localActions??u?.localActions??y?.localActions??C?.localActions??$?.localActions??f?.localActions??B?.localActions??R?.localActions??["exec_approve","exec_reject"],U,F,["connector_rollback","connector_upgrade_push",c,"configure_gateway_provider","skill_upload"]),skills:W},agent:{command:s.command,args:[...s.args??[],...t.args??[]],env:t.env,clientType:r,...t.provider?{provider:{...t.provider.provider_id?.trim()?{providerId:t.provider.provider_id.trim()}:{},...t.provider.base_url?.trim()?{baseUrl:t.provider.base_url.trim()}:{},...t.provider.quota_base_url?.trim()?{quotaBaseUrl:t.provider.quota_base_url.trim()}:{},...t.provider.api_key?.trim()?{apiKey:t.provider.api_key.trim()}:{},...t.provider.model?.trim()?{model:t.provider.model.trim()}:{}}}:{}},adapterOptions:_,acpAuthMethod:_.auth_method,acpInitialMode:_.initial_mode,acpMcpTools:_.acp_mcp_tools,promptTimeoutMs:t.prompt_timeout_ms,bindingsPath:fe(t.name,t.agent_id),activeEventStorePath:_e(t.name,t.agent_id),...s.enableSessionBinding||g?{enableSessionBinding:!0}:{},...s.autoInjectArgs||r==="reasonix"&&t.provider?.model?.trim()?{autoInjectArgs:{...s.autoInjectArgs??{},...r==="reasonix"&&t.provider?.model?.trim()?{model:t.provider.model.trim()}:{}}}:{},poolMaxSize:t.pool?.maxSize,poolIdleTimeoutMs:t.pool?.idleTimeoutMs,eventQueue:we(a,t.event_queue),logDir:w.log,providerBaseUrl:t.provider_base_url?.trim()||void 0,providerApiKey:t.provider_api_key?.trim()||void 0}}function Se(){const t=process.env.GRIX_AGENT_STARTUP_WAIT_MS,e=Number(t);return Number.isFinite(e)&&e>=500?Math.floor(e):me}function N(t){const e=[],n=[],r=[],s=G(t).filter(i=>i.endsWith(".json")).sort();for(const i of s)try{const o=JSON.parse(K(v(t,i),"utf-8"));if(o===null||typeof o!="object"||Array.isArray(o)||!("agents"in o)){r.push(i);continue}if(!Array.isArray(o.agents)){n.push({file:i,error:"agents is not an array"});continue}if(o.agents.length===0){n.push({file:i,error:"agents array is empty"});continue}for(const p of o.agents)e.push({entry:p,file:i})}catch(o){n.push({file:i,error:o instanceof Error?o.message:String(o)})}return{entries:e,fileErrors:n,skipped:r}}function k(t){if(t===null||typeof t!="object")return JSON.stringify(t)??"null";if(Array.isArray(t))return`[${t.map(k).join(",")}]`;const e=t;return`{${Object.keys(e).sort().map(r=>`${JSON.stringify(r)}:${k(e[r])}`).join(",")}}`}function ve(t,e){return k(t)===k(e)}class Le{instances=[];configMap=new Map;rawConfigMap=new Map;reloadChain=Promise.resolve();sharedInstances=new Map;shareSyncChains=new Map;stopping=!1;deletedAgentCleanups=new Set;upgradeChecker=null;skillSyncer=null;globalConfigStore;configDir=w.config;sharedOwnersCache=new Y(w.data);installer=new de;async start(e){const n=e??w.config;this.configDir=n,l.info("manager",`Loading configs from ${n}`),oe(),P(),this.globalConfigStore=new ne(v(w.data,"agent-global-configs.json")),this.globalConfigStore.load();const{entries:r,fileErrors:s,skipped:i}=N(n);for(const a of s)l.error("manager",`Failed to load ${a.file}: ${a.error}`);for(const a of i)l.info("manager",`Skipped ${a}: not an agent config (no "agents" key)`);if(r.length===0&&s.length===0&&i.length===0)throw new Error(`No config files found in ${n}`);const o=[];let p=s.length;for(const{entry:a,file:g}of r)try{const m=b(a);o.push({config:m,entry:a,file:g}),l.info("manager",`Loaded ${m.name} (${m.adapterType??"acp"}) from ${g}`)}catch(m){const _=typeof a?.name=="string"?a.name:"<unknown>";l.error("manager",`Invalid agent config in ${g} (name=${_}): ${m}`),p++}let c=0;if(o.length>0){const a=Se();l.info("manager",`Starting ${o.length} agent(s), startup wait=${a}ms`);const g=()=>this.upgradeChecker?.triggerCheck(),m=d=>{this.instances=this.instances.filter(f=>f!==d)},_=o.map(({config:d,entry:f,file:h})=>{const u=new T(d,this.globalConfigStore);return u.setUpgradeTrigger(g),u.setShareSetHandler(y=>this.onShareSet(d,y)),u.setAgentDeletedHandler(()=>this.onAgentDeletedByPlatform(d.name)),u.setSkillSyncHandler(()=>this.skillSyncer?.triggerSync()),this.instances.push(u),this.configMap.set(d.name,d),this.rawConfigMap.set(d.name,{entry:f,file:h}),{config:d,instance:u,startPromise:u.start()}}),S=await Promise.all(_.map(async d=>{const f=await new Promise(h=>{let u=!1;const y=setTimeout(()=>{u||(u=!0,h({kind:"timeout"}))},a);d.startPromise.then(()=>{u||(u=!0,clearTimeout(y),h({kind:"started"}))}).catch(A=>{u||(u=!0,clearTimeout(y),h({kind:"failed",error:A}))})});return{task:d,outcome:f}}));for(const{task:d,outcome:f}of S){if(f.kind==="started"){this.restoreCachedSharedInstances(d.config);continue}if(f.kind==="failed"){if(D(f.error)){this.onAgentDeletedByPlatform(d.config.name);continue}m(d.instance),l.error("manager",`Failed to start ${d.config.name}: ${f.error}`);continue}c++,l.warn("manager",`Startup pending for ${d.config.name}, continue retrying in background`),d.startPromise.then(()=>{l.info("manager",`Delayed start succeeded: ${d.config.name}`),this.restoreCachedSharedInstances(d.config)}).catch(h=>{if(D(h)){this.onAgentDeletedByPlatform(d.config.name);return}m(d.instance),l.error("manager",`Delayed start failed: ${d.config.name}: ${h}`)})}if(this.instances.length>0){const d=Math.max(0,this.instances.length-c);l.info("manager",`${d}/${o.length} agent(s) running now`)}c>0&&l.warn("manager",`${c} agent(s) still connecting in background`)}if(this.instances.length===0&&o.length>0)throw new Error("All agent configurations failed to start");o.length>0&&(this.upgradeChecker=new ee(o.map(({config:a})=>({apiKey:a.aibot.apiKey,wsUrl:a.aibot.url})),()=>this.instances.some(a=>a.hasPendingOrBackgroundWork())||[...this.sharedInstances.values()].some(a=>a.hasPendingOrBackgroundWork())),await this.upgradeChecker.start()),o.length>0&&(this.skillSyncer=new te(o.map(({config:a})=>({apiKey:a.aibot.apiKey,wsUrl:a.aibot.url})),w.skills),await this.skillSyncer.start())}async stop(){l.info("manager","Stopping all agents..."),this.stopping=!0,this.upgradeChecker?.stop(),this.skillSyncer?.stop(),await Promise.allSettled([...this.shareSyncChains.values()]),this.shareSyncChains.clear();const e=[...this.sharedInstances.values()];this.sharedInstances.clear(),await Promise.allSettled([...this.instances.map(n=>n.stop()),...e.map(n=>n.stop())]),await this.globalConfigStore?.flush(),this.instances=[],P(),l.info("manager","All stopped")}onShareSet(e,n){this.sharedOwnersCache.save(e.name,n);const s=(this.shareSyncChains.get(e.name)??Promise.resolve()).catch(()=>{}).then(()=>this.syncSharedInstances(e,n));this.shareSyncChains.set(e.name,s)}async syncSharedInstances(e,n){if(this.stopping)return;const{toAdd:r,toRemove:s}=V(e.name,this.sharedInstances.keys(),n);for(const i of r){if(this.stopping)break;const o=X(e.name,i);try{const p=J(e,i),c=new T(p,this.globalConfigStore);this.sharedInstances.set(o,c),await c.start(),l.info("manager",`shared instance started: ${o}`)}catch(p){this.sharedInstances.delete(o);const c=p instanceof Error?p.message:String(p);/Auth failed/i.test(c)?(this.sharedOwnersCache.remove(e.name,i),l.warn("manager",`shared instance auth rejected, removed from cache: ${o}`)):l.error("manager",`start shared instance failed ${o}: ${p}`)}}for(const{key:i}of s){const o=this.sharedInstances.get(i);o&&(this.sharedInstances.delete(i),o.stop().catch(p=>l.error("manager",`stop shared instance failed ${i}: ${p}`)),l.info("manager",`shared instance stopped: ${i}`))}}restoreCachedSharedInstances(e){const n=this.sharedOwnersCache.load(e.name);n.length!==0&&(l.info("manager",`restoring ${n.length} cached shared instance(s) for ${e.name}`),this.onShareSet(e,n))}getAgentsStatus(){return this.instances.map(e=>e.getStatus())}async addAgent(e){await this.addAgentInternal(e,"agents.json"),this.persistAgentsConfig(),l.info("manager",`Added agent: ${e.name}`)}async addAgentInternal(e,n){const r=b(e);if(this.instances.some(s=>s.name===r.name))throw new Error(`Agent "${r.name}" already exists`);await this.startInstanceFromConfig(r,e,n)}async startInstanceFromConfig(e,n,r){const s=new T(e,this.globalConfigStore);s.setUpgradeTrigger(()=>this.upgradeChecker?.triggerCheck()),s.setShareSetHandler(i=>this.onShareSet(e,i)),s.setAgentDeletedHandler(()=>this.onAgentDeletedByPlatform(e.name)),s.setSkillSyncHandler(()=>this.skillSyncer?.triggerSync()),s.setProviderConfigHandler(i=>this.setAgentProvider(e.name,i)),s.setRelayEnvSettledHandler(()=>this.relayEnvSettledHandler?.()),await s.start(),this.instances.push(s),this.configMap.set(e.name,e),this.rawConfigMap.set(e.name,{entry:n,file:r}),this.relayEnvSettledHandler?.()}async removeAgent(e){if(!this.instances.some(n=>n.name===e))throw Object.assign(new Error(`Agent "${e}" not found`),{code:"NOT_FOUND"});await this.removeAgentInternal(e),this.persistAgentsConfig(),l.info("manager",`Removed agent: ${e}`),this.relayEnvSettledHandler?.()}async removeAgentInternal(e,n){const r=this.instances.findIndex(p=>p.name===e),s=r===-1?void 0:this.instances[r];r!==-1&&this.instances.splice(r,1),this.configMap.delete(e),this.rawConfigMap.delete(e);const i=this.shareSyncChains.get(e);this.shareSyncChains.delete(e),i&&await i.catch(()=>{});const o=`${e}#shared:`;for(const[p,c]of[...this.sharedInstances.entries()])p.startsWith(o)&&(this.sharedInstances.delete(p),c.stop().catch(a=>l.error("manager",`stop shared instance failed ${p}: ${a}`)));n?.keepShareCache||this.sharedOwnersCache.delete(e),s&&await s.stop()}onAgentDeletedByPlatform(e){this.stopping||this.deletedAgentCleanups.has(e)||(this.deletedAgentCleanups.add(e),(async()=>{try{if(!(this.rawConfigMap.has(e)||this.instances.some(r=>r.name===e)))return;l.warn("manager",`Agent "${e}" \u5DF2\u5728\u5E73\u53F0\u5220\u9664\uFF0C\u505C\u6B62\u5B9E\u4F8B\u5E76\u6E05\u7406\u672C\u5730\u914D\u7F6E`),await this.removeAgentInternal(e),this.persistAgentsConfig(),l.info("manager",`Agent "${e}" \u5DF2\u5728\u5E73\u53F0\u5220\u9664\uFF0C\u672C\u5730\u914D\u7F6E\u5DF2\u6E05\u7406\uFF08agents.json \u5DF2\u79FB\u9664\u8BE5\u6761\u76EE\uFF09`)}catch(n){l.error("manager",`cleanup for deleted agent "${e}" failed: ${n}`)}finally{this.deletedAgentCleanups.delete(e)}})())}persistAgentsConfig(){const e=v(this.configDir,"agents.json");try{const n=[];for(const[,s]of this.configMap){const i=this.rawConfigMap.get(s.name)?.entry.provider;n.push({name:s.name,ws_url:s.aibot.url,agent_id:s.aibot.agentId,api_key:s.aibot.apiKey,client_type:s.aibot.clientType,...i?{provider:i}:{}})}z(e,JSON.stringify({agents:n},null,4)+`
|
|
2
2
|
`,"utf-8")}catch(n){l.error("manager",`Failed to persist agents config: ${n}`)}}isAgentBusy(e){return this.instances.find(r=>r.name===e)?.hasPendingWork()??!1}getAgentsWithStaleRelayEnv(){return this.instances.filter(e=>e.hasStaleRelayEnv()).map(e=>e.name)}markAgentRelayEnvStale(e){const n=this.instances.find(r=>r.name===e);return n?(n.markRelayEnvStale(),!0):!1}relayEnvSettledHandler;setRelayEnvSettledHandler(e){this.relayEnvSettledHandler=e}async restartAgent(e){const n=this.rawConfigMap.get(e);if(!n)throw Object.assign(new Error(`Agent "${e}" not found`),{code:"NOT_FOUND"});await this.replaceInstance(e,n.entry,n.file),l.info("manager",`Restarted agent: ${e}`)}async setAgentProvider(e,n){const r=this.rawConfigMap.get(e);if(!r)throw Object.assign(new Error(`Agent "${e}" not found`),{code:"NOT_FOUND"});const s={...r.entry,provider:n};await this.replaceInstance(e,s,r.file),this.persistAgentsConfig(),l.info("manager",`Updated provider config for agent: ${e}`)}async clearDirectProviderConfigs(){const e=[...this.rawConfigMap.entries()].filter(([,r])=>!!r.entry.provider&&ie(r.entry.client_type)).map(([r])=>r),n=[];for(const r of e)try{await this.setAgentProvider(r,void 0),n.push(r)}catch(s){l.warn("manager",`Failed to clear provider config for "${r}": ${s instanceof Error?s.message:String(s)}`)}return n}async replaceInstance(e,n,r){const s=b(n);await this.removeAgentInternal(e,{keepShareCache:!0}),await this.startInstanceFromConfig(s,n,r),this.restoreCachedSharedInstances(s)}async reload(){const e=this.reloadChain.catch(()=>{}).then(()=>this.doReload());return this.reloadChain=e.catch(()=>{}),e}async doReload(){if(this.stopping)throw Object.assign(new Error("manager is stopping"),{code:"RELOAD_UNSAFE"});const{entries:e,fileErrors:n,skipped:r}=N(this.configDir);for(const c of r)l.info("manager",`reload: skipped ${c}: not an agent config (no "agents" key)`);if(n.length>0){const c=n.map(a=>`${a.file}: ${a.error}`).join("; ");throw Object.assign(new Error(`reload aborted: ${n.length} config file(s) failed to parse [${c}]`),{code:"RELOAD_UNSAFE"})}if(e.length===0)throw Object.assign(new Error("reload aborted: no valid agent config found"),{code:"RELOAD_UNSAFE"});const s=[];for(const{entry:c}of e)try{b(c,{quiet:!0})}catch(a){const g=c?.name,m=typeof g=="string"&&g.trim()?g:"<unknown>";s.push({name:m,error:a instanceof Error?a.message:String(a)})}if(s.length>0){const c=s.map(a=>`${a.name}: ${a.error}`).join("; ");throw Object.assign(new Error(`reload aborted: ${s.length} agent config(s) invalid [${c}]`),{code:"RELOAD_UNSAFE"})}const i=new Map;for(const c of e){const a=String(c.entry?.name??"").trim();a&&(i.has(a)&&l.warn("manager",`reload: duplicate agent name "${a}", last one wins`),i.set(a,c))}const o={added:[],removed:[],restarted:[],unchanged:[],failed:[]},p=new Set(this.rawConfigMap.keys());for(const c of p)if(!i.has(c))try{await this.removeAgentInternal(c),o.removed.push(c)}catch(a){o.failed.push({name:c,error:a instanceof Error?a.message:String(a)})}for(const[c,a]of i){const g=this.rawConfigMap.get(c);if(g)if(ve(g.entry,a.entry))this.rawConfigMap.set(c,a),o.unchanged.push(c);else try{await this.replaceInstance(c,a.entry,a.file),o.restarted.push(c)}catch(m){o.failed.push({name:c,error:m instanceof Error?m.message:String(m)})}else try{await this.addAgentInternal(a.entry,a.file),o.added.push(c)}catch(m){o.failed.push({name:c,error:m instanceof Error?m.message:String(m)})}}return l.info("manager",`reload done: +${o.added.length} -${o.removed.length} ~${o.restarted.length} =${o.unchanged.length} !${o.failed.length}`),o}async checkUpgrade(){return this.upgradeChecker?this.upgradeChecker.checkForUpdate():{available:!1}}triggerUpgrade(){this.upgradeChecker?.triggerCheck()}async probeAll(e={}){return Ae(this.instances,e)}async probeOne(e,n={}){return be(this.instances,e,n)}listInstallable(){return this.installer.listInstallable()}async installAgent(e){const n=await this.installer.install(e);return pe(n),n}getInstallProgress(e){return this.installer.getProgress(e)??null}}async function Ae(t,e){const n=e.concurrency??4,r=Date.now(),s=new Array(t.length);await new Promise(a=>{let g=0,m=0;const _=t.length;if(_===0){a();return}function S(h){const u=t[h];u.probe(e).then(y=>{s[h]=y,d()},y=>{s[h]={agent_name:u.name,client_type:"unknown",adapter_type:"acp",ok:!1,status:"error",probed_at:Date.now(),duration_ms:0,cached:!1,cli:{command:"",installed:!1,path:null,version:null,error:{code:"internal",message:y?.message??String(y)}},conversation:{attempted:!1,ok:!1,latency_ms:null},config:{model:null,base_url:null,source:{model:"unknown",base_url:"unknown"}},process:{started:!1,alive:!1,busy:!1}},d()})}function d(){m++,g<_?S(g++):m===_&&a()}const f=Math.min(n,_);for(let h=0;h<f;h++)S(g++)});const i=s.filter(a=>a.status==="healthy").length,o=s.filter(a=>a.status==="degraded").length,p=s.filter(a=>a.status==="unavailable").length,c=await ge();return{ok:i===s.length&&s.length>0,total:s.length,healthy:i,degraded:o,unavailable:p,installed_clients:c,agents:s,probed_at:r,duration_ms:Date.now()-r}}async function be(t,e,n){const r=t.find(s=>s.name===e);if(!r)throw Object.assign(new Error(`Agent "${e}" not found`),{code:"NOT_FOUND"});return r.probe(n)}export{Le as Manager,ge as probeInstalledClientCommands,Ae as probeInstances,be as probeOneInstance};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
function a(
|
|
1
|
+
function a(o){const e=new Set([`http://127.0.0.1:${o.serverPort}`,`http://localhost:${o.serverPort}`,...o.allowedOrigins]),t=new Set([`127.0.0.1:${o.serverPort}`,`localhost:${o.serverPort}`,...o.allowedHosts]);return{validateRequest(s){const r=i(s,e);if(!r.ok)return r;const n=l(s,t);return n.ok?{ok:!0}:n}}}function i(o,e){const t=o.headers.origin;return t?e.has(t)?{ok:!0}:{ok:!1,statusCode:403,message:`Origin not allowed: ${t}`}:{ok:!0}}function l(o,e){const t=o.headers.host;return t?e.has(t)?{ok:!0}:{ok:!1,statusCode:403,message:`Host not allowed: ${t}`}:{ok:!1,statusCode:403,message:"Missing Host header"}}export{a as createSecurityPolicy};
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "grix-connector",
|
|
3
|
-
"version": "3.
|
|
3
|
+
"version": "3.15.0",
|
|
4
4
|
"description": "Connect local AI coding agents (Claude, Codex, Gemini, Qwen, DeepSeek, Cursor, OpenCode, Pi, OpenHuman, Reasonix) to the Grix scheduling platform. Also serves as an OpenClaw plugin for Grix channel transport.",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|