@cryptiklemur/lattice 1.31.1 → 1.32.1

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, Loader2 } from "lucide-react";
 import { useWebSocket } from "../../hooks/useWebSocket";
 import { useMesh } from "../../hooks/useMesh";
 import { PairingDialog } from "../mesh/PairingDialog";
@@ -8,10 +8,12 @@ import type { NodeInfo } from "@lattice/shared";
 interface NodeRowProps {
   node: NodeInfo;
   onUnpair: (nodeId: string) => void;
+  onReconnect: (nodeId: string) => void;
 }
 
 function NodeRow(props: NodeRowProps) {
   var [confirming, setConfirming] = useState(false);
+  var [reconnecting, setReconnecting] = useState(false);
 
   function handleUnpair() {
     if (!confirming) {
@@ -56,6 +58,27 @@ function NodeRow(props: NodeRowProps) {
 
       {!props.node.isLocal && (
         <div className="flex gap-1.5 flex-shrink-0">
+          {!props.node.online && (
+            reconnecting ? (
+              <span className="flex items-center gap-1 text-[11px] text-base-content/40">
+                <Loader2 size={10} className="animate-spin" />
+                Connecting...
+              </span>
+            ) : (
+              <button
+                onClick={function () {
+                  setReconnecting(true);
+                  props.onReconnect(props.node.id);
+                  setTimeout(function () { setReconnecting(false); }, 5000);
+                }}
+                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 +119,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 +133,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 +166,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.1",
+  "version": "1.32.1",
   "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

@@ -4,8 +4,8 @@ import { sendTo, broadcast } from "../ws/broadcast";
 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, connectToPeer } from "../mesh/connector";
+import { addPeer, removePeer, loadPeers, getPeer } from "../mesh/peers";
+import { getConnectedPeerIds, connectToPeer, reconnectPeer } from "../mesh/connector";
 import type { PeerInfo } from "@lattice/shared";
 import { networkInterfaces } from "node:os";
 
@@ -172,6 +172,20 @@ 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; port?: number; addresses?: string[]; projects: Array<{ slug: string; title: string }> };
 
+    var knownPeer = hello.nodeId ? getPeer(hello.nodeId) : undefined;
+
+    if (knownPeer) {
+      var identity = loadOrCreateIdentity();
+      sendTo(clientId, {
+        type: "mesh:hello" as any,
+        nodeId: identity.id,
+        name: loadConfig().name,
+        projects: [],
+      });
+      broadcast({ type: "mesh:nodes", nodes: buildNodesMessage() });
+      return;
+    }
+
     if (!hello.token || !validatePairingToken(hello.token)) {
       sendTo(clientId, { type: "mesh:hello_rejected" as any, error: "Invalid or expired invite code" });
       return;
@@ -193,10 +207,10 @@ registerHandler("mesh", function (clientId: string, message: ClientMessage) {
       connectToPeer(peer.id, peerAddresses[0]);
     }
 
-    var identity = loadOrCreateIdentity();
+    var identity2 = loadOrCreateIdentity();
     sendTo(clientId, {
       type: "mesh:hello" as any,
-      nodeId: identity.id,
+      nodeId: identity2.id,
       name: loadConfig().name,
       projects: [],
     });
@@ -205,6 +219,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]);
+    }
   }
 }
 
@@ -223,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