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
|
const jwt = require("jsonwebtoken");
const io = require("socket.io");
const whitelist = require("./whitelist");
const Sequelize = require("sequelize");
/* eslint-disable-next-line */
const pick = (O, ...K) => K.reduce((o, k) => ((o[k] = O[k]), o), {});
const l = require("pino")();;
module.exports = (app, db, server) => {
const socket = io(server, { origins: "*:*" });
const sids = {};
const emit = (username, msg, data) => {
if (data.username !== undefined) data = pick(data, ...whitelist);
if (!sids[username]) return;
sids[username].map((sid, i) => {
try {
socket.to(sid).emit(msg, data);
} catch (e) {
sids[username].splice(i, 1);
}
});
};
socket.use((socket, next) => {
try {
let token = socket.handshake.query.token;
let user = jwt.decode(token).username;
socket.request.user = user;
sids[user] ? sids[user].push(socket.id) : (sids[user] = [socket.id]);
sids[socket.id] = user;
} catch (e) {
l.error("error creating socket", e);
}
next();
});
socket.sockets.on("connect", async socket => {
socket.emit("connected");
if (app.get("rates")) socket.emit("rate", app.get("rates").ask);
socket.on("getuser", async (data, callback) => {
l.info(`logging in ${socket.request.user}`);
const user = await db.User.findOne({
include: [
{
model: db.Payment,
as: "payments",
order: [["id", "DESC"]],
received: {
[Sequelize.Op.ne]: null
},
limit: 12
}
],
where: {
username: socket.request.user
}
});
callback(user);
});
socket.on("disconnect", s => {
let user = sids[socket.id];
l.info("logging out", user);
sids[user].splice(sids[user].indexOf(socket.id), 1);
delete sids[socket.id];
});
});
return [socket, emit];
};
|
