Created socket session middleware for sockets

Author: muki119

services/devicecontrol/package.json

@@ -33,6 +33,7 @@
 		"pino-loki": "^3.0.0",
 		"prom-client": "^15.1.3",
 		"redis": "^5.10.0",
+		"socket.io": "^4.8.3",
 		"tsx": "^4.21.0"
 	},
 	"devDependencies": {

services/devicecontrol/src/helpers/access_tokens.ts

@@ -0,0 +1,4 @@
+import { CreateVerifyAccessTokenInstance } from "@services/common/helpers"
+import Logger from "../config/logger";
+
+export const VerifyAccessToken = CreateVerifyAccessTokenInstance(Logger);

services/devicecontrol/src/helpers/index.ts

@@ -1,2 +1,3 @@
 export * from "./generate_pairing_code"
-export * from "./device_tokens"
+export * from "./device_tokens"
+export * from "./access_tokens"

services/devicecontrol/src/middleware/error_handler_middleware.ts

@@ -1,5 +1,4 @@
 import type { Request, Response, NextFunction } from 'express';
 import logger from '../config/logger';
 import { ErrorHandlerMiddleware } from '@services/common/middleware';
-const ErrorHandler = new ErrorHandlerMiddleware(logger).middleware;
-export default ErrorHandler;
+export const ErrorHandler = new ErrorHandlerMiddleware(logger).middleware;

services/devicecontrol/src/middleware/index.ts

@@ -1,3 +1,4 @@
 
-export { default as ValidSessionMiddleware } from "./valid_session_middleware";
-export { default as ErrorHandlerMiddleware } from "./error_handler_middleware";
+export * from "./valid_session_middleware";
+export * from "./error_handler_middleware";
+export * from "./valid_socket_session_middleware"

services/devicecontrol/src/middleware/valid_session_middleware.ts

@@ -1,4 +1,3 @@
 import { SessionMiddleware } from '@services/common/middleware';
 import logger from '../config/logger';
-const ValidSessionMiddleware = new SessionMiddleware(logger).middleware;
-export default ValidSessionMiddleware;
+export const ValidSessionMiddleware = new SessionMiddleware(logger).middleware;

services/devicecontrol/src/middleware/valid_socket_session_middleware.ts

@@ -0,0 +1,27 @@
+import type { Socket, Event } from "socket.io"
+import { VerifyAccessToken } from "../helpers"
+import logger from "../config/logger";
+
+
+
+export async function ValidSocketSessionMiddleware(socket: Socket, next: (err?: Error) => void) {
+    try {
+        const token = socket.handshake.auth.token;
+        if (!token) {
+            return next(new Error("Authentication error: No token provided"));
+        }
+        const payload = await VerifyAccessToken(token);
+        if (!payload) {
+            return next(new Error("Authentication error: Invalid token"));
+        }
+        (socket as any).user = payload; // cba to extend for user payload
+        const tokenExpiry = payload.exp * 1000 - Date.now();
+        setTimeout(() => { // disconnect socket when token expires - security measure
+            socket.disconnect(true);
+        }, tokenExpiry);
+        next();
+    } catch (err) {
+        logger.error({ error: err }, "Error in ValidSocketSessionMiddleware:");
+        next(new Error("Authentication error"));
+    }
+}