import { ForwardedRef, useEffect, useRef } from "react";

export default <T>(ref?: ForwardedRef<T>) => {
    const innerRef = useRef<T>(null);

    useEffect(() => {
        if (!ref) {
            return;
        }

        if (typeof ref === "function") {
            ref(innerRef.current);
        } else {
            ref.current = innerRef.current;
        }
    });

    return innerRef;
};