@algochad/archcoder 2.0.2

package/server/lib/utils/sse.js

@@ -0,0 +1,121 @@
+export function createSseHandler(options) {
+  const {
+    idleTimeoutMs = 5 * 60 * 1000,
+    heartbeatMs = 30 * 1000,
+    onConnect,
+    onDisconnect,
+    onEvent,
+    onError,
+    onEnd,
+  } = options || {};
+
+  const clients = new Set();
+  const clientStates = new WeakMap();
+
+  const getOrCreateClientState = (client) => {
+    let state = clientStates.get(client);
+    if (!state) {
+      state = {
+        lastActivityAt: Date.now(),
+        connectedAt: Date.now(),
+      };
+      clientStates.set(client, state);
+    }
+    return state;
+  };
+
+  const resetIdleTimeout = (client, idleTimer, abortController) => {
+    clearTimeout(idleTimer);
+    const newTimer = setTimeout(() => {
+      const reason = 'idle-timeout';
+      if (onEnd) onEnd(reason, client);
+      abortController.abort();
+    }, idleTimeoutMs);
+    return newTimer;
+  };
+
+  const recordActivity = (client) => {
+    const state = clientStates.get(client);
+    if (state) {
+      state.lastActivityAt = Date.now();
+    }
+  };
+
+  const registerClient = (client) => {
+    clients.add(client);
+    getOrCreateClientState(client);
+    if (onConnect) onConnect(client);
+  };
+
+  const unregisterClient = (client) => {
+    clients.delete(client);
+    clientStates.delete(client);
+    if (onDisconnect) onDisconnect(client);
+  };
+
+  const broadcastEvent = (event) => {
+    for (const client of clients) {
+      if (onEvent) onEvent(event, client);
+    }
+  };
+
+  const forEachClient = (callback) => {
+    for (const client of clients) {
+      callback(client);
+    }
+  };
+
+  const getClientState = (client) => {
+    return clientStates.get(client);
+  };
+
+  const setClientSessionId = (client, sessionId) => {
+    const state = clientStates.get(client);
+    if (state) {
+      state.boundSessionId = sessionId;
+    }
+  };
+
+  const getClientCount = () => {
+    return clients.size;
+  };
+
+  let globalHeartbeatInterval = null;
+
+  const startGlobalHeartbeat = () => {
+    if (globalHeartbeatInterval) {
+      return;
+    }
+    globalHeartbeatInterval = setInterval(() => {
+      for (const client of clients) {
+        if (onEvent) onEvent({ type: 'archcoder:heartbeat', timestamp: Date.now() }, client);
+      }
+    }, heartbeatMs);
+  };
+
+  const stopGlobalHeartbeat = () => {
+    if (globalHeartbeatInterval) {
+      clearInterval(globalHeartbeatInterval);
+      globalHeartbeatInterval = null;
+    }
+  };
+
+  return {
+    registerClient,
+    unregisterClient,
+    recordActivity,
+    broadcastEvent,
+    forEachClient,
+    getClientState,
+    setClientSessionId,
+    getClientCount,
+    startGlobalHeartbeat,
+    stopGlobalHeartbeat,
+    clients,
+    clientStates,
+    options: {
+      idleTimeoutMs,
+      heartbeatMs,
+    },
+  };
+}