TypeScript implementation of react `ref` merge for `HTMLElement`

useRefMerge.ts

export function useRefMerge<T>(...refs: (RefCallback<T> | MutableRefObject<T | null> | ForwardedRef<T>)[]) {
  const assignedObject = useRef<T>()

  return useCallback((node: T) => {
    assignedObject.current = node

    for (const ref of refs) {
      if (typeof ref === 'function') {
        ref(node)
      } else if (typeof ref === 'object' && ref) {
        ref.current = node
      }
    }
  }, refs)
}