go-bds/Maneger
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Releases 71 Wiki Activity

Update api.js #247

Merged
Sirherobrine23 merged 1 commits from Socket into main 2021-11-23 19:22:38 +00:00
Conversation 0 Commits 1 Files Changed 1 +9 -4
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/api.js
View File

@ -35,14 +35,19 @@ const Server = require("http").createServer(app);
const SocketIo = require("socket.io"); const SocketIo = require("socket.io");
const io = new SocketIo.Server(Server); const io = new SocketIo.Server(Server);
io.use(function (socket, next) { io.use(function (socket, next) {
if (socket.handshake.query.token) { const { headers, query } = socket.handshake;
if (BdsChecks.token_verify(socket.handshake.query.token)) { const Token = headers["AuthorizationToken"] || query["token"] || query["Token"];
socket.token = socket.handshake.query.token; if (Token) {
next(); if (BdsChecks.token_verify(Token)) {
socket.token = Token;
return next();
} }
} }
return next(new Error("Token is not valid")); return next(new Error("Token is not valid"));
}); });
io.on("connection", socket => {
console.log("Socket.io connection ID:", socket.token);
});
module.exports.SocketIO = io; module.exports.SocketIO = io;
// Routes // Routes