43 lines
1.1 KiB
TypeScript
43 lines
1.1 KiB
TypeScript
|
"use client"
|
|||
|
|
|
|||
|
|
import { useEffect } from 'react'
|
|||
|
|
import { useRouter, usePathname } from 'next/navigation'
|
|||
|
|
import { useAuth } from '@/hooks/useAuth'
|
|||
|
|
|
|||
|
|
// 不需要登录的公共页面路径
|
|||
|
|
const PUBLIC_PATHS = [
|
|||
|
|
'/login',
|
|||
|
|
'/register',
|
|||
|
|
'/forgot-password',
|
|||
|
|
'/reset-password',
|
|||
|
|
'/404',
|
|||
|
|
'/500'
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
export function AuthCheck({ children }: { children: React.ReactNode }) {
|
|||
|
|
const router = useRouter()
|
|||
|
|
const pathname = usePathname()
|
|||
|
|
const { isAuthenticated, isLoading } = useAuth()
|
|||
|
|
|
|||
|
|
useEffect(() => {
|
|||
|
|
if (!isLoading && !isAuthenticated && !PUBLIC_PATHS.includes(pathname)) {
|
|||
|
|
// 保存当前URL，登录后可以重定向回来
|
|||
|
|
const returnUrl = encodeURIComponent(window.location.href)
|
|||
|
|
router.push(`/login?returnUrl=${returnUrl}`)
|
|||
|
|
}
|
|||
|
|
}, [isAuthenticated, isLoading, pathname, router])
|
|||
|
|
|
|||
|
|
if (isLoading) {
|
|||
|
|
return (
|
|||
|
|
<div className="flex h-screen w-screen items-center justify-center">
|
|||
|
|
<div className="animate-spin rounded-full h-8 w-8 border-t-2 border-b-2 border-blue-500"></div>
|
|||
|
|
</div>
|
|||
|
|
)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!isAuthenticated && !PUBLIC_PATHS.includes(pathname)) {
|
|||
|
|
return null
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return <>{children}</>
|
|||
|
|
}
|