useRefsWithKeys.ts

import { useRef } from "react"

const useRefsWithKeys = <T>(): {
  setRefForKey: (key: string) => (element: T) => void
  getRefForKey: (key: string) => T
} => {
  const ref = useRef<Record<string, T>>({})

  const setRefForKey =
    (key: string) =>
    (element: T): void => {
      ref.current[key] = element
    }

  const getRefForKey = (key: string): T => {
    return ref.current[key]
  }

  return {
    setRefForKey,
    getRefForKey,
  }
}

export default useRefsWithKeys