flo-utils/es/filterTree.js

some utils

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; }

/**
 * @name filterTree, 树状结构数据过滤集合
 * @param {*} data
 * @param {*} id
 * @param {string} [key="id"]
 * @returns
 */
import isArray from './isArray';

function filterTree(data, id) {
  var key = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 'id';

  if (!isArray(data)) {
    return new Error('data params is array type');
  }

  var newData = data.filter(function (x) {
    return x[key] !== id;
  });
  newData = newData.map(function (item) {
    var x = _objectSpread({}, item);

    x.children && (x.children = filterTree(x.children, id, key));
    return x;
  });
  return newData;
}

export default filterTree;