ensl_gathers/config/socketio.js

"use strict";

var winston = require("winston");
var User = require("../lib/user/user");
var config = require("./config");
var EnslClient = require("../lib/ensl/client");
var chatController = require("../lib/chat/controller");
var gatherController = require("../lib/gather/controller");
var userController = require("../lib/user/controller");
var usersHelper = require("../lib/user/helper");
var env = process.env.NODE_ENV || "development";
var parseCookies = EnslClient.parseCookies;

var assignRandomUser = (socket, next) => {
	usersHelper.getRandomUser(function (error, user) {
		if (error) {
			winston.error(error);
			return next(new Error("Authentication Failed"))
		}
		socket._user = user;
		return next();
	});
};

module.exports = io => {
	var rootNamespace = io.of('/')

	// Authentication
	io.use((socket, next) => {
		let cookies = parseCookies(socket);

		let session;
		if (cookies) {
			session = EnslClient.decodeSession(cookies[config.session_store_name]);
		}

		if (!session || typeof session.user !== 'number') {
			if (env === 'staging') {
				return assignRandomUser(socket, next);
			} else {
				return next(new Error("Authentication Failed"));
			}
		}

		User.find(session.user, (error, user) => {
			if (error) {
				winston.error(error);
				return next(new Error("Authentication failed"));
			}
			socket._user = user;
			if (socket._user.bans.gather) return next(new Error("Gather Banned"));
			winston.info("Logged in:", user.username, user.id);
			return next();
		});
	});

	userController(rootNamespace);
	chatController(rootNamespace);
	gatherController(rootNamespace);
};