packages/components/hooks/useAsync.js

   1 import { useCallback, useReducer, useRef } from 'react';
   2
   3 import useIsMounted from '@proton/hooks/useIsMounted';
   4
   5 const DEFAULT_STATE = {
   6     loading: false,
   7 };
   8
   9 const reducer = (state, action) => {
  10     switch (action.type) {
  11         case 'loading':
  12             return {
  13                 ...state,
  14                 loading: true,
  15             };
  16         case 'success':
  17             return {
  18                 error: undefined,
  19                 loading: false,
  20                 result: action.payload,
  21             };
  22         default:
  23         case 'error':
  24             return {
  25                 result: undefined,
  26                 loading: false,
  27                 error: action.payload,
  28             };
  29     }
  30 };
  31
  32 const useAsync = (setResults = true) => {
  33     const [{ loading, result, error }, dispatch] = useReducer(reducer, DEFAULT_STATE);
  34     const isMounted = useIsMounted();
  35     const promiseRef = useRef();
  36
  37     const run = useCallback(async (promise) => {
  38         const isCurrentPromise = () => promiseRef.current === promise;
  39         promiseRef.current = promise;
  40
  41         dispatch({ type: 'loading' });
  42         try {
  43             const data = await promise;
  44             if (isMounted() && isCurrentPromise()) {
  45                 dispatch({ type: 'success', payload: setResults ? data : undefined });
  46             }
  47             return data;
  48         } catch (e) {
  49             if (isMounted() && isCurrentPromise() && e.name !== 'AbortError') {
  50                 dispatch({ type: 'error', payload: setResults ? e : undefined });
  51             }
  52             throw e;
  53         }
  54     }, []);
  55
  56     return {
  57         result,
  58         error,
  59         loading,
  60         run,
  61     };
  62 };
  63
  64 export default useAsync;