105 lines
2.7 KiB
TypeScript
105 lines
2.7 KiB
TypeScript
import {
|
|
QueryKey,
|
|
UseMutationOptions,
|
|
UseQueryOptions,
|
|
useMutation,
|
|
useQuery,
|
|
} from "@tanstack/react-query"
|
|
import { client } from "../../lib/client"
|
|
import { queryClient } from "../../lib/medusa"
|
|
import { queryKeysFactory } from "../../lib/query-key-factory"
|
|
import { UpdateUserReq } from "../../types/api-payloads"
|
|
import { UserDeleteRes, UserListRes, UserRes } from "../../types/api-responses"
|
|
|
|
const USERS_QUERY_KEY = "users" as const
|
|
const usersQueryKeys = {
|
|
...queryKeysFactory(USERS_QUERY_KEY),
|
|
me: () => [USERS_QUERY_KEY, "me"],
|
|
}
|
|
|
|
export const useMe = (
|
|
options?: UseQueryOptions<UserRes, Error, UserRes, QueryKey>
|
|
) => {
|
|
const { data, ...rest } = useQuery({
|
|
queryFn: () => client.users.me(),
|
|
queryKey: usersQueryKeys.me(),
|
|
...options,
|
|
})
|
|
|
|
return {
|
|
...data,
|
|
...rest,
|
|
}
|
|
}
|
|
|
|
export const useUser = (
|
|
id: string,
|
|
query?: Record<string, any>,
|
|
options?: Omit<
|
|
UseQueryOptions<UserRes, Error, UserRes, QueryKey>,
|
|
"queryFn" | "queryKey"
|
|
>
|
|
) => {
|
|
const { data, ...rest } = useQuery({
|
|
queryFn: () => client.users.retrieve(id, query),
|
|
queryKey: usersQueryKeys.detail(id),
|
|
...options,
|
|
})
|
|
|
|
return { ...data, ...rest }
|
|
}
|
|
|
|
export const useUsers = (
|
|
query?: Record<string, any>,
|
|
options?: Omit<
|
|
UseQueryOptions<UserListRes, Error, UserListRes, QueryKey>,
|
|
"queryFn" | "queryKey"
|
|
>
|
|
) => {
|
|
const { data, ...rest } = useQuery({
|
|
queryFn: () => client.users.list(query),
|
|
queryKey: usersQueryKeys.list(query),
|
|
...options,
|
|
})
|
|
|
|
return { ...data, ...rest }
|
|
}
|
|
|
|
export const useUpdateUser = (
|
|
id: string,
|
|
options?: UseMutationOptions<UserRes, Error, UpdateUserReq>
|
|
) => {
|
|
return useMutation({
|
|
mutationFn: (payload) => client.users.update(id, payload),
|
|
onSuccess: (data, variables, context) => {
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.detail(id) })
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.lists() })
|
|
|
|
// We invalidate the me query in case the user updates their own profile
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.me() })
|
|
|
|
options?.onSuccess?.(data, variables, context)
|
|
},
|
|
...options,
|
|
})
|
|
}
|
|
|
|
export const useDeleteUser = (
|
|
id: string,
|
|
options?: UseMutationOptions<UserDeleteRes, Error, void>
|
|
) => {
|
|
return useMutation({
|
|
mutationFn: () => client.users.delete(id),
|
|
onSuccess: (data, variables, context) => {
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.detail(id) })
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.lists() })
|
|
|
|
// We invalidate the me query in case the user updates their own profile
|
|
queryClient.invalidateQueries({ queryKey: usersQueryKeys.me() })
|
|
|
|
options?.onSuccess?.(data, variables, context)
|
|
},
|
|
...options,
|
|
})
|
|
}
|