/**
 * tdesign v0.39.1-alpha.1
 * (c) 2022 tdesign
 * @license MIT
 */

'use strict';

Object.defineProperty(exports, '__esModule', { value: true });

var _toConsumableArray = require('@babel/runtime/helpers/toConsumableArray');
var _defineProperty = require('@babel/runtime/helpers/defineProperty');
var _slicedToArray = require('@babel/runtime/helpers/slicedToArray');
var VueCompositionAPI = require('@vue/composition-api');
var tdesignIconsVue = require('tdesign-icons-vue');
var cloneDeep = require('lodash/cloneDeep');
var get = require('lodash/get');
var table_hooks_treeStore = require('./tree-store.js');
var table_hooks_useClassName = require('./useClassName.js');
var table_tr = require('../tr.js');
require('@babel/runtime/helpers/classCallCheck');
require('@babel/runtime/helpers/createClass');
require('../utils.js');
require('@babel/runtime/helpers/typeof');
require('lodash/isFunction');
require('../../_common/js/log/log.js');
require('../../config-provider/useConfig.js');
require('lodash/mergeWith');
require('../../config-provider/context.js');
require('lodash/merge');
require('../../_common/js/global-config/default-config.js');
require('../../_common/js/global-config/locale/zh_CN.js');
require('../../_chunks/dep-33ee4bce.js');
require('lodash/upperFirst');
require('lodash/isString');
require('lodash/pick');
require('./useFixed.js');
require('../../_common/js/utils/getScrollbarWidth.js');
require('../../utils/dom.js');
require('vue');
require('raf');
require('../../utils/easing.js');
require('../ellipsis.js');
require('../../utils/render-tnode.js');
require('lodash/isObject');
require('../../popup/index.js');
require('../../popup/popup.js');
require('@popperjs/core');
require('../../config.js');
require('../../utils/classnames.js');
require('../../utils/helper.js');
require('@babel/runtime/helpers/objectWithoutProperties');
require('lodash/camelCase');
require('../../_common/js/utils/set-style.js');
require('../../popup/props.js');
require('../../popup/container.js');
require('../../utils/map-props.js');
require('../../utils/withInstall.js');
require('lodash/capitalize');
require('../base-table-props.js');

function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }

var _toConsumableArray__default = /*#__PURE__*/_interopDefaultLegacy(_toConsumableArray);
var _defineProperty__default = /*#__PURE__*/_interopDefaultLegacy(_defineProperty);
var _slicedToArray__default = /*#__PURE__*/_interopDefaultLegacy(_slicedToArray);
var cloneDeep__default = /*#__PURE__*/_interopDefaultLegacy(cloneDeep);
var get__default = /*#__PURE__*/_interopDefaultLegacy(get);

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__default["default"](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 useTreeData(props, context) {
  var _toRefs = VueCompositionAPI.toRefs(props),
      data = _toRefs.data,
      columns = _toRefs.columns;

  var store = VueCompositionAPI.ref(new table_hooks_treeStore["default"]());
  var treeNodeCol = VueCompositionAPI.ref();
  var dataSource = VueCompositionAPI.ref([]);

  var _useClassName = table_hooks_useClassName["default"](),
      tableTreeClasses = _useClassName.tableTreeClasses;

  var rowDataKeys = VueCompositionAPI.computed(function () {
    var _props$tree;

    return {
      rowKey: props.rowKey || "id",
      childrenKey: ((_props$tree = props.tree) === null || _props$tree === void 0 ? void 0 : _props$tree.childrenKey) || "children"
    };
  });
  VueCompositionAPI.watch([data], function (_ref) {
    var _ref2 = _slicedToArray__default["default"](_ref, 1),
        val = _ref2[0];

    if (!val) return [];

    if (!props.tree || !Object.keys(props.tree).length) {
      dataSource.value = val;
      return;
    }

    var newVal = cloneDeep__default["default"](val);
    dataSource.value = newVal;
    store.value.initialTreeStore(newVal, props.columns, rowDataKeys.value);
  }, {
    immediate: true
  });
  VueCompositionAPI.onUnmounted(function () {
    var _store$value$treeData;

    if (!props.tree || !Object.keys(props.tree).length) return;
    (_store$value$treeData = store.value.treeDataMap) === null || _store$value$treeData === void 0 ? void 0 : _store$value$treeData.clear();
    store.value = null;
  });
  VueCompositionAPI.watch([columns], function () {
    treeNodeCol.value = getTreeNodeColumnCol();
  }, {
    immediate: true
  });

  function getTreeNodeStyle(level) {
    var _props$tree2;

    if (level === void 0) return;
    var indent = ((_props$tree2 = props.tree) === null || _props$tree2 === void 0 ? void 0 : _props$tree2.indent) || 24;
    return {
      paddingLeft: "".concat(level * indent || 1, "px")
    };
  }

  function toggleExpandData(p) {
    dataSource.value = store.value.toggleExpandData(p, dataSource.value, rowDataKeys.value);
  }

  function getTreeNodeColumnCol() {
    var _props$tree3;

    var columns2 = props.columns;
    var treeNodeColumnIndex = ((_props$tree3 = props.tree) === null || _props$tree3 === void 0 ? void 0 : _props$tree3.treeNodeColumnIndex) || 0;

    while ((_columns2$treeNodeCol = columns2[treeNodeColumnIndex]) !== null && _columns2$treeNodeCol !== void 0 && _columns2$treeNodeCol.type || ((_columns2$treeNodeCol2 = columns2[treeNodeColumnIndex]) === null || _columns2$treeNodeCol2 === void 0 ? void 0 : _columns2$treeNodeCol2.colKey) === "__EXPAND_ROW_ICON_COLUMN__") {
      var _columns2$treeNodeCol, _columns2$treeNodeCol2;

      treeNodeColumnIndex += 1;
    }

    return columns2[treeNodeColumnIndex];
  }

  function formatTreeColum(col) {
    if (!props.tree || !Object.keys(props.tree).length || col.colKey !== treeNodeCol.value.colKey) return col;

    var newCol = _objectSpread({}, treeNodeCol.value);

    newCol.cell = function (h, p) {
      var cellInfo = table_tr.renderCell(_objectSpread(_objectSpread({}, p), {}, {
        col: _objectSpread({}, treeNodeCol.value)
      }), context.slots);
      var currentState = store.value.treeDataMap.get(get__default["default"](p.row, rowDataKeys.value.rowKey));
      var colStyle = getTreeNodeStyle(currentState === null || currentState === void 0 ? void 0 : currentState.level);

      var classes = _defineProperty__default["default"]({}, tableTreeClasses.inlineCol, !!col.ellipsis);

      var childrenNodes = get__default["default"](p.row, rowDataKeys.value.childrenKey);

      if (childrenNodes && childrenNodes instanceof Array) {
        var _store$value$treeData2;

        var IconNode = (_store$value$treeData2 = store.value.treeDataMap.get(get__default["default"](p.row, rowDataKeys.value.rowKey))) !== null && _store$value$treeData2 !== void 0 && _store$value$treeData2.expanded ? tdesignIconsVue.MinusRectangleIcon : tdesignIconsVue.AddRectangleIcon;
        return h("div", {
          "class": [tableTreeClasses.col, classes],
          "style": colStyle
        }, [!!childrenNodes.length && h(IconNode, {
          "class": tableTreeClasses.icon,
          "on": {
            "click": function click() {
              return toggleExpandData(p);
            }
          }
        }), cellInfo]);
      }

      return h("div", {
        "style": colStyle,
        "class": classes
      }, [cellInfo]);
    };

    if (newCol.ellipsis === true) {
      newCol.ellipsis = function (h, p) {
        return table_tr.renderCell(_objectSpread(_objectSpread({}, p), {}, {
          col: _objectSpread({}, treeNodeCol.value)
        }), context.slots);
      };
    }

    return newCol;
  }

  function setData(key, newRowData) {
    var rowIndex = store.value.updateData(key, newRowData, dataSource.value, rowDataKeys.value);

    var newData = _toConsumableArray__default["default"](dataSource.value);

    newData[rowIndex] = newRowData;
    dataSource.value = newData;
  }

  function getData(key) {
    return store.value.getData(key);
  }

  function remove(key) {
    dataSource.value = store.value.remove(key, dataSource.value, rowDataKeys.value);
  }

  function appendTo(key, newData) {
    dataSource.value = store.value.appendTo(key, newData, dataSource.value, rowDataKeys.value);
  }

  return {
    store: store,
    rowDataKeys: rowDataKeys,
    dataSource: dataSource,
    setData: setData,
    getData: getData,
    remove: remove,
    appendTo: appendTo,
    formatTreeColum: formatTreeColum
  };
}

exports["default"] = useTreeData;
//# sourceMappingURL=useTreeData.js.map
