| import { getLoginUrl } from "@/const"; |
| import { trpc } from "@/lib/trpc"; |
| import { TRPCClientError } from "@trpc/client"; |
| import { useCallback, useEffect, useMemo } from "react"; |
|
|
| type UseAuthOptions = { |
| redirectOnUnauthenticated?: boolean; |
| redirectPath?: string; |
| }; |
|
|
| export function useAuth(options?: UseAuthOptions) { |
| const { redirectOnUnauthenticated = false, redirectPath = getLoginUrl() } = |
| options ?? {}; |
| const utils = trpc.useUtils(); |
|
|
| const meQuery = trpc.auth.me.useQuery(undefined, { |
| retry: false, |
| refetchOnWindowFocus: false, |
| }); |
|
|
| const logoutMutation = trpc.auth.logout.useMutation({ |
| onSuccess: () => { |
| utils.auth.me.setData(undefined, null); |
| }, |
| }); |
|
|
| const logout = useCallback(async () => { |
| try { |
| await logoutMutation.mutateAsync(); |
| } catch (error: unknown) { |
| if ( |
| error instanceof TRPCClientError && |
| error.data?.code === "UNAUTHORIZED" |
| ) { |
| return; |
| } |
| throw error; |
| } finally { |
| utils.auth.me.setData(undefined, null); |
| await utils.auth.me.invalidate(); |
| } |
| }, [logoutMutation, utils]); |
|
|
| const state = useMemo(() => { |
| |
| |
| return { |
| user: meQuery.data ?? null, |
| loading: meQuery.isLoading || logoutMutation.isPending, |
| error: meQuery.error ?? logoutMutation.error ?? null, |
| |
| isAuthenticated: true, |
| }; |
| }, [ |
| meQuery.data, |
| meQuery.error, |
| meQuery.isLoading, |
| logoutMutation.error, |
| logoutMutation.isPending, |
| ]); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| return { |
| ...state, |
| refresh: () => meQuery.refetch(), |
| logout, |
| }; |
| } |
|
|
