45 lines
1.1 KiB
TypeScript
45 lines
1.1 KiB
TypeScript
'use client';
|
|
|
|
import { usePathname, useRouter } from 'next/navigation';
|
|
import { useEffect, useState } from 'react';
|
|
import { useAuthStore } from '@/stores/auth-store';
|
|
|
|
export function RequireAuth({ children }: { children: React.ReactNode }) {
|
|
const router = useRouter();
|
|
const pathname = usePathname();
|
|
const accessToken = useAuthStore((s) => s.accessToken);
|
|
const [mounted, setMounted] = useState(false);
|
|
|
|
useEffect(() => {
|
|
setMounted(true);
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
if (!mounted) {
|
|
return;
|
|
}
|
|
if (!accessToken) {
|
|
const from = pathname || '/dashboard';
|
|
router.replace(`/login?from=${encodeURIComponent(from)}`);
|
|
}
|
|
}, [mounted, accessToken, pathname, router]);
|
|
|
|
if (!mounted) {
|
|
return (
|
|
<div className="flex min-h-screen items-center justify-center bg-neutral-100 text-sm text-neutral-500">
|
|
加载中…
|
|
</div>
|
|
);
|
|
}
|
|
|
|
if (!accessToken) {
|
|
return (
|
|
<div className="flex min-h-screen items-center justify-center bg-neutral-100 text-sm text-neutral-500">
|
|
跳转登录…
|
|
</div>
|
|
);
|
|
}
|
|
|
|
return <>{children}</>;
|
|
}
|