@katyella/legio 0.1.0

package/src/server/public/lib/ws.js

@@ -0,0 +1,79 @@
+// Legio Web UI — WebSocket Client
+// Auto-reconnecting WebSocket that pushes snapshots into Preact signals.
+
+import { appState, setConnected, setLastUpdated } from "./state.js";
+
+let ws = null;
+let reconnectTimer = null;
+
+function handleSnapshot(d) {
+	if (d.agents !== undefined) appState.agents.value = d.agents;
+	// WS sends mail as { unreadCount, recent }; REST API sends flat MailMessage[].
+	// Normalize to a flat array in both cases.
+	if (d.mail !== undefined) {
+		appState.mail.value = Array.isArray(d.mail) ? d.mail : d.mail.recent || [];
+	}
+	if (d.mergeQueue !== undefined) appState.mergeQueue.value = d.mergeQueue;
+	if (d.metrics !== undefined && Array.isArray(d.metrics)) appState.metrics.value = d.metrics;
+	if (d.snapshots !== undefined) appState.snapshots.value = d.snapshots;
+	if (d.runs !== undefined) appState.runs.value = d.runs;
+	if (d.events !== undefined) appState.events.value = d.events;
+	if (d.errors !== undefined) appState.errors.value = d.errors;
+	if (d.config !== undefined) appState.config.value = d.config;
+	if (d.status !== undefined) appState.status.value = d.status;
+	setLastUpdated();
+}
+
+export function connectWS() {
+	if (reconnectTimer) {
+		clearTimeout(reconnectTimer);
+		reconnectTimer = null;
+	}
+
+	ws = new WebSocket(`ws://${location.host}/ws`);
+
+	ws.addEventListener("open", () => {
+		setConnected(true);
+	});
+
+	ws.addEventListener("message", (event) => {
+		let msg;
+		try {
+			msg = JSON.parse(event.data);
+		} catch (_e) {
+			console.warn("[legio] unparseable WS message:", event.data);
+			return;
+		}
+		if (msg.type === "snapshot") {
+			handleSnapshot(msg.data || {});
+		} else if (msg.type === "mail_new") {
+			// Prepend new message to mail array if not already present
+			const current = appState.mail.value;
+			const newMsg = msg.data;
+			if (newMsg && !current.some((m) => m.id === newMsg.id)) {
+				appState.mail.value = [newMsg, ...current];
+			}
+			setLastUpdated();
+		} else if (msg.type === "coordinator_output") {
+			const text = msg.data?.text ?? "";
+			if (text) {
+				window.dispatchEvent(new CustomEvent("coordinator-output", { detail: { text } }));
+			}
+		}
+	});
+
+	ws.addEventListener("close", () => {
+		setConnected(false);
+		reconnectTimer = setTimeout(connectWS, 3000);
+	});
+
+	ws.addEventListener("error", (event) => {
+		console.error("[legio] WebSocket error:", event);
+	});
+}
+
+export function requestRefresh() {
+	if (ws && ws.readyState === WebSocket.OPEN) {
+		ws.send(JSON.stringify({ type: "refresh" }));
+	}
+}