import { ForwardedRef, useEffect, useRef } from 'react'

export const useForwardedRef = <T,>(ref: ForwardedRef<T>, initialValue = null) => {
  const targetRef = useRef<T>(initialValue)

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

    if (typeof ref === 'function') {
      ref(targetRef.current)
    } else {
      ref.current = targetRef.current
    }
  }, [ref])

  return targetRef
}