"use strict";

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

var _cloneDeep = _interopRequireDefault(require("clone-deep"));

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); 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 = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : 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; }

// 扁平数据转换为树
var toTree = function toTree(data) {
  var cloneData = (0, _cloneDeep["default"])(data);
  var map = {};
  var res = [];
  cloneData.forEach(function (item) {
    return map[item.taskId] = item;
  });
  cloneData.forEach(function (item) {
    var parent = map[item.parentTaskId];

    if (parent) {
      if (!parent.children) parent.children = [];
      parent.children.push(item);
    } else {
      res.push(item);
    }
  });
  return res;
};

var getTaskNodeData = function getTaskNodeData(tasks) {
  if (!tasks) return [];
  return tasks.map(function (task, index) {
    return {
      key: task.taskId,
      index: index,
      title: task.taskName,
      nodeType: task.objType === 4 ? "e-project" : "e-task",
      isLeaf: task.children && task.children.length === 0,
      children: getTaskNodeData(task.children),
      data: _objectSpread({}, task)
    };
  });
};

var formatTasksToTreeData = function formatTasksToTreeData(tasks) {
  var taskTreeData = toTree(tasks);
  var taskNodeData = getTaskNodeData(taskTreeData);
  return taskNodeData;
};

var _default = formatTasksToTreeData;
exports["default"] = _default;