import * as http from 'http'; import * as websocket from 'websocket'; import * as redis from 'redis'; import User from './models/user'; import Userkey from './models/userkey'; import homeStream from './stream/home'; import messagingStream from './stream/messaging'; module.exports = (server: http.Server) => { /** * Init websocket server */ const ws = new websocket.server({ httpServer: server }); ws.on('request', async (request) => { const connection = request.accept(); const user = await authenticate(connection, request.resourceURL.query.i); if (user == null) { connection.send('authentication-failed'); connection.close(); return; } // Connect to Redis const subscriber = redis.createClient( config.redis.port, config.redis.host); connection.on('close', () => { subscriber.unsubscribe(); subscriber.quit(); }); const channel = request.resourceURL.pathname === '/' ? homeStream : request.resourceURL.pathname === '/messaging' ? messagingStream : null; if (channel !== null) { channel(request, connection, subscriber, user); } else { connection.close(); } }); }; function authenticate(connection: websocket.connection, token: string): Promise { return new Promise(async (resolve, reject) => { if (token[0] == '!') { // Fetch user // SELECT _id const user = await User .findOne({ token: token }, { _id: true }); resolve(user); } else { const userkey = await Userkey.findOne({ key: token }); if (userkey == null) { return reject('invalid userkey'); } // Fetch user // SELECT _id const user = await User .findOne({ _id: userkey.user_id }, { _id: true }); resolve(user); } }); }