fix: add overloads to useIsMutating (TanStack#2451)

Author: TkDodo

src/core/utils.ts

@@ -166,6 +166,13 @@ export function parseFilterArgs<
     : [arg1 || {}, arg2]) as [TFilters, TOptions]
 }
 
+export function parseMutationFilterArgs(
+  arg1?: QueryKey | MutationFilters,
+  arg2?: MutationFilters
+): MutationFilters | undefined {
+  return isQueryKey(arg1) ? { ...arg2, mutationKey: arg1 } : arg1
+}
+
 export function mapQueryStatusFilter(
   active?: boolean,
   inactive?: boolean

src/react/tests/useIsMutating.test.tsx

@@ -45,7 +45,7 @@ describe('useIsMutating', () => {
     const queryClient = new QueryClient()
 
     function IsMutating() {
-      const isMutating = useIsMutating({ mutationKey: 'mutation1' })
+      const isMutating = useIsMutating('mutation1')
       isMutatings.push(isMutating)
       return null
     }

src/react/useIsMutating.ts

@@ -1,11 +1,21 @@
 import React from 'react'
 
 import { notifyManager } from '../core/notifyManager'
-import { MutationFilters } from '../core/utils'
+import { QueryKey } from '../core/types'
+import { MutationFilters, parseMutationFilterArgs } from '../core/utils'
 import { useQueryClient } from './QueryClientProvider'
 
-export function useIsMutating(filters?: MutationFilters): number {
+export function useIsMutating(filters?: MutationFilters): number
+export function useIsMutating(
+  queryKey?: QueryKey,
+  filters?: MutationFilters
+): number
+export function useIsMutating(
+  arg1?: QueryKey | MutationFilters,
+  arg2?: MutationFilters
+): number {
   const mountedRef = React.useRef(false)
+  const filters = parseMutationFilterArgs(arg1, arg2)
 
   const queryClient = useQueryClient()