File size: 1,244 Bytes
803f4c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | import React from 'react';
import { useAuth } from '@/lib/auth';
type Role = 'STUDENT' | 'ADMIN' | 'ORG_MEMBER' | 'ORG_ADMIN' | 'SUPER_ADMIN';
interface RoleGuardProps {
children: React.ReactNode;
allowedRoles?: Role[];
requireSuperAdmin?: boolean;
fallback?: React.ReactNode;
}
/**
* RoleGuard component to conditionally render content based on user roles.
*
* Usage:
* <RoleGuard allowedRoles={['SUPER_ADMIN', 'ADMIN']}>
* <SecretComponent />
* </RoleGuard>
*/
export const RoleGuard: React.FC<RoleGuardProps> = ({
children,
allowedRoles,
requireSuperAdmin = false,
fallback = null
}) => {
const { user } = useAuth();
if (!user) return <>{fallback}</>;
const userRole = user.role as Role;
// Special case for Super Admin logic
if (requireSuperAdmin) {
if (userRole === 'SUPER_ADMIN' || userRole === 'ADMIN') {
return <>{children}</>;
}
return <>{fallback}</>;
}
// Default role check
if (allowedRoles && allowedRoles.length > 0) {
if (allowedRoles.includes(userRole)) {
return <>{children}</>;
}
return <>{fallback}</>;
}
return <>{children}</>;
};
export default RoleGuard;
|