add solution

package.json

@@ -17,7 +17,7 @@
   "license": "GPL-3.0",
   "devDependencies": {
     "@mate-academy/eslint-config": "latest",
-    "@mate-academy/scripts": "^1.8.6",
+    "@mate-academy/scripts": "^2.1.3",
     "eslint": "^8.57.0",
     "eslint-plugin-jest": "^28.6.0",
     "eslint-plugin-node": "^11.1.0",
@@ -26,5 +26,10 @@
   },
   "mateAcademy": {
     "projectType": "javascript"
+  },
+  "dependencies": {
+    "cors": "^2.8.6",
+    "express": "^5.2.1",
+    "ws": "^8.20.0"
   }
 }

src/handlers.js

@@ -0,0 +1,78 @@
+const { rooms } = require('./store');
+const { broadcastRooms, broadcastToRoom } = require('./utils');
+
+function handleMessage(ws, wss, messageAsString) {
+  const data = JSON.parse(messageAsString);
+
+  switch (data.type) {
+    case 'JOIN_ROOM':
+      ws.roomId = data.roomId;
+      ws.username = data.username;
+
+      if (!rooms[data.roomId]) {
+        rooms[data.roomId] = {
+          id: data.roomId,
+          name: data.roomId,
+          messages: [],
+        };
+      }
+
+      ws.send(
+        JSON.stringify({
+          type: 'ROOM_HISTORY',
+          roomId: data.roomId,
+          messages: rooms[data.roomId].messages,
+        }),
+      );
+      break;
+
+    case 'NEW_MESSAGE':
+      if (ws.roomId && rooms[ws.roomId]) {
+        const newMessage = {
+          id: Date.now().toString(),
+          author: ws.username,
+          time: new Date().toISOString(),
+          text: data.text,
+        };
+
+        rooms[ws.roomId].messages.push(newMessage);
+
+        broadcastToRoom(wss, ws.roomId, {
+          type: 'MESSAGE',
+          roomId: ws.roomId,
+          message: newMessage,
+        });
+      }
+      break;
+
+    case 'CREATE_ROOM':
+      const newRoomId = 'room_' + Date.now();
+
+      rooms[newRoomId] = { id: newRoomId, name: data.roomName, messages: [] };
+      broadcastRooms(wss);
+      break;
+
+    case 'RENAME_ROOM':
+      if (rooms[data.roomId] && data.roomId !== 'general') {
+        rooms[data.roomId].name = data.newName;
+        broadcastRooms(wss);
+      }
+      break;
+
+    case 'DELETE_ROOM':
+      if (rooms[data.roomId] && data.roomId !== 'general') {
+        delete rooms[data.roomId];
+
+        wss.clients.forEach((client) => {
+          if (client.roomId === data.roomId) {
+            client.roomId = 'general';
+            client.send(JSON.stringify({ type: 'ROOM_DELETED' }));
+          }
+        });
+        broadcastRooms(wss);
+      }
+      break;
+  }
+}
+
+module.exports = { handleMessage };

src/index.js

@@ -1 +1,13 @@
-'use strict';
+/* eslint-disable no-console */
+const http = require('http');
+const { setupWebSocket } = require('./src/websocket');
+
+const server = http.createServer();
+
+setupWebSocket(server);
+
+const PORT = 8080;
+
+server.listen(PORT, () => {
+  console.log(`Server is listening on port ${PORT}`);
+});

src/store.js

@@ -0,0 +1,5 @@
+const rooms = {
+  general: { id: 'general', name: 'General', messages: [] },
+};
+
+module.exports = { rooms };

src/utils.js

@@ -0,0 +1,28 @@
+const WebSocket = require('ws');
+const { rooms } = require('./store');
+
+function broadcastRooms(wss) {
+  const roomList = Object.values(rooms).map((room) => ({
+    id: room.id,
+    name: room.name,
+  }));
+  const message = JSON.stringify({ type: 'ROOMS_LIST', rooms: roomList });
+
+  wss.clients.forEach((client) => {
+    if (client.readyState === WebSocket.OPEN) {
+      client.send(message);
+    }
+  });
+}
+
+function broadcastToRoom(wss, roomId, messageObj) {
+  const message = JSON.stringify(messageObj);
+
+  wss.clients.forEach((client) => {
+    if (client.readyState === WebSocket.OPEN && client.roomId === roomId) {
+      client.send(message);
+    }
+  });
+}
+
+module.exports = { broadcastRooms, broadcastToRoom };

src/websocket.js

@@ -0,0 +1,47 @@
+const WebSocket = require('ws');
+const { rooms } = require('./store');
+const { handleMessage } = require('./handlers');
+
+function setupWebSocket(server) {
+  const wss = new WebSocket.Server({ server });
+
+  const interval = setInterval(() => {
+    wss.clients.forEach((ws) => {
+      if (ws.isAlive === false) {
+        return ws.terminate();
+      }
+
+      ws.isAlive = false;
+      ws.ping();
+    });
+  }, 30000);
+
+  wss.on('close', function close() {
+    clearInterval(interval);
+  });
+
+  wss.on('connection', (ws) => {
+    ws.isAlive = true;
+
+    ws.on('pong', () => {
+      ws.isAlive = true;
+    });
+
+    ws.send(
+      JSON.stringify({
+        type: 'ROOMS_LIST',
+        rooms: Object.values(rooms).map((r) => ({ id: r.id, name: r.name })),
+      }),
+    );
+
+    ws.on('message', (message) => handleMessage(ws, wss, message));
+
+    ws.on('close', () => {
+      ws.roomId = null;
+    });
+  });
+
+  return wss;
+}
+
+module.exports = { setupWebSocket };