import { useState } from 'react';

const useLoading = (initialState: boolean = false) => {
  const [isLoading, setIsLoading] = useState<boolean>(initialState);

  const setPromise = <T>(promise: Promise<T>) => {
    setIsLoading(true);

    promise
      .then(() => {
        setIsLoading(false);
      })
      .catch(() => {
        setIsLoading(false);
      });

    return promise;
  };

  return [isLoading, setPromise] as const;
};

export { useLoading };