@aerostack/gateway 0.15.29 → 0.16.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/aerostack-connector.js +10 -14
- package/dist/exec-approval-daemon.js +1 -1
- package/dist/index.js +28 -29
- package/dist/logger.js +2 -2
- package/dist/openclaw-connector.js +1 -1
- package/dist/resolution.js +1 -1
- package/package.json +1 -1
|
@@ -1,18 +1,14 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
|
|
3
|
-
import{homedir as
|
|
4
|
-
`)}function
|
|
5
|
-
`)}function
|
|
6
|
-
`)}async function
|
|
7
|
-
`).pop()||"";try{const
|
|
8
|
-
`+t.slice(-
|
|
3
|
+
import{homedir as U}from"node:os";import{readFile as se}from"node:fs/promises";import{join as D}from"node:path";import{execFile as z}from"node:child_process";const ae=process.env.AEROSTACK_WORKSPACE_URL||"https://api.aerostack.dev/api/gateway/ws/notification",w=process.env.AEROSTACK_TOKEN||"",L=parseInt(process.env.AEROSTACK_OPENCLAW_PORT||"18789",10),oe=6e4,re=1e3,ie=3e4,ce=3e4,F=12e4,x=`aerostack-connector-${Date.now().toString(36)}`,m=ae.replace(/\/+$/,"");function o(e,t){const n=new Date().toISOString(),s=t?` ${JSON.stringify(t)}`:"";process.stderr.write(`[${n}] [INFO] ${e}${s}
|
|
4
|
+
`)}function d(e,t){const n=new Date().toISOString(),s=t?` ${JSON.stringify(t)}`:"";process.stderr.write(`[${n}] [WARN] ${e}${s}
|
|
5
|
+
`)}function S(e,t){const n=new Date().toISOString(),s=t?` ${JSON.stringify(t)}`:"";process.stderr.write(`[${n}] [ERROR] ${e}${s}
|
|
6
|
+
`)}async function T(e,t={},n=ce){const s=new AbortController,a=setTimeout(()=>s.abort(),n);try{return await fetch(e,{...t,signal:s.signal})}finally{clearTimeout(a)}}let I=null,k=!1,N=null,E=0,q=0;const R=new Map,le=100,ue=2e3,de=3e4,ge=3e4;let j=null;async function ye(){try{const e=D(U(),".openclaw","openclaw.json"),t=await se(e,"utf-8");return JSON.parse(t)?.gateway?.auth?.token??null}catch{return null}}async function v(){const e=await ye();if(!e)return d("No OpenClaw gateway token found, skipping WS control"),!1;const t=`ws://127.0.0.1:${L}`;return new Promise(n=>{try{const s=new WebSocket(t);I=s;const a=setTimeout(()=>{o("Gateway WS connect timeout");try{s.close()}catch{}n(!1)},1e4);s.onopen=()=>{clearTimeout(a),o("Gateway WS connected, sending handshake"),X({type:"req",id:String(++q),method:"connect",params:{minProtocol:3,maxProtocol:3,client:{id:"openclaw-tui",displayName:"Aerostack Connector",version:"1.0.0",platform:process.platform,mode:"cli"},auth:{token:e},scopes:["operator.read","operator.write","operator.admin"],caps:["tool-events","sessions"]}})},s.onmessage=r=>{try{const i=JSON.parse(String(r.data));fe(i,n)}catch{}},s.onerror=()=>{clearTimeout(a),n(!1)},s.onclose=()=>{clearTimeout(a),I=null,k=!1,G||we()}}catch{n(!1)}})}function X(e){try{I?.send(JSON.stringify(e))}catch{}}function fe(e,t){if(e.type==="res"&&!k){e.ok?(k=!0,E=0,o("Gateway WS authenticated",{port:L}),t?.(!0),b("sessions.subscribe").then(n=>{n.ok&&o("Subscribed to gateway session events")})):(d("Gateway WS auth rejected",{error:e.error?.message}),t?.(!1));return}if(e.type==="res"&&e.id&&R.has(e.id)){const n=R.get(e.id);R.delete(e.id),n(e);return}e.type==="event"&&pe(e)}const $=new Map;function pe(e){const t=e.event,n=e.payload??{};if(t==="session.tool"){const s=n.data,a=s?.name??n.toolName??"unknown",r=s?.phase??"";(r==="start"||r==="end")&&Te(a,r,"tool_call")}else if(t==="session.message"){const s=n.sessionKey??"",a=n.message,r=typeof a=="object"?a?.role:void 0;let i="";if(typeof a=="string")i=a;else if(a){const l=a.content;typeof l=="string"?i=l:Array.isArray(l)&&(i=l.map(c=>typeof c=="string"?c:c?.text??"").join("")),i||(i=a.text??n.text??"")}o("session.message",{sessionKey:s,role:r,textLength:i?.length??0});const g=$.get(s);g&&i&&r!=="user"&&(o("WS fast path: got agent response",{sessionKey:s,textLength:i.length}),g({text:i}))}else t==="sessions.changed"&&o("Gateway session changed",{key:n.sessionKey})}function b(e,t){return he(e,t,1e4)}function he(e,t,n=1e4){return new Promise(s=>{const a=String(++q),r=setTimeout(()=>{R.delete(a),s({type:"res",id:a,ok:!1,error:{code:"TIMEOUT",message:"timeout"}})},n);R.set(a,i=>{clearTimeout(r),s(i)}),X({type:"req",id:a,method:e,params:t})})}async function me(){if(!k){H("disconnected");return}const e=await b("health");e.ok?H("healthy",e.payload):H("unhealthy",{error:e.error?.message})}async function H(e,t){try{await T(`${m}/chat/heartbeat`,{method:"POST",headers:{Authorization:`Bearer ${w}`,"Content-Type":"application/json"},body:JSON.stringify({agent_type:"openclaw-connector",user_agent:"aerostack-connector/1.0.0",bridge_id:x,gateway_status:e,gateway_data:t})})}catch{}}function we(){if(G||N||E>=le)return;const e=Math.min(ue*Math.pow(2,Math.min(E,5)),de);E++,o(`Gateway WS reconnecting in ${e/1e3}s`,{attempt:E}),N=setTimeout(async()=>{N=null,k=!1,R.clear(),await v()},e)}function Se(){return new Promise(e=>{o("Restarting OpenClaw gateway"),z("openclaw",["gateway","restart"],{timeout:3e4},t=>{t?(S("Gateway restart failed",{error:t.message}),e(!1)):(o("Gateway restart triggered"),setTimeout(()=>{v()},5e3),e(!0))})})}async function _e(e,t){const n=Date.now();o("Agent turn starting",{sessionKey:e,messageLength:t.length});const s=e;if(k)try{const a=`aerostack-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,r=`agent:main:${s}`,i=new Promise(l=>{const c=setTimeout(()=>{$.delete(r),l(null)},F);$.set(r,f=>{clearTimeout(c),$.delete(r),l(f)})});b("sessions.messages.subscribe",{key:r});let g=await b("sessions.send",{key:r,message:t,idempotencyKey:a});if(!g.ok&&g.error?.message?.includes("session not found")&&(o("Creating new gateway session for thread",{key:s}),(await b("sessions.create",{key:s,agentId:"main"})).ok&&(g=await b("sessions.send",{key:r,message:t,idempotencyKey:a}))),!g.ok)$.delete(r),d("Gateway WS sessions.send rejected, falling back to CLI",{error:g.error?.message});else{const l=await i,c=Date.now()-n;if(l?.text)return o("Agent turn completed (WS fast path)",{sessionKey:r,durationMs:c}),[{text:l.text}];d("No response received via WS events, falling back to CLI",{durationMs:c})}}catch(a){$.delete(`agent:main:${s}`),d("Gateway WS fast path error, falling back to CLI",{error:a?.message})}return k||(d("Gateway disconnected, attempting reconnect before agent turn"),await v()||(d("Gateway unreachable, attempting restart"),await Se(),await new Promise(r=>setTimeout(r,8e3)))),new Promise((a,r)=>{z("openclaw",["agent","--agent","main","--json","--session-id",e,"-m",t],{timeout:F},(g,l)=>{const c=Date.now()-n;if(g){S("Agent turn failed",{sessionKey:e,durationMs:c,error:g.message}),r(g);return}let f=[];if(l)try{const u=JSON.parse(l.trim()),p=u?.result?.payloads;Array.isArray(p)&&p.length>0?f=p:(u.text||u.reply||u.message)&&(f=[{text:u.text||u.reply||u.message}])}catch{const u=l.trim().split(`
|
|
7
|
+
`).pop()||"";try{const p=JSON.parse(u);f=[{text:p.text||p.reply||p.message||u}]}catch{u&&(f=[{text:u}])}}f.length===0&&(f=[{text:`Processed your message (${c}ms).`}]),o("Agent turn completed (CLI fallback)",{sessionKey:e,durationMs:c,payloads:f.length}),a(f)})})}let G=!1,M=null;async function V(){try{const e=await T(`${m}/chat/heartbeat`,{method:"POST",headers:{Authorization:`Bearer ${w}`,"Content-Type":"application/json"},body:JSON.stringify({agent_type:"openclaw-connector",user_agent:"aerostack-connector/1.0.0",bridge_id:x})});e.ok?o("Heartbeat OK"):d("Heartbeat failed",{status:e.status})}catch(e){d("Heartbeat error",{error:e?.message})}}async function Y(e,t){try{const n=await T(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${w}`,"Content-Type":"application/json"},body:JSON.stringify({content:e.slice(0,1e4),session_id:t||M})});n.ok?o("Reply sent",{length:e.length,sessionId:t}):d("Reply failed",{status:n.status})}catch(n){S("Reply error",{error:n?.message})}}async function Te(e,t,n){try{await T(`${m}/chat/tool-status`,{method:"POST",headers:{Authorization:`Bearer ${w}`,"Content-Type":"application/json"},body:JSON.stringify({tool_name:e,phase:t,category:n})})}catch{}}const J=new Map,Q=50,Z=10;function K(e,t,n){let s=J.get(e);s||(s=[],J.set(e,s)),s.push({role:t,content:n,ts:Date.now()}),s.length>Q&&s.splice(0,s.length-Q)}function ke(e){const t=J.get(e);return!t||t.length===0?"":`Previous conversation:
|
|
8
|
+
`+t.slice(-Z).map(n=>`[${n.role==="user"?"Admin":"You (Agent)"}] ${n.content}`).join(`
|
|
9
9
|
`)+`
|
|
10
10
|
|
|
11
|
-
`}async function
|
|
12
|
-
[
|
|
13
|
-
${
|
|
14
|
-
|
|
15
|
-
[Image data (base64)]
|
|
16
|
-
${y.join(`
|
|
17
|
-
`)}`)}}j(n||"_default","user",a||`[${r.map(l=>l.filename).join(", ")}]`);try{const l=await he(u,f);for(const d of l){const y=(typeof d.text=="string"?d.text:String(d.text??"")).trim();y&&y.length>=2&&(j(n||"_default","agent",y),await z(y,n))}}catch(l){p("Agent turn failed, sending fallback",{error:l?.message}),await z("I received your message but encountered an error processing it. Please try again.",n)}}catch(t){p("handleChatMessage error",{error:t?.message})}}async function q(){if(!A){A=!0,R?.abort(),R=new AbortController;try{o("SSE connecting",{url:h});const e=await fetch(h,{method:"GET",headers:{Authorization:`Bearer ${m}`,Accept:"text/event-stream"},signal:R.signal});if(!e.ok||!e.body){g("SSE connect failed",{status:e.status}),k++,A=!1,B();return}o("SSE connected"),k=0;const t=e.body.getReader(),n=new TextDecoder;let s="";for(;;){const{done:a,value:c}=await t.read();if(a)break;s+=n.decode(c,{stream:!0});const i=s.split(`
|
|
18
|
-
`);s=i.pop()??"";let r="",u="";for(const f of i)f.startsWith("event: ")?r=f.slice(7).trim():f.startsWith("data: ")?u=f.slice(6):f===""&&r&&u&&(r==="aerostack_event"&&Ae(u),r="",u="")}o("SSE stream ended, reconnecting"),A=!1,B()}catch(e){if(e?.name==="AbortError"){A=!1;return}g("SSE error",{error:e?.message}),k++,A=!1,B()}}}function B(){if(O)return;const e=Math.min(te*Math.pow(2,k),ne);o(`SSE reconnecting in ${e/1e3}s`,{failures:k}),O=setTimeout(()=>{O=null,q()},e)}async function $e(){try{const e=await w(`${h}/chat/connection-test`,{headers:{Authorization:`Bearer ${m}`}},1e4);if(e.ok){const t=await e.json();return o(`Connected to "${t.workspace}"`,{tools:t.mcp_servers,token:t.token_name,role:t.token_role}),!0}else return e.status===401?(p("Token invalid \u2014 regenerate at dashboard \u2192 Workspace \u2192 Tokens"),!1):(g("Connection check failed",{status:e.status}),!1)}catch(e){return p("Connection check error",{error:e?.message}),!1}}async function Ee(){o("Aerostack Connector starting",{workspace:h,openclawPort:v,bridgeId:x}),m||(p("AEROSTACK_TOKEN is required"),process.exit(1)),await $e()||g("Connection verification failed, continuing anyway (will retry)"),await we("_default"),await b()?o("Gateway WS control layer active"):g("Gateway WS not reachable \u2014 will keep retrying"),M=setInterval(de,ie),await J();const n=setInterval(J,ee);q(),o("Connector ready \u2014 listening for dashboard chat messages");const s=()=>{I=!0,o("Shutting down"),clearInterval(n),M&&clearInterval(M),R?.abort(),O&&clearTimeout(O),$&&clearTimeout($);try{C?.close(1e3)}catch{}process.exit(0)};process.on("SIGTERM",s),process.on("SIGINT",s)}Ee().catch(e=>{p("Fatal error",{error:e?.message}),process.exit(1)});
|
|
11
|
+
`}async function Oe(e){try{const t=await T(`${m}/chat/history?limit=${Z}`,{headers:{Authorization:`Bearer ${w}`}});if(!t.ok)return;const n=await t.json();n.session_id&&(M=n.session_id);for(const s of n.messages??[]){const a=s.sender_type==="agent"?"agent":"user";K(n.session_id||e,a,s.content)}o("History seeded",{sessionId:n.session_id,count:n.messages?.length??0})}catch{}}let h=null,W=0,P=null;function Ae(e){return(e||"").replace(/[\r\n]+/g," ").replace(/\[/g,"(").replace(/\]/g,")").replace(/---/g,"\u2014")}import{writeFile as Ce,mkdir as Re}from"node:fs/promises";const ee=D(U(),".openclaw","media","inbound");async function $e(e,t){try{const n=await T(`${m}/chat/media/${e}/${t.filename}`,{headers:{Authorization:`Bearer ${w}`}},15e3);if(!n.ok)return d("Media fetch failed",{messageId:e,filename:t.filename,status:n.status}),null;const s=Buffer.from(await n.arrayBuffer());await Re(ee,{recursive:!0});const a=t.filename.split(".").pop()||"bin",r=D(ee,`file_0---${t.id||e}.${a}`);return await Ce(r,s),o("Media saved locally",{localPath:r,size:s.length}),r}catch(n){return d("Media download error",{messageId:e,filename:t.filename,error:n?.message}),null}}function be(e){return e.map(t=>`[media attached: ${t.path} (${t.mime}) | ${t.path}]`).join(`
|
|
12
|
+
`)}async function Ne(e){const t=Date.now();try{const n=JSON.parse(e);if(n.type!=="chat_message")return;const s=n.session_id,a=n.session_title||null,r=Ae(n.sender_name||"Admin"),i=n.content||"",g=n.content_type||"text",l=n.message_id;let c=[];if(n.metadata_json)try{const y=JSON.parse(n.metadata_json);y?.v===1&&Array.isArray(y.attachments)&&(c=y.attachments)}catch{}if(!i&&c.length===0)return;o(">>> Chat message received via WebSocket (ChatRelayDO)",{transport:"websocket",wsDeliveryMs:Date.now()-t,sender:r,sessionId:s,sessionTitle:a,contentType:g,textLength:i.length,attachments:c.length}),s&&(M=s);const f=`aerostack-chat-${s||"_default"}`;let u=`[Style: concise, numbered lists, bold headers, short sub-bullets. No walls of text.]
|
|
13
|
+
[${r}]:`;if(i&&(u+=` ${i}`),c.length>0&&l){const y=[],O=await Promise.all(c.slice(0,5).map(_=>$e(l,_).then(A=>A?{path:A,mime:_.mime_type}:null)));for(const _ of O)_&&y.push(_);y.length>0&&(u=be(y)+`
|
|
14
|
+
`+u)}K(s||"_default","user",i||`[${c.map(y=>y.filename).join(", ")}]`),B(s,!0);const p=Date.now();o(">>> Agent turn starting",{timeSinceWsReceived:`${p-t}ms`});try{const y=await _e(f,u),O=Date.now();B(s,!1),o(">>> Agent turn completed \u2014 timing breakdown",{transport:"websocket",agentProcessingMs:O-p,totalFromWsReceiveMs:O-t});for(const A of y){const C=(typeof A.text=="string"?A.text:String(A.text??"")).trim();C&&C.length>=2&&(K(s||"_default","agent",C),Ee(C,s)?o(">>> Reply sent via WS fast path (no HTTP POST needed)",{textLength:C.length}):await Y(C,s))}const _=Date.now();o(">>> Full round-trip complete",{wsToAgentStartMs:p-t,agentProcessingMs:O-p,replyPostMs:_-O,totalRoundTripMs:_-t})}catch(y){B(s,!1),S("Agent turn failed, sending fallback",{error:y?.message}),await Y("I received your message but encountered an error processing it. Please try again.",s)}}catch(n){S("handleChatMessage error",{error:n?.message})}}function Ee(e,t){if(!h||h.readyState!==WebSocket.OPEN)return!1;try{return h.send(JSON.stringify({type:"chat_reply",content:e.slice(0,1e4),session_id:t||M,sender_name:"Agent",content_type:"text"})),!0}catch{return!1}}function B(e,t=!0){if(!(!h||h.readyState!==WebSocket.OPEN))try{h.send(JSON.stringify({type:"typing",session_id:e||M,is_typing:t}))}catch{}}function te(){if(h){try{h.close()}catch{}h=null}try{const e=m.replace(/^http/,"ws")+`/chat-relay?token=${encodeURIComponent(w)}&client_type=connector&client_id=${encodeURIComponent(x)}`;o("Chat relay WS connecting",{url:m+"/chat-relay"});const t=new WebSocket(e);h=t,t.onopen=()=>{o("Chat relay WS connected"),W=0},t.onmessage=n=>{const s=typeof n.data=="string"?n.data:"";if(s)try{const a=JSON.parse(s);if(a.type==="pong")return;o(">>> WebSocket frame received from ChatRelayDO",{type:a.type,dataLength:s.length}),Ne(s)}catch{}},t.onclose=()=>{h=null,G||ne()},t.onerror=()=>{W++}}catch(e){d("Chat relay WS connect error",{error:e?.message}),W++,ne()}}function ne(){if(P)return;const e=Math.min(re*Math.pow(2,W),ie);o(`Chat relay WS reconnecting in ${e/1e3}s`,{failures:W}),P=setTimeout(()=>{P=null,te()},e)}async function Me(){try{const e=await T(`${m}/chat/connection-test`,{headers:{Authorization:`Bearer ${w}`}},1e4);if(e.ok){const t=await e.json();return o(`Connected to "${t.workspace}"`,{tools:t.mcp_servers,token:t.token_name,role:t.token_role}),!0}else return e.status===401?(S("Token invalid \u2014 regenerate at dashboard \u2192 Workspace \u2192 Tokens"),!1):(d("Connection check failed",{status:e.status}),!1)}catch(e){return S("Connection check error",{error:e?.message}),!1}}async function We(){o("Aerostack Connector starting",{workspace:m,openclawPort:L,bridgeId:x}),w||(S("AEROSTACK_TOKEN is required"),process.exit(1)),await Me()||d("Connection verification failed, continuing anyway (will retry)"),await Oe("_default"),await v()?o("Gateway WS control layer active"):d("Gateway WS not reachable \u2014 will keep retrying"),j=setInterval(me,ge),await V();const n=setInterval(V,oe);te(),o("Connector ready \u2014 listening for dashboard chat messages");const s=()=>{G=!0,o("Shutting down"),clearInterval(n),j&&clearInterval(j);try{h?.close(1e3)}catch{}P&&clearTimeout(P),N&&clearTimeout(N);try{I?.close(1e3)}catch{}process.exit(0)};process.on("SIGTERM",s),process.on("SIGINT",s)}We().catch(e=>{S("Fatal error",{error:e?.message}),process.exit(1)});
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
|
|
3
3
|
import{join as p}from"node:path";import{homedir as f}from"node:os";import{readFile as u}from"node:fs/promises";import{info as i,warn as l,error as S}from"./logger.js";import{startExecApprovalServer as x}from"./exec-approval-server.js";const d=process.env.AEROSTACK_WORKSPACE_URL,v=process.env.AEROSTACK_TOKEN;(!d||!v)&&(process.stderr.write(`ERROR: AEROSTACK_WORKSPACE_URL and AEROSTACK_TOKEN are required
|
|
4
|
-
`),process.exit(1));const m=d.replace(/\/+$/,"");async function O(t,a){const n={jsonrpc:"2.0",id:Date.now(),method:t,params:a??{}};try{const e=await fetch(m,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${v}`,"User-Agent":"aerostack-exec-daemon/0.15.
|
|
4
|
+
`),process.exit(1));const m=d.replace(/\/+$/,"");async function O(t,a){const n={jsonrpc:"2.0",id:Date.now(),method:t,params:a??{}};try{const e=await fetch(m,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${v}`,"User-Agent":"aerostack-exec-daemon/0.15.29","X-Agent-Id":"aerostack-exec-daemon"},body:JSON.stringify(n),signal:AbortSignal.timeout(15e3)});if((e.headers.get("content-type")??"").includes("text/event-stream")){const o=(await e.text()).split(`
|
|
5
5
|
`).reverse().find(s=>s.startsWith("data: "));return o?JSON.parse(o.slice(6)):null}return await e.json()}catch(e){return l("rpcCall failed",{method:t,error:e instanceof Error?e.message:String(e)}),null}}async function g(){i("Aerostack exec approval daemon starting",{workspace:m});const t=p(f(),".openclaw","exec-approvals.json");let a={};try{a=JSON.parse(await u(t,"utf-8"))}catch{l("exec-approvals.json not found, will retry on interval",{path:t})}const n=async()=>{try{const w=await u(t,"utf-8");a=JSON.parse(w)}catch{return null}const o=a?.socket?.path??p(f(),".openclaw","exec-approvals.sock"),s=a?.socket?.token;return s?(i("Starting exec approval socket server",{path:o}),x({socketPath:o,token:s,rpcCall:O})):(l("No socket token in exec-approvals.json, waiting for OpenClaw to write it"),null)};let e=await n(),r=null;e||(r=setInterval(async()=>{e=await n(),e&&r&&(clearInterval(r),r=null)},3e4)),i("Exec approval daemon ready \u2014 waiting for OpenClaw approval requests");const c=()=>{r&&(clearInterval(r),r=null),e?.stop(),process.exit(0)};process.on("SIGTERM",c),process.on("SIGINT",c)}g().catch(t=>{S("Fatal",{error:t instanceof Error?t.message:String(t)}),process.exit(1)});
|
package/dist/index.js
CHANGED
|
@@ -1,20 +1,20 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
|
|
3
|
-
import{Server as
|
|
4
|
-
`),process.exit(1)),
|
|
5
|
-
`),process.exit(1));let B;try{if(B=new URL(
|
|
3
|
+
import{Server as we}from"@modelcontextprotocol/sdk/server/index.js";import{StdioServerTransport as Ae}from"@modelcontextprotocol/sdk/server/stdio.js";import{ListToolsRequestSchema as Se,CallToolRequestSchema as Te,ListResourcesRequestSchema as Ee,ReadResourceRequestSchema as ke,ListPromptsRequestSchema as Re,GetPromptRequestSchema as Oe}from"@modelcontextprotocol/sdk/types.js";import{readFile as Ce,writeFile as ve}from"node:fs/promises";import{join as $e}from"node:path";import{homedir as be}from"node:os";import{resolveApproval as Q}from"./resolution.js";import{startHookServer as Ne,installClaudeHook as Pe,stopHookServer as Ie}from"./hook-server.js";import{OpenClawConnector as xe,resolveOpenClawToken as Ue}from"./openclaw-connector.js";import{info as u,warn as y,error as Le}from"./logger.js";const ie=$e(be(),".openclaw","pre-authorized.json");async function De(e){try{let t={};try{t=JSON.parse(await Ce(ie,"utf-8"))}catch{}t[e]=Date.now(),await ve(ie,JSON.stringify(t))}catch{}}const Z=process.env.AEROSTACK_WORKSPACE_URL,h=process.env.AEROSTACK_TOKEN;function W(e,t,a){const s=parseInt(e??String(t),10);return Number.isFinite(s)&&s>=a?s:t}const ee=W(process.env.AEROSTACK_APPROVAL_POLL_MS,3e3,500),te=W(process.env.AEROSTACK_APPROVAL_TIMEOUT_MS,864e5,5e3),ce=W(process.env.AEROSTACK_REQUEST_TIMEOUT_MS,3e4,1e3),Me=process.env.AEROSTACK_HOOK_SERVER!=="false",je=W(process.env.AEROSTACK_HOOK_PORT,18321,1024),Ke=process.env.AEROSTACK_HOOK_AUTO_INSTALL!=="false",He=process.env.AEROSTACK_OPENCLAW_ENABLED!=="false",le=W(process.env.AEROSTACK_OPENCLAW_PORT,18789,1024),We=process.env.AEROSTACK_OPENCLAW_TOKEN;Z||(process.stderr.write(`ERROR: AEROSTACK_WORKSPACE_URL is required
|
|
4
|
+
`),process.exit(1)),h||(process.stderr.write(`ERROR: AEROSTACK_TOKEN is required
|
|
5
|
+
`),process.exit(1));let B;try{if(B=new URL(Z),B.protocol!=="https:"&&B.protocol!=="http:")throw new Error("must be http or https")}catch{process.stderr.write(`ERROR: AEROSTACK_WORKSPACE_URL must be a valid HTTP(S) URL
|
|
6
6
|
`),process.exit(1)}B.protocol==="http:"&&!B.hostname.match(/^(localhost|127\.0\.0\.1)$/)&&process.stderr.write(`WARNING: Using HTTP (not HTTPS) \u2014 token will be sent in plaintext
|
|
7
|
-
`);const
|
|
8
|
-
`);let
|
|
9
|
-
${
|
|
7
|
+
`);const m=Z.replace(/\/+$/,""),se=crypto.randomUUID();function Be(){return process.env.AEROSTACK_AGENT_TYPE?process.env.AEROSTACK_AGENT_TYPE:process.env.CLAUDECODE||process.env.CLAUDE_CODE?"claude-code":process.env.MCP_TRANSPORT==="stdio"?"mcp-stdio":"unknown"}const ue=Be();let $=null,L=Date.now();const ze=3e4;async function O(e,t){const a={jsonrpc:"2.0",id:Date.now(),method:e,params:t??{}},s=new AbortController,n=setTimeout(()=>s.abort(),ce);try{const o=await fetch(m,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${h}`,"User-Agent":"aerostack-gateway/0.15.29","X-Agent-Id":"aerostack-gateway","X-Bridge-Id":se,"X-Agent-Type":ue},body:JSON.stringify(a),signal:s.signal});if(clearTimeout(n),(o.headers.get("content-type")??"").includes("text/event-stream")){const c=await o.text();return Fe(c,a.id)}return await o.json()}catch(o){clearTimeout(n);const r=o instanceof Error?o.message:"Unknown error";return o instanceof Error&&o.name==="AbortError"?{jsonrpc:"2.0",id:a.id,error:{code:-32603,message:"Request timed out"}}:{jsonrpc:"2.0",id:a.id,error:{code:-32603,message:`HTTP error: ${r}`}}}}function Fe(e,t){const a=e.split(`
|
|
8
|
+
`);let s=null;for(const n of a)if(n.startsWith("data: "))try{s=JSON.parse(n.slice(6))}catch{}return s??{jsonrpc:"2.0",id:t,error:{code:-32603,message:"Empty SSE response"}}}const qe=new Set(["aerostack__guardian_report","aerostack__check_approval","aerostack__guardian_check"]);function Je(e,t){if(qe.has(e))return;let a="other";const s=e.toLowerCase();s.includes("exec")||s.includes("bash")||s.includes("shell")||s.includes("command")||s.includes("run")?a="exec_command":s.includes("write")||s.includes("edit")||s.includes("create")||s.includes("patch")?a="file_write":s.includes("delete")||s.includes("remove")||s.includes("trash")||s.includes("unlink")?a="file_delete":s.includes("fetch")||s.includes("http")||s.includes("request")||s.includes("api")||s.includes("get")||s.includes("post")?a="api_call":s.includes("install")||s.includes("package")||s.includes("npm")||s.includes("pip")?a="package_install":s.includes("config")||s.includes("setting")||s.includes("env")?a="config_change":s.includes("deploy")||s.includes("publish")||s.includes("release")?a="deploy":s.includes("send")||s.includes("message")||s.includes("email")||s.includes("notify")||s.includes("slack")||s.includes("telegram")?a="message_send":(s.includes("read")||s.includes("query")||s.includes("search")||s.includes("list")||s.includes("get"))&&(a="data_access");let n;try{const o=JSON.stringify(t);n=o.length>500?o.slice(0,500)+"...":o}catch{n="(unable to serialize)"}O("tools/call",{name:"aerostack__guardian_report",arguments:{action:`${e}(${Object.keys(t).join(", ")})`,category:a,risk_level:"low",details:n}}).catch(()=>{})}const Ge=new Set(["aerostack__check_approval"]),de=[{name:"aerostack__chat_check",description:"Check for new messages from the workspace owner. Call this periodically or when asked to check for instructions.",inputSchema:{type:"object",properties:{},required:[]}},{name:"aerostack__chat_reply",description:"Send a reply message to the workspace owner via Agent Chat.",inputSchema:{type:"object",properties:{message:{type:"string",description:"Your reply message to the workspace owner"}},required:["message"]}}],Ve=new Set(de.map(e=>e.name));function _(e){return e.replace(/[\r\n]+/g," ").replace(/\[/g,"(").replace(/\]/g,")").replace(/---/g,"\u2014")}const Ye=new Set(["image/jpeg","image/png","image/gif","image/webp"]),Xe=2*1024*1024;async function ae(e,t,a,s,n){if(a!=="media"&&a!=="rich")return[{type:"text",text:`${_(n)}: ${_(s)}`}];let o=[];if(t)try{const c=JSON.parse(t);c?.v===1&&Array.isArray(c.attachments)&&(o=c.attachments)}catch{}if(o.length===0)return[{type:"text",text:`${_(n)}: ${_(s)}`}];const r=[];a==="rich"&&s?r.push({type:"text",text:`${_(n)}: ${_(s)}`}):o.length>0&&r.push({type:"text",text:`${_(n)} sent ${o.length} file(s):`});const i=e&&/^[a-zA-Z0-9_-]{1,128}$/.test(e)?e:null;for(const c of o){if(Ye.has(c.mime_type)&&c.size<=Xe&&i)try{const H=`${m}/chat/media/${i}/${encodeURIComponent(c.filename)}`,R=await S(H,{headers:{Authorization:`Bearer ${h}`}});if(R.ok){const g=Buffer.from(await R.arrayBuffer()).toString("base64");r.push({type:"image",data:g,mimeType:c.mime_type});continue}}catch{}const b=Math.round(c.size/1024),D=b>=1024?`${(b/1024).toFixed(1)} MB`:`${b} KB`,k=_(c.filename.slice(0,200)),C=_(c.mime_type.slice(0,100)),re=i?`${m}/chat/media/${i}/${encodeURIComponent(c.filename)}`:"(unavailable)";r.push({type:"text",text:`Attached: ${k} (${C}, ${D}) \u2014 ${re}`})}return r}const z=[],Qe=100,N=new Map,pe=50,Ze=100,he=10;let P=null;const me=new Set,F=new Map;let I=null;function j(e,t,a){let s=N.get(e);if(!s){if(N.size>=Ze){const n=N.keys().next().value;n&&N.delete(n)}s=[],N.set(e,s)}s.push({role:t,content:a,ts:Date.now()}),s.length>pe&&s.splice(0,s.length-pe)}async function et(e){if(!me.has(e))try{const t=await S(`${m}/chat/history?limit=${he}`,{headers:{Authorization:`Bearer ${h}`}});if(!t.ok)return;const a=await t.json();a.session_id&&(P=a.session_id);for(const s of a.messages??[]){const n=s.sender_type==="agent"?"agent":"user",o=a.session_id||e;let r=N.get(o);r||(r=[],N.set(o,r));const i=s.content_type==="media"||s.content_type==="rich"?`${s.content} [has media attachment]`:s.content;r.push({role:n,content:i,ts:s.created_at,content_type:s.content_type})}me.add(e),u("Thread history seeded from D1",{sessionId:a.session_id,count:a.messages?.length??0})}catch{}}function fe(e){const t=N.get(e);return!t||t.length===0?"":`Previous conversation:
|
|
9
|
+
${t.slice(-he).map(n=>`[${n.role==="user"?"Admin":"You (Agent)"}] ${n.content}`).join(`
|
|
10
10
|
`)}
|
|
11
11
|
|
|
12
|
-
`}async function
|
|
13
|
-
`):["[APPROVAL RECEIVED] Your request has been approved by the workspace owner.",
|
|
14
|
-
`);case"changes_requested":return["[CHANGES REQUESTED] The workspace owner reviewed your action and requested changes.","",`Original action: ${
|
|
15
|
-
`);case"rejected":return["[REJECTED] The workspace owner denied this action.","",`Action: ${
|
|
16
|
-
`);case"expired":return[`[EXPIRED] The approval request for "${
|
|
17
|
-
`);default:return`[APPROVAL UPDATE] Status for "${
|
|
12
|
+
`}async function S(e,t){const a=new AbortController,s=setTimeout(()=>a.abort(),ce);try{return await fetch(e,{...t,signal:a.signal})}finally{clearTimeout(s)}}async function tt(){const e=P||"_default";if(await et(e),z.length>0){const t=z.splice(0);L=Date.now();const a=fe(e),s=[];a&&s.push({type:"text",text:a}),s.push({type:"text",text:"New messages from workspace owner:"});for(const n of t){const o=await ae(n.message_id??null,n.metadata_json??null,n.content_type,n.content,n.sender_name);s.push(...o)}return s}try{const a=await S(`${m}/chat/pending?since=${L}`,{headers:{Authorization:`Bearer ${h}`}});if(!a.ok)return[{type:"text",text:"No new messages."}];const n=(await a.json()).messages??[];if(L=Date.now(),n.length===0)return[{type:"text",text:"No new messages from the workspace owner."}];for(const i of n)j(e,"user",i.content);const o=fe(e),r=[];o&&r.push({type:"text",text:o}),r.push({type:"text",text:"New messages from workspace owner:"});for(const i of n){const c=await ae(i.id??null,i.metadata_json??null,i.content_type,i.content,i.sender_name);r.push(...c)}return r}catch{return[{type:"text",text:"Failed to check messages."}]}}async function st(e){const t=typeof e.message=="string"?e.message.trim().slice(0,1e4):"";if(!t)return[{type:"text",text:"Error: message is required."}];try{const a=await S(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({content:t,session_id:P})});return a.ok?(j(P||"_default","agent",t),[{type:"text",text:"Reply sent to workspace owner."}]):[{type:"text",text:`Failed to send reply (HTTP ${a.status}).`}]}catch{return[{type:"text",text:"Failed to send reply."}]}}async function at(e,t){Je(e,t);const a=await O("tools/call",{name:e,arguments:t});if(a.error?.code===-32050){const o=a.error.data,r=o?.approval_id;if(!r||!/^[a-zA-Z0-9_-]{4,128}$/.test(r))return{jsonrpc:"2.0",id:a.id,error:{code:-32603,message:"Approval required but no approval_id returned"}};u("Tool gate: blocking until approval resolves",{tool:e,approvalId:r});const i=await Q({approvalId:r,wsUrl:o?.ws_url,pollUrl:o?.polling_url??`${m}/approval-status/${r}`,pollIntervalMs:ee,timeoutMs:te,token:h});u("Tool gate resolved",{tool:e,status:i.status}),q({approvalId:r,toolName:e,toolArgs:t,sessionKey:null,gate:"tool_gate",pollUrl:"",authToken:h},i.status,!0);const c=ne({approvalId:r,toolName:e,toolArgs:t,sessionKey:null,gate:"tool_gate",pollUrl:"",authToken:h},i);return{jsonrpc:"2.0",id:a.id,result:{content:[{type:"text",text:c}]}}}const n=a.result?._meta;if(n?.approval_id&&n?.status==="pending"){const o=n.approval_id;if(!/^[a-zA-Z0-9_-]{4,128}$/.test(o))return a;u("Permission gate: blocking until approval resolves",{tool:e,approvalId:o});const r=await Q({approvalId:o,wsUrl:n.ws_url,pollUrl:n.polling_url??`${m}/approval-status/${o}`,pollIntervalMs:ee,timeoutMs:te,token:h});u("Permission gate resolved",{tool:e,status:r.status}),q({approvalId:o,toolName:e,toolArgs:t,sessionKey:null,gate:"permission_gate",pollUrl:"",authToken:h},r.status,!0);const i=ne({approvalId:o,toolName:e,toolArgs:t,sessionKey:null,gate:"permission_gate",pollUrl:"",authToken:h},r);return{jsonrpc:"2.0",id:a.id,result:{content:[{type:"text",text:i}]}}}return a}function kt(e){Q({approvalId:e.approvalId,wsUrl:e.wsUrl,pollUrl:e.pollUrl,pollIntervalMs:ee,timeoutMs:te,token:e.authToken}).then(async t=>{u("Approval resolved",{tool:e.toolName,status:t.status,session:e.sessionKey});const a=ne(e,t),s=e.sessionKey??$?.getLastActiveSession()??null;if((t.status==="approved"||t.status==="executed")&&s&&De(s).catch(()=>{}),s&&$){if(await $.sendToSession(s,a)){u("Agent resumed via sessions.send",{session:s,status:t.status}),q(e,t.status,!0);return}y("sessions.send failed, falling back to channel notification",{session:s})}try{await x.notification({method:"notifications/claude/channel",params:{content:a,meta:{tool_name:e.toolName,approval_id:e.approvalId,status:t.status,gate:e.gate}}}),u("Agent resumed via channel notification",{tool:e.toolName,status:t.status}),q(e,t.status,!0)}catch(n){y("Channel notification failed",{error:n instanceof Error?n.message:String(n)}),q(e,t.status,!1)}}).catch(t=>{y("Background approval resolver error",{error:t instanceof Error?t.message:String(t)})})}function q(e,t,a){const s=typeof e.toolArgs.action=="string"?e.toolArgs.action:null,n=s??e.toolName.replace(/^(?:\w+__)*aerostack__/,"")??e.toolName,o=s?`${n}`:`${n}(${Object.keys(e.toolArgs).join(", ")})`,r=a?"resume sent to agent":"session unavailable \u2014 resume not delivered";let i;t==="approved"||t==="executed"?i=a?`[RESUMED] ${o} \u2014 ${r}`:`[RESUME FAILED] ${o} \u2014 ${r}`:t==="rejected"?i=`[REJECTED] ${o} \u2014 rejection sent to agent`:t==="changes_requested"?i=`[CHANGES REQUESTED] ${o} \u2014 feedback sent to agent`:i=`[EXPIRED] ${o} \u2014 approval timed out, expiry sent to agent`,O("tools/call",{name:"aerostack__guardian_report",arguments:{action:i.slice(0,500),category:"approval",risk_level:"low",details:JSON.stringify({tool:e.toolName,approvalId:e.approvalId,status:t,delivered:a,session:e.sessionKey}).slice(0,500)}}).catch(()=>{})}function ne(e,t){const a=t.reviewer_note?.trim()||null,s=`${e.toolName}(${Object.keys(e.toolArgs).join(", ")})`;switch(t.status){case"approved":case"executed":return e.gate==="tool_gate"?["[APPROVAL RECEIVED] Your pending action has been approved.","",`Action: ${s}`,a?`Reviewer note: "${a}"`:null,"",`Please now execute this action: call ${e.toolName} with the same parameters as before.`].filter(n=>n!==null).join(`
|
|
13
|
+
`):["[APPROVAL RECEIVED] Your request has been approved by the workspace owner.",a?`Reviewer note: "${a}"`:null,"","You may now proceed with the action you requested approval for."].filter(n=>n!==null).join(`
|
|
14
|
+
`);case"changes_requested":return["[CHANGES REQUESTED] The workspace owner reviewed your action and requested changes.","",`Original action: ${s}`,`Reviewer feedback: "${a??"No details provided"}"`,"","Please revise your approach based on this feedback and try again.","Do NOT proceed with the original plan."].join(`
|
|
15
|
+
`);case"rejected":return["[REJECTED] The workspace owner denied this action.","",`Action: ${s}`,a?`Reason: "${a}"`:"No reason provided.","","Do not retry this action. If you believe this was a mistake, inform the user."].join(`
|
|
16
|
+
`);case"expired":return[`[EXPIRED] The approval request for "${s}" timed out \u2014 no decision was made.`,"","You may resubmit the request if the action is still needed."].join(`
|
|
17
|
+
`);default:return`[APPROVAL UPDATE] Status for "${s}": ${t.status}.`}}const V=`
|
|
18
18
|
=== AEROSTACK GUARDIAN POLICY (MANDATORY) ===
|
|
19
19
|
|
|
20
20
|
You are operating under Aerostack workspace monitoring. These rules apply to ALL actions:
|
|
@@ -42,27 +42,26 @@ or before starting a new one, call aerostack__chat_check to see if the owner sen
|
|
|
42
42
|
If there are messages, read them carefully and respond using aerostack__chat_reply. This keeps
|
|
43
43
|
the owner informed and allows them to steer your work in real time.
|
|
44
44
|
=== END WORKSPACE CHAT ===
|
|
45
|
-
`.trim(),
|
|
45
|
+
`.trim(),nt=["delete","remove","drop","truncate","destroy","wipe","reset","deploy","publish","release","push","exec","shell","bash","run","command","terminal","install","uninstall","send","email","notify","webhook"];function ot(e,t){const a=e.toLowerCase(),s=nt.some(o=>a.includes(o)),n=t??"";return s?`[REQUIRES GUARDIAN APPROVAL] ${n}`.trim():n}let ye=null;async function K(){if(ye)return;const e=await O("initialize",{protocolVersion:"2024-11-05",capabilities:{},clientInfo:{name:"aerostack-gateway",version:"0.15.29"}});if(e.result){const t=e.result,a=t.instructions??"";ye={protocolVersion:t.protocolVersion??"2024-11-05",instructions:a?`${a}
|
|
46
46
|
|
|
47
|
-
${
|
|
47
|
+
${V}`:V}}}const x=new we({name:"aerostack-gateway",version:"0.15.29"},{capabilities:{tools:{},resources:{},prompts:{},experimental:{"claude/channel":{}}},instructions:V});x.setRequestHandler(Se,async()=>{await K();const e=await O("tools/list");if(e.error)throw new Error(e.error.message);const a=(e.result.tools??[]).filter(s=>!Ge.has(s.name)).map(s=>({...s,description:ot(s.name,s.description)}));return a.push(...de),{tools:a}}),x.setRequestHandler(Te,async e=>{await K();const{name:t,arguments:a}=e.params;if(Ve.has(t))return{content:t==="aerostack__chat_check"?await tt():await st(a??{})};const s=await at(t,a??{});if(s.error)return{content:[{type:"text",text:`Error: ${s.error.message}`}],isError:!0};const o=s.result.content??[{type:"text",text:JSON.stringify(s.result)}];if(Date.now()-L>ze)try{const i=await S(`${m}/chat/pending?since=${L}`,{headers:{Authorization:`Bearer ${h}`}});if(i.ok){const E=(await i.json()).messages??[];if(L=Date.now(),E.length>0){const b=[];for(const k of E){const C=await ae(k.id??null,k.metadata_json??null,k.content_type,k.content,k.sender_name);b.push(...C)}const D=[{type:"text",text:"--- WORKSPACE OWNER MESSAGE ---"},...b,{type:"text",text:"--- END MESSAGE \u2014 please acknowledge ---"}];o.unshift(...D)}}}catch{}return{content:o}});const rt={uri:"aerostack://guardian/policy",name:"Aerostack Guardian Policy",description:"Mandatory workspace monitoring policy. Read this to understand required approval workflows.",mimeType:"text/plain"};x.setRequestHandler(Ee,async()=>{await K();const e=await O("resources/list");if(e.error)throw new Error(e.error.message);const t=e.result;return{resources:[rt,...t.resources??[]]}}),x.setRequestHandler(ke,async e=>{if(await K(),e.params.uri==="aerostack://guardian/policy")return{contents:[{uri:e.params.uri,text:V,mimeType:"text/plain"}]};const t=await O("resources/read",{uri:e.params.uri});if(t.error)throw new Error(t.error.message);return{contents:t.result.contents??[]}}),x.setRequestHandler(Re,async()=>{await K();const e=await O("prompts/list");if(e.error)throw new Error(e.error.message);return{prompts:e.result.prompts??[]}}),x.setRequestHandler(Oe,async e=>{await K();const t=await O("prompts/get",{name:e.params.name,arguments:e.params.arguments});if(t.error)throw new Error(t.error.message);return{messages:t.result.messages??[]}});async function it(){const e=t=>process.stderr.write(t+`
|
|
48
48
|
`);e(`
|
|
49
49
|
Aerostack Connection Check
|
|
50
|
-
`);try{const
|
|
50
|
+
`);try{const t=await fetch(`${m}/chat/connection-test`,{headers:{Authorization:`Bearer ${h}`},signal:AbortSignal.timeout(1e4)});t.ok||(t.status===401&&(e(" \u2705 Workspace URL reachable"),e(" \u274C Token invalid (HTTP 401)"),e(`
|
|
51
51
|
Fix: Regenerate token at your dashboard \u2192 Workspace \u2192 Tokens
|
|
52
|
-
`),process.exit(1)),e(` \u274C Workspace unreachable (HTTP ${
|
|
52
|
+
`),process.exit(1)),e(` \u274C Workspace unreachable (HTTP ${t.status})`),e(`
|
|
53
53
|
Fix: Check AEROSTACK_WORKSPACE_URL is correct
|
|
54
|
-
`),process.exit(1));const
|
|
55
|
-
All checks passed. Your agent is connected.`),e(` Dashboard: ${
|
|
56
|
-
`),
|
|
57
|
-
`),process.exit(0)}catch(
|
|
54
|
+
`),process.exit(1));const a=await t.json();e(` \u2705 Workspace URL reachable (${m})`),e(` \u2705 Token valid (${a.token_name}, role: ${a.token_role})`),e(` \u2705 Workspace connected ("${a.workspace}")`),e(` \u2705 Tools available (${a.mcp_servers} tools)`);try{const s=await fetch(`${m}/chat/heartbeat`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({agent_type:"check",user_agent:"aerostack-gateway/check",bridge_id:"check"}),signal:AbortSignal.timeout(1e4)});e(s.ok?" \u2705 Dashboard registration (bridge visible in admin)":` \u26A0\uFE0F Dashboard registration (returned ${s.status})`)}catch{e(" \u26A0\uFE0F Dashboard registration (heartbeat failed \u2014 non-fatal)")}e(`
|
|
55
|
+
All checks passed. Your agent is connected.`),e(` Dashboard: ${a.dashboard_url||"https://app.aerostack.dev"}
|
|
56
|
+
`),a.mcp_servers===0&&e(` Note: 0 tools found. Add MCP servers or functions to your workspace first.
|
|
57
|
+
`),process.exit(0)}catch(t){t?.name==="TimeoutError"||t?.name==="AbortError"?(e(" \u274C Workspace unreachable (timeout after 10s)"),e(`
|
|
58
58
|
Fix: Check AEROSTACK_WORKSPACE_URL and your network connection
|
|
59
|
-
`)):(e(` \u274C Connection failed (${
|
|
59
|
+
`)):(e(` \u274C Connection failed (${t?.message??"unknown error"})`),e(`
|
|
60
60
|
Fix: Check AEROSTACK_WORKSPACE_URL is correct
|
|
61
|
-
`)),process.exit(1)}}async function
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
`)
|
|
67
|
-
`)
|
|
68
|
-
`).pop()||k.trim();l&&(m=[{text:l}])}m.length===0&&(m=[{text:`Message received. I'll process "${O.slice(0,100)}" shortly.`}]),await new Promise(l=>setTimeout(l,500));for(const l of w)l.status==="running"&&(l.status="timeout",l.ended_at=Date.now(),l.duration_ms=l.ended_at-l.started_at);if(w.length>0){const l=w.reduce((g,U)=>g+(U.duration_ms||0),0),h=w.length;try{await E(`${p}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${d}`,"Content-Type":"application/json"},body:JSON.stringify({content:`Used ${h} tool${h>1?"s":""}`,content_type:"tool_activity",session_id:a,metadata:JSON.stringify({v:1,tools:w.map(g=>({name:g.name,category:g.category,duration_ms:g.duration_ms,status:g.status,args_summary:g.args_summary?.slice(0,500),error:g.error?.slice(0,500)})),total_duration_ms:l})})})}catch{}}z.delete(C),P===C&&(P=null);for(const l of m){const h=l.text?.trim()||"";if(l.mediaUrl){let g=!1;try{g=new URL(l.mediaUrl).protocol==="https:"}catch{}if(g)try{const U=await E(`${p}/chat/media`,{method:"POST",headers:{Authorization:`Bearer ${d}`,"Content-Type":"application/json"},body:JSON.stringify({media_url:l.mediaUrl,content:h||void 0,session_id:a})});U.ok?u("Chat media sent to workspace",{mediaUrl:l.mediaUrl}):_("Chat media upload failed",{status:U.status})}catch(U){_("Failed to send chat media",{error:U?.message})}else _("Skipping invalid media URL",{mediaUrl:l.mediaUrl});K(f,"agent",h||`[media: ${l.mediaUrl}]`)}else if(h&&h.length>=2){K(f,"agent",h);try{const g=await E(`${p}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${d}`,"Content-Type":"application/json"},body:JSON.stringify({content:h.slice(0,1e4),session_id:a})});g.ok?u("Chat reply sent to workspace"):_("Chat reply failed",{status:g.status})}catch(g){_("Failed to send chat reply",{error:g?.message})}}}})}catch(y){_("Failed to spawn agent turn",{error:y?.message}),z.delete(C),P===C&&(P=null)}}catch{}}function me(){rt(),b?.stop(),ve(),process.exit(0)}process.on("SIGTERM",()=>{me()}),process.on("SIGINT",()=>{me()}),process.argv.includes("--check")?st():at().catch(e=>{Ne("Fatal error",{error:e instanceof Error?e.message:String(e)}),process.exit(1)});
|
|
61
|
+
`)),process.exit(1)}}async function ct(){try{const e=await fetch(`${m}/chat/connection-test`,{headers:{Authorization:`Bearer ${h}`},signal:AbortSignal.timeout(5e3)});if(e.ok){const t=await e.json();u(`Connected to "${t.workspace}" (${t.mcp_servers} tools)`,{dashboard:"https://app.aerostack.dev"})}else e.status===401?y("Token invalid \u2014 regenerate at dashboard \u2192 Workspace \u2192 Tokens"):y("Connection check failed",{status:e.status})}catch{y("Could not verify connection (non-fatal)")}}async function lt(){u("Connecting to workspace",{url:m});const e=new Ae;await x.connect(e),u("Ready",{url:m}),ct().catch(()=>{});const t=()=>{S(`${m}/chat/heartbeat`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({agent_type:ue,user_agent:"aerostack-gateway/0.15.29",bridge_id:se})}).catch(()=>{})};t();const a=setInterval(t,12e4);if(process.on("exit",()=>clearInterval(a)),Me)try{const n=await Ne(async o=>{try{const r=await fetch(`${m}/guardian-batch`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${h}`,"User-Agent":"aerostack-gateway/0.15.29","X-Agent-Id":"aerostack-gateway"},body:JSON.stringify({events:o})});return r.ok?(await r.json()).config?.hook_tracking??null:null}catch{return null}},je);Ke&&await Pe(n)&&u("Claude Code hook auto-installed",{port:n})}catch(s){y("Hook server failed to start (non-fatal)",{error:s instanceof Error?s.message:String(s)})}if(He)try{const s=We??await Ue();s?($=new xe({port:le,token:s,rpcCall:O,onToolEvent:o=>{const r=I;if(!r)return;const i=F.get(r);if(i){if(o.phase==="start")i.push({name:o.toolName,category:o.category,started_at:Date.now(),ended_at:null,duration_ms:null,status:"running",args_summary:o.summary});else if(o.phase==="end"){const c=[...i].reverse().find(E=>E.name===o.toolName&&E.status==="running");c&&(c.ended_at=Date.now(),c.duration_ms=c.ended_at-c.started_at,c.status=o.error?"error":"success",o.error&&(c.error=o.error))}S(`${m}/chat/tool-status`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({tool_name:o.toolName,phase:o.phase,category:o.category})}).catch(()=>{})}}}),await $.connect()?u("OpenClaw connector started",{port:le}):(u("OpenClaw gateway not reachable, skipping connector"),$=null)):u("OpenClaw integration skipped (no token found)")}catch(s){y("OpenClaw connector failed (non-fatal)",{error:s instanceof Error?s.message:String(s)})}process.env.AEROSTACK_CHAT_LISTENER!=="false"&&dt()}let T=null,J=!1;async function ut(){return typeof globalThis.WebSocket<"u"?globalThis.WebSocket:(await import("ws")).default}function dt(){J=!1;let e=null,t=0;const a=async()=>{if(!J){if(T){try{T.close()}catch{}T=null}try{const n=await ut(),o=m.replace(/^http/,"ws")+`/chat-relay?token=${encodeURIComponent(h||"")}&client_type=bridge&client_id=${encodeURIComponent(se)}`,r=new n(o);T=r,r.onopen=()=>{u("[chat-relay] WebSocket connected to workspace"),t=0},r.onmessage=i=>{const c=typeof i.data=="string"?i.data:"";if(c)try{if(JSON.parse(c).type==="pong")return;u("[chat-relay] WS frame received",{dataLength:c.length}),ht(c)}catch{}},r.onclose=i=>{u("[chat-relay] WebSocket closed",{code:i.code,reason:i.reason}),T=null,J||s()},r.onerror=()=>{y("[chat-relay] WebSocket error"),t++}}catch(n){y("[chat-relay] Connection failed",{error:n?.message}),t++,s()}}},s=()=>{if(J||e)return;const n=Math.min(2e3*Math.pow(2,t),3e4);u(`[chat-relay] Reconnecting in ${n/1e3}s (failures: ${t})`),e=setTimeout(()=>{e=null,a()},n)};a()}function ge(e,t){if(!T)return!1;try{return T.send(JSON.stringify({type:"chat_reply",content:e.slice(0,1e4),session_id:t||P,sender_name:"Agent",content_type:"text"})),!0}catch{return!1}}function oe(e,t=!0){if(T)try{T.send(JSON.stringify({type:"typing",session_id:e||P,is_typing:t}))}catch{}}function pt(){if(J=!0,T){try{T.close()}catch{}T=null}}async function ht(e){try{const t=JSON.parse(e);if(t.type!=="chat_message"){u("[chat-trace] Event ignored (not chat_message)",{type:t.type});return}const a=Date.now();u("[chat-trace] Chat event received",{type:t.type,sender:t.sender_name,contentType:t.content_type,hasContent:!!t.content,sessionId:t.session_id,messageId:t.message_id});let s=[];if(t.metadata_json)try{const g=JSON.parse(t.metadata_json);g?.v===1&&Array.isArray(g.attachments)&&(s=g.attachments)}catch{}const n=!!t.content&&t.content_type!=="media",o=s.length>0;if(!n&&!o)return;const r=t.session_id;u("Chat message received from admin",{sender:t.sender_name,sessionId:r,contentType:t.content_type,attachments:s.length});const i=/^[a-zA-Z0-9_-]{1,128}$/.test(t.message_id??"")?t.message_id:null,c=s.map(g=>{const U=Math.round(g.size/1024),Y=U>=1024?`${(U/1024).toFixed(1)} MB`:`${U} KB`,X=_((g.filename||"file").slice(0,200)),M=_((g.mime_type||"application/octet-stream").slice(0,100)),v=i?`${m}/chat/media/${i}/${encodeURIComponent(g.filename)}`:"(unavailable)";return`[Attached: ${X} (${M}, ${Y}) \u2014 ${v}]`}),E=c.length>0?`
|
|
62
|
+
`+c.join(`
|
|
63
|
+
`):"",b=n?t.content+(E?`
|
|
64
|
+
`+c.map(g=>g).join(`
|
|
65
|
+
`):""):c.join(`
|
|
66
|
+
`);r&&(P=r,j(r,"user",b)),z.length>=Qe&&z.shift(),z.push({sender_name:t.sender_name??"Admin",content:n?t.content:c.join(", "),created_at:Date.now(),session_id:r,content_type:t.content_type,metadata_json:t.metadata_json,message_id:t.message_id}),L=Date.now();const D=_(t.sender_name??"Admin"),k=n?_(t.content):"",C=r||P||"_default",H=`aerostack-chat-${/^[a-zA-Z0-9_-]{1,64}$/.test(C)?C:"_default"}`,R=`turn-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,w=[];F.set(R,w),I=R;try{const g=k?k+E:E.trim(),U=`[${D} via Aerostack Dashboard]: ${g}`;if(oe(r,!0),$){const M=Date.now();u("[chat-trace] WS fast path starting",{session:H,promptLength:U.length,elapsedMs:M-a});const v=await $.sendChatMessage(U);if(v){const A=v.trim(),l=Date.now();u("[chat-trace] WS fast path got response",{responseLength:A.length,llmMs:l-M,totalMs:l-a}),await new Promise(d=>setTimeout(d,500));for(const d of w)d.status==="running"&&(d.status="timeout",d.ended_at=Date.now(),d.duration_ms=d.ended_at-d.started_at);if(w.length>0){const d=w.reduce((p,G)=>p+(G.duration_ms||0),0),f=w.length;try{await S(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({content:`Used ${f} tool${f>1?"s":""}`,content_type:"tool_activity",session_id:r,metadata:JSON.stringify({v:1,tools:w.map(p=>({name:p.name,category:p.category,duration_ms:p.duration_ms,status:p.status,args_summary:p.args_summary?.slice(0,500),error:p.error?.slice(0,500)})),total_duration_ms:d})})})}catch{}}if(F.delete(R),I===R&&(I=null),oe(r,!1),A.length>=2)if(j(C,"agent",A),ge(A,r))u("[chat-trace] Reply sent via WS upstream fast path (no HTTP POST needed)",{totalMs:Date.now()-a});else{const f=Date.now();try{const p=await S(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({content:A.slice(0,1e4),session_id:r})}),G=Date.now()-f;p.ok?u("[chat-trace] Reply sent (HTTP fallback)",{replyMs:G,totalMs:Date.now()-a}):y("[chat-trace] Reply failed (HTTP fallback)",{status:p.status,replyMs:G})}catch(p){y("[chat-trace] Reply error (HTTP fallback)",{error:p?.message,replyMs:Date.now()-f})}}return}u("[chat-trace] WS fast path failed, falling back to CLI",{elapsedMs:Date.now()-a})}const{execFile:Y}=await import("child_process"),X=Date.now();u("[chat-trace] CLI fallback starting",{sender:D,session:H,elapsedMs:X-a}),Y("openclaw",["agent","--agent","main","--json","--session-id",H,"-m",U],{timeout:12e4},async(M,v)=>{let A=[];if(!M&&v)try{const l=JSON.parse(v.trim()),d=l?.result?.payloads;Array.isArray(d)&&d.length>0?A=d:(l.text||l.reply||l.message)&&(A=[{text:l.text||l.reply||l.message}])}catch{const l=v.trim().split(`
|
|
67
|
+
`).pop()||v.trim();l&&(A=[{text:l}])}A.length===0&&(A=[{text:`Message received. I'll process "${k.slice(0,100)}" shortly.`}]),await new Promise(l=>setTimeout(l,500));for(const l of w)l.status==="running"&&(l.status="timeout",l.ended_at=Date.now(),l.duration_ms=l.ended_at-l.started_at);if(w.length>0){const l=w.reduce((f,p)=>f+(p.duration_ms||0),0),d=w.length;try{await S(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({content:`Used ${d} tool${d>1?"s":""}`,content_type:"tool_activity",session_id:r,metadata:JSON.stringify({v:1,tools:w.map(f=>({name:f.name,category:f.category,duration_ms:f.duration_ms,status:f.status,args_summary:f.args_summary?.slice(0,500),error:f.error?.slice(0,500)})),total_duration_ms:l})})})}catch{}}F.delete(R),I===R&&(I=null),oe(r,!1);for(const l of A){const d=l.text?.trim()||"";if(l.mediaUrl){let f=!1;try{f=new URL(l.mediaUrl).protocol==="https:"}catch{}if(f)try{const p=await S(`${m}/chat/media`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({media_url:l.mediaUrl,content:d||void 0,session_id:r})});p.ok?u("Chat media sent to workspace",{mediaUrl:l.mediaUrl}):y("Chat media upload failed",{status:p.status})}catch(p){y("Failed to send chat media",{error:p?.message})}else y("Skipping invalid media URL",{mediaUrl:l.mediaUrl});j(C,"agent",d||`[media: ${l.mediaUrl}]`)}else if(d&&d.length>=2)if(j(C,"agent",d),ge(d,r))u("Chat reply sent via WS upstream (no HTTP POST needed)");else try{const p=await S(`${m}/chat/messages`,{method:"POST",headers:{Authorization:`Bearer ${h}`,"Content-Type":"application/json"},body:JSON.stringify({content:d.slice(0,1e4),session_id:r})});p.ok?u("Chat reply sent to workspace"):y("Chat reply failed",{status:p.status})}catch(p){y("Failed to send chat reply",{error:p?.message})}}})}catch(g){y("Failed to spawn agent turn",{error:g?.message}),F.delete(R),I===R&&(I=null)}}catch{}}function _e(){pt(),$?.stop(),Ie(),process.exit(0)}process.on("SIGTERM",()=>{_e()}),process.on("SIGINT",()=>{_e()}),process.argv.includes("--check")?it():lt().catch(e=>{Le("Fatal error",{error:e instanceof Error?e.message:String(e)}),process.exit(1)});
|
package/dist/logger.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
const
|
|
2
|
-
|
|
1
|
+
import{appendFileSync as p,mkdirSync as u}from"node:fs";import{join as i}from"node:path";import{homedir as a}from"node:os";const n={debug:0,info:1,warn:2,error:3},g=(()=>{const r=(process.env.AEROSTACK_LOG_LEVEL??"info").toLowerCase();return r in n?r:"info"})(),c=(()=>{const r=process.env.AEROSTACK_LOG_FILE;if(r)return r;try{const o=i(a(),".aerostack","logs");return u(o,{recursive:!0}),i(o,"bridge.log")}catch{return null}})();function t(r,o,s){if(n[r]<n[g])return;const f={ts:new Date().toISOString(),level:r,msg:o,...s},e=JSON.stringify(f)+`
|
|
2
|
+
`;if(process.stderr.write(e),c)try{p(c,e)}catch{}}const L=(r,o)=>t("debug",r,o),S=(r,o)=>t("info",r,o),w=(r,o)=>t("warn",r,o),E=(r,o)=>t("error",r,o);export{L as debug,E as error,S as info,t as log,w as warn};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
import{readFile as m}from"node:fs/promises";import{join as f}from"node:path";import{homedir as w}from"node:os";import{info as u,warn as c,debug as d}from"./logger.js";import{addToBatch as p,detectCategory as b,summarizeToolInput as y}from"./hook-server.js";function S(l){return l.replace(/^(?:\w+__)*aerostack__/,"").replace(/__/g,":")||l}async function N(){try{const l=f(w(),".openclaw","openclaw.json"),e=await m(l,"utf-8");return JSON.parse(e)?.gateway?.auth?.token??null}catch{return null}}async function R(){try{const l=f(w(),".openclaw","exec-approvals.json"),e=await m(l,"utf-8");return JSON.parse(e)?.socket?.token??null}catch{return null}}const k=1e3,_=3e4;class A{opts;ws=null;destroyed=!1;reconnectMs=k;reconnectTimer=null;requestId=0;seenSessions=new Set;pendingRequests=new Map;lastActiveSession=null;pendingChatTurns=new Map;constructor(e){this.opts=e}async connect(){if(this.destroyed)return!1;const e=await this.getWebSocket(),s=`ws://127.0.0.1:${this.opts.port}`;return new Promise(t=>{try{const n=new e(s);this.ws=n;const o=setTimeout(()=>{d("OpenClaw connect timeout");try{n.close()}catch{}t(!1)},1e4);n.onopen=()=>{clearTimeout(o),d("OpenClaw WS connected, sending handshake"),this.sendHandshake()},n.onmessage=i=>{try{const r=JSON.parse(String(i.data));this.handleFrame(r,t)}catch{}},n.onerror=()=>{clearTimeout(o),t(!1)},n.onclose=()=>{clearTimeout(o),this.ws=null,this.destroyed||this.scheduleReconnect()}}catch{t(!1)}})}getLastActiveSession(){return this.lastActiveSession}async sendToSession(e,s){if(!this.ws||!this.connected)return!1;const t=await this.sendRequest("sessions.send",{key:e,message:s,idempotencyKey:`aerostack-${Date.now()}`});return t.ok||c("sendToSession failed",{key:e,error:t.error?.message}),t.ok??!1}async sendChatMessage(e,s=12e4){if(!this.ws||!this.connected)return null;const t="agent:main:main";this.subscribeToSessionMessages(t);const n=new Promise(i=>{const r=setTimeout(()=>{this.pendingChatTurns.delete(t),i(null)},s);this.pendingChatTurns.set(t,a=>{clearTimeout(r),i(a)})}),o=await this.sendRequest("sessions.send",{key:t,message:e,idempotencyKey:`aerostack-${Date.now()}-${Math.random().toString(36).slice(2,8)}`});return o.ok?n:(this.pendingChatTurns.delete(t),c("sendChatMessage: sessions.send rejected",{error:o.error?.message}),null)}stop(){if(this.destroyed=!0,this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null),this.ws){try{this.ws.close(1e3)}catch{}this.ws=null}this.pendingRequests.clear(),this.pendingChatTurns.clear()}sendHandshake(){this.send({type:"req",id:String(++this.requestId),method:"connect",params:{minProtocol:3,maxProtocol:3,client:{id:"openclaw-tui",displayName:"Aerostack Guardian",version:"0.15.
|
|
1
|
+
import{readFile as m}from"node:fs/promises";import{join as f}from"node:path";import{homedir as w}from"node:os";import{info as u,warn as c,debug as d}from"./logger.js";import{addToBatch as p,detectCategory as b,summarizeToolInput as y}from"./hook-server.js";function S(l){return l.replace(/^(?:\w+__)*aerostack__/,"").replace(/__/g,":")||l}async function N(){try{const l=f(w(),".openclaw","openclaw.json"),e=await m(l,"utf-8");return JSON.parse(e)?.gateway?.auth?.token??null}catch{return null}}async function R(){try{const l=f(w(),".openclaw","exec-approvals.json"),e=await m(l,"utf-8");return JSON.parse(e)?.socket?.token??null}catch{return null}}const k=1e3,_=3e4;class A{opts;ws=null;destroyed=!1;reconnectMs=k;reconnectTimer=null;requestId=0;seenSessions=new Set;pendingRequests=new Map;lastActiveSession=null;pendingChatTurns=new Map;constructor(e){this.opts=e}async connect(){if(this.destroyed)return!1;const e=await this.getWebSocket(),s=`ws://127.0.0.1:${this.opts.port}`;return new Promise(t=>{try{const n=new e(s);this.ws=n;const o=setTimeout(()=>{d("OpenClaw connect timeout");try{n.close()}catch{}t(!1)},1e4);n.onopen=()=>{clearTimeout(o),d("OpenClaw WS connected, sending handshake"),this.sendHandshake()},n.onmessage=i=>{try{const r=JSON.parse(String(i.data));this.handleFrame(r,t)}catch{}},n.onerror=()=>{clearTimeout(o),t(!1)},n.onclose=()=>{clearTimeout(o),this.ws=null,this.destroyed||this.scheduleReconnect()}}catch{t(!1)}})}getLastActiveSession(){return this.lastActiveSession}async sendToSession(e,s){if(!this.ws||!this.connected)return!1;const t=await this.sendRequest("sessions.send",{key:e,message:s,idempotencyKey:`aerostack-${Date.now()}`});return t.ok||c("sendToSession failed",{key:e,error:t.error?.message}),t.ok??!1}async sendChatMessage(e,s=12e4){if(!this.ws||!this.connected)return null;const t="agent:main:main";this.subscribeToSessionMessages(t);const n=new Promise(i=>{const r=setTimeout(()=>{this.pendingChatTurns.delete(t),i(null)},s);this.pendingChatTurns.set(t,a=>{clearTimeout(r),i(a)})}),o=await this.sendRequest("sessions.send",{key:t,message:e,idempotencyKey:`aerostack-${Date.now()}-${Math.random().toString(36).slice(2,8)}`});return o.ok?n:(this.pendingChatTurns.delete(t),c("sendChatMessage: sessions.send rejected",{error:o.error?.message}),null)}stop(){if(this.destroyed=!0,this.reconnectTimer&&(clearTimeout(this.reconnectTimer),this.reconnectTimer=null),this.ws){try{this.ws.close(1e3)}catch{}this.ws=null}this.pendingRequests.clear(),this.pendingChatTurns.clear()}sendHandshake(){this.send({type:"req",id:String(++this.requestId),method:"connect",params:{minProtocol:3,maxProtocol:3,client:{id:"openclaw-tui",displayName:"Aerostack Guardian",version:"0.15.29",platform:process.platform,mode:"cli"},auth:{token:this.opts.token},scopes:["operator.read","operator.write"],caps:["tool-events"]}})}connected=!1;subscribedSessions=new Set;handleFrame(e,s){if(e.type==="res"&&!this.connected){e.ok?(this.connected=!0,this.reconnectMs=k,u("OpenClaw connector connected",{port:this.opts.port}),s?.(!0),this.subscribeAllSessions().catch(t=>c("subscribeAllSessions failed",{error:t?.message}))):(c("OpenClaw connect rejected",{error:e.error?.message}),s?.(!1));return}if(e.type==="res"&&e.id&&this.pendingRequests.has(e.id)){const t=this.pendingRequests.get(e.id);this.pendingRequests.delete(e.id),t(e);return}if(e.type==="res"&&e.ok===!1&&e.error){c("OpenClaw RPC error",{error:e.error.message,code:e.error.code});return}e.type==="event"&&this.handleEvent(e)}subscribeToSessionMessages(e){if(this.subscribedSessions.has(e))return;this.subscribedSessions.add(e);const s=String(++this.requestId);this.pendingRequests.set(s,t=>{t.ok?d("OpenClaw subscribed to session messages",{key:e}):c("OpenClaw messages.subscribe failed",{key:e,error:t.error?.message})}),this.send({type:"req",id:s,method:"sessions.messages.subscribe",params:{key:e}}),setTimeout(()=>this.pendingRequests.delete(s),1e4)}async subscribeAllSessions(){const e=await this.sendRequest("sessions.subscribe");e.ok?u("OpenClaw subscribed to all session events (tool + lifecycle)"):c("OpenClaw sessions.subscribe failed",{error:e.error?.message});const s=await this.sendRequest("sessions.list",{limit:50});if(!s.ok){c("OpenClaw sessions.list failed",{error:s.error?.message});return}const t=s.payload?.sessions,n=Array.isArray(t)?t:[];u("OpenClaw active sessions",{count:n.length});for(const o of n)o.key&&this.subscribeToSessionMessages(o.key)}sendRequest(e,s){return new Promise(t=>{const n=String(++this.requestId),o=setTimeout(()=>{this.pendingRequests.delete(n),t({type:"res",id:n,ok:!1,error:{code:"TIMEOUT",message:"request timeout"}})},1e4);this.pendingRequests.set(n,i=>{clearTimeout(o),t(i)}),this.send({type:"req",id:n,method:e,params:s})})}handleEvent(e){const s=e.event,t=e.payload??{};s==="agent"?this.handleAgentEvent(t):s==="session.tool"?this.handleToolEvent(t):s==="session.message"?this.handleMessageEvent(t):s==="sessions.changed"?this.handleSessionChanged(t):(s==="exec.approval"||s==="exec.request"||s==="tool.approval")&&this.handleExecApprovalEvent(t)}handleAgentEvent(e){const s=e.stream,t=e.data,n=e.sessionKey??"";if(n&&this.seenSessions.add(n),s==="tool"){const o=t?.name??t?.toolName??e.toolName??"unknown",i=t?.phase??"",r=t?.args??e.args??{};if(i==="start"){const{category:a,risk:h}=b(o,r),g=y(o,r);p({action:`OpenClaw ${S(o)}: ${g}`.slice(0,500),category:a,risk_level:h,details:JSON.stringify({tool:o,session:n,...r}).slice(0,500),agent_name:"OpenClaw"}),d("OpenClaw agent tool event",{tool:o,category:a,risk:h})}}if(s==="lifecycle"&&t){const o=t.phase??"";(o==="start"||o==="end")&&d("OpenClaw agent lifecycle",{phase:o,session:n})}}handleToolEvent(e){const s=e.data,t=s?.name??s?.toolName??e.toolName??"unknown",n=s?.phase??"",o=s?.args??e.args??{},i=e.sessionKey??"";if(n!=="start"&&n!=="end")return;const{category:r,risk:a}=b(t,o),h=y(t,o);if(n==="start"&&p({action:`OpenClaw ${S(t)}: ${h}`.slice(0,500),category:r,risk_level:a,details:JSON.stringify({tool:t,session:i,...o}).slice(0,500),agent_name:"OpenClaw"}),d("OpenClaw tool event",{tool:t,phase:n,category:r,risk:a,session:i}),!new Set(["aerostack__guardian_report","aerostack__chat_check","aerostack__chat_reply","aerostack__check_approval","aerostack__local_guardian","aerostack__guardian_check"]).has(t)){const T=s?.error??e.error??void 0;this.opts.onToolEvent?.({toolName:t,phase:n,category:r,summary:h,error:T})}i&&(this.seenSessions.add(i),this.lastActiveSession=i)}handleExecApprovalEvent(e){const s=e.command??e.cmd??"",t=e.sessionKey??"",n=e.agentId??"";if(!s)return;const o=s.toLowerCase();let i="medium",r="exec_command";/\brm\s+-rf?\b|\bdrop\b|\bdelete\b|\btruncate\b/.test(o)?(i="critical",r="file_delete"):/\brm\b|\bgit\s+push\b|\bgit\s+reset\b|\bdeploy\b|\bkill\b/.test(o)?(i="high",r="deploy"):/\binstall\b|\bpip\b|\bnpm\b|\bcurl\b|\bwget\b/.test(o)&&(i="medium",r="package_install"),p({action:`[approval requested] exec: ${s}`.slice(0,500),category:r,risk_level:i,details:JSON.stringify({command:s,agent:n,session:t,source:"exec.approval"}).slice(0,500),agent_name:"OpenClaw"}),this.opts.rpcCall("tools/call",{name:"aerostack__guardian_report",arguments:{action:`OpenClaw exec approval: ${s}`.slice(0,500),category:r,risk_level:i,details:JSON.stringify({command:s,agent:n,session:t}).slice(0,500)}}).catch(()=>{}),u("OpenClaw exec approval event",{command:s.slice(0,100),risk:i})}handleMessageEvent(e){const s=e.sessionKey??"";s&&(this.seenSessions.add(s),this.lastActiveSession=s);const t=this.pendingChatTurns.get(s);if(!t)return;const n=e.message,o=typeof n=="object"?n?.role:void 0;let i="";if(typeof n=="string")i=n;else if(n){const r=n.content;typeof r=="string"?i=r:Array.isArray(r)&&(i=r.map(a=>typeof a=="string"?a:a?.text??"").join(""))}i&&o!=="user"&&(this.pendingChatTurns.delete(s),t(i))}handleSessionChanged(e){const s=e.sessionKey??"";s&&(this.seenSessions.add(s),this.subscribeToSessionMessages(s))}send(e){if(this.ws)try{this.ws.send(JSON.stringify(e))}catch{}}scheduleReconnect(){this.destroyed||this.reconnectTimer||(d("OpenClaw reconnecting in",{ms:this.reconnectMs}),this.reconnectTimer=setTimeout(async()=>{this.reconnectTimer=null,this.connected=!1,this.seenSessions.clear(),this.subscribedSessions.clear(),this.pendingRequests.clear(),await this.connect()},this.reconnectMs),this.reconnectMs=Math.min(this.reconnectMs*2,_))}async getWebSocket(){return typeof globalThis.WebSocket<"u"?globalThis.WebSocket:(await import("ws")).default}}export{A as OpenClawConnector,R as resolveExecApprovalToken,N as resolveOpenClawToken};
|
package/dist/resolution.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import{info as y,warn as _,debug as h}from"./logger.js";function J(e,o){const{approvalId:n}=e;let a=!1,s=null,r=null,g=null;const v=()=>{if(a=!0,r&&(clearInterval(r),r=null),g&&(clearTimeout(g),g=null),s){try{s.close(1e3)}catch{}s=null}},c=f=>{a||(o.resolve(n,f.status,f.reviewer_note),y("Background resolver: approval resolved",{approvalId:n,status:f.status}),v())};return(async()=>{if(e.wsUrl)try{const d=await O(),l=new d(e.wsUrl);s=l,l.onmessage=S=>{try{const i=typeof S.data=="string"?JSON.parse(S.data):JSON.parse(String(S.data)),p=i?.status;(p==="executed"||p==="approved"||p==="rejected"||p==="expired"||p==="changes_requested")&&c({status:p,reviewer_note:i?.reviewer_note})}catch{}},l.onerror=()=>{h("Background resolver WS error",{approvalId:n})},l.onclose=()=>{s=null},typeof l.on=="function"&&l.on("unexpected-response",async(S,i)=>{try{const p=[];for await(const w of i)p.push(w);const k=Buffer.concat(p).toString(),t=JSON.parse(k);t?.status&&t.status!=="pending"&&c({status:t.status,reviewer_note:t?.reviewer_note})}catch{}})}catch{h("Background resolver WS connect failed, using polling only",{approvalId:n})}r=setInterval(async()=>{if(!a)try{const d=await W(e.pollUrl);d&&c(d)}catch{}},3e4);try{const d=await W(e.pollUrl);if(d){c(d);return}}catch{}const m=e.timeoutMs??2*60*60*1e3;g=setTimeout(()=>{a||(o.resolve(n,"expired"),v())},m)})().catch(()=>{}),{cancel:v}}async function N(e){const{approvalId:o,wsUrl:n,pollUrl:a,pollIntervalMs:s,timeoutMs:r,token:g}=e,v=Date.now()+r;if(n)try{return await L(o,n,a,v,g)}catch(c){const f=c instanceof Error?c.message:"Unknown WS error";_("WebSocket resolution failed, falling back to polling",{approvalId:o,error:f})}return P(o,a,s,v,g)}async function L(e,o,n,a,s){const r=await O();return new Promise((g,v)=>{let c=!1,f=null,m=null;const d=()=>{c=!0,f&&clearInterval(f),m&&clearTimeout(m)},l=t=>{c||(d(),g(t))},S=t=>{c||(d(),v(t))};h("Connecting WebSocket",{approvalId:e,wsUrl:o});const i=new r(o);i.onopen=()=>{y("WebSocket connected, waiting for resolution",{approvalId:e})},i.onmessage=t=>{try{const w=typeof t.data=="string"?JSON.parse(t.data):JSON.parse(String(t.data)),u=w?.status;if(h("WebSocket message received",{approvalId:e,status:u}),u==="executed"||u==="approved"||u==="rejected"||u==="expired"||u==="changes_requested"){y("Approval resolved via WebSocket",{approvalId:e,status:u}),l({status:u,reviewer_note:w?.reviewer_note});try{i.close(1e3)}catch{}}}catch{_("Failed to parse WebSocket message",{approvalId:e})}},i.onerror=t=>{const w=t instanceof Error?t.message:"WebSocket error";h("WebSocket error",{approvalId:e,error:w}),S(new Error(`WebSocket error: ${w}`))},i.onclose=t=>{c||(h("WebSocket closed without resolution",{approvalId:e,code:t.code}),S(new Error(`WebSocket closed unexpectedly (code ${t.code})`)))},typeof i.on=="function"&&i.on("unexpected-response",async(t,w)=>{try{const u=[];for await(const A of w)u.push(A);const T=Buffer.concat(u).toString(),x=JSON.parse(T),b=x?.status;if(b&&b!=="pending"){y("Approval already resolved (WS endpoint returned JSON)",{approvalId:e,status:b}),l({status:b,reviewer_note:x?.reviewer_note});return}}catch{}S(new Error("WebSocket upgrade rejected by server"))}),f=setInterval(async()=>{if(!c)try{const t=await W(n,s);if(t){y("Approval resolved via safety-net poll",{approvalId:e,status:t.status}),l(t);try{i.close(1e3)}catch{}}}catch{}},3e4);const k=a-Date.now();if(k<=0){S(new Error("Approval timeout"));return}m=setTimeout(()=>{if(!c){_("Approval timed out",{approvalId:e}),l({status:"expired"});try{i.close(1e3)}catch{}}},k)})}async function P(e,o,n,a,s){for(y("Polling for approval resolution",{approvalId:e,intervalMs:n});Date.now()<a;){await B(n);try{const r=await W(o,s);if(r)return y("Approval resolved via polling",{approvalId:e,status:r.status}),r}catch{}}return _("Approval polling timed out",{approvalId:e}),{status:"expired"}}async function W(e,o){const n={"User-Agent":"aerostack-gateway/0.15.
|
|
1
|
+
import{info as y,warn as _,debug as h}from"./logger.js";function J(e,o){const{approvalId:n}=e;let a=!1,s=null,r=null,g=null;const v=()=>{if(a=!0,r&&(clearInterval(r),r=null),g&&(clearTimeout(g),g=null),s){try{s.close(1e3)}catch{}s=null}},c=f=>{a||(o.resolve(n,f.status,f.reviewer_note),y("Background resolver: approval resolved",{approvalId:n,status:f.status}),v())};return(async()=>{if(e.wsUrl)try{const d=await O(),l=new d(e.wsUrl);s=l,l.onmessage=S=>{try{const i=typeof S.data=="string"?JSON.parse(S.data):JSON.parse(String(S.data)),p=i?.status;(p==="executed"||p==="approved"||p==="rejected"||p==="expired"||p==="changes_requested")&&c({status:p,reviewer_note:i?.reviewer_note})}catch{}},l.onerror=()=>{h("Background resolver WS error",{approvalId:n})},l.onclose=()=>{s=null},typeof l.on=="function"&&l.on("unexpected-response",async(S,i)=>{try{const p=[];for await(const w of i)p.push(w);const k=Buffer.concat(p).toString(),t=JSON.parse(k);t?.status&&t.status!=="pending"&&c({status:t.status,reviewer_note:t?.reviewer_note})}catch{}})}catch{h("Background resolver WS connect failed, using polling only",{approvalId:n})}r=setInterval(async()=>{if(!a)try{const d=await W(e.pollUrl);d&&c(d)}catch{}},3e4);try{const d=await W(e.pollUrl);if(d){c(d);return}}catch{}const m=e.timeoutMs??2*60*60*1e3;g=setTimeout(()=>{a||(o.resolve(n,"expired"),v())},m)})().catch(()=>{}),{cancel:v}}async function N(e){const{approvalId:o,wsUrl:n,pollUrl:a,pollIntervalMs:s,timeoutMs:r,token:g}=e,v=Date.now()+r;if(n)try{return await L(o,n,a,v,g)}catch(c){const f=c instanceof Error?c.message:"Unknown WS error";_("WebSocket resolution failed, falling back to polling",{approvalId:o,error:f})}return P(o,a,s,v,g)}async function L(e,o,n,a,s){const r=await O();return new Promise((g,v)=>{let c=!1,f=null,m=null;const d=()=>{c=!0,f&&clearInterval(f),m&&clearTimeout(m)},l=t=>{c||(d(),g(t))},S=t=>{c||(d(),v(t))};h("Connecting WebSocket",{approvalId:e,wsUrl:o});const i=new r(o);i.onopen=()=>{y("WebSocket connected, waiting for resolution",{approvalId:e})},i.onmessage=t=>{try{const w=typeof t.data=="string"?JSON.parse(t.data):JSON.parse(String(t.data)),u=w?.status;if(h("WebSocket message received",{approvalId:e,status:u}),u==="executed"||u==="approved"||u==="rejected"||u==="expired"||u==="changes_requested"){y("Approval resolved via WebSocket",{approvalId:e,status:u}),l({status:u,reviewer_note:w?.reviewer_note});try{i.close(1e3)}catch{}}}catch{_("Failed to parse WebSocket message",{approvalId:e})}},i.onerror=t=>{const w=t instanceof Error?t.message:"WebSocket error";h("WebSocket error",{approvalId:e,error:w}),S(new Error(`WebSocket error: ${w}`))},i.onclose=t=>{c||(h("WebSocket closed without resolution",{approvalId:e,code:t.code}),S(new Error(`WebSocket closed unexpectedly (code ${t.code})`)))},typeof i.on=="function"&&i.on("unexpected-response",async(t,w)=>{try{const u=[];for await(const A of w)u.push(A);const T=Buffer.concat(u).toString(),x=JSON.parse(T),b=x?.status;if(b&&b!=="pending"){y("Approval already resolved (WS endpoint returned JSON)",{approvalId:e,status:b}),l({status:b,reviewer_note:x?.reviewer_note});return}}catch{}S(new Error("WebSocket upgrade rejected by server"))}),f=setInterval(async()=>{if(!c)try{const t=await W(n,s);if(t){y("Approval resolved via safety-net poll",{approvalId:e,status:t.status}),l(t);try{i.close(1e3)}catch{}}}catch{}},3e4);const k=a-Date.now();if(k<=0){S(new Error("Approval timeout"));return}m=setTimeout(()=>{if(!c){_("Approval timed out",{approvalId:e}),l({status:"expired"});try{i.close(1e3)}catch{}}},k)})}async function P(e,o,n,a,s){for(y("Polling for approval resolution",{approvalId:e,intervalMs:n});Date.now()<a;){await B(n);try{const r=await W(o,s);if(r)return y("Approval resolved via polling",{approvalId:e,status:r.status}),r}catch{}}return _("Approval polling timed out",{approvalId:e}),{status:"expired"}}async function W(e,o){const n={"User-Agent":"aerostack-gateway/0.15.29"};o&&(n.Authorization=`Bearer ${o}`);const a=await fetch(e,{headers:n});if(!a.ok)return null;const s=await a.json(),r=s.status;return r==="executed"||r==="approved"?{status:r,reviewer_note:s.reviewer_note}:r==="rejected"?{status:"rejected",reviewer_note:s.reviewer_note}:r==="changes_requested"?{status:"changes_requested",reviewer_note:s.reviewer_note}:r==="expired"?{status:"expired"}:null}async function O(){if(typeof globalThis.WebSocket<"u")return globalThis.WebSocket;try{return(await import("ws")).default}catch{throw new Error('WebSocket not available. Install the "ws" package: npm install ws')}}function B(e){return new Promise(o=>setTimeout(o,e))}export{N as resolveApproval,J as startBackgroundResolver};
|