use-refs-map.ts

import {
  createRef,
  useCallback,
  useRef,
  RefObject,
} from 'react';

type RefsMap<T> = Partial<Record<string, RefObject<T>>>;
type LinkRef<T> = (key: string) => RefObject<T>;

export const useRefsMap = (): [RefsMap<T>, LinkRef<T>] => {
  const refsMap = useRef<RefsMap<T>>({});
  
  const linkRef = useCallback(
    (key: string) => {
      if (!refsMap.current[key]) {
        refsMap.current[key] = createRef<T>();
      }
  
      return refsMap.current[key] as RefsMap<T>;
    },
    [],
  );
  
  return [refsMap.current, linkRef];
};