@messenger-agent/messenger-agent 0.37.0-alpha.1 → 0.37.0

package/dist/tunnel-client.js

@@ -1 +1 @@
-import{createRequire as f}from"node:module";import p from"ws";import{logger as c}from"@messenger-agent/shared/logger";import{createIncomingRequestStream as m}from"@messenger-agent/shared/incoming-request-stream";import{reconnectDelayWithJitter as k}from"@messenger-agent/shared/reconnect-backoff";import{createAgentFileRoutes as w}from"@messenger-agent/shared/agent-files";import{createPrefixedRouteRequest as R,createWorkspaceRouteRequest as q,isAgentFileHttpTunnelRequest as g,isWorkspaceHttpTunnelRequest as y,sendHttpTunnelResponse as b}from"@messenger-agent/shared/tunnel-http";import{CODING_AGENT_TUNNEL_DUPLICATE_CLOSE_CODE as h,CODING_AGENT_TUNNEL_PROTOCOL_VERSION as T}from"@messenger-agent/shared/tunnel-protocol";import{createAgentWorkspaceRoutes as v}from"@messenger-agent/shared/workspace-routes";import{appConfig as E}from"./config.js";const M=f(import.meta.url)("../package.json").version,I=l=>new Promise(e=>setTimeout(e,l)),W=3;function S(l){try{const e=JSON.parse(l.toString());return e&&typeof e=="object"&&"type"in e?e:void 0}catch{return}}class P{config;stopped=!1;socket;running=new Map;streamedRequests=new Map;workspaceRoutes;fileRoutes=w();constructor(e=E){this.config=e,this.workspaceRoutes=v({config:e,textMaxBytes:e.textMaxBytes,uploadLimits:e.workspaceUploadLimits,allowManagement:!0,onWorkspacesChanged:()=>{process.send?.({type:"client.workspaces.changed"})}})}async start(){if(!this.config.tunnel.enabled)return;if(!this.config.tunnel.serverUrl||!this.config.tunnel.tunnelId||!this.config.tunnel.token)throw new Error("Workspace tunnel requires server_url, tunnel_id, and token when enabled");let e=this.config.tunnel.reconnectInitialMs;for(;!this.stopped;){try{e=await this.connectOnce()==="duplicate"?this.config.tunnel.reconnectMaxMs:this.config.tunnel.reconnectInitialMs}catch(r){this.stopped||c.warn("[WorkspaceTunnel] connection failed:",r)}this.stopped||(await I(k(e)),e=Math.min(e*2,this.config.tunnel.reconnectMaxMs))}}stop(){this.stopped=!0;for(const e of this.running.values())e.abort();this.running.clear();for(const e of this.streamedRequests.values())e.stream.error(new Error("Workspace tunnel stopped"));this.streamedRequests.clear(),this.socket?.close()}connectOnce(){return new Promise((e,r)=>{const s=new p(this.config.tunnel.serverUrl,{headers:{Authorization:`Bearer ${this.config.tunnel.token}`}});this.socket=s;let o,i=0,a=!1;const u=()=>{o&&clearInterval(o),this.socket===s&&(this.socket=void 0);for(const n of this.running.values())n.abort(new Error("Workspace tunnel closed"));this.running.clear();for(const n of this.streamedRequests.values())n.stream.error(new Error("Workspace tunnel closed"));this.streamedRequests.clear()};s.on("open",()=>{i=Date.now(),this.send({type:"hello",protocolVersion:T,tunnelId:this.config.tunnel.tunnelId,agentType:"workspace",capabilities:["workspace.files","workspace.files.upload.v1","workspace.git","workspace.tasks"],clientVersion:M,workspaces:[...this.config.workspaces.values()].map(({id:n,name:t,path:d})=>({id:n,name:t,path:d})).sort((n,t)=>n.id.localeCompare(t.id))}),o=setInterval(()=>{if(Date.now()-i>=this.config.tunnel.heartbeatIntervalMs*W){c.warn("[WorkspaceTunnel] websocket tunnel server is not responding; terminating stale connection"),s.terminate();return}this.send({type:"heartbeat"})},this.config.tunnel.heartbeatIntervalMs),c.info(`[WorkspaceTunnel] connected to ${this.config.tunnel.serverUrl} as ${this.config.tunnel.tunnelId}`)}),s.on("message",n=>{i=Date.now();const t=S(n);if(!(!t||t.type==="heartbeat"))if(t.type==="cancel")this.running.get(t.id)?.abort(),this.streamedRequests.get(t.id)?.stream.error(new Error("Workspace tunnel request cancelled")),this.streamedRequests.delete(t.id);else if(t.type==="request")this.handleRequest(t);else if(t.type==="request.start")this.startStreamedRequest(t);else if(t.type==="request.chunk"){const d=this.streamedRequests.get(t.id);d&&!d.stream.enqueue(Buffer.from(t.body.data,"base64"))&&(this.running.get(t.id)?.abort(new Error("Streamed request body exceeded its buffer limit")),this.streamedRequests.delete(t.id))}else t.type==="request.end"&&(this.streamedRequests.get(t.id)?.stream.close(),this.streamedRequests.delete(t.id))}),s.once("error",n=>{a||(a=!0,u(),r(n))}),s.once("close",(n,t)=>{u(),n===h?c.warn(`[WorkspaceTunnel] duplicate connection rejected; retrying with backoff: ${t.toString()}`):n===1008&&(this.stopped=!0,c.warn(`[WorkspaceTunnel] connection rejected with code ${n}; reconnect disabled until service restart: ${t.toString()}`)),a||(a=!0,e(n===h?"duplicate":"closed"))})})}send(e){this.socket?.readyState===p.OPEN&&this.socket.send(JSON.stringify(e))}startStreamedRequest(e){if(this.streamedRequests.has(e.id)||this.running.has(e.id))return;const r=m({onConsumed:s=>this.send({type:"request.ack",id:e.id,bytes:s}),onCancel:()=>this.running.get(e.id)?.abort()});this.streamedRequests.set(e.id,{stream:r,request:e}),this.handleRequest(e,r.body)}async handleRequest(e,r){const s=g(e);if(!s&&!y(e)){this.send({type:"response.error",id:e.id,code:"unsupported_request",errorText:`Unsupported workspace tunnel request: ${e.method} ${e.path}`}),this.streamedRequests.delete(e.id);return}const o=new AbortController;this.running.set(e.id,o);try{const i=s?R(e,"/v1/files",o.signal,r):q(e,o.signal,r),a=await(s?this.fileRoutes:this.workspaceRoutes).fetch(i);await b(e.id,a,u=>this.send(u))}catch(i){this.send({type:"response.error",id:e.id,code:"request_failed",errorText:i instanceof Error?i.message:"Workspace tunnel request failed"})}finally{this.running.delete(e.id),this.streamedRequests.delete(e.id)}}}export{P as WorkspaceTunnelClient};
+import{createRequire as f}from"node:module";import{hostname as m,platform as k,userInfo as w}from"node:os";import p from"ws";import{logger as u}from"@messenger-agent/shared/logger";import{createIncomingRequestStream as R}from"@messenger-agent/shared/incoming-request-stream";import{reconnectDelayWithJitter as g}from"@messenger-agent/shared/reconnect-backoff";import{createAgentFileRoutes as q}from"@messenger-agent/shared/agent-files";import{createPrefixedRouteRequest as y,createWorkspaceRouteRequest as b,isAgentFileHttpTunnelRequest as T,isWorkspaceHttpTunnelRequest as v,sendHttpTunnelResponse as E}from"@messenger-agent/shared/tunnel-http";import{CODING_AGENT_TUNNEL_DUPLICATE_CLOSE_CODE as h,CODING_AGENT_TUNNEL_PROTOCOL_VERSION as I}from"@messenger-agent/shared/tunnel-protocol";import{createAgentWorkspaceRoutes as M}from"@messenger-agent/shared/workspace-routes";import{appConfig as W}from"./config.js";const S=f(import.meta.url)("../package.json").version;function _(){const o=m().trim();if(!o)return{};if(k()!=="linux")return{hostname:o};try{const e=w().username.trim();return{hostname:o,...e?{username:e}:{}}}catch{return{hostname:o}}}const O=o=>new Promise(e=>setTimeout(e,o)),C=3;function x(o){try{const e=JSON.parse(o.toString());return e&&typeof e=="object"&&"type"in e?e:void 0}catch{return}}class V{config;stopped=!1;socket;running=new Map;streamedRequests=new Map;workspaceRoutes;fileRoutes=q();constructor(e=W){this.config=e,this.workspaceRoutes=M({config:e,textMaxBytes:e.textMaxBytes,uploadLimits:e.workspaceUploadLimits,allowManagement:!0,onWorkspacesChanged:()=>{process.send?.({type:"client.workspaces.changed"})}})}async start(){if(!this.config.tunnel.enabled)return;if(!this.config.tunnel.serverUrl||!this.config.tunnel.tunnelId||!this.config.tunnel.token)throw new Error("Workspace tunnel requires server_url, tunnel_id, and token when enabled");let e=this.config.tunnel.reconnectInitialMs;for(;!this.stopped;){try{e=await this.connectOnce()==="duplicate"?this.config.tunnel.reconnectMaxMs:this.config.tunnel.reconnectInitialMs}catch(s){this.stopped||u.warn("[WorkspaceTunnel] connection failed:",s)}this.stopped||(await O(g(e)),e=Math.min(e*2,this.config.tunnel.reconnectMaxMs))}}stop(){this.stopped=!0;for(const e of this.running.values())e.abort();this.running.clear();for(const e of this.streamedRequests.values())e.stream.error(new Error("Workspace tunnel stopped"));this.streamedRequests.clear(),this.socket?.close()}connectOnce(){return new Promise((e,s)=>{const r=new p(this.config.tunnel.serverUrl,{headers:{Authorization:`Bearer ${this.config.tunnel.token}`}});this.socket=r;let a,i=0,c=!1;const l=()=>{a&&clearInterval(a),this.socket===r&&(this.socket=void 0);for(const n of this.running.values())n.abort(new Error("Workspace tunnel closed"));this.running.clear();for(const n of this.streamedRequests.values())n.stream.error(new Error("Workspace tunnel closed"));this.streamedRequests.clear()};r.on("open",()=>{i=Date.now(),this.send({type:"hello",protocolVersion:I,tunnelId:this.config.tunnel.tunnelId,agentType:"workspace",capabilities:["workspace.files","workspace.files.upload.v1","workspace.git","workspace.tasks"],clientVersion:S,..._(),workspaces:[...this.config.workspaces.values()].map(({id:n,name:t,path:d})=>({id:n,name:t,path:d})).sort((n,t)=>n.id.localeCompare(t.id))}),a=setInterval(()=>{if(Date.now()-i>=this.config.tunnel.heartbeatIntervalMs*C){u.warn("[WorkspaceTunnel] websocket tunnel server is not responding; terminating stale connection"),r.terminate();return}this.send({type:"heartbeat"})},this.config.tunnel.heartbeatIntervalMs),u.info(`[WorkspaceTunnel] connected to ${this.config.tunnel.serverUrl} as ${this.config.tunnel.tunnelId}`)}),r.on("message",n=>{i=Date.now();const t=x(n);if(!(!t||t.type==="heartbeat"))if(t.type==="cancel")this.running.get(t.id)?.abort(),this.streamedRequests.get(t.id)?.stream.error(new Error("Workspace tunnel request cancelled")),this.streamedRequests.delete(t.id);else if(t.type==="request")this.handleRequest(t);else if(t.type==="request.start")this.startStreamedRequest(t);else if(t.type==="request.chunk"){const d=this.streamedRequests.get(t.id);d&&!d.stream.enqueue(Buffer.from(t.body.data,"base64"))&&(this.running.get(t.id)?.abort(new Error("Streamed request body exceeded its buffer limit")),this.streamedRequests.delete(t.id))}else t.type==="request.end"&&(this.streamedRequests.get(t.id)?.stream.close(),this.streamedRequests.delete(t.id))}),r.once("error",n=>{c||(c=!0,l(),s(n))}),r.once("close",(n,t)=>{l(),n===h?u.warn(`[WorkspaceTunnel] duplicate connection rejected; retrying with backoff: ${t.toString()}`):n===1008&&(this.stopped=!0,u.warn(`[WorkspaceTunnel] connection rejected with code ${n}; reconnect disabled until service restart: ${t.toString()}`)),c||(c=!0,e(n===h?"duplicate":"closed"))})})}send(e){this.socket?.readyState===p.OPEN&&this.socket.send(JSON.stringify(e))}startStreamedRequest(e){if(this.streamedRequests.has(e.id)||this.running.has(e.id))return;const s=R({onConsumed:r=>this.send({type:"request.ack",id:e.id,bytes:r}),onCancel:()=>this.running.get(e.id)?.abort()});this.streamedRequests.set(e.id,{stream:s,request:e}),this.handleRequest(e,s.body)}async handleRequest(e,s){const r=T(e);if(!r&&!v(e)){this.send({type:"response.error",id:e.id,code:"unsupported_request",errorText:`Unsupported workspace tunnel request: ${e.method} ${e.path}`}),this.streamedRequests.delete(e.id);return}const a=new AbortController;this.running.set(e.id,a);try{const i=r?y(e,"/v1/files",a.signal,s):b(e,a.signal,s),c=await(r?this.fileRoutes:this.workspaceRoutes).fetch(i);await E(e.id,c,l=>this.send(l))}catch(i){this.send({type:"response.error",id:e.id,code:"request_failed",errorText:i instanceof Error?i.message:"Workspace tunnel request failed"})}finally{this.running.delete(e.id),this.streamedRequests.delete(e.id)}}}export{V as WorkspaceTunnelClient};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@messenger-agent/messenger-agent",
-  "version": "0.37.0-alpha.1",
+  "version": "0.37.0",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -20,7 +20,7 @@
     "ws": "^8.18.3",
     "yaml": "^2.9.0",
     "zod": "^4.4.3",
-    "@messenger-agent/shared": "0.37.0-alpha.1"
+    "@messenger-agent/shared": "0.37.0"
   },
   "devDependencies": {
     "@types/ws": "^8.18.1"