SocketHandler base class for Primus

socket-handler.js

class SocketHandler {
  get room() {
    return this.primus.rooms(this.roomName);
  }

  constructor(primus, roomName, actionHandlers, disconnectionHandler) {
    this.primus = primus;
    this.roomName = roomName;

    primus.on('connection', (spark) => {
      spark.on('data', ({ room, action, payload = {} }) => {
        // Only handle messages which are sent to the room
        if (room !== roomName) return;

        // Forward payload to the corresponding action handler
        const actionHandler = actionHandlers[action];
        if (actionHandler) {
          actionHandler(spark, payload);
        }
      });
    });

    primus.on('disconnection', (spark) => {
      if (disconnectionHandler) {
        disconnectionHandler(spark, {});
      }
    });
  }
}

module.exports = SocketHandler;