4dkankan-components/packages/utils/vue/refs.ts

import { isFunction } from '../types';

import type { ComponentPublicInstance, Ref } from 'vue';

export type RefSetter = (el: Element | ComponentPublicInstance | undefined) => void;

export const composeRefs = (...refs: (Ref<HTMLElement | undefined> | RefSetter)[]) => {
  return (el: Element | ComponentPublicInstance | null) => {
    refs.forEach((ref) => {
      if (isFunction(ref)) {
        ref(el as Element | ComponentPublicInstance);
      } else {
        ref.value = el as HTMLElement | undefined;
      }
    });
  };
};