Spaces:
Paused
Paused
File size: 2,133 Bytes
99a7ebb | 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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | "use client";
import { useEffect, useState } from "react";
import { useRouter } from "next/navigation";
import { getValidatedAuthSession } from "@/lib/auth-session";
import {
getDefaultRouteForRole,
type AuthRole,
type StoredAuthSession,
} from "@/store/auth";
type UseAuthGuardResult = {
isCheckingAuth: boolean;
session: StoredAuthSession | null;
};
export function useAuthGuard(allowedRoles?: AuthRole[]): UseAuthGuardResult {
const router = useRouter();
const [session, setSession] = useState<StoredAuthSession | null>(null);
const [isCheckingAuth, setIsCheckingAuth] = useState(true);
const allowedRolesKey = (allowedRoles || []).join(",");
useEffect(() => {
let active = true;
const load = async () => {
const roleList = allowedRolesKey ? (allowedRolesKey.split(",") as AuthRole[]) : [];
const storedSession = await getValidatedAuthSession();
if (!active) {
return;
}
if (!storedSession) {
setSession(null);
setIsCheckingAuth(false);
router.replace("/login");
return;
}
if (roleList.length > 0 && !roleList.includes(storedSession.role)) {
setSession(storedSession);
setIsCheckingAuth(false);
router.replace(getDefaultRouteForRole(storedSession.role));
return;
}
setSession(storedSession);
setIsCheckingAuth(false);
};
void load();
return () => {
active = false;
};
}, [allowedRolesKey, router]);
return { isCheckingAuth, session };
}
export function useRedirectIfAuthenticated() {
const router = useRouter();
const [isCheckingAuth, setIsCheckingAuth] = useState(true);
useEffect(() => {
let active = true;
const load = async () => {
const storedSession = await getValidatedAuthSession();
if (!active) {
return;
}
if (storedSession) {
router.replace(getDefaultRouteForRole(storedSession.role));
return;
}
setIsCheckingAuth(false);
};
void load();
return () => {
active = false;
};
}, [router]);
return { isCheckingAuth };
}
|