From bce94bb738c22d7ebd6ab887da5844a1212884d7 Mon Sep 17 00:00:00 2001
From: Jozef Sovcik <jozef@sovcik.net>
Date: Mon, 4 Jan 2021 10:14:30 +0100
Subject: [PATCH] extending socket interface #4

---
 src/authorize.ts | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/authorize.ts b/src/authorize.ts
index c445eb7..0dfed57 100644
--- a/src/authorize.ts
+++ b/src/authorize.ts
@@ -7,8 +7,12 @@ interface ExtendedError extends Error {
   data?: any
 }
 
+interface ExtendedSocket extends Socket {
+  encodedToken?: string
+  decodedToken?: any
+
 type SocketIOMiddleware = (
-  socket: Socket,
+  socket: ExtendedSocket,
   next: (err?: ExtendedError) => void
 ) => void
 
@@ -41,7 +45,7 @@ export const authorize = (options: AuthorizeOptions): SocketIOMiddleware => {
       )
     }
     // Store encoded JWT
-    socket = Object.assign(socket, { encodedToken: token })
+    socket.encodedToken = token
     let payload: any
     try {
       payload = jwt.verify(token, secret, { algorithms })
@@ -53,7 +57,7 @@ export const authorize = (options: AuthorizeOptions): SocketIOMiddleware => {
       )
     }
     // Store decoded JWT
-    socket = Object.assign(socket, { decodedToken: payload })
+    socket.decodedToken = payload
     return next()
   }
 }