useDataEntities.js 1.4 KB

1234567891011121314151617181920212223242526272829303132333435
  1. import _extends from "@babel/runtime/helpers/esm/extends";
  2. import { convertDataToEntities } from '../../vc-tree/utils/treeUtil';
  3. import { isNil } from '../utils/valueUtil';
  4. import { shallowRef, watchEffect } from 'vue';
  5. import { warning } from '../../vc-util/warning';
  6. export default ((treeData, fieldNames) => {
  7. const valueEntities = shallowRef(new Map());
  8. const keyEntities = shallowRef({});
  9. watchEffect(() => {
  10. const fieldNamesValue = fieldNames.value;
  11. const collection = convertDataToEntities(treeData.value, {
  12. fieldNames: fieldNamesValue,
  13. initWrapper: wrapper => _extends(_extends({}, wrapper), {
  14. valueEntities: new Map()
  15. }),
  16. processEntity: (entity, wrapper) => {
  17. const val = entity.node[fieldNamesValue.value];
  18. // Check if exist same value
  19. if (process.env.NODE_ENV !== 'production') {
  20. const key = entity.node.key;
  21. warning(!isNil(val), 'TreeNode `value` is invalidate: undefined');
  22. warning(!wrapper.valueEntities.has(val), `Same \`value\` exist in the tree: ${val}`);
  23. warning(!key || String(key) === String(val), `\`key\` or \`value\` with TreeNode must be the same or you can remove one of them. key: ${key}, value: ${val}.`);
  24. }
  25. wrapper.valueEntities.set(val, entity);
  26. }
  27. });
  28. valueEntities.value = collection.valueEntities;
  29. keyEntities.value = collection.keyEntities;
  30. });
  31. return {
  32. valueEntities,
  33. keyEntities
  34. };
  35. });