Update AfterLogin.tsx

uhawaii-system-its-ti-iam · Apr 17, 2024 · 09399ec · 09399ec
1 parent 71bc9ba
commit 09399ec
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/ui/src/app/(index)/_components/AfterLogin.tsx b/ui/src/app/(index)/_components/AfterLogin.tsx
@@ -3,7 +3,7 @@ import Role from '@/access/Role';
 import Image from 'next/image';
 import { KeyRound } from 'lucide-react';
 import Link from 'next/link';
-import {Button} from '@/components/ui/button';
+import { Button } from '@/components/ui/button';
 
 const AfterLogin = ({
     currentUser,
@@ -34,6 +34,7 @@ const AfterLogin = ({
                 width: 48,
                 height: 48,
             },
+            role: Role.ADMIN
         },
         {
             title: 'Memberships',
@@ -45,7 +46,8 @@ const AfterLogin = ({
                 width:54,
                 height:48,
             },
-            number:numberOfMemberships
+            number:numberOfMemberships,
+            role: Role.UH
         },
         {
             title:'Groupings',
@@ -58,7 +60,8 @@ const AfterLogin = ({
                 width:48,
                 height:48
             },
-            number:numberOfGroupings
+            number:numberOfGroupings,
+            role: Role.OWNER
         }
     ]
 
@@ -96,12 +99,7 @@ const AfterLogin = ({
 
             <div className="container grid grid-cols-1 md:grid-cols-3 gap-6 md:gap-8 mt-10 mb-10">
                 {items
-                    .filter((item, index) =>{
-                        if (isAdmin) return true;
-                        else if (isOwner) return index === 1 || index === 2;
-                        else return index === 1;
-
-                    })
+                    .filter((item) => currentUser.roles.includes(item.role))
                     .map((item, index)=>(
                         <div key={index} className="flex flex-col justify-between">
                             <div>