useAnimationDelay.ts

export function useAnimationDelay(
    delay: number,
    shouldShow: boolean,
    onHide: () => void = () => {
        return
    },
): boolean {
    const [display, setDisplay] = useState(shouldShow)

    useEffect(() => {
        if (shouldShow) {
            setDisplay(true)
            return _.identity
        }
        // Wait for the animation to end before
        // removing the element.
        let tm: number
        if (display) {
            tm = window.setTimeout(() => {
                setDisplay(false)
                onHide()
            }, delay)
        }
        return (): void => {
            clearTimeout(tm)
        }
    }, [delay, display, shouldShow, onHide])

    return display
}