37 lines
1001 B
TypeScript
37 lines
1001 B
TypeScript
import { useState, ReactNode } from 'react';
|
|
import { Navigate, useLocation } from 'react-router-dom';
|
|
// hooks
|
|
import useAuth from '../hooks/useAuth';
|
|
// components
|
|
import LoadingScreen from '../components/LoadingScreen';
|
|
|
|
// ----------------------------------------------------------------------
|
|
|
|
type AuthGuardProps = {
|
|
children: ReactNode;
|
|
};
|
|
|
|
export default function AuthGuard({ children }: AuthGuardProps) {
|
|
const { isAuthenticated, isInitialized } = useAuth();
|
|
const { pathname } = useLocation();
|
|
const [requestedLocation, setRequestedLocation] = useState<string | null>(null);
|
|
|
|
if (!isInitialized) {
|
|
return <LoadingScreen />;
|
|
}
|
|
|
|
if (!isAuthenticated) {
|
|
if (pathname !== requestedLocation) {
|
|
setRequestedLocation(pathname);
|
|
}
|
|
return <Navigate to="/auth/login" replace={true} />;
|
|
}
|
|
|
|
if (requestedLocation && pathname !== requestedLocation) {
|
|
setRequestedLocation(null);
|
|
return <Navigate to={requestedLocation} />;
|
|
}
|
|
|
|
return <>{children}</>;
|
|
}
|