import { useEffect, useRef } from 'react'

export const useTimeout = (callback: Function, delay: number | null) => {
  const timeoutRef = useRef<any>(null)
  const savedCallback = useRef<Function>()

  // Remember the latest callback
  useEffect(() => {
    savedCallback.current = callback
  }, [callback])

  // Set up the timeout
  useEffect(() => {
    function tick() {
      savedCallback.current?.()
    }

    if (delay !== null) {
      timeoutRef.current = setTimeout(tick, delay)
      return () => clearTimeout(timeoutRef.current)
    }
  }, [delay])
}