@cryptiklemur/lattice 1.36.1 → 1.36.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cryptiklemur/lattice",
-  "version": "1.36.1",
+  "version": "1.36.2",
   "description": "Multi-machine agentic dashboard for Claude Code. Monitor sessions, manage MCP servers and skills, orchestrate across mesh-networked nodes.",
   "license": "MIT",
   "author": "Aaron Scherer <me@aaronscherer.me>",

package/server/src/mesh/proxy.ts

@@ -1,7 +1,7 @@
 import { randomUUID } from "node:crypto";
 import type { ClientMessage, MeshProxyRequestMessage, MeshProxyResponseMessage, ServerMessage } from "@lattice/shared";
 import { getPeerConnection } from "./connector";
-import { sendTo, broadcast } from "../ws/broadcast";
+import { sendTo, broadcast, registerVirtualClient, removeVirtualClient } from "../ws/broadcast";
 import { routeMessage } from "../ws/router";
 
 var pendingRequests = new Map<string, string>();
@@ -30,29 +30,26 @@ export function proxyToRemoteNode(nodeId: string, projectSlug: string, clientId:
 export function handleProxyRequest(sourceNodeId: string, msg: MeshProxyRequestMessage): void {
   var proxyClientId = "mesh-proxy:" + sourceNodeId + ":" + msg.requestId;
 
-  var originalBroadcast = broadcast;
-  void originalBroadcast;
-
-  var interceptedSendTo = function (targetId: string, response: object): void {
-    if (targetId === proxyClientId) {
-      var ws = getPeerConnection(sourceNodeId);
-      if (!ws) {
-        console.warn("[mesh/proxy] Cannot send response, no connection to: " + sourceNodeId);
-        return;
-      }
-
-      var envelope: MeshProxyResponseMessage = {
-        type: "mesh:proxy_response",
-        projectSlug: msg.projectSlug,
-        requestId: msg.requestId,
-        payload: response as ServerMessage,
-      };
-
-      ws.send(JSON.stringify(envelope));
+  registerVirtualClient(proxyClientId, function (response: object) {
+    var ws = getPeerConnection(sourceNodeId);
+    if (!ws) {
+      console.warn("[mesh/proxy] Cannot send response, no connection to: " + sourceNodeId);
+      removeVirtualClient(proxyClientId);
+      return;
     }
-  };
 
-  proxyRouteMessage(proxyClientId, msg.payload, interceptedSendTo);
+    var envelope: MeshProxyResponseMessage = {
+      type: "mesh:proxy_response",
+      projectSlug: msg.projectSlug,
+      requestId: msg.requestId,
+      payload: response as ServerMessage,
+    };
+
+    ws.send(JSON.stringify(envelope));
+    removeVirtualClient(proxyClientId);
+  });
+
+  routeMessage(proxyClientId, msg.payload);
 }
 
 export function handleProxyResponse(msg: MeshProxyResponseMessage): void {

package/server/src/ws/broadcast.ts

@@ -1,6 +1,15 @@
 import type { ServerWebSocket } from "bun";
 
 var clients = new Map<string, ServerWebSocket<{ id: string }>>();
+var virtualSendHandlers = new Map<string, (message: object) => void>();
+
+export function registerVirtualClient(id: string, handler: (message: object) => void): void {
+  virtualSendHandlers.set(id, handler);
+}
+
+export function removeVirtualClient(id: string): void {
+  virtualSendHandlers.delete(id);
+}
 
 export function addClient(ws: ServerWebSocket<{ id: string }>): void {
   clients.set(ws.data.id, ws);
@@ -23,6 +32,11 @@ export function sendTo(id: string, message: object): void {
   var ws = clients.get(id);
   if (ws) {
     ws.send(JSON.stringify(message));
+    return;
+  }
+  var virtualHandler = virtualSendHandlers.get(id);
+  if (virtualHandler) {
+    virtualHandler(message);
   }
 }