Refactor 4 with bugfixes

server/src/websocket/index.ts

@@ -0,0 +1,49 @@
+import {
+  IncomingMessage,
+  Server,
+  ServerResponse
+} from 'http';
+import { WebSocketServer, WebSocket } from 'ws';
+import { v4 as uuidv4 } from 'uuid';
+import { connectedUsers } from './state';
+import { broadcastToServer } from './broadcast';
+import { handleWebSocketMessage } from './handler';
+
+export function setupWebSocket(server: Server<typeof IncomingMessage, typeof ServerResponse>): void {
+  const wss = new WebSocketServer({ server });
+
+  wss.on('connection', (ws: WebSocket) => {
+    const connectionId = uuidv4();
+
+    connectedUsers.set(connectionId, { oderId: connectionId, ws, serverIds: new Set() });
+
+    ws.on('message', (data) => {
+      try {
+        const message = JSON.parse(data.toString());
+
+        handleWebSocketMessage(connectionId, message);
+      } catch (err) {
+        console.error('Invalid WebSocket message:', err);
+      }
+    });
+
+    ws.on('close', () => {
+      const user = connectedUsers.get(connectionId);
+
+      if (user) {
+        user.serverIds.forEach((sid) => {
+          broadcastToServer(sid, {
+            type: 'user_left',
+            oderId: user.oderId,
+            displayName: user.displayName,
+            serverId: sid
+          }, user.oderId);
+        });
+      }
+
+      connectedUsers.delete(connectionId);
+    });
+
+    ws.send(JSON.stringify({ type: 'connected', connectionId, serverTime: Date.now() }));
+  });
+}