"use strict";

var __assign = void 0 && (void 0).__assign || function () {
  __assign = Object.assign || function (t) {
    for (var s, i = 1, n = arguments.length; i < n; i++) {
      s = arguments[i];

      for (var p in s) {
        if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p];
      }
    }

    return t;
  };

  return __assign.apply(this, arguments);
};

var __createBinding = void 0 && (void 0).__createBinding || (Object.create ? function (o, m, k, k2) {
  if (k2 === undefined) k2 = k;
  Object.defineProperty(o, k2, {
    enumerable: true,
    get: function get() {
      return m[k];
    }
  });
} : function (o, m, k, k2) {
  if (k2 === undefined) k2 = k;
  o[k2] = m[k];
});

var __setModuleDefault = void 0 && (void 0).__setModuleDefault || (Object.create ? function (o, v) {
  Object.defineProperty(o, "default", {
    enumerable: true,
    value: v
  });
} : function (o, v) {
  o["default"] = v;
});

var __importStar = void 0 && (void 0).__importStar || function (mod) {
  if (mod && mod.__esModule) return mod;
  var result = {};
  if (mod != null) for (var k in mod) {
    if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
  }

  __setModuleDefault(result, mod);

  return result;
};

var __rest = void 0 && (void 0).__rest || function (s, e) {
  var t = {};

  for (var p in s) {
    if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  }

  if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
    if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  }
  return t;
};

var __importDefault = void 0 && (void 0).__importDefault || function (mod) {
  return mod && mod.__esModule ? mod : {
    "default": mod
  };
};

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

var jsx_runtime_1 = require("react/jsx-runtime");

var React = __importStar(require("react"));

var rc_tree_1 = __importStar(require("rc-tree"));

Object.defineProperty(exports, "TreeNode", {
  enumerable: true,
  get: function get() {
    return rc_tree_1.TreeNode;
  }
});

var classnames_1 = __importDefault(require("classnames"));

var useMeasure_1 = __importDefault(require("react-use/lib/useMeasure"));

var checkbox_1 = __importDefault(require("../checkbox"));

var icon_1 = __importDefault(require("../icon"));

var size_1 = __importDefault(require("lodash/size"));

require("./style/tree.css");

var react_1 = require("react");

var dropIndicator_1 = __importDefault(require("./dropIndicator"));

var prefixCls = 'dmc-tree';
var Tree = React.forwardRef(function (props, treeRef) {
  var _a;

  var className = props.className,
      showIcon = props.showIcon,
      switcherIcon = props.switcherIcon,
      blockNode = props.blockNode,
      children = props.children,
      checkable = props.checkable,
      direction = props.direction,
      icon = props.icon,
      fullSelect = props.fullSelect,
      height = props.height,
      type = props.type,
      withoutSelectEffect = props.withoutSelectEffect,
      itemHeight = props.itemHeight,
      style = props.style,
      emptyContent = props.emptyContent,
      treeData = props.treeData,
      withoutHoverEffect = props.withoutHoverEffect,
      restProps = __rest(props, ["className", "showIcon", "switcherIcon", "blockNode", "children", "checkable", "direction", "icon", "fullSelect", "height", "type", "withoutSelectEffect", "itemHeight", "style", "emptyContent", "treeData", "withoutHoverEffect"]);

  var _b = useMeasure_1["default"](),
      measureRefCallback = _b[0],
      eleHeight = _b[1].height;

  var wrapperRef = react_1.useRef(null);
  var rcTreeRef = react_1.useRef(null);
  var wrapperRefCallback = react_1.useCallback(function (node) {
    if (node) {
      measureRefCallback(node);
    }

    wrapperRef.current = node;
  }, [measureRefCallback]);
  var rcTreeRefCallback = react_1.useCallback(function (node) {
    rcTreeRef.current = node;
  }, []);

  var mixIcon = function mixIcon(props) {
    var _a;

    var checkboxDisabled = props.disabled || ((_a = props.data) === null || _a === void 0 ? void 0 : _a.disableCheckbox);
    var iconRender = typeof icon === 'function' ? icon(props) : icon;
    return jsx_runtime_1.jsxs(jsx_runtime_1.Fragment, {
      children: [checkable && jsx_runtime_1.jsx(checkbox_1["default"], {
        checked: props.checked,
        indeterminate: props.halfChecked,
        disabled: checkboxDisabled
      }, void 0), iconRender]
    }, void 0);
  };

  var treeStyle = React.useMemo(function () {
    return itemHeight ? __assign(__assign({}, style), {
      lineHeight: itemHeight - 6 + "px"
    }) : style;
  }, [itemHeight, style]);
  var cls = classnames_1["default"](className, (_a = {}, _a[prefixCls + "-icon-hide"] = !showIcon, _a[prefixCls + "-block-node"] = blockNode, _a[prefixCls + "-checkable"] = checkable, _a[prefixCls + "-rtl"] = direction === 'rtl', _a[prefixCls + "-full"] = fullSelect, _a[prefixCls + "-without-select-effect"] = withoutSelectEffect, _a[prefixCls + "-without-hover-effect"] = withoutHoverEffect, _a[prefixCls + "-" + type] = type, _a));
  react_1.useImperativeHandle(treeRef, function () {
    return {
      treeComponent: rcTreeRef.current,
      componentWrapper: wrapperRef.current
    };
  });
  return jsx_runtime_1.jsx("div", __assign({
    className: prefixCls + "-wrapper",
    style: {
      height: height
    },
    ref: wrapperRefCallback
  }, {
    children: size_1["default"](treeData) === 0 && emptyContent ? emptyContent : eleHeight && jsx_runtime_1.jsx(rc_tree_1["default"], __assign({}, restProps, {
      style: treeStyle,
      ref: rcTreeRefCallback,
      treeData: treeData,
      height: eleHeight,
      prefixCls: prefixCls,
      className: cls,
      checkable: checkable,
      icon: mixIcon,
      itemHeight: itemHeight,
      dropIndicatorRender: dropIndicator_1["default"],
      switcherIcon: switcherIcon || function (nodeProps) {
        var isLeaf = nodeProps.isLeaf,
            expanded = nodeProps.expanded;
        var icons = ['arrow-right', 'arrow-down'];
        return !isLeaf ? jsx_runtime_1.jsx(icon_1["default"], {
          icon: icons[Number(!!expanded)]
        }, void 0) : null;
      }
    }, {
      children: children
    }), void 0)
  }), void 0);
});
Tree.defaultProps = {
  checkable: false,
  showIcon: true,
  blockNode: false,
  virtual: true,
  fullSelect: false,
  withoutSelectEffect: false
};
var TreeMemo = React.memo(Tree);
exports.TreeMemo = TreeMemo;
TreeMemo.displayName = 'TreeMemo';
Tree.displayName = 'Tree';
exports["default"] = Tree;

