framer-motion/src/utils/use-force-update.ts

An open source and production-ready motion library for React on the web

import { useState, useCallback, useRef } from "react"
import { useUnmountEffect } from "./use-unmount-effect"

export function useForceUpdate() {
    const unloadingRef = useRef(false)
    const [forcedRenderCount, setForcedRenderCount] = useState(0)

    useUnmountEffect(() => (unloadingRef.current = true))

    return useCallback(() => {
        !unloadingRef.current && setForcedRenderCount(forcedRenderCount + 1)
    }, [forcedRenderCount])
}