io.js 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. 'use strict';
  2. // This file contains all the logic for Socket.IO
  3. const app = require('./app');
  4. const actions = require('./actions');
  5. const cache = require('./cache');
  6. const utils = require('./utils');
  7. module.exports = {
  8. io: null,
  9. init: (cb) => {
  10. //TODO Check every 30s/60s, for all sockets, if they are still allowed to be in the rooms they are in, and on socket at all (permission changing/banning)
  11. this.io = require('socket.io')(app.server);
  12. this.io.use((socket, next) => {
  13. let cookies = socket.request.headers.cookie;
  14. // set the sessionId for the socket (this will have to be checked every request, this allows us to have a logout all devices option)
  15. if (cookies) socket.sessionId = utils.cookies.parseCookies(cookies).SID;
  16. return next();
  17. });
  18. this.io.on('connection', socket => {
  19. socket.join("SomeRoom");
  20. console.log("io: User has connected");
  21. // catch when the socket has been disconnected
  22. socket.on('disconnect', () => {
  23. // remove the user from their current station (if any)
  24. if (socket.sessionId) {
  25. //actions.stations.leave(socket.sessionId, result => {});
  26. delete socket.sessionId;
  27. }
  28. console.log('io: User has disconnected');
  29. });
  30. // catch errors on the socket (internal to socket.io)
  31. socket.on('error', err => console.log(err));
  32. // have the socket listen for each action
  33. Object.keys(actions).forEach((namespace) => {
  34. Object.keys(actions[namespace]).forEach((action) => {
  35. // the full name of the action
  36. let name = `${namespace}.${action}`;
  37. // listen for this action to be called
  38. socket.on(name, function () {
  39. let args = Array.prototype.slice.call(arguments, 0, -1);
  40. let cb = arguments[arguments.length - 1];
  41. // load the session from the cache
  42. cache.hget('sessions', socket.sessionId, (err, session) => {
  43. if (err && err !== true) {
  44. return cb({
  45. status: 'error',
  46. message: 'An error occurred while obtaining your session'
  47. });
  48. }
  49. // make sure the sockets sessionId isn't set if there is no session
  50. if (socket.sessionId && session === null) delete socket.sessionId;
  51. // call the action, passing it the session, and the arguments socket.io passed us
  52. actions[namespace][action].apply(null, [session].concat(args).concat([
  53. (result) => {
  54. // store the session id
  55. if (name == 'users.login' && result.user) socket.sessionId = result.user.sessionId;
  56. // respond to the socket with our message
  57. cb(result);
  58. }
  59. ]));
  60. });
  61. })
  62. })
  63. });
  64. //TODO check if session is valid before returning true/false
  65. if (socket.sessionId !== undefined) cache.hget('sessions', socket.sessionId, (err, session) => {
  66. if (err && err !== true) {
  67. socket.emit('ready', false);
  68. } else if (session) {
  69. socket.emit('ready', true);
  70. } else {
  71. socket.emit('ready', false);
  72. }
  73. });
  74. });
  75. cb();
  76. }
  77. };