You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

111 lines
3.7 KiB

'use strict';
const xtend = require('xtend');
function parseCookie(auth, cookieHeader) {
const cookieParser = auth.cookieParser(auth.secret);
const req = {
headers:{
cookie: cookieHeader
}
};
let result;
cookieParser(req, {}, err => {
if (err) throw err;
result = req.signedCookies || req.cookies;
});
return result;
}
function authorize(options) {
const defaults = {
passport: require('passport'),
key: 'connect.sid',
secret: null,
store: null,
success: (data, accept) => {
if (data.socketio_version_1) {
accept();
} else {
accept(null, true);
}
},
fail: (data, message, critical, accept) => {
if (data.socketio_version_1) {
accept(new Error(message));
} else {
accept(null, false);
}
}
};
const auth = xtend(defaults, options);
auth.userProperty = auth.passport._userProperty || 'user';
if (!auth.cookieParser) {
throw new Error('cookieParser is required use connect.cookieParser or express.cookieParser');
}
return function (data, accept) {
// socket.io v1.0 now provides socket handshake data via `socket.request`
if (data.request) {
data = data.request;
data.socketio_version_1 = true;
}
data.cookie = parseCookie(auth, data.headers.cookie || '');
data.sessionID = (data.query && data.query.session_id) || data.cookie[auth.key] || '';
data[auth.userProperty] = {
logged_in: false
};
if (data.xdomain && !data.sessionID) {
return auth.fail(data, 'Can not read cookies from CORS-Requests. See CORS-Workaround in the readme.', false, accept);
}
auth.store.get(data.sessionID, (err, session) => {
if (err) {
return auth.fail(data, 'Error in session store:\n' + err.message, true, accept);
} else
if (!session) {
return auth.fail(data, 'No session found', false, accept);
} else
if (!session[auth.passport._key]) {
return auth.fail(data, 'Passport was not initialized', true, accept);
}
const userKey = session[auth.passport._key][auth.userProperty];
if (!userKey) {
return auth.fail(data, 'User not authorized through passport. (User Property not found)', false, accept);
}
// Because of authentication error removed
/*auth.passport.deserializeUser(userKey, function(err, user) {
if (err)
return auth.fail(data, err, true, accept);
if (!user)
return auth.fail(data, "User not found", false, accept);
data[auth.userProperty] = user;
data[auth.userProperty].logged_in = true;
auth.success(data, accept);
});*/
data[auth.userProperty] = userKey;
data[auth.userProperty].logged_in = true;
data.client._user = userKey;
auth.success(data, accept);
});
};
}
function filterSocketsByUser(socketIo, filter) {
const handshaken = socketIo.sockets.manager.handshaken;
return Object.keys(handshaken || {})
.filter(skey => filter(handshaken[skey].user))
.map(skey => socketIo.sockets.manager.sockets.sockets[skey]);
}
exports.authorize = authorize;
exports.filterSocketsByUser = filterSocketsByUser;