71 lines
1.8 KiB
TypeScript
71 lines
1.8 KiB
TypeScript
|
import { createContext, useContext, useEffect, useState, useCallback, type ReactNode } from 'react'
|
||
|
|
import { api } from './api'
|
||
|
|
import type { User } from './types'
|
||
|
|
|
||
|
|
export type AuthStatus = 'loading' | 'setup' | 'unauthenticated' | 'authenticated'
|
||
|
|
|
||
|
|
interface AuthContextValue {
|
||
|
|
status: AuthStatus
|
||
|
|
user: User | null
|
||
|
|
setupNeeded: boolean
|
||
|
|
refresh: () => Promise<void>
|
||
|
|
onSetupComplete: (user: User) => void
|
||
|
|
onLogin: (user: User) => void
|
||
|
|
onLogout: () => void
|
||
|
|
}
|
||
|
|
|
||
|
|
const AuthContext = createContext<AuthContextValue | null>(null)
|
||
|
|
|
||
|
|
export function AuthProvider({ children }: { children: ReactNode }) {
|
||
|
|
const [status, setStatus] = useState<AuthStatus>('loading')
|
||
|
|
const [user, setUser] = useState<User | null>(null)
|
||
|
|
const [setupNeeded, setSetupNeeded] = useState(false)
|
||
|
|
|
||
|
|
const refresh = useCallback(async () => {
|
||
|
|
try {
|
||
|
|
const { needed } = await api.setupStatus()
|
||
|
|
if (needed) {
|
||
|
|
setSetupNeeded(true)
|
||
|
|
setStatus('setup')
|
||
|
|
return
|
||
|
|
}
|
||
|
|
setSetupNeeded(false)
|
||
|
|
const { user: me } = await api.me()
|
||
|
|
setUser(me)
|
||
|
|
setStatus('authenticated')
|
||
|
|
} catch {
|
||
|
|
setStatus('unauthenticated')
|
||
|
|
}
|
||
|
|
}, [])
|
||
|
|
|
||
|
|
useEffect(() => {
|
||
|
|
void refresh()
|
||
|
|
}, [refresh])
|
||
|
|
|
||
|
|
const onSetupComplete = useCallback((u: User) => {
|
||
|
|
setSetupNeeded(false)
|
||
|
|
setUser(u)
|
||
|
|
setStatus('authenticated')
|
||
|
|
}, [])
|
||
|
|
const onLogin = useCallback((u: User) => {
|
||
|
|
setUser(u)
|
||
|
|
setStatus('authenticated')
|
||
|
|
}, [])
|
||
|
|
const onLogout = useCallback(() => {
|
||
|
|
setUser(null)
|
||
|
|
setStatus('unauthenticated')
|
||
|
|
}, [])
|
||
|
|
|
||
|
|
return (
|
||
|
|
<AuthContext.Provider value={{ status, user, setupNeeded, refresh, onSetupComplete, onLogin, onLogout }}>
|
||
|
|
{children}
|
||
|
|
</AuthContext.Provider>
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
export function useAuth(): AuthContextValue {
|
||
|
|
const ctx = useContext(AuthContext)
|
||
|
|
if (!ctx) throw new Error('useAuth outside AuthProvider')
|
||
|
|
return ctx
|
||
|
|
}
|