react query mutation

gistfile1.txt

import { apiClient, REACT_QUERY_KEYS } from '@/config'
import { getErrorMessage } from '@/utils/utils'
import { useMutation, useQueryClient } from '@tanstack/react-query'
import { useRouter } from 'next/router'
import { toast } from 'react-toastify'

export const useDeleteDevice = ({
  isOnDeviceListPage,
  callbackOnSuccess,
}: {
  isOnDeviceListPage: boolean
  callbackOnSuccess: Function
}) => {
  const queryClient = useQueryClient()
  const router = useRouter()

  const mutation = useMutation(
    async (id: string) => {
      const { data } = await apiClient.delete(`/device/${id}`)
      return data.success
    },
    {
      onSuccess: () => {
        toast.success('Device deleted successfully')
        if (isOnDeviceListPage) {
          queryClient.invalidateQueries([REACT_QUERY_KEYS.LIST_DEVICE])
        } else {
          router.push('/device')
        }
        callbackOnSuccess()
      },
      onError: (error: any) => {
        toast.error(getErrorMessage(error))
      },
    }
  )

  return mutation
}