hyperzlib
/
nodebb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

socket.io index refactor

Browse Source
v1.18.x
barisusakli 11 years ago
parent 14a5de6d9d
commit 3a172711f4
1 changed files with 133 additions and 121 deletions
Show Stats Download Patch File Download Diff File

254
src/socket.io/index.js
Unescape Escape View File

@ -19,173 +19,185 @@ var SocketIO = require('socket.io'),
Sockets = {}, Sockets = {},
Namespaces = {}; Namespaces = {};
/* === */
var io; var io;
Sockets.init = function(server) { Sockets.init = function(server) {
var config = { var config = {
transports: ['polling', 'websocket'], transports: ['websocket', 'polling'],
path: nconf.get('relative_path') + '/socket.io' path: nconf.get('relative_path') + '/socket.io'
}; };
requireModules();
io = new SocketIO(); io = new SocketIO();
addRedisAdapter(io); addRedisAdapter(io);
io.use(socketioWildcard); io.use(socketioWildcard);
io.use(authorize);
io.on('connection', onConnection);
io.listen(server, config); io.listen(server, config);
Sockets.server = io; Sockets.server = io;
};
fs.readdir(__dirname, function(err, files) { function onConnection(socket) {
files.splice(files.indexOf('index.js'), 1); socket.ip = socket.request.connection.remoteAddress;
async.each(files, function(lib, next) { logger.io_one(socket, socket.uid);
if (lib.substr(lib.length - 3) === '.js') {
lib = lib.slice(0, -3);
Namespaces[lib] = require('./' + lib);
}
next(); onConnect(socket);
});
socket.on('disconnect', function() {
onDisconnect(socket);
}); });
io.use(function(socket, next) { socket.on('*', function(payload) {
console.log('AUTH'); onMessage(socket, payload);
});
}
var handshake = socket.request, function onConnect(socket) {
sessionID; if (socket.uid) {
socket.join('uid_' + socket.uid);
socket.join('online_users');
async.parallel({
user: function(next) {
user.getUserFields(socket.uid, ['username', 'userslug', 'picture', 'status'], next);
},
isAdmin: function(next) {
user.isAdministrator(socket.uid, next);
}
}, function(err, userData) {
if (err || !userData.user) {
return;
}
socket.emit('event:connect', {
username: userData.user.username,
userslug: userData.user.userslug,
picture: userData.user.picture,
isAdmin: userData.isAdmin,
uid: socket.uid
});
if (!handshake) { socket.broadcast.emit('event:user_status_change', {uid: socket.uid, status: userData.user.status});
return next(new Error('[[error:not-authorized]]')); });
} } else {
socket.join('online_guests');
socket.emit('event:connect', {
username: '[[global:guest]]',
isAdmin: false,
uid: 0
});
}
}
cookieParser(handshake, {}, function(err) { function onDisconnect(socket) {
if (err) { var socketCount = Sockets.getUserSocketCount(socket.uid);
return next(err);
}
var sessionID = handshake.signedCookies['express.sid']; if (socket.uid && socketCount <= 0) {
socket.broadcast.emit('event:user_status_change', {uid: socket.uid, status: 'offline'});
}
db.sessionStore.get(sessionID, function(err, sessionData) { // TODO : needs fixing for cluster
if (err) {
return next(err);
}
if (sessionData && sessionData.passport && sessionData.passport.user) {
socket.uid = parseInt(sessionData.passport.user, 10);
} else {
socket.uid = 0;
}
next();
});
});
});
io.on('connection', function(socket) { // for(var roomName in io.sockets.manager.roomClients[socket.id]) {
console.log('CONNECTED', socket.uid, socket.id); // if (roomName.indexOf('topic') !== -1) {
// io.sockets.in(roomName.slice(1)).emit('event:user_leave', socket.uid);
// }
// }
}
socket.ip = socket.request.connection.remoteAddress; function onMessage(socket, payload) {
if (!payload.data.length) {
return winston.warn('[socket.io] Empty payload');
}
logger.io_one(socket, socket.uid); var eventName = payload.data[0];
var params = payload.data[1];
var callback = typeof payload.data[payload.data.length - 1] === 'function' ? payload.data[payload.data.length - 1] : function() {};
if (socket.uid) { if (!eventName) {
socket.join('uid_' + socket.uid); return winston.warn('[socket.io] Empty method name');
socket.join('online_users'); }
async.parallel({ if (ratelimit.isFlooding(socket)) {
user: function(next) { winston.warn('[socket.io] Too many emits! Disconnecting uid : ' + socket.uid + '. Message : ' + payload.name);
user.getUserFields(socket.uid, ['username', 'userslug', 'picture', 'status'], next); return socket.disconnect();
}, }
isAdmin: function(next) {
user.isAdministrator(socket.uid, next);
}
}, function(err, userData) {
if (err || !userData.user) {
return;
}
socket.emit('event:connect', {
status: 1,
username: userData.user.username,
userslug: userData.user.userslug,
picture: userData.user.picture,
isAdmin: userData.isAdmin,
uid: socket.uid
});
socket.broadcast.emit('event:user_status_change', {uid: socket.uid, status: userData.user.status});
});
} else {
socket.join('online_guests');
socket.emit('event:connect', {
status: 1,
username: '[[global:guest]]',
isAdmin: false,
uid: 0
});
}
socket.on('disconnect', function() { var parts = eventName.toString().split('.'),
var socketCount = Sockets.getUserSocketCount(socket.uid); namespace = parts[0],
console.log('DISCONNECT', socket.uid, socket.id); methodToCall = parts.reduce(function(prev, cur) {
if (socket.uid && socketCount <= 0) { if (prev !== null && prev[cur]) {
socket.broadcast.emit('event:user_status_change', {uid: socket.uid, status: 'offline'}); return prev[cur];
} else {
return null;
} }
}, Namespaces);
if(!methodToCall) {
if (process.env.NODE_ENV === 'development') {
winston.warn('[socket.io] Unrecognized message: ' + eventName);
}
return;
}
// for(var roomName in io.sockets.manager.roomClients[socket.id]) { if (Namespaces[namespace].before) {
// if (roomName.indexOf('topic') !== -1) { Namespaces[namespace].before(socket, eventName, function() {
// io.sockets.in(roomName.slice(1)).emit('event:user_leave', socket.uid); callMethod(methodToCall, socket, params, callback);
// }
// }
}); });
} else {
callMethod(methodToCall, socket, params, callback);
}
}
function requireModules() {
fs.readdir(__dirname, function(err, files) {
files.splice(files.indexOf('index.js'), 1);
socket.on('*', function(payload) { async.each(files, function(lib, next) {
if (!payload.data.length) { if (lib.substr(lib.length - 3) === '.js') {
return winston.warn('[socket.io] Empty payload'); lib = lib.slice(0, -3);
Namespaces[lib] = require('./' + lib);
} }
var eventName = payload.data[0]; next();
var params = payload.data[1]; });
var callback = typeof payload.data[payload.data.length - 1] === 'function' ? payload.data[payload.data.length - 1] : function() {}; });
}
if (!eventName) { function authorize(socket, next) {
return winston.warn('[socket.io] Empty method name'); var handshake = socket.request,
} sessionID;
if (ratelimit.isFlooding(socket)) { if (!handshake) {
winston.warn('[socket.io] Too many emits! Disconnecting uid : ' + socket.uid + '. Message : ' + payload.name); return next(new Error('[[error:not-authorized]]'));
return socket.disconnect(); }
}
var parts = eventName.toString().split('.'), cookieParser(handshake, {}, function(err) {
namespace = parts[0], if (err) {
methodToCall = parts.reduce(function(prev, cur) { return next(err);
if (prev !== null && prev[cur]) { }
return prev[cur];
} else {
return null;
}
}, Namespaces);
if(!methodToCall) {
if (process.env.NODE_ENV === 'development') {
winston.warn('[socket.io] Unrecognized message: ' + eventName);
}
return;
}
if (Namespaces[namespace].before) { var sessionID = handshake.signedCookies['express.sid'];
Namespaces[namespace].before(socket, eventName, function() {
callMethod(methodToCall, socket, params, callback); db.sessionStore.get(sessionID, function(err, sessionData) {
}); if (err) {
return next(err);
}
if (sessionData && sessionData.passport && sessionData.passport.user) {
socket.uid = parseInt(sessionData.passport.user, 10);
} else { } else {
callMethod(methodToCall, socket, params, callback); socket.uid = 0;
} }
next();
}); });
}); });
}; }
function addRedisAdapter(io) { function addRedisAdapter(io) {
if (nconf.get('redis')) { if (nconf.get('redis')) {

Write Preview
Loading…
Cancel
Save