feat: add Express + WebSocket server entry point

2026-04-20 18:25:10 -04:00
parent 55a49134ee
commit 421d591c1e
2 changed files with 80 additions and 0 deletions
@@ -0,0 +1,46 @@
+import { WebSocketServer, WebSocket } from "ws";
+import type { Server } from "http";
+
+type EventType = "docker" | "build" | "log" | "toolset" | "git";
+
+interface ForgeEvent {
+  type: EventType;
+  action: string;
+  data: unknown;
+  timestamp: number;
+}
+
+const clients = new Set<WebSocket>();
+
+let wss: WebSocketServer;
+
+export function initWebSocket(server: Server): WebSocketServer {
+  wss = new WebSocketServer({ server, path: "/ws" });
+
+  wss.on("connection", (ws) => {
+    clients.add(ws);
+    ws.on("close", () => clients.delete(ws));
+    ws.on("error", () => clients.delete(ws));
+  });
+
+  return wss;
+}
+
+export function broadcast(type: EventType, action: string, data: unknown): void {
+  const event: ForgeEvent = {
+    type,
+    action,
+    data,
+    timestamp: Date.now(),
+  };
+  const message = JSON.stringify(event);
+  for (const client of clients) {
+    if (client.readyState === WebSocket.OPEN) {
+      client.send(message);
+    }
+  }
+}
+
+export function getClientCount(): number {
+  return clients.size;
+}