All files / src/components/guards AuthGuard.tsx

100% Statements 21/21
100% Branches 22/22
100% Functions 3/3
100% Lines 21/21

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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                        15x 5x       10x                 10x           36x 36x 36x 36x   36x 36x 4x     36x 15x 15x                     36x 4x     32x 4x       28x 15x     13x    
import { useEffect } from "react";
import { useNavigate, useRouterState } from "@tanstack/react-router";
import { useAuth } from "../../lib/auth-context";
import { LoadingSpinner } from "./LoadingSpinner";
 
type AuthGuardProps = {
	children: React.ReactNode;
};
 
// Map pro routes to their admin equivalents
function getAdminEquivalentRoute(proPath: string): string {
	// Handle project detail pages: /projects/:id -> /admin/projects/:id
	if (proPath.startsWith("/projects/")) {
		return `/admin${proPath}`;
	}
 
	// Handle other pro routes
	const routeMap: Record<string, string> = {
		"/": "/admin",
		"/projects": "/admin/projects",
 
		"/analytics": "/admin/analytics",
		"/profile": "/admin/pros",
		"/team": "/admin/users",
	};
 
	return routeMap[proPath] || "/admin";
}
 
// Auth guard component - redirects to login if not authenticated
// Also redirects admin-only users (no pro access) to equivalent admin routes
export function AuthGuard({ children }: AuthGuardProps) {
	const { isAuthenticated, isLoading, isAdmin, hasProAccess } = useAuth();
	const navigate = useNavigate();
	const routerState = useRouterState();
	const currentPath = routerState.location.pathname;
 
	useEffect(() => {
		if (!isLoading && !isAuthenticated) {
			navigate({ to: "/login" });
		}
		// Redirect admin-only users (no pro access) to equivalent admin route
		if (!isLoading && isAuthenticated && isAdmin && !hasProAccess) {
			const adminRoute = getAdminEquivalentRoute(currentPath);
			navigate({ to: adminRoute });
		}
	}, [
		isLoading,
		isAuthenticated,
		isAdmin,
		hasProAccess,
		currentPath,
		navigate,
	]);
 
	if (isLoading) {
		return <LoadingSpinner />;
	}
 
	if (!isAuthenticated) {
		return <LoadingSpinner />; // Show spinner while redirecting
	}
 
	// Admin-only users should be redirected to admin equivalent
	if (isAdmin && !hasProAccess) {
		return <LoadingSpinner />; // Show spinner while redirecting
	}
 
	return <>{children}</>;
}