whale-code 6.5.5 → 6.5.7

package/dist/server/local-agent-gateway-init.js

@@ -0,0 +1,143 @@
+// local-agent-gateway-init.ts — Gateway initialization and authentication
+import { WebSocketServer, WebSocket } from "ws";
+import { createHash } from "node:crypto";
+import { createClient } from "@supabase/supabase-js";
+import { agentPool, HEARTBEAT_INTERVAL_MS, PONG_TIMEOUT_MS, send } from "./local-agent-gateway-types.js";
+import { handleConnection } from "./local-agent-gateway-handler.js";
+let wss = null;
+let heartbeatTimer = null;
+export let supabaseClient = null;
+export function getWss() {
+  return wss;
+}
+export function initLocalAgentGateway(server) {
+  const SUPABASE_URL = process.env.SUPABASE_URL;
+  const SUPABASE_SERVICE_ROLE_KEY = process.env.SUPABASE_SERVICE_ROLE_KEY;
+  supabaseClient = createClient(SUPABASE_URL, SUPABASE_SERVICE_ROLE_KEY);
+  wss = new WebSocketServer({
+    noServer: true
+  });
+
+  // Handle HTTP upgrade on /agent/ws path
+  server.on("upgrade", (req, socket, head) => {
+    const url = new URL(req.url || "/", `http://${req.headers.host || "localhost"}`);
+    if (url.pathname !== "/agent/ws") {
+      socket.destroy();
+      return;
+    }
+    wss.handleUpgrade(req, socket, head, ws => {
+      wss.emit("connection", ws, req);
+    });
+  });
+  wss.on("connection", handleConnection);
+
+  // Heartbeat — ping all agents, disconnect stale ones
+  heartbeatTimer = setInterval(() => {
+    const now = Date.now();
+    for (const [storeId, agents] of agentPool) {
+      for (let i = agents.length - 1; i >= 0; i--) {
+        const agent = agents[i];
+        if (now - agent.lastPong > PONG_TIMEOUT_MS) {
+          console.log(`[local-agent] Disconnecting stale agent ${agent.agentId} (store=${storeId})`);
+          agent.ws.terminate();
+          agents.splice(i, 1);
+        } else if (agent.ws.readyState === WebSocket.OPEN) {
+          send(agent.ws, {
+            type: "ping"
+          });
+        }
+      }
+      if (agents.length === 0) agentPool.delete(storeId);
+    }
+  }, HEARTBEAT_INTERVAL_MS);
+  heartbeatTimer.unref();
+  console.log("[local-agent] Gateway initialized on /agent/ws");
+}
+
+// AUTHENTICATION
+// ============================================================================
+
+export async function authenticateAgent(msg) {
+  if (!supabaseClient || !msg.api_key) return null;
+  const keyHash = createHash("sha256").update(msg.api_key).digest("hex");
+
+  // 1. Try api_keys table first (standard API keys)
+  const {
+    data,
+    error
+  } = await supabaseClient.from("api_keys").select("id, store_id, owner_user_id").eq("key_hash", keyHash).eq("is_active", true).single();
+  if (data && !error) {
+    // Update last_used_at
+    supabaseClient.from("api_keys").update({
+      last_used_at: new Date().toISOString()
+    }).eq("id", data.id).then(() => {});
+    return {
+      storeId: data.store_id,
+      userId: data.owner_user_id || null,
+      nodeId: null
+    };
+  }
+
+  // 2. Fall back to nodes table (node daemon API keys)
+  const {
+    data: nodeData,
+    error: nodeErr
+  } = await supabaseClient.from("nodes").select("id, store_id").eq("api_key_hash", keyHash).single();
+  if (nodeData && !nodeErr) {
+    // Update node last_heartbeat on gateway connect
+    supabaseClient.from("nodes").update({
+      status: "online",
+      last_heartbeat: new Date().toISOString()
+    }).eq("id", nodeData.id).then(() => {});
+    return {
+      storeId: nodeData.store_id,
+      userId: null,
+      nodeId: nodeData.id
+    };
+  }
+  return null;
+}
+
+/**
+ * Authenticate a viewer connection using Supabase JWT.
+ */
+export async function authenticateViewer(msg) {
+  if (!supabaseClient || !msg.token) return null;
+  try {
+    // Verify JWT with Supabase auth
+    const {
+      data: {
+        user
+      },
+      error
+    } = await supabaseClient.auth.getUser(msg.token);
+    if (error || !user) return null;
+
+    // Verify user has access to the requested store
+    const {
+      data: storeAccess
+    } = await supabaseClient.from("store_users").select("store_id").eq("user_id", user.id).eq("store_id", msg.store_id).single();
+    if (!storeAccess) return null;
+    return {
+      storeId: msg.store_id,
+      userId: user.id
+    };
+  } catch {
+    return null;
+  }
+}
+
+// Shutdown helpers — called from local-agent-gateway-api.ts
+export function clearHeartbeat() {
+  if (heartbeatTimer) {
+    clearInterval(heartbeatTimer);
+    heartbeatTimer = null;
+  }
+}
+export function closeWss() {
+  if (wss) {
+    wss.close();
+    wss = null;
+  }
+}
+//# sourceMappingURL=local-agent-gateway-init.js.map

package/dist/server/local-agent-gateway-init.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-agent-gateway-init.js","names":["WebSocketServer","WebSocket","createHash","createClient","agentPool","HEARTBEAT_INTERVAL_MS","PONG_TIMEOUT_MS","send","handleConnection","wss","heartbeatTimer","supabaseClient","getWss","initLocalAgentGateway","server","SUPABASE_URL","process","env","SUPABASE_SERVICE_ROLE_KEY","noServer","on","req","socket","head","url","URL","headers","host","pathname","destroy","handleUpgrade","ws","emit","setInterval","now","Date","storeId","agents","i","length","agent","lastPong","console","log","agentId","terminate","splice","readyState","OPEN","type","delete","unref","authenticateAgent","msg","api_key","keyHash","update","digest","data","error","from","select","eq","single","last_used_at","toISOString","id","then","store_id","userId","owner_user_id","nodeId","nodeData","nodeErr","status","last_heartbeat","authenticateViewer","token","user","auth","getUser","storeAccess","clearHeartbeat","clearInterval","closeWss","close"],"sources":["../../src/server/local-agent-gateway-init.ts"],"sourcesContent":["// local-agent-gateway-init.ts — Gateway initialization and authentication\nimport { WebSocketServer, WebSocket } from \"ws\";\nimport { createHash } from \"node:crypto\";\nimport type http from \"node:http\";\nimport type { SupabaseClient } from \"@supabase/supabase-js\";\nimport { createClient } from \"@supabase/supabase-js\";\nimport {\n  type AgentAuthMessage, type ViewerAuthMessage,\n  agentPool, HEARTBEAT_INTERVAL_MS, PONG_TIMEOUT_MS,\n  send,\n} from \"./local-agent-gateway-types.js\";\nimport { handleConnection } from \"./local-agent-gateway-handler.js\";\n\nlet wss: WebSocketServer | null = null;\nlet heartbeatTimer: ReturnType<typeof setInterval> | null = null;\nexport let supabaseClient: SupabaseClient | null = null;\n\nexport function getWss(): WebSocketServer | null { return wss; }\n\nexport function initLocalAgentGateway(server: http.Server): void {\n  const SUPABASE_URL = process.env.SUPABASE_URL!;\n  const SUPABASE_SERVICE_ROLE_KEY = process.env.SUPABASE_SERVICE_ROLE_KEY!;\n  supabaseClient = createClient(SUPABASE_URL, SUPABASE_SERVICE_ROLE_KEY);\n\n  wss = new WebSocketServer({ noServer: true });\n\n  // Handle HTTP upgrade on /agent/ws path\n  server.on(\"upgrade\", (req: http.IncomingMessage, socket, head) => {\n    const url = new URL(req.url || \"/\", `http://${req.headers.host || \"localhost\"}`);\n    if (url.pathname !== \"/agent/ws\") {\n      socket.destroy();\n      return;\n    }\n    wss!.handleUpgrade(req, socket as any, head, (ws) => {\n      wss!.emit(\"connection\", ws, req);\n    });\n  });\n\n  wss.on(\"connection\", handleConnection);\n\n  // Heartbeat — ping all agents, disconnect stale ones\n  heartbeatTimer = setInterval(() => {\n    const now = Date.now();\n    for (const [storeId, agents] of agentPool) {\n      for (let i = agents.length - 1; i >= 0; i--) {\n        const agent = agents[i];\n        if (now - agent.lastPong > PONG_TIMEOUT_MS) {\n          console.log(`[local-agent] Disconnecting stale agent ${agent.agentId} (store=${storeId})`);\n          agent.ws.terminate();\n          agents.splice(i, 1);\n        } else if (agent.ws.readyState === WebSocket.OPEN) {\n          send(agent.ws, { type: \"ping\" });\n        }\n      }\n      if (agents.length === 0) agentPool.delete(storeId);\n    }\n  }, HEARTBEAT_INTERVAL_MS);\n  heartbeatTimer.unref();\n\n  console.log(\"[local-agent] Gateway initialized on /agent/ws\");\n}\n\n// AUTHENTICATION\n// ============================================================================\n\nexport async function authenticateAgent(msg: AgentAuthMessage): Promise<{ storeId: string; userId: string | null; nodeId: string | null } | null> {\n  if (!supabaseClient || !msg.api_key) return null;\n\n  const keyHash = createHash(\"sha256\").update(msg.api_key).digest(\"hex\");\n\n  // 1. Try api_keys table first (standard API keys)\n  const { data, error } = await supabaseClient\n    .from(\"api_keys\")\n    .select(\"id, store_id, owner_user_id\")\n    .eq(\"key_hash\", keyHash)\n    .eq(\"is_active\", true)\n    .single();\n\n  if (data && !error) {\n    // Update last_used_at\n    supabaseClient.from(\"api_keys\")\n      .update({ last_used_at: new Date().toISOString() })\n      .eq(\"id\", data.id)\n      .then(() => {});\n    return { storeId: data.store_id, userId: data.owner_user_id || null, nodeId: null };\n  }\n\n  // 2. Fall back to nodes table (node daemon API keys)\n  const { data: nodeData, error: nodeErr } = await supabaseClient\n    .from(\"nodes\")\n    .select(\"id, store_id\")\n    .eq(\"api_key_hash\", keyHash)\n    .single();\n\n  if (nodeData && !nodeErr) {\n    // Update node last_heartbeat on gateway connect\n    supabaseClient.from(\"nodes\")\n      .update({ status: \"online\", last_heartbeat: new Date().toISOString() })\n      .eq(\"id\", nodeData.id)\n      .then(() => {});\n    return { storeId: nodeData.store_id, userId: null, nodeId: nodeData.id };\n  }\n\n  return null;\n}\n\n/**\n * Authenticate a viewer connection using Supabase JWT.\n */\nexport async function authenticateViewer(msg: ViewerAuthMessage): Promise<{ storeId: string; userId: string } | null> {\n  if (!supabaseClient || !msg.token) return null;\n\n  try {\n    // Verify JWT with Supabase auth\n    const { data: { user }, error } = await supabaseClient.auth.getUser(msg.token);\n    if (error || !user) return null;\n\n    // Verify user has access to the requested store\n    const { data: storeAccess } = await supabaseClient\n      .from(\"store_users\")\n      .select(\"store_id\")\n      .eq(\"user_id\", user.id)\n      .eq(\"store_id\", msg.store_id)\n      .single();\n\n    if (!storeAccess) return null;\n\n    return { storeId: msg.store_id, userId: user.id };\n  } catch {\n    return null;\n  }\n}\n\n// Shutdown helpers — called from local-agent-gateway-api.ts\nexport function clearHeartbeat(): void {\n  if (heartbeatTimer) { clearInterval(heartbeatTimer); heartbeatTimer = null; }\n}\nexport function closeWss(): void {\n  if (wss) { wss.close(); wss = null; }\n}\n"],"mappings":"AAAA;AACA,SAASA,eAAe,EAAEC,SAAS,QAAQ,IAAI;AAC/C,SAASC,UAAU,QAAQ,aAAa;AAGxC,SAASC,YAAY,QAAQ,uBAAuB;AACpD,SAEEC,SAAS,EAAEC,qBAAqB,EAAEC,eAAe,EACjDC,IAAI,QACC,gCAAgC;AACvC,SAASC,gBAAgB,QAAQ,kCAAkC;AAEnE,IAAIC,GAA2B,GAAG,IAAI;AACtC,IAAIC,cAAqD,GAAG,IAAI;AAChE,OAAO,IAAIC,cAAqC,GAAG,IAAI;AAEvD,OAAO,SAASC,MAAMA,CAAA,EAA2B;EAAE,OAAOH,GAAG;AAAE;AAE/D,OAAO,SAASI,qBAAqBA,CAACC,MAAmB,EAAQ;EAC/D,MAAMC,YAAY,GAAGC,OAAO,CAACC,GAAG,CAACF,YAAa;EAC9C,MAAMG,yBAAyB,GAAGF,OAAO,CAACC,GAAG,CAACC,yBAA0B;EACxEP,cAAc,GAAGR,YAAY,CAACY,YAAY,EAAEG,yBAAyB,CAAC;EAEtET,GAAG,GAAG,IAAIT,eAAe,CAAC;IAAEmB,QAAQ,EAAE;EAAK,CAAC,CAAC;;EAE7C;EACAL,MAAM,CAACM,EAAE,CAAC,SAAS,EAAE,CAACC,GAAyB,EAAEC,MAAM,EAAEC,IAAI,KAAK;IAChE,MAAMC,GAAG,GAAG,IAAIC,GAAG,CAACJ,GAAG,CAACG,GAAG,IAAI,GAAG,EAAE,UAAUH,GAAG,CAACK,OAAO,CAACC,IAAI,IAAI,WAAW,EAAE,CAAC;IAChF,IAAIH,GAAG,CAACI,QAAQ,KAAK,WAAW,EAAE;MAChCN,MAAM,CAACO,OAAO,CAAC,CAAC;MAChB;IACF;IACApB,GAAG,CAAEqB,aAAa,CAACT,GAAG,EAAEC,MAAM,EAASC,IAAI,EAAGQ,EAAE,IAAK;MACnDtB,GAAG,CAAEuB,IAAI,CAAC,YAAY,EAAED,EAAE,EAAEV,GAAG,CAAC;IAClC,CAAC,CAAC;EACJ,CAAC,CAAC;EAEFZ,GAAG,CAACW,EAAE,CAAC,YAAY,EAAEZ,gBAAgB,CAAC;;EAEtC;EACAE,cAAc,GAAGuB,WAAW,CAAC,MAAM;IACjC,MAAMC,GAAG,GAAGC,IAAI,CAACD,GAAG,CAAC,CAAC;IACtB,KAAK,MAAM,CAACE,OAAO,EAAEC,MAAM,CAAC,IAAIjC,SAAS,EAAE;MACzC,KAAK,IAAIkC,CAAC,GAAGD,MAAM,CAACE,MAAM,GAAG,CAAC,EAAED,CAAC,IAAI,CAAC,EAAEA,CAAC,EAAE,EAAE;QAC3C,MAAME,KAAK,GAAGH,MAAM,CAACC,CAAC,CAAC;QACvB,IAAIJ,GAAG,GAAGM,KAAK,CAACC,QAAQ,GAAGnC,eAAe,EAAE;UAC1CoC,OAAO,CAACC,GAAG,CAAC,2CAA2CH,KAAK,CAACI,OAAO,WAAWR,OAAO,GAAG,CAAC;UAC1FI,KAAK,CAACT,EAAE,CAACc,SAAS,CAAC,CAAC;UACpBR,MAAM,CAACS,MAAM,CAACR,CAAC,EAAE,CAAC,CAAC;QACrB,CAAC,MAAM,IAAIE,KAAK,CAACT,EAAE,CAACgB,UAAU,KAAK9C,SAAS,CAAC+C,IAAI,EAAE;UACjDzC,IAAI,CAACiC,KAAK,CAACT,EAAE,EAAE;YAAEkB,IAAI,EAAE;UAAO,CAAC,CAAC;QAClC;MACF;MACA,IAAIZ,MAAM,CAACE,MAAM,KAAK,CAAC,EAAEnC,SAAS,CAAC8C,MAAM,CAACd,OAAO,CAAC;IACpD;EACF,CAAC,EAAE/B,qBAAqB,CAAC;EACzBK,cAAc,CAACyC,KAAK,CAAC,CAAC;EAEtBT,OAAO,CAACC,GAAG,CAAC,gDAAgD,CAAC;AAC/D;;AAEA;AACA;;AAEA,OAAO,eAAeS,iBAAiBA,CAACC,GAAqB,EAAqF;EAChJ,IAAI,CAAC1C,cAAc,IAAI,CAAC0C,GAAG,CAACC,OAAO,EAAE,OAAO,IAAI;EAEhD,MAAMC,OAAO,GAAGrD,UAAU,CAAC,QAAQ,CAAC,CAACsD,MAAM,CAACH,GAAG,CAACC,OAAO,CAAC,CAACG,MAAM,CAAC,KAAK,CAAC;;EAEtE;EACA,MAAM;IAAEC,IAAI;IAAEC;EAAM,CAAC,GAAG,MAAMhD,cAAc,CACzCiD,IAAI,CAAC,UAAU,CAAC,CAChBC,MAAM,CAAC,6BAA6B,CAAC,CACrCC,EAAE,CAAC,UAAU,EAAEP,OAAO,CAAC,CACvBO,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,CACrBC,MAAM,CAAC,CAAC;EAEX,IAAIL,IAAI,IAAI,CAACC,KAAK,EAAE;IAClB;IACAhD,cAAc,CAACiD,IAAI,CAAC,UAAU,CAAC,CAC5BJ,MAAM,CAAC;MAAEQ,YAAY,EAAE,IAAI7B,IAAI,CAAC,CAAC,CAAC8B,WAAW,CAAC;IAAE,CAAC,CAAC,CAClDH,EAAE,CAAC,IAAI,EAAEJ,IAAI,CAACQ,EAAE,CAAC,CACjBC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IACjB,OAAO;MAAE/B,OAAO,EAAEsB,IAAI,CAACU,QAAQ;MAAEC,MAAM,EAAEX,IAAI,CAACY,aAAa,IAAI,IAAI;MAAEC,MAAM,EAAE;IAAK,CAAC;EACrF;;EAEA;EACA,MAAM;IAAEb,IAAI,EAAEc,QAAQ;IAAEb,KAAK,EAAEc;EAAQ,CAAC,GAAG,MAAM9D,cAAc,CAC5DiD,IAAI,CAAC,OAAO,CAAC,CACbC,MAAM,CAAC,cAAc,CAAC,CACtBC,EAAE,CAAC,cAAc,EAAEP,OAAO,CAAC,CAC3BQ,MAAM,CAAC,CAAC;EAEX,IAAIS,QAAQ,IAAI,CAACC,OAAO,EAAE;IACxB;IACA9D,cAAc,CAACiD,IAAI,CAAC,OAAO,CAAC,CACzBJ,MAAM,CAAC;MAAEkB,MAAM,EAAE,QAAQ;MAAEC,cAAc,EAAE,IAAIxC,IAAI,CAAC,CAAC,CAAC8B,WAAW,CAAC;IAAE,CAAC,CAAC,CACtEH,EAAE,CAAC,IAAI,EAAEU,QAAQ,CAACN,EAAE,CAAC,CACrBC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IACjB,OAAO;MAAE/B,OAAO,EAAEoC,QAAQ,CAACJ,QAAQ;MAAEC,MAAM,EAAE,IAAI;MAAEE,MAAM,EAAEC,QAAQ,CAACN;IAAG,CAAC;EAC1E;EAEA,OAAO,IAAI;AACb;;AAEA;AACA;AACA;AACA,OAAO,eAAeU,kBAAkBA,CAACvB,GAAsB,EAAuD;EACpH,IAAI,CAAC1C,cAAc,IAAI,CAAC0C,GAAG,CAACwB,KAAK,EAAE,OAAO,IAAI;EAE9C,IAAI;IACF;IACA,MAAM;MAAEnB,IAAI,EAAE;QAAEoB;MAAK,CAAC;MAAEnB;IAAM,CAAC,GAAG,MAAMhD,cAAc,CAACoE,IAAI,CAACC,OAAO,CAAC3B,GAAG,CAACwB,KAAK,CAAC;IAC9E,IAAIlB,KAAK,IAAI,CAACmB,IAAI,EAAE,OAAO,IAAI;;IAE/B;IACA,MAAM;MAAEpB,IAAI,EAAEuB;IAAY,CAAC,GAAG,MAAMtE,cAAc,CAC/CiD,IAAI,CAAC,aAAa,CAAC,CACnBC,MAAM,CAAC,UAAU,CAAC,CAClBC,EAAE,CAAC,SAAS,EAAEgB,IAAI,CAACZ,EAAE,CAAC,CACtBJ,EAAE,CAAC,UAAU,EAAET,GAAG,CAACe,QAAQ,CAAC,CAC5BL,MAAM,CAAC,CAAC;IAEX,IAAI,CAACkB,WAAW,EAAE,OAAO,IAAI;IAE7B,OAAO;MAAE7C,OAAO,EAAEiB,GAAG,CAACe,QAAQ;MAAEC,MAAM,EAAES,IAAI,CAACZ;IAAG,CAAC;EACnD,CAAC,CAAC,MAAM;IACN,OAAO,IAAI;EACb;AACF;;AAEA;AACA,OAAO,SAASgB,cAAcA,CAAA,EAAS;EACrC,IAAIxE,cAAc,EAAE;IAAEyE,aAAa,CAACzE,cAAc,CAAC;IAAEA,cAAc,GAAG,IAAI;EAAE;AAC9E;AACA,OAAO,SAAS0E,QAAQA,CAAA,EAAS;EAC/B,IAAI3E,GAAG,EAAE;IAAEA,GAAG,CAAC4E,KAAK,CAAC,CAAC;IAAE5E,GAAG,GAAG,IAAI;EAAE;AACtC","ignoreList":[]}

package/dist/server/local-agent-gateway-messages.d.ts

@@ -0,0 +1,12 @@
+import { WebSocket } from "ws";
+import { type LocalAgent } from "./local-agent-gateway-types.js";
+/** Handle portal_request from an agent wanting to connect to a target node. */
+export declare function handlePortalRequest(ws: WebSocket, msg: any, agent: LocalAgent): void;
+/** Handle portal_accept from target agent. */
+export declare function handlePortalAccept(msg: any, agent: LocalAgent): void;
+/** Handle portal_reject from target agent. */
+export declare function handlePortalReject(msg: any, agent: LocalAgent): void;
+/** Handle portal_close from either side. */
+export declare function handlePortalClose(msg: any, agent: LocalAgent): void;
+/** Handle tool result from agent. */
+export declare function handleResult(msg: any): void;

package/dist/server/local-agent-gateway-messages.js

@@ -0,0 +1,108 @@
+// local-agent-gateway-messages.ts — Message routing handlers for portal and result messages
+
+import { portalSessions, pendingRequests, pickAgent, pickAgentById, send, truncate } from "./local-agent-gateway-types.js";
+
+/** Handle portal_request from an agent wanting to connect to a target node. */
+export function handlePortalRequest(ws, msg, agent) {
+  const targetAgent = pickAgent(agent.storeId, undefined, msg.target_node_id);
+  if (!targetAgent) {
+    send(ws, {
+      type: "portal_reject",
+      session_id: msg.session_id,
+      reason: "Target node not online"
+    });
+    return;
+  }
+  // Forward request to target agent
+  send(targetAgent.ws, {
+    type: "portal_request",
+    session_id: msg.session_id,
+    capabilities: msg.capabilities,
+    requester_node_id: agent.nodeId,
+    requester_hostname: agent.hostname,
+    requester_role: msg.requester_role || "admin"
+  });
+  // Pre-register session so binary routing works once accepted
+  portalSessions.set(msg.session_id, {
+    sessionId: msg.session_id,
+    initiatorAgentId: agent.agentId,
+    targetAgentId: targetAgent.agentId,
+    storeId: agent.storeId,
+    capabilities: msg.capabilities || [],
+    startedAt: new Date(),
+    bytesRelayed: 0
+  });
+  console.log(`[portal] Session ${msg.session_id}: ${agent.hostname} -> ${targetAgent.hostname}`);
+}
+
+/** Handle portal_accept from target agent. */
+export function handlePortalAccept(msg, agent) {
+  const session = portalSessions.get(msg.session_id);
+  if (session) {
+    const initiator = pickAgentById(session.initiatorAgentId);
+    if (initiator) {
+      send(initiator.ws, {
+        type: "portal_accept",
+        session_id: msg.session_id,
+        hostname: agent.hostname
+      });
+    }
+  }
+}
+
+/** Handle portal_reject from target agent. */
+export function handlePortalReject(msg, agent) {
+  const session = portalSessions.get(msg.session_id);
+  if (session) {
+    const initiator = pickAgentById(session.initiatorAgentId);
+    if (initiator) {
+      send(initiator.ws, {
+        type: "portal_reject",
+        session_id: msg.session_id,
+        reason: msg.reason
+      });
+    }
+    portalSessions.delete(msg.session_id);
+  }
+}
+
+/** Handle portal_close from either side. */
+export function handlePortalClose(msg, agent) {
+  const session = portalSessions.get(msg.session_id);
+  if (session) {
+    const otherAgentId = agent.agentId === session.initiatorAgentId ? session.targetAgentId : session.initiatorAgentId;
+    const other = pickAgentById(otherAgentId);
+    if (other) {
+      send(other.ws, {
+        type: "portal_close",
+        session_id: msg.session_id
+      });
+    }
+    portalSessions.delete(msg.session_id);
+    console.log(`[portal] Session ${msg.session_id} closed (${session.bytesRelayed} bytes relayed)`);
+  }
+}
+
+/** Handle tool result from agent. */
+export function handleResult(msg) {
+  const pending = pendingRequests.get(msg.request_id);
+  if (!pending) return; // Already timed out or cancelled
+
+  clearTimeout(pending.timer);
+  pendingRequests.delete(msg.request_id);
+
+  // Truncate large outputs
+  const result = {
+    success: msg.success,
+    stdout: truncate(msg.stdout),
+    stderr: truncate(msg.stderr),
+    exit_code: msg.exit_code,
+    data: msg.data,
+    error: msg.error,
+    killed: msg.killed,
+    cwd: msg.cwd,
+    duration_ms: msg.duration_ms
+  };
+  pending.resolve(result);
+}
+//# sourceMappingURL=local-agent-gateway-messages.js.map

package/dist/server/local-agent-gateway-messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-agent-gateway-messages.js","names":["portalSessions","pendingRequests","pickAgent","pickAgentById","send","truncate","handlePortalRequest","ws","msg","agent","targetAgent","storeId","undefined","target_node_id","type","session_id","reason","capabilities","requester_node_id","nodeId","requester_hostname","hostname","requester_role","set","sessionId","initiatorAgentId","agentId","targetAgentId","startedAt","Date","bytesRelayed","console","log","handlePortalAccept","session","get","initiator","handlePortalReject","delete","handlePortalClose","otherAgentId","other","handleResult","pending","request_id","clearTimeout","timer","result","success","stdout","stderr","exit_code","data","error","killed","cwd","duration_ms","resolve"],"sources":["../../src/server/local-agent-gateway-messages.ts"],"sourcesContent":["// local-agent-gateway-messages.ts — Message routing handlers for portal and result messages\nimport { WebSocket } from \"ws\";\nimport {\n  type LocalAgent, type AgentResult,\n  portalSessions, pendingRequests,\n  pickAgent, pickAgentById, send, truncate,\n} from \"./local-agent-gateway-types.js\";\n\n/** Handle portal_request from an agent wanting to connect to a target node. */\nexport function handlePortalRequest(ws: WebSocket, msg: any, agent: LocalAgent): void {\n  const targetAgent = pickAgent(agent.storeId, undefined, msg.target_node_id);\n  if (!targetAgent) {\n    send(ws, { type: \"portal_reject\", session_id: msg.session_id, reason: \"Target node not online\" });\n    return;\n  }\n  // Forward request to target agent\n  send(targetAgent.ws, {\n    type: \"portal_request\",\n    session_id: msg.session_id,\n    capabilities: msg.capabilities,\n    requester_node_id: agent.nodeId,\n    requester_hostname: agent.hostname,\n    requester_role: msg.requester_role || \"admin\",\n  });\n  // Pre-register session so binary routing works once accepted\n  portalSessions.set(msg.session_id, {\n    sessionId: msg.session_id,\n    initiatorAgentId: agent.agentId,\n    targetAgentId: targetAgent.agentId,\n    storeId: agent.storeId,\n    capabilities: msg.capabilities || [],\n    startedAt: new Date(),\n    bytesRelayed: 0,\n  });\n  console.log(`[portal] Session ${msg.session_id}: ${agent.hostname} -> ${targetAgent.hostname}`);\n}\n\n/** Handle portal_accept from target agent. */\nexport function handlePortalAccept(msg: any, agent: LocalAgent): void {\n  const session = portalSessions.get(msg.session_id);\n  if (session) {\n    const initiator = pickAgentById(session.initiatorAgentId);\n    if (initiator) {\n      send(initiator.ws, { type: \"portal_accept\", session_id: msg.session_id, hostname: agent.hostname });\n    }\n  }\n}\n\n/** Handle portal_reject from target agent. */\nexport function handlePortalReject(msg: any, agent: LocalAgent): void {\n  const session = portalSessions.get(msg.session_id);\n  if (session) {\n    const initiator = pickAgentById(session.initiatorAgentId);\n    if (initiator) {\n      send(initiator.ws, { type: \"portal_reject\", session_id: msg.session_id, reason: msg.reason });\n    }\n    portalSessions.delete(msg.session_id);\n  }\n}\n\n/** Handle portal_close from either side. */\nexport function handlePortalClose(msg: any, agent: LocalAgent): void {\n  const session = portalSessions.get(msg.session_id);\n  if (session) {\n    const otherAgentId = agent.agentId === session.initiatorAgentId\n      ? session.targetAgentId : session.initiatorAgentId;\n    const other = pickAgentById(otherAgentId);\n    if (other) {\n      send(other.ws, { type: \"portal_close\", session_id: msg.session_id });\n    }\n    portalSessions.delete(msg.session_id);\n    console.log(`[portal] Session ${msg.session_id} closed (${session.bytesRelayed} bytes relayed)`);\n  }\n}\n\n/** Handle tool result from agent. */\nexport function handleResult(msg: any): void {\n  const pending = pendingRequests.get(msg.request_id);\n  if (!pending) return; // Already timed out or cancelled\n\n  clearTimeout(pending.timer);\n  pendingRequests.delete(msg.request_id);\n\n  // Truncate large outputs\n  const result: AgentResult = {\n    success: msg.success,\n    stdout: truncate(msg.stdout),\n    stderr: truncate(msg.stderr),\n    exit_code: msg.exit_code,\n    data: msg.data,\n    error: msg.error,\n    killed: msg.killed,\n    cwd: msg.cwd,\n    duration_ms: msg.duration_ms,\n  };\n\n  pending.resolve(result);\n}\n"],"mappings":"AAAA;;AAEA,SAEEA,cAAc,EAAEC,eAAe,EAC/BC,SAAS,EAAEC,aAAa,EAAEC,IAAI,EAAEC,QAAQ,QACnC,gCAAgC;;AAEvC;AACA,OAAO,SAASC,mBAAmBA,CAACC,EAAa,EAAEC,GAAQ,EAAEC,KAAiB,EAAQ;EACpF,MAAMC,WAAW,GAAGR,SAAS,CAACO,KAAK,CAACE,OAAO,EAAEC,SAAS,EAAEJ,GAAG,CAACK,cAAc,CAAC;EAC3E,IAAI,CAACH,WAAW,EAAE;IAChBN,IAAI,CAACG,EAAE,EAAE;MAAEO,IAAI,EAAE,eAAe;MAAEC,UAAU,EAAEP,GAAG,CAACO,UAAU;MAAEC,MAAM,EAAE;IAAyB,CAAC,CAAC;IACjG;EACF;EACA;EACAZ,IAAI,CAACM,WAAW,CAACH,EAAE,EAAE;IACnBO,IAAI,EAAE,gBAAgB;IACtBC,UAAU,EAAEP,GAAG,CAACO,UAAU;IAC1BE,YAAY,EAAET,GAAG,CAACS,YAAY;IAC9BC,iBAAiB,EAAET,KAAK,CAACU,MAAM;IAC/BC,kBAAkB,EAAEX,KAAK,CAACY,QAAQ;IAClCC,cAAc,EAAEd,GAAG,CAACc,cAAc,IAAI;EACxC,CAAC,CAAC;EACF;EACAtB,cAAc,CAACuB,GAAG,CAACf,GAAG,CAACO,UAAU,EAAE;IACjCS,SAAS,EAAEhB,GAAG,CAACO,UAAU;IACzBU,gBAAgB,EAAEhB,KAAK,CAACiB,OAAO;IAC/BC,aAAa,EAAEjB,WAAW,CAACgB,OAAO;IAClCf,OAAO,EAAEF,KAAK,CAACE,OAAO;IACtBM,YAAY,EAAET,GAAG,CAACS,YAAY,IAAI,EAAE;IACpCW,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC;IACrBC,YAAY,EAAE;EAChB,CAAC,CAAC;EACFC,OAAO,CAACC,GAAG,CAAC,oBAAoBxB,GAAG,CAACO,UAAU,KAAKN,KAAK,CAACY,QAAQ,OAAOX,WAAW,CAACW,QAAQ,EAAE,CAAC;AACjG;;AAEA;AACA,OAAO,SAASY,kBAAkBA,CAACzB,GAAQ,EAAEC,KAAiB,EAAQ;EACpE,MAAMyB,OAAO,GAAGlC,cAAc,CAACmC,GAAG,CAAC3B,GAAG,CAACO,UAAU,CAAC;EAClD,IAAImB,OAAO,EAAE;IACX,MAAME,SAAS,GAAGjC,aAAa,CAAC+B,OAAO,CAACT,gBAAgB,CAAC;IACzD,IAAIW,SAAS,EAAE;MACbhC,IAAI,CAACgC,SAAS,CAAC7B,EAAE,EAAE;QAAEO,IAAI,EAAE,eAAe;QAAEC,UAAU,EAAEP,GAAG,CAACO,UAAU;QAAEM,QAAQ,EAAEZ,KAAK,CAACY;MAAS,CAAC,CAAC;IACrG;EACF;AACF;;AAEA;AACA,OAAO,SAASgB,kBAAkBA,CAAC7B,GAAQ,EAAEC,KAAiB,EAAQ;EACpE,MAAMyB,OAAO,GAAGlC,cAAc,CAACmC,GAAG,CAAC3B,GAAG,CAACO,UAAU,CAAC;EAClD,IAAImB,OAAO,EAAE;IACX,MAAME,SAAS,GAAGjC,aAAa,CAAC+B,OAAO,CAACT,gBAAgB,CAAC;IACzD,IAAIW,SAAS,EAAE;MACbhC,IAAI,CAACgC,SAAS,CAAC7B,EAAE,EAAE;QAAEO,IAAI,EAAE,eAAe;QAAEC,UAAU,EAAEP,GAAG,CAACO,UAAU;QAAEC,MAAM,EAAER,GAAG,CAACQ;MAAO,CAAC,CAAC;IAC/F;IACAhB,cAAc,CAACsC,MAAM,CAAC9B,GAAG,CAACO,UAAU,CAAC;EACvC;AACF;;AAEA;AACA,OAAO,SAASwB,iBAAiBA,CAAC/B,GAAQ,EAAEC,KAAiB,EAAQ;EACnE,MAAMyB,OAAO,GAAGlC,cAAc,CAACmC,GAAG,CAAC3B,GAAG,CAACO,UAAU,CAAC;EAClD,IAAImB,OAAO,EAAE;IACX,MAAMM,YAAY,GAAG/B,KAAK,CAACiB,OAAO,KAAKQ,OAAO,CAACT,gBAAgB,GAC3DS,OAAO,CAACP,aAAa,GAAGO,OAAO,CAACT,gBAAgB;IACpD,MAAMgB,KAAK,GAAGtC,aAAa,CAACqC,YAAY,CAAC;IACzC,IAAIC,KAAK,EAAE;MACTrC,IAAI,CAACqC,KAAK,CAAClC,EAAE,EAAE;QAAEO,IAAI,EAAE,cAAc;QAAEC,UAAU,EAAEP,GAAG,CAACO;MAAW,CAAC,CAAC;IACtE;IACAf,cAAc,CAACsC,MAAM,CAAC9B,GAAG,CAACO,UAAU,CAAC;IACrCgB,OAAO,CAACC,GAAG,CAAC,oBAAoBxB,GAAG,CAACO,UAAU,YAAYmB,OAAO,CAACJ,YAAY,iBAAiB,CAAC;EAClG;AACF;;AAEA;AACA,OAAO,SAASY,YAAYA,CAAClC,GAAQ,EAAQ;EAC3C,MAAMmC,OAAO,GAAG1C,eAAe,CAACkC,GAAG,CAAC3B,GAAG,CAACoC,UAAU,CAAC;EACnD,IAAI,CAACD,OAAO,EAAE,OAAO,CAAC;;EAEtBE,YAAY,CAACF,OAAO,CAACG,KAAK,CAAC;EAC3B7C,eAAe,CAACqC,MAAM,CAAC9B,GAAG,CAACoC,UAAU,CAAC;;EAEtC;EACA,MAAMG,MAAmB,GAAG;IAC1BC,OAAO,EAAExC,GAAG,CAACwC,OAAO;IACpBC,MAAM,EAAE5C,QAAQ,CAACG,GAAG,CAACyC,MAAM,CAAC;IAC5BC,MAAM,EAAE7C,QAAQ,CAACG,GAAG,CAAC0C,MAAM,CAAC;IAC5BC,SAAS,EAAE3C,GAAG,CAAC2C,SAAS;IACxBC,IAAI,EAAE5C,GAAG,CAAC4C,IAAI;IACdC,KAAK,EAAE7C,GAAG,CAAC6C,KAAK;IAChBC,MAAM,EAAE9C,GAAG,CAAC8C,MAAM;IAClBC,GAAG,EAAE/C,GAAG,CAAC+C,GAAG;IACZC,WAAW,EAAEhD,GAAG,CAACgD;EACnB,CAAC;EAEDb,OAAO,CAACc,OAAO,CAACV,MAAM,CAAC;AACzB","ignoreList":[]}

package/dist/server/local-agent-gateway-stats.d.ts

@@ -0,0 +1,53 @@
+import { type AgentResult } from "./local-agent-gateway-types.js";
+/**
+ * Get gateway statistics.
+ */
+export declare function getGatewayStats(): {
+    total_agents: number;
+    total_stores: number;
+    pending_requests: number;
+    agents_by_store: Record<string, number>;
+};
+/**
+ * Get active remote desktop sessions.
+ */
+export declare function getRemoteDesktopSessions(): Array<{
+    session_id: string;
+    store_id: string;
+    user_id: string;
+    agent_id: string;
+    started_at: string;
+    frames_relayed: number;
+    bytes_relayed: number;
+}>;
+/**
+ * Get active portal sessions.
+ */
+export declare function getPortalSessions(): Array<{
+    session_id: string;
+    store_id: string;
+    initiator_agent_id: string;
+    target_agent_id: string;
+    capabilities: string[];
+    started_at: string;
+    bytes_relayed: number;
+}>;
+export interface ClusterCommandArgs {
+    action: string;
+    task?: string;
+    max_cells?: number;
+    working_directory?: string;
+    cluster_id?: string;
+    channel_id?: string;
+    conversation_id?: string;
+}
+/**
+ * Execute a cluster command on the connected whale-node.
+ * Sends via WebSocket and waits for the result.
+ */
+export declare function executeClusterCommand(storeId: string, args: ClusterCommandArgs, options?: {
+    timeout?: number;
+    agent_id?: string;
+    node_id?: string;
+}): Promise<AgentResult>;
+export declare function shutdownGateway(): void;

package/dist/server/local-agent-gateway-stats.js

@@ -0,0 +1,129 @@
+// local-agent-gateway-stats.ts — Stats, sessions, cluster commands, and shutdown
+
+import { WebSocket } from "ws";
+import { randomUUID } from "node:crypto";
+import { agentPool, pendingRequests, rdSessions, portalSessions, pickAgent, send } from "./local-agent-gateway-types.js";
+import { clearHeartbeat, closeWss } from "./local-agent-gateway-init.js";
+
+/**
+ * Get gateway statistics.
+ */
+export function getGatewayStats() {
+  const agentsByStore = {};
+  let totalAgents = 0;
+  for (const [storeId, agents] of agentPool) {
+    const active = agents.filter(a => a.ws.readyState === WebSocket.OPEN).length;
+    if (active > 0) {
+      agentsByStore[storeId] = active;
+      totalAgents += active;
+    }
+  }
+  return {
+    total_agents: totalAgents,
+    total_stores: Object.keys(agentsByStore).length,
+    pending_requests: pendingRequests.size,
+    agents_by_store: agentsByStore
+  };
+}
+
+/**
+ * Get active remote desktop sessions.
+ */
+export function getRemoteDesktopSessions() {
+  return Array.from(rdSessions.values()).map(s => ({
+    session_id: s.sessionId,
+    store_id: s.storeId,
+    user_id: s.userId,
+    agent_id: s.agentId,
+    started_at: s.startedAt.toISOString(),
+    frames_relayed: s.framesRelayed,
+    bytes_relayed: s.bytesRelayed
+  }));
+}
+
+/**
+ * Get active portal sessions.
+ */
+export function getPortalSessions() {
+  return Array.from(portalSessions.values()).map(s => ({
+    session_id: s.sessionId,
+    store_id: s.storeId,
+    initiator_agent_id: s.initiatorAgentId,
+    target_agent_id: s.targetAgentId,
+    capabilities: s.capabilities,
+    started_at: s.startedAt.toISOString(),
+    bytes_relayed: s.bytesRelayed
+  }));
+}
+
+// ============================================================================
+// CLUSTER COMMANDS — routes cluster operations to whale-node
+// ============================================================================
+
+/**
+ * Execute a cluster command on the connected whale-node.
+ * Sends via WebSocket and waits for the result.
+ */
+export async function executeClusterCommand(storeId, args, options = {}) {
+  const agent = pickAgent(storeId, options.agent_id, options.node_id);
+  if (!agent) {
+    return {
+      success: false,
+      error: "No local agent connected for cluster commands."
+    };
+  }
+  const requestId = randomUUID();
+  const timeout = Math.min(options.timeout || 60000, 600000);
+  const sent = send(agent.ws, {
+    type: "cluster_command",
+    request_id: requestId,
+    args
+  });
+  if (!sent) return {
+    success: false,
+    error: "Agent connection closed"
+  };
+  return new Promise((resolve, reject) => {
+    const timer = setTimeout(() => {
+      pendingRequests.delete(requestId);
+      resolve({
+        success: false,
+        error: `Cluster command timed out after ${timeout}ms`
+      });
+    }, timeout + 5000);
+    pendingRequests.set(requestId, {
+      resolve,
+      reject,
+      timer,
+      agentId: agent.agentId
+    });
+  });
+}
+export function shutdownGateway() {
+  clearHeartbeat();
+
+  // Close all agent connections
+  for (const [, agents] of agentPool) {
+    for (const agent of agents) {
+      send(agent.ws, {
+        type: "shutdown",
+        message: "Server shutting down"
+      });
+      agent.ws.close(1001, "Server shutdown");
+    }
+  }
+  agentPool.clear();
+
+  // Reject all pending requests
+  for (const [, pending] of pendingRequests) {
+    clearTimeout(pending.timer);
+    pending.resolve({
+      success: false,
+      error: "Gateway shutting down"
+    });
+  }
+  pendingRequests.clear();
+  closeWss();
+  console.log("[local-agent] Gateway shut down");
+}
+//# sourceMappingURL=local-agent-gateway-stats.js.map

package/dist/server/local-agent-gateway-stats.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local-agent-gateway-stats.js","names":["WebSocket","randomUUID","agentPool","pendingRequests","rdSessions","portalSessions","pickAgent","send","clearHeartbeat","closeWss","getGatewayStats","agentsByStore","totalAgents","storeId","agents","active","filter","a","ws","readyState","OPEN","length","total_agents","total_stores","Object","keys","pending_requests","size","agents_by_store","getRemoteDesktopSessions","Array","from","values","map","s","session_id","sessionId","store_id","user_id","userId","agent_id","agentId","started_at","startedAt","toISOString","frames_relayed","framesRelayed","bytes_relayed","bytesRelayed","getPortalSessions","initiator_agent_id","initiatorAgentId","target_agent_id","targetAgentId","capabilities","executeClusterCommand","args","options","agent","node_id","success","error","requestId","timeout","Math","min","sent","type","request_id","Promise","resolve","reject","timer","setTimeout","delete","set","shutdownGateway","message","close","clear","pending","clearTimeout","console","log"],"sources":["../../src/server/local-agent-gateway-stats.ts"],"sourcesContent":["// local-agent-gateway-stats.ts — Stats, sessions, cluster commands, and shutdown\n\nimport { WebSocket } from \"ws\";\nimport { randomUUID } from \"node:crypto\";\nimport {\n  type AgentResult,\n  agentPool, pendingRequests, rdSessions, portalSessions,\n  pickAgent, send,\n} from \"./local-agent-gateway-types.js\";\nimport { clearHeartbeat, closeWss } from \"./local-agent-gateway-init.js\";\n\n/**\n * Get gateway statistics.\n */\nexport function getGatewayStats(): {\n  total_agents: number;\n  total_stores: number;\n  pending_requests: number;\n  agents_by_store: Record<string, number>;\n} {\n  const agentsByStore: Record<string, number> = {};\n  let totalAgents = 0;\n  for (const [storeId, agents] of agentPool) {\n    const active = agents.filter(a => a.ws.readyState === WebSocket.OPEN).length;\n    if (active > 0) {\n      agentsByStore[storeId] = active;\n      totalAgents += active;\n    }\n  }\n  return {\n    total_agents: totalAgents,\n    total_stores: Object.keys(agentsByStore).length,\n    pending_requests: pendingRequests.size,\n    agents_by_store: agentsByStore,\n  };\n}\n\n/**\n * Get active remote desktop sessions.\n */\nexport function getRemoteDesktopSessions(): Array<{\n  session_id: string;\n  store_id: string;\n  user_id: string;\n  agent_id: string;\n  started_at: string;\n  frames_relayed: number;\n  bytes_relayed: number;\n}> {\n  return Array.from(rdSessions.values()).map(s => ({\n    session_id: s.sessionId,\n    store_id: s.storeId,\n    user_id: s.userId,\n    agent_id: s.agentId,\n    started_at: s.startedAt.toISOString(),\n    frames_relayed: s.framesRelayed,\n    bytes_relayed: s.bytesRelayed,\n  }));\n}\n\n/**\n * Get active portal sessions.\n */\nexport function getPortalSessions(): Array<{\n  session_id: string;\n  store_id: string;\n  initiator_agent_id: string;\n  target_agent_id: string;\n  capabilities: string[];\n  started_at: string;\n  bytes_relayed: number;\n}> {\n  return Array.from(portalSessions.values()).map(s => ({\n    session_id: s.sessionId,\n    store_id: s.storeId,\n    initiator_agent_id: s.initiatorAgentId,\n    target_agent_id: s.targetAgentId,\n    capabilities: s.capabilities,\n    started_at: s.startedAt.toISOString(),\n    bytes_relayed: s.bytesRelayed,\n  }));\n}\n\n// ============================================================================\n// CLUSTER COMMANDS — routes cluster operations to whale-node\n// ============================================================================\n\nexport interface ClusterCommandArgs {\n  action: string;\n  task?: string;\n  max_cells?: number;\n  working_directory?: string;\n  cluster_id?: string;\n  channel_id?: string;\n  conversation_id?: string;\n}\n\n/**\n * Execute a cluster command on the connected whale-node.\n * Sends via WebSocket and waits for the result.\n */\nexport async function executeClusterCommand(\n  storeId: string,\n  args: ClusterCommandArgs,\n  options: { timeout?: number; agent_id?: string; node_id?: string } = {},\n): Promise<AgentResult> {\n  const agent = pickAgent(storeId, options.agent_id, options.node_id);\n  if (!agent) {\n    return {\n      success: false,\n      error: \"No local agent connected for cluster commands.\",\n    };\n  }\n\n  const requestId = randomUUID();\n  const timeout = Math.min(options.timeout || 60000, 600000);\n\n  const sent = send(agent.ws, {\n    type: \"cluster_command\",\n    request_id: requestId,\n    args,\n  });\n  if (!sent) return { success: false, error: \"Agent connection closed\" };\n\n  return new Promise<AgentResult>((resolve, reject) => {\n    const timer = setTimeout(() => {\n      pendingRequests.delete(requestId);\n      resolve({\n        success: false,\n        error: `Cluster command timed out after ${timeout}ms`,\n      });\n    }, timeout + 5000);\n\n    pendingRequests.set(requestId, { resolve, reject, timer, agentId: agent.agentId });\n  });\n}\n\nexport function shutdownGateway(): void {\n  clearHeartbeat();\n\n  // Close all agent connections\n  for (const [, agents] of agentPool) {\n    for (const agent of agents) {\n      send(agent.ws, { type: \"shutdown\", message: \"Server shutting down\" });\n      agent.ws.close(1001, \"Server shutdown\");\n    }\n  }\n  agentPool.clear();\n\n  // Reject all pending requests\n  for (const [, pending] of pendingRequests) {\n    clearTimeout(pending.timer);\n    pending.resolve({ success: false, error: \"Gateway shutting down\" });\n  }\n  pendingRequests.clear();\n\n  closeWss();\n\n  console.log(\"[local-agent] Gateway shut down\");\n}\n"],"mappings":"AAAA;;AAEA,SAASA,SAAS,QAAQ,IAAI;AAC9B,SAASC,UAAU,QAAQ,aAAa;AACxC,SAEEC,SAAS,EAAEC,eAAe,EAAEC,UAAU,EAAEC,cAAc,EACtDC,SAAS,EAAEC,IAAI,QACV,gCAAgC;AACvC,SAASC,cAAc,EAAEC,QAAQ,QAAQ,+BAA+B;;AAExE;AACA;AACA;AACA,OAAO,SAASC,eAAeA,CAAA,EAK7B;EACA,MAAMC,aAAqC,GAAG,CAAC,CAAC;EAChD,IAAIC,WAAW,GAAG,CAAC;EACnB,KAAK,MAAM,CAACC,OAAO,EAAEC,MAAM,CAAC,IAAIZ,SAAS,EAAE;IACzC,MAAMa,MAAM,GAAGD,MAAM,CAACE,MAAM,CAACC,CAAC,IAAIA,CAAC,CAACC,EAAE,CAACC,UAAU,KAAKnB,SAAS,CAACoB,IAAI,CAAC,CAACC,MAAM;IAC5E,IAAIN,MAAM,GAAG,CAAC,EAAE;MACdJ,aAAa,CAACE,OAAO,CAAC,GAAGE,MAAM;MAC/BH,WAAW,IAAIG,MAAM;IACvB;EACF;EACA,OAAO;IACLO,YAAY,EAAEV,WAAW;IACzBW,YAAY,EAAEC,MAAM,CAACC,IAAI,CAACd,aAAa,CAAC,CAACU,MAAM;IAC/CK,gBAAgB,EAAEvB,eAAe,CAACwB,IAAI;IACtCC,eAAe,EAAEjB;EACnB,CAAC;AACH;;AAEA;AACA;AACA;AACA,OAAO,SAASkB,wBAAwBA,CAAA,EAQrC;EACD,OAAOC,KAAK,CAACC,IAAI,CAAC3B,UAAU,CAAC4B,MAAM,CAAC,CAAC,CAAC,CAACC,GAAG,CAACC,CAAC,KAAK;IAC/CC,UAAU,EAAED,CAAC,CAACE,SAAS;IACvBC,QAAQ,EAAEH,CAAC,CAACrB,OAAO;IACnByB,OAAO,EAAEJ,CAAC,CAACK,MAAM;IACjBC,QAAQ,EAAEN,CAAC,CAACO,OAAO;IACnBC,UAAU,EAAER,CAAC,CAACS,SAAS,CAACC,WAAW,CAAC,CAAC;IACrCC,cAAc,EAAEX,CAAC,CAACY,aAAa;IAC/BC,aAAa,EAAEb,CAAC,CAACc;EACnB,CAAC,CAAC,CAAC;AACL;;AAEA;AACA;AACA;AACA,OAAO,SAASC,iBAAiBA,CAAA,EAQ9B;EACD,OAAOnB,KAAK,CAACC,IAAI,CAAC1B,cAAc,CAAC2B,MAAM,CAAC,CAAC,CAAC,CAACC,GAAG,CAACC,CAAC,KAAK;IACnDC,UAAU,EAAED,CAAC,CAACE,SAAS;IACvBC,QAAQ,EAAEH,CAAC,CAACrB,OAAO;IACnBqC,kBAAkB,EAAEhB,CAAC,CAACiB,gBAAgB;IACtCC,eAAe,EAAElB,CAAC,CAACmB,aAAa;IAChCC,YAAY,EAAEpB,CAAC,CAACoB,YAAY;IAC5BZ,UAAU,EAAER,CAAC,CAACS,SAAS,CAACC,WAAW,CAAC,CAAC;IACrCG,aAAa,EAAEb,CAAC,CAACc;EACnB,CAAC,CAAC,CAAC;AACL;;AAEA;AACA;AACA;;AAYA;AACA;AACA;AACA;AACA,OAAO,eAAeO,qBAAqBA,CACzC1C,OAAe,EACf2C,IAAwB,EACxBC,OAAkE,GAAG,CAAC,CAAC,EACjD;EACtB,MAAMC,KAAK,GAAGpD,SAAS,CAACO,OAAO,EAAE4C,OAAO,CAACjB,QAAQ,EAAEiB,OAAO,CAACE,OAAO,CAAC;EACnE,IAAI,CAACD,KAAK,EAAE;IACV,OAAO;MACLE,OAAO,EAAE,KAAK;MACdC,KAAK,EAAE;IACT,CAAC;EACH;EAEA,MAAMC,SAAS,GAAG7D,UAAU,CAAC,CAAC;EAC9B,MAAM8D,OAAO,GAAGC,IAAI,CAACC,GAAG,CAACR,OAAO,CAACM,OAAO,IAAI,KAAK,EAAE,MAAM,CAAC;EAE1D,MAAMG,IAAI,GAAG3D,IAAI,CAACmD,KAAK,CAACxC,EAAE,EAAE;IAC1BiD,IAAI,EAAE,iBAAiB;IACvBC,UAAU,EAAEN,SAAS;IACrBN;EACF,CAAC,CAAC;EACF,IAAI,CAACU,IAAI,EAAE,OAAO;IAAEN,OAAO,EAAE,KAAK;IAAEC,KAAK,EAAE;EAA0B,CAAC;EAEtE,OAAO,IAAIQ,OAAO,CAAc,CAACC,OAAO,EAAEC,MAAM,KAAK;IACnD,MAAMC,KAAK,GAAGC,UAAU,CAAC,MAAM;MAC7BtE,eAAe,CAACuE,MAAM,CAACZ,SAAS,CAAC;MACjCQ,OAAO,CAAC;QACNV,OAAO,EAAE,KAAK;QACdC,KAAK,EAAE,mCAAmCE,OAAO;MACnD,CAAC,CAAC;IACJ,CAAC,EAAEA,OAAO,GAAG,IAAI,CAAC;IAElB5D,eAAe,CAACwE,GAAG,CAACb,SAAS,EAAE;MAAEQ,OAAO;MAAEC,MAAM;MAAEC,KAAK;MAAE/B,OAAO,EAAEiB,KAAK,CAACjB;IAAQ,CAAC,CAAC;EACpF,CAAC,CAAC;AACJ;AAEA,OAAO,SAASmC,eAAeA,CAAA,EAAS;EACtCpE,cAAc,CAAC,CAAC;;EAEhB;EACA,KAAK,MAAM,GAAGM,MAAM,CAAC,IAAIZ,SAAS,EAAE;IAClC,KAAK,MAAMwD,KAAK,IAAI5C,MAAM,EAAE;MAC1BP,IAAI,CAACmD,KAAK,CAACxC,EAAE,EAAE;QAAEiD,IAAI,EAAE,UAAU;QAAEU,OAAO,EAAE;MAAuB,CAAC,CAAC;MACrEnB,KAAK,CAACxC,EAAE,CAAC4D,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC;IACzC;EACF;EACA5E,SAAS,CAAC6E,KAAK,CAAC,CAAC;;EAEjB;EACA,KAAK,MAAM,GAAGC,OAAO,CAAC,IAAI7E,eAAe,EAAE;IACzC8E,YAAY,CAACD,OAAO,CAACR,KAAK,CAAC;IAC3BQ,OAAO,CAACV,OAAO,CAAC;MAAEV,OAAO,EAAE,KAAK;MAAEC,KAAK,EAAE;IAAwB,CAAC,CAAC;EACrE;EACA1D,eAAe,CAAC4E,KAAK,CAAC,CAAC;EAEvBtE,QAAQ,CAAC,CAAC;EAEVyE,OAAO,CAACC,GAAG,CAAC,iCAAiC,CAAC;AAChD","ignoreList":[]}

package/dist/server/local-agent-gateway-types.d.ts

@@ -0,0 +1,94 @@
+import { WebSocket } from "ws";
+export interface LocalAgent {
+    ws: WebSocket;
+    storeId: string;
+    userId: string | null;
+    nodeId: string | null;
+    capabilities: string[];
+    connectedAt: Date;
+    lastPong: number;
+    agentId: string;
+    platform: string;
+    hostname: string;
+}
+interface AgentRequest {
+    resolve: (result: AgentResult) => void;
+    reject: (err: Error) => void;
+    timer: ReturnType<typeof setTimeout>;
+    agentId: string;
+}
+export interface AgentResult {
+    success: boolean;
+    stdout?: string;
+    stderr?: string;
+    exit_code?: number;
+    data?: unknown;
+    error?: string;
+    killed?: boolean;
+    cwd?: string;
+    duration_ms?: number;
+    tools?: string[];
+}
+export interface AgentAuthMessage {
+    type: "auth";
+    api_key: string;
+    capabilities?: string[];
+    platform?: string;
+    hostname?: string;
+    version?: string;
+}
+export interface RemoteDesktopSession {
+    sessionId: string;
+    viewerWs: WebSocket;
+    agentId: string;
+    storeId: string;
+    userId: string;
+    startedAt: Date;
+    framesRelayed: number;
+    bytesRelayed: number;
+}
+export interface ViewerAuthMessage {
+    type: "auth";
+    role: "viewer";
+    token: string;
+    store_id: string;
+    target_node_id?: string;
+}
+interface PortalRelaySession {
+    sessionId: string;
+    initiatorAgentId: string;
+    targetAgentId: string;
+    storeId: string;
+    capabilities: string[];
+    startedAt: Date;
+    bytesRelayed: number;
+}
+/** Map<storeId, LocalAgent[]> — multiple agents per store */
+export declare const agentPool: Map<string, LocalAgent[]>;
+/** Map<requestId, pending request> */
+export declare const pendingRequests: Map<string, AgentRequest>;
+/** Map<sessionId, RemoteDesktopSession> — active remote desktop relay sessions */
+export declare const rdSessions: Map<string, RemoteDesktopSession>;
+/** Map<sessionId, PortalRelaySession> — active portal sessions */
+export declare const portalSessions: Map<string, PortalRelaySession>;
+export declare const PORTAL_MARKER = 1;
+export declare const MAX_AGENTS_PER_STORE = 25;
+export declare const HEARTBEAT_INTERVAL_MS = 30000;
+export declare const PONG_TIMEOUT_MS = 45000;
+export declare const MAX_OUTPUT_CHARS: number;
+export declare function pickAgent(storeId: string, agentId?: string, nodeId?: string): LocalAgent | null;
+export declare function pickAgentById(agentId: string): LocalAgent | null;
+export declare function isJsonString(data: any): boolean;
+export declare function send(ws: WebSocket, data: unknown): boolean;
+export declare function failPendingRequestsForAgent(agentId: string, reason: string): void;
+export declare function truncate(text?: string): string | undefined;
+export interface ClusterCommandArgs {
+    action: string;
+    task?: string;
+    max_cells?: number;
+    working_directory?: string;
+    cluster_id?: string;
+    channel_id?: string;
+    conversation_id?: string;
+}
+export {};