agent.ts 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325
  1. import { randomUUID, createHash } from 'node:crypto'
  2. import { DatabaseSync } from 'node:sqlite'
  3. import { realpath,access } from 'node:fs/promises'
  4. import { resolve, sep, join } from 'node:path'
  5. import {LocalVmImages} from './computers/localVm.js'
  6. import {LOCAL_VM_IMAGE_KEYS,MAX_VM_IDLE_STOP_MINUTES} from '../shared/localVm.js'
  7. import {ComputerError} from './computers/container.js'
  8. import {z} from 'zod'
  9. import {ComputerRuntime,ComputerGateway,type ComputerTarget} from './worker/gateway.js'
  10. import { HttpError } from '../server/errors.js'
  11. import { UpstreamClient } from '../server/upstream.js'
  12. import { UpstreamServiceSession } from '../server/localAuth.js'
  13. import { WorkspaceGateway, type GatewayTarget, type GatewayFrame } from '../server/workspaceGateway.js'
  14. import { createWorkspaceToolLease, type WorkspaceToolLease } from '../server/workspaceToolLease.js'
  15. import {UNCONFIGURED_COMPUTER_IMAGE} from '../shared/runner.js'
  16. import type { RunnerCommand, RunnerConfiguration } from '../shared/runner.js'
  17. import { LoopbackTransport, isLocalAuthorizationTarget } from '../server/loopbackAuthorization.js'
  18. const commands=new Set(['session.create','session.resume','session.close','session.usage','session.interrupt','session.cwd.set','prompt.submit','session.steer','image.attach_bytes','file.attach','approval.respond','clarify.respond'])
  19. interface Connection {cleanupOnly?:boolean;gateway:WorkspaceGateway|ComputerGateway;sessions:Map<string,string>;running:Set<string>;events:Promise<void>}
  20. interface Lease {connectionId:string;profile:string;lease:WorkspaceToolLease;controller:AbortController;session:{runtimeId:string;storedId:string}}
  21. export class RunnerAgent {
  22. readonly instance=randomUUID()
  23. private active=true
  24. private connected=false
  25. private disconnectRevision=0
  26. private disconnecting?:Promise<void>
  27. private connections=new Map<string,Connection>()
  28. private leases=new Map<string,Lease>()
  29. private closedConnections=new Map<string,number>()
  30. private closedLeases=new Map<string,number>()
  31. private inflight=new Map<string,Promise<unknown>>()
  32. private db:DatabaseSync
  33. private computers?:ComputerRuntime
  34. private localVm?:LocalVmImages
  35. private target:GatewayTarget
  36. private controlTransport=new LoopbackTransport()
  37. private serverEpoch?:string
  38. private controlAbort=new AbortController()
  39. constructor(readonly config:RunnerConfiguration,home:string,readonly fetchImpl:typeof fetch=fetch,readonly onStatus:(message:string)=>void=()=>{}) {
  40. const web=new URL(config.serverURL),hermes=new URL(config.hermesURL)
  41. if(web.username||web.password||web.search||web.hash||!['http:','https:'].includes(web.protocol))throw new Error('Web 地址无效')
  42. if(web.protocol==='http:'&&!['127.0.0.1','localhost','[::1]'].includes(web.hostname)&&!config.allowInsecureLan)throw new Error('远程执行节点必须使用 HTTPS,或明确配置可信局域网 HTTP')
  43. if(hermes.username||hermes.password||hermes.search||hermes.hash||!['127.0.0.1','[::1]'].includes(hermes.hostname)||!['http:','https:'].includes(hermes.protocol))throw new Error('Runner 必须连接本机 Hermes')
  44. const client=new UpstreamClient(hermes)
  45. this.target={url:hermes,client,session:new UpstreamServiceSession(client,()=>config.hermesCredentials)}
  46. this.db=new DatabaseSync(join(home,'runner-commands.sqlite3'))
  47. this.db.exec('CREATE TABLE IF NOT EXISTS commands(id TEXT PRIMARY KEY,fingerprint TEXT NOT NULL,state TEXT NOT NULL,result TEXT,created INTEGER NOT NULL)')
  48. if(config.computers){this.computers=new ComputerRuntime(this.db,config,home,undefined,(id,operation,body)=>this.api('desktop',{id,operation,...body}));this.localVm=new LocalVmImages(this.computers,this.db)}
  49. this.compactReceipts()
  50. }
  51. private async api(path:string,body?:unknown):Promise<any> {
  52. const url=new URL(`/api/runner/v1/${this.config.runnerId}/${path}`,this.config.serverURL)
  53. if(url.hostname==='localhost')url.hostname='127.0.0.1'
  54. const response=await (isLocalAuthorizationTarget(url)&&this.fetchImpl===fetch?this.controlTransport.fetch.bind(this.controlTransport):this.fetchImpl)(url,{
  55. method:body===undefined?'GET':'POST',redirect:'error',signal:AbortSignal.any([this.controlAbort.signal,AbortSignal.timeout(path==='desktop'?75000:25000)]),
  56. headers:{Authorization:`Bearer ${this.config.token}`,'x-runner-instance':this.instance,'x-runner-protocol':'1','x-runner-features':this.computers?'profile-computer-v1,host-computer-tools-v1,idle-stop-policy-v1,computer-worker-v1,artifact-chunks-v1,computer-control-v1,shared-computer-v1,local-vm-v1'+(this.computers.provider.fixedCapacity?',compose-desktops-v1':',helper-retirement-v1,image-options-v1')+(this.computers.config.imageId!==UNCONFIGURED_COMPUTER_IMAGE?',image-ready-v1':''):'',...(this.serverEpoch?{'x-runner-epoch':this.serverEpoch}:{}),'Content-Type':'application/json',...(path==='poll'&&!this.connected?{'x-runner-reset':'1'}:{})},
  57. ...(body===undefined?{}:{body:JSON.stringify(body)}),
  58. })
  59. if(!response.ok){
  60. if(path==='desktop'){const error=await response.json().catch(()=>({})) as {error?:string;code?:string};throw new HttpError(response.status,error.error?.slice(0,300)||'Compose 桌面连接失败',error.code||'compose_desktop_error')}
  61. throw new HttpError(response.status,`执行节点连接被拒绝(HTTP ${response.status})`,response.status===403?'runner_unauthorized':'runner_request_failed')
  62. }
  63. const reader=response.body?.getReader();const chunks:Uint8Array[]=[];let bytes=0
  64. if(!reader)throw new Error('执行节点响应为空')
  65. try{for(;;){const part=await reader.read();if(part.done)break;bytes+=part.value.length;if(bytes>40*1024*1024)throw new Error('执行节点响应超过大小限制');chunks.push(part.value)}}finally{await reader.cancel()}
  66. return JSON.parse(Buffer.concat(chunks).toString())
  67. }
  68. private requireProfile(profile:unknown):string {
  69. if(typeof profile!=='string'||!this.config.allowedProfiles.includes(profile))throw new HttpError(403,'Profile 未授权','runner_profile_forbidden')
  70. return profile
  71. }
  72. private async closeConnection(id:string,interrupt=true) {
  73. this.closedConnections.set(id,Date.now()+60000)
  74. const connection=this.connections.get(id)
  75. if(!connection)return
  76. this.connections.delete(id)
  77. for(const [leaseId,lease] of this.leases)if(lease.connectionId===id){this.closedLeases.set(leaseId,Date.now()+60000);lease.controller.abort();this.leases.delete(leaseId);void lease.lease.dispose()}
  78. if(interrupt){let timer:ReturnType<typeof setTimeout>|undefined;try{await Promise.race([Promise.all([...connection.running].map(session_id=>connection.gateway.rpc('session.interrupt',{session_id}).catch(()=>{}))),new Promise(resolve=>{timer=setTimeout(resolve,3000)})])}finally{clearTimeout(timer)}}
  79. await connection.gateway.close()
  80. }
  81. private disconnect():Promise<void> {
  82. if(this.disconnecting)return this.disconnecting
  83. this.disconnectRevision++;this.connected=false
  84. const leases=[...this.leases.values()];this.leases.clear()
  85. for(const value of leases)value.controller.abort()
  86. const pending=Promise.all([
  87. ...leases.map(value=>value.lease.dispose()),
  88. ...[...this.connections.keys()].map(id=>this.closeConnection(id)),
  89. ]).then(()=>{}).finally(()=>{if(this.disconnecting===pending)this.disconnecting=undefined})
  90. this.disconnecting=pending;return pending
  91. }
  92. private async execute(command:RunnerCommand):Promise<any> {
  93. const p=command.payload
  94. if(!this.active||!this.connected)throw new Error('执行节点已断开')
  95. if(command.kind==='local-vm.manage'){
  96. if(!this.localVm)throw new HttpError(409,'本地虚拟机尚未配置','computer_unavailable')
  97. const input=z.object({owner:z.string().regex(/^[a-f0-9]{64}$/),op:z.enum(['status','prepare','policy','idle-policy','instance']),idleStopMinutes:z.number().int().min(0).max(MAX_VM_IDLE_STOP_MINUTES).optional(),id:z.string().uuid().optional(),imageKey:z.enum(LOCAL_VM_IMAGE_KEYS).optional(),mode:z.enum(['shared','per-bot']).optional(),maxInstances:z.number().int().min(1).max(4).optional(),environmentId:z.string().uuid().optional(),action:z.enum(['stop','remove']).optional()}).strict().parse(p)
  98. const check=async()=>{if(!this.connected||!this.active||(await this.api('local-vm-check',{id:input.id})).allowed!==true)throw new ComputerError('local_vm_authorization_revoked','本地虚拟机配置授权已失效')}
  99. if(input.op==='status')return this.localVm.status(input.owner)
  100. if(input.op==='instance'){
  101. await check()
  102. const spec=this.computers!.pool.definition(input.owner,input.environmentId!)
  103. if(spec)await this.computers!.pool.desktop(spec,input.action!,()=>{if(!this.active||!this.connected)throw new Error('节点已断开')})
  104. await check();return {ok:true}
  105. }
  106. if(input.op==='prepare')return this.localVm.prepare(input.id!,check,input.imageKey)
  107. await check();if(input.op==='idle-policy')return this.localVm.idlePolicy(input.idleStopMinutes!)
  108. return this.localVm.policy(input.mode!,input.maxInstances!)
  109. }
  110. if(command.kind==='computer.control'){
  111. if(!this.computers)throw new HttpError(409,'电脑服务未配置','computer_unavailable')
  112. const input=z.object({target:z.object({environmentId:z.string().uuid(),agentId:z.string().uuid(),ownerKey:z.string().regex(/^[a-f0-9]{64}$/)}).strict(),profile:z.string(),op:z.enum(['status','detail','image','lifecycle','frame','take','input','giveback']),imageKey:z.enum(LOCAL_VM_IMAGE_KEYS).optional(),controlId:z.string().uuid().optional(),requestId:z.string().uuid().optional(),generation:z.number().int().optional(),frameId:z.string().uuid().optional(),action:z.unknown().optional(),notes:z.string().max(4000).optional()}).strict().parse(p)
  113. this.requireProfile(input.profile)
  114. const check=async()=>{if(!this.connected||!input.controlId||(await this.api('control-check',{controlId:input.controlId})).allowed!==true)throw new HttpError(403,'电脑控制授权已失效','computer_control_expired')}
  115. const authorize=()=>{if(!this.connected||!this.active||Date.now()>command.expiresAt)throw new HttpError(403,'电脑请求已失效','computer_control_expired')}
  116. await this.computers.ready
  117. if(input.op==='detail'){
  118. const vm=await this.localVm!.status(),spec=this.computers.pool.definition(input.target.ownerKey,input.target.environmentId)??(this.computers.provider.fixedCapacity?{id:input.target.environmentId,ownerKey:input.target.ownerKey,imageId:this.computers.config.imageId}:undefined)
  119. const state=await this.computers.controls.status(input.target)
  120. const actual=spec?await this.computers.provider.inspect(spec):undefined
  121. let ready=false
  122. if(actual?.running)try{await this.computers.provider.health(spec!,authorize);ready=true}catch{}
  123. const imageId=this.computers.imageFor(input.target),option=vm.images?.find(image=>image.imageId===imageId)
  124. let image=!!option?.ready
  125. if(!image&&!vm.fixedCapacity&&imageId!==UNCONFIGURED_COMPUTER_IMAGE)try{await this.localVm!.images.inspect(imageId);image=true}catch{}
  126. return {...state,container:actual?.running?'running':spec?'stopped':'missing',ready,inUse:['agent','pausing','human','resuming'].includes(state.mode),mode:vm.mode,maxInstances:vm.maxInstances,idleStopMinutes:vm.idleStopMinutes,image:vm.fixedCapacity?vm.image:image,imageId,images:vm.images,imageKey:option?.key,controlMode:state.mode,fixedCapacity:vm.fixedCapacity}
  127. }
  128. if(input.op==='image')return this.localVm!.select(input.target,z.enum(LOCAL_VM_IMAGE_KEYS).parse(input.imageKey),authorize)
  129. if(input.op==='lifecycle'){
  130. const action=z.enum(['create','start','stop','recreate','remove']).parse(input.action)
  131. await this.computers.desktop(input.target,input.profile,action,authorize)
  132. return {ok:true}
  133. }
  134. if(input.op==='status')return this.computers.controls.status(input.target)
  135. if(input.op==='frame')return this.computers.controls.frame(input.target,authorize)
  136. if(input.op==='take')return this.computers.controls.take(input.target,input.profile,input.controlId!,check)
  137. if(input.op==='input')return this.computers.controls.input(input.target,input.controlId!,input.requestId!,input.generation!,input.frameId!,input.action)
  138. return this.computers.controls.giveBack(input.target,input.controlId!,input.notes??'')
  139. }
  140. if(command.kind==='computer.retire'){
  141. const meta=p as unknown as ComputerTarget
  142. if(!this.computers||![meta.environmentId,meta.agentId].every(value=>/^[0-9a-f-]{36}$/.test(value))||!/^[a-f0-9]{64}$/.test(meta.ownerKey))throw new HttpError(400,'电脑清理请求无效','computer_cleanup_invalid')
  143. await this.computers.retire(meta);return {ok:true}
  144. }
  145. if(command.kind==='http') {
  146. const path=String(p.path),search=new URLSearchParams(String(p.search??''))
  147. if (path === '/api/plugins/yaoyao-bot-bridge/memory-extract') {
  148. if (p.method !== 'POST' || !this.computers) throw new HttpError(409, '执行节点尚不支持无工具记忆提炼', 'memory_extraction_unavailable')
  149. const body = z.object({ profile: z.string().min(1).max(256), prompt: z.string().min(1).max(65000) }).strict().parse(p.body)
  150. if (body.profile !== (search.get('profile') ?? 'default')) throw new HttpError(403, '提炼 Profile 与授权来源不一致', 'memory_profile_forbidden')
  151. this.requireProfile(body.profile)
  152. const text = await this.computers.extractMemory(body.profile, body.prompt, this.controlAbort.signal)
  153. return { status: 200, headers: {}, body: Buffer.from(JSON.stringify({ text })).toString('base64') }
  154. }
  155. if(path==='/api/computer/capabilities'||(p.computer&&!(p.computer as ComputerTarget).profileSession&&path==='/api/plugins/yaoyao-bot-bridge/capabilities')){
  156. this.requireProfile(search.get('profile')??'default')
  157. if(p.method!=='GET'||!this.computers)throw new HttpError(409,'执行节点未配置隔离电脑','computer_unavailable')
  158. await Promise.all([access(this.computers.config.python),access(this.computers.config.hermesSource),access(this.computers.script)])
  159. return {status:200,headers:{},body:Buffer.from(JSON.stringify({version:1,ready:true,native_tools:true,in_process:true,memory_isolation:true,memory_extraction:true,network:this.config.computers?.network??'none'})).toString('base64')}
  160. }
  161. if(p.computer&&path==='/api/files/download')throw new HttpError(409,'隔离电脑产物需通过产物导出接口读取','computer_artifact_required')
  162. if(p.computer&&/^\/api\/sessions\/[^/]+\/messages$/.test(path)){
  163. if(p.method!=='GET'||!this.computers)throw new HttpError(403,'电脑历史不可用','computer_unavailable')
  164. const profile=this.requireProfile(search.get('profile')??'default'),session=this.computers.session(decodeURIComponent(path.split('/')[3]!),p.computer as ComputerTarget,profile)
  165. if(session.vmExecution==='profile'){
  166. if(!session.profileSessionId)throw new HttpError(409,'本机 Profile 会话尚未建立','computer_profile_session_invalid')
  167. const response=await this.target.session.request(`/api/sessions/${encodeURIComponent(session.profileSessionId)}/messages`,{search,maxResponseBytes:8*1024*1024})
  168. return {status:response.status,headers:Object.fromEntries(response.headers.entries()),body:response.body.toString('base64')}
  169. }
  170. const offset=Math.max(0,Number(search.get('offset')??0)),history=session.history
  171. return {status:200,headers:{},body:Buffer.from(JSON.stringify({messages:history.slice(Math.max(0,history.length-offset-500),history.length-offset)})).toString('base64')}
  172. }
  173. if(p.method!=='GET'||!(/^\/api\/(?:status|profiles|config)$/.test(path)||/^\/api\/sessions\/[^/]+\/messages$/.test(path)||path==='/api/plugins/yaoyao-bot-bridge/capabilities'||path==='/api/files/download'))throw new HttpError(403,'Runner 不支持此 HTTP 操作','runner_http_forbidden')
  174. if(!['/api/status','/api/profiles'].includes(path))this.requireProfile(search.get('profile')??'default')
  175. if(path==='/api/files/download') {
  176. const requested=await realpath(String(search.get('path')??''))
  177. const roots=await Promise.all(this.config.artifactRoots.map(root=>realpath(root)))
  178. if(!roots.some(root=>requested===root||requested.startsWith(root+sep)))throw new HttpError(403,'文件不在允许的产物目录','runner_artifact_forbidden')
  179. }
  180. const response=await this.target.session.request(path,{search,maxResponseBytes:8*1024*1024})
  181. return {status:response.status,headers:Object.fromEntries(response.headers.entries()),body:response.body.toString('base64')}
  182. }
  183. const connectionId=String(p.connectionId??'')
  184. if(command.kind==='gateway.open') {
  185. if(this.closedConnections.has(connectionId)||this.connections.has(connectionId)||!/^[0-9a-f-]{36}$/.test(connectionId)||this.connections.size>=64)throw new Error('执行通道无效或已达上限')
  186. let gateway:WorkspaceGateway|ComputerGateway
  187. if(p.computer){
  188. const meta=p.computer as ComputerTarget
  189. if(!this.computers||![meta.environmentId,meta.agentId,String(p.workId)].every(value=>/^[0-9a-f-]{36}$/.test(value))||!/^[a-f0-9]{64}$/.test(meta.ownerKey))throw new HttpError(409,'电脑任务配置或授权无效','computer_unavailable')
  190. gateway=new ComputerGateway(this.computers,meta,String(p.workId),async()=>{if(!this.active||!this.connected||(await this.api('check',{connectionId})).allowed!==true)throw new HttpError(403,'电脑任务授权已失效','computer_authorization_revoked')},body=>this.api('artifact',{connectionId,...body}),meta.profileSession?this.target:undefined)
  191. }else gateway=new WorkspaceGateway(this.target)
  192. const connection:Connection={cleanupOnly:p.cleanupOnly===true,gateway,sessions:new Map(),running:new Set(),events:Promise.resolve()}
  193. this.connections.set(connectionId,connection)
  194. gateway.onEvent=(frame:GatewayFrame)=>{
  195. if(frame.session_id&&!connection.sessions.has(frame.session_id))return
  196. if(frame.session_id&&['message.complete','run.completed','run.failed'].includes(frame.type))connection.running.delete(frame.session_id)
  197. if(frame.type==='session.info'&&frame.session_id)for(const lease of this.leases.values())if(lease.session.runtimeId===frame.session_id&&typeof frame.payload?.stored_session_id==='string')lease.session.storedId=frame.payload.stored_session_id
  198. connection.events=connection.events.then(()=>this.api('event',{connectionId,frame})).then(()=>{},error=>{if(error instanceof HttpError&&error.status===410)void this.closeConnection(connectionId);else void this.disconnect()})
  199. }
  200. gateway.onDisconnect=()=>{void this.api('event',{connectionId,closed:true}).catch(()=>{})}
  201. try {await gateway.connect();if(!this.active||!this.connected||this.connections.get(connectionId)!==connection)throw new Error('执行节点已断开');return {ok:true}}catch(error){this.connections.delete(connectionId);gateway.close();throw error}
  202. }
  203. if(command.kind==='gateway.close'){await this.closeConnection(connectionId);return {ok:true}}
  204. if(command.kind==='gateway.rpc') {
  205. const connection=this.connections.get(connectionId),method=String(p.method),params=p.params as Record<string,unknown>
  206. if(!connection||!commands.has(method)||!params||typeof params!=='object')throw new HttpError(403,'执行通道或命令无效','runner_command_forbidden')
  207. if(connection.cleanupOnly&&!['session.resume','session.interrupt','session.close'].includes(method))throw new HttpError(403,'清理通道不允许执行新工作','runner_cleanup_forbidden')
  208. if(method==='session.create'||method==='session.resume'){if(!connection.cleanupOnly)this.requireProfile(params.profile)}
  209. else if(!connection.sessions.has(String(params.session_id)))throw new HttpError(403,'会话不属于这个通道','runner_session_forbidden')
  210. if(method==='prompt.submit')connection.running.add(String(params.session_id))
  211. const result=await connection.gateway.rpc(method,params)
  212. if(method==='session.create'||method==='session.resume'){
  213. if(this.connections.get(connectionId)!==connection)throw new HttpError(409,'执行通道已关闭','runner_command_uncertain')
  214. connection.sessions.set(String(result.session_id),String(params.profile))
  215. }
  216. if(method==='session.interrupt')connection.running.delete(String(params.session_id))
  217. return result
  218. }
  219. const leaseId=String(p.leaseId??'')
  220. if(command.kind==='lease.create') {
  221. this.requireProfile(p.profile)
  222. if(this.closedLeases.has(leaseId)||this.leases.has(leaseId)||!/^[0-9a-f-]{36}$/.test(leaseId)||this.leases.size>=64)throw new Error('工具授权无效或已达上限')
  223. const session=p.session as {runtimeId:string;storedId:string}
  224. const source=[...this.connections].find(([,c])=>c.sessions.get(session?.runtimeId)===p.profile)
  225. if(!source)throw new HttpError(403,'工具会话不属于此节点','runner_session_forbidden')
  226. const controller=new AbortController()
  227. const virtual=source[1].gateway
  228. if(virtual instanceof ComputerGateway){
  229. virtual.installTeamLease(leaseId,p.catalog as any[],async(toolId,args,callId)=>(await this.api('tool',{leaseId,toolId,arguments:args,callId})).result)
  230. const lease={bind:async()=>{},dispose:async()=>{virtual.removeTeamLease(leaseId)}}
  231. this.leases.set(leaseId,{lease,controller,session,connectionId:source[0],profile:String(p.profile)});return {ok:true}
  232. }
  233. const lease=await createWorkspaceToolLease({target:this.target,profile:String(p.profile),workId:String(p.workId),session:()=>session,signal:controller.signal,
  234. assertActive:()=>{if(!this.connected||controller.signal.aborted)throw new Error('工具授权已失效')},
  235. catalog:()=>p.catalog as any[],call:async(toolId,args,callId)=>(await this.api('tool',{leaseId,toolId,arguments:args,callId:callId??randomUUID()})).result,
  236. onFailure:()=>{controller.abort()},
  237. })
  238. if(!this.active||!this.connected||this.closedLeases.has(leaseId)||!this.connections.has(source[0])){controller.abort();await lease.dispose();throw new Error('工具授权已停止')}
  239. this.leases.set(leaseId,{lease,controller,session,connectionId:source[0],profile:String(p.profile)});return {ok:true}
  240. }
  241. if(command.kind==='lease.bind') {
  242. const lease=this.leases.get(leaseId)
  243. if(!lease)throw new HttpError(410,'工具授权已关闭','runner_lease_expired')
  244. const update=p.session as {runtimeId?:string;storedId?:string}|undefined
  245. if(update?.runtimeId!==lease.session.runtimeId||!this.connections.get(lease.connectionId)?.sessions.has(update.runtimeId))throw new HttpError(403,'工具不能改绑其他会话','runner_session_forbidden')
  246. if(typeof update.storedId!=='string')throw new Error('存储会话编号无效')
  247. lease.session.storedId=update.storedId;await lease.lease.bind();return {ok:true}
  248. }
  249. if(command.kind==='lease.close') {
  250. this.closedLeases.set(leaseId,Date.now()+60000)
  251. const lease=this.leases.get(leaseId);this.leases.delete(leaseId)
  252. if(lease){lease.controller.abort();await lease.lease.dispose()}return {ok:true}
  253. }
  254. throw new HttpError(400,'未知执行命令','runner_command_unknown')
  255. }
  256. private compactReceipts(){
  257. // Keep fingerprints as replay tombstones; large HTTP/file replies expire.
  258. this.db.prepare("UPDATE commands SET state='retired',result=NULL WHERE state='complete' AND created<?").run(Date.now()-3600000)
  259. }
  260. private handled=0
  261. async handle(command:RunnerCommand):Promise<unknown> {
  262. if(++this.handled%256===0)this.compactReceipts()
  263. for(const map of [this.closedConnections,this.closedLeases])for(const [id,expires] of map)if(expires<Date.now())map.delete(id)
  264. const existing=this.inflight.get(command.id)
  265. const fingerprint=createHash('sha256').update(JSON.stringify(command)).digest('hex')
  266. const previous=this.db.prepare('SELECT * FROM commands WHERE id=?').get(command.id)
  267. if(previous){
  268. if(previous.fingerprint!==fingerprint)throw new HttpError(409,'命令编号已用于其他内容','idempotency_conflict')
  269. if(existing)return existing
  270. if(previous.state==='complete')return JSON.parse(String(previous.result))
  271. throw new HttpError(409,'原命令结果不确定,禁止重复执行','runner_command_uncertain')
  272. }
  273. this.db.prepare("INSERT INTO commands VALUES(?,?,'started',NULL,?)").run(command.id,fingerprint,Date.now())
  274. const pending=Promise.resolve().then(async()=>{
  275. if(command.expiresAt<=Date.now()||!(await this.api('admit',{id:command.id})).allowed||command.expiresAt<=Date.now())throw new HttpError(403,'命令授权已经失效,未执行','runner_command_not_admitted')
  276. return this.execute(command)
  277. }).then(result=>({result}),error=>({error:{message:error instanceof HttpError||error instanceof ComputerError?error.message:'执行节点操作失败',code:error instanceof HttpError||error instanceof ComputerError?error.code:'runner_error'}}))
  278. .then(reply=>{this.db.prepare("UPDATE commands SET state='complete',result=? WHERE id=?").run(JSON.stringify(reply),command.id);return reply})
  279. .finally(()=>this.inflight.delete(command.id))
  280. this.inflight.set(command.id,pending);return pending
  281. }
  282. private lastStatus=''
  283. private status(message:string){if(message!==this.lastStatus){this.lastStatus=message;this.onStatus(message)}}
  284. async run(signal:AbortSignal):Promise<void> {
  285. const deliveries=new Map<string,Promise<void>>()
  286. const cleanupTimer=setInterval(()=>{void this.computers?.pool.expire().catch(()=>{})},5000);cleanupTimer.unref()
  287. const abort=()=>{this.active=false;this.controlAbort.abort();void this.disconnect()};signal.addEventListener('abort',abort,{once:true})
  288. try {
  289. while(this.active&&!signal.aborted) {
  290. try {
  291. const revision=this.disconnectRevision
  292. const response=await this.api('poll')
  293. await this.disconnecting
  294. if(revision!==this.disconnectRevision)continue
  295. if(this.serverEpoch&&this.serverEpoch!==response.epoch)await this.disconnect()
  296. this.serverEpoch=response.epoch;this.connected=true;this.status('执行节点已连接')
  297. for(const command of response.commands as RunnerCommand[]){
  298. if(deliveries.has(command.id))continue
  299. const delivery=(async()=>{
  300. let reply
  301. try {reply=await this.handle(command)}catch(error){reply={error:{message:'原命令需要核对',code:error instanceof HttpError?error.code:'runner_error'}}}
  302. await this.api('result',{id:command.id,...reply as object})
  303. })().catch(()=>{}).finally(()=>deliveries.delete(command.id))
  304. deliveries.set(command.id,delivery)
  305. }
  306. }catch(error){
  307. this.status(error instanceof HttpError?error.message:'连接暂时中断,正在重连')
  308. await this.disconnect()
  309. if(error instanceof HttpError&&error.code==='runner_unauthorized')throw error
  310. if(this.active)await new Promise(resolve=>setTimeout(resolve,1000))
  311. }
  312. }
  313. }finally{
  314. clearInterval(cleanupTimer)
  315. signal.removeEventListener('abort',abort)
  316. try{
  317. await this.disconnect().catch(()=>{})
  318. await Promise.allSettled(deliveries.values());await Promise.allSettled(this.inflight.values())
  319. await this.localVm?.close();await this.computers?.controls.close();await this.computers?.pool.close()
  320. }finally{this.target.client.close();this.controlTransport.close();this.db.close();this.status('执行节点已停止')}
  321. }
  322. }
  323. }