Extend useMutation to support retries

useMutationWithRetry.ts

import {
  useMutation,
  MutationHookOptions,
  MutationFunctionOptions,
} from 'react-apollo'
import { DocumentNode, ExecutionResult } from 'graphql'

interface MutationOptions extends MutationHookOptions {
  retry: {
    max: number
  }
}

export const useMutationWithRetry = (
  mutation: DocumentNode,
  options: MutationOptions
): any => {
  const {
    retry: { max },
    ...fnOptions
  } = options

  const [_fnMutation, result] = useMutation(mutation, {
    ...fnOptions,
  })

  const fnMutation = (
    newOptions: MutationFunctionOptions,
    attemps = 0
  ): Promise<ExecutionResult<any>> => {
    return _fnMutation(newOptions).catch((error) => {
      if (attemps + 1 >= max) {
        throw error
      }

      return fnMutation(newOptions, attemps + 1)
    })
  }

  return [fnMutation, { ...result }]
}