91 lines
2.3 KiB
TypeScript
91 lines
2.3 KiB
TypeScript
|
import { create } from 'zustand'
|
||
|
|
import { persist } from 'zustand/middleware'
|
||
|
|
import type { User } from '@/types'
|
||
|
|
import { authApi, type RegisterData, type LoginData } from '@/api/auth'
|
||
|
|
|
||
|
|
interface AuthState {
|
||
|
|
user: User | null
|
||
|
|
token: string | null
|
||
|
|
isAuthenticated: boolean
|
||
|
|
isLoading: boolean
|
||
|
|
error: string | null
|
||
|
|
|
||
|
|
login: (data: LoginData) => Promise<void>
|
||
|
|
register: (data: RegisterData) => Promise<void>
|
||
|
|
logout: () => void
|
||
|
|
clearError: () => void
|
||
|
|
}
|
||
|
|
|
||
|
|
export const useAuthStore = create<AuthState>()(
|
||
|
|
persist(
|
||
|
|
(set) => ({
|
||
|
|
user: null,
|
||
|
|
token: null,
|
||
|
|
isAuthenticated: false,
|
||
|
|
isLoading: false,
|
||
|
|
error: null,
|
||
|
|
|
||
|
|
login: async (data) => {
|
||
|
|
set({ isLoading: true, error: null })
|
||
|
|
try {
|
||
|
|
const response = await authApi.login(data)
|
||
|
|
localStorage.setItem('token', response.access_token)
|
||
|
|
set({
|
||
|
|
user: response.user,
|
||
|
|
token: response.access_token,
|
||
|
|
isAuthenticated: true,
|
||
|
|
isLoading: false,
|
||
|
|
})
|
||
|
|
} catch (err: unknown) {
|
||
|
|
const error = err as { response?: { data?: { detail?: string } } }
|
||
|
|
set({
|
||
|
|
error: error.response?.data?.detail || 'Login failed',
|
||
|
|
isLoading: false,
|
||
|
|
})
|
||
|
|
throw err
|
||
|
|
}
|
||
|
|
},
|
||
|
|
|
||
|
|
register: async (data) => {
|
||
|
|
set({ isLoading: true, error: null })
|
||
|
|
try {
|
||
|
|
const response = await authApi.register(data)
|
||
|
|
localStorage.setItem('token', response.access_token)
|
||
|
|
set({
|
||
|
|
user: response.user,
|
||
|
|
token: response.access_token,
|
||
|
|
isAuthenticated: true,
|
||
|
|
isLoading: false,
|
||
|
|
})
|
||
|
|
} catch (err: unknown) {
|
||
|
|
const error = err as { response?: { data?: { detail?: string } } }
|
||
|
|
set({
|
||
|
|
error: error.response?.data?.detail || 'Registration failed',
|
||
|
|
isLoading: false,
|
||
|
|
})
|
||
|
|
throw err
|
||
|
|
}
|
||
|
|
},
|
||
|
|
|
||
|
|
logout: () => {
|
||
|
|
localStorage.removeItem('token')
|
||
|
|
set({
|
||
|
|
user: null,
|
||
|
|
token: null,
|
||
|
|
isAuthenticated: false,
|
||
|
|
})
|
||
|
|
},
|
||
|
|
|
||
|
|
clearError: () => set({ error: null }),
|
||
|
|
}),
|
||
|
|
{
|
||
|
|
name: 'auth-storage',
|
||
|
|
partialize: (state) => ({
|
||
|
|
user: state.user,
|
||
|
|
token: state.token,
|
||
|
|
isAuthenticated: state.isAuthenticated,
|
||
|
|
}),
|
||
|
|
}
|
||
|
|
)
|
||
|
|
)
|