Strider-CD/strider

const io = require('socket.io');
const cookieParser = require('cookie-parser');
const config = require('./config');
const common = require('./common');
const UserSocket = require('./utils/user-socket');
const expressParser = cookieParser(config.session_secret);

/*
 * websockets.init()
 *
 * Initialize the Socket.io server.
 */
// sio: socketio server. ex: io.listen(server)
function UserSockets(sio, sessionStore) {
  this.sio = sio;
  this.sockets = {};
  this.sessionStore = sessionStore;
  //sio.enable('browser client minification');  // send minified client
  //sio.enable('browser client etag');          // apply etag caching logic based on version number
  //sio.enable('browser client gzip');
  //sio.set('log level', 1);
  //sio.set('authorization', authorize.bind(this, sessionStore))
  sio.use(authorize.bind(this, sessionStore));
  sio.sockets.on('connection', this.connected.bind(this));
}

module.exports = UserSockets;

UserSockets.prototype = {
  addSocket: function(uid, socket) {
    if (!this.sockets[uid]) {
      this.sockets[uid] = new UserSocket(uid);
    }
    this.sockets[uid].add(socket);
  },
  // -> true if the socket was found and removed. false if it wasn't found
  removeSocket: function(uid, socket) {
    const socks = this.sockets[uid];
    if (!socks) return false;
    return socks.remove(socket);
  },
  // socket callback. Adds a new socket
  connected: function(socket) {
    const session = socket.handshake.session;

    if (session && session.passport) {
      this.addSocket(session.passport.user, socket);
    } else {
      console.debug(
        'Websocket connection does not have authorization - nothing to do.'
      );
    }
  },
  // send a message to a number of users
  // send([uid, uid, ...], arguments)
  send: function(users, args) {
    for (let i = 0; i < users.length; i++) {
      if (!this.sockets[users[i]]) continue;
      this.sockets[users[i]].emit(args);
    }
  },
  // send a message to a number of users running callback to get args
  // send([uid, uid, ...], callback)
  sendEach: function(users, fn) {
    for (let i = 0; i < users.length; i++) {
      if (!this.sockets[users[i]] || !this.sockets[users[i]].user) continue;
      this.sockets[users[i]].emit(fn(this.sockets[users[i]].user));
    }
  },
  // send a public message - to all /but/ the specified users
  sendPublic: function(users, args) {
    for (const id in this.sockets) {
      if (users.indexOf(id) !== -1) continue;
      this.sockets[id].emit(args);
    }
  }
};

function authorize(sessionStore, data, next) {
  if (data.handshake.headers.cookie) {
    const req = data.handshake;
    expressParser(req, {}, function() {
      const sessionID = req.signedCookies['connect.sid'];
      sessionStore.get(sessionID, function(err, session) {
        if (err || !session) {
          next(new Error('not authorized'));
        } else {
          req.session = session;
          next();
        }
      });
    });
  } else {
    return next(new Error('not authorized'));
  }
}

module.exports.init = function(server, sessionStore) {
  return (common.ws = new UserSockets(io.listen(server), sessionStore));
};