baixingwenzheng-pc/node_modules/naive-ui/es/_utils/composable/use-collection.mjs

import { getCurrentInstance, inject, onBeforeUnmount, onMounted, watch } from 'vue';
// injection.collection {
//   key1: [insta, instb]
//   key2: [instc]
// }
export function useInjectionInstanceCollection(injectionName, collectionKey, registerKeyRef) {
  var _a;
  const injection = inject(injectionName, null);
  if (injection === null) return;
  const vm = (_a = getCurrentInstance()) === null || _a === void 0 ? void 0 : _a.proxy;
  watch(registerKeyRef, registerInstance);
  registerInstance(registerKeyRef.value);
  onBeforeUnmount(() => {
    registerInstance(undefined, registerKeyRef.value);
  });
  function registerInstance(key, oldKey) {
    if (!injection) return;
    const collection = injection[collectionKey];
    if (oldKey !== undefined) removeInstance(collection, oldKey);
    if (key !== undefined) addInstance(collection, key);
  }
  function removeInstance(collection, key) {
    if (!collection[key]) collection[key] = [];
    collection[key].splice(collection[key].findIndex(instance => instance === vm), 1);
  }
  function addInstance(collection, key) {
    if (!collection[key]) collection[key] = [];
    if (!~collection[key].findIndex(instance => instance === vm)) {
      collection[key].push(vm);
    }
  }
}
// injection.collection {
//   key1: [insta.value, instb.value]
//   key2: [instc.value]
// }
export function useInjectionCollection(injectionName, collectionKey, valueRef) {
  const injection = inject(injectionName, null);
  if (injection === null) return;
  if (!(collectionKey in injection)) {
    injection[collectionKey] = [];
  }
  injection[collectionKey].push(valueRef.value);
  watch(valueRef, (value, prevValue) => {
    const collectionArray = injection[collectionKey];
    const index = collectionArray.findIndex(collectionValue => collectionValue === prevValue);
    if (~index) collectionArray.splice(index, 1);
    collectionArray.push(value);
  });
  onBeforeUnmount(() => {
    const collectionArray = injection[collectionKey];
    const index = collectionArray.findIndex(collectionValue => collectionValue === valueRef.value);
    if (~index) collectionArray.splice(index, 1);
  });
}
// injection.collection {
//   key1: [insta.$el, instb.$el]
//   key2: [instc.$el]
// }
export function useInjectionElementCollection(injectionName, collectionKey, getElement) {
  const injection = inject(injectionName, null);
  if (injection === null) return;
  if (!(collectionKey in injection)) {
    injection[collectionKey] = [];
  }
  onMounted(() => {
    const el = getElement();
    if (!el) return;
    injection[collectionKey].push(el);
  });
  onBeforeUnmount(() => {
    const collectionArray = injection[collectionKey];
    const element = getElement();
    const index = collectionArray.findIndex(collectionElement => collectionElement === element);
    if (~index) collectionArray.splice(index, 1);
  });
}