Update server.js

server.js

@@ -3,17 +3,26 @@ const PORT = 7860;
 
 const wss = new WebSocket.Server({ port: PORT });
 
-wss.on('connection', (ws) => {
-  console.log('Client connected');
+// This counter represents number of connected clients
+function broadcastClientCount() {
+  const count = wss.clients.size;
+  const message = JSON.stringify({ clients: count });
 
-  ws.on('message', (message) => {
-    console.log(`Received: ${message}`);
-    ws.send(`Echo: ${message}`);
+  wss.clients.forEach((client) => {
+    if (client.readyState === WebSocket.OPEN) {
+      client.send(message);
+    }
   });
+}
+
+wss.on('connection', (ws) => {
+  console.log('Client connected');
+  broadcastClientCount(); // On connect
 
   ws.on('close', () => {
     console.log('Client disconnected');
+    broadcastClientCount(); // On disconnect
   });
 });
 
-console.log(`WebSocket server is running on ws://localhost:${PORT}`);
+console.log(`WebSocket server running on ws://localhost:${PORT}`);