Tree.js 51 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. var _typeof3 = require("@babel/runtime/helpers/typeof");
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = void 0;
  8. var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
  9. var _typeof2 = _interopRequireDefault(require("@babel/runtime/helpers/typeof"));
  10. var _objectSpread3 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
  11. var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
  12. var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));
  13. var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));
  14. var _assertThisInitialized2 = _interopRequireDefault(require("@babel/runtime/helpers/assertThisInitialized"));
  15. var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));
  16. var _createSuper2 = _interopRequireDefault(require("@babel/runtime/helpers/createSuper"));
  17. var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
  18. var _classnames = _interopRequireDefault(require("classnames"));
  19. var _KeyCode = _interopRequireDefault(require("rc-util/lib/KeyCode"));
  20. var _pickAttrs = _interopRequireDefault(require("rc-util/lib/pickAttrs"));
  21. var _warning = _interopRequireDefault(require("rc-util/lib/warning"));
  22. var React = _interopRequireWildcard(require("react"));
  23. var _contextTypes = require("./contextTypes");
  24. var _DropIndicator = _interopRequireDefault(require("./DropIndicator"));
  25. var _NodeList = _interopRequireWildcard(require("./NodeList"));
  26. var _TreeNode = _interopRequireDefault(require("./TreeNode"));
  27. var _util = require("./util");
  28. var _conductUtil = require("./utils/conductUtil");
  29. var _keyUtil = _interopRequireDefault(require("./utils/keyUtil"));
  30. var _treeUtil = require("./utils/treeUtil");
  31. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
  32. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof3(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  33. // TODO: https://www.w3.org/TR/2017/NOTE-wai-aria-practices-1.1-20171214/examples/treeview/treeview-2/treeview-2a.html
  34. // Fully accessibility support
  35. var MAX_RETRY_TIMES = 10;
  36. var Tree = /*#__PURE__*/function (_React$Component) {
  37. (0, _inherits2.default)(Tree, _React$Component);
  38. var _super = (0, _createSuper2.default)(Tree);
  39. function Tree() {
  40. var _this;
  41. (0, _classCallCheck2.default)(this, Tree);
  42. for (var _len = arguments.length, _args = new Array(_len), _key = 0; _key < _len; _key++) {
  43. _args[_key] = arguments[_key];
  44. }
  45. _this = _super.call.apply(_super, [this].concat(_args));
  46. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "destroyed", false);
  47. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "delayedDragEnterLogic", void 0);
  48. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "loadingRetryTimes", {});
  49. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "state", {
  50. keyEntities: {},
  51. indent: null,
  52. selectedKeys: [],
  53. checkedKeys: [],
  54. halfCheckedKeys: [],
  55. loadedKeys: [],
  56. loadingKeys: [],
  57. expandedKeys: [],
  58. draggingNodeKey: null,
  59. dragChildrenKeys: [],
  60. // dropTargetKey is the key of abstract-drop-node
  61. // the abstract-drop-node is the real drop node when drag and drop
  62. // not the DOM drag over node
  63. dropTargetKey: null,
  64. dropPosition: null,
  65. // the drop position of abstract-drop-node, inside 0, top -1, bottom 1
  66. dropContainerKey: null,
  67. // the container key of abstract-drop-node if dropPosition is -1 or 1
  68. dropLevelOffset: null,
  69. // the drop level offset of abstract-drag-over-node
  70. dropTargetPos: null,
  71. // the pos of abstract-drop-node
  72. dropAllowed: true,
  73. // if drop to abstract-drop-node is allowed
  74. // the abstract-drag-over-node
  75. // if mouse is on the bottom of top dom node or no the top of the bottom dom node
  76. // abstract-drag-over-node is the top node
  77. dragOverNodeKey: null,
  78. treeData: [],
  79. flattenNodes: [],
  80. focused: false,
  81. activeKey: null,
  82. listChanging: false,
  83. prevProps: null,
  84. fieldNames: (0, _treeUtil.fillFieldNames)()
  85. });
  86. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "dragStartMousePosition", null);
  87. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "dragNodeProps", null);
  88. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "currentMouseOverDroppableNodeKey", null);
  89. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "listRef", /*#__PURE__*/React.createRef());
  90. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDragStart", function (event, nodeProps) {
  91. var _this$state = _this.state,
  92. expandedKeys = _this$state.expandedKeys,
  93. keyEntities = _this$state.keyEntities;
  94. var onDragStart = _this.props.onDragStart;
  95. var eventKey = nodeProps.eventKey;
  96. _this.dragNodeProps = nodeProps;
  97. _this.dragStartMousePosition = {
  98. x: event.clientX,
  99. y: event.clientY
  100. };
  101. var newExpandedKeys = (0, _util.arrDel)(expandedKeys, eventKey);
  102. _this.setState({
  103. draggingNodeKey: eventKey,
  104. dragChildrenKeys: (0, _util.getDragChildrenKeys)(eventKey, keyEntities),
  105. indent: _this.listRef.current.getIndentWidth()
  106. });
  107. _this.setExpandedKeys(newExpandedKeys);
  108. window.addEventListener('dragend', _this.onWindowDragEnd);
  109. onDragStart === null || onDragStart === void 0 || onDragStart({
  110. event: event,
  111. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps)
  112. });
  113. });
  114. /**
  115. * [Legacy] Select handler is smaller than node,
  116. * so that this will trigger when drag enter node or select handler.
  117. * This is a little tricky if customize css without padding.
  118. * Better for use mouse move event to refresh drag state.
  119. * But let's just keep it to avoid event trigger logic change.
  120. */
  121. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDragEnter", function (event, nodeProps) {
  122. var _this$state2 = _this.state,
  123. expandedKeys = _this$state2.expandedKeys,
  124. keyEntities = _this$state2.keyEntities,
  125. dragChildrenKeys = _this$state2.dragChildrenKeys,
  126. flattenNodes = _this$state2.flattenNodes,
  127. indent = _this$state2.indent;
  128. var _this$props = _this.props,
  129. onDragEnter = _this$props.onDragEnter,
  130. onExpand = _this$props.onExpand,
  131. allowDrop = _this$props.allowDrop,
  132. direction = _this$props.direction;
  133. var pos = nodeProps.pos,
  134. eventKey = nodeProps.eventKey;
  135. // record the key of node which is latest entered, used in dragleave event.
  136. if (_this.currentMouseOverDroppableNodeKey !== eventKey) {
  137. _this.currentMouseOverDroppableNodeKey = eventKey;
  138. }
  139. if (!_this.dragNodeProps) {
  140. _this.resetDragState();
  141. return;
  142. }
  143. var _calcDropPosition = (0, _util.calcDropPosition)(event, _this.dragNodeProps, nodeProps, indent, _this.dragStartMousePosition, allowDrop, flattenNodes, keyEntities, expandedKeys, direction),
  144. dropPosition = _calcDropPosition.dropPosition,
  145. dropLevelOffset = _calcDropPosition.dropLevelOffset,
  146. dropTargetKey = _calcDropPosition.dropTargetKey,
  147. dropContainerKey = _calcDropPosition.dropContainerKey,
  148. dropTargetPos = _calcDropPosition.dropTargetPos,
  149. dropAllowed = _calcDropPosition.dropAllowed,
  150. dragOverNodeKey = _calcDropPosition.dragOverNodeKey;
  151. if (
  152. // don't allow drop inside its children
  153. dragChildrenKeys.includes(dropTargetKey) ||
  154. // don't allow drop when drop is not allowed caculated by calcDropPosition
  155. !dropAllowed) {
  156. _this.resetDragState();
  157. return;
  158. }
  159. // Side effect for delay drag
  160. if (!_this.delayedDragEnterLogic) {
  161. _this.delayedDragEnterLogic = {};
  162. }
  163. Object.keys(_this.delayedDragEnterLogic).forEach(function (key) {
  164. clearTimeout(_this.delayedDragEnterLogic[key]);
  165. });
  166. if (_this.dragNodeProps.eventKey !== nodeProps.eventKey) {
  167. // hoist expand logic here
  168. // since if logic is on the bottom
  169. // it will be blocked by abstract dragover node check
  170. // => if you dragenter from top, you mouse will still be consider as in the top node
  171. event.persist();
  172. _this.delayedDragEnterLogic[pos] = window.setTimeout(function () {
  173. if (_this.state.draggingNodeKey === null) {
  174. return;
  175. }
  176. var newExpandedKeys = (0, _toConsumableArray2.default)(expandedKeys);
  177. var entity = (0, _keyUtil.default)(keyEntities, nodeProps.eventKey);
  178. if (entity && (entity.children || []).length) {
  179. newExpandedKeys = (0, _util.arrAdd)(expandedKeys, nodeProps.eventKey);
  180. }
  181. if (!_this.props.hasOwnProperty('expandedKeys')) {
  182. _this.setExpandedKeys(newExpandedKeys);
  183. }
  184. onExpand === null || onExpand === void 0 || onExpand(newExpandedKeys, {
  185. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps),
  186. expanded: true,
  187. nativeEvent: event.nativeEvent
  188. });
  189. }, 800);
  190. }
  191. // Skip if drag node is self
  192. if (_this.dragNodeProps.eventKey === dropTargetKey && dropLevelOffset === 0) {
  193. _this.resetDragState();
  194. return;
  195. }
  196. // Update drag over node and drag state
  197. _this.setState({
  198. dragOverNodeKey: dragOverNodeKey,
  199. dropPosition: dropPosition,
  200. dropLevelOffset: dropLevelOffset,
  201. dropTargetKey: dropTargetKey,
  202. dropContainerKey: dropContainerKey,
  203. dropTargetPos: dropTargetPos,
  204. dropAllowed: dropAllowed
  205. });
  206. onDragEnter === null || onDragEnter === void 0 || onDragEnter({
  207. event: event,
  208. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps),
  209. expandedKeys: expandedKeys
  210. });
  211. });
  212. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDragOver", function (event, nodeProps) {
  213. var _this$state3 = _this.state,
  214. dragChildrenKeys = _this$state3.dragChildrenKeys,
  215. flattenNodes = _this$state3.flattenNodes,
  216. keyEntities = _this$state3.keyEntities,
  217. expandedKeys = _this$state3.expandedKeys,
  218. indent = _this$state3.indent;
  219. var _this$props2 = _this.props,
  220. onDragOver = _this$props2.onDragOver,
  221. allowDrop = _this$props2.allowDrop,
  222. direction = _this$props2.direction;
  223. if (!_this.dragNodeProps) {
  224. return;
  225. }
  226. var _calcDropPosition2 = (0, _util.calcDropPosition)(event, _this.dragNodeProps, nodeProps, indent, _this.dragStartMousePosition, allowDrop, flattenNodes, keyEntities, expandedKeys, direction),
  227. dropPosition = _calcDropPosition2.dropPosition,
  228. dropLevelOffset = _calcDropPosition2.dropLevelOffset,
  229. dropTargetKey = _calcDropPosition2.dropTargetKey,
  230. dropContainerKey = _calcDropPosition2.dropContainerKey,
  231. dropTargetPos = _calcDropPosition2.dropTargetPos,
  232. dropAllowed = _calcDropPosition2.dropAllowed,
  233. dragOverNodeKey = _calcDropPosition2.dragOverNodeKey;
  234. if (dragChildrenKeys.includes(dropTargetKey) || !dropAllowed) {
  235. // don't allow drop inside its children
  236. // don't allow drop when drop is not allowed calculated by calcDropPosition
  237. return;
  238. }
  239. // Update drag position
  240. if (_this.dragNodeProps.eventKey === dropTargetKey && dropLevelOffset === 0) {
  241. if (!(_this.state.dropPosition === null && _this.state.dropLevelOffset === null && _this.state.dropTargetKey === null && _this.state.dropContainerKey === null && _this.state.dropTargetPos === null && _this.state.dropAllowed === false && _this.state.dragOverNodeKey === null)) {
  242. _this.resetDragState();
  243. }
  244. } else if (!(dropPosition === _this.state.dropPosition && dropLevelOffset === _this.state.dropLevelOffset && dropTargetKey === _this.state.dropTargetKey && dropContainerKey === _this.state.dropContainerKey && dropTargetPos === _this.state.dropTargetPos && dropAllowed === _this.state.dropAllowed && dragOverNodeKey === _this.state.dragOverNodeKey)) {
  245. _this.setState({
  246. dropPosition: dropPosition,
  247. dropLevelOffset: dropLevelOffset,
  248. dropTargetKey: dropTargetKey,
  249. dropContainerKey: dropContainerKey,
  250. dropTargetPos: dropTargetPos,
  251. dropAllowed: dropAllowed,
  252. dragOverNodeKey: dragOverNodeKey
  253. });
  254. }
  255. onDragOver === null || onDragOver === void 0 || onDragOver({
  256. event: event,
  257. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps)
  258. });
  259. });
  260. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDragLeave", function (event, nodeProps) {
  261. // if it is outside the droppable area
  262. // currentMouseOverDroppableNodeKey will be updated in dragenter event when into another droppable receiver.
  263. if (_this.currentMouseOverDroppableNodeKey === nodeProps.eventKey && !event.currentTarget.contains(event.relatedTarget)) {
  264. _this.resetDragState();
  265. _this.currentMouseOverDroppableNodeKey = null;
  266. }
  267. var onDragLeave = _this.props.onDragLeave;
  268. onDragLeave === null || onDragLeave === void 0 || onDragLeave({
  269. event: event,
  270. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps)
  271. });
  272. });
  273. // since stopPropagation() is called in treeNode
  274. // if onWindowDrag is called, whice means state is keeped, drag state should be cleared
  275. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onWindowDragEnd", function (event) {
  276. _this.onNodeDragEnd(event, null, true);
  277. window.removeEventListener('dragend', _this.onWindowDragEnd);
  278. });
  279. // if onNodeDragEnd is called, onWindowDragEnd won't be called since stopPropagation() is called
  280. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDragEnd", function (event, nodeProps) {
  281. var onDragEnd = _this.props.onDragEnd;
  282. _this.setState({
  283. dragOverNodeKey: null
  284. });
  285. _this.cleanDragState();
  286. onDragEnd === null || onDragEnd === void 0 || onDragEnd({
  287. event: event,
  288. node: (0, _treeUtil.convertNodePropsToEventData)(nodeProps)
  289. });
  290. _this.dragNodeProps = null;
  291. window.removeEventListener('dragend', _this.onWindowDragEnd);
  292. });
  293. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDrop", function (event, _) {
  294. var _this$getActiveItem;
  295. var outsideTree = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
  296. var _this$state4 = _this.state,
  297. dragChildrenKeys = _this$state4.dragChildrenKeys,
  298. dropPosition = _this$state4.dropPosition,
  299. dropTargetKey = _this$state4.dropTargetKey,
  300. dropTargetPos = _this$state4.dropTargetPos,
  301. dropAllowed = _this$state4.dropAllowed;
  302. if (!dropAllowed) {
  303. return;
  304. }
  305. var onDrop = _this.props.onDrop;
  306. _this.setState({
  307. dragOverNodeKey: null
  308. });
  309. _this.cleanDragState();
  310. if (dropTargetKey === null) return;
  311. var abstractDropNodeProps = (0, _objectSpread3.default)((0, _objectSpread3.default)({}, (0, _treeUtil.getTreeNodeProps)(dropTargetKey, _this.getTreeNodeRequiredProps())), {}, {
  312. active: ((_this$getActiveItem = _this.getActiveItem()) === null || _this$getActiveItem === void 0 ? void 0 : _this$getActiveItem.key) === dropTargetKey,
  313. data: (0, _keyUtil.default)(_this.state.keyEntities, dropTargetKey).node
  314. });
  315. var dropToChild = dragChildrenKeys.includes(dropTargetKey);
  316. (0, _warning.default)(!dropToChild, "Can not drop to dragNode's children node. This is a bug of rc-tree. Please report an issue.");
  317. var posArr = (0, _util.posToArr)(dropTargetPos);
  318. var dropResult = {
  319. event: event,
  320. node: (0, _treeUtil.convertNodePropsToEventData)(abstractDropNodeProps),
  321. dragNode: _this.dragNodeProps ? (0, _treeUtil.convertNodePropsToEventData)(_this.dragNodeProps) : null,
  322. dragNodesKeys: [_this.dragNodeProps.eventKey].concat(dragChildrenKeys),
  323. dropToGap: dropPosition !== 0,
  324. dropPosition: dropPosition + Number(posArr[posArr.length - 1])
  325. };
  326. if (!outsideTree) {
  327. onDrop === null || onDrop === void 0 || onDrop(dropResult);
  328. }
  329. _this.dragNodeProps = null;
  330. });
  331. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "cleanDragState", function () {
  332. var draggingNodeKey = _this.state.draggingNodeKey;
  333. if (draggingNodeKey !== null) {
  334. _this.setState({
  335. draggingNodeKey: null,
  336. dropPosition: null,
  337. dropContainerKey: null,
  338. dropTargetKey: null,
  339. dropLevelOffset: null,
  340. dropAllowed: true,
  341. dragOverNodeKey: null
  342. });
  343. }
  344. _this.dragStartMousePosition = null;
  345. _this.currentMouseOverDroppableNodeKey = null;
  346. });
  347. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "triggerExpandActionExpand", function (e, treeNode) {
  348. var _this$state5 = _this.state,
  349. expandedKeys = _this$state5.expandedKeys,
  350. flattenNodes = _this$state5.flattenNodes;
  351. var expanded = treeNode.expanded,
  352. key = treeNode.key,
  353. isLeaf = treeNode.isLeaf;
  354. if (isLeaf || e.shiftKey || e.metaKey || e.ctrlKey) {
  355. return;
  356. }
  357. var node = flattenNodes.filter(function (nodeItem) {
  358. return nodeItem.key === key;
  359. })[0];
  360. var eventNode = (0, _treeUtil.convertNodePropsToEventData)((0, _objectSpread3.default)((0, _objectSpread3.default)({}, (0, _treeUtil.getTreeNodeProps)(key, _this.getTreeNodeRequiredProps())), {}, {
  361. data: node.data
  362. }));
  363. _this.setExpandedKeys(expanded ? (0, _util.arrDel)(expandedKeys, key) : (0, _util.arrAdd)(expandedKeys, key));
  364. _this.onNodeExpand(e, eventNode);
  365. });
  366. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeClick", function (e, treeNode) {
  367. var _this$props3 = _this.props,
  368. onClick = _this$props3.onClick,
  369. expandAction = _this$props3.expandAction;
  370. if (expandAction === 'click') {
  371. _this.triggerExpandActionExpand(e, treeNode);
  372. }
  373. onClick === null || onClick === void 0 || onClick(e, treeNode);
  374. });
  375. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeDoubleClick", function (e, treeNode) {
  376. var _this$props4 = _this.props,
  377. onDoubleClick = _this$props4.onDoubleClick,
  378. expandAction = _this$props4.expandAction;
  379. if (expandAction === 'doubleClick') {
  380. _this.triggerExpandActionExpand(e, treeNode);
  381. }
  382. onDoubleClick === null || onDoubleClick === void 0 || onDoubleClick(e, treeNode);
  383. });
  384. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeSelect", function (e, treeNode) {
  385. var selectedKeys = _this.state.selectedKeys;
  386. var _this$state6 = _this.state,
  387. keyEntities = _this$state6.keyEntities,
  388. fieldNames = _this$state6.fieldNames;
  389. var _this$props5 = _this.props,
  390. onSelect = _this$props5.onSelect,
  391. multiple = _this$props5.multiple;
  392. var selected = treeNode.selected;
  393. var key = treeNode[fieldNames.key];
  394. var targetSelected = !selected;
  395. // Update selected keys
  396. if (!targetSelected) {
  397. selectedKeys = (0, _util.arrDel)(selectedKeys, key);
  398. } else if (!multiple) {
  399. selectedKeys = [key];
  400. } else {
  401. selectedKeys = (0, _util.arrAdd)(selectedKeys, key);
  402. }
  403. // [Legacy] Not found related usage in doc or upper libs
  404. var selectedNodes = selectedKeys.map(function (selectedKey) {
  405. var entity = (0, _keyUtil.default)(keyEntities, selectedKey);
  406. return entity ? entity.node : null;
  407. }).filter(Boolean);
  408. _this.setUncontrolledState({
  409. selectedKeys: selectedKeys
  410. });
  411. onSelect === null || onSelect === void 0 || onSelect(selectedKeys, {
  412. event: 'select',
  413. selected: targetSelected,
  414. node: treeNode,
  415. selectedNodes: selectedNodes,
  416. nativeEvent: e.nativeEvent
  417. });
  418. });
  419. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeCheck", function (e, treeNode, checked) {
  420. var _this$state7 = _this.state,
  421. keyEntities = _this$state7.keyEntities,
  422. oriCheckedKeys = _this$state7.checkedKeys,
  423. oriHalfCheckedKeys = _this$state7.halfCheckedKeys;
  424. var _this$props6 = _this.props,
  425. checkStrictly = _this$props6.checkStrictly,
  426. onCheck = _this$props6.onCheck;
  427. var key = treeNode.key;
  428. // Prepare trigger arguments
  429. var checkedObj;
  430. var eventObj = {
  431. event: 'check',
  432. node: treeNode,
  433. checked: checked,
  434. nativeEvent: e.nativeEvent
  435. };
  436. if (checkStrictly) {
  437. var checkedKeys = checked ? (0, _util.arrAdd)(oriCheckedKeys, key) : (0, _util.arrDel)(oriCheckedKeys, key);
  438. var halfCheckedKeys = (0, _util.arrDel)(oriHalfCheckedKeys, key);
  439. checkedObj = {
  440. checked: checkedKeys,
  441. halfChecked: halfCheckedKeys
  442. };
  443. eventObj.checkedNodes = checkedKeys.map(function (checkedKey) {
  444. return (0, _keyUtil.default)(keyEntities, checkedKey);
  445. }).filter(Boolean).map(function (entity) {
  446. return entity.node;
  447. });
  448. _this.setUncontrolledState({
  449. checkedKeys: checkedKeys
  450. });
  451. } else {
  452. // Always fill first
  453. var _conductCheck = (0, _conductUtil.conductCheck)([].concat((0, _toConsumableArray2.default)(oriCheckedKeys), [key]), true, keyEntities),
  454. _checkedKeys = _conductCheck.checkedKeys,
  455. _halfCheckedKeys = _conductCheck.halfCheckedKeys;
  456. // If remove, we do it again to correction
  457. if (!checked) {
  458. var keySet = new Set(_checkedKeys);
  459. keySet.delete(key);
  460. var _conductCheck2 = (0, _conductUtil.conductCheck)(Array.from(keySet), {
  461. checked: false,
  462. halfCheckedKeys: _halfCheckedKeys
  463. }, keyEntities);
  464. _checkedKeys = _conductCheck2.checkedKeys;
  465. _halfCheckedKeys = _conductCheck2.halfCheckedKeys;
  466. }
  467. checkedObj = _checkedKeys;
  468. // [Legacy] This is used for `rc-tree-select`
  469. eventObj.checkedNodes = [];
  470. eventObj.checkedNodesPositions = [];
  471. eventObj.halfCheckedKeys = _halfCheckedKeys;
  472. _checkedKeys.forEach(function (checkedKey) {
  473. var entity = (0, _keyUtil.default)(keyEntities, checkedKey);
  474. if (!entity) return;
  475. var node = entity.node,
  476. pos = entity.pos;
  477. eventObj.checkedNodes.push(node);
  478. eventObj.checkedNodesPositions.push({
  479. node: node,
  480. pos: pos
  481. });
  482. });
  483. _this.setUncontrolledState({
  484. checkedKeys: _checkedKeys
  485. }, false, {
  486. halfCheckedKeys: _halfCheckedKeys
  487. });
  488. }
  489. onCheck === null || onCheck === void 0 || onCheck(checkedObj, eventObj);
  490. });
  491. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeLoad", function (treeNode) {
  492. var _entity$children;
  493. var key = treeNode.key;
  494. var keyEntities = _this.state.keyEntities;
  495. // Skip if has children already
  496. var entity = (0, _keyUtil.default)(keyEntities, key);
  497. if (entity !== null && entity !== void 0 && (_entity$children = entity.children) !== null && _entity$children !== void 0 && _entity$children.length) {
  498. return;
  499. }
  500. var loadPromise = new Promise(function (resolve, reject) {
  501. // We need to get the latest state of loading/loaded keys
  502. _this.setState(function (_ref) {
  503. var _ref$loadedKeys = _ref.loadedKeys,
  504. loadedKeys = _ref$loadedKeys === void 0 ? [] : _ref$loadedKeys,
  505. _ref$loadingKeys = _ref.loadingKeys,
  506. loadingKeys = _ref$loadingKeys === void 0 ? [] : _ref$loadingKeys;
  507. var _this$props7 = _this.props,
  508. loadData = _this$props7.loadData,
  509. onLoad = _this$props7.onLoad;
  510. if (!loadData || loadedKeys.includes(key) || loadingKeys.includes(key)) {
  511. return null;
  512. }
  513. // Process load data
  514. var promise = loadData(treeNode);
  515. promise.then(function () {
  516. var currentLoadedKeys = _this.state.loadedKeys;
  517. var newLoadedKeys = (0, _util.arrAdd)(currentLoadedKeys, key);
  518. // onLoad should trigger before internal setState to avoid `loadData` trigger twice.
  519. // https://github.com/ant-design/ant-design/issues/12464
  520. onLoad === null || onLoad === void 0 || onLoad(newLoadedKeys, {
  521. event: 'load',
  522. node: treeNode
  523. });
  524. _this.setUncontrolledState({
  525. loadedKeys: newLoadedKeys
  526. });
  527. _this.setState(function (prevState) {
  528. return {
  529. loadingKeys: (0, _util.arrDel)(prevState.loadingKeys, key)
  530. };
  531. });
  532. resolve();
  533. }).catch(function (e) {
  534. _this.setState(function (prevState) {
  535. return {
  536. loadingKeys: (0, _util.arrDel)(prevState.loadingKeys, key)
  537. };
  538. });
  539. // If exceed max retry times, we give up retry
  540. _this.loadingRetryTimes[key] = (_this.loadingRetryTimes[key] || 0) + 1;
  541. if (_this.loadingRetryTimes[key] >= MAX_RETRY_TIMES) {
  542. var currentLoadedKeys = _this.state.loadedKeys;
  543. (0, _warning.default)(false, 'Retry for `loadData` many times but still failed. No more retry.');
  544. _this.setUncontrolledState({
  545. loadedKeys: (0, _util.arrAdd)(currentLoadedKeys, key)
  546. });
  547. resolve();
  548. }
  549. reject(e);
  550. });
  551. return {
  552. loadingKeys: (0, _util.arrAdd)(loadingKeys, key)
  553. };
  554. });
  555. });
  556. // Not care warning if we ignore this
  557. loadPromise.catch(function () {});
  558. return loadPromise;
  559. });
  560. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeMouseEnter", function (event, node) {
  561. var onMouseEnter = _this.props.onMouseEnter;
  562. onMouseEnter === null || onMouseEnter === void 0 || onMouseEnter({
  563. event: event,
  564. node: node
  565. });
  566. });
  567. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeMouseLeave", function (event, node) {
  568. var onMouseLeave = _this.props.onMouseLeave;
  569. onMouseLeave === null || onMouseLeave === void 0 || onMouseLeave({
  570. event: event,
  571. node: node
  572. });
  573. });
  574. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeContextMenu", function (event, node) {
  575. var onRightClick = _this.props.onRightClick;
  576. if (onRightClick) {
  577. event.preventDefault();
  578. onRightClick({
  579. event: event,
  580. node: node
  581. });
  582. }
  583. });
  584. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onFocus", function () {
  585. var onFocus = _this.props.onFocus;
  586. _this.setState({
  587. focused: true
  588. });
  589. for (var _len2 = arguments.length, args = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
  590. args[_key2] = arguments[_key2];
  591. }
  592. onFocus === null || onFocus === void 0 || onFocus.apply(void 0, args);
  593. });
  594. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onBlur", function () {
  595. var onBlur = _this.props.onBlur;
  596. _this.setState({
  597. focused: false
  598. });
  599. _this.onActiveChange(null);
  600. for (var _len3 = arguments.length, args = new Array(_len3), _key3 = 0; _key3 < _len3; _key3++) {
  601. args[_key3] = arguments[_key3];
  602. }
  603. onBlur === null || onBlur === void 0 || onBlur.apply(void 0, args);
  604. });
  605. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "getTreeNodeRequiredProps", function () {
  606. var _this$state8 = _this.state,
  607. expandedKeys = _this$state8.expandedKeys,
  608. selectedKeys = _this$state8.selectedKeys,
  609. loadedKeys = _this$state8.loadedKeys,
  610. loadingKeys = _this$state8.loadingKeys,
  611. checkedKeys = _this$state8.checkedKeys,
  612. halfCheckedKeys = _this$state8.halfCheckedKeys,
  613. dragOverNodeKey = _this$state8.dragOverNodeKey,
  614. dropPosition = _this$state8.dropPosition,
  615. keyEntities = _this$state8.keyEntities;
  616. return {
  617. expandedKeys: expandedKeys || [],
  618. selectedKeys: selectedKeys || [],
  619. loadedKeys: loadedKeys || [],
  620. loadingKeys: loadingKeys || [],
  621. checkedKeys: checkedKeys || [],
  622. halfCheckedKeys: halfCheckedKeys || [],
  623. dragOverNodeKey: dragOverNodeKey,
  624. dropPosition: dropPosition,
  625. keyEntities: keyEntities
  626. };
  627. });
  628. // =========================== Expanded ===========================
  629. /** Set uncontrolled `expandedKeys`. This will also auto update `flattenNodes`. */
  630. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "setExpandedKeys", function (expandedKeys) {
  631. var _this$state9 = _this.state,
  632. treeData = _this$state9.treeData,
  633. fieldNames = _this$state9.fieldNames;
  634. var flattenNodes = (0, _treeUtil.flattenTreeData)(treeData, expandedKeys, fieldNames);
  635. _this.setUncontrolledState({
  636. expandedKeys: expandedKeys,
  637. flattenNodes: flattenNodes
  638. }, true);
  639. });
  640. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onNodeExpand", function (e, treeNode) {
  641. var expandedKeys = _this.state.expandedKeys;
  642. var _this$state10 = _this.state,
  643. listChanging = _this$state10.listChanging,
  644. fieldNames = _this$state10.fieldNames;
  645. var _this$props8 = _this.props,
  646. onExpand = _this$props8.onExpand,
  647. loadData = _this$props8.loadData;
  648. var expanded = treeNode.expanded;
  649. var key = treeNode[fieldNames.key];
  650. // Do nothing when motion is in progress
  651. if (listChanging) {
  652. return;
  653. }
  654. // Update selected keys
  655. var certain = expandedKeys.includes(key);
  656. var targetExpanded = !expanded;
  657. (0, _warning.default)(expanded && certain || !expanded && !certain, 'Expand state not sync with index check');
  658. expandedKeys = targetExpanded ? (0, _util.arrAdd)(expandedKeys, key) : (0, _util.arrDel)(expandedKeys, key);
  659. _this.setExpandedKeys(expandedKeys);
  660. onExpand === null || onExpand === void 0 || onExpand(expandedKeys, {
  661. node: treeNode,
  662. expanded: targetExpanded,
  663. nativeEvent: e.nativeEvent
  664. });
  665. // Async Load data
  666. if (targetExpanded && loadData) {
  667. var loadPromise = _this.onNodeLoad(treeNode);
  668. if (loadPromise) {
  669. loadPromise.then(function () {
  670. // [Legacy] Refresh logic
  671. var newFlattenTreeData = (0, _treeUtil.flattenTreeData)(_this.state.treeData, expandedKeys, fieldNames);
  672. _this.setUncontrolledState({
  673. flattenNodes: newFlattenTreeData
  674. });
  675. }).catch(function () {
  676. var currentExpandedKeys = _this.state.expandedKeys;
  677. var expandedKeysToRestore = (0, _util.arrDel)(currentExpandedKeys, key);
  678. _this.setExpandedKeys(expandedKeysToRestore);
  679. });
  680. }
  681. }
  682. });
  683. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onListChangeStart", function () {
  684. _this.setUncontrolledState({
  685. listChanging: true
  686. });
  687. });
  688. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onListChangeEnd", function () {
  689. setTimeout(function () {
  690. _this.setUncontrolledState({
  691. listChanging: false
  692. });
  693. });
  694. });
  695. // =========================== Keyboard ===========================
  696. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onActiveChange", function (newActiveKey) {
  697. var activeKey = _this.state.activeKey;
  698. var _this$props9 = _this.props,
  699. onActiveChange = _this$props9.onActiveChange,
  700. _this$props9$itemScro = _this$props9.itemScrollOffset,
  701. itemScrollOffset = _this$props9$itemScro === void 0 ? 0 : _this$props9$itemScro;
  702. if (activeKey === newActiveKey) {
  703. return;
  704. }
  705. _this.setState({
  706. activeKey: newActiveKey
  707. });
  708. if (newActiveKey !== null) {
  709. _this.scrollTo({
  710. key: newActiveKey,
  711. offset: itemScrollOffset
  712. });
  713. }
  714. onActiveChange === null || onActiveChange === void 0 || onActiveChange(newActiveKey);
  715. });
  716. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "getActiveItem", function () {
  717. var _this$state11 = _this.state,
  718. activeKey = _this$state11.activeKey,
  719. flattenNodes = _this$state11.flattenNodes;
  720. if (activeKey === null) {
  721. return null;
  722. }
  723. return flattenNodes.find(function (_ref2) {
  724. var key = _ref2.key;
  725. return key === activeKey;
  726. }) || null;
  727. });
  728. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "offsetActiveKey", function (offset) {
  729. var _this$state12 = _this.state,
  730. flattenNodes = _this$state12.flattenNodes,
  731. activeKey = _this$state12.activeKey;
  732. var index = flattenNodes.findIndex(function (_ref3) {
  733. var key = _ref3.key;
  734. return key === activeKey;
  735. });
  736. // Align with index
  737. if (index === -1 && offset < 0) {
  738. index = flattenNodes.length;
  739. }
  740. index = (index + offset + flattenNodes.length) % flattenNodes.length;
  741. var item = flattenNodes[index];
  742. if (item) {
  743. var _key4 = item.key;
  744. _this.onActiveChange(_key4);
  745. } else {
  746. _this.onActiveChange(null);
  747. }
  748. });
  749. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "onKeyDown", function (event) {
  750. var _this$state13 = _this.state,
  751. activeKey = _this$state13.activeKey,
  752. expandedKeys = _this$state13.expandedKeys,
  753. checkedKeys = _this$state13.checkedKeys,
  754. fieldNames = _this$state13.fieldNames;
  755. var _this$props10 = _this.props,
  756. onKeyDown = _this$props10.onKeyDown,
  757. checkable = _this$props10.checkable,
  758. selectable = _this$props10.selectable;
  759. // >>>>>>>>>> Direction
  760. switch (event.which) {
  761. case _KeyCode.default.UP:
  762. {
  763. _this.offsetActiveKey(-1);
  764. event.preventDefault();
  765. break;
  766. }
  767. case _KeyCode.default.DOWN:
  768. {
  769. _this.offsetActiveKey(1);
  770. event.preventDefault();
  771. break;
  772. }
  773. }
  774. // >>>>>>>>>> Expand & Selection
  775. var activeItem = _this.getActiveItem();
  776. if (activeItem && activeItem.data) {
  777. var treeNodeRequiredProps = _this.getTreeNodeRequiredProps();
  778. var expandable = activeItem.data.isLeaf === false || !!(activeItem.data[fieldNames.children] || []).length;
  779. var eventNode = (0, _treeUtil.convertNodePropsToEventData)((0, _objectSpread3.default)((0, _objectSpread3.default)({}, (0, _treeUtil.getTreeNodeProps)(activeKey, treeNodeRequiredProps)), {}, {
  780. data: activeItem.data,
  781. active: true
  782. }));
  783. switch (event.which) {
  784. // >>> Expand
  785. case _KeyCode.default.LEFT:
  786. {
  787. // Collapse if possible
  788. if (expandable && expandedKeys.includes(activeKey)) {
  789. _this.onNodeExpand({}, eventNode);
  790. } else if (activeItem.parent) {
  791. _this.onActiveChange(activeItem.parent.key);
  792. }
  793. event.preventDefault();
  794. break;
  795. }
  796. case _KeyCode.default.RIGHT:
  797. {
  798. // Expand if possible
  799. if (expandable && !expandedKeys.includes(activeKey)) {
  800. _this.onNodeExpand({}, eventNode);
  801. } else if (activeItem.children && activeItem.children.length) {
  802. _this.onActiveChange(activeItem.children[0].key);
  803. }
  804. event.preventDefault();
  805. break;
  806. }
  807. // Selection
  808. case _KeyCode.default.ENTER:
  809. case _KeyCode.default.SPACE:
  810. {
  811. if (checkable && !eventNode.disabled && eventNode.checkable !== false && !eventNode.disableCheckbox) {
  812. _this.onNodeCheck({}, eventNode, !checkedKeys.includes(activeKey));
  813. } else if (!checkable && selectable && !eventNode.disabled && eventNode.selectable !== false) {
  814. _this.onNodeSelect({}, eventNode);
  815. }
  816. break;
  817. }
  818. }
  819. }
  820. onKeyDown === null || onKeyDown === void 0 || onKeyDown(event);
  821. });
  822. /**
  823. * Only update the value which is not in props
  824. */
  825. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "setUncontrolledState", function (state) {
  826. var atomic = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false;
  827. var forceState = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : null;
  828. if (!_this.destroyed) {
  829. var needSync = false;
  830. var allPassed = true;
  831. var newState = {};
  832. Object.keys(state).forEach(function (name) {
  833. if (_this.props.hasOwnProperty(name)) {
  834. allPassed = false;
  835. return;
  836. }
  837. needSync = true;
  838. newState[name] = state[name];
  839. });
  840. if (needSync && (!atomic || allPassed)) {
  841. _this.setState((0, _objectSpread3.default)((0, _objectSpread3.default)({}, newState), forceState));
  842. }
  843. }
  844. });
  845. (0, _defineProperty2.default)((0, _assertThisInitialized2.default)(_this), "scrollTo", function (scroll) {
  846. _this.listRef.current.scrollTo(scroll);
  847. });
  848. return _this;
  849. }
  850. (0, _createClass2.default)(Tree, [{
  851. key: "componentDidMount",
  852. value: function componentDidMount() {
  853. this.destroyed = false;
  854. this.onUpdated();
  855. }
  856. }, {
  857. key: "componentDidUpdate",
  858. value: function componentDidUpdate() {
  859. this.onUpdated();
  860. }
  861. }, {
  862. key: "onUpdated",
  863. value: function onUpdated() {
  864. var _this$props11 = this.props,
  865. activeKey = _this$props11.activeKey,
  866. _this$props11$itemScr = _this$props11.itemScrollOffset,
  867. itemScrollOffset = _this$props11$itemScr === void 0 ? 0 : _this$props11$itemScr;
  868. if (activeKey !== undefined && activeKey !== this.state.activeKey) {
  869. this.setState({
  870. activeKey: activeKey
  871. });
  872. if (activeKey !== null) {
  873. this.scrollTo({
  874. key: activeKey,
  875. offset: itemScrollOffset
  876. });
  877. }
  878. }
  879. }
  880. }, {
  881. key: "componentWillUnmount",
  882. value: function componentWillUnmount() {
  883. window.removeEventListener('dragend', this.onWindowDragEnd);
  884. this.destroyed = true;
  885. }
  886. }, {
  887. key: "resetDragState",
  888. value: function resetDragState() {
  889. this.setState({
  890. dragOverNodeKey: null,
  891. dropPosition: null,
  892. dropLevelOffset: null,
  893. dropTargetKey: null,
  894. dropContainerKey: null,
  895. dropTargetPos: null,
  896. dropAllowed: false
  897. });
  898. }
  899. }, {
  900. key: "render",
  901. value: function render() {
  902. var _this$state14 = this.state,
  903. focused = _this$state14.focused,
  904. flattenNodes = _this$state14.flattenNodes,
  905. keyEntities = _this$state14.keyEntities,
  906. draggingNodeKey = _this$state14.draggingNodeKey,
  907. activeKey = _this$state14.activeKey,
  908. dropLevelOffset = _this$state14.dropLevelOffset,
  909. dropContainerKey = _this$state14.dropContainerKey,
  910. dropTargetKey = _this$state14.dropTargetKey,
  911. dropPosition = _this$state14.dropPosition,
  912. dragOverNodeKey = _this$state14.dragOverNodeKey,
  913. indent = _this$state14.indent;
  914. var _this$props12 = this.props,
  915. prefixCls = _this$props12.prefixCls,
  916. className = _this$props12.className,
  917. style = _this$props12.style,
  918. showLine = _this$props12.showLine,
  919. focusable = _this$props12.focusable,
  920. _this$props12$tabInde = _this$props12.tabIndex,
  921. tabIndex = _this$props12$tabInde === void 0 ? 0 : _this$props12$tabInde,
  922. selectable = _this$props12.selectable,
  923. showIcon = _this$props12.showIcon,
  924. icon = _this$props12.icon,
  925. switcherIcon = _this$props12.switcherIcon,
  926. draggable = _this$props12.draggable,
  927. checkable = _this$props12.checkable,
  928. checkStrictly = _this$props12.checkStrictly,
  929. disabled = _this$props12.disabled,
  930. motion = _this$props12.motion,
  931. loadData = _this$props12.loadData,
  932. filterTreeNode = _this$props12.filterTreeNode,
  933. height = _this$props12.height,
  934. itemHeight = _this$props12.itemHeight,
  935. scrollWidth = _this$props12.scrollWidth,
  936. virtual = _this$props12.virtual,
  937. titleRender = _this$props12.titleRender,
  938. dropIndicatorRender = _this$props12.dropIndicatorRender,
  939. onContextMenu = _this$props12.onContextMenu,
  940. onScroll = _this$props12.onScroll,
  941. direction = _this$props12.direction,
  942. rootClassName = _this$props12.rootClassName,
  943. rootStyle = _this$props12.rootStyle;
  944. var domProps = (0, _pickAttrs.default)(this.props, {
  945. aria: true,
  946. data: true
  947. });
  948. // It's better move to hooks but we just simply keep here
  949. var draggableConfig;
  950. if (draggable) {
  951. if ((0, _typeof2.default)(draggable) === 'object') {
  952. draggableConfig = draggable;
  953. } else if (typeof draggable === 'function') {
  954. draggableConfig = {
  955. nodeDraggable: draggable
  956. };
  957. } else {
  958. draggableConfig = {};
  959. }
  960. }
  961. var contextValue = {
  962. prefixCls: prefixCls,
  963. selectable: selectable,
  964. showIcon: showIcon,
  965. icon: icon,
  966. switcherIcon: switcherIcon,
  967. draggable: draggableConfig,
  968. draggingNodeKey: draggingNodeKey,
  969. checkable: checkable,
  970. checkStrictly: checkStrictly,
  971. disabled: disabled,
  972. keyEntities: keyEntities,
  973. dropLevelOffset: dropLevelOffset,
  974. dropContainerKey: dropContainerKey,
  975. dropTargetKey: dropTargetKey,
  976. dropPosition: dropPosition,
  977. dragOverNodeKey: dragOverNodeKey,
  978. indent: indent,
  979. direction: direction,
  980. dropIndicatorRender: dropIndicatorRender,
  981. loadData: loadData,
  982. filterTreeNode: filterTreeNode,
  983. titleRender: titleRender,
  984. onNodeClick: this.onNodeClick,
  985. onNodeDoubleClick: this.onNodeDoubleClick,
  986. onNodeExpand: this.onNodeExpand,
  987. onNodeSelect: this.onNodeSelect,
  988. onNodeCheck: this.onNodeCheck,
  989. onNodeLoad: this.onNodeLoad,
  990. onNodeMouseEnter: this.onNodeMouseEnter,
  991. onNodeMouseLeave: this.onNodeMouseLeave,
  992. onNodeContextMenu: this.onNodeContextMenu,
  993. onNodeDragStart: this.onNodeDragStart,
  994. onNodeDragEnter: this.onNodeDragEnter,
  995. onNodeDragOver: this.onNodeDragOver,
  996. onNodeDragLeave: this.onNodeDragLeave,
  997. onNodeDragEnd: this.onNodeDragEnd,
  998. onNodeDrop: this.onNodeDrop
  999. };
  1000. return /*#__PURE__*/React.createElement(_contextTypes.TreeContext.Provider, {
  1001. value: contextValue
  1002. }, /*#__PURE__*/React.createElement("div", {
  1003. className: (0, _classnames.default)(prefixCls, className, rootClassName, (0, _defineProperty2.default)((0, _defineProperty2.default)((0, _defineProperty2.default)({}, "".concat(prefixCls, "-show-line"), showLine), "".concat(prefixCls, "-focused"), focused), "".concat(prefixCls, "-active-focused"), activeKey !== null)),
  1004. style: rootStyle
  1005. }, /*#__PURE__*/React.createElement(_NodeList.default, (0, _extends2.default)({
  1006. ref: this.listRef,
  1007. prefixCls: prefixCls,
  1008. style: style,
  1009. data: flattenNodes,
  1010. disabled: disabled,
  1011. selectable: selectable,
  1012. checkable: !!checkable,
  1013. motion: motion,
  1014. dragging: draggingNodeKey !== null,
  1015. height: height,
  1016. itemHeight: itemHeight,
  1017. virtual: virtual,
  1018. focusable: focusable,
  1019. focused: focused,
  1020. tabIndex: tabIndex,
  1021. activeItem: this.getActiveItem(),
  1022. onFocus: this.onFocus,
  1023. onBlur: this.onBlur,
  1024. onKeyDown: this.onKeyDown,
  1025. onActiveChange: this.onActiveChange,
  1026. onListChangeStart: this.onListChangeStart,
  1027. onListChangeEnd: this.onListChangeEnd,
  1028. onContextMenu: onContextMenu,
  1029. onScroll: onScroll,
  1030. scrollWidth: scrollWidth
  1031. }, this.getTreeNodeRequiredProps(), domProps))));
  1032. }
  1033. }], [{
  1034. key: "getDerivedStateFromProps",
  1035. value: function getDerivedStateFromProps(props, prevState) {
  1036. var prevProps = prevState.prevProps;
  1037. var newState = {
  1038. prevProps: props
  1039. };
  1040. function needSync(name) {
  1041. return !prevProps && props.hasOwnProperty(name) || prevProps && prevProps[name] !== props[name];
  1042. }
  1043. // ================== Tree Node ==================
  1044. var treeData;
  1045. // fieldNames
  1046. var fieldNames = prevState.fieldNames;
  1047. if (needSync('fieldNames')) {
  1048. fieldNames = (0, _treeUtil.fillFieldNames)(props.fieldNames);
  1049. newState.fieldNames = fieldNames;
  1050. }
  1051. // Check if `treeData` or `children` changed and save into the state.
  1052. if (needSync('treeData')) {
  1053. treeData = props.treeData;
  1054. } else if (needSync('children')) {
  1055. (0, _warning.default)(false, '`children` of Tree is deprecated. Please use `treeData` instead.');
  1056. treeData = (0, _treeUtil.convertTreeToData)(props.children);
  1057. }
  1058. // Save flatten nodes info and convert `treeData` into keyEntities
  1059. if (treeData) {
  1060. newState.treeData = treeData;
  1061. var entitiesMap = (0, _treeUtil.convertDataToEntities)(treeData, {
  1062. fieldNames: fieldNames
  1063. });
  1064. newState.keyEntities = (0, _objectSpread3.default)((0, _defineProperty2.default)({}, _NodeList.MOTION_KEY, _NodeList.MotionEntity), entitiesMap.keyEntities);
  1065. // Warning if treeNode not provide key
  1066. if (process.env.NODE_ENV !== 'production') {
  1067. (0, _treeUtil.warningWithoutKey)(treeData, fieldNames);
  1068. }
  1069. }
  1070. var keyEntities = newState.keyEntities || prevState.keyEntities;
  1071. // ================ expandedKeys =================
  1072. if (needSync('expandedKeys') || prevProps && needSync('autoExpandParent')) {
  1073. newState.expandedKeys = props.autoExpandParent || !prevProps && props.defaultExpandParent ? (0, _util.conductExpandParent)(props.expandedKeys, keyEntities) : props.expandedKeys;
  1074. } else if (!prevProps && props.defaultExpandAll) {
  1075. var cloneKeyEntities = (0, _objectSpread3.default)({}, keyEntities);
  1076. delete cloneKeyEntities[_NodeList.MOTION_KEY];
  1077. // Only take the key who has the children to enhance the performance
  1078. var nextExpandedKeys = [];
  1079. Object.keys(cloneKeyEntities).forEach(function (key) {
  1080. var entity = cloneKeyEntities[key];
  1081. if (entity.children && entity.children.length) {
  1082. nextExpandedKeys.push(entity.key);
  1083. }
  1084. });
  1085. newState.expandedKeys = nextExpandedKeys;
  1086. } else if (!prevProps && props.defaultExpandedKeys) {
  1087. newState.expandedKeys = props.autoExpandParent || props.defaultExpandParent ? (0, _util.conductExpandParent)(props.defaultExpandedKeys, keyEntities) : props.defaultExpandedKeys;
  1088. }
  1089. if (!newState.expandedKeys) {
  1090. delete newState.expandedKeys;
  1091. }
  1092. // ================ flattenNodes =================
  1093. if (treeData || newState.expandedKeys) {
  1094. var flattenNodes = (0, _treeUtil.flattenTreeData)(treeData || prevState.treeData, newState.expandedKeys || prevState.expandedKeys, fieldNames);
  1095. newState.flattenNodes = flattenNodes;
  1096. }
  1097. // ================ selectedKeys =================
  1098. if (props.selectable) {
  1099. if (needSync('selectedKeys')) {
  1100. newState.selectedKeys = (0, _util.calcSelectedKeys)(props.selectedKeys, props);
  1101. } else if (!prevProps && props.defaultSelectedKeys) {
  1102. newState.selectedKeys = (0, _util.calcSelectedKeys)(props.defaultSelectedKeys, props);
  1103. }
  1104. }
  1105. // ================= checkedKeys =================
  1106. if (props.checkable) {
  1107. var checkedKeyEntity;
  1108. if (needSync('checkedKeys')) {
  1109. checkedKeyEntity = (0, _util.parseCheckedKeys)(props.checkedKeys) || {};
  1110. } else if (!prevProps && props.defaultCheckedKeys) {
  1111. checkedKeyEntity = (0, _util.parseCheckedKeys)(props.defaultCheckedKeys) || {};
  1112. } else if (treeData) {
  1113. // If `treeData` changed, we also need check it
  1114. checkedKeyEntity = (0, _util.parseCheckedKeys)(props.checkedKeys) || {
  1115. checkedKeys: prevState.checkedKeys,
  1116. halfCheckedKeys: prevState.halfCheckedKeys
  1117. };
  1118. }
  1119. if (checkedKeyEntity) {
  1120. var _checkedKeyEntity = checkedKeyEntity,
  1121. _checkedKeyEntity$che = _checkedKeyEntity.checkedKeys,
  1122. checkedKeys = _checkedKeyEntity$che === void 0 ? [] : _checkedKeyEntity$che,
  1123. _checkedKeyEntity$hal = _checkedKeyEntity.halfCheckedKeys,
  1124. halfCheckedKeys = _checkedKeyEntity$hal === void 0 ? [] : _checkedKeyEntity$hal;
  1125. if (!props.checkStrictly) {
  1126. var conductKeys = (0, _conductUtil.conductCheck)(checkedKeys, true, keyEntities);
  1127. checkedKeys = conductKeys.checkedKeys;
  1128. halfCheckedKeys = conductKeys.halfCheckedKeys;
  1129. }
  1130. newState.checkedKeys = checkedKeys;
  1131. newState.halfCheckedKeys = halfCheckedKeys;
  1132. }
  1133. }
  1134. // ================= loadedKeys ==================
  1135. if (needSync('loadedKeys')) {
  1136. newState.loadedKeys = props.loadedKeys;
  1137. }
  1138. return newState;
  1139. }
  1140. }]);
  1141. return Tree;
  1142. }(React.Component);
  1143. (0, _defineProperty2.default)(Tree, "defaultProps", {
  1144. prefixCls: 'rc-tree',
  1145. showLine: false,
  1146. showIcon: true,
  1147. selectable: true,
  1148. multiple: false,
  1149. checkable: false,
  1150. disabled: false,
  1151. checkStrictly: false,
  1152. draggable: false,
  1153. defaultExpandParent: true,
  1154. autoExpandParent: false,
  1155. defaultExpandAll: false,
  1156. defaultExpandedKeys: [],
  1157. defaultCheckedKeys: [],
  1158. defaultSelectedKeys: [],
  1159. dropIndicatorRender: _DropIndicator.default,
  1160. allowDrop: function allowDrop() {
  1161. return true;
  1162. },
  1163. expandAction: false
  1164. });
  1165. (0, _defineProperty2.default)(Tree, "TreeNode", _TreeNode.default);
  1166. var _default = exports.default = Tree;