util.js 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. var _toConsumableArray2 = require('babel-runtime/helpers/toConsumableArray');
  6. var _toConsumableArray3 = _interopRequireDefault(_toConsumableArray2);
  7. var _extends2 = require('babel-runtime/helpers/extends');
  8. var _extends3 = _interopRequireDefault(_extends2);
  9. exports.flatArray = flatArray;
  10. exports.treeMap = treeMap;
  11. exports.flatFilter = flatFilter;
  12. exports.generateValueMaps = generateValueMaps;
  13. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { 'default': obj }; }
  14. function flatArray() {
  15. var data = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
  16. var childrenName = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'children';
  17. var result = [];
  18. var loop = function loop(array) {
  19. array.forEach(function (item) {
  20. if (item[childrenName]) {
  21. var newItem = (0, _extends3['default'])({}, item);
  22. delete newItem[childrenName];
  23. result.push(newItem);
  24. if (item[childrenName].length > 0) {
  25. loop(item[childrenName]);
  26. }
  27. } else {
  28. result.push(item);
  29. }
  30. });
  31. };
  32. loop(data);
  33. return result;
  34. }
  35. function treeMap(tree, mapper) {
  36. var childrenName = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 'children';
  37. return tree.map(function (node, index) {
  38. var extra = {};
  39. if (node[childrenName]) {
  40. extra[childrenName] = treeMap(node[childrenName], mapper, childrenName);
  41. }
  42. return (0, _extends3['default'])({}, mapper(node, index), extra);
  43. });
  44. }
  45. function flatFilter(tree, callback) {
  46. return tree.reduce(function (acc, node) {
  47. if (callback(node)) {
  48. acc.push(node);
  49. }
  50. if (node.children) {
  51. var children = flatFilter(node.children, callback);
  52. acc.push.apply(acc, (0, _toConsumableArray3['default'])(children));
  53. }
  54. return acc;
  55. }, []);
  56. }
  57. // export function normalizeColumns (elements) {
  58. // const columns = []
  59. // React.Children.forEach(elements, (element) => {
  60. // if (!React.isValidElement(element)) {
  61. // return
  62. // }
  63. // const column = {
  64. // ...element.props,
  65. // }
  66. // if (element.key) {
  67. // column.key = element.key
  68. // }
  69. // if (element.type && element.type.__ANT_TABLE_COLUMN_GROUP) {
  70. // column.children = normalizeColumns(column.children)
  71. // }
  72. // columns.push(column)
  73. // })
  74. // return columns
  75. // }
  76. function generateValueMaps(items) {
  77. var maps = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
  78. (items || []).forEach(function (_ref) {
  79. var value = _ref.value,
  80. children = _ref.children;
  81. maps[value.toString()] = value;
  82. generateValueMaps(children, maps);
  83. });
  84. return maps;
  85. }