30 lines
871 B
TypeScript
30 lines
871 B
TypeScript
|
import { useIdentityAuthSession } from "@erp/identity/client";
|
||
|
|
import { LoadingOverlay } from "@repo/rdx-ui/components";
|
||
|
|
import type { ReactNode } from "react";
|
||
|
|
import { Navigate, useLocation } from "react-router-dom";
|
||
|
|
|
||
|
|
interface RequireCompanyRouteGuardProps {
|
||
|
|
children: ReactNode;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const RequireCompanyRouteGuard = ({ children }: RequireCompanyRouteGuardProps) => {
|
||
|
|
const location = useLocation();
|
||
|
|
const { activeCompany, availableCompanies, isLoading } = useIdentityAuthSession();
|
||
|
|
|
||
|
|
if (isLoading) {
|
||
|
|
return <LoadingOverlay />;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (activeCompany) {
|
||
|
|
return <>{children}</>;
|
||
|
|
}
|
||
|
|
|
||
|
|
const returnTo = `${location.pathname}${location.search}`;
|
||
|
|
|
||
|
|
if (availableCompanies.length === 0) {
|
||
|
|
return <Navigate replace to="/no-companies" />;
|
||
|
|
}
|
||
|
|
|
||
|
|
return <Navigate replace to={`/company-selection?returnTo=${encodeURIComponent(returnTo)}`} />;
|
||
|
|
};
|