OpenFlare/frontend/contexts/user-context.tsx

"use client"

import {createContext, ReactNode, useCallback, useContext, useEffect, useRef, useState} from 'react'

import services from '@/lib/services'
import {User} from '@/lib/services/auth/types'


/** 用户状态接口 */
interface UserState {
  user: User | null
  loading: boolean
  error: string | null
}

/** 用户上下文接口 */
interface UserContextValue extends UserState {
  setUser: (user: User) => void
  refetch: () => Promise<void>
  logout: () => Promise<void>
}



/** 用户上下文 */
const UserContext = createContext<UserContextValue | undefined>(undefined)

/**
 * 用户Provider组件
 *
 * @param {React.ReactNode} children - 用户 Provider 的子元素
 * @returns {React.ReactNode} 用户 Provider 组件
 * @example
 * ```tsx
 * <UserProvider>
 *   <UserContext.Provider value={{ user, loading, error, refetch, updatePayKey, getTrustLevelLabel, getPayLevelLabel, logout }}>
 *     {children}
 *   </UserContext.Provider>
 * </UserProvider>
 * ```
 */
export function UserProvider({ children }: { children: ReactNode }) {
  const [state, setState] = useState<UserState>({
    user: null,
    loading: true,
    error: null,
  })

  const isMountedRef = useRef(true)



  /** 获取用户信息 */
  const fetchUser = useCallback(async () => {
    try {
      setState(prev => ({ ...prev, loading: true, error: null }))
      const user = await services.auth.getUserInfo()

      if (!isMountedRef.current) return

      setState({ user, loading: false, error: null })
    } catch (error) {
      if (!isMountedRef.current) return

      setState({
        user: null,
        loading: false,
        error: error instanceof Error ? error.message : '获取用户信息失败',
      })
    }
  }, [])

  /** 重新获取用户信息 */
  const refetch = useCallback(async () => {
    await fetchUser()
  }, [fetchUser])

  /** 直接设置用户信息（登录/注册后免二次请求） */
  const setUser = useCallback((user: User) => {
    setState({ user, loading: false, error: null })
  }, [])

  /** 用户登出 */
  const logout = useCallback(async () => {
    try {
      await services.auth.logout()

      if (!isMountedRef.current) return

      setState({ user: null, loading: false, error: null })
      window.location.href = '/login'
    } catch (error) {
      if (!isMountedRef.current) {
        throw error
      }

      const errorMessage = error instanceof Error ? error.message : '登出失败'
      setState(prev => ({
        ...prev,
        error: errorMessage,
      }))
      throw new Error(errorMessage)
    }
  }, [])

  /** 组件挂载时获取用户信息 */
  useEffect(() => {
    isMountedRef.current = true
    fetchUser()

    return () => {
      isMountedRef.current = false
    }
  }, [fetchUser])

  return (
    <UserContext.Provider
      value={{
        ...state,
        setUser,
        refetch,
        logout,
      }}
    >
      {children}
    </UserContext.Provider>
  )
}

/**
 * 使用用户上下文的Hook
 *
 * @returns {UserContextValue} 用户上下文值
 * @example
 * ```tsx
 * const { user, loading, error, refetch, updatePayKey, getTrustLevelLabel, getPayLevelLabel, logout } = useUser()
 * ```
 */
export function useUser(): UserContextValue {
  const context = useContext(UserContext)
  if (context === undefined) {
    throw new Error('useUser must be used within a UserProvider')
  }
  return context
}