@hi-ui/hiui 2.15.11 → 2.15.12

package/es/tree/TreeItem.js

@@ -1,74 +1,91 @@
 "use strict";
 
+var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard");
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports["default"] = void 0;
 
-var _react = _interopRequireWildcard(require("react"));
+var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
 
-var _index = _interopRequireDefault(require("../table/checkbox/index"));
+var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
 
-var _reactDnd = require("react-dnd");
+var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
 
-var _input = _interopRequireDefault(require("../input"));
+var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn"));
 
-var _reactDom = require("react-dom");
+var _getPrototypeOf4 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf"));
 
-var _TreeDivider = _interopRequireDefault(require("./TreeDivider"));
+var _assertThisInitialized2 = _interopRequireDefault(require("@babel/runtime/helpers/assertThisInitialized"));
 
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
+var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));
 
-function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }
+var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
 
-function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { "default": obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj["default"] = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
-
-function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
-
-function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
-
-function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
+var _react = _interopRequireWildcard(require("react"));
 
-function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
+var _index = _interopRequireDefault(require("../table/checkbox/index"));
 
-function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); if (superClass) _setPrototypeOf(subClass, superClass); }
+var _reactDnd = require("react-dnd");
 
-function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
+var _classnames = _interopRequireDefault(require("classnames"));
 
-function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
+var _input = _interopRequireDefault(require("../input"));
 
-function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } return _assertThisInitialized(self); }
+var _reactDom = require("react-dom");
 
-function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
+var _TreeDivider = _interopRequireDefault(require("./TreeDivider"));
 
-function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Date.prototype.toString.call(Reflect.construct(Date, [], function () {})); return true; } catch (e) { return false; } }
+var _IconLoading = _interopRequireDefault(require("./IconLoading"));
 
-function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
+var _context = _interopRequireDefault(require("../context"));
 
 var Types = {
   TreeNode: 'treeNode'
 };
 
-var TreeItem = /*#__PURE__*/function (_Component) {
-  _inherits(TreeItem, _Component);
-
-  var _super = _createSuper(TreeItem);
+var TreeItem =
+/*#__PURE__*/
+function (_Component) {
+  (0, _inherits2["default"])(TreeItem, _Component);
 
   function TreeItem() {
-    _classCallCheck(this, TreeItem);
+    var _getPrototypeOf2;
+
+    var _this;
 
-    return _super.apply(this, arguments);
+    (0, _classCallCheck2["default"])(this, TreeItem);
+
+    for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
+      args[_key] = arguments[_key];
+    }
+
+    _this = (0, _possibleConstructorReturn2["default"])(this, (_getPrototypeOf2 = (0, _getPrototypeOf4["default"])(TreeItem)).call.apply(_getPrototypeOf2, [this].concat(args)));
+    (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this), "state", {
+      loading: false
+    });
+    (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this), "loadingTimer", null);
+    return _this;
   }
 
-  _createClass(TreeItem, [{
+  (0, _createClass2["default"])(TreeItem, [{
+    key: "componentWillUnmount",
+    value: function componentWillUnmount() {
+      clearTimeout(this.loadingTimer);
+    }
+  }, {
     key: "render",
     value: function render() {
-      var _this = this;
+      var _this2 = this;
 
       var _this$props = this.props,
+          level = _this$props.level,
           editable = _this$props.editable,
           draggable = _this$props.draggable,
-          dropDividerPosition = _this$props.dropDividerPosition,
+          direction = _this$props.direction,
           checked = _this$props.checked,
           expanded = _this$props.expanded,
           highlight = _this$props.highlight,
@@ -76,11 +93,9 @@ var TreeItem = /*#__PURE__*/function (_Component) {
           editingNodes = _this$props.editingNodes,
           prefixCls = _this$props.prefixCls,
           semiChecked = _this$props.semiChecked,
-          onNodeClick = _this$props.onNodeClick,
           _onClick = _this$props.onClick,
           highlightable = _this$props.highlightable,
           item = _this$props.item,
-          draggingNode = _this$props.draggingNode,
           checkable = _this$props.checkable,
           itemStyle = _this$props.itemStyle,
           onExpanded = _this$props.onExpanded,
@@ -88,7 +103,6 @@ var TreeItem = /*#__PURE__*/function (_Component) {
           cancelEditNode = _this$props.cancelEditNode,
           cancelAddSiblingNode = _this$props.cancelAddSiblingNode,
           renderTree = _this$props.renderTree,
-          renderRightClickMenu = _this$props.renderRightClickMenu,
           onCheckChange = _this$props.onCheckChange,
           onSetHighlight = _this$props.onSetHighlight,
           showRightClickMenu = _this$props.showRightClickMenu,
@@ -97,36 +111,86 @@ var TreeItem = /*#__PURE__*/function (_Component) {
           connectDragSource = _this$props.connectDragSource,
           connectDropTarget = _this$props.connectDropTarget,
           isOver = _this$props.isOver,
-          targetNode = _this$props.targetNode,
+          isDragging = _this$props.isDragging,
           saveEditNode = _this$props.saveEditNode,
           origin = _this$props.origin,
-          loadChildren = _this$props.loadChildren;
-
-      var treeItem = /*#__PURE__*/_react["default"].createElement("li", {
-        key: item.id
-      }, /*#__PURE__*/_react["default"].createElement("div", {
-        style: {
-          display: 'flex'
-        }
-      }, targetNode === item.id && dropDividerPosition === 'down' && isOver && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], {
-        top: true
-      }), /*#__PURE__*/_react["default"].createElement("span", {
+          loadChildren = _this$props.loadChildren,
+          isRoot = _this$props.isRoot,
+          showLine = _this$props.showLine,
+          localeDatas = _this$props.localeDatas,
+          apperance = _this$props.apperance,
+          theme = _this$props.theme;
+      var themeColor = {
+        'orange': '#f63',
+        'cyan': '#46bc99',
+        'magenta': '#ff5975',
+        'lavender': '#b450de',
+        'blue': '#3da8f5',
+        'purple': '#8a8acb',
+        'hiui-blue': '#4284f5'
+      };
+      var _localeDatas$tree = localeDatas.tree,
+          nodePlaceholder = _localeDatas$tree.nodePlaceholder,
+          confirm = _localeDatas$tree.confirm,
+          cancel = _localeDatas$tree.cancel;
+
+      var treeItem = _react["default"].createElement("li", {
+        key: item.id,
+        className: (0, _classnames["default"])({
+          'is-root': isRoot
+        }, {
+          'no-expanded': !item.children || item.children.length === 0 || item.children && !expanded
+        })
+      }, direction === 'up' && isOver && _react["default"].createElement(_TreeDivider["default"], {
+        placement: "top",
+        theme: theme
+      }), _react["default"].createElement("div", {
+        className: (0, _classnames["default"])('item__wrapper', {
+          'can-expand': item.children && item.children.length > 0,
+          'item__wrapper--expanded': expanded
+        })
+      }, (!item.children || item.children && !expanded) && direction === 'down' && isOver && _react["default"].createElement(_TreeDivider["default"], {
+        theme: theme,
+        placement: "bottom"
+      }), _react["default"].createElement("span", {
         onClick: function onClick() {
           onExpanded(expanded, item);
 
-          if (origin) {
-            loadChildren(item.id);
+          if (origin && (!item.children || item.children.length === 0)) {
+            _this2.setState({
+              loading: true
+            });
+
+            loadChildren(item.id)["finally"](function () {
+              _this2.loadingTimer = setTimeout(function () {
+                _this2.setState({
+                  loading: false
+                });
+              }, 300);
+            });
           }
         },
+        style: {
+          position: 'relative'
+        },
         className: "".concat(prefixCls, "_item-icon")
-      }, (item.children && item.children.length > 0 || origin && !expanded) && renderSwitcher(expanded)), checkable ? /*#__PURE__*/_react["default"].createElement(_index["default"], {
+      }, item.children && item.children.length > 0 || origin && !expanded ? renderSwitcher(expanded) : showLine && _react["default"].createElement("span", {
+        className: "hi-tree__dot"
+      }) || apperance === 'folder' && _react["default"].createElement("i", {
+        className: "hi-icon icon-File"
+      }), this.state.loading && _react["default"].createElement(_IconLoading["default"], {
+        theme: theme
+      })), checkable ? _react["default"].createElement(_index["default"], {
         semi: semiChecked.includes(item.id),
         checked: checked,
         onChange: function onChange() {
           return onCheckChange(checked, item);
         },
         onTitleClick: function onTitleClick(e) {
-          onNodeClick && onNodeClick(item);
+          if (item.disabled) {
+            return false;
+          }
+
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
@@ -136,28 +200,39 @@ var TreeItem = /*#__PURE__*/function (_Component) {
         disabled: item.disabled
       }) : item.status === 'editable' || editNodes.map(function (node) {
         return node.id;
-      }).includes(item.id) ? /*#__PURE__*/_react["default"].createElement("div", {
+      }).includes(item.id) ? _react["default"].createElement("div", {
         className: "editing"
-      }, /*#__PURE__*/_react["default"].createElement(_input["default"], {
-        placeholder: "\u8BF7\u8F93\u5165\u83DC\u5355\u540D\u79F0",
+      }, _react["default"].createElement(_input["default"], {
+        placeholder: nodePlaceholder,
         value: (editingNodes.find(function (node) {
           return node.id === item.id;
         }) || {}).title,
         onChange: function onChange(e) {
           onValueChange(e.target.value, item.id);
         }
-      }), /*#__PURE__*/_react["default"].createElement("span", {
-        style: {
+      }), _react["default"].createElement("span", {
+        style: (editingNodes.find(function (node) {
+          return node.id === item.id;
+        }) || {}).title === '' ? {
+          marginRight: 12,
+          color: '#999',
+          cursor: 'not-allowed'
+        } : {
           cursor: 'pointer',
           marginRight: 12,
-          color: '#4284F5'
+          color: themeColor[theme]
         },
         onClick: function onClick() {
-          saveEditNode(item.id);
+          if ((editingNodes.find(function (node) {
+            return node.id === item.id;
+          }) || {}).title !== '') {
+            saveEditNode(item.id, level);
+          }
         }
-      }, "\u786E\u5B9A"), /*#__PURE__*/_react["default"].createElement("span", {
+      }, confirm), _react["default"].createElement("span", {
         style: {
-          cursor: 'pointer'
+          cursor: 'pointer',
+          color: '#999'
         },
         onClick: function onClick() {
           if (editNodes.map(function (node) {
@@ -168,44 +243,63 @@ var TreeItem = /*#__PURE__*/function (_Component) {
             cancelAddSiblingNode(item.id);
           }
         }
-      }, "\u53D6\u6D88")) : draggable ? connectDragSource( /*#__PURE__*/_react["default"].createElement("span", {
+      }, cancel)) : draggable ? connectDragSource(_react["default"].createElement("span", {
         style: item.style,
-        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(draggingNode === item.id ? 'dragging' : ''),
+        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(isDragging ? 'dragging' : '', " ").concat(item.disabled ? prefixCls + '_item-text--disabled' : ''),
         onContextMenu: function onContextMenu(e) {
+          if (item.disabled) {
+            return false;
+          }
+
           if (editable) {
             e.preventDefault();
-            showRightClickMenu(item);
+            showRightClickMenu(item, e, level);
           }
         },
         onClick: function onClick(e) {
+          if (item.disabled) {
+            return false;
+          }
+
           closeRightClickMenu();
-          onNodeClick && onNodeClick(item);
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
         }
-      }, item.title, renderRightClickMenu(item), targetNode === item.id && dropDividerPosition === 'sub' && isOver && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], null))) : /*#__PURE__*/_react["default"].createElement("span", {
+      }, item.title, direction === 'sub' && isOver && _react["default"].createElement(_TreeDivider["default"], {
+        placement: "inner",
+        theme: this.props.theme
+      }))) : _react["default"].createElement("span", {
         style: item.style,
-        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(draggingNode === item.id ? 'dragging' : ''),
+        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(isDragging ? 'dragging' : '', " ").concat(item.disabled ? prefixCls + '_item-text--disabled' : ''),
         onContextMenu: function onContextMenu(e) {
-          if (_this.props.editable) {
+          if (item.disabled) {
+            return false;
+          }
+
+          if (_this2.props.editable) {
             e.preventDefault();
-            showRightClickMenu(item);
+            showRightClickMenu(item, e, level);
           }
         },
         onClick: function onClick(e) {
+          if (item.disabled) {
+            return false;
+          }
+
           closeRightClickMenu();
-          onNodeClick && onNodeClick(item);
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
         }
-      }, item.title, renderRightClickMenu(item))), item.children && item.children.length > 0 && expanded ? renderTree(item.children) : null);
+      }, item.title)), item.children && item.children.length > 0 && expanded ? renderTree(item.children, [], level + 1) : null, item.children && expanded && direction === 'down' && isOver && _react["default"].createElement(_TreeDivider["default"], {
+        placement: "bottom",
+        theme: this.props.theme
+      }));
 
       return draggable ? connectDropTarget(treeItem) : treeItem;
     }
   }]);
-
   return TreeItem;
 }(_react.Component);
 
@@ -219,65 +313,46 @@ var source = {
     props.onDragStart(props.item);
     return {
       sourceItem: props.item,
-      originalExpandStatus: props.expanded
+      originalExpandStatus: props.expanded,
+      sourceLevel: props.level
     };
-  },
-  endDrag: function endDrag(props, monitor) {
-    var dropResult = monitor.getDropResult();
-
-    if (!dropResult) {
-      var removeTargetNode = props.removeTargetNode,
-          removeDraggingNode = props.removeDraggingNode;
-      removeDraggingNode();
-      removeTargetNode();
-    }
   }
 };
 var target = {
   drop: function drop(props, monitor) {
     var _monitor$getItem = monitor.getItem(),
         sourceItem = _monitor$getItem.sourceItem,
-        originalExpandStatus = _monitor$getItem.originalExpandStatus;
+        originalExpandStatus = _monitor$getItem.originalExpandStatus,
+        sourceLevel = _monitor$getItem.sourceLevel;
 
     var targetItem = props.item,
         dropNode = props.dropNode,
-        removeDraggingNode = props.removeDraggingNode,
         expandTreeNode = props.expandTreeNode,
-        removeTargetNode = props.removeTargetNode,
-        dropDividerPosition = props.dropDividerPosition; // 先看下是不是在最近得组件
+        direction = props.direction,
+        level = props.level; // 先看下是不是在最近得组件
 
     if (monitor.isOver({
       shallow: true
     })) {
       if (sourceItem.id === targetItem.id || targetItem.children && targetItem.children.map(function (t) {
         return t.id;
-      }).includes(sourceItem.id)) {
+      }).includes(sourceItem.id) && direction === 'sub') {
         // 如果源节点就是目的节点或者源节点是目的节点的子节点（直系）再或者源节点是目的节点的父节点，那么什么都不做
         // 如果什么都不做，原来展开则现在还展开
         if (originalExpandStatus) {
           expandTreeNode(sourceItem.id);
         }
-
-        removeDraggingNode();
-        removeTargetNode();
       } else {
         // 移动节点到相应位置
-        dropNode(sourceItem, targetItem, dropDividerPosition);
-        removeDraggingNode();
-        removeTargetNode();
+        dropNode(sourceItem, targetItem, direction, {
+          before: sourceLevel,
+          after: level
+        });
       }
     }
   },
   hover: function hover(props, monitor, component) {
-    var _monitor$getItem2 = monitor.getItem(),
-        sourceItem = _monitor$getItem2.sourceItem;
-
-    var targetItem = props.item,
-        setDraggingNode = props.setDraggingNode,
-        setTargetNode = props.setTargetNode,
-        positionX = props.positionX,
-        positionY = props.positionY,
-        setPosition = props.setPosition; // 先看下是不是在最近得组件
+    var setDirection = props.setDirection; // 先看下是不是在最近得组件
 
     if (monitor.isOver({
       shallow: true
@@ -285,16 +360,12 @@ var target = {
       var sourcePosition = monitor.getClientOffset();
       var targetComponent = (0, _reactDom.findDOMNode)(component).getBoundingClientRect();
 
-      if (!(sourcePosition.x === positionX && sourcePosition.y === positionY)) {
-        setPosition(sourcePosition.x, sourcePosition.y); // 如果在节点的上半部分，则为移动其内部，如果为下半部分，则为节点下方
-
-        if (sourcePosition.y <= targetComponent.y + targetComponent.height / 2) {
-          setTargetNode(targetItem.id, 'sub');
-        } else {
-          setTargetNode(targetItem.id, 'down');
-        }
-
-        setDraggingNode(sourceItem.id);
+      if (sourcePosition.y <= targetComponent.y + targetComponent.height / 3) {
+        setDirection('up');
+      } else if (targetComponent.y + targetComponent.height / 3 < sourcePosition.y && sourcePosition.y <= targetComponent.y + targetComponent.height * 2 / 3) {
+        setDirection('sub');
+      } else {
+        setDirection('down');
       }
     }
   }
@@ -310,36 +381,59 @@ function sourceCollect(connect, monitor) {
 function targetCollect(connect, monitor) {
   return {
     connectDropTarget: connect.dropTarget(),
-    isOver: monitor.isOver()
+    isOver: monitor.isOver({
+      shallow: true
+    })
   };
 }
 
 var DraggableTreeItem = (0, _reactDnd.DropTarget)(Types['TreeNode'], target, targetCollect)((0, _reactDnd.DragSource)(Types['TreeNode'], source, sourceCollect)(TreeItem));
 
 var HOCTreeItem = function HOCTreeItem(TreeItemComponent) {
-  return /*#__PURE__*/function (_Component2) {
-    _inherits(WrapperTreeItem, _Component2);
+  var _temp;
 
-    var _super2 = _createSuper(WrapperTreeItem);
+  return _temp =
+  /*#__PURE__*/
+  function (_Component2) {
+    (0, _inherits2["default"])(WrapperTreeItem, _Component2);
 
     function WrapperTreeItem() {
-      _classCallCheck(this, WrapperTreeItem);
+      var _getPrototypeOf3;
+
+      var _this3;
+
+      (0, _classCallCheck2["default"])(this, WrapperTreeItem);
 
-      return _super2.apply(this, arguments);
+      for (var _len2 = arguments.length, args = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
+        args[_key2] = arguments[_key2];
+      }
+
+      _this3 = (0, _possibleConstructorReturn2["default"])(this, (_getPrototypeOf3 = (0, _getPrototypeOf4["default"])(WrapperTreeItem)).call.apply(_getPrototypeOf3, [this].concat(args)));
+      (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this3), "state", {
+        direction: null
+      });
+      (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this3), "setDirection", function (direction) {
+        _this3.setState({
+          direction: direction
+        });
+      });
+      return _this3;
     }
 
-    _createClass(WrapperTreeItem, [{
+    (0, _createClass2["default"])(WrapperTreeItem, [{
       key: "render",
       value: function render() {
         var draggable = this.props.draggable;
-        return draggable ? /*#__PURE__*/_react["default"].createElement(DraggableTreeItem, this.props) : /*#__PURE__*/_react["default"].createElement(TreeItemComponent, this.props);
+        return draggable ? _react["default"].createElement(DraggableTreeItem, (0, _extends2["default"])({}, this.props, {
+          direction: this.state.direction,
+          setDirection: this.setDirection
+        })) : _react["default"].createElement(TreeItemComponent, this.props);
       }
     }]);
-
     return WrapperTreeItem;
-  }(_react.Component);
+  }(_react.Component), _temp;
 };
 
-var _default = HOCTreeItem(TreeItem);
+var _default = (0, _context["default"])(HOCTreeItem(TreeItem));
 
 exports["default"] = _default;