ws-rooms-implementation.js

const wss = new ws.Server({ server: server });

_.broadcast = (data, room) => {
  wss.clients.forEach(client => {
    if (client.readyState === ws.OPEN) {
      if (!room || client.room === room)
        client.send(JSON.stringify(data), error => {
          if (error) console.log(error);
        });
    }
  });
};

_.users = 0;

wss.on('connection', (socket, req) => {
  socket.parse = content => socket.send(JSON.stringify(content));
  socket.on('message', msg => {
    msg = JSON.parse(msg);
    console.log(msg);
    switch (msg.type) {
      case 'setup':
        /**
         * {
         *  type: 'setup',
         *  data: {
         *    name: String,
         *    room: String
         *   }
         * }
         */
        if (!msg.data.name) break;
        socket._name = msg.data.name;
        socket._room = msg.data.room;
        socket._uid = _.users++;
        socket.parse({ type: 'id', data: { uid: socket._uid } });
        _.broadcast(
          {
            type: 'join',
            data: {
              name: socket._name,
              uid: socket._uid
            }
          },
          socket._room
        );
        break;

      case 'message':
        /**
         * {
         *  type: 'message',
         *  data: {
         *    text: String,
         *    time: Date
         *   }
         * }
         */
        if (!msg.data.text || !msg.data.time) break;
        _.broadcast(
          {
            type: 'message',
            data: {
              name: socket._name,
              uid: socket._uid,
              text: msg.data.text,
              time: new Date(msg.data.time).toUTCString(),
              msgID: uuidv4()
            }
          },
          socket._room
        );
    }
  });

  socket.on('close', () => {
    if (socket._name && socket._uid && socket._room)
      _.broadcast(
        {
          type: 'leave',
          data: {
            name: socket._name,
            uid: socket._uid
          }
        },
        socket._room
      );
  });
});