84 lines
1.7 KiB
TypeScript
84 lines
1.7 KiB
TypeScript
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<any> {
|
|
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);
|
|
}
|
|
});
|
|
}
|