interim commit - removed calls to websockets.js, beginning porting to namespaced files
Julian Lam
11 years ago
parent
1caaa8c20a
commit
adb8f80b86
@ -0,0 +1,297 @@
|
|||||||
|
var SocketIO = require('socket.io'),
|
||||||
|
socketioWildcard = require('socket.io-wildcard'),
|
||||||
|
util = require('util'),
|
||||||
|
async = require('async'),
|
||||||
|
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
|
||||||
|
});
|
||||||
|
|
||||||
|
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) {
|
||||||
|
// Ignore all non-api messages
|
||||||
|
if (payload.name.substr(0, 4) !== 'api:') {
|
||||||
|
return;
|
||||||
|
} else {
|
||||||
|
// Deconstruct the message
|
||||||
|
var parts = payload.name.split('.'),
|
||||||
|
namespace = parts[0],
|
||||||
|
command = parts[1],
|
||||||
|
subcommand = parts[2], // MUST ADD RECURSION (:P)
|
||||||
|
executeHandler = function(args) {
|
||||||
|
if (!subcommand) {
|
||||||
|
Namespaces[namespace][command](args);
|
||||||
|
} else {
|
||||||
|
Namespaces[namespace][command][subcommand](args);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if (Namespaces[namespace]) {
|
||||||
|
executeHandler(payload.args);
|
||||||
|
} else {
|
||||||
|
fs.exists(path.join(__dirname, namespace + '.js'), function(exists) {
|
||||||
|
if (exists) {
|
||||||
|
Namespaces[namespace] = require('./' + namespace);
|
||||||
|
executeHandler(payload.args);
|
||||||
|
} else {
|
||||||
|
winston.warn('[socket.io] Unrecognized message: ' + payload.name);
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
console.log('message!', arguments);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
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;
|
||||||
|
|
||||||
|
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 });
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function emitTopicPostStats() {
|
||||||
|
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
|
||||||
|
};
|
||||||
|
|
||||||
|
io.sockets.emit('post.stats', stats);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function emitOnlineUserCount() {
|
||||||
|
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
|
||||||
|
};
|
||||||
|
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;
|
||||||
Loading…
Reference in New Issue