useDataEntities.js 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. "use strict";
  2. var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
  3. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = void 0;
  8. var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
  9. var React = _interopRequireWildcard(require("react"));
  10. var _treeUtil = require("rc-tree/lib/utils/treeUtil");
  11. var _warning = _interopRequireDefault(require("rc-util/lib/warning"));
  12. var _valueUtil = require("../utils/valueUtil");
  13. var _default = exports.default = function _default(treeData, fieldNames) {
  14. return React.useMemo(function () {
  15. var collection = (0, _treeUtil.convertDataToEntities)(treeData, {
  16. fieldNames: fieldNames,
  17. initWrapper: function initWrapper(wrapper) {
  18. return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, wrapper), {}, {
  19. valueEntities: new Map()
  20. });
  21. },
  22. processEntity: function processEntity(entity, wrapper) {
  23. var val = entity.node[fieldNames.value];
  24. // Check if exist same value
  25. if (process.env.NODE_ENV !== 'production') {
  26. var key = entity.node.key;
  27. (0, _warning.default)(!(0, _valueUtil.isNil)(val), 'TreeNode `value` is invalidate: undefined');
  28. (0, _warning.default)(!wrapper.valueEntities.has(val), "Same `value` exist in the tree: ".concat(val));
  29. (0, _warning.default)(!key || String(key) === String(val), "`key` or `value` with TreeNode must be the same or you can remove one of them. key: ".concat(key, ", value: ").concat(val, "."));
  30. }
  31. wrapper.valueEntities.set(val, entity);
  32. }
  33. });
  34. return collection;
  35. }, [treeData, fieldNames]);
  36. };