@easbot/gateway 0.2.24 → 0.2.26
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/chunks/{chunk-OBW6CNOG.mjs → chunk-7OFQ65NG.mjs} +6 -6
- package/dist/chunks/{chunk-RIWYHEQ6.cjs → chunk-AUPHFE34.cjs} +6 -6
- package/dist/chunks/chunk-BJV43DSB.mjs +14 -0
- package/dist/chunks/chunk-SOCQU5UV.cjs +14 -0
- package/dist/chunks/{global-DVGSJ3V3.mjs → global-5GMGC2RN.mjs} +1 -1
- package/dist/chunks/{global-WNXPJXAS.cjs → global-TEPBDNQM.cjs} +1 -1
- package/dist/chunks/log-JFEBRH2W.cjs +1 -0
- package/dist/chunks/log-MIT4H7S5.mjs +1 -0
- package/dist/chunks/server-UXFF63KL.cjs +1 -0
- package/dist/chunks/server-Y336TWLM.mjs +1 -0
- package/dist/index.cjs +2 -2
- package/dist/index.d.cts +30 -14
- package/dist/index.d.ts +30 -14
- package/dist/index.mjs +2 -2
- package/package.json +5 -5
- package/dist/chunks/chunk-GQFBXITJ.mjs +0 -5
- package/dist/chunks/chunk-LR37LRDG.cjs +0 -5
- package/dist/chunks/log-7WPRHZWH.cjs +0 -1
- package/dist/chunks/log-MKRLBH4R.mjs +0 -1
- package/dist/chunks/server-HGRZI7JT.mjs +0 -1
- package/dist/chunks/server-ZWJPPFTK.cjs +0 -1
package/dist/index.cjs
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
'use strict';var chunkLR37LRDG_cjs=require('./chunks/chunk-LR37LRDG.cjs'),chunkRIWYHEQ6_cjs=require('./chunks/chunk-RIWYHEQ6.cjs'),chunkGY3SWWW3_cjs=require('./chunks/chunk-GY3SWWW3.cjs'),utils=require('@easbot/utils');var a=chunkRIWYHEQ6_cjs.a.create({service:"gateway"}),m=class i extends Error{constructor(t,n,s){super(t);chunkGY3SWWW3_cjs.a(this,"type");chunkGY3SWWW3_cjs.a(this,"stage");chunkGY3SWWW3_cjs.a(this,"cause");this.type="GatewayInitializationError",this.stage=n,this.cause=s,Error.captureStackTrace&&Error.captureStackTrace(this,i);}};function k(i){let e=["ECONNRESET","EPIPE","ETIMEDOUT","ECONNREFUSED","ENOTFOUND","EHOSTUNREACH","ENETUNREACH"],t=i.code;return e.includes(t??"")}function Ze(){process.on("unhandledRejection",i=>{let e=i instanceof Error?i:new Error(String(i));k(e)||a.error("unhandled promise rejection",{error:e.message,stack:e.stack});}),process.on("uncaughtException",i=>{k(i)||(a.error("uncaught exception",{error:i.message,stack:i.stack}),process.exit(1));});}exports.Gateway=void 0;(rt=>{let i={initialized:false,initPromise:null,directory:".easbot",config:null},e,t={server:null,config:null,initialized:false,status:"stopped"};async function n(u={}){if(i.initialized)return;if(i.initPromise)return i.initPromise;let d={directory:u.directory??".easbot",printLogs:u.printLogs??false,logLevel:u.logLevel??"INFO"};return i.directory=d.directory,i.initPromise=(async()=>{let{Global:C}=await import('./chunks/global-WNXPJXAS.cjs');try{await C.init();}catch(g){throw new m("Failed to initialize global directories","global",g instanceof Error?g:void 0)}try{await chunkRIWYHEQ6_cjs.a.init({logDir:C.Path.log,print:d.printLogs,level:d.logLevel});}catch(g){throw new m("Failed to initialize log infrastructure","log",g instanceof Error?g:void 0)}try{Ze();}catch(g){throw new m("Failed to install unhandled exception handlers","server",g instanceof Error?g:void 0)}utils.Fetch.hasProxyConfigured()&&(utils.Fetch.enableProxy({connectTimeout:1e4,keepAliveTimeout:3e4}),a.info("global proxy enabled",{proxyUrl:utils.Fetch.getProxyUrl()}));try{i.config=await chunkLR37LRDG_cjs.v(d.directory);}catch(g){throw new m("Failed to load gateway configuration","config",g instanceof Error?g:void 0)}i.initialized=true,a.info("gateway runtime initialized",{directory:d.directory,logLevel:d.logLevel});})(),i.initPromise}rt.init=n;function s(){return i.initialized}rt.isInitialized=s;function r(){return e||(e=async()=>{if(!i.initialized)throw new m("Gateway not initialized. Call init() first.","server");let u=chunkLR37LRDG_cjs.Q(),d=await chunkLR37LRDG_cjs.v(u.directory);return d.server?.enabled?(t={server:null,config:d,initialized:true,status:"stopped"},a.info("gateway initialized",{port:d.server?.port,hostname:d.server?.hostname}),t):(a.debug("gateway server disabled or not configured"),{server:null,config:d,initialized:true,status:"stopped"})}),e}function c(){return r()()}rt.state=c;async function l(){return (await c()).server}rt.get=l;async function y(){return (await c()).config?.server?.enabled??false}rt.isEnabled=y;async function et(){return (await c()).config}rt.config=et;function tt(){return t.status}rt.getStatus=tt;async function H(u){a.debug("Gateway.start: called"),i.initialized||(a.debug("Gateway.start: calling init()"),await n(),a.debug("Gateway.start: init() completed")),a.debug("Gateway.start: calling state()");let d=await c();if(a.debug("Gateway.start: state() completed",{status:d.status}),!(d.config?.server?.enabled??true))throw a.warn("gateway server is disabled"),new Error("Gateway server is disabled");if(t.status==="running"){a.info("gateway server is already running");return}if(t.status==="starting"){a.info("gateway server is starting");return}t.status="starting",a.debug("Gateway.start: status set to starting");try{let g={...d.config?.server,...u};a.debug("gateway start: creating server with config",{port:g.port,hostname:g.hostname,path:g.path,https:g.https?.enabled}),a.debug("Gateway.start: calling createGatewayServer");let w=await F(g);a.debug("Gateway.start: createGatewayServer completed"),a.debug("gateway start: server created, updating state"),t.server=w,t.status="running",t.error=void 0,a.info("gateway server started",{port:g.port,hostname:g.hostname});}catch(g){throw t.status="error",t.error=g instanceof Error?g.message:String(g),a.error("failed to start gateway server",{error:t.error}),g}}rt.start=H;async function _(){if(t.status==="stopped"){a.info("gateway server is already stopped");return}if(t.status==="stopping"){a.info("gateway server is stopping");return}t.status="stopping";try{t.server&&await t.server.stop(),t.server=null,t.status="stopped",t.error=void 0,a.info("gateway server stopped");}catch(u){throw t.status="error",t.error=u instanceof Error?u.message:String(u),a.error("failed to stop gateway server",{error:t.error}),u}}rt.stop=_;async function nt(u){a.info("restarting gateway server"),(t.status==="running"||t.status==="starting")&&await _(),await H(u);}rt.restart=nt;async function st(){let u=chunkLR37LRDG_cjs.Q(),d=await chunkLR37LRDG_cjs.v(u.directory);return t.config=d,i.config=d,a.info("gateway config reloaded"),d}rt.reloadConfig=st;async function F(u,d){a.debug("createGatewayServer: starting");let{GatewayServer:C}=await import('./chunks/server-ZWJPPFTK.cjs');a.debug("createGatewayServer: GatewayServer imported");let g=new C(u);a.debug("createGatewayServer: server instance created"),a.debug("createGatewayServer: created server instance, calling start()");let w=d?.startupTimeout??3e4;if(await(async()=>{let b=new Promise((A,O)=>{setTimeout(()=>{O(new Error(`Gateway server startup timeout after ${w}ms`));},w);});try{a.debug("createGatewayServer: calling server.start()"),await Promise.race([g.start(),b]),a.debug("createGatewayServer: server.start() completed");}catch(A){a.error("gateway server start failed, attempting cleanup",{error:A instanceof Error?A.message:String(A)});try{await g.stop();}catch{}throw A}})(),d?.onStarted)try{await d.onStarted(g);}catch(b){a.warn("server started but onStarted callback failed",{error:b instanceof Error?b.message:String(b)});}return a.info("gateway server created and started",{port:u.port,hostname:u.hostname}),g}rt.createGatewayServer=F;})(exports.Gateway||(exports.Gateway={}));var f=chunkRIWYHEQ6_cjs.a.create({service:"gateway:agent-registry"}),P=class{constructor(e={}){chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"agents",new Map);chunkGY3SWWW3_cjs.a(this,"heartbeatTimer",null);chunkGY3SWWW3_cjs.a(this,"roundRobinIndex",0);this.config={...chunkLR37LRDG_cjs._,...e};}async register(e){if(this.agents.has(e.id))throw new Error(`Agent with ID "${e.id}" is already registered`);if(this.agents.size>=this.config.maxAgents)throw new Error(`Maximum number of agents (${this.config.maxAgents}) reached`);let t=Date.now(),n={...e,status:"healthy",lastHeartbeat:t,registeredAt:t,connectionCount:0};this.agents.set(e.id,n),f.info("Agent registered",{agentId:e.id,name:e.name,capabilities:e.capabilities,address:e.address,totalAgents:this.agents.size});}async deregister(e){let t=this.agents.get(e);if(!t){f.warn("Attempted to deregister unknown agent",{agentId:e});return}this.agents.delete(e),f.info("Agent deregistered",{agentId:e,name:t.name,totalAgents:this.agents.size});}async heartbeat(e,t){let n=this.agents.get(e);if(!n)throw f.warn("Heartbeat received from unknown agent",{agentId:e}),new Error(`Agent "${e}" is not registered`);n.lastHeartbeat=Date.now(),n.status="healthy",t&&(n.metadata={...n.metadata,...t}),f.debug("Agent heartbeat received",{agentId:e,lastHeartbeat:n.lastHeartbeat});}getAgent(e){return this.agents.get(e)}getAllAgents(){return Array.from(this.agents.values())}getHealthyAgents(){return this.getAllAgents().filter(e=>e.status==="healthy")}selectAgent(e){let{strategy:t,capabilities:n,healthyOnly:s=true}=e,r=s?this.getHealthyAgents():this.getAllAgents();if(n&&n.length>0&&(r=r.filter(l=>n.every(y=>l.capabilities.includes(y)))),r.length===0){f.warn("No agents available for selection",{strategy:t,capabilities:n,healthyOnly:s,totalAgents:this.agents.size});return}let c;switch(t){case "random":c=this.selectRandom(r);break;case "round-robin":c=this.selectRoundRobin(r);break;case "least-connections":c=this.selectLeastConnections(r);break;default:c=this.selectRandom(r);}return f.debug("Agent selected",{agentId:c.id,strategy:t,connectionCount:c.connectionCount}),c}incrementConnection(e){let t=this.agents.get(e);t&&(t.connectionCount++,f.debug("Agent connection incremented",{agentId:e,connectionCount:t.connectionCount}));}decrementConnection(e){let t=this.agents.get(e);t&&t.connectionCount>0&&(t.connectionCount--,f.debug("Agent connection decremented",{agentId:e,connectionCount:t.connectionCount}));}startHeartbeatCheck(){if(this.heartbeatTimer){f.warn("Heartbeat check is already running");return}f.info("Starting heartbeat check",{interval:this.config.heartbeatCheckInterval,timeout:this.config.heartbeatTimeout}),this.heartbeatTimer=setInterval(()=>this.checkHeartbeats(),this.config.heartbeatCheckInterval);}stopHeartbeatCheck(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null,f.info("Heartbeat check stopped"));}checkHeartbeats(){let e=Date.now(),{heartbeatTimeout:t,autoDeregisterTimeout:n}=this.config,s=[],r=0;for(let[c,l]of this.agents.entries()){let y=e-l.lastHeartbeat;if(y>n){s.push(c);continue}y>t&&l.status==="healthy"&&(l.status="unhealthy",r++,f.warn("Agent marked as unhealthy due to heartbeat timeout",{agentId:c,lastHeartbeat:l.lastHeartbeat,timeSinceLastHeartbeat:y,timeout:t}));}for(let c of s){let l=this.agents.get(c);l&&(f.info("Auto-deregistering agent due to timeout",{agentId:c,name:l.name,lastHeartbeat:l.lastHeartbeat,timeout:n}),this.agents.delete(c));}(r>0||s.length>0)&&f.info("Heartbeat check completed",{unhealthyCount:r,deregisteredCount:s.length,totalAgents:this.agents.size});}selectRandom(e){let t=Math.floor(Math.random()*e.length);return e[t]}selectRoundRobin(e){this.roundRobinIndex=this.roundRobinIndex%e.length;let t=e[this.roundRobinIndex];return this.roundRobinIndex++,t}selectLeastConnections(e){return e.reduce((t,n)=>n.connectionCount<t.connectionCount?n:t)}getStats(){let e=this.getAllAgents();return {totalAgents:e.length,healthyAgents:e.filter(t=>t.status==="healthy").length,unhealthyAgents:e.filter(t=>t.status==="unhealthy").length,offlineAgents:e.filter(t=>t.status==="offline").length,totalConnections:e.reduce((t,n)=>t+n.connectionCount,0)}}setAgentStatus(e,t){let n=this.agents.get(e);n&&(n.status=t);}setLastHeartbeat(e,t){let n=this.agents.get(e);n&&(n.lastHeartbeat=t);}};var h=chunkRIWYHEQ6_cjs.a.create({service:"gateway:agent-sync-manager"}),M=class{constructor(e){chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"registry");chunkGY3SWWW3_cjs.a(this,"localNodeId");chunkGY3SWWW3_cjs.a(this,"localNodeName");chunkGY3SWWW3_cjs.a(this,"remoteNodes",new Map);chunkGY3SWWW3_cjs.a(this,"remoteAgents",new Map);chunkGY3SWWW3_cjs.a(this,"syncTimer",null);chunkGY3SWWW3_cjs.a(this,"running",false);this.registry=e.registry,this.localNodeId=e.localNodeId,this.localNodeName=e.localNodeName,this.config={...chunkLR37LRDG_cjs.$,...e.config};}async start(){if(this.running){h.warn("Agent sync manager is already running");return}h.info("Starting agent sync manager",{localNodeId:this.localNodeId,mode:this.config.mode,interval:this.config.interval,remoteNodes:this.config.remoteNodes.length});for(let e of this.config.remoteNodes)this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"});(this.config.mode==="pull"||this.config.mode==="both")&&(this.syncTimer=setInterval(()=>this.pullFromAllNodes(),this.config.interval)),this.running=true,h.info("Agent sync manager started");}async stop(){if(!this.running){h.warn("Agent sync manager is not running");return}h.info("Stopping agent sync manager"),this.syncTimer&&(clearInterval(this.syncTimer),this.syncTimer=null),this.remoteNodes.clear(),this.remoteAgents.clear(),this.running=false,h.info("Agent sync manager stopped");}async pullFromAllNodes(){for(let[e]of this.remoteNodes.entries())try{await this.pullFromNode(e);}catch(t){h.error("Failed to pull from node",{nodeId:e,error:t.message});}}async pullFromNode(e){let t=this.remoteNodes.get(e);if(!t)throw new Error(`Unknown remote node: ${e}`);t.status="syncing";let n=this.config.remoteNodes.find(s=>s.id===e);if(!n)throw new Error(`Remote node not found in config: ${e}`);h.debug("Pulling agent list from node",{nodeId:e,address:n.address});try{let s=`http://${n.address}:${n.port}/sync/agents`,r=await utils.Fetch.get(s,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);let c=r.data.agents||[];for(let l of c)l.sourceGatewayId=e,this.remoteAgents.set(l.id,l);return t.lastSyncAt=Date.now(),t.status="synced",t.info.lastSyncAt=t.lastSyncAt,t.info.status="online",delete t.error,h.info("Pulled agent list from node",{nodeId:e,agentCount:c.length,lastSyncAt:t.lastSyncAt}),c}catch(s){throw t.status="error",t.error=s.message,t.info.status="offline",h.error("Failed to pull from node",{nodeId:e,error:s.message}),s}}async pushEvent(e){if(this.config.mode!=="push"&&this.config.mode!=="both")return;let t=[];for(let[n]of this.remoteNodes.entries())t.push(this.pushToNode(n,e));await Promise.allSettled(t);}async pushToNode(e,t){let n=this.config.remoteNodes.find(s=>s.id===e);if(!n){h.warn("Unknown remote node for push",{nodeId:e});return}try{let s=`http://${n.address}:${n.port}/sync/events`,r=await utils.Fetch.post(s,t,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);h.debug("Pushed event to node",{nodeId:e,eventType:t.type});}catch(s){h.error("Failed to push to node",{nodeId:e,eventType:t.type,error:s.message});}}async handleSyncMessage(e){switch(h.debug("Received sync message",{type:e.type,sourceNodeId:e.sourceNodeId}),e.type){case "agent_list_request":await this.handleAgentListRequest(e);break;case "agent_list_response":await this.handleAgentListResponse(e);break;case "agent_register":await this.handleAgentRegister(e);break;case "agent_deregister":await this.handleAgentDeregister(e);break;case "agent_heartbeat":await this.handleAgentHeartbeat(e);break;case "agent_status_change":await this.handleAgentStatusChange(e);break;default:h.warn("Unknown sync message type",{type:e.type});}}async handleAgentListRequest(e){let t=this.getLocalAgentsForSync(),n={type:"agent_list_response",sourceNodeId:this.localNodeId,timestamp:Date.now(),requestId:e.requestId,agents:t,fullList:true,syncAt:Date.now()};await this.pushToNode(e.sourceNodeId,n);}async handleAgentListResponse(e){for(let t of e.agents)this.remoteAgents.set(t.id,t);h.info("Received agent list response",{sourceNodeId:e.sourceNodeId,agentCount:e.agents.length});}async handleAgentRegister(e){let{agent:t}=e,n=this.remoteAgents.get(t.id);if(n){let s=this.resolveConflict(n,t);this.remoteAgents.set(t.id,s);}else this.remoteAgents.set(t.id,t);h.info("Remote agent registered",{agentId:t.id,name:t.name,sourceNodeId:e.sourceNodeId});}async handleAgentDeregister(e){let{agentId:t}=e;this.remoteAgents.delete(t),h.info("Remote agent deregistered",{agentId:t,sourceNodeId:e.sourceNodeId});}async handleAgentHeartbeat(e){let{agentId:t,heartbeatAt:n,status:s}=e,r=this.remoteAgents.get(t);r&&(r.updatedAt=n,r.status=s),h.debug("Remote agent heartbeat received",{agentId:t,status:s,sourceNodeId:e.sourceNodeId});}async handleAgentStatusChange(e){let{agentId:t,newStatus:n,changedAt:s}=e,r=this.remoteAgents.get(t);r&&(r.status=n,r.updatedAt=s),h.info("Remote agent status changed",{agentId:t,newStatus:n,sourceNodeId:e.sourceNodeId});}resolveConflict(e,t){switch(this.config.conflictResolution){case "latest":return t.updatedAt>e.updatedAt?t:e;case "local":return e;case "remote":return t;default:return t.updatedAt>e.updatedAt?t:e}}getLocalAgentsForSync(){return this.registry.getAllAgents().map(t=>({id:t.id,name:t.name,sourceGatewayId:this.localNodeId,address:t.address,capabilities:t.capabilities,status:t.status,updatedAt:t.lastHeartbeat,metadata:t.metadata}))}getMergedAgentList(){let e=new Map;for(let t of this.getLocalAgentsForSync())e.set(t.id,t);for(let[t,n]of this.remoteAgents.entries())e.has(t)||e.set(t,n);return Array.from(e.values())}getRemoteAgents(){return Array.from(this.remoteAgents.values())}getStats(){let e=this.registry.getAllAgents(),t=0,n=0;for(let r of this.remoteNodes.values())r.info.status==="online"?t++:n++;let s=null;for(let r of this.remoteNodes.values())r.lastSyncAt>(s||0)&&(s=r.lastSyncAt);return {localAgentCount:e.length,remoteAgentCount:this.remoteAgents.size,totalAgentCount:e.length+this.remoteAgents.size,onlineNodeCount:t,offlineNodeCount:n,lastSyncAt:s}}isRunning(){return this.running}addRemoteNode(e){this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"}),h.info("Remote node added",{nodeId:e.id,address:e.address});}removeRemoteNode(e){this.remoteNodes.delete(e);for(let[t,n]of this.remoteAgents.entries())n.sourceGatewayId===e&&this.remoteAgents.delete(t);h.info("Remote node removed",{nodeId:e});}};var T=class{constructor(e){chunkGY3SWWW3_cjs.a(this,"log");chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"ws",null);chunkGY3SWWW3_cjs.a(this,"state","disconnected");chunkGY3SWWW3_cjs.a(this,"subscriptions",new Map);chunkGY3SWWW3_cjs.a(this,"messageCallbacks",new Set);chunkGY3SWWW3_cjs.a(this,"agentListCallbacks",new Set);chunkGY3SWWW3_cjs.a(this,"cachedAgentList",[]);chunkGY3SWWW3_cjs.a(this,"localAgentList",[]);chunkGY3SWWW3_cjs.a(this,"reconnectAttempts",0);chunkGY3SWWW3_cjs.a(this,"reconnectTimer",null);chunkGY3SWWW3_cjs.a(this,"heartbeatTimer",null);chunkGY3SWWW3_cjs.a(this,"agentListRequestId",null);chunkGY3SWWW3_cjs.a(this,"agentListRequestResolve",null);this.config={url:e.url,type:e.type,id:e.id??`client_${Date.now()}_${Math.random().toString(36).substring(2,9)}`,reconnect:{enabled:e.reconnect?.enabled??true,maxAttempts:e.reconnect?.maxAttempts??5,delay:e.reconnect?.delay??3e3},heartbeat:{enabled:e.heartbeat?.enabled??true,interval:e.heartbeat?.interval??3e4}},this.log=chunkRIWYHEQ6_cjs.a.create({service:`gateway:client:${this.config.id}`});}async connect(){if(this.state==="connected"||this.state==="connecting"){this.log.warn("already connected or connecting");return}return this.state="connecting",this.log.info("connecting to gateway",{url:this.config.url}),new Promise((e,t)=>{try{this.ws=new WebSocket(this.config.url),this.ws.onopen=()=>{this.state="connected",this.reconnectAttempts=0,this.log.info("connected to gateway"),this.config.heartbeat.enabled&&this.startHeartbeat(),this.resubscribeAll(),e();},this.ws.onmessage=n=>{this.handleMessage(n.data);},this.ws.onclose=n=>{this.handleClose(n.code,n.reason);},this.ws.onerror=n=>{this.log.error("WebSocket error",{error:String(n)}),this.state==="connecting"&&t(new Error("Connection failed"));};}catch(n){this.state="disconnected",t(n);}})}async disconnect(){this.state!=="disconnected"&&(this.log.info("disconnecting from gateway"),this.stopHeartbeat(),this.stopReconnect(),this.ws&&(this.ws.close(1e3,"Client disconnect"),this.ws=null),this.state="disconnected",this.log.info("disconnected from gateway"));}async subscribe(e,t){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"subscribe",sessionId:e,backendSessionId:t,clientId:this.config.id}),this.log.info("subscribed to session",{sessionId:e,backendSessionId:t});}async unsubscribe(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"unsubscribe",sessionId:e,clientId:this.config.id}),this.subscriptions.delete(e),this.log.info("unsubscribed from session",{sessionId:e});}async send(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"message",message:e});}onMessage(e){this.messageCallbacks.add(e);}offMessage(e){this.messageCallbacks.delete(e);}getState(){return this.state}getId(){return this.config.id}getSubscriptions(){return [...this.subscriptions.keys()]}setLocalAgents(e){this.localAgentList=e,this.updateMergedAgentList();}async fetchAgentList(){if(this.state!=="connected")throw new Error("Not connected to gateway");return new Promise((e,t)=>{let n=`req_${Date.now()}_${Math.random().toString(36).substring(2,9)}`;this.agentListRequestId=n,this.agentListRequestResolve=e,this.sendData({type:"agent_list_request",requestId:n}),setTimeout(()=>{this.agentListRequestId===n&&(this.agentListRequestId=null,this.agentListRequestResolve=null,t(new Error("Agent list request timeout")));},1e4);})}getConnectableAgents(e={}){let{localAgents:t,preferLocal:n=true,filterOffline:s=true}=e,r=t||this.localAgentList,c=this.cachedAgentList,l=this.mergeAgentLists(r,c,n);return s?l.filter(y=>y.status!=="offline"):l}onAgentListChange(e){this.agentListCallbacks.add(e);}offAgentListChange(e){this.agentListCallbacks.delete(e);}mergeAgentLists(e,t,n){let s=new Map,r=n?t:e,c=n?e:t;for(let l of r)s.set(l.id,l);for(let l of c)s.set(l.id,l);return Array.from(s.values())}updateMergedAgentList(){let e=this.getConnectableAgents({filterOffline:false});for(let t of this.agentListCallbacks)try{t(e);}catch(n){this.log.error("agent list callback error",{error:String(n)});}}handleAgentListResponse(e,t){t&&this.agentListRequestId===t&&this.agentListRequestResolve&&(this.agentListRequestId=null,this.agentListRequestResolve(e),this.agentListRequestResolve=null),this.cachedAgentList=e,this.updateMergedAgentList(),this.log.debug("received agent list",{count:e.length});}sendData(e){if(!this.ws||this.ws.readyState!==WebSocket.OPEN){this.log.warn("WebSocket not ready, cannot send");return}this.ws.send(JSON.stringify(e));}handleMessage(e){try{let t=JSON.parse(e);switch(t.type){case "message":this.handleGatewayMessage(t.message);break;case "subscribed":this.subscriptions.set(t.sessionId,t.subscription);break;case "unsubscribed":this.subscriptions.delete(t.sessionId);break;case "pong":break;case "agent_list_response":this.handleAgentListResponse(t.agents||[],t.requestId);break;case "agent_update":this.handleAgentUpdate(t.agent,t.action);break;default:this.log.debug("unknown message type",{type:t.type});}}catch(t){this.log.error("failed to parse message",{error:String(t)});}}handleAgentUpdate(e,t){switch(t){case "add":case "update":{let n=this.cachedAgentList.findIndex(s=>s.id===e.id);n>=0?this.cachedAgentList[n]=e:this.cachedAgentList.push(e);break}case "remove":this.cachedAgentList=this.cachedAgentList.filter(n=>n.id!==e.id);break}this.updateMergedAgentList();}handleGatewayMessage(e){for(let t of this.messageCallbacks)try{t(e);}catch(n){this.log.error("message callback error",{error:String(n)});}}handleClose(e,t){this.log.info("connection closed",{code:e,reason:t}),this.state="disconnected",this.ws=null,this.stopHeartbeat(),this.config.reconnect.enabled&&this.scheduleReconnect();}scheduleReconnect(){if(this.reconnectAttempts>=this.config.reconnect.maxAttempts){this.log.error("max reconnect attempts reached");return}this.reconnectAttempts++,this.state="reconnecting";let e=this.config.reconnect.delay*this.reconnectAttempts;this.log.info("scheduling reconnect",{attempt:this.reconnectAttempts,delay:e}),this.reconnectTimer=setTimeout(async()=>{try{await this.connect();}catch(t){this.log.error("reconnect failed",{error:String(t)});}},e);}stopReconnect(){this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null);}startHeartbeat(){this.heartbeatTimer=setInterval(()=>{this.state==="connected"&&this.sendData({type:"ping"});},this.config.heartbeat.interval);}stopHeartbeat(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null);}async resubscribeAll(){for(let e of this.subscriptions.keys())try{await this.subscribe(e);}catch(t){this.log.error("failed to resubscribe",{sessionId:e,error:String(t)});}}getHttpUrl(){return this.config.url.replace(/^ws:/,"http:").replace(/^wss:/,"https:")}async httpRequest(e,t={}){let n=`${this.getHttpUrl()}${e}`,s=await fetch(n,{...t,headers:{"Content-Type":"application/json",...t.headers}});if(!s.ok){let r=await s.text().catch(()=>"Unknown error");throw new Error(`HTTP ${s.status}: ${r}`)}return s.json()}async healthCheck(){return this.httpRequest("/health")}async getStatus(){return this.httpRequest("/status")}async listContacts(e,t={}){let n=new URLSearchParams({platform:e,limit:String(t.limit??100),offset:String(t.offset??0)});return this.httpRequest(`/contacts?${n}`)}async getContact(e){return this.httpRequest(`/contacts/${encodeURIComponent(e)}`)}async listSessions(e={}){let t=new URLSearchParams({limit:String(e.limit??100),offset:String(e.offset??0)});return this.httpRequest(`/sessions?${t}`)}async getSession(e){return this.httpRequest(`/sessions/${encodeURIComponent(e)}`)}async createSession(e){return this.httpRequest("/session",{method:"POST",body:JSON.stringify(e??{})})}async sendMessage(e,t){return this.httpRequest(`/session/${encodeURIComponent(e)}/prompt`,{method:"POST",body:JSON.stringify(t)})}};async function qt(i){let{Log:e}=await import('./chunks/log-7WPRHZWH.cjs'),t=false;await e.init({logDir:i.logDir??process.env.EASBOT_LOG_PATH??process.cwd(),print:i.print??false,dev:i.dev??t,level:i.level??("INFO")});}
|
|
2
|
-
Object.defineProperty(exports,"AgentRegistryConfigSchema",{enumerable:true,get:function(){return
|
|
1
|
+
'use strict';var chunkSOCQU5UV_cjs=require('./chunks/chunk-SOCQU5UV.cjs'),chunkAUPHFE34_cjs=require('./chunks/chunk-AUPHFE34.cjs'),chunkGY3SWWW3_cjs=require('./chunks/chunk-GY3SWWW3.cjs'),utils=require('@easbot/utils');var a=chunkAUPHFE34_cjs.a.create({service:"gateway"}),m=class g extends Error{constructor(t,n,s){super(t);chunkGY3SWWW3_cjs.a(this,"type");chunkGY3SWWW3_cjs.a(this,"stage");chunkGY3SWWW3_cjs.a(this,"cause");this.type="GatewayInitializationError",this.stage=n,this.cause=s,Error.captureStackTrace&&Error.captureStackTrace(this,g);}};function G(g){let e=["ECONNRESET","EPIPE","ETIMEDOUT","ECONNREFUSED","ENOTFOUND","EHOSTUNREACH","ENETUNREACH"],t=g.code;return e.includes(t??"")}function et(){process.on("unhandledRejection",g=>{let e=g instanceof Error?g:new Error(String(g));G(e)||a.error("unhandled promise rejection",{error:e.message,stack:e.stack});}),process.on("uncaughtException",g=>{G(g)||(a.error("uncaught exception",{error:g.message,stack:g.stack}),process.exit(1));});}exports.Gateway=void 0;(at=>{let g={initialized:false,initPromise:null,directory:".easbot",config:null},e,t={server:null,config:null,initialized:false,status:"stopped"};async function n(u={}){if(g.initialized)return;if(g.initPromise)return g.initPromise;let d={directory:u.directory??".easbot",printLogs:u.printLogs??false,logLevel:u.logLevel??"INFO"};return g.directory=d.directory,g.initPromise=(async()=>{let{Global:C}=await import('./chunks/global-TEPBDNQM.cjs');try{await C.init();}catch(l){throw new m("Failed to initialize global directories","global",l instanceof Error?l:void 0)}try{await chunkAUPHFE34_cjs.a.init({logDir:C.Path.log,print:d.printLogs,level:d.logLevel});}catch(l){throw new m("Failed to initialize log infrastructure","log",l instanceof Error?l:void 0)}try{et();}catch(l){throw new m("Failed to install unhandled exception handlers","server",l instanceof Error?l:void 0)}utils.Fetch.hasProxyConfigured()&&(utils.Fetch.enableProxy({connectTimeout:1e4,keepAliveTimeout:3e4}),a.info("global proxy enabled",{proxyUrl:utils.Fetch.getProxyUrl()}));try{g.config=await chunkSOCQU5UV_cjs.v(d.directory);}catch(l){throw new m("Failed to load gateway configuration","config",l instanceof Error?l:void 0)}g.initialized=true,a.info("gateway runtime initialized",{directory:d.directory,logLevel:d.logLevel});})(),g.initPromise}at.init=n;function s(){return g.initialized}at.isInitialized=s;function r(){return e||(e=async()=>{if(!g.initialized)throw new m("Gateway not initialized. Call init() first.","server");let u=chunkSOCQU5UV_cjs.Q(),d=await chunkSOCQU5UV_cjs.v(u.directory);return d.server?.enabled?(t={server:null,config:d,initialized:true,status:"stopped"},a.info("gateway initialized",{port:d.server?.port,hostname:d.server?.hostname}),t):(a.debug("gateway server disabled or not configured"),{server:null,config:d,initialized:true,status:"stopped"})}),e}function i(){return r()()}at.state=i;async function c(){return (await i()).server}at.get=c;async function p(){return (await i()).config?.server?.enabled??false}at.isEnabled=p;async function x(){return (await i()).config}at.config=x;function nt(){return t.status}at.getStatus=nt;async function _(u){a.debug("Gateway.start: called"),g.initialized||(a.debug("Gateway.start: calling init()"),await n(),a.debug("Gateway.start: init() completed")),a.debug("Gateway.start: calling state()");let d=await i();if(a.debug("Gateway.start: state() completed",{status:d.status}),!(d.config?.server?.enabled??true))throw a.warn("gateway server is disabled"),new Error("Gateway server is disabled");if(t.status==="running"){a.info("gateway server is already running");return}if(t.status==="starting"){a.info("gateway server is starting");return}t.status="starting",a.debug("Gateway.start: status set to starting");try{let l={...d.config?.server,...u};a.debug("gateway start: creating server with config",{port:l.port,hostname:l.hostname,path:l.path,https:l.https?.enabled}),a.debug("Gateway.start: calling createGatewayServer");let w=await F(l);a.debug("Gateway.start: createGatewayServer completed"),a.debug("gateway start: server created, updating state"),t.server=w,t.status="running",t.error=void 0,a.info("gateway server started",{port:l.port,hostname:l.hostname});}catch(l){throw t.status="error",t.error=l instanceof Error?l.message:String(l),a.error("failed to start gateway server",{error:t.error}),l}}at.start=_;async function O(){if(t.status==="stopped"){a.info("gateway server is already stopped");return}if(t.status==="stopping"){a.info("gateway server is stopping");return}t.status="stopping";try{t.server&&await t.server.stop(),t.server=null,t.status="stopped",t.error=void 0,await chunkAUPHFE34_cjs.a.close(),a.info("gateway server stopped");}catch(u){throw t.status="error",t.error=u instanceof Error?u.message:String(u),a.error("failed to stop gateway server",{error:t.error}),u}}at.stop=O;async function st(u){a.info("restarting gateway server"),(t.status==="running"||t.status==="starting")&&await O(),await _(u);}at.restart=st;async function rt(){let u=chunkSOCQU5UV_cjs.Q(),d=await chunkSOCQU5UV_cjs.v(u.directory);return t.config=d,g.config=d,a.info("gateway config reloaded"),d}at.reloadConfig=rt;async function F(u,d){a.debug("createGatewayServer: starting");let{GatewayServer:C}=await import('./chunks/server-UXFF63KL.cjs');a.debug("createGatewayServer: GatewayServer imported");let l=new C(u);a.debug("createGatewayServer: server instance created"),a.debug("createGatewayServer: created server instance, calling start()");let w=d?.startupTimeout??3e4;if(await(async()=>{let b=new Promise((A,D)=>{setTimeout(()=>{D(new Error(`Gateway server startup timeout after ${w}ms`));},w);});try{a.debug("createGatewayServer: calling server.start()"),await Promise.race([l.start(),b]),a.debug("createGatewayServer: server.start() completed");}catch(A){a.error("gateway server start failed, attempting cleanup",{error:A instanceof Error?A.message:String(A)});try{await l.stop();}catch{}throw A}})(),d?.onStarted)try{await d.onStarted(l);}catch(b){a.warn("server started but onStarted callback failed",{error:b instanceof Error?b.message:String(b)});}return a.info("gateway server created and started",{port:u.port,hostname:u.hostname}),l}at.createGatewayServer=F;})(exports.Gateway||(exports.Gateway={}));var f=chunkAUPHFE34_cjs.a.create({service:"gateway:agent-registry"}),P=class{constructor(e={}){chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"agents",new Map);chunkGY3SWWW3_cjs.a(this,"heartbeatTimer",null);chunkGY3SWWW3_cjs.a(this,"roundRobinIndex",0);this.config={...chunkSOCQU5UV_cjs._,...e};}async register(e){if(this.agents.has(e.id))throw new Error(`Agent with ID "${e.id}" is already registered`);if(this.agents.size>=this.config.maxAgents)throw new Error(`Maximum number of agents (${this.config.maxAgents}) reached`);let t=Date.now(),n={...e,status:"healthy",lastHeartbeat:t,registeredAt:t,connectionCount:0};this.agents.set(e.id,n),f.info("Agent registered",{agentId:e.id,name:e.name,capabilities:e.capabilities,address:e.address,totalAgents:this.agents.size});}async deregister(e){let t=this.agents.get(e);if(!t){f.warn("Attempted to deregister unknown agent",{agentId:e});return}this.agents.delete(e),f.info("Agent deregistered",{agentId:e,name:t.name,totalAgents:this.agents.size});}async heartbeat(e,t){let n=this.agents.get(e);if(!n)throw f.warn("Heartbeat received from unknown agent",{agentId:e}),new Error(`Agent "${e}" is not registered`);n.lastHeartbeat=Date.now(),n.status="healthy",t&&(n.metadata={...n.metadata,...t}),f.debug("Agent heartbeat received",{agentId:e,lastHeartbeat:n.lastHeartbeat});}getAgent(e){return this.agents.get(e)}getAllAgents(){return Array.from(this.agents.values())}getHealthyAgents(){return this.getAllAgents().filter(e=>e.status==="healthy")}selectAgent(e){let{strategy:t,capabilities:n,healthyOnly:s=true}=e,r=s?this.getHealthyAgents():this.getAllAgents();if(n&&n.length>0&&(r=r.filter(c=>n.every(p=>c.capabilities.includes(p)))),r.length===0){f.warn("No agents available for selection",{strategy:t,capabilities:n,healthyOnly:s,totalAgents:this.agents.size});return}let i;switch(t){case "random":i=this.selectRandom(r);break;case "round-robin":i=this.selectRoundRobin(r);break;case "least-connections":i=this.selectLeastConnections(r);break;default:i=this.selectRandom(r);}return f.debug("Agent selected",{agentId:i.id,strategy:t,connectionCount:i.connectionCount}),i}incrementConnection(e){let t=this.agents.get(e);t&&(t.connectionCount++,f.debug("Agent connection incremented",{agentId:e,connectionCount:t.connectionCount}));}decrementConnection(e){let t=this.agents.get(e);t&&t.connectionCount>0&&(t.connectionCount--,f.debug("Agent connection decremented",{agentId:e,connectionCount:t.connectionCount}));}startHeartbeatCheck(){if(this.heartbeatTimer){f.warn("Heartbeat check is already running");return}f.info("Starting heartbeat check",{interval:this.config.heartbeatCheckInterval,timeout:this.config.heartbeatTimeout}),this.heartbeatTimer=setInterval(()=>this.checkHeartbeats(),this.config.heartbeatCheckInterval);}stopHeartbeatCheck(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null,f.info("Heartbeat check stopped"));}checkHeartbeats(){let e=Date.now(),{heartbeatTimeout:t,autoDeregisterTimeout:n}=this.config,s=[],r=0;for(let[i,c]of this.agents.entries()){let p=e-c.lastHeartbeat;if(p>n){s.push(i);continue}p>t&&c.status==="healthy"&&(c.status="unhealthy",r++,f.warn("Agent marked as unhealthy due to heartbeat timeout",{agentId:i,lastHeartbeat:c.lastHeartbeat,timeSinceLastHeartbeat:p,timeout:t}));}for(let i of s){let c=this.agents.get(i);c&&(f.info("Auto-deregistering agent due to timeout",{agentId:i,name:c.name,lastHeartbeat:c.lastHeartbeat,timeout:n}),this.agents.delete(i));}(r>0||s.length>0)&&f.info("Heartbeat check completed",{unhealthyCount:r,deregisteredCount:s.length,totalAgents:this.agents.size});}selectRandom(e){let t=Math.floor(Math.random()*e.length);return e[t]}selectRoundRobin(e){this.roundRobinIndex=this.roundRobinIndex%e.length;let t=e[this.roundRobinIndex];return this.roundRobinIndex++,t}selectLeastConnections(e){return e.reduce((t,n)=>n.connectionCount<t.connectionCount?n:t)}getStats(){let e=this.getAllAgents();return {totalAgents:e.length,healthyAgents:e.filter(t=>t.status==="healthy").length,unhealthyAgents:e.filter(t=>t.status==="unhealthy").length,offlineAgents:e.filter(t=>t.status==="offline").length,totalConnections:e.reduce((t,n)=>t+n.connectionCount,0)}}setAgentStatus(e,t){let n=this.agents.get(e);n&&(n.status=t);}setLastHeartbeat(e,t){let n=this.agents.get(e);n&&(n.lastHeartbeat=t);}};var h=chunkAUPHFE34_cjs.a.create({service:"gateway:agent-sync-manager"}),M=class{constructor(e){chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"registry");chunkGY3SWWW3_cjs.a(this,"localNodeId");chunkGY3SWWW3_cjs.a(this,"localNodeName");chunkGY3SWWW3_cjs.a(this,"remoteNodes",new Map);chunkGY3SWWW3_cjs.a(this,"remoteAgents",new Map);chunkGY3SWWW3_cjs.a(this,"syncTimer",null);chunkGY3SWWW3_cjs.a(this,"running",false);this.registry=e.registry,this.localNodeId=e.localNodeId,this.localNodeName=e.localNodeName,this.config={...chunkSOCQU5UV_cjs.$,...e.config};}async start(){if(this.running){h.warn("Agent sync manager is already running");return}h.info("Starting agent sync manager",{localNodeId:this.localNodeId,mode:this.config.mode,interval:this.config.interval,remoteNodes:this.config.remoteNodes.length});for(let e of this.config.remoteNodes)this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"});(this.config.mode==="pull"||this.config.mode==="both")&&(this.syncTimer=setInterval(()=>this.pullFromAllNodes(),this.config.interval)),this.running=true,h.info("Agent sync manager started");}async stop(){if(!this.running){h.warn("Agent sync manager is not running");return}h.info("Stopping agent sync manager"),this.syncTimer&&(clearInterval(this.syncTimer),this.syncTimer=null),this.remoteNodes.clear(),this.remoteAgents.clear(),this.running=false,h.info("Agent sync manager stopped");}async pullFromAllNodes(){for(let[e]of this.remoteNodes.entries())try{await this.pullFromNode(e);}catch(t){h.error("Failed to pull from node",{nodeId:e,error:t.message});}}async pullFromNode(e){let t=this.remoteNodes.get(e);if(!t)throw new Error(`Unknown remote node: ${e}`);t.status="syncing";let n=this.config.remoteNodes.find(s=>s.id===e);if(!n)throw new Error(`Remote node not found in config: ${e}`);h.debug("Pulling agent list from node",{nodeId:e,address:n.address});try{let s=`http://${n.address}:${n.port}/sync/agents`,r=await utils.Fetch.get(s,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);let i=r.data.agents||[];for(let c of i)c.sourceGatewayId=e,this.remoteAgents.set(c.id,c);return t.lastSyncAt=Date.now(),t.status="synced",t.info.lastSyncAt=t.lastSyncAt,t.info.status="online",delete t.error,h.info("Pulled agent list from node",{nodeId:e,agentCount:i.length,lastSyncAt:t.lastSyncAt}),i}catch(s){throw t.status="error",t.error=s.message,t.info.status="offline",h.error("Failed to pull from node",{nodeId:e,error:s.message}),s}}async pushEvent(e){if(this.config.mode!=="push"&&this.config.mode!=="both")return;let t=[];for(let[n]of this.remoteNodes.entries())t.push(this.pushToNode(n,e));await Promise.allSettled(t);}async pushToNode(e,t){let n=this.config.remoteNodes.find(s=>s.id===e);if(!n){h.warn("Unknown remote node for push",{nodeId:e});return}try{let s=`http://${n.address}:${n.port}/sync/events`,r=await utils.Fetch.post(s,t,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);h.debug("Pushed event to node",{nodeId:e,eventType:t.type});}catch(s){h.error("Failed to push to node",{nodeId:e,eventType:t.type,error:s.message});}}async handleSyncMessage(e){switch(h.debug("Received sync message",{type:e.type,sourceNodeId:e.sourceNodeId}),e.type){case "agent_list_request":await this.handleAgentListRequest(e);break;case "agent_list_response":await this.handleAgentListResponse(e);break;case "agent_register":await this.handleAgentRegister(e);break;case "agent_deregister":await this.handleAgentDeregister(e);break;case "agent_heartbeat":await this.handleAgentHeartbeat(e);break;case "agent_status_change":await this.handleAgentStatusChange(e);break;default:h.warn("Unknown sync message type",{type:e.type});}}async handleAgentListRequest(e){let t=this.getLocalAgentsForSync(),n={type:"agent_list_response",sourceNodeId:this.localNodeId,timestamp:Date.now(),requestId:e.requestId,agents:t,fullList:true,syncAt:Date.now()};await this.pushToNode(e.sourceNodeId,n);}async handleAgentListResponse(e){for(let t of e.agents)this.remoteAgents.set(t.id,t);h.info("Received agent list response",{sourceNodeId:e.sourceNodeId,agentCount:e.agents.length});}async handleAgentRegister(e){let{agent:t}=e,n=this.remoteAgents.get(t.id);if(n){let s=this.resolveConflict(n,t);this.remoteAgents.set(t.id,s);}else this.remoteAgents.set(t.id,t);h.info("Remote agent registered",{agentId:t.id,name:t.name,sourceNodeId:e.sourceNodeId});}async handleAgentDeregister(e){let{agentId:t}=e;this.remoteAgents.delete(t),h.info("Remote agent deregistered",{agentId:t,sourceNodeId:e.sourceNodeId});}async handleAgentHeartbeat(e){let{agentId:t,heartbeatAt:n,status:s}=e,r=this.remoteAgents.get(t);r&&(r.updatedAt=n,r.status=s),h.debug("Remote agent heartbeat received",{agentId:t,status:s,sourceNodeId:e.sourceNodeId});}async handleAgentStatusChange(e){let{agentId:t,newStatus:n,changedAt:s}=e,r=this.remoteAgents.get(t);r&&(r.status=n,r.updatedAt=s),h.info("Remote agent status changed",{agentId:t,newStatus:n,sourceNodeId:e.sourceNodeId});}resolveConflict(e,t){switch(this.config.conflictResolution){case "latest":return t.updatedAt>e.updatedAt?t:e;case "local":return e;case "remote":return t;default:return t.updatedAt>e.updatedAt?t:e}}getLocalAgentsForSync(){return this.registry.getAllAgents().map(t=>({id:t.id,name:t.name,sourceGatewayId:this.localNodeId,address:t.address,capabilities:t.capabilities,status:t.status,updatedAt:t.lastHeartbeat,metadata:t.metadata}))}getMergedAgentList(){let e=new Map;for(let t of this.getLocalAgentsForSync())e.set(t.id,t);for(let[t,n]of this.remoteAgents.entries())e.has(t)||e.set(t,n);return Array.from(e.values())}getRemoteAgents(){return Array.from(this.remoteAgents.values())}getStats(){let e=this.registry.getAllAgents(),t=0,n=0;for(let r of this.remoteNodes.values())r.info.status==="online"?t++:n++;let s=null;for(let r of this.remoteNodes.values())r.lastSyncAt>(s||0)&&(s=r.lastSyncAt);return {localAgentCount:e.length,remoteAgentCount:this.remoteAgents.size,totalAgentCount:e.length+this.remoteAgents.size,onlineNodeCount:t,offlineNodeCount:n,lastSyncAt:s}}isRunning(){return this.running}addRemoteNode(e){this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"}),h.info("Remote node added",{nodeId:e.id,address:e.address});}removeRemoteNode(e){this.remoteNodes.delete(e);for(let[t,n]of this.remoteAgents.entries())n.sourceGatewayId===e&&this.remoteAgents.delete(t);h.info("Remote node removed",{nodeId:e});}};var T=class{constructor(e){chunkGY3SWWW3_cjs.a(this,"log");chunkGY3SWWW3_cjs.a(this,"config");chunkGY3SWWW3_cjs.a(this,"ws",null);chunkGY3SWWW3_cjs.a(this,"state","disconnected");chunkGY3SWWW3_cjs.a(this,"subscriptions",new Map);chunkGY3SWWW3_cjs.a(this,"messageCallbacks",new Set);chunkGY3SWWW3_cjs.a(this,"agentListCallbacks",new Set);chunkGY3SWWW3_cjs.a(this,"cachedAgentList",[]);chunkGY3SWWW3_cjs.a(this,"localAgentList",[]);chunkGY3SWWW3_cjs.a(this,"reconnectAttempts",0);chunkGY3SWWW3_cjs.a(this,"reconnectTimer",null);chunkGY3SWWW3_cjs.a(this,"heartbeatTimer",null);chunkGY3SWWW3_cjs.a(this,"agentListRequestId",null);chunkGY3SWWW3_cjs.a(this,"agentListRequestResolve",null);this.config={url:e.url,type:e.type,id:e.id??`client_${Date.now()}_${Math.random().toString(36).substring(2,9)}`,reconnect:{enabled:e.reconnect?.enabled??true,maxAttempts:e.reconnect?.maxAttempts??5,delay:e.reconnect?.delay??3e3},heartbeat:{enabled:e.heartbeat?.enabled??true,interval:e.heartbeat?.interval??3e4}},this.log=chunkAUPHFE34_cjs.a.create({service:`gateway:client:${this.config.id}`});}async connect(){if(this.state==="connected"||this.state==="connecting"){this.log.warn("already connected or connecting");return}return this.state="connecting",this.log.info("connecting to gateway",{url:this.config.url}),new Promise((e,t)=>{try{this.ws=new WebSocket(this.config.url),this.ws.onopen=()=>{this.state="connected",this.reconnectAttempts=0,this.log.info("connected to gateway"),this.config.heartbeat.enabled&&this.startHeartbeat(),this.resubscribeAll(),e();},this.ws.onmessage=n=>{this.handleMessage(n.data);},this.ws.onclose=n=>{this.handleClose(n.code,n.reason);},this.ws.onerror=n=>{this.log.error("WebSocket error",{error:String(n)}),this.state==="connecting"&&t(new Error("Connection failed"));};}catch(n){this.state="disconnected",t(n);}})}async disconnect(){this.state!=="disconnected"&&(this.log.info("disconnecting from gateway"),this.stopHeartbeat(),this.stopReconnect(),this.ws&&(this.ws.close(1e3,"Client disconnect"),this.ws=null),this.state="disconnected",this.log.info("disconnected from gateway"));}async subscribe(e,t){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"subscribe",sessionId:e,backendSessionId:t,clientId:this.config.id}),this.log.info("subscribed to session",{sessionId:e,backendSessionId:t});}async unsubscribe(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"unsubscribe",sessionId:e,clientId:this.config.id}),this.subscriptions.delete(e),this.log.info("unsubscribed from session",{sessionId:e});}async send(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"message",message:e});}onMessage(e){this.messageCallbacks.add(e);}offMessage(e){this.messageCallbacks.delete(e);}getState(){return this.state}getId(){return this.config.id}getSubscriptions(){return [...this.subscriptions.keys()]}setLocalAgents(e){this.localAgentList=e,this.updateMergedAgentList();}async fetchAgentList(){if(this.state!=="connected")throw new Error("Not connected to gateway");return new Promise((e,t)=>{let n=`req_${Date.now()}_${Math.random().toString(36).substring(2,9)}`;this.agentListRequestId=n,this.agentListRequestResolve=e,this.sendData({type:"agent_list_request",requestId:n}),setTimeout(()=>{this.agentListRequestId===n&&(this.agentListRequestId=null,this.agentListRequestResolve=null,t(new Error("Agent list request timeout")));},1e4);})}getConnectableAgents(e={}){let{localAgents:t,preferLocal:n=true,filterOffline:s=true}=e,r=t||this.localAgentList,i=this.cachedAgentList,c=this.mergeAgentLists(r,i,n);return s?c.filter(p=>p.status!=="offline"):c}onAgentListChange(e){this.agentListCallbacks.add(e);}offAgentListChange(e){this.agentListCallbacks.delete(e);}mergeAgentLists(e,t,n){let s=new Map,r=n?t:e,i=n?e:t;for(let c of r)s.set(c.id,c);for(let c of i)s.set(c.id,c);return Array.from(s.values())}updateMergedAgentList(){let e=this.getConnectableAgents({filterOffline:false});for(let t of this.agentListCallbacks)try{t(e);}catch(n){this.log.error("agent list callback error",{error:String(n)});}}handleAgentListResponse(e,t){t&&this.agentListRequestId===t&&this.agentListRequestResolve&&(this.agentListRequestId=null,this.agentListRequestResolve(e),this.agentListRequestResolve=null),this.cachedAgentList=e,this.updateMergedAgentList(),this.log.debug("received agent list",{count:e.length});}sendData(e){if(!this.ws||this.ws.readyState!==WebSocket.OPEN){this.log.warn("WebSocket not ready, cannot send");return}this.ws.send(JSON.stringify(e));}handleMessage(e){try{let t=JSON.parse(e);switch(t.type){case "message":this.handleGatewayMessage(t.message);break;case "subscribed":this.subscriptions.set(t.sessionId,t.subscription);break;case "unsubscribed":this.subscriptions.delete(t.sessionId);break;case "pong":break;case "agent_list_response":this.handleAgentListResponse(t.agents||[],t.requestId);break;case "agent_update":this.handleAgentUpdate(t.agent,t.action);break;default:this.log.debug("unknown message type",{type:t.type});}}catch(t){this.log.error("failed to parse message",{error:String(t)});}}handleAgentUpdate(e,t){switch(t){case "add":case "update":{let n=this.cachedAgentList.findIndex(s=>s.id===e.id);n>=0?this.cachedAgentList[n]=e:this.cachedAgentList.push(e);break}case "remove":this.cachedAgentList=this.cachedAgentList.filter(n=>n.id!==e.id);break}this.updateMergedAgentList();}handleGatewayMessage(e){for(let t of this.messageCallbacks)try{t(e);}catch(n){this.log.error("message callback error",{error:String(n)});}}handleClose(e,t){this.log.info("connection closed",{code:e,reason:t}),this.state="disconnected",this.ws=null,this.stopHeartbeat(),this.config.reconnect.enabled&&this.scheduleReconnect();}scheduleReconnect(){if(this.reconnectAttempts>=this.config.reconnect.maxAttempts){this.log.error("max reconnect attempts reached");return}this.reconnectAttempts++,this.state="reconnecting";let e=this.config.reconnect.delay*this.reconnectAttempts;this.log.info("scheduling reconnect",{attempt:this.reconnectAttempts,delay:e}),this.reconnectTimer=setTimeout(async()=>{try{await this.connect();}catch(t){this.log.error("reconnect failed",{error:String(t)});}},e);}stopReconnect(){this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null);}startHeartbeat(){this.heartbeatTimer=setInterval(()=>{this.state==="connected"&&this.sendData({type:"ping"});},this.config.heartbeat.interval);}stopHeartbeat(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null);}async resubscribeAll(){for(let e of this.subscriptions.keys())try{await this.subscribe(e);}catch(t){this.log.error("failed to resubscribe",{sessionId:e,error:String(t)});}}getHttpUrl(){return this.config.url.replace(/^ws:/,"http:").replace(/^wss:/,"https:").replace(/\/$/,"")}async httpRequest(e,t={}){let n=`${this.getHttpUrl()}${e}`,s=t.method||"GET",{signal:r,...i}=t,c=await utils.Fetch.request(n,{method:s,...i,headers:{"Content-Type":"application/json",...i.headers},...r?{signal:r}:{}});if(!c.ok){let p=c.data,x=typeof p=="object"?JSON.stringify(p):String(p||"Unknown error");throw new Error(`HTTP ${c.status}: ${x}`)}return c.data}async healthCheck(){return this.httpRequest("/health")}async getStatus(){return this.httpRequest("/status")}async listChannels(){return this.httpRequest("/channels")}async listContacts(e,t={}){let n=new URLSearchParams({platform:e,limit:String(t.limit??100),offset:String(t.offset??0)});return this.httpRequest(`/contacts?${n}`)}async getContact(e){return this.httpRequest(`/contacts/${encodeURIComponent(e)}`)}async getSession(e){return this.httpRequest(`/sessions/${encodeURIComponent(e)}`)}async createSession(e){return this.httpRequest("/session",{method:"POST",body:JSON.stringify(e??{})})}async getOrCreateSession(e){if(e.sessionId)try{return await this.getSession(e.sessionId),{sessionId:e.sessionId,created:!1}}catch{this.log.debug("session not found, will create new",{sessionId:e.sessionId});}if(e.platform||e.channel||e.userId){try{let r=await this.listSessions({platform:e.platform,channelId:e.channel,userId:e.userId});if(r.sessions.length>0){let i=r.sessions[0];if(i)return this.log.info("found existing session",{sessionId:i.id,channel:e.channel,userId:e.userId}),{sessionId:i.id,created:!1}}}catch(r){this.log.warn("failed to search sessions",{error:String(r)});}let n={};e.channel&&(n.channel=e.channel),e.userId&&(n.userId=e.userId),e.platform&&(n.platform=e.platform);let s=await this.createSession(n);return this.log.info("session created",{sessionId:s.id,params:n}),{sessionId:s.id,created:true}}let t=await this.createSession({});return this.log.info("session created (default)",{sessionId:t.id}),{sessionId:t.id,created:true}}async listSessions(e={}){let t=new URLSearchParams;return t.set("limit",String(e.limit??100)),t.set("offset",String(e.offset??0)),e.platform&&t.set("platform",e.platform),e.channelId&&t.set("channelId",e.channelId),e.userId&&t.set("userId",e.userId),this.httpRequest(`/sessions?${t}`)}async sendMessage(e,t){return this.httpRequest(`/session/${encodeURIComponent(e)}/prompt`,{method:"POST",body:JSON.stringify(t)})}};async function zt(g){let{Log:e}=await import('./chunks/log-JFEBRH2W.cjs'),t=false;await e.init({logDir:g.logDir??process.env.EASBOT_LOG_PATH??process.cwd(),print:g.print??false,dev:g.dev??t,level:g.level??("INFO")});}
|
|
2
|
+
Object.defineProperty(exports,"AgentRegistryConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.p}});Object.defineProperty(exports,"AgentSyncConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.k}});Object.defineProperty(exports,"AuthConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.j}});Object.defineProperty(exports,"BaseChannelPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ha}});Object.defineProperty(exports,"ChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.h}});Object.defineProperty(exports,"ChannelPluginLoader",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.fa}});Object.defineProperty(exports,"ChannelPluginRegistry",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ra}});Object.defineProperty(exports,"CircuitBreakerConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.o}});Object.defineProperty(exports,"ConnectionPoolConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.m}});Object.defineProperty(exports,"DEFAULT_AGENT_REGISTRY_CONFIG",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs._}});Object.defineProperty(exports,"DEFAULT_CONNECTION_POOL_CONFIG",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.aa}});Object.defineProperty(exports,"DEFAULT_SYNC_CONFIG",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.$}});Object.defineProperty(exports,"DEFAULT_TOKEN_AUTH_CONFIG",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.Z}});Object.defineProperty(exports,"DEFAULT_WEBSOCKET_SERVER_CONFIG",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.Y}});Object.defineProperty(exports,"DiscordPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ja}});Object.defineProperty(exports,"FeishuChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.c}});Object.defineProperty(exports,"FeishuPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.la}});Object.defineProperty(exports,"GatewayClusterConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.l}});Object.defineProperty(exports,"GatewayConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.s}});Object.defineProperty(exports,"GatewayServer",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ta}});Object.defineProperty(exports,"GatewayServerConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.r}});Object.defineProperty(exports,"GatewaySessionManager",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ea}});Object.defineProperty(exports,"HTTPSConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.i}});Object.defineProperty(exports,"MessageLockManager",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.R}});Object.defineProperty(exports,"MessageQueueConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.n}});Object.defineProperty(exports,"MessageRetryManager",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.S}});Object.defineProperty(exports,"MessageRouter",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ba}});Object.defineProperty(exports,"MessageStore",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ca}});Object.defineProperty(exports,"NostrChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.g}});Object.defineProperty(exports,"NostrPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.pa}});Object.defineProperty(exports,"SessionConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.q}});Object.defineProperty(exports,"SessionStore",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.da}});Object.defineProperty(exports,"SignalChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.f}});Object.defineProperty(exports,"SignalPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.oa}});Object.defineProperty(exports,"SlackChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.b}});Object.defineProperty(exports,"SlackPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ka}});Object.defineProperty(exports,"TelegramChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.a}});Object.defineProperty(exports,"TelegramPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ia}});Object.defineProperty(exports,"WeChatChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.d}});Object.defineProperty(exports,"WeChatPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.ma}});Object.defineProperty(exports,"WebChatChannelConfigSchema",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.e}});Object.defineProperty(exports,"WebChatPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.na}});Object.defineProperty(exports,"clearConfigCache",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.y}});Object.defineProperty(exports,"createChannelPlugin",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.qa}});Object.defineProperty(exports,"createDefaultSessionState",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.W}});Object.defineProperty(exports,"createGatewayMessage",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.T}});Object.defineProperty(exports,"createTextMessage",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.U}});Object.defineProperty(exports,"generateSessionId",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.V}});Object.defineProperty(exports,"generateSubscriptionId",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.X}});Object.defineProperty(exports,"getAgentAdapter",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.N}});Object.defineProperty(exports,"getAgentRegistryConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.F}});Object.defineProperty(exports,"getAuthConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.I}});Object.defineProperty(exports,"getChannelConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.B}});Object.defineProperty(exports,"getCircuitBreakerConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.E}});Object.defineProperty(exports,"getClusterConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.J}});Object.defineProperty(exports,"getConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.w}});Object.defineProperty(exports,"getConfigDirectory",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.z}});Object.defineProperty(exports,"getConnectionPoolConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.C}});Object.defineProperty(exports,"getDefaultAgent",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.K}});Object.defineProperty(exports,"getHTTPSConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.H}});Object.defineProperty(exports,"getLogLevel",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.L}});Object.defineProperty(exports,"getMessageQueueConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.D}});Object.defineProperty(exports,"getServerConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.A}});Object.defineProperty(exports,"getSessionConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.G}});Object.defineProperty(exports,"getSupportedPlatforms",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.sa}});Object.defineProperty(exports,"hasAgentAdapter",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.O}});Object.defineProperty(exports,"isConfigLoaded",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.x}});Object.defineProperty(exports,"loadConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.v}});Object.defineProperty(exports,"parsePartialConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.u}});Object.defineProperty(exports,"requireAgentAdapter",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.P}});Object.defineProperty(exports,"setAgentAdapter",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.M}});Object.defineProperty(exports,"validateConfig",{enumerable:true,get:function(){return chunkSOCQU5UV_cjs.t}});exports.AgentRegistry=P;exports.AgentSyncManager=M;exports.GatewayClient=T;exports.initLog=zt;
|
package/dist/index.d.cts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { IGatewayServer, GatewayConfig as GatewayConfig$1, GatewayServerConfig as GatewayServerConfig$1, PlatformType, GatewayMessage, SubAgentConfig, ChannelInfo, AgentAdapter } from '@easbot/types';
|
|
1
|
+
import { IGatewayServer, GatewayConfig as GatewayConfig$1, GatewayServerConfig as GatewayServerConfig$1, PlatformType, GatewayMessage, SubAgentConfig, ChannelInfo, ChannelListResponse, AgentAdapter } from '@easbot/types';
|
|
2
2
|
export { AgentAdapter, AgentInfo, ChannelInfo, GatewayMessage, GatewayMessagePromptPart, IGatewayServer, IGlobal, IHookRegistry, IInstance, ISubAgentRunner, MessageContext, MessageMetadata, MessageStorage, MessageType, PlatformType } from '@easbot/types';
|
|
3
3
|
import z from 'zod';
|
|
4
4
|
import { WSContext } from '@easbot/utils';
|
|
@@ -1656,16 +1656,23 @@ declare class MessageStore {
|
|
|
1656
1656
|
private sessionMessages;
|
|
1657
1657
|
private cleanupTimer;
|
|
1658
1658
|
private accessOrder;
|
|
1659
|
+
private storagePath;
|
|
1660
|
+
private dirty;
|
|
1659
1661
|
constructor(config?: Partial<MessageStoreConfig>);
|
|
1662
|
+
private ensureDir;
|
|
1663
|
+
private getMessageFilePath;
|
|
1664
|
+
private getSessionDir;
|
|
1665
|
+
private sanitizeFileName;
|
|
1660
1666
|
private startCleanupTimer;
|
|
1661
1667
|
private performCleanup;
|
|
1662
1668
|
store(message: GatewayMessage): Promise<MessageRecord>;
|
|
1669
|
+
private saveMessageToFile;
|
|
1663
1670
|
get(messageId: string): MessageRecord | undefined;
|
|
1664
1671
|
updateStatus(messageId: string, status: MessageStatus, options?: {
|
|
1665
1672
|
processedBy?: string;
|
|
1666
1673
|
error?: string;
|
|
1667
1674
|
}): Promise<void>;
|
|
1668
|
-
incrementRetry(messageId: string): number
|
|
1675
|
+
incrementRetry(messageId: string): Promise<number>;
|
|
1669
1676
|
isProcessed(messageId: string): boolean;
|
|
1670
1677
|
getSessionHistory(sessionId: string, limit?: number): MessageRecord[];
|
|
1671
1678
|
getPendingMessages(sessionId?: string): MessageRecord[];
|
|
@@ -1776,14 +1783,8 @@ interface ResolveChannelParams {
|
|
|
1776
1783
|
bindBackendSession(gatewaySessionId: string, backendSessionId: string): Promise<void>;
|
|
1777
1784
|
};
|
|
1778
1785
|
pluginLoader?: {
|
|
1779
|
-
getPlugin(pluginId: string):
|
|
1780
|
-
|
|
1781
|
-
platform: PlatformType;
|
|
1782
|
-
} | undefined;
|
|
1783
|
-
getAllPlugins(): Array<{
|
|
1784
|
-
id: string;
|
|
1785
|
-
platform: PlatformType;
|
|
1786
|
-
}>;
|
|
1786
|
+
getPlugin(pluginId: string): ChannelPlugin | undefined;
|
|
1787
|
+
getAllPlugins(): ChannelPlugin[];
|
|
1787
1788
|
isRunning(pluginId: string): boolean;
|
|
1788
1789
|
};
|
|
1789
1790
|
contactStore?: {
|
|
@@ -2061,6 +2062,7 @@ declare class GatewayClient {
|
|
|
2061
2062
|
private httpRequest;
|
|
2062
2063
|
healthCheck(): Promise<HealthResponse>;
|
|
2063
2064
|
getStatus(): Promise<StatusResponse>;
|
|
2065
|
+
listChannels(): Promise<ChannelListResponse>;
|
|
2064
2066
|
listContacts(platform: string, options?: {
|
|
2065
2067
|
limit?: number;
|
|
2066
2068
|
offset?: number;
|
|
@@ -2068,16 +2070,30 @@ declare class GatewayClient {
|
|
|
2068
2070
|
getContact(uid: string): Promise<ContactInfo & {
|
|
2069
2071
|
aliases?: Record<string, unknown>;
|
|
2070
2072
|
}>;
|
|
2071
|
-
listSessions(options?: {
|
|
2072
|
-
limit?: number;
|
|
2073
|
-
offset?: number;
|
|
2074
|
-
}): Promise<SessionListResponse>;
|
|
2075
2073
|
getSession(sessionId: string): Promise<SessionInfo & {
|
|
2076
2074
|
subscriberCount?: number;
|
|
2077
2075
|
}>;
|
|
2078
2076
|
createSession(params?: {
|
|
2079
2077
|
channel?: string;
|
|
2078
|
+
userId?: string;
|
|
2079
|
+
platform?: string;
|
|
2080
2080
|
}): Promise<CreateSessionResponse>;
|
|
2081
|
+
getOrCreateSession(options: {
|
|
2082
|
+
sessionId?: string;
|
|
2083
|
+
channel?: string;
|
|
2084
|
+
userId?: string;
|
|
2085
|
+
platform?: string;
|
|
2086
|
+
}): Promise<{
|
|
2087
|
+
sessionId: string;
|
|
2088
|
+
created: boolean;
|
|
2089
|
+
}>;
|
|
2090
|
+
listSessions(options?: {
|
|
2091
|
+
limit?: number;
|
|
2092
|
+
offset?: number;
|
|
2093
|
+
platform?: string;
|
|
2094
|
+
channelId?: string;
|
|
2095
|
+
userId?: string;
|
|
2096
|
+
}): Promise<SessionListResponse>;
|
|
2081
2097
|
sendMessage(sessionId: string, message: {
|
|
2082
2098
|
content: Array<{
|
|
2083
2099
|
type: string;
|
package/dist/index.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { IGatewayServer, GatewayConfig as GatewayConfig$1, GatewayServerConfig as GatewayServerConfig$1, PlatformType, GatewayMessage, SubAgentConfig, ChannelInfo, AgentAdapter } from '@easbot/types';
|
|
1
|
+
import { IGatewayServer, GatewayConfig as GatewayConfig$1, GatewayServerConfig as GatewayServerConfig$1, PlatformType, GatewayMessage, SubAgentConfig, ChannelInfo, ChannelListResponse, AgentAdapter } from '@easbot/types';
|
|
2
2
|
export { AgentAdapter, AgentInfo, ChannelInfo, GatewayMessage, GatewayMessagePromptPart, IGatewayServer, IGlobal, IHookRegistry, IInstance, ISubAgentRunner, MessageContext, MessageMetadata, MessageStorage, MessageType, PlatformType } from '@easbot/types';
|
|
3
3
|
import z from 'zod';
|
|
4
4
|
import { WSContext } from '@easbot/utils';
|
|
@@ -1656,16 +1656,23 @@ declare class MessageStore {
|
|
|
1656
1656
|
private sessionMessages;
|
|
1657
1657
|
private cleanupTimer;
|
|
1658
1658
|
private accessOrder;
|
|
1659
|
+
private storagePath;
|
|
1660
|
+
private dirty;
|
|
1659
1661
|
constructor(config?: Partial<MessageStoreConfig>);
|
|
1662
|
+
private ensureDir;
|
|
1663
|
+
private getMessageFilePath;
|
|
1664
|
+
private getSessionDir;
|
|
1665
|
+
private sanitizeFileName;
|
|
1660
1666
|
private startCleanupTimer;
|
|
1661
1667
|
private performCleanup;
|
|
1662
1668
|
store(message: GatewayMessage): Promise<MessageRecord>;
|
|
1669
|
+
private saveMessageToFile;
|
|
1663
1670
|
get(messageId: string): MessageRecord | undefined;
|
|
1664
1671
|
updateStatus(messageId: string, status: MessageStatus, options?: {
|
|
1665
1672
|
processedBy?: string;
|
|
1666
1673
|
error?: string;
|
|
1667
1674
|
}): Promise<void>;
|
|
1668
|
-
incrementRetry(messageId: string): number
|
|
1675
|
+
incrementRetry(messageId: string): Promise<number>;
|
|
1669
1676
|
isProcessed(messageId: string): boolean;
|
|
1670
1677
|
getSessionHistory(sessionId: string, limit?: number): MessageRecord[];
|
|
1671
1678
|
getPendingMessages(sessionId?: string): MessageRecord[];
|
|
@@ -1776,14 +1783,8 @@ interface ResolveChannelParams {
|
|
|
1776
1783
|
bindBackendSession(gatewaySessionId: string, backendSessionId: string): Promise<void>;
|
|
1777
1784
|
};
|
|
1778
1785
|
pluginLoader?: {
|
|
1779
|
-
getPlugin(pluginId: string):
|
|
1780
|
-
|
|
1781
|
-
platform: PlatformType;
|
|
1782
|
-
} | undefined;
|
|
1783
|
-
getAllPlugins(): Array<{
|
|
1784
|
-
id: string;
|
|
1785
|
-
platform: PlatformType;
|
|
1786
|
-
}>;
|
|
1786
|
+
getPlugin(pluginId: string): ChannelPlugin | undefined;
|
|
1787
|
+
getAllPlugins(): ChannelPlugin[];
|
|
1787
1788
|
isRunning(pluginId: string): boolean;
|
|
1788
1789
|
};
|
|
1789
1790
|
contactStore?: {
|
|
@@ -2061,6 +2062,7 @@ declare class GatewayClient {
|
|
|
2061
2062
|
private httpRequest;
|
|
2062
2063
|
healthCheck(): Promise<HealthResponse>;
|
|
2063
2064
|
getStatus(): Promise<StatusResponse>;
|
|
2065
|
+
listChannels(): Promise<ChannelListResponse>;
|
|
2064
2066
|
listContacts(platform: string, options?: {
|
|
2065
2067
|
limit?: number;
|
|
2066
2068
|
offset?: number;
|
|
@@ -2068,16 +2070,30 @@ declare class GatewayClient {
|
|
|
2068
2070
|
getContact(uid: string): Promise<ContactInfo & {
|
|
2069
2071
|
aliases?: Record<string, unknown>;
|
|
2070
2072
|
}>;
|
|
2071
|
-
listSessions(options?: {
|
|
2072
|
-
limit?: number;
|
|
2073
|
-
offset?: number;
|
|
2074
|
-
}): Promise<SessionListResponse>;
|
|
2075
2073
|
getSession(sessionId: string): Promise<SessionInfo & {
|
|
2076
2074
|
subscriberCount?: number;
|
|
2077
2075
|
}>;
|
|
2078
2076
|
createSession(params?: {
|
|
2079
2077
|
channel?: string;
|
|
2078
|
+
userId?: string;
|
|
2079
|
+
platform?: string;
|
|
2080
2080
|
}): Promise<CreateSessionResponse>;
|
|
2081
|
+
getOrCreateSession(options: {
|
|
2082
|
+
sessionId?: string;
|
|
2083
|
+
channel?: string;
|
|
2084
|
+
userId?: string;
|
|
2085
|
+
platform?: string;
|
|
2086
|
+
}): Promise<{
|
|
2087
|
+
sessionId: string;
|
|
2088
|
+
created: boolean;
|
|
2089
|
+
}>;
|
|
2090
|
+
listSessions(options?: {
|
|
2091
|
+
limit?: number;
|
|
2092
|
+
offset?: number;
|
|
2093
|
+
platform?: string;
|
|
2094
|
+
channelId?: string;
|
|
2095
|
+
userId?: string;
|
|
2096
|
+
}): Promise<SessionListResponse>;
|
|
2081
2097
|
sendMessage(sessionId: string, message: {
|
|
2082
2098
|
content: Array<{
|
|
2083
2099
|
type: string;
|
package/dist/index.mjs
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
import {v,Q,_,$}from'./chunks/chunk-GQFBXITJ.mjs';export{p as AgentRegistryConfigSchema,k as AgentSyncConfigSchema,j as AuthConfigSchema,ha as BaseChannelPlugin,h as ChannelConfigSchema,fa as ChannelPluginLoader,ra as ChannelPluginRegistry,o as CircuitBreakerConfigSchema,m as ConnectionPoolConfigSchema,_ as DEFAULT_AGENT_REGISTRY_CONFIG,aa as DEFAULT_CONNECTION_POOL_CONFIG,$ as DEFAULT_SYNC_CONFIG,Z as DEFAULT_TOKEN_AUTH_CONFIG,Y as DEFAULT_WEBSOCKET_SERVER_CONFIG,ja as DiscordPlugin,c as FeishuChannelConfigSchema,la as FeishuPlugin,l as GatewayClusterConfigSchema,s as GatewayConfigSchema,ta as GatewayServer,r as GatewayServerConfigSchema,ea as GatewaySessionManager,i as HTTPSConfigSchema,R as MessageLockManager,n as MessageQueueConfigSchema,S as MessageRetryManager,ba as MessageRouter,ca as MessageStore,g as NostrChannelConfigSchema,pa as NostrPlugin,q as SessionConfigSchema,da as SessionStore,f as SignalChannelConfigSchema,oa as SignalPlugin,b as SlackChannelConfigSchema,ka as SlackPlugin,a as TelegramChannelConfigSchema,ia as TelegramPlugin,d as WeChatChannelConfigSchema,ma as WeChatPlugin,e as WebChatChannelConfigSchema,na as WebChatPlugin,y as clearConfigCache,qa as createChannelPlugin,W as createDefaultSessionState,T as createGatewayMessage,U as createTextMessage,V as generateSessionId,X as generateSubscriptionId,N as getAgentAdapter,F as getAgentRegistryConfig,I as getAuthConfig,B as getChannelConfig,E as getCircuitBreakerConfig,J as getClusterConfig,w as getConfig,z as getConfigDirectory,C as getConnectionPoolConfig,K as getDefaultAgent,H as getHTTPSConfig,L as getLogLevel,D as getMessageQueueConfig,A as getServerConfig,G as getSessionConfig,sa as getSupportedPlatforms,O as hasAgentAdapter,x as isConfigLoaded,v as loadConfig,u as parsePartialConfig,P as requireAgentAdapter,M as setAgentAdapter,t as validateConfig}from'./chunks/chunk-GQFBXITJ.mjs';import {a as a$1}from'./chunks/chunk-OBW6CNOG.mjs';import {a as a$2}from'./chunks/chunk-HAMGVOQD.mjs';import {Fetch}from'@easbot/utils';var a=a$1.create({service:"gateway"}),m=class i extends Error{constructor(t,n,s){super(t);a$2(this,"type");a$2(this,"stage");a$2(this,"cause");this.type="GatewayInitializationError",this.stage=n,this.cause=s,Error.captureStackTrace&&Error.captureStackTrace(this,i);}};function G(i){let e=["ECONNRESET","EPIPE","ETIMEDOUT","ECONNREFUSED","ENOTFOUND","EHOSTUNREACH","ENETUNREACH"],t=i.code;return e.includes(t??"")}function et(){process.on("unhandledRejection",i=>{let e=i instanceof Error?i:new Error(String(i));G(e)||a.error("unhandled promise rejection",{error:e.message,stack:e.stack});}),process.on("uncaughtException",i=>{G(i)||(a.error("uncaught exception",{error:i.message,stack:i.stack}),process.exit(1));});}var x;(at=>{let i={initialized:false,initPromise:null,directory:".easbot",config:null},e,t={server:null,config:null,initialized:false,status:"stopped"};async function n(u={}){if(i.initialized)return;if(i.initPromise)return i.initPromise;let d={directory:u.directory??".easbot",printLogs:u.printLogs??false,logLevel:u.logLevel??"INFO"};return i.directory=d.directory,i.initPromise=(async()=>{let{Global:b}=await import('./chunks/global-DVGSJ3V3.mjs');try{await b.init();}catch(g){throw new m("Failed to initialize global directories","global",g instanceof Error?g:void 0)}try{await a$1.init({logDir:b.Path.log,print:d.printLogs,level:d.logLevel});}catch(g){throw new m("Failed to initialize log infrastructure","log",g instanceof Error?g:void 0)}try{et();}catch(g){throw new m("Failed to install unhandled exception handlers","server",g instanceof Error?g:void 0)}Fetch.hasProxyConfigured()&&(Fetch.enableProxy({connectTimeout:1e4,keepAliveTimeout:3e4}),a.info("global proxy enabled",{proxyUrl:Fetch.getProxyUrl()}));try{i.config=await v(d.directory);}catch(g){throw new m("Failed to load gateway configuration","config",g instanceof Error?g:void 0)}i.initialized=true,a.info("gateway runtime initialized",{directory:d.directory,logLevel:d.logLevel});})(),i.initPromise}at.init=n;function s(){return i.initialized}at.isInitialized=s;function r(){return e||(e=async()=>{if(!i.initialized)throw new m("Gateway not initialized. Call init() first.","server");let u=Q(),d=await v(u.directory);return d.server?.enabled?(t={server:null,config:d,initialized:true,status:"stopped"},a.info("gateway initialized",{port:d.server?.port,hostname:d.server?.hostname}),t):(a.debug("gateway server disabled or not configured"),{server:null,config:d,initialized:true,status:"stopped"})}),e}function c(){return r()()}at.state=c;async function l(){return (await c()).server}at.get=l;async function y(){return (await c()).config?.server?.enabled??false}at.isEnabled=y;async function tt(){return (await c()).config}at.config=tt;function nt(){return t.status}at.getStatus=nt;async function _(u){a.debug("Gateway.start: called"),i.initialized||(a.debug("Gateway.start: calling init()"),await n(),a.debug("Gateway.start: init() completed")),a.debug("Gateway.start: calling state()");let d=await c();if(a.debug("Gateway.start: state() completed",{status:d.status}),!(d.config?.server?.enabled??true))throw a.warn("gateway server is disabled"),new Error("Gateway server is disabled");if(t.status==="running"){a.info("gateway server is already running");return}if(t.status==="starting"){a.info("gateway server is starting");return}t.status="starting",a.debug("Gateway.start: status set to starting");try{let g={...d.config?.server,...u};a.debug("gateway start: creating server with config",{port:g.port,hostname:g.hostname,path:g.path,https:g.https?.enabled}),a.debug("Gateway.start: calling createGatewayServer");let R=await O(g);a.debug("Gateway.start: createGatewayServer completed"),a.debug("gateway start: server created, updating state"),t.server=R,t.status="running",t.error=void 0,a.info("gateway server started",{port:g.port,hostname:g.hostname});}catch(g){throw t.status="error",t.error=g instanceof Error?g.message:String(g),a.error("failed to start gateway server",{error:t.error}),g}}at.start=_;async function F(){if(t.status==="stopped"){a.info("gateway server is already stopped");return}if(t.status==="stopping"){a.info("gateway server is stopping");return}t.status="stopping";try{t.server&&await t.server.stop(),t.server=null,t.status="stopped",t.error=void 0,a.info("gateway server stopped");}catch(u){throw t.status="error",t.error=u instanceof Error?u.message:String(u),a.error("failed to stop gateway server",{error:t.error}),u}}at.stop=F;async function st(u){a.info("restarting gateway server"),(t.status==="running"||t.status==="starting")&&await F(),await _(u);}at.restart=st;async function rt(){let u=Q(),d=await v(u.directory);return t.config=d,i.config=d,a.info("gateway config reloaded"),d}at.reloadConfig=rt;async function O(u,d){a.debug("createGatewayServer: starting");let{GatewayServer:b}=await import('./chunks/server-HGRZI7JT.mjs');a.debug("createGatewayServer: GatewayServer imported");let g=new b(u);a.debug("createGatewayServer: server instance created"),a.debug("createGatewayServer: created server instance, calling start()");let R=d?.startupTimeout??3e4;if(await(async()=>{let A=new Promise((v,D)=>{setTimeout(()=>{D(new Error(`Gateway server startup timeout after ${R}ms`));},R);});try{a.debug("createGatewayServer: calling server.start()"),await Promise.race([g.start(),A]),a.debug("createGatewayServer: server.start() completed");}catch(v){a.error("gateway server start failed, attempting cleanup",{error:v instanceof Error?v.message:String(v)});try{await g.stop();}catch{}throw v}})(),d?.onStarted)try{await d.onStarted(g);}catch(A){a.warn("server started but onStarted callback failed",{error:A instanceof Error?A.message:String(A)});}return a.info("gateway server created and started",{port:u.port,hostname:u.hostname}),g}at.createGatewayServer=O;})(x||(x={}));var f=a$1.create({service:"gateway:agent-registry"}),M=class{constructor(e={}){a$2(this,"config");a$2(this,"agents",new Map);a$2(this,"heartbeatTimer",null);a$2(this,"roundRobinIndex",0);this.config={..._,...e};}async register(e){if(this.agents.has(e.id))throw new Error(`Agent with ID "${e.id}" is already registered`);if(this.agents.size>=this.config.maxAgents)throw new Error(`Maximum number of agents (${this.config.maxAgents}) reached`);let t=Date.now(),n={...e,status:"healthy",lastHeartbeat:t,registeredAt:t,connectionCount:0};this.agents.set(e.id,n),f.info("Agent registered",{agentId:e.id,name:e.name,capabilities:e.capabilities,address:e.address,totalAgents:this.agents.size});}async deregister(e){let t=this.agents.get(e);if(!t){f.warn("Attempted to deregister unknown agent",{agentId:e});return}this.agents.delete(e),f.info("Agent deregistered",{agentId:e,name:t.name,totalAgents:this.agents.size});}async heartbeat(e,t){let n=this.agents.get(e);if(!n)throw f.warn("Heartbeat received from unknown agent",{agentId:e}),new Error(`Agent "${e}" is not registered`);n.lastHeartbeat=Date.now(),n.status="healthy",t&&(n.metadata={...n.metadata,...t}),f.debug("Agent heartbeat received",{agentId:e,lastHeartbeat:n.lastHeartbeat});}getAgent(e){return this.agents.get(e)}getAllAgents(){return Array.from(this.agents.values())}getHealthyAgents(){return this.getAllAgents().filter(e=>e.status==="healthy")}selectAgent(e){let{strategy:t,capabilities:n,healthyOnly:s=true}=e,r=s?this.getHealthyAgents():this.getAllAgents();if(n&&n.length>0&&(r=r.filter(l=>n.every(y=>l.capabilities.includes(y)))),r.length===0){f.warn("No agents available for selection",{strategy:t,capabilities:n,healthyOnly:s,totalAgents:this.agents.size});return}let c;switch(t){case "random":c=this.selectRandom(r);break;case "round-robin":c=this.selectRoundRobin(r);break;case "least-connections":c=this.selectLeastConnections(r);break;default:c=this.selectRandom(r);}return f.debug("Agent selected",{agentId:c.id,strategy:t,connectionCount:c.connectionCount}),c}incrementConnection(e){let t=this.agents.get(e);t&&(t.connectionCount++,f.debug("Agent connection incremented",{agentId:e,connectionCount:t.connectionCount}));}decrementConnection(e){let t=this.agents.get(e);t&&t.connectionCount>0&&(t.connectionCount--,f.debug("Agent connection decremented",{agentId:e,connectionCount:t.connectionCount}));}startHeartbeatCheck(){if(this.heartbeatTimer){f.warn("Heartbeat check is already running");return}f.info("Starting heartbeat check",{interval:this.config.heartbeatCheckInterval,timeout:this.config.heartbeatTimeout}),this.heartbeatTimer=setInterval(()=>this.checkHeartbeats(),this.config.heartbeatCheckInterval);}stopHeartbeatCheck(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null,f.info("Heartbeat check stopped"));}checkHeartbeats(){let e=Date.now(),{heartbeatTimeout:t,autoDeregisterTimeout:n}=this.config,s=[],r=0;for(let[c,l]of this.agents.entries()){let y=e-l.lastHeartbeat;if(y>n){s.push(c);continue}y>t&&l.status==="healthy"&&(l.status="unhealthy",r++,f.warn("Agent marked as unhealthy due to heartbeat timeout",{agentId:c,lastHeartbeat:l.lastHeartbeat,timeSinceLastHeartbeat:y,timeout:t}));}for(let c of s){let l=this.agents.get(c);l&&(f.info("Auto-deregistering agent due to timeout",{agentId:c,name:l.name,lastHeartbeat:l.lastHeartbeat,timeout:n}),this.agents.delete(c));}(r>0||s.length>0)&&f.info("Heartbeat check completed",{unhealthyCount:r,deregisteredCount:s.length,totalAgents:this.agents.size});}selectRandom(e){let t=Math.floor(Math.random()*e.length);return e[t]}selectRoundRobin(e){this.roundRobinIndex=this.roundRobinIndex%e.length;let t=e[this.roundRobinIndex];return this.roundRobinIndex++,t}selectLeastConnections(e){return e.reduce((t,n)=>n.connectionCount<t.connectionCount?n:t)}getStats(){let e=this.getAllAgents();return {totalAgents:e.length,healthyAgents:e.filter(t=>t.status==="healthy").length,unhealthyAgents:e.filter(t=>t.status==="unhealthy").length,offlineAgents:e.filter(t=>t.status==="offline").length,totalConnections:e.reduce((t,n)=>t+n.connectionCount,0)}}setAgentStatus(e,t){let n=this.agents.get(e);n&&(n.status=t);}setLastHeartbeat(e,t){let n=this.agents.get(e);n&&(n.lastHeartbeat=t);}};var h=a$1.create({service:"gateway:agent-sync-manager"}),T=class{constructor(e){a$2(this,"config");a$2(this,"registry");a$2(this,"localNodeId");a$2(this,"localNodeName");a$2(this,"remoteNodes",new Map);a$2(this,"remoteAgents",new Map);a$2(this,"syncTimer",null);a$2(this,"running",false);this.registry=e.registry,this.localNodeId=e.localNodeId,this.localNodeName=e.localNodeName,this.config={...$,...e.config};}async start(){if(this.running){h.warn("Agent sync manager is already running");return}h.info("Starting agent sync manager",{localNodeId:this.localNodeId,mode:this.config.mode,interval:this.config.interval,remoteNodes:this.config.remoteNodes.length});for(let e of this.config.remoteNodes)this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"});(this.config.mode==="pull"||this.config.mode==="both")&&(this.syncTimer=setInterval(()=>this.pullFromAllNodes(),this.config.interval)),this.running=true,h.info("Agent sync manager started");}async stop(){if(!this.running){h.warn("Agent sync manager is not running");return}h.info("Stopping agent sync manager"),this.syncTimer&&(clearInterval(this.syncTimer),this.syncTimer=null),this.remoteNodes.clear(),this.remoteAgents.clear(),this.running=false,h.info("Agent sync manager stopped");}async pullFromAllNodes(){for(let[e]of this.remoteNodes.entries())try{await this.pullFromNode(e);}catch(t){h.error("Failed to pull from node",{nodeId:e,error:t.message});}}async pullFromNode(e){let t=this.remoteNodes.get(e);if(!t)throw new Error(`Unknown remote node: ${e}`);t.status="syncing";let n=this.config.remoteNodes.find(s=>s.id===e);if(!n)throw new Error(`Remote node not found in config: ${e}`);h.debug("Pulling agent list from node",{nodeId:e,address:n.address});try{let s=`http://${n.address}:${n.port}/sync/agents`,r=await Fetch.get(s,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);let c=r.data.agents||[];for(let l of c)l.sourceGatewayId=e,this.remoteAgents.set(l.id,l);return t.lastSyncAt=Date.now(),t.status="synced",t.info.lastSyncAt=t.lastSyncAt,t.info.status="online",delete t.error,h.info("Pulled agent list from node",{nodeId:e,agentCount:c.length,lastSyncAt:t.lastSyncAt}),c}catch(s){throw t.status="error",t.error=s.message,t.info.status="offline",h.error("Failed to pull from node",{nodeId:e,error:s.message}),s}}async pushEvent(e){if(this.config.mode!=="push"&&this.config.mode!=="both")return;let t=[];for(let[n]of this.remoteNodes.entries())t.push(this.pushToNode(n,e));await Promise.allSettled(t);}async pushToNode(e,t){let n=this.config.remoteNodes.find(s=>s.id===e);if(!n){h.warn("Unknown remote node for push",{nodeId:e});return}try{let s=`http://${n.address}:${n.port}/sync/events`,r=await Fetch.post(s,t,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);h.debug("Pushed event to node",{nodeId:e,eventType:t.type});}catch(s){h.error("Failed to push to node",{nodeId:e,eventType:t.type,error:s.message});}}async handleSyncMessage(e){switch(h.debug("Received sync message",{type:e.type,sourceNodeId:e.sourceNodeId}),e.type){case "agent_list_request":await this.handleAgentListRequest(e);break;case "agent_list_response":await this.handleAgentListResponse(e);break;case "agent_register":await this.handleAgentRegister(e);break;case "agent_deregister":await this.handleAgentDeregister(e);break;case "agent_heartbeat":await this.handleAgentHeartbeat(e);break;case "agent_status_change":await this.handleAgentStatusChange(e);break;default:h.warn("Unknown sync message type",{type:e.type});}}async handleAgentListRequest(e){let t=this.getLocalAgentsForSync(),n={type:"agent_list_response",sourceNodeId:this.localNodeId,timestamp:Date.now(),requestId:e.requestId,agents:t,fullList:true,syncAt:Date.now()};await this.pushToNode(e.sourceNodeId,n);}async handleAgentListResponse(e){for(let t of e.agents)this.remoteAgents.set(t.id,t);h.info("Received agent list response",{sourceNodeId:e.sourceNodeId,agentCount:e.agents.length});}async handleAgentRegister(e){let{agent:t}=e,n=this.remoteAgents.get(t.id);if(n){let s=this.resolveConflict(n,t);this.remoteAgents.set(t.id,s);}else this.remoteAgents.set(t.id,t);h.info("Remote agent registered",{agentId:t.id,name:t.name,sourceNodeId:e.sourceNodeId});}async handleAgentDeregister(e){let{agentId:t}=e;this.remoteAgents.delete(t),h.info("Remote agent deregistered",{agentId:t,sourceNodeId:e.sourceNodeId});}async handleAgentHeartbeat(e){let{agentId:t,heartbeatAt:n,status:s}=e,r=this.remoteAgents.get(t);r&&(r.updatedAt=n,r.status=s),h.debug("Remote agent heartbeat received",{agentId:t,status:s,sourceNodeId:e.sourceNodeId});}async handleAgentStatusChange(e){let{agentId:t,newStatus:n,changedAt:s}=e,r=this.remoteAgents.get(t);r&&(r.status=n,r.updatedAt=s),h.info("Remote agent status changed",{agentId:t,newStatus:n,sourceNodeId:e.sourceNodeId});}resolveConflict(e,t){switch(this.config.conflictResolution){case "latest":return t.updatedAt>e.updatedAt?t:e;case "local":return e;case "remote":return t;default:return t.updatedAt>e.updatedAt?t:e}}getLocalAgentsForSync(){return this.registry.getAllAgents().map(t=>({id:t.id,name:t.name,sourceGatewayId:this.localNodeId,address:t.address,capabilities:t.capabilities,status:t.status,updatedAt:t.lastHeartbeat,metadata:t.metadata}))}getMergedAgentList(){let e=new Map;for(let t of this.getLocalAgentsForSync())e.set(t.id,t);for(let[t,n]of this.remoteAgents.entries())e.has(t)||e.set(t,n);return Array.from(e.values())}getRemoteAgents(){return Array.from(this.remoteAgents.values())}getStats(){let e=this.registry.getAllAgents(),t=0,n=0;for(let r of this.remoteNodes.values())r.info.status==="online"?t++:n++;let s=null;for(let r of this.remoteNodes.values())r.lastSyncAt>(s||0)&&(s=r.lastSyncAt);return {localAgentCount:e.length,remoteAgentCount:this.remoteAgents.size,totalAgentCount:e.length+this.remoteAgents.size,onlineNodeCount:t,offlineNodeCount:n,lastSyncAt:s}}isRunning(){return this.running}addRemoteNode(e){this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"}),h.info("Remote node added",{nodeId:e.id,address:e.address});}removeRemoteNode(e){this.remoteNodes.delete(e);for(let[t,n]of this.remoteAgents.entries())n.sourceGatewayId===e&&this.remoteAgents.delete(t);h.info("Remote node removed",{nodeId:e});}};var N=class{constructor(e){a$2(this,"log");a$2(this,"config");a$2(this,"ws",null);a$2(this,"state","disconnected");a$2(this,"subscriptions",new Map);a$2(this,"messageCallbacks",new Set);a$2(this,"agentListCallbacks",new Set);a$2(this,"cachedAgentList",[]);a$2(this,"localAgentList",[]);a$2(this,"reconnectAttempts",0);a$2(this,"reconnectTimer",null);a$2(this,"heartbeatTimer",null);a$2(this,"agentListRequestId",null);a$2(this,"agentListRequestResolve",null);this.config={url:e.url,type:e.type,id:e.id??`client_${Date.now()}_${Math.random().toString(36).substring(2,9)}`,reconnect:{enabled:e.reconnect?.enabled??true,maxAttempts:e.reconnect?.maxAttempts??5,delay:e.reconnect?.delay??3e3},heartbeat:{enabled:e.heartbeat?.enabled??true,interval:e.heartbeat?.interval??3e4}},this.log=a$1.create({service:`gateway:client:${this.config.id}`});}async connect(){if(this.state==="connected"||this.state==="connecting"){this.log.warn("already connected or connecting");return}return this.state="connecting",this.log.info("connecting to gateway",{url:this.config.url}),new Promise((e,t)=>{try{this.ws=new WebSocket(this.config.url),this.ws.onopen=()=>{this.state="connected",this.reconnectAttempts=0,this.log.info("connected to gateway"),this.config.heartbeat.enabled&&this.startHeartbeat(),this.resubscribeAll(),e();},this.ws.onmessage=n=>{this.handleMessage(n.data);},this.ws.onclose=n=>{this.handleClose(n.code,n.reason);},this.ws.onerror=n=>{this.log.error("WebSocket error",{error:String(n)}),this.state==="connecting"&&t(new Error("Connection failed"));};}catch(n){this.state="disconnected",t(n);}})}async disconnect(){this.state!=="disconnected"&&(this.log.info("disconnecting from gateway"),this.stopHeartbeat(),this.stopReconnect(),this.ws&&(this.ws.close(1e3,"Client disconnect"),this.ws=null),this.state="disconnected",this.log.info("disconnected from gateway"));}async subscribe(e,t){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"subscribe",sessionId:e,backendSessionId:t,clientId:this.config.id}),this.log.info("subscribed to session",{sessionId:e,backendSessionId:t});}async unsubscribe(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"unsubscribe",sessionId:e,clientId:this.config.id}),this.subscriptions.delete(e),this.log.info("unsubscribed from session",{sessionId:e});}async send(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"message",message:e});}onMessage(e){this.messageCallbacks.add(e);}offMessage(e){this.messageCallbacks.delete(e);}getState(){return this.state}getId(){return this.config.id}getSubscriptions(){return [...this.subscriptions.keys()]}setLocalAgents(e){this.localAgentList=e,this.updateMergedAgentList();}async fetchAgentList(){if(this.state!=="connected")throw new Error("Not connected to gateway");return new Promise((e,t)=>{let n=`req_${Date.now()}_${Math.random().toString(36).substring(2,9)}`;this.agentListRequestId=n,this.agentListRequestResolve=e,this.sendData({type:"agent_list_request",requestId:n}),setTimeout(()=>{this.agentListRequestId===n&&(this.agentListRequestId=null,this.agentListRequestResolve=null,t(new Error("Agent list request timeout")));},1e4);})}getConnectableAgents(e={}){let{localAgents:t,preferLocal:n=true,filterOffline:s=true}=e,r=t||this.localAgentList,c=this.cachedAgentList,l=this.mergeAgentLists(r,c,n);return s?l.filter(y=>y.status!=="offline"):l}onAgentListChange(e){this.agentListCallbacks.add(e);}offAgentListChange(e){this.agentListCallbacks.delete(e);}mergeAgentLists(e,t,n){let s=new Map,r=n?t:e,c=n?e:t;for(let l of r)s.set(l.id,l);for(let l of c)s.set(l.id,l);return Array.from(s.values())}updateMergedAgentList(){let e=this.getConnectableAgents({filterOffline:false});for(let t of this.agentListCallbacks)try{t(e);}catch(n){this.log.error("agent list callback error",{error:String(n)});}}handleAgentListResponse(e,t){t&&this.agentListRequestId===t&&this.agentListRequestResolve&&(this.agentListRequestId=null,this.agentListRequestResolve(e),this.agentListRequestResolve=null),this.cachedAgentList=e,this.updateMergedAgentList(),this.log.debug("received agent list",{count:e.length});}sendData(e){if(!this.ws||this.ws.readyState!==WebSocket.OPEN){this.log.warn("WebSocket not ready, cannot send");return}this.ws.send(JSON.stringify(e));}handleMessage(e){try{let t=JSON.parse(e);switch(t.type){case "message":this.handleGatewayMessage(t.message);break;case "subscribed":this.subscriptions.set(t.sessionId,t.subscription);break;case "unsubscribed":this.subscriptions.delete(t.sessionId);break;case "pong":break;case "agent_list_response":this.handleAgentListResponse(t.agents||[],t.requestId);break;case "agent_update":this.handleAgentUpdate(t.agent,t.action);break;default:this.log.debug("unknown message type",{type:t.type});}}catch(t){this.log.error("failed to parse message",{error:String(t)});}}handleAgentUpdate(e,t){switch(t){case "add":case "update":{let n=this.cachedAgentList.findIndex(s=>s.id===e.id);n>=0?this.cachedAgentList[n]=e:this.cachedAgentList.push(e);break}case "remove":this.cachedAgentList=this.cachedAgentList.filter(n=>n.id!==e.id);break}this.updateMergedAgentList();}handleGatewayMessage(e){for(let t of this.messageCallbacks)try{t(e);}catch(n){this.log.error("message callback error",{error:String(n)});}}handleClose(e,t){this.log.info("connection closed",{code:e,reason:t}),this.state="disconnected",this.ws=null,this.stopHeartbeat(),this.config.reconnect.enabled&&this.scheduleReconnect();}scheduleReconnect(){if(this.reconnectAttempts>=this.config.reconnect.maxAttempts){this.log.error("max reconnect attempts reached");return}this.reconnectAttempts++,this.state="reconnecting";let e=this.config.reconnect.delay*this.reconnectAttempts;this.log.info("scheduling reconnect",{attempt:this.reconnectAttempts,delay:e}),this.reconnectTimer=setTimeout(async()=>{try{await this.connect();}catch(t){this.log.error("reconnect failed",{error:String(t)});}},e);}stopReconnect(){this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null);}startHeartbeat(){this.heartbeatTimer=setInterval(()=>{this.state==="connected"&&this.sendData({type:"ping"});},this.config.heartbeat.interval);}stopHeartbeat(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null);}async resubscribeAll(){for(let e of this.subscriptions.keys())try{await this.subscribe(e);}catch(t){this.log.error("failed to resubscribe",{sessionId:e,error:String(t)});}}getHttpUrl(){return this.config.url.replace(/^ws:/,"http:").replace(/^wss:/,"https:")}async httpRequest(e,t={}){let n=`${this.getHttpUrl()}${e}`,s=await fetch(n,{...t,headers:{"Content-Type":"application/json",...t.headers}});if(!s.ok){let r=await s.text().catch(()=>"Unknown error");throw new Error(`HTTP ${s.status}: ${r}`)}return s.json()}async healthCheck(){return this.httpRequest("/health")}async getStatus(){return this.httpRequest("/status")}async listContacts(e,t={}){let n=new URLSearchParams({platform:e,limit:String(t.limit??100),offset:String(t.offset??0)});return this.httpRequest(`/contacts?${n}`)}async getContact(e){return this.httpRequest(`/contacts/${encodeURIComponent(e)}`)}async listSessions(e={}){let t=new URLSearchParams({limit:String(e.limit??100),offset:String(e.offset??0)});return this.httpRequest(`/sessions?${t}`)}async getSession(e){return this.httpRequest(`/sessions/${encodeURIComponent(e)}`)}async createSession(e){return this.httpRequest("/session",{method:"POST",body:JSON.stringify(e??{})})}async sendMessage(e,t){return this.httpRequest(`/session/${encodeURIComponent(e)}/prompt`,{method:"POST",body:JSON.stringify(t)})}};async function Ut(i){let{Log:e}=await import('./chunks/log-MKRLBH4R.mjs'),t=false;await e.init({logDir:i.logDir??process.env.EASBOT_LOG_PATH??process.cwd(),print:i.print??false,dev:i.dev??t,level:i.level??("INFO")});}
|
|
2
|
-
export{M as AgentRegistry,T as AgentSyncManager,x as Gateway,N as GatewayClient
|
|
1
|
+
import {v,Q,_,$}from'./chunks/chunk-BJV43DSB.mjs';export{p as AgentRegistryConfigSchema,k as AgentSyncConfigSchema,j as AuthConfigSchema,ha as BaseChannelPlugin,h as ChannelConfigSchema,fa as ChannelPluginLoader,ra as ChannelPluginRegistry,o as CircuitBreakerConfigSchema,m as ConnectionPoolConfigSchema,_ as DEFAULT_AGENT_REGISTRY_CONFIG,aa as DEFAULT_CONNECTION_POOL_CONFIG,$ as DEFAULT_SYNC_CONFIG,Z as DEFAULT_TOKEN_AUTH_CONFIG,Y as DEFAULT_WEBSOCKET_SERVER_CONFIG,ja as DiscordPlugin,c as FeishuChannelConfigSchema,la as FeishuPlugin,l as GatewayClusterConfigSchema,s as GatewayConfigSchema,ta as GatewayServer,r as GatewayServerConfigSchema,ea as GatewaySessionManager,i as HTTPSConfigSchema,R as MessageLockManager,n as MessageQueueConfigSchema,S as MessageRetryManager,ba as MessageRouter,ca as MessageStore,g as NostrChannelConfigSchema,pa as NostrPlugin,q as SessionConfigSchema,da as SessionStore,f as SignalChannelConfigSchema,oa as SignalPlugin,b as SlackChannelConfigSchema,ka as SlackPlugin,a as TelegramChannelConfigSchema,ia as TelegramPlugin,d as WeChatChannelConfigSchema,ma as WeChatPlugin,e as WebChatChannelConfigSchema,na as WebChatPlugin,y as clearConfigCache,qa as createChannelPlugin,W as createDefaultSessionState,T as createGatewayMessage,U as createTextMessage,V as generateSessionId,X as generateSubscriptionId,N as getAgentAdapter,F as getAgentRegistryConfig,I as getAuthConfig,B as getChannelConfig,E as getCircuitBreakerConfig,J as getClusterConfig,w as getConfig,z as getConfigDirectory,C as getConnectionPoolConfig,K as getDefaultAgent,H as getHTTPSConfig,L as getLogLevel,D as getMessageQueueConfig,A as getServerConfig,G as getSessionConfig,sa as getSupportedPlatforms,O as hasAgentAdapter,x as isConfigLoaded,v as loadConfig,u as parsePartialConfig,P as requireAgentAdapter,M as setAgentAdapter,t as validateConfig}from'./chunks/chunk-BJV43DSB.mjs';import {a as a$1}from'./chunks/chunk-7OFQ65NG.mjs';import {a as a$2}from'./chunks/chunk-HAMGVOQD.mjs';import {Fetch}from'@easbot/utils';var a=a$1.create({service:"gateway"}),m=class g extends Error{constructor(t,n,s){super(t);a$2(this,"type");a$2(this,"stage");a$2(this,"cause");this.type="GatewayInitializationError",this.stage=n,this.cause=s,Error.captureStackTrace&&Error.captureStackTrace(this,g);}};function H(g){let e=["ECONNRESET","EPIPE","ETIMEDOUT","ECONNREFUSED","ENOTFOUND","EHOSTUNREACH","ENETUNREACH"],t=g.code;return e.includes(t??"")}function tt(){process.on("unhandledRejection",g=>{let e=g instanceof Error?g:new Error(String(g));H(e)||a.error("unhandled promise rejection",{error:e.message,stack:e.stack});}),process.on("uncaughtException",g=>{H(g)||(a.error("uncaught exception",{error:g.message,stack:g.stack}),process.exit(1));});}var x;(it=>{let g={initialized:false,initPromise:null,directory:".easbot",config:null},e,t={server:null,config:null,initialized:false,status:"stopped"};async function n(u={}){if(g.initialized)return;if(g.initPromise)return g.initPromise;let d={directory:u.directory??".easbot",printLogs:u.printLogs??false,logLevel:u.logLevel??"INFO"};return g.directory=d.directory,g.initPromise=(async()=>{let{Global:b}=await import('./chunks/global-5GMGC2RN.mjs');try{await b.init();}catch(l){throw new m("Failed to initialize global directories","global",l instanceof Error?l:void 0)}try{await a$1.init({logDir:b.Path.log,print:d.printLogs,level:d.logLevel});}catch(l){throw new m("Failed to initialize log infrastructure","log",l instanceof Error?l:void 0)}try{tt();}catch(l){throw new m("Failed to install unhandled exception handlers","server",l instanceof Error?l:void 0)}Fetch.hasProxyConfigured()&&(Fetch.enableProxy({connectTimeout:1e4,keepAliveTimeout:3e4}),a.info("global proxy enabled",{proxyUrl:Fetch.getProxyUrl()}));try{g.config=await v(d.directory);}catch(l){throw new m("Failed to load gateway configuration","config",l instanceof Error?l:void 0)}g.initialized=true,a.info("gateway runtime initialized",{directory:d.directory,logLevel:d.logLevel});})(),g.initPromise}it.init=n;function s(){return g.initialized}it.isInitialized=s;function r(){return e||(e=async()=>{if(!g.initialized)throw new m("Gateway not initialized. Call init() first.","server");let u=Q(),d=await v(u.directory);return d.server?.enabled?(t={server:null,config:d,initialized:true,status:"stopped"},a.info("gateway initialized",{port:d.server?.port,hostname:d.server?.hostname}),t):(a.debug("gateway server disabled or not configured"),{server:null,config:d,initialized:true,status:"stopped"})}),e}function i(){return r()()}it.state=i;async function c(){return (await i()).server}it.get=c;async function p(){return (await i()).config?.server?.enabled??false}it.isEnabled=p;async function k(){return (await i()).config}it.config=k;function st(){return t.status}it.getStatus=st;async function O(u){a.debug("Gateway.start: called"),g.initialized||(a.debug("Gateway.start: calling init()"),await n(),a.debug("Gateway.start: init() completed")),a.debug("Gateway.start: calling state()");let d=await i();if(a.debug("Gateway.start: state() completed",{status:d.status}),!(d.config?.server?.enabled??true))throw a.warn("gateway server is disabled"),new Error("Gateway server is disabled");if(t.status==="running"){a.info("gateway server is already running");return}if(t.status==="starting"){a.info("gateway server is starting");return}t.status="starting",a.debug("Gateway.start: status set to starting");try{let l={...d.config?.server,...u};a.debug("gateway start: creating server with config",{port:l.port,hostname:l.hostname,path:l.path,https:l.https?.enabled}),a.debug("Gateway.start: calling createGatewayServer");let R=await D(l);a.debug("Gateway.start: createGatewayServer completed"),a.debug("gateway start: server created, updating state"),t.server=R,t.status="running",t.error=void 0,a.info("gateway server started",{port:l.port,hostname:l.hostname});}catch(l){throw t.status="error",t.error=l instanceof Error?l.message:String(l),a.error("failed to start gateway server",{error:t.error}),l}}it.start=O;async function F(){if(t.status==="stopped"){a.info("gateway server is already stopped");return}if(t.status==="stopping"){a.info("gateway server is stopping");return}t.status="stopping";try{t.server&&await t.server.stop(),t.server=null,t.status="stopped",t.error=void 0,await a$1.close(),a.info("gateway server stopped");}catch(u){throw t.status="error",t.error=u instanceof Error?u.message:String(u),a.error("failed to stop gateway server",{error:t.error}),u}}it.stop=F;async function rt(u){a.info("restarting gateway server"),(t.status==="running"||t.status==="starting")&&await F(),await O(u);}it.restart=rt;async function at(){let u=Q(),d=await v(u.directory);return t.config=d,g.config=d,a.info("gateway config reloaded"),d}it.reloadConfig=at;async function D(u,d){a.debug("createGatewayServer: starting");let{GatewayServer:b}=await import('./chunks/server-Y336TWLM.mjs');a.debug("createGatewayServer: GatewayServer imported");let l=new b(u);a.debug("createGatewayServer: server instance created"),a.debug("createGatewayServer: created server instance, calling start()");let R=d?.startupTimeout??3e4;if(await(async()=>{let A=new Promise((v,q)=>{setTimeout(()=>{q(new Error(`Gateway server startup timeout after ${R}ms`));},R);});try{a.debug("createGatewayServer: calling server.start()"),await Promise.race([l.start(),A]),a.debug("createGatewayServer: server.start() completed");}catch(v){a.error("gateway server start failed, attempting cleanup",{error:v instanceof Error?v.message:String(v)});try{await l.stop();}catch{}throw v}})(),d?.onStarted)try{await d.onStarted(l);}catch(A){a.warn("server started but onStarted callback failed",{error:A instanceof Error?A.message:String(A)});}return a.info("gateway server created and started",{port:u.port,hostname:u.hostname}),l}it.createGatewayServer=D;})(x||(x={}));var f=a$1.create({service:"gateway:agent-registry"}),M=class{constructor(e={}){a$2(this,"config");a$2(this,"agents",new Map);a$2(this,"heartbeatTimer",null);a$2(this,"roundRobinIndex",0);this.config={..._,...e};}async register(e){if(this.agents.has(e.id))throw new Error(`Agent with ID "${e.id}" is already registered`);if(this.agents.size>=this.config.maxAgents)throw new Error(`Maximum number of agents (${this.config.maxAgents}) reached`);let t=Date.now(),n={...e,status:"healthy",lastHeartbeat:t,registeredAt:t,connectionCount:0};this.agents.set(e.id,n),f.info("Agent registered",{agentId:e.id,name:e.name,capabilities:e.capabilities,address:e.address,totalAgents:this.agents.size});}async deregister(e){let t=this.agents.get(e);if(!t){f.warn("Attempted to deregister unknown agent",{agentId:e});return}this.agents.delete(e),f.info("Agent deregistered",{agentId:e,name:t.name,totalAgents:this.agents.size});}async heartbeat(e,t){let n=this.agents.get(e);if(!n)throw f.warn("Heartbeat received from unknown agent",{agentId:e}),new Error(`Agent "${e}" is not registered`);n.lastHeartbeat=Date.now(),n.status="healthy",t&&(n.metadata={...n.metadata,...t}),f.debug("Agent heartbeat received",{agentId:e,lastHeartbeat:n.lastHeartbeat});}getAgent(e){return this.agents.get(e)}getAllAgents(){return Array.from(this.agents.values())}getHealthyAgents(){return this.getAllAgents().filter(e=>e.status==="healthy")}selectAgent(e){let{strategy:t,capabilities:n,healthyOnly:s=true}=e,r=s?this.getHealthyAgents():this.getAllAgents();if(n&&n.length>0&&(r=r.filter(c=>n.every(p=>c.capabilities.includes(p)))),r.length===0){f.warn("No agents available for selection",{strategy:t,capabilities:n,healthyOnly:s,totalAgents:this.agents.size});return}let i;switch(t){case "random":i=this.selectRandom(r);break;case "round-robin":i=this.selectRoundRobin(r);break;case "least-connections":i=this.selectLeastConnections(r);break;default:i=this.selectRandom(r);}return f.debug("Agent selected",{agentId:i.id,strategy:t,connectionCount:i.connectionCount}),i}incrementConnection(e){let t=this.agents.get(e);t&&(t.connectionCount++,f.debug("Agent connection incremented",{agentId:e,connectionCount:t.connectionCount}));}decrementConnection(e){let t=this.agents.get(e);t&&t.connectionCount>0&&(t.connectionCount--,f.debug("Agent connection decremented",{agentId:e,connectionCount:t.connectionCount}));}startHeartbeatCheck(){if(this.heartbeatTimer){f.warn("Heartbeat check is already running");return}f.info("Starting heartbeat check",{interval:this.config.heartbeatCheckInterval,timeout:this.config.heartbeatTimeout}),this.heartbeatTimer=setInterval(()=>this.checkHeartbeats(),this.config.heartbeatCheckInterval);}stopHeartbeatCheck(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null,f.info("Heartbeat check stopped"));}checkHeartbeats(){let e=Date.now(),{heartbeatTimeout:t,autoDeregisterTimeout:n}=this.config,s=[],r=0;for(let[i,c]of this.agents.entries()){let p=e-c.lastHeartbeat;if(p>n){s.push(i);continue}p>t&&c.status==="healthy"&&(c.status="unhealthy",r++,f.warn("Agent marked as unhealthy due to heartbeat timeout",{agentId:i,lastHeartbeat:c.lastHeartbeat,timeSinceLastHeartbeat:p,timeout:t}));}for(let i of s){let c=this.agents.get(i);c&&(f.info("Auto-deregistering agent due to timeout",{agentId:i,name:c.name,lastHeartbeat:c.lastHeartbeat,timeout:n}),this.agents.delete(i));}(r>0||s.length>0)&&f.info("Heartbeat check completed",{unhealthyCount:r,deregisteredCount:s.length,totalAgents:this.agents.size});}selectRandom(e){let t=Math.floor(Math.random()*e.length);return e[t]}selectRoundRobin(e){this.roundRobinIndex=this.roundRobinIndex%e.length;let t=e[this.roundRobinIndex];return this.roundRobinIndex++,t}selectLeastConnections(e){return e.reduce((t,n)=>n.connectionCount<t.connectionCount?n:t)}getStats(){let e=this.getAllAgents();return {totalAgents:e.length,healthyAgents:e.filter(t=>t.status==="healthy").length,unhealthyAgents:e.filter(t=>t.status==="unhealthy").length,offlineAgents:e.filter(t=>t.status==="offline").length,totalConnections:e.reduce((t,n)=>t+n.connectionCount,0)}}setAgentStatus(e,t){let n=this.agents.get(e);n&&(n.status=t);}setLastHeartbeat(e,t){let n=this.agents.get(e);n&&(n.lastHeartbeat=t);}};var h=a$1.create({service:"gateway:agent-sync-manager"}),T=class{constructor(e){a$2(this,"config");a$2(this,"registry");a$2(this,"localNodeId");a$2(this,"localNodeName");a$2(this,"remoteNodes",new Map);a$2(this,"remoteAgents",new Map);a$2(this,"syncTimer",null);a$2(this,"running",false);this.registry=e.registry,this.localNodeId=e.localNodeId,this.localNodeName=e.localNodeName,this.config={...$,...e.config};}async start(){if(this.running){h.warn("Agent sync manager is already running");return}h.info("Starting agent sync manager",{localNodeId:this.localNodeId,mode:this.config.mode,interval:this.config.interval,remoteNodes:this.config.remoteNodes.length});for(let e of this.config.remoteNodes)this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"});(this.config.mode==="pull"||this.config.mode==="both")&&(this.syncTimer=setInterval(()=>this.pullFromAllNodes(),this.config.interval)),this.running=true,h.info("Agent sync manager started");}async stop(){if(!this.running){h.warn("Agent sync manager is not running");return}h.info("Stopping agent sync manager"),this.syncTimer&&(clearInterval(this.syncTimer),this.syncTimer=null),this.remoteNodes.clear(),this.remoteAgents.clear(),this.running=false,h.info("Agent sync manager stopped");}async pullFromAllNodes(){for(let[e]of this.remoteNodes.entries())try{await this.pullFromNode(e);}catch(t){h.error("Failed to pull from node",{nodeId:e,error:t.message});}}async pullFromNode(e){let t=this.remoteNodes.get(e);if(!t)throw new Error(`Unknown remote node: ${e}`);t.status="syncing";let n=this.config.remoteNodes.find(s=>s.id===e);if(!n)throw new Error(`Remote node not found in config: ${e}`);h.debug("Pulling agent list from node",{nodeId:e,address:n.address});try{let s=`http://${n.address}:${n.port}/sync/agents`,r=await Fetch.get(s,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);let i=r.data.agents||[];for(let c of i)c.sourceGatewayId=e,this.remoteAgents.set(c.id,c);return t.lastSyncAt=Date.now(),t.status="synced",t.info.lastSyncAt=t.lastSyncAt,t.info.status="online",delete t.error,h.info("Pulled agent list from node",{nodeId:e,agentCount:i.length,lastSyncAt:t.lastSyncAt}),i}catch(s){throw t.status="error",t.error=s.message,t.info.status="offline",h.error("Failed to pull from node",{nodeId:e,error:s.message}),s}}async pushEvent(e){if(this.config.mode!=="push"&&this.config.mode!=="both")return;let t=[];for(let[n]of this.remoteNodes.entries())t.push(this.pushToNode(n,e));await Promise.allSettled(t);}async pushToNode(e,t){let n=this.config.remoteNodes.find(s=>s.id===e);if(!n){h.warn("Unknown remote node for push",{nodeId:e});return}try{let s=`http://${n.address}:${n.port}/sync/events`,r=await Fetch.post(s,t,{headers:{"Content-Type":"application/json","X-Gateway-Node-Id":this.localNodeId}});if(!r.ok)throw new Error(`HTTP ${r.status}`);h.debug("Pushed event to node",{nodeId:e,eventType:t.type});}catch(s){h.error("Failed to push to node",{nodeId:e,eventType:t.type,error:s.message});}}async handleSyncMessage(e){switch(h.debug("Received sync message",{type:e.type,sourceNodeId:e.sourceNodeId}),e.type){case "agent_list_request":await this.handleAgentListRequest(e);break;case "agent_list_response":await this.handleAgentListResponse(e);break;case "agent_register":await this.handleAgentRegister(e);break;case "agent_deregister":await this.handleAgentDeregister(e);break;case "agent_heartbeat":await this.handleAgentHeartbeat(e);break;case "agent_status_change":await this.handleAgentStatusChange(e);break;default:h.warn("Unknown sync message type",{type:e.type});}}async handleAgentListRequest(e){let t=this.getLocalAgentsForSync(),n={type:"agent_list_response",sourceNodeId:this.localNodeId,timestamp:Date.now(),requestId:e.requestId,agents:t,fullList:true,syncAt:Date.now()};await this.pushToNode(e.sourceNodeId,n);}async handleAgentListResponse(e){for(let t of e.agents)this.remoteAgents.set(t.id,t);h.info("Received agent list response",{sourceNodeId:e.sourceNodeId,agentCount:e.agents.length});}async handleAgentRegister(e){let{agent:t}=e,n=this.remoteAgents.get(t.id);if(n){let s=this.resolveConflict(n,t);this.remoteAgents.set(t.id,s);}else this.remoteAgents.set(t.id,t);h.info("Remote agent registered",{agentId:t.id,name:t.name,sourceNodeId:e.sourceNodeId});}async handleAgentDeregister(e){let{agentId:t}=e;this.remoteAgents.delete(t),h.info("Remote agent deregistered",{agentId:t,sourceNodeId:e.sourceNodeId});}async handleAgentHeartbeat(e){let{agentId:t,heartbeatAt:n,status:s}=e,r=this.remoteAgents.get(t);r&&(r.updatedAt=n,r.status=s),h.debug("Remote agent heartbeat received",{agentId:t,status:s,sourceNodeId:e.sourceNodeId});}async handleAgentStatusChange(e){let{agentId:t,newStatus:n,changedAt:s}=e,r=this.remoteAgents.get(t);r&&(r.status=n,r.updatedAt=s),h.info("Remote agent status changed",{agentId:t,newStatus:n,sourceNodeId:e.sourceNodeId});}resolveConflict(e,t){switch(this.config.conflictResolution){case "latest":return t.updatedAt>e.updatedAt?t:e;case "local":return e;case "remote":return t;default:return t.updatedAt>e.updatedAt?t:e}}getLocalAgentsForSync(){return this.registry.getAllAgents().map(t=>({id:t.id,name:t.name,sourceGatewayId:this.localNodeId,address:t.address,capabilities:t.capabilities,status:t.status,updatedAt:t.lastHeartbeat,metadata:t.metadata}))}getMergedAgentList(){let e=new Map;for(let t of this.getLocalAgentsForSync())e.set(t.id,t);for(let[t,n]of this.remoteAgents.entries())e.has(t)||e.set(t,n);return Array.from(e.values())}getRemoteAgents(){return Array.from(this.remoteAgents.values())}getStats(){let e=this.registry.getAllAgents(),t=0,n=0;for(let r of this.remoteNodes.values())r.info.status==="online"?t++:n++;let s=null;for(let r of this.remoteNodes.values())r.lastSyncAt>(s||0)&&(s=r.lastSyncAt);return {localAgentCount:e.length,remoteAgentCount:this.remoteAgents.size,totalAgentCount:e.length+this.remoteAgents.size,onlineNodeCount:t,offlineNodeCount:n,lastSyncAt:s}}isRunning(){return this.running}addRemoteNode(e){this.remoteNodes.set(e.id,{info:{id:e.id,name:e.id,address:e.address,port:e.port,lastSyncAt:0,status:"unknown"},lastSyncAt:0,status:"synced"}),h.info("Remote node added",{nodeId:e.id,address:e.address});}removeRemoteNode(e){this.remoteNodes.delete(e);for(let[t,n]of this.remoteAgents.entries())n.sourceGatewayId===e&&this.remoteAgents.delete(t);h.info("Remote node removed",{nodeId:e});}};var N=class{constructor(e){a$2(this,"log");a$2(this,"config");a$2(this,"ws",null);a$2(this,"state","disconnected");a$2(this,"subscriptions",new Map);a$2(this,"messageCallbacks",new Set);a$2(this,"agentListCallbacks",new Set);a$2(this,"cachedAgentList",[]);a$2(this,"localAgentList",[]);a$2(this,"reconnectAttempts",0);a$2(this,"reconnectTimer",null);a$2(this,"heartbeatTimer",null);a$2(this,"agentListRequestId",null);a$2(this,"agentListRequestResolve",null);this.config={url:e.url,type:e.type,id:e.id??`client_${Date.now()}_${Math.random().toString(36).substring(2,9)}`,reconnect:{enabled:e.reconnect?.enabled??true,maxAttempts:e.reconnect?.maxAttempts??5,delay:e.reconnect?.delay??3e3},heartbeat:{enabled:e.heartbeat?.enabled??true,interval:e.heartbeat?.interval??3e4}},this.log=a$1.create({service:`gateway:client:${this.config.id}`});}async connect(){if(this.state==="connected"||this.state==="connecting"){this.log.warn("already connected or connecting");return}return this.state="connecting",this.log.info("connecting to gateway",{url:this.config.url}),new Promise((e,t)=>{try{this.ws=new WebSocket(this.config.url),this.ws.onopen=()=>{this.state="connected",this.reconnectAttempts=0,this.log.info("connected to gateway"),this.config.heartbeat.enabled&&this.startHeartbeat(),this.resubscribeAll(),e();},this.ws.onmessage=n=>{this.handleMessage(n.data);},this.ws.onclose=n=>{this.handleClose(n.code,n.reason);},this.ws.onerror=n=>{this.log.error("WebSocket error",{error:String(n)}),this.state==="connecting"&&t(new Error("Connection failed"));};}catch(n){this.state="disconnected",t(n);}})}async disconnect(){this.state!=="disconnected"&&(this.log.info("disconnecting from gateway"),this.stopHeartbeat(),this.stopReconnect(),this.ws&&(this.ws.close(1e3,"Client disconnect"),this.ws=null),this.state="disconnected",this.log.info("disconnected from gateway"));}async subscribe(e,t){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"subscribe",sessionId:e,backendSessionId:t,clientId:this.config.id}),this.log.info("subscribed to session",{sessionId:e,backendSessionId:t});}async unsubscribe(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"unsubscribe",sessionId:e,clientId:this.config.id}),this.subscriptions.delete(e),this.log.info("unsubscribed from session",{sessionId:e});}async send(e){if(this.state!=="connected")throw new Error("Not connected to gateway");this.sendData({type:"message",message:e});}onMessage(e){this.messageCallbacks.add(e);}offMessage(e){this.messageCallbacks.delete(e);}getState(){return this.state}getId(){return this.config.id}getSubscriptions(){return [...this.subscriptions.keys()]}setLocalAgents(e){this.localAgentList=e,this.updateMergedAgentList();}async fetchAgentList(){if(this.state!=="connected")throw new Error("Not connected to gateway");return new Promise((e,t)=>{let n=`req_${Date.now()}_${Math.random().toString(36).substring(2,9)}`;this.agentListRequestId=n,this.agentListRequestResolve=e,this.sendData({type:"agent_list_request",requestId:n}),setTimeout(()=>{this.agentListRequestId===n&&(this.agentListRequestId=null,this.agentListRequestResolve=null,t(new Error("Agent list request timeout")));},1e4);})}getConnectableAgents(e={}){let{localAgents:t,preferLocal:n=true,filterOffline:s=true}=e,r=t||this.localAgentList,i=this.cachedAgentList,c=this.mergeAgentLists(r,i,n);return s?c.filter(p=>p.status!=="offline"):c}onAgentListChange(e){this.agentListCallbacks.add(e);}offAgentListChange(e){this.agentListCallbacks.delete(e);}mergeAgentLists(e,t,n){let s=new Map,r=n?t:e,i=n?e:t;for(let c of r)s.set(c.id,c);for(let c of i)s.set(c.id,c);return Array.from(s.values())}updateMergedAgentList(){let e=this.getConnectableAgents({filterOffline:false});for(let t of this.agentListCallbacks)try{t(e);}catch(n){this.log.error("agent list callback error",{error:String(n)});}}handleAgentListResponse(e,t){t&&this.agentListRequestId===t&&this.agentListRequestResolve&&(this.agentListRequestId=null,this.agentListRequestResolve(e),this.agentListRequestResolve=null),this.cachedAgentList=e,this.updateMergedAgentList(),this.log.debug("received agent list",{count:e.length});}sendData(e){if(!this.ws||this.ws.readyState!==WebSocket.OPEN){this.log.warn("WebSocket not ready, cannot send");return}this.ws.send(JSON.stringify(e));}handleMessage(e){try{let t=JSON.parse(e);switch(t.type){case "message":this.handleGatewayMessage(t.message);break;case "subscribed":this.subscriptions.set(t.sessionId,t.subscription);break;case "unsubscribed":this.subscriptions.delete(t.sessionId);break;case "pong":break;case "agent_list_response":this.handleAgentListResponse(t.agents||[],t.requestId);break;case "agent_update":this.handleAgentUpdate(t.agent,t.action);break;default:this.log.debug("unknown message type",{type:t.type});}}catch(t){this.log.error("failed to parse message",{error:String(t)});}}handleAgentUpdate(e,t){switch(t){case "add":case "update":{let n=this.cachedAgentList.findIndex(s=>s.id===e.id);n>=0?this.cachedAgentList[n]=e:this.cachedAgentList.push(e);break}case "remove":this.cachedAgentList=this.cachedAgentList.filter(n=>n.id!==e.id);break}this.updateMergedAgentList();}handleGatewayMessage(e){for(let t of this.messageCallbacks)try{t(e);}catch(n){this.log.error("message callback error",{error:String(n)});}}handleClose(e,t){this.log.info("connection closed",{code:e,reason:t}),this.state="disconnected",this.ws=null,this.stopHeartbeat(),this.config.reconnect.enabled&&this.scheduleReconnect();}scheduleReconnect(){if(this.reconnectAttempts>=this.config.reconnect.maxAttempts){this.log.error("max reconnect attempts reached");return}this.reconnectAttempts++,this.state="reconnecting";let e=this.config.reconnect.delay*this.reconnectAttempts;this.log.info("scheduling reconnect",{attempt:this.reconnectAttempts,delay:e}),this.reconnectTimer=setTimeout(async()=>{try{await this.connect();}catch(t){this.log.error("reconnect failed",{error:String(t)});}},e);}stopReconnect(){this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null);}startHeartbeat(){this.heartbeatTimer=setInterval(()=>{this.state==="connected"&&this.sendData({type:"ping"});},this.config.heartbeat.interval);}stopHeartbeat(){this.heartbeatTimer&&(clearInterval(this.heartbeatTimer),this.heartbeatTimer=null);}async resubscribeAll(){for(let e of this.subscriptions.keys())try{await this.subscribe(e);}catch(t){this.log.error("failed to resubscribe",{sessionId:e,error:String(t)});}}getHttpUrl(){return this.config.url.replace(/^ws:/,"http:").replace(/^wss:/,"https:").replace(/\/$/,"")}async httpRequest(e,t={}){let n=`${this.getHttpUrl()}${e}`,s=t.method||"GET",{signal:r,...i}=t,c=await Fetch.request(n,{method:s,...i,headers:{"Content-Type":"application/json",...i.headers},...r?{signal:r}:{}});if(!c.ok){let p=c.data,k=typeof p=="object"?JSON.stringify(p):String(p||"Unknown error");throw new Error(`HTTP ${c.status}: ${k}`)}return c.data}async healthCheck(){return this.httpRequest("/health")}async getStatus(){return this.httpRequest("/status")}async listChannels(){return this.httpRequest("/channels")}async listContacts(e,t={}){let n=new URLSearchParams({platform:e,limit:String(t.limit??100),offset:String(t.offset??0)});return this.httpRequest(`/contacts?${n}`)}async getContact(e){return this.httpRequest(`/contacts/${encodeURIComponent(e)}`)}async getSession(e){return this.httpRequest(`/sessions/${encodeURIComponent(e)}`)}async createSession(e){return this.httpRequest("/session",{method:"POST",body:JSON.stringify(e??{})})}async getOrCreateSession(e){if(e.sessionId)try{return await this.getSession(e.sessionId),{sessionId:e.sessionId,created:!1}}catch{this.log.debug("session not found, will create new",{sessionId:e.sessionId});}if(e.platform||e.channel||e.userId){try{let r=await this.listSessions({platform:e.platform,channelId:e.channel,userId:e.userId});if(r.sessions.length>0){let i=r.sessions[0];if(i)return this.log.info("found existing session",{sessionId:i.id,channel:e.channel,userId:e.userId}),{sessionId:i.id,created:!1}}}catch(r){this.log.warn("failed to search sessions",{error:String(r)});}let n={};e.channel&&(n.channel=e.channel),e.userId&&(n.userId=e.userId),e.platform&&(n.platform=e.platform);let s=await this.createSession(n);return this.log.info("session created",{sessionId:s.id,params:n}),{sessionId:s.id,created:true}}let t=await this.createSession({});return this.log.info("session created (default)",{sessionId:t.id}),{sessionId:t.id,created:true}}async listSessions(e={}){let t=new URLSearchParams;return t.set("limit",String(e.limit??100)),t.set("offset",String(e.offset??0)),e.platform&&t.set("platform",e.platform),e.channelId&&t.set("channelId",e.channelId),e.userId&&t.set("userId",e.userId),this.httpRequest(`/sessions?${t}`)}async sendMessage(e,t){return this.httpRequest(`/session/${encodeURIComponent(e)}/prompt`,{method:"POST",body:JSON.stringify(t)})}};async function $t(g){let{Log:e}=await import('./chunks/log-MIT4H7S5.mjs'),t=false;await e.init({logDir:g.logDir??process.env.EASBOT_LOG_PATH??process.cwd(),print:g.print??false,dev:g.dev??t,level:g.level??("INFO")});}
|
|
2
|
+
export{M as AgentRegistry,T as AgentSyncManager,x as Gateway,N as GatewayClient,$t as initLog};
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@easbot/gateway",
|
|
3
|
-
"version": "0.2.
|
|
3
|
+
"version": "0.2.26",
|
|
4
4
|
"description": "EASBot Gateway - AI Agent Server and Multi-channel Integration Platform - 支持 WebSocket、HTTP、Discord、Telegram、Slack 等多渠道集成",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.cjs",
|
|
@@ -67,10 +67,10 @@
|
|
|
67
67
|
"ws": "^8.20.0",
|
|
68
68
|
"xdg-basedir": "^5.1.0",
|
|
69
69
|
"zod": "^4.4.3",
|
|
70
|
-
"@easbot/plugin": "0.2.
|
|
71
|
-
"@easbot/
|
|
72
|
-
"@easbot/
|
|
73
|
-
"@easbot/utils": "0.2.
|
|
70
|
+
"@easbot/plugin": "0.2.26",
|
|
71
|
+
"@easbot/types": "0.2.26",
|
|
72
|
+
"@easbot/sdk": "0.2.26",
|
|
73
|
+
"@easbot/utils": "0.2.26"
|
|
74
74
|
},
|
|
75
75
|
"devDependencies": {
|
|
76
76
|
"@biomejs/biome": "^2.4.14",
|