import { useEffect, useRef } from 'react';

export function useUnmountEffect(callback: () => void) {
  const callbackRef = useRef(callback);
  callbackRef.current = callback;

  useEffect(() => {
    return () => {
      const callbackFunction = callbackRef.current;

      if (!callbackFunction) {
        return;
      }

      callbackFunction();
    };
  }, []);
}