Refactor authOptions to update user role in token if ID is provided

src/lib/auth-options.ts

@@ -5,6 +5,7 @@ import { PrismaAdapter } from "@next-auth/prisma-adapter";
 import prisma from "@/server/db";
 import { UserRoleType } from "@/types";
 import { JWT } from "next-auth/jwt";
+import { getUserById } from "@/app/actions/get-user-by-id";
 
 declare module "next-auth" {
     interface Session {
@@ -39,6 +40,13 @@ export const authOptions: NextAuthOptions = {
                     role: user.role,
                 };
             }
+            if (token.id) {
+                const updatedUser = await getUserById(token.id);
+                return {
+                    ...token,
+                    role: updatedUser?.role,
+                };
+            }
             return token;
         },
         async session({ session, token }: { session: NextAuthSession; token: JWT }): Promise<any> {