const WebSocket = require('ws');
const PORT = 7860;

const wss = new WebSocket.Server({ port: PORT });

// This counter represents number of connected clients
function broadcastClientCount() {
  const count = wss.clients.size;
  const message = JSON.stringify({ clients: count });

  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 running on ws://localhost:${PORT}`);