useRetry.ts

import { useState } from 'react';

export type RetryCallback<T> = (...args: any[]) => void | T | Promise<T>;

function useRetry<T, E = any>(callback: RetryCallback<T>) {
  const [error, setError] = useState<E>();
  const [retryCount, setRetryCount] = useState(0);

  function watchRetry(callback: RetryCallback<T>) {
    return async function (...args: any[]) {
      setError(undefined);

      try {
        return await callback(...args);
      } catch (error) {
        setError(error as E);
        setRetryCount((prev) => prev + 1);

        throw error;
      }
    };
  }

  return { execute: watchRetry(callback), retryCount, error };
}

export default useRetry;