@cryptiklemur/lattice 1.31.0 → 1.32.0

package/client/src/components/settings/MeshStatus.tsx

@@ -1,5 +1,5 @@
 import { useState, useCallback, memo } from "react";
-import { Plus, CircleDot, Circle } from "lucide-react";
+import { Plus, CircleDot, Circle, RefreshCw } from "lucide-react";
 import { useWebSocket } from "../../hooks/useWebSocket";
 import { useMesh } from "../../hooks/useMesh";
 import { PairingDialog } from "../mesh/PairingDialog";
@@ -8,6 +8,7 @@ import type { NodeInfo } from "@lattice/shared";
 interface NodeRowProps {
   node: NodeInfo;
   onUnpair: (nodeId: string) => void;
+  onReconnect: (nodeId: string) => void;
 }
 
 function NodeRow(props: NodeRowProps) {
@@ -56,6 +57,16 @@ function NodeRow(props: NodeRowProps) {
 
       {!props.node.isLocal && (
         <div className="flex gap-1.5 flex-shrink-0">
+          {!props.node.online && (
+            <button
+              onClick={function () { props.onReconnect(props.node.id); }}
+              className="btn btn-ghost btn-xs border border-base-content/20 hover:btn-info hover:border-info gap-1"
+              title="Attempt to reconnect"
+            >
+              <RefreshCw size={10} />
+              Reconnect
+            </button>
+          )}
           {confirming ? (
             <>
               <button
@@ -96,6 +107,10 @@ export function MeshStatus() {
     ws.send({ type: "mesh:unpair", nodeId });
   }
 
+  function handleReconnect(nodeId: string) {
+    ws.send({ type: "mesh:reconnect", nodeId } as any);
+  }
+
   var localNode = nodes.find(function (n) { return n.isLocal; });
   var remoteNodes = nodes.filter(function (n) { return !n.isLocal; });
 
@@ -106,7 +121,7 @@ export function MeshStatus() {
           This Node
         </div>
         {localNode ? (
-          <NodeRow node={localNode} onUnpair={handleUnpair} />
+          <NodeRow node={localNode} onUnpair={handleUnpair} onReconnect={handleReconnect} />
         ) : (
           <div className="text-[12px] text-base-content/40 italic">
             Waiting for node info...
@@ -139,6 +154,7 @@ export function MeshStatus() {
                 key={node.id}
                 node={node}
                 onUnpair={handleUnpair}
+                onReconnect={handleReconnect}
               />
             );
           })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cryptiklemur/lattice",
-  "version": "1.31.0",
+  "version": "1.32.0",
   "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/handlers/mesh.ts

@@ -5,7 +5,7 @@ import { loadConfig } from "../config";
 import { loadOrCreateIdentity } from "../identity";
 import { generateInviteCode, parseInviteCode, validatePairingToken, consumePairingToken } from "../mesh/pairing";
 import { addPeer, removePeer, loadPeers } from "../mesh/peers";
-import { getConnectedPeerIds } from "../mesh/connector";
+import { getConnectedPeerIds, connectToPeer, reconnectPeer } from "../mesh/connector";
 import type { PeerInfo } from "@lattice/shared";
 import { networkInterfaces } from "node:os";
 
@@ -106,11 +106,14 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
 
     pairWs.addEventListener("open", function () {
       var identity = loadOrCreateIdentity();
+      var pairConfig = loadConfig();
       pairWs.send(JSON.stringify({
         type: "mesh:hello",
         nodeId: identity.id,
-        name: loadConfig().name,
+        name: pairConfig.name,
         token: parsed!.token,
+        port: pairConfig.port,
+        addresses: getAllAddresses().map(function (a) { return a.address + ":" + pairConfig.port; }),
         projects: [],
       }));
     });
@@ -128,21 +131,24 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
 
         if (data.type === "mesh:hello" && data.nodeId && data.name) {
           clearTimeout(pairTimeout);
+          var peerAddr = parsed!.address + ":" + parsed!.port;
           var peer: PeerInfo = {
             id: data.nodeId,
             name: data.name,
-            addresses: [parsed!.address],
+            addresses: [peerAddr],
             publicKey: "",
             pairedAt: Date.now(),
           };
           addPeer(peer);
           pairWs.close();
 
+          connectToPeer(peer.id, peerAddr);
+
           var nodeInfo: NodeInfo = {
             id: peer.id,
             name: peer.name,
-            address: parsed!.address,
-            addresses: [parsed!.address + ":" + parsed!.port],
+            address: peerAddr,
+            addresses: [peerAddr],
             port: parsed!.port,
             online: true,
             isLocal: false,
@@ -164,7 +170,7 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
   }
 
   if ((message as any).type === "mesh:hello") {
-    var hello = message as any as { type: "mesh:hello"; nodeId: string; name: string; token?: string; projects: Array<{ slug: string; title: string }> };
+    var hello = message as any as { type: "mesh:hello"; nodeId: string; name: string; token?: string; port?: number; addresses?: string[]; projects: Array<{ slug: string; title: string }> };
 
     if (!hello.token || !validatePairingToken(hello.token)) {
       sendTo(clientId, { type: "mesh:hello_rejected" as any, error: "Invalid or expired invite code" });
@@ -172,15 +178,21 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
     }
     consumePairingToken(hello.token);
 
+    var peerAddresses = hello.addresses ?? [];
+
     var peer: PeerInfo = {
       id: hello.nodeId,
       name: hello.name,
-      addresses: [],
+      addresses: peerAddresses,
       publicKey: "",
       pairedAt: Date.now(),
     };
     addPeer(peer);
 
+    if (peerAddresses.length > 0) {
+      connectToPeer(peer.id, peerAddresses[0]);
+    }
+
     var identity = loadOrCreateIdentity();
     sendTo(clientId, {
       type: "mesh:hello" as any,
@@ -193,6 +205,15 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
     return;
   }
 
+  if (message.type === "mesh:reconnect") {
+    var reconnectMsg = message as { type: "mesh:reconnect"; nodeId: string };
+    reconnectPeer(reconnectMsg.nodeId);
+    setTimeout(function () {
+      broadcast({ type: "mesh:nodes", nodes: buildNodesMessage() });
+    }, 2000);
+    return;
+  }
+
   if (message.type === "mesh:unpair") {
     var unpairMsg = message as MeshUnpairMessage;
     var removed = removePeer(unpairMsg.nodeId);

package/server/src/mesh/connector.ts

@@ -35,10 +35,25 @@ interface CircuitState {
 
 var circuitBreakers = new Map<string, CircuitState>();
 
+var RECONNECT_INTERVAL_MS = 15000;
+
 export function startMeshConnections(): void {
+  reconcilePeers();
+  setInterval(reconcilePeers, RECONNECT_INTERVAL_MS);
+}
+
+function reconcilePeers(): void {
   var peers = loadPeers();
   for (var i = 0; i < peers.length; i++) {
-    connectToPeer(peers[i].id, peers[i].addresses[0]);
+    var peer = peers[i];
+    if (!peer.addresses || peer.addresses.length === 0) continue;
+    var existing = connections.get(peer.id);
+    if (existing && !existing.dead && existing.ws.readyState === WebSocket.OPEN) continue;
+    if (existing && !existing.dead && existing.retryTimer !== null) continue;
+    if (!existing || existing.dead) {
+      connections.delete(peer.id);
+      connectToPeer(peer.id, peer.addresses[0]);
+    }
   }
 }
 
@@ -66,9 +81,10 @@ export function connectToPeer(nodeId: string, address: string): void {
   }
 
   var config = loadConfig();
-  var port = config.port;
   var protocol = config.tls ? "wss" : "ws";
-  var url = protocol + "://" + address + ":" + port + "/ws";
+  var url = address.includes(":")
+    ? protocol + "://" + address + "/ws"
+    : protocol + "://" + address + ":" + config.port + "/ws";
 
   var conn: PeerConnection = {
     nodeId: nodeId,
@@ -222,6 +238,26 @@ export function getPeerConnection(nodeId: string): WebSocket | undefined {
   return conn.ws;
 }
 
+export function reconnectPeer(nodeId: string): void {
+  var existing = connections.get(nodeId);
+  if (existing) {
+    existing.dead = true;
+    if (existing.retryTimer !== null) {
+      clearTimeout(existing.retryTimer);
+      existing.retryTimer = null;
+    }
+    existing.ws.close();
+    connections.delete(nodeId);
+  }
+  circuitBreakers.delete(nodeId);
+
+  var peers = loadPeers();
+  var peer = peers.find(function (p) { return p.id === nodeId; });
+  if (peer && peer.addresses && peer.addresses.length > 0) {
+    connectToPeer(nodeId, peer.addresses[0]);
+  }
+}
+
 export function getConnectedPeerIds(): string[] {
   var ids: string[] = [];
   for (var [nodeId, conn] of connections) {

package/shared/src/messages.ts

@@ -218,6 +218,11 @@ export interface MeshUnpairMessage {
   nodeId: string;
 }
 
+export interface MeshReconnectMessage {
+  type: "mesh:reconnect";
+  nodeId: string;
+}
+
 export interface LoopStartMessage {
   type: "loop:start";
   projectSlug: string;
@@ -537,6 +542,7 @@ export type ClientMessage =
   | MeshPairMessage
   | MeshGenerateInviteMessage
   | MeshUnpairMessage
+  | MeshReconnectMessage
   | LoopStartMessage
   | LoopStopMessage
   | LoopStatusRequestMessage