packages/pass/hooks/useEnsureMounted.ts

   1 import { useCallback, useState } from 'react';
   2
   3 import useIsMounted from '@proton/hooks/useIsMounted';
   4 import type { Callback, Maybe } from '@proton/pass/types';
   5
   6 export const useEnsureMounted = () => {
   7     const isMounted = useIsMounted();
   8     return useCallback(
   9         <T extends Callback>(fn: T) =>
  10             ((...args: Parameters<T>): Maybe<ReturnType<T>> => {
  11                 if (isMounted()) return fn(...args);
  12             }) as T,
  13         []
  14     );
  15 };
  16
  17 export const useMountedState = <T>(initial: T) => {
  18     const [state, setState] = useState<T>(initial);
  19     const ensureMounted = useEnsureMounted();
  20     const setStateSafe = useCallback(ensureMounted(setState), []);
  21
  22     return [state, setStateSafe] as const;
  23 };