import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query" import { useNavigate } from "@tanstack/react-router" import { useState } from "react" import { type Body_login_login_access_token as AccessToken, type ApiError, LoginService, type UserPublic, type UserRegister, UsersService, } from "../client" import { handleError } from "../utils" const isLoggedIn = () => { return localStorage.getItem("access_token") !== null } const useAuth = () => { const [error, setError] = useState(null) const navigate = useNavigate() const queryClient = useQueryClient() const { data: user } = useQuery({ queryKey: ["currentUser"], queryFn: UsersService.readUserMe, enabled: isLoggedIn(), }) const signUpMutation = useMutation({ mutationFn: (data: UserRegister) => UsersService.registerUser({ requestBody: data }), onSuccess: () => { navigate({ to: "/login" }) }, onError: (err: ApiError) => { handleError(err) }, onSettled: () => { queryClient.invalidateQueries({ queryKey: ["users"] }) }, }) const login = async (data: AccessToken) => { const response = await LoginService.loginAccessToken({ formData: data, }) localStorage.setItem("access_token", response.access_token) } const loginMutation = useMutation({ mutationFn: login, onSuccess: () => { navigate({ to: "/" }) }, onError: (err: ApiError) => { handleError(err) }, }) const logout = () => { localStorage.removeItem("access_token") navigate({ to: "/login" }) } return { signUpMutation, loginMutation, logout, user, error, resetError: () => setError(null), } } export { isLoggedIn } export default useAuth