From c35af15d5502a31cf46583efb4892df04c579d52 Mon Sep 17 00:00:00 2001 From: Matheus Sampaio Queiroga Date: Tue, 23 Nov 2021 16:01:15 -0300 Subject: [PATCH] Update api.js --- src/api.js | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/src/api.js b/src/api.js index 0b03dd0..6fb43e7 100644 --- a/src/api.js +++ b/src/api.js @@ -35,14 +35,19 @@ const Server = require("http").createServer(app); const SocketIo = require("socket.io"); const io = new SocketIo.Server(Server); io.use(function (socket, next) { - if (socket.handshake.query.token) { - if (BdsChecks.token_verify(socket.handshake.query.token)) { - socket.token = socket.handshake.query.token; - next(); + const { headers, query } = socket.handshake; + const Token = headers["AuthorizationToken"] || query["token"] || query["Token"]; + if (Token) { + if (BdsChecks.token_verify(Token)) { + socket.token = Token; + return next(); } } return next(new Error("Token is not valid")); }); +io.on("connection", socket => { + console.log("Socket.io connection ID:", socket.token); +}); module.exports.SocketIO = io; // Routes -- 2.49.0