Daniel Howard
93 linhas
3.0 KiB
JavaScript
93 linhas
3.0 KiB
JavaScript
var url = require('url'),
|
|
Hub = require('./im/hub');
|
|
|
|
module.exports = function setupHub(options) {
|
|
options = options || {};
|
|
|
|
store = new Hub(options);
|
|
|
|
return function session(req, res, next) {
|
|
req.sessionStore = store;
|
|
|
|
if(!req.cookies) {
|
|
next(new Error('session requires cookieParser to work properly'));
|
|
return;
|
|
}
|
|
|
|
if (!(options.authentication.cookie in req.cookies)) {
|
|
if (options.authentication.cookie in req.query) {
|
|
req.cookies[options.authentication.cookie] = req.query[options.authentication.cookie];
|
|
} else if (options.authentication.cookie in req.body) {
|
|
req.cookies[options.authentication.cookie] = req.body[options.authentication.cookie];
|
|
}
|
|
}
|
|
|
|
if (!('callback' in req.cookies)) {
|
|
if ('callback' in req.query) {
|
|
req.cookies.callback = req.query.callback;
|
|
} else if ('callback' in req.body) {
|
|
req.cookies.callback = req.body.callback;
|
|
}
|
|
}
|
|
|
|
req.sessionID = req.cookies[options.authentication.cookie];
|
|
req.jsonpCallback = req.cookies.callback;
|
|
|
|
delete req.query[options.authentication.cookie];
|
|
delete req.body[options.authentication.cookie];
|
|
delete req.query.callback;
|
|
delete req.body.callback;
|
|
|
|
if(req.dev) {
|
|
next();
|
|
return;
|
|
}
|
|
|
|
if(req.sessionID) {
|
|
store.get(req, function(err, sess) {
|
|
if(err) {
|
|
next(err);
|
|
return;
|
|
}
|
|
|
|
if(!sess) {
|
|
next(new Error(JSON.stringify({
|
|
type: 'error',
|
|
error: 'not authenticated'})));
|
|
return;
|
|
}
|
|
|
|
sess.touch();
|
|
if(url.parse(req.url).pathname === '/listen') {
|
|
req.connection.setTimeout(5 * 60 * 1000);
|
|
sess.listener(res);
|
|
store.set(req.sessionID, sess);
|
|
|
|
if(msg = sess.message_queue.shift())
|
|
sess._send.apply(sess, msg);
|
|
} else {
|
|
sess.connection = res;
|
|
}
|
|
|
|
req.session = sess;
|
|
res.find = store.find.bind(store);
|
|
res.message = function(to, package) {
|
|
store.message(req.session, to, package);
|
|
};
|
|
res.status = function(value, message) {
|
|
req.session.status(value, message);
|
|
};
|
|
res.signOff = function() { store.signOff(req.sessionID); };
|
|
|
|
if(url.parse(req.url).pathname !== '/listen') {
|
|
next();
|
|
}
|
|
});
|
|
} else {
|
|
next(new Error(JSON.stringify({
|
|
type: 'error',
|
|
error: 'not authenticated'})));
|
|
}
|
|
};
|
|
};
|