-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
84 lines (76 loc) · 2.09 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
require('dotenv').config();
const WebSocket = require('ws');
const url = require('url');
const Rx = require('rxjs/Rx');
const auth = require('./auth');
const messagesRepository = require('./messagesRepository');
const connectedUsers = new Map();
const messages = Rx.Observable.create(function(observer) {
messagesRepository.on('message', (data) => {
observer.next(data);
});
});
messages.filter((message, index) => {
console.log(message);
return connectedUsers.has(message.to);
}).subscribe(message => {
connectedUsers.get(message.to).send(JSON.stringify({
event: 'message',
data: message.data
}));
});
const validEvents = new Set();
validEvents.add('message');
validEvents.add('file');
const invalidEventError = (event) => {
return JSON.stringify({
'event': 'error',
'data': `Unsupported event ${event}`
});
};
const wss = new WebSocket.Server({
port: process.env.PORT,
verifyClient: function(info, callback) {
const location = url.parse(info.req.url, true);
const decodeToken = auth.decodeToken(location.query.token);
auth
.authenticate(decodeToken.type, location.query.token)
.then(response => {
const {status, data} = JSON.parse(response);
info.req.user = Object.assign(decodeToken, data);
callback(true);
})
.catch(error => {
if('statusCode' in error) {
callback(false, error.statusCode);
}
callback(false);
});
}
});
function addMessageToUserQueue(message){
return messagesRepository.add(message);
}
wss.on('connection', (ws, request) => {
const user = request.user;
const key = `${user.type}:${user.sub}`;
connectedUsers.set(key, ws);
ws.send(JSON.stringify({
event: 'message',
data: `Welcome ${user.username}`
}));
ws.on('message', (message) => {
const { event, data } = JSON.parse(message);
if (!validEvents.has(event)) {
ws.send(invalidEventError(event));
}
switch (event) {
case 'message':
addMessageToUserQueue(data);
break;
case 'file':
break;
default:
}
});
});