27 lines
809 B
TypeScript
27 lines
809 B
TypeScript
"use client"
|
||
|
||
import { useEffect } from "react"
|
||
import { useRouter, usePathname } from "next/navigation"
|
||
|
||
export function AuthCheck({ children }: { children: React.ReactNode }) {
|
||
const router = useRouter()
|
||
const pathname = usePathname()
|
||
|
||
useEffect(() => {
|
||
// 排除不需要登录的页面
|
||
const publicPaths = ['/login', '/register', '/forgot-password']
|
||
if (publicPaths.includes(pathname)) {
|
||
return
|
||
}
|
||
|
||
const token = localStorage.getItem('token')
|
||
if (!token) {
|
||
// 如果没有token，重定向到登录页面，并携带当前页面URL作为回调
|
||
const currentPath = window.location.pathname + window.location.search
|
||
router.push(`/login?redirect=${encodeURIComponent(currentPath)}`)
|
||
return
|
||
}
|
||
}, [router, pathname])
|
||
|
||
return <>{children}</>
|
||
}
|