flo-utils/lib/arrayTreeCallBack.js

some utils

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;

var _isArray = _interopRequireDefault(require("./isArray"));

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

function arrayTreeCallBack(treeNodes, Fn) {
  var config = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};

  if (!(0, _isArray.default)(treeNodes)) {
    return new Error('data params is array type');
  }

  if (!treeNodes.length) return [];
  var defaultOpt = {
    childrenKeyName: 'children',
    traversal: 'DLR' // LRD 后序遍历；DLR 前序遍历

  };

  var options = _objectSpread(_objectSpread({}, defaultOpt), config);

  function treeMap(tree, parent) {
    return tree.map(function (t, i) {
      var item = _objectSpread({}, t);

      if (options.traversal === 'DLR') {
        item = Fn(item, i, parent);

        if (item[options.childrenKeyName] && item[options.childrenKeyName].length) {
          var children = item[options.childrenKeyName];
          item[options.childrenKeyName] = treeMap(children, item);
        }
      } else if (options.traversal === 'LRD') {
        if (item[options.childrenKeyName] && item[options.childrenKeyName].length) {
          var _children = item[options.childrenKeyName];
          item[options.childrenKeyName] = treeMap(_children, item);
        }

        item = Fn(item, i, parent);
      }

      return item;
    });
  }

  return treeMap(treeNodes);
}

var _default = arrayTreeCallBack;
exports.default = _default;