Autenticando o Websocket com token

rtenorioh · Oct 25, 2024 · cbada47 · cbada47
1 parent 980be20
commit cbada47
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/backend/src/libs/socket.ts b/backend/src/libs/socket.ts
@@ -1,5 +1,7 @@
-import { Server as SocketIO } from "socket.io";
 import { Server } from "http";
+import { verify } from "jsonwebtoken";
+import { Server as SocketIO } from "socket.io";
+import authConfig from "../config/auth";
 import AppError from "../errors/AppError";
 import { logger } from "../utils/logger";
 
@@ -13,6 +15,16 @@ export const initIO = (httpServer: Server): SocketIO => {
   });
 
   io.on("connection", socket => {
+    const { token } = socket.handshake.query;
+    let tokenData = null;
+    try {
+      tokenData = verify(token, authConfig.secret);
+      logger.debug(JSON.stringify(tokenData), "io-onConnection: tokenData");
+    } catch (error) {
+      logger.error(JSON.stringify(error), "Error decoding token");
+      socket.disconnect();
+      return io;
+    }
     logger.info("Client Connected");
     socket.on("joinChatBox", (ticketId: string) => {
       logger.info("A client joined a ticket channel");
@@ -32,6 +44,7 @@ export const initIO = (httpServer: Server): SocketIO => {
     socket.on("disconnect", () => {
       logger.info("Client disconnected");
     });
+    return socket;
   });
   return io;
 };

diff --git a/frontend/src/services/socket-io.js b/frontend/src/services/socket-io.js
@@ -2,7 +2,13 @@ import openSocket from "socket.io-client";
 import { getBackendUrl } from "../config";
 
 function connectToSocket() {
-    return openSocket(getBackendUrl());
+    const token = localStorage.getItem("token");
+    return openSocket(getBackendUrl(), {
+        transports: ["websocket", "polling", "flashsocket"],
+        query: {
+            token: JSON.parse(token),
+        },
+    });
 }
 
 export default connectToSocket;