NodeBB/src/socket.io/index.js

var	SocketIO = require('socket.io'),
	socketioWildcard = require('socket.io-wildcard'),
	util = require('util'),
	async = require('async'),
	path = require('path'),
	fs = require('fs'),
	nconf = require('nconf'),
	express = require('express'),
	socketCookieParser = express.cookieParser(nconf.get('secret')),
	winston = require('winston'),

	db = require('../database'),
	user = require('../user'),
	topics = require('../topics'),
	logger = require('../logger'),

	Sockets = {},
	Namespaces = {};

/* === */

var users = {},
	userSockets = {},
	rooms = {},
	io;

Sockets.init = function() {
	io = socketioWildcard(SocketIO).listen(global.server, {
		log: false,
		transports: ['websocket', 'xhr-polling', 'jsonp-polling', 'flashsocket'],
		'browser client minification': true
	});

	fs.readdir(__dirname, function(err, files) {
		files.splice(files.indexOf('index.js'), 1);

		async.each(files, function(lib, next) {
			lib = lib.slice(0, -3);
			Namespaces[lib] = require('./' + lib);
			next();
		});
	});

	io.sockets.on('connection', function(socket) {
		var hs = socket.handshake,
			sessionID, uid, lastPostTime = 0;

		// Validate the session, if present
		socketCookieParser(hs, {}, function(err) {
			sessionID = socket.handshake.signedCookies["express.sid"];
			db.sessionStore.get(sessionID, function(err, sessionData) {
				if (!err && sessionData && sessionData.passport && sessionData.passport.user) {
					uid = users[sessionID] = sessionData.passport.user;
				} else {
					uid = users[sessionID] = 0;
				}

				userSockets[uid] = userSockets[uid] || [];
				userSockets[uid].push(socket);

				/* Need to save some state for the logger & maybe some other modules later on */
				socket.state = {
					user : {
						uid : uid
					}
				};

				/* If meta.config.loggerIOStatus > 0, logger.io_one will hook into this socket */
				logger.io_one(socket,uid);

				if (uid) {

					db.sortedSetAdd('users:online', Date.now(), uid, function(err, data) {
						socket.join('uid_' + uid);

						user.getUserField(uid, 'username', function(err, username) {
							socket.emit('event:connect', {
								status: 1,
								username: username,
								uid: uid
							});
						});
					});
				}

				io.sockets.in('global').emit('api:user.isOnline', isUserOnline(uid));
			});
		});

		socket.on('disconnect', function() {

			var index = (userSockets[uid] || []).indexOf(socket);
			if (index !== -1) {
				userSockets[uid].splice(index, 1);
			}

			if (userSockets[uid] && userSockets[uid].length === 0) {
				delete users[sessionID];
				delete userSockets[uid];
				if (uid) {
					db.sortedSetRemove('users:online', uid, function(err, data) {
					});
				}
			}

			io.sockets.in('global').emit('api:user.isOnline', isUserOnline(uid));

			emitOnlineUserCount();

			for (var roomName in rooms) {

				socket.leave(roomName);

				if (rooms[roomName][socket.id]) {
					delete rooms[roomName][socket.id];
				}

				updateRoomBrowsingText(roomName);
			}
		});

		socket.on('reconnected', function() {
			if (uid) {
				topics.pushUnreadCount(uid);
				user.pushNotifCount(uid);
			}

			if (process.env.NODE_ENV === 'development') {
				if (uid) {
					winston.info('[socket] uid ' + uid + ' (' + sessionID + ') has successfully reconnected.');
				} else {
					winston.info('[socket] An anonymous user (' + sessionID + ') has successfully reconnected.');
				}
			}
		});

		socket.on('*', function(payload, callback) {
			// Ignore all non-api messages
			if (payload.name.substr(0, 4) !== 'api:') {
				return;
			} else {
				// Deconstruct the message
				var parts = payload.name.slice(4).split('.'),
					namespace = parts[0],
					command = parts[1],
					subcommand = parts[2],	// MUST ADD RECURSION (:P)
					executeHandler = function(args) {
						// Session data
						var	sessionData = {
								uid: uid,
								socket: socket,
								rooms: rooms,
								server: io
							},
							socketArgs = [];

						// Construct the arguments that'll get passed into each socket method
						if (args.length) {
							socketArgs = socketArgs.concat(args);
						}
						if (callback !== undefined) {
							socketArgs.push(callback);
						}
						socketArgs.push(sessionData);

						// winston.info('[socket.io] Executing: ' + payload.name);
						if (!subcommand) {
							Namespaces[namespace][command].apply(Namespaces[namespace], socketArgs);
						} else {
							Namespaces[namespace][command][subcommand].apply(Namespaces[namespace][command], socketArgs);
						}
					};

				if (Namespaces[namespace]) {
					executeHandler(payload.args);
				} else {
					winston.warn('[socket.io] Unrecognized message: ' + payload.name);
				}
			}
		});
	});
}

Sockets.logoutUser = function(uid) {
	if(userSockets[uid] && userSockets[uid].length) {
		for(var i=0; i< userSockets[uid].length; ++i) {
			userSockets[uid][i].emit('event:disconnect');
			userSockets[uid][i].disconnect();

			if(!userSockets[uid]) {
				return;
			}
		}
	}
}

Sockets.emitUserCount = function() {
	db.getObjectField('global', 'userCount', function(err, count) {
		io.sockets.emit('user.count', {
			count: count
		});
	});
};

Sockets.in = function(room) {
	return io.sockets.in(room);
};

Sockets.getConnectedClients = function() {
	return userSockets;
}

Sockets.getOnlineAnonCount = function () {
	return userSockets[0] ? userSockets[0].length : 0;
};

/* Helpers */

function isUserOnline(uid) {
	return !!userSockets[uid] && userSockets[uid].length > 0;
}
Sockets.isUserOnline = isUserOnline;

Sockets.updateRoomBrowsingText = updateRoomBrowsingText;
function updateRoomBrowsingText(roomName) {

	function getUidsInRoom(room) {
		var uids = [];
		for (var socketId in room) {
			if (uids.indexOf(room[socketId]) === -1)
				uids.push(room[socketId]);
		}
		return uids;
	}

	function getAnonymousCount(roomName) {
		var clients = io.sockets.clients(roomName);
		var anonCount = 0;

		for (var i = 0; i < clients.length; ++i) {
			var hs = clients[i].handshake;
			if (hs && clients[i].state && clients[i].state.user.uid === 0) {
				++anonCount;
			}
		}
		return anonCount;
	}

	var	uids = getUidsInRoom(rooms[roomName]),
		anonymousCount = getAnonymousCount(roomName);

	if (uids.length === 0) {
		io.sockets.in(roomName).emit('api:get_users_in_room', { users: [], anonymousCount: anonymousCount });
	} else {
		user.getMultipleUserFields(uids, ['uid', 'username', 'userslug', 'picture'], function(err, users) {
			if(!err)
				io.sockets.in(roomName).emit('api:get_users_in_room', { users: users, anonymousCount: anonymousCount });
		});
	}
}

Sockets.emitTopicPostStats = emitTopicPostStats;
function emitTopicPostStats(callback) {
	db.getObjectFields('global', ['topicCount', 'postCount'], function(err, data) {
		if (err) {
			return winston.err(err);
		}

		var stats = {
			topics: data.topicCount ? data.topicCount : 0,
			posts: data.postCount ? data.postCount : 0
		};

		if (!callback) {
			io.sockets.emit('post.stats', stats);
		} else {
			callback(stats);
		}
	});
}

Sockets.emitOnlineUserCount = emitOnlineUserCount;
function emitOnlineUserCount(callback) {
	var anon = userSockets[0] ? userSockets[0].length : 0;
	var registered = Object.keys(userSockets).length;
	if (anon)
		registered = registered - 1;

	var returnObj = {
		users: registered + anon,
		anon: anon
	};

	if (callback) {
		callback(returnObj);
	} else {
		io.sockets.emit('api:user.active.get', returnObj)
	}
}

function emitAlert(socket, title, message) {
	socket.emit('event:alert', {
		type: 'danger',
		timeout: 2000,
		title: title,
		message: message,
		alert_id: 'post_error'
	});
}

function emitContentTooShortAlert(socket) {
	socket.emit('event:alert', {
		type: 'danger',
		timeout: 2000,
		title: 'Content too short',
		message: "Please enter a longer post. At least " + meta.config.minimumPostLength + " characters.",
		alert_id: 'post_error'
	});
}

function emitTooManyPostsAlert(socket) {
	socket.emit('event:alert', {
		title: 'Too many posts!',
		message: 'You can only post every ' + meta.config.postDelay + ' seconds.',
		type: 'danger',
		timeout: 2000
	});
}

/* Exporting */
module.exports = Sockets;