chore: fixed behavior related to role changes

Author: mikita-kandratsyeu

components/common/premium-banner.tsx

@@ -5,7 +5,7 @@ import { Button } from '../ui';
 export const PremiumBanner = () => {
   return (
     <div className="border rounded-sm flex flex-col p-4 gap-y-2">
-      <h2 className="font-semibold tracking-tight text-base">Upgrade to Premium</h2>
+      <h2 className="font-semibold tracking-tight text-base">Upgrade to Nova Plus</h2>
       <p className="text-muted-foreground text-sm mb-2">
         Unlock premium courses, get access to Nova AI, and more.
       </p>

lib/auth.ts

@@ -115,23 +115,6 @@ export const authOptions = {
         token.email = user.email;
         token.isPublic = user.isPublic;
         token.role = user.role;
-
-        const updatedToken = await fetchCachedData(
-          `token-change-${token.email}`,
-          async () => {
-            const updatedUser = await db.user.findUnique({
-              where: { id: user?.id },
-              select: { role: true },
-            });
-
-            return { role: updatedUser?.role };
-          },
-          TEN_MINUTE_SEC,
-        );
-
-        if (updatedToken?.role && updatedToken.role !== user.role) {
-          token.role = updatedToken.role;
-        }
       }
 
       if (trigger === 'update') {
@@ -157,6 +140,23 @@ export const authOptions = {
         if (token.role) {
           session.user.role = (token.role as string) || UserRole.STUDENT;
         }
+
+        const updatedToken = await fetchCachedData(
+          `token-change-${token.email}`,
+          async () => {
+            const updatedUser = await db.user.findUnique({
+              where: { id: session?.user?.userId },
+              select: { role: true },
+            });
+
+            return { role: updatedUser?.role };
+          },
+          TEN_MINUTE_SEC,
+        );
+
+        if (updatedToken?.role && updatedToken.role !== session?.user?.role) {
+          session.user.role = updatedToken.role;
+        }
       }
 
       return session;