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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
const jwt = require("jsonwebtoken");
const uuidv4 = require("uuid/v4");
const WebSocket = require("ws");
wss = new WebSocket.Server({ clientTracking: false, noServer: true });
/* eslint-disable-next-line */
const pick = (O, ...K) => K.reduce((o, k) => ((o[k] = O[k]), o), {});
emit = (username, type, data) => {
if (data.username !== undefined) data = pick(data, ...whitelist);
if (!sockets[username]) return;
for (id in sockets[username]) {
const ws = sockets[username][id];
ws.send(JSON.stringify({ type, data }));
}
};
broadcast = (type, data) => {
for (const username in sockets) {
for (const id in sockets[username]) {
const ws = sockets[username][id];
ws.send(JSON.stringify({ type, data }));
}
}
};
server.on("upgrade", function(req, socket, head) {
wss.handleUpgrade(req, socket, head, function(ws) {
wss.emit("connection", ws, req);
});
});
const names = {};
wss.on("connection", function(ws, req) {
const id = uuidv4();
ws.send(JSON.stringify({ type: "connected" }));
ws.on("message", async message => {
let type, data;
try {
({ type, data } = JSON.parse(message));
} catch (e) {
l.error("coudn't parse socket message");
}
let token;
switch (type) {
case "login":
token = data;
if (!token || token === "null") return;
try {
const username = jwt.decode(token).username;
if (!sockets[username]) sockets[username] = {};
sockets[username][id] = ws;
names[id] = username;
let user = await getUser(username);
let payments = await user.getPayments({
where: {
account_id: user.account_id
},
order: [["id", "DESC"]],
limit: 12,
include: {
model: db.Account,
as: "account"
}
});
let accounts = await user.getAccounts();
let keys = await user.getKeys();
user = user.get({ plain: true });
user.accounts = accounts;
user.keys = keys;
user.payments = payments;
ws.user = user;
ws.send(JSON.stringify({ type: "login", data: user }));
ws.send(
JSON.stringify({ type: "version", data: config.clientVersion })
);
l.info("socket open", username, id);
} catch (e) {
ws.close();
l.error(
"problem opening socket",
token,
e.message,
id,
req.connection.remoteAddress
);
}
break;
case "token":
token = data;
if (!ws.user) return;
ws.token = token;
break;
case "lnurl":
sessions[data] = ws;
break;
default:
l.warn("received socket message of unknown type", type, data);
}
});
ws.on("close", function() {
try {
const username = names[id];
if (sockets[username] && sockets[username][id])
delete sockets[username][id];
l.info("socket closed", username, id);
} catch (e) {
l.error("problem closing socket", e.message);
}
});
});
|
