@hi-ui/hiui 2.15.10 → 2.15.11

package/es/tree/TreeItem.js

@@ -1,97 +1,74 @@
 "use strict";
 
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
-
-var _typeof = require("@babel/runtime/helpers/typeof");
-
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports["default"] = void 0;
 
-var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
+var _react = _interopRequireWildcard(require("react"));
 
-var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
+var _index = _interopRequireDefault(require("../table/checkbox/index"));
 
-var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
+var _reactDnd = require("react-dnd");
 
-var _assertThisInitialized2 = _interopRequireDefault(require("@babel/runtime/helpers/assertThisInitialized"));
+var _input = _interopRequireDefault(require("../input"));
 
-var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));
+var _reactDom = require("react-dom");
 
-var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn"));
+var _TreeDivider = _interopRequireDefault(require("./TreeDivider"));
 
-var _getPrototypeOf2 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf"));
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
 
-var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
+function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }
 
-var _react = _interopRequireWildcard(require("react"));
+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; }
 
-var _index = _interopRequireDefault(require("../table/checkbox/index"));
+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); }
 
-var _reactDnd = require("react-dnd");
+function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
 
-var _classnames = _interopRequireDefault(require("classnames"));
+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 _input = _interopRequireDefault(require("../input"));
+function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
 
-var _reactDom = require("react-dom");
+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 _TreeDivider = _interopRequireDefault(require("./TreeDivider"));
+function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
 
-var _IconLoading = _interopRequireDefault(require("./IconLoading"));
+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 _context = _interopRequireDefault(require("../context"));
+function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } return _assertThisInitialized(self); }
 
-function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
+function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
 
-function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { "default": obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && 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 _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; } }
 
-function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = (0, _getPrototypeOf2["default"])(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = (0, _getPrototypeOf2["default"])(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return (0, _possibleConstructorReturn2["default"])(this, result); }; }
-
-function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
+function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
 
 var Types = {
   TreeNode: 'treeNode'
 };
 
 var TreeItem = /*#__PURE__*/function (_Component) {
-  (0, _inherits2["default"])(TreeItem, _Component);
+  _inherits(TreeItem, _Component);
 
   var _super = _createSuper(TreeItem);
 
   function TreeItem() {
-    var _this;
-
-    (0, _classCallCheck2["default"])(this, TreeItem);
-
-    for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
-      args[_key] = arguments[_key];
-    }
+    _classCallCheck(this, TreeItem);
 
-    _this = _super.call.apply(_super, [this].concat(args));
-    (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this), "state", {
-      loading: false
-    });
-    (0, _defineProperty2["default"])((0, _assertThisInitialized2["default"])(_this), "loadingTimer", null);
-    return _this;
+    return _super.apply(this, arguments);
   }
 
-  (0, _createClass2["default"])(TreeItem, [{
-    key: "componentWillUnmount",
-    value: function componentWillUnmount() {
-      clearTimeout(this.loadingTimer);
-    }
-  }, {
+  _createClass(TreeItem, [{
     key: "render",
     value: function render() {
-      var _this2 = this;
+      var _this = this;
 
       var _this$props = this.props,
-          level = _this$props.level,
           editable = _this$props.editable,
           draggable = _this$props.draggable,
-          direction = _this$props.direction,
+          dropDividerPosition = _this$props.dropDividerPosition,
           checked = _this$props.checked,
           expanded = _this$props.expanded,
           highlight = _this$props.highlight,
@@ -99,9 +76,11 @@ 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,
@@ -109,6 +88,7 @@ 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,
@@ -117,86 +97,36 @@ var TreeItem = /*#__PURE__*/function (_Component) {
           connectDragSource = _this$props.connectDragSource,
           connectDropTarget = _this$props.connectDropTarget,
           isOver = _this$props.isOver,
-          isDragging = _this$props.isDragging,
+          targetNode = _this$props.targetNode,
           saveEditNode = _this$props.saveEditNode,
           origin = _this$props.origin,
-          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;
+          loadChildren = _this$props.loadChildren;
 
       var treeItem = /*#__PURE__*/_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 && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], {
-        placement: "top",
-        theme: theme
-      }), /*#__PURE__*/_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 && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], {
-        theme: theme,
-        placement: "bottom"
+        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", {
         onClick: function onClick() {
           onExpanded(expanded, item);
 
-          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);
-            });
+          if (origin) {
+            loadChildren(item.id);
           }
         },
-        style: {
-          position: 'relative'
-        },
         className: "".concat(prefixCls, "_item-icon")
-      }, item.children && item.children.length > 0 || origin && !expanded ? renderSwitcher(expanded) : showLine && /*#__PURE__*/_react["default"].createElement("span", {
-        className: "hi-tree__dot"
-      }) || apperance === 'folder' && /*#__PURE__*/_react["default"].createElement("i", {
-        className: "hi-icon icon-File"
-      }), this.state.loading && /*#__PURE__*/_react["default"].createElement(_IconLoading["default"], {
-        theme: theme
-      })), checkable ? /*#__PURE__*/_react["default"].createElement(_index["default"], {
+      }, (item.children && item.children.length > 0 || origin && !expanded) && renderSwitcher(expanded)), checkable ? /*#__PURE__*/_react["default"].createElement(_index["default"], {
         semi: semiChecked.includes(item.id),
         checked: checked,
         onChange: function onChange() {
           return onCheckChange(checked, item);
         },
         onTitleClick: function onTitleClick(e) {
-          if (item.disabled) {
-            return false;
-          }
-
+          onNodeClick && onNodeClick(item);
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
@@ -209,7 +139,7 @@ var TreeItem = /*#__PURE__*/function (_Component) {
       }).includes(item.id) ? /*#__PURE__*/_react["default"].createElement("div", {
         className: "editing"
       }, /*#__PURE__*/_react["default"].createElement(_input["default"], {
-        placeholder: nodePlaceholder,
+        placeholder: "\u8BF7\u8F93\u5165\u83DC\u5355\u540D\u79F0",
         value: (editingNodes.find(function (node) {
           return node.id === item.id;
         }) || {}).title,
@@ -217,28 +147,17 @@ var TreeItem = /*#__PURE__*/function (_Component) {
           onValueChange(e.target.value, item.id);
         }
       }), /*#__PURE__*/_react["default"].createElement("span", {
-        style: (editingNodes.find(function (node) {
-          return node.id === item.id;
-        }) || {}).title === '' ? {
-          marginRight: 12,
-          color: '#999',
-          cursor: 'not-allowed'
-        } : {
+        style: {
           cursor: 'pointer',
           marginRight: 12,
-          color: themeColor[theme]
+          color: '#4284F5'
         },
         onClick: function onClick() {
-          if ((editingNodes.find(function (node) {
-            return node.id === item.id;
-          }) || {}).title !== '') {
-            saveEditNode(item.id, level);
-          }
+          saveEditNode(item.id);
         }
-      }, confirm), /*#__PURE__*/_react["default"].createElement("span", {
+      }, "\u786E\u5B9A"), /*#__PURE__*/_react["default"].createElement("span", {
         style: {
-          cursor: 'pointer',
-          color: '#999'
+          cursor: 'pointer'
         },
         onClick: function onClick() {
           if (editNodes.map(function (node) {
@@ -249,63 +168,44 @@ var TreeItem = /*#__PURE__*/function (_Component) {
             cancelAddSiblingNode(item.id);
           }
         }
-      }, cancel)) : draggable ? connectDragSource( /*#__PURE__*/_react["default"].createElement("span", {
+      }, "\u53D6\u6D88")) : draggable ? connectDragSource( /*#__PURE__*/_react["default"].createElement("span", {
         style: item.style,
-        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(isDragging ? 'dragging' : '', " ").concat(item.disabled ? prefixCls + '_item-text--disabled' : ''),
+        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(draggingNode === item.id ? 'dragging' : ''),
         onContextMenu: function onContextMenu(e) {
-          if (item.disabled) {
-            return false;
-          }
-
           if (editable) {
             e.preventDefault();
-            showRightClickMenu(item, e, level);
+            showRightClickMenu(item);
           }
         },
         onClick: function onClick(e) {
-          if (item.disabled) {
-            return false;
-          }
-
           closeRightClickMenu();
+          onNodeClick && onNodeClick(item);
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
         }
-      }, item.title, direction === 'sub' && isOver && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], {
-        placement: "inner",
-        theme: this.props.theme
-      }))) : /*#__PURE__*/_react["default"].createElement("span", {
+      }, item.title, renderRightClickMenu(item), targetNode === item.id && dropDividerPosition === 'sub' && isOver && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], null))) : /*#__PURE__*/_react["default"].createElement("span", {
         style: item.style,
-        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(isDragging ? 'dragging' : '', " ").concat(item.disabled ? prefixCls + '_item-text--disabled' : ''),
+        className: "".concat(prefixCls, "_item-text ").concat(itemStyle, " ").concat(highlight === item.id ? 'highlight' : '', " ").concat(draggingNode === item.id ? 'dragging' : ''),
         onContextMenu: function onContextMenu(e) {
-          if (item.disabled) {
-            return false;
-          }
-
-          if (_this2.props.editable) {
+          if (_this.props.editable) {
             e.preventDefault();
-            showRightClickMenu(item, e, level);
+            showRightClickMenu(item);
           }
         },
         onClick: function onClick(e) {
-          if (item.disabled) {
-            return false;
-          }
-
           closeRightClickMenu();
+          onNodeClick && onNodeClick(item);
           _onClick && _onClick(item);
           highlightable && onSetHighlight(item);
           e.stopPropagation();
         }
-      }, item.title)), item.children && item.children.length > 0 && expanded ? renderTree(item.children, [], level + 1) : null, item.children && expanded && direction === 'down' && isOver && /*#__PURE__*/_react["default"].createElement(_TreeDivider["default"], {
-        placement: "bottom",
-        theme: this.props.theme
-      }));
+      }, item.title, renderRightClickMenu(item))), item.children && item.children.length > 0 && expanded ? renderTree(item.children) : null);
 
       return draggable ? connectDropTarget(treeItem) : treeItem;
     }
   }]);
+
   return TreeItem;
 }(_react.Component);
 
@@ -319,46 +219,65 @@ var source = {
     props.onDragStart(props.item);
     return {
       sourceItem: props.item,
-      originalExpandStatus: props.expanded,
-      sourceLevel: props.level
+      originalExpandStatus: props.expanded
     };
+  },
+  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,
-        sourceLevel = _monitor$getItem.sourceLevel;
+        originalExpandStatus = _monitor$getItem.originalExpandStatus;
 
     var targetItem = props.item,
         dropNode = props.dropNode,
+        removeDraggingNode = props.removeDraggingNode,
         expandTreeNode = props.expandTreeNode,
-        direction = props.direction,
-        level = props.level; // 先看下是不是在最近得组件
+        removeTargetNode = props.removeTargetNode,
+        dropDividerPosition = props.dropDividerPosition; // 先看下是不是在最近得组件
 
     if (monitor.isOver({
       shallow: true
     })) {
       if (sourceItem.id === targetItem.id || targetItem.children && targetItem.children.map(function (t) {
         return t.id;
-      }).includes(sourceItem.id) && direction === 'sub') {
+      }).includes(sourceItem.id)) {
         // 如果源节点就是目的节点或者源节点是目的节点的子节点（直系）再或者源节点是目的节点的父节点，那么什么都不做
         // 如果什么都不做，原来展开则现在还展开
         if (originalExpandStatus) {
           expandTreeNode(sourceItem.id);
         }
+
+        removeDraggingNode();
+        removeTargetNode();
       } else {
         // 移动节点到相应位置
-        dropNode(sourceItem, targetItem, direction, {
-          before: sourceLevel,
-          after: level
-        });
+        dropNode(sourceItem, targetItem, dropDividerPosition);
+        removeDraggingNode();
+        removeTargetNode();
       }
     }
   },
   hover: function hover(props, monitor, component) {
-    var setDirection = props.setDirection; // 先看下是不是在最近得组件
+    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; // 先看下是不是在最近得组件
 
     if (monitor.isOver({
       shallow: true
@@ -366,12 +285,16 @@ var target = {
       var sourcePosition = monitor.getClientOffset();
       var targetComponent = (0, _reactDom.findDOMNode)(component).getBoundingClientRect();
 
-      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');
+      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);
       }
     }
   }
@@ -387,9 +310,7 @@ function sourceCollect(connect, monitor) {
 function targetCollect(connect, monitor) {
   return {
     connectDropTarget: connect.dropTarget(),
-    isOver: monitor.isOver({
-      shallow: true
-    })
+    isOver: monitor.isOver()
   };
 }
 
@@ -397,45 +318,28 @@ var DraggableTreeItem = (0, _reactDnd.DropTarget)(Types['TreeNode'], target, tar
 
 var HOCTreeItem = function HOCTreeItem(TreeItemComponent) {
   return /*#__PURE__*/function (_Component2) {
-    (0, _inherits2["default"])(WrapperTreeItem, _Component2);
+    _inherits(WrapperTreeItem, _Component2);
 
     var _super2 = _createSuper(WrapperTreeItem);
 
     function WrapperTreeItem() {
-      var _this3;
-
-      (0, _classCallCheck2["default"])(this, WrapperTreeItem);
+      _classCallCheck(this, WrapperTreeItem);
 
-      for (var _len2 = arguments.length, args = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
-        args[_key2] = arguments[_key2];
-      }
-
-      _this3 = _super2.call.apply(_super2, [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;
+      return _super2.apply(this, arguments);
     }
 
-    (0, _createClass2["default"])(WrapperTreeItem, [{
+    _createClass(WrapperTreeItem, [{
       key: "render",
       value: function render() {
         var draggable = this.props.draggable;
-        return draggable ? /*#__PURE__*/_react["default"].createElement(DraggableTreeItem, (0, _extends2["default"])({}, this.props, {
-          direction: this.state.direction,
-          setDirection: this.setDirection
-        })) : /*#__PURE__*/_react["default"].createElement(TreeItemComponent, this.props);
+        return draggable ? /*#__PURE__*/_react["default"].createElement(DraggableTreeItem, this.props) : /*#__PURE__*/_react["default"].createElement(TreeItemComponent, this.props);
       }
     }]);
+
     return WrapperTreeItem;
   }(_react.Component);
 };
 
-var _default = (0, _context["default"])(HOCTreeItem(TreeItem));
+var _default = HOCTreeItem(TreeItem);
 
 exports["default"] = _default;