1
0

index.tsx 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769
  1. /* eslint-disable max-lines-per-function */
  2. import React, { MouseEvent } from 'react';
  3. import cls from 'classnames';
  4. import PropTypes from 'prop-types';
  5. import ConfigContext from '../configProvider/context';
  6. import TreeFoundation, { TreeAdapter } from '@douyinfe/semi-foundation/tree/foundation';
  7. import {
  8. convertDataToEntities,
  9. flattenTreeData,
  10. calcExpandedKeysForValues,
  11. calcMotionKeys,
  12. convertJsonToData,
  13. findKeysForValues,
  14. calcCheckedKeys,
  15. calcExpandedKeys,
  16. filterTreeData,
  17. normalizeValue,
  18. updateKeys,
  19. calcDisabledKeys
  20. } from '@douyinfe/semi-foundation/tree/treeUtil';
  21. import { cssClasses, strings } from '@douyinfe/semi-foundation/tree/constants';
  22. import BaseComponent from '../_base/baseComponent';
  23. import { isEmpty, isEqual, get, isFunction } from 'lodash';
  24. import { cloneDeep } from './treeUtil';
  25. import Input from '../input/index';
  26. import { FixedSizeList as VirtualList } from 'react-window';
  27. import AutoSizer from './autoSizer';
  28. import TreeContext from './treeContext';
  29. import TreeNode from './treeNode';
  30. import NodeList from './nodeList';
  31. import LocaleConsumer from '../locale/localeConsumer';
  32. import '@douyinfe/semi-foundation/tree/tree.scss';
  33. import { IconSearch } from '@douyinfe/semi-icons';
  34. import { Locale as LocaleObject } from '../locale/interface';
  35. import {
  36. TreeProps,
  37. TreeState,
  38. TreeNodeProps,
  39. TreeNodeData,
  40. FlattenNode,
  41. KeyEntity,
  42. OptionProps
  43. } from './interface';
  44. export * from './interface';
  45. export { AutoSizerProps } from './autoSizer';
  46. const prefixcls = cssClasses.PREFIX;
  47. class Tree extends BaseComponent<TreeProps, TreeState> {
  48. static contextType = ConfigContext;
  49. static propTypes = {
  50. blockNode: PropTypes.bool,
  51. className: PropTypes.string,
  52. showClear: PropTypes.bool,
  53. defaultExpandAll: PropTypes.bool,
  54. defaultExpandedKeys: PropTypes.array,
  55. defaultValue: PropTypes.oneOfType([PropTypes.string, PropTypes.array]),
  56. directory: PropTypes.bool,
  57. disabled: PropTypes.bool,
  58. emptyContent: PropTypes.node,
  59. expandAll: PropTypes.bool,
  60. expandedKeys: PropTypes.array,
  61. filterTreeNode: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
  62. icon: PropTypes.node,
  63. onChangeWithObject: PropTypes.bool,
  64. motion: PropTypes.bool,
  65. multiple: PropTypes.bool,
  66. onChange: PropTypes.func,
  67. onExpand: PropTypes.func,
  68. onSearch: PropTypes.func,
  69. onSelect: PropTypes.func,
  70. onContextMenu: PropTypes.func,
  71. onDoubleClick: PropTypes.func,
  72. searchClassName: PropTypes.string,
  73. searchPlaceholder: PropTypes.string,
  74. searchStyle: PropTypes.object,
  75. selectedKey: PropTypes.string,
  76. showFilteredOnly: PropTypes.bool,
  77. style: PropTypes.object,
  78. treeData: PropTypes.arrayOf(
  79. PropTypes.shape({
  80. key: PropTypes.string.isRequired,
  81. value: PropTypes.oneOfType([PropTypes.string, PropTypes.number]),
  82. label: PropTypes.any,
  83. isLeaf: PropTypes.bool,
  84. })
  85. ),
  86. treeDataSimpleJson: PropTypes.object,
  87. treeNodeFilterProp: PropTypes.string,
  88. value: PropTypes.oneOfType([PropTypes.string, PropTypes.number, PropTypes.array, PropTypes.object]),
  89. virtualize: PropTypes.object,
  90. autoExpandParent: PropTypes.bool,
  91. expandAction: PropTypes.oneOf(strings.EXPAND_ACTION),
  92. searchRender: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
  93. renderLabel: PropTypes.func,
  94. renderFullLabel: PropTypes.func,
  95. leafOnly: PropTypes.bool,
  96. loadedKeys: PropTypes.array,
  97. loadData: PropTypes.func,
  98. onLoad: PropTypes.func,
  99. disableStrictly: PropTypes.bool,
  100. draggable: PropTypes.bool,
  101. autoExpandWhenDragEnter: PropTypes.bool,
  102. hideDraggingNode: PropTypes.bool,
  103. renderDraggingNode: PropTypes.func,
  104. onDragEnd: PropTypes.func,
  105. onDragEnter: PropTypes.func,
  106. onDragLeave: PropTypes.func,
  107. onDragOver: PropTypes.func,
  108. onDragStart: PropTypes.func,
  109. onDrop: PropTypes.func,
  110. labelEllipsis: PropTypes.bool,
  111. checkRelation: PropTypes.string,
  112. 'aria-label': PropTypes.string,
  113. };
  114. static defaultProps = {
  115. showClear: true,
  116. disabled: false,
  117. blockNode: true,
  118. multiple: false,
  119. filterTreeNode: false,
  120. autoExpandParent: false,
  121. treeNodeFilterProp: 'label',
  122. defaultExpandAll: false,
  123. expandAll: false,
  124. onChangeWithObject: false,
  125. motion: true,
  126. leafOnly: false,
  127. showFilteredOnly: false,
  128. expandAction: false,
  129. disableStrictly: false,
  130. draggable: false,
  131. autoExpandWhenDragEnter: true,
  132. checkRelation: 'related',
  133. };
  134. static TreeNode: typeof TreeNode;
  135. inputRef: React.RefObject<typeof Input>;
  136. optionsRef: React.RefObject<any>;
  137. dragNode: any;
  138. onNodeClick: any;
  139. onMotionEnd: any;
  140. constructor(props: TreeProps) {
  141. super(props);
  142. this.state = {
  143. inputValue: '',
  144. keyEntities: {},
  145. treeData: [],
  146. flattenNodes: [],
  147. selectedKeys: [],
  148. checkedKeys: new Set(),
  149. halfCheckedKeys: new Set(),
  150. realCheckedKeys: new Set([]),
  151. motionKeys: new Set([]),
  152. motionType: 'hide',
  153. expandedKeys: new Set(props.expandedKeys),
  154. filteredKeys: new Set(),
  155. filteredExpandedKeys: new Set(),
  156. filteredShownKeys: new Set(),
  157. prevProps: null,
  158. loadedKeys: new Set(),
  159. loadingKeys: new Set(),
  160. cachedFlattenNodes: undefined,
  161. cachedKeyValuePairs: {},
  162. disabledKeys: new Set(),
  163. dragging: false,
  164. dragNodesKeys: new Set(),
  165. dragOverNodeKey: null,
  166. dropPosition: null,
  167. };
  168. this.inputRef = React.createRef();
  169. this.optionsRef = React.createRef();
  170. this.foundation = new TreeFoundation(this.adapter);
  171. this.dragNode = null;
  172. }
  173. /**
  174. * Process of getDerivedStateFromProps was inspired by rc-tree
  175. * https://github.com/react-component/tree
  176. */
  177. static getDerivedStateFromProps(props: TreeProps, prevState: TreeState) {
  178. const { prevProps } = prevState;
  179. let treeData;
  180. let keyEntities = prevState.keyEntities || {};
  181. let valueEntities = prevState.cachedKeyValuePairs || {};
  182. const isSeaching = Boolean(props.filterTreeNode && prevState.inputValue && prevState.inputValue.length);
  183. const newState: Partial<TreeState> = {
  184. prevProps: props,
  185. };
  186. // Accept a props field as a parameter to determine whether to update the field
  187. const needUpdate = (name: string) => {
  188. const firstInProps = !prevProps && name in props;
  189. const nameHasChange = prevProps && !isEqual(prevProps[name], props[name]);
  190. return firstInProps || nameHasChange;
  191. };
  192. // Determine whether treeData has changed
  193. const needUpdateData = () => {
  194. const firstInProps = !prevProps && 'treeData' in props;
  195. const treeDataHasChange = prevProps && prevProps.treeData !== props.treeData;
  196. return firstInProps || treeDataHasChange;
  197. };
  198. // Update the data of tree in state
  199. if (needUpdate('treeData') || (props.draggable && needUpdateData())) {
  200. // eslint-disable-next-line prefer-destructuring
  201. treeData = props.treeData;
  202. newState.treeData = treeData;
  203. const entitiesMap = convertDataToEntities(treeData);
  204. newState.keyEntities = {
  205. ...entitiesMap.keyEntities,
  206. };
  207. keyEntities = newState.keyEntities;
  208. newState.cachedKeyValuePairs = { ...entitiesMap.valueEntities };
  209. valueEntities = newState.cachedKeyValuePairs;
  210. } else if (needUpdate('treeDataSimpleJson')) {
  211. // Convert treeDataSimpleJson to treeData
  212. treeData = convertJsonToData(props.treeDataSimpleJson);
  213. newState.treeData = treeData;
  214. const entitiesMap = convertDataToEntities(treeData);
  215. newState.keyEntities = {
  216. ...entitiesMap.keyEntities,
  217. };
  218. keyEntities = newState.keyEntities;
  219. newState.cachedKeyValuePairs = { ...entitiesMap.valueEntities };
  220. valueEntities = newState.cachedKeyValuePairs;
  221. }
  222. // If treeData keys changes, we won't show animation
  223. if (treeData && props.motion) {
  224. if (prevProps && props.motion) {
  225. newState.motionKeys = new Set([]);
  226. newState.motionType = null;
  227. }
  228. }
  229. const expandAllWhenDataChange = (needUpdate('treeDataSimpleJson') || needUpdate('treeData')) && props.expandAll;
  230. if (!isSeaching) {
  231. // Update expandedKeys
  232. if (needUpdate('expandedKeys') || (prevProps && needUpdate('autoExpandParent'))) {
  233. newState.expandedKeys = calcExpandedKeys(
  234. props.expandedKeys,
  235. keyEntities,
  236. props.autoExpandParent || !prevProps
  237. );
  238. // only show animation when treeData does not change
  239. if (prevProps && props.motion && !treeData) {
  240. const { motionKeys, motionType } = calcMotionKeys(
  241. prevState.expandedKeys,
  242. newState.expandedKeys,
  243. keyEntities
  244. );
  245. newState.motionKeys = new Set(motionKeys);
  246. newState.motionType = motionType;
  247. if (motionType === 'hide') {
  248. // cache flatten nodes: expandedKeys changed may not be triggered by interaction
  249. newState.cachedFlattenNodes = cloneDeep(prevState.flattenNodes);
  250. }
  251. }
  252. } else if ((!prevProps && (props.defaultExpandAll || props.expandAll)) || expandAllWhenDataChange) {
  253. newState.expandedKeys = new Set(Object.keys(keyEntities));
  254. } else if (!prevProps && props.defaultExpandedKeys) {
  255. newState.expandedKeys = calcExpandedKeys(props.defaultExpandedKeys, keyEntities);
  256. } else if (!prevProps && props.defaultValue) {
  257. newState.expandedKeys = calcExpandedKeysForValues(
  258. props.defaultValue,
  259. keyEntities,
  260. props.multiple,
  261. valueEntities
  262. );
  263. } else if (!prevProps && props.value) {
  264. newState.expandedKeys = calcExpandedKeysForValues(
  265. props.value,
  266. keyEntities,
  267. props.multiple,
  268. valueEntities
  269. );
  270. }
  271. if (!newState.expandedKeys) {
  272. delete newState.expandedKeys;
  273. }
  274. // Update flattenNodes
  275. if (treeData || newState.expandedKeys) {
  276. const flattenNodes = flattenTreeData(
  277. treeData || prevState.treeData,
  278. newState.expandedKeys || prevState.expandedKeys
  279. );
  280. newState.flattenNodes = flattenNodes;
  281. }
  282. } else {
  283. let filteredState;
  284. // treeData changed while searching
  285. if (treeData) {
  286. // Get filter data
  287. filteredState = filterTreeData({
  288. treeData,
  289. inputValue: prevState.inputValue,
  290. filterTreeNode: props.filterTreeNode,
  291. filterProps: props.treeNodeFilterProp,
  292. showFilteredOnly: props.showFilteredOnly,
  293. keyEntities: newState.keyEntities,
  294. prevExpandedKeys: [...prevState.filteredExpandedKeys],
  295. });
  296. newState.flattenNodes = filteredState.flattenNodes;
  297. newState.motionKeys = new Set([]);
  298. newState.filteredKeys = filteredState.filteredKeys;
  299. newState.filteredShownKeys = filteredState.filteredShownKeys;
  300. newState.filteredExpandedKeys = filteredState.filteredExpandedKeys;
  301. }
  302. // expandedKeys changed while searching
  303. if (props.expandedKeys) {
  304. newState.filteredExpandedKeys = calcExpandedKeys(
  305. props.expandedKeys,
  306. keyEntities,
  307. props.autoExpandParent || !prevProps
  308. );
  309. if (prevProps && props.motion) {
  310. const prevKeys = prevState ? prevState.filteredExpandedKeys : new Set([]);
  311. // only show animation when treeData does not change
  312. if (!treeData) {
  313. const motionResult = calcMotionKeys(
  314. prevKeys,
  315. newState.filteredExpandedKeys,
  316. keyEntities
  317. );
  318. let { motionKeys } = motionResult;
  319. const { motionType } = motionResult;
  320. if (props.showFilteredOnly) {
  321. motionKeys = motionKeys.filter(key => prevState.filteredShownKeys.has(key));
  322. }
  323. if (motionType === 'hide') {
  324. // cache flatten nodes: expandedKeys changed may not be triggered by interaction
  325. newState.cachedFlattenNodes = cloneDeep(prevState.flattenNodes);
  326. }
  327. newState.motionKeys = new Set(motionKeys);
  328. newState.motionType = motionType;
  329. }
  330. }
  331. newState.flattenNodes = flattenTreeData(
  332. treeData || prevState.treeData,
  333. newState.filteredExpandedKeys || prevState.filteredExpandedKeys,
  334. props.showFilteredOnly && prevState.filteredShownKeys
  335. );
  336. }
  337. }
  338. // Handle single selection and multiple selection in controlled mode
  339. const withObject = props.onChangeWithObject;
  340. const isMultiple = props.multiple;
  341. if (!isMultiple) {
  342. // When getting single selection, the selected node
  343. if (needUpdate('value')) {
  344. newState.selectedKeys = findKeysForValues(
  345. // In both cases whether withObject is turned on, the value is standardized to string
  346. normalizeValue(props.value, withObject),
  347. valueEntities,
  348. isMultiple
  349. );
  350. } else if (!prevProps && props.defaultValue) {
  351. newState.selectedKeys = findKeysForValues(
  352. normalizeValue(props.defaultValue, withObject),
  353. valueEntities,
  354. isMultiple
  355. );
  356. } else if (treeData) {
  357. // If `treeData` changed, we also need check it
  358. if (props.value) {
  359. newState.selectedKeys = findKeysForValues(
  360. normalizeValue(props.value, withObject) || '',
  361. valueEntities,
  362. isMultiple
  363. );
  364. }
  365. }
  366. } else {
  367. let checkedKeyValues;
  368. // Get the selected node during multiple selection
  369. if (needUpdate('value')) {
  370. checkedKeyValues = findKeysForValues(
  371. normalizeValue(props.value, withObject),
  372. valueEntities,
  373. isMultiple
  374. );
  375. } else if (!prevProps && props.defaultValue) {
  376. checkedKeyValues = findKeysForValues(
  377. normalizeValue(props.defaultValue, withObject),
  378. valueEntities,
  379. isMultiple
  380. );
  381. } else if (treeData) {
  382. // If `treeData` changed, we also need check it
  383. if (props.value) {
  384. checkedKeyValues = findKeysForValues(
  385. normalizeValue(props.value, withObject) || [],
  386. valueEntities,
  387. isMultiple
  388. );
  389. } else {
  390. checkedKeyValues = updateKeys(prevState.checkedKeys, keyEntities);
  391. }
  392. }
  393. if (checkedKeyValues) {
  394. if (props.checkRelation === 'unRelated') {
  395. newState.realCheckedKeys = new Set(checkedKeyValues);
  396. } else if (props.checkRelation === 'related') {
  397. const { checkedKeys, halfCheckedKeys } = calcCheckedKeys(checkedKeyValues, keyEntities);
  398. newState.checkedKeys = checkedKeys;
  399. newState.halfCheckedKeys = halfCheckedKeys;
  400. }
  401. }
  402. }
  403. // update loadedKeys
  404. if (needUpdate('loadedKeys')) {
  405. newState.loadedKeys = new Set(props.loadedKeys);
  406. }
  407. // update disableStrictly
  408. if (treeData && props.disableStrictly && props.checkRelation === 'related') {
  409. newState.disabledKeys = calcDisabledKeys(keyEntities);
  410. }
  411. return newState;
  412. }
  413. get adapter(): TreeAdapter {
  414. const filterAdapter: Pick<TreeAdapter, 'updateInputValue' | 'focusInput'> = {
  415. updateInputValue: value => {
  416. this.setState({ inputValue: value });
  417. },
  418. focusInput: () => {
  419. if (this.inputRef && this.inputRef.current) {
  420. (this.inputRef.current as any).focus();
  421. }
  422. },
  423. };
  424. return {
  425. ...super.adapter,
  426. ...filterAdapter,
  427. updateState: states => {
  428. this.setState({ ...states } as TreeState);
  429. },
  430. notifyExpand: (expandedKeys, { expanded: bool, node }) => {
  431. this.props.onExpand && this.props.onExpand([...expandedKeys], { expanded: bool, node });
  432. if (bool && this.props.loadData) {
  433. this.onNodeLoad(node);
  434. }
  435. },
  436. notifySelect: (selectKey, bool, node) => {
  437. this.props.onSelect && this.props.onSelect(selectKey, bool, node);
  438. },
  439. notifyChange: value => {
  440. this.props.onChange && this.props.onChange(value);
  441. },
  442. notifySearch: input => {
  443. this.props.onSearch && this.props.onSearch(input);
  444. },
  445. notifyRightClick: (e, node) => {
  446. this.props.onContextMenu && this.props.onContextMenu(e, node);
  447. },
  448. notifyDoubleClick: (e, node) => {
  449. this.props.onDoubleClick && this.props.onDoubleClick(e, node);
  450. },
  451. cacheFlattenNodes: bool => {
  452. this.setState({ cachedFlattenNodes: bool ? cloneDeep(this.state.flattenNodes) : undefined });
  453. },
  454. setDragNode: treeNode => {
  455. this.dragNode = treeNode;
  456. },
  457. };
  458. }
  459. search = (value: string) => {
  460. this.foundation.handleInputChange(value);
  461. };
  462. renderInput() {
  463. const {
  464. searchClassName,
  465. searchStyle,
  466. searchRender,
  467. searchPlaceholder,
  468. showClear
  469. } = this.props;
  470. const inputcls = cls(`${prefixcls}-input`);
  471. const { inputValue } = this.state;
  472. const inputProps = {
  473. value: inputValue,
  474. className: inputcls,
  475. onChange: (value: string) => this.search(value),
  476. prefix: <IconSearch />,
  477. showClear,
  478. placeholder: searchPlaceholder,
  479. };
  480. const wrapperCls = cls(`${prefixcls}-search-wrapper`, searchClassName);
  481. return (
  482. <div className={wrapperCls} style={searchStyle}>
  483. <LocaleConsumer componentName="Tree">
  484. {(locale: LocaleObject) => {
  485. inputProps.placeholder = searchPlaceholder || get(locale, 'searchPlaceholder');
  486. if (isFunction(searchRender)) {
  487. return searchRender({ ...inputProps });
  488. }
  489. if (searchRender === false) {
  490. return null;
  491. }
  492. return (
  493. <Input
  494. aria-label='Filter Tree'
  495. ref={this.inputRef as any}
  496. {...inputProps}
  497. />
  498. );
  499. }}
  500. </LocaleConsumer>
  501. </div>
  502. );
  503. }
  504. renderEmpty = () => {
  505. const { emptyContent } = this.props;
  506. if (emptyContent) {
  507. return <TreeNode empty emptyContent={this.props.emptyContent} />;
  508. } else {
  509. return (
  510. <LocaleConsumer componentName="Tree">
  511. {(locale: LocaleObject) => <TreeNode empty emptyContent={get(locale, 'emptyText')} />}
  512. </LocaleConsumer>
  513. );
  514. }
  515. };
  516. onNodeSelect = (e: MouseEvent, treeNode: TreeNodeProps) => {
  517. this.foundation.handleNodeSelect(e, treeNode);
  518. };
  519. onNodeLoad = (data: TreeNodeData) => (
  520. new Promise(resolve => {
  521. // We need to get the latest state of loading/loaded keys
  522. this.setState(({ loadedKeys = new Set([]), loadingKeys = new Set([]) }) => (
  523. this.foundation.handleNodeLoad(loadedKeys, loadingKeys, data, resolve)
  524. ));
  525. })
  526. );
  527. onNodeCheck = (e: MouseEvent, treeNode: TreeNodeProps) => {
  528. this.foundation.handleNodeSelect(e, treeNode);
  529. };
  530. onNodeExpand = (e: MouseEvent, treeNode: TreeNodeProps) => {
  531. this.foundation.handleNodeExpand(e, treeNode);
  532. };
  533. onNodeRightClick = (e: MouseEvent, treeNode: TreeNodeProps) => {
  534. this.foundation.handleNodeRightClick(e, treeNode);
  535. };
  536. onNodeDoubleClick = (e: MouseEvent, treeNode: TreeNodeProps) => {
  537. this.foundation.handleNodeDoubleClick(e, treeNode);
  538. };
  539. onNodeDragStart = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  540. this.foundation.handleNodeDragStart(e, treeNode);
  541. };
  542. onNodeDragEnter = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  543. this.foundation.handleNodeDragEnter(e, treeNode, this.dragNode);
  544. };
  545. onNodeDragOver = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  546. this.foundation.handleNodeDragOver(e, treeNode, this.dragNode);
  547. };
  548. onNodeDragLeave = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  549. this.foundation.handleNodeDragLeave(e, treeNode);
  550. };
  551. onNodeDragEnd = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  552. this.foundation.handleNodeDragEnd(e, treeNode);
  553. };
  554. onNodeDrop = (e: React.DragEvent<HTMLDivElement>, treeNode: TreeNodeProps) => {
  555. this.foundation.handleNodeDrop(e, treeNode, this.dragNode);
  556. };
  557. getTreeNodeRequiredProps = () => {
  558. const { expandedKeys, selectedKeys, checkedKeys, halfCheckedKeys, keyEntities, filteredKeys } = this.state;
  559. return {
  560. expandedKeys: expandedKeys || new Set(),
  561. selectedKeys: selectedKeys || [],
  562. checkedKeys: checkedKeys || new Set(),
  563. halfCheckedKeys: halfCheckedKeys || new Set(),
  564. filteredKeys: filteredKeys || new Set(),
  565. keyEntities,
  566. };
  567. };
  568. getTreeNodeKey = (treeNode: TreeNodeData) => {
  569. const { data } = treeNode;
  570. const { key } = data;
  571. return key;
  572. };
  573. renderTreeNode = (treeNode: FlattenNode, ind?: number, style?: React.CSSProperties) => {
  574. const { data } = treeNode;
  575. const { key } = data;
  576. const treeNodeProps = this.foundation.getTreeNodeProps(key);
  577. if (!treeNodeProps) {
  578. return null;
  579. }
  580. return <TreeNode {...treeNodeProps} {...data} key={key} data={data} style={isEmpty(style) ? {} : style} />;
  581. };
  582. itemKey = (index: number, data: KeyEntity) => {
  583. // Find the item at the specified index.
  584. const item = data[index];
  585. // Return a value that uniquely identifies this item.
  586. return item.key;
  587. };
  588. renderNodeList() {
  589. const { flattenNodes, cachedFlattenNodes, motionKeys, motionType } = this.state;
  590. const { virtualize, motion } = this.props;
  591. const { direction } = this.context;
  592. if (isEmpty(flattenNodes)) {
  593. return undefined;
  594. }
  595. if (!virtualize || isEmpty(virtualize)) {
  596. return (
  597. <NodeList
  598. flattenNodes={flattenNodes}
  599. flattenList={cachedFlattenNodes}
  600. motionKeys={motion ? motionKeys : new Set([])}
  601. motionType={motionType}
  602. onMotionEnd={this.onMotionEnd}
  603. renderTreeNode={this.renderTreeNode}
  604. />
  605. );
  606. }
  607. const option = ({ index, style, data }: OptionProps) => (
  608. this.renderTreeNode(data[index], index, style)
  609. );
  610. return (
  611. <AutoSizer defaultHeight={virtualize.height} defaultWidth={virtualize.width}>
  612. {({ height, width }: { width: string | number; height: string | number }) => (
  613. <VirtualList
  614. itemCount={flattenNodes.length}
  615. itemSize={virtualize.itemSize}
  616. height={height}
  617. width={width}
  618. itemKey={this.itemKey}
  619. itemData={flattenNodes as any}
  620. className={`${prefixcls}-virtual-list`}
  621. style={{ direction }}
  622. >
  623. {option}
  624. </VirtualList>
  625. )}
  626. </AutoSizer>
  627. );
  628. }
  629. render() {
  630. const {
  631. keyEntities,
  632. motionKeys,
  633. motionType,
  634. inputValue,
  635. filteredKeys,
  636. dragOverNodeKey,
  637. dropPosition,
  638. } = this.state;
  639. const {
  640. blockNode,
  641. className,
  642. style,
  643. filterTreeNode,
  644. disabled,
  645. icon,
  646. directory,
  647. multiple,
  648. showFilteredOnly,
  649. motion,
  650. expandAction,
  651. loadData,
  652. renderLabel,
  653. draggable,
  654. renderFullLabel,
  655. labelEllipsis,
  656. virtualize,
  657. } = this.props;
  658. const wrapperCls = cls(`${prefixcls}-wrapper`, className);
  659. const listCls = cls(`${prefixcls}-option-list`, {
  660. [`${prefixcls}-option-list-block`]: blockNode,
  661. });
  662. const searchNoRes = Boolean(inputValue) && !filteredKeys.size;
  663. const noData = isEmpty(keyEntities) || (showFilteredOnly && searchNoRes);
  664. const ariaAttr = {
  665. role: noData ? 'none' : 'tree'
  666. };
  667. if (ariaAttr.role === 'tree') {
  668. ariaAttr['aria-multiselectable'] = multiple ? true : false;
  669. }
  670. return (
  671. <TreeContext.Provider
  672. value={{
  673. treeDisabled: disabled,
  674. treeIcon: icon,
  675. motion,
  676. motionKeys,
  677. motionType,
  678. filterTreeNode,
  679. keyEntities,
  680. onNodeClick: this.onNodeClick,
  681. onNodeExpand: this.onNodeExpand,
  682. onNodeSelect: this.onNodeSelect,
  683. onNodeCheck: this.onNodeCheck,
  684. onNodeRightClick: this.onNodeRightClick,
  685. onNodeDoubleClick: this.onNodeDoubleClick,
  686. renderTreeNode: this.renderTreeNode,
  687. onNodeDragStart: this.onNodeDragStart,
  688. onNodeDragEnter: this.onNodeDragEnter,
  689. onNodeDragOver: this.onNodeDragOver,
  690. onNodeDragLeave: this.onNodeDragLeave,
  691. onNodeDragEnd: this.onNodeDragEnd,
  692. onNodeDrop: this.onNodeDrop,
  693. expandAction,
  694. directory,
  695. multiple,
  696. showFilteredOnly,
  697. isSearching: Boolean(inputValue),
  698. loadData,
  699. onNodeLoad: this.onNodeLoad,
  700. renderLabel,
  701. draggable,
  702. renderFullLabel,
  703. dragOverNodeKey,
  704. dropPosition,
  705. labelEllipsis: typeof labelEllipsis === 'undefined' ? virtualize : labelEllipsis,
  706. }}
  707. >
  708. <div aria-label={this.props['aria-label']} className={wrapperCls} style={style}>
  709. {filterTreeNode ? this.renderInput() : null}
  710. <div className={listCls} {...ariaAttr}>
  711. {noData ? this.renderEmpty() : this.renderNodeList()}
  712. </div>
  713. </div>
  714. </TreeContext.Provider>
  715. );
  716. }
  717. }
  718. Tree.TreeNode = TreeNode;
  719. export default Tree;