From 65c1001751642c73ddc560805006b8894749004b Mon Sep 17 00:00:00 2001
From: Mauricio Siu <47042324+Siumauricio@users.noreply.github.com>
Date: Fri, 27 Sep 2024 11:22:52 -0600
Subject: [PATCH] fix(swagger): add mising validation

---
 apps/dokploy/server/auth/token.ts | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/apps/dokploy/server/auth/token.ts b/apps/dokploy/server/auth/token.ts
index 54162fdcb..64009fef3 100644
--- a/apps/dokploy/server/auth/token.ts
+++ b/apps/dokploy/server/auth/token.ts
@@ -2,6 +2,8 @@ import type { IncomingMessage } from "node:http";
 import { TimeSpan } from "lucia";
 import { Lucia } from "lucia/dist/core.js";
 import { type ReturnValidateToken, adapter } from "./auth";
+import { findAdminByAuthId } from "../api/services/admin";
+import { findUserByAuthId } from "../api/services/user";
 
 export const luciaToken = new Lucia(adapter, {
 	sessionCookie: {
@@ -31,6 +33,16 @@ export const validateBearerToken = async (
 		};
 	}
 	const result = await luciaToken.validateSession(sessionId);
+
+	if (result.user) {
+		if (result.user?.rol === "admin") {
+			const admin = await findAdminByAuthId(result.user.id);
+			result.user.adminId = admin.adminId;
+		} else if (result.user?.rol === "user") {
+			const userResult = await findUserByAuthId(result.user.id);
+			result.user.adminId = userResult.adminId;
+		}
+	}
 	return {
 		session: result.session,
 		...((result.user && {