Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
| import { Navigate, Outlet } from 'react-router-dom'; | |
| import { useAuth } from './AuthContext'; | |
| export const ProtectedRoute = ({ requiredRole, children }) => { | |
| const { token, loading, user } = useAuth(); | |
| if (loading) { | |
| return ( | |
| <div className="min-h-screen flex flex-col items-center justify-center bg-surface-container-lowest font-body-md text-on-surface-variant"> | |
| <div className="flex flex-col items-center justify-center gap-lg animate-pulse"> | |
| <img src="/logo.png" alt="LarShield Logo" className="h-16 object-contain" /> | |
| <div className="flex items-center gap-sm"> | |
| <span className="material-symbols-outlined animate-spin text-[20px] text-primary">sync</span> | |
| <span className="font-bold tracking-wide">Verifying LarShield Session...</span> | |
| </div> | |
| </div> | |
| </div> | |
| ); | |
| } | |
| if (!token) return <Navigate to="/login" replace />; | |
| if (requiredRole) { | |
| const isMasterAuth = sessionStorage.getItem('superAdminAuth') === 'true'; | |
| const isImpersonating = !!localStorage.getItem('original_admin_token'); | |
| const role = user?.role; | |
| let hasPermission = false; | |
| if (Array.isArray(requiredRole)) { | |
| hasPermission = requiredRole.includes(role) || role === 'super_admin' || isMasterAuth; | |
| } else if (requiredRole === 'super_admin') { | |
| hasPermission = role === 'super_admin' || isMasterAuth || isImpersonating; | |
| } else if (requiredRole === 'admin') { | |
| hasPermission = role === 'admin' || role === 'super_admin' || isMasterAuth || isImpersonating; | |
| } else if (requiredRole === 'support_engineer') { | |
| hasPermission = role === 'support_engineer' || role === 'admin' || role === 'super_admin' || isMasterAuth; | |
| } else { | |
| hasPermission = role === requiredRole || role === 'super_admin' || isMasterAuth; | |
| } | |
| if (!hasPermission) { | |
| return <Navigate to="/dashboard" replace />; | |
| } | |
| } | |
| return children ? children : <Outlet />; | |
| }; | |