index.tsx 54 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431
  1. import React, { Fragment } from 'react';
  2. import ReactDOM from 'react-dom';
  3. import cls from 'classnames';
  4. import PropTypes from 'prop-types';
  5. import { isEqual, isString, isEmpty, noop, get, isFunction } from 'lodash';
  6. import TreeSelectFoundation, {
  7. Size,
  8. BasicTriggerRenderProps,
  9. /* Corresponding props */
  10. BasicTreeSelectProps,
  11. /* Corresponding state */
  12. BasicTreeSelectInnerData,
  13. TreeSelectAdapter
  14. } from '@douyinfe/semi-foundation/treeSelect/foundation';
  15. import {
  16. convertDataToEntities,
  17. flattenTreeData,
  18. calcExpandedKeysForValues,
  19. calcMotionKeys,
  20. findKeysForValues,
  21. calcCheckedKeys,
  22. calcExpandedKeys,
  23. getValueOrKey,
  24. normalizeKeyList,
  25. calcDisabledKeys,
  26. normalizeValue,
  27. updateKeys,
  28. } from '@douyinfe/semi-foundation/tree/treeUtil';
  29. import { cssClasses, strings } from '@douyinfe/semi-foundation/treeSelect/constants';
  30. import { numbers as popoverNumbers } from '@douyinfe/semi-foundation/popover/constants';
  31. import { FixedSizeList as VirtualList, ListItemKeySelector } from 'react-window';
  32. import '@douyinfe/semi-foundation/tree/tree.scss';
  33. import '@douyinfe/semi-foundation/treeSelect/treeSelect.scss';
  34. import BaseComponent, { ValidateStatus } from '../_base/baseComponent';
  35. import ConfigContext, { ContextValue } from '../configProvider/context';
  36. import TagGroup from '../tag/group';
  37. import Tag, { TagProps } from '../tag/index';
  38. import Input, { InputProps } from '../input/index';
  39. import AutoSizer from '../tree/autoSizer';
  40. import TreeContext from '../tree/treeContext';
  41. import TreeNode from '../tree/treeNode';
  42. import NodeList from '../tree/nodeList';
  43. import { cloneDeep } from '../tree/treeUtil';
  44. import LocaleConsumer from '../locale/localeConsumer';
  45. import { Locale } from '../locale/interface';
  46. import Trigger from '../trigger';
  47. import TagInput from '../tagInput';
  48. import { isSemiIcon } from '../_utils';
  49. import { OptionProps, TreeProps, TreeState, FlattenNode, TreeNodeData, TreeNodeProps } from '../tree/interface';
  50. import { IconChevronDown, IconClear, IconSearch } from '@douyinfe/semi-icons';
  51. import CheckboxGroup from '../checkbox/checkboxGroup';
  52. import Popover, { PopoverProps } from '../popover/index';
  53. export type ExpandAction = false | 'click' | 'doubleClick';
  54. export interface TriggerRenderProps extends Omit<BasicTriggerRenderProps, 'componentProps'> {
  55. [x: string]: any;
  56. componentProps: TreeSelectProps;
  57. value: TreeNodeData[];
  58. onClear: (e: React.MouseEvent) => void
  59. }
  60. export interface OnChange {
  61. /* onChangeWithObject is false */
  62. (
  63. value: TreeNodeData['value'] | Array<TreeNodeData['value']>,
  64. node: TreeNodeData[] | TreeNodeData,
  65. e: React.MouseEvent
  66. ): void;
  67. /* onChangeWithObject is true */
  68. (node: TreeNodeData[] | TreeNodeData, e: React.MouseEvent): void
  69. }
  70. export type RenderSelectedItemInSingle = (treeNode: TreeNodeData) => React.ReactNode;
  71. export type RenderSelectedItemInMultiple = (
  72. treeNode: TreeNodeData,
  73. otherProps: { index: number | string; onClose: (tagContent: any, e: React.MouseEvent) => void }
  74. ) => {
  75. isRenderInTag: boolean;
  76. content: React.ReactNode
  77. };
  78. export type RenderSelectedItem = RenderSelectedItemInSingle | RenderSelectedItemInMultiple;
  79. export type OverrideCommonProps =
  80. 'renderFullLabel'
  81. | 'renderLabel'
  82. | 'defaultValue'
  83. | 'emptyContent'
  84. | 'filterTreeNode'
  85. | 'style'
  86. | 'treeData'
  87. | 'value'
  88. | 'onExpand';
  89. /**
  90. * Type definition description:
  91. * TreeSelectProps inherits some properties from BasicTreeSelectProps (from foundation) and TreeProps (from semi-ui-react).
  92. */
  93. // eslint-disable-next-line max-len
  94. export interface TreeSelectProps extends Omit<BasicTreeSelectProps, OverrideCommonProps | 'validateStatus' | 'searchRender'>, Pick<TreeProps, OverrideCommonProps> {
  95. 'aria-describedby'?: React.AriaAttributes['aria-describedby'];
  96. 'aria-errormessage'?: React.AriaAttributes['aria-errormessage'];
  97. 'aria-invalid'?: React.AriaAttributes['aria-invalid'];
  98. 'aria-labelledby'?: React.AriaAttributes['aria-labelledby'];
  99. 'aria-required'?: React.AriaAttributes['aria-required'];
  100. motion?: boolean;
  101. mouseEnterDelay?: number;
  102. mouseLeaveDelay?: number;
  103. arrowIcon?: React.ReactNode;
  104. autoAdjustOverflow?: boolean;
  105. clickToHide?: boolean;
  106. defaultOpen?: boolean;
  107. dropdownClassName?: string;
  108. dropdownMatchSelectWidth?: boolean;
  109. dropdownStyle?: React.CSSProperties;
  110. insetLabel?: React.ReactNode;
  111. insetLabelId?: string;
  112. maxTagCount?: number;
  113. motionExpand?: boolean;
  114. optionListStyle?: React.CSSProperties;
  115. outerBottomSlot?: React.ReactNode;
  116. outerTopSlot?: React.ReactNode;
  117. placeholder?: string;
  118. prefix?: React.ReactNode;
  119. searchAutoFocus?: boolean;
  120. searchPlaceholder?: string;
  121. showSearchClear?: boolean;
  122. size?: Size;
  123. suffix?: React.ReactNode;
  124. treeNodeLabelProp?: string;
  125. validateStatus?: ValidateStatus;
  126. zIndex?: number;
  127. searchPosition?: string;
  128. stopPropagation?: boolean | string;
  129. restTagsPopoverProps?: PopoverProps;
  130. searchRender?: boolean | ((inputProps: InputProps) => React.ReactNode);
  131. onSelect?: (selectedKeys: string, selected: boolean, selectedNode: TreeNodeData) => void;
  132. renderSelectedItem?: RenderSelectedItem;
  133. getPopupContainer?: () => HTMLElement;
  134. triggerRender?: (props?: TriggerRenderProps) => React.ReactNode;
  135. onBlur?: (e: React.MouseEvent) => void;
  136. onChange?: OnChange;
  137. onFocus?: (e: React.MouseEvent) => void;
  138. onVisibleChange?: (isVisible: boolean) => void
  139. }
  140. export type OverrideCommonState =
  141. 'keyEntities'
  142. | 'treeData'
  143. | 'disabledKeys'
  144. | 'flattenNodes';
  145. // eslint-disable-next-line max-len
  146. export interface TreeSelectState extends Omit<BasicTreeSelectInnerData, OverrideCommonState | 'prevProps'>, Pick<TreeState, OverrideCommonState> {
  147. inputTriggerFocus: boolean;
  148. isOpen: boolean;
  149. isInput: boolean;
  150. rePosKey: number;
  151. dropdownMinWidth: null | number;
  152. isHovering: boolean;
  153. prevProps: TreeSelectProps
  154. }
  155. const prefixcls = cssClasses.PREFIX;
  156. const prefixTree = cssClasses.PREFIX_TREE;
  157. const key = 0;
  158. class TreeSelect extends BaseComponent<TreeSelectProps, TreeSelectState> {
  159. static contextType = ConfigContext;
  160. static propTypes = {
  161. 'aria-describedby': PropTypes.string,
  162. 'aria-errormessage': PropTypes.string,
  163. 'aria-invalid': PropTypes.bool,
  164. 'aria-labelledby': PropTypes.string,
  165. 'aria-required': PropTypes.bool,
  166. loadedKeys: PropTypes.arrayOf(PropTypes.string),
  167. loadData: PropTypes.func,
  168. onLoad: PropTypes.func,
  169. arrowIcon: PropTypes.node,
  170. defaultOpen: PropTypes.bool,
  171. defaultValue: PropTypes.oneOfType([PropTypes.string, PropTypes.array]),
  172. defaultExpandAll: PropTypes.bool,
  173. defaultExpandedKeys: PropTypes.array,
  174. expandAll: PropTypes.bool,
  175. disabled: PropTypes.bool,
  176. disableStrictly: PropTypes.bool,
  177. // Whether to turn on the input box filtering function, when it is a function, it represents a custom filtering function
  178. filterTreeNode: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
  179. multiple: PropTypes.bool,
  180. searchPlaceholder: PropTypes.string,
  181. searchAutoFocus: PropTypes.bool,
  182. virtualize: PropTypes.object,
  183. treeNodeFilterProp: PropTypes.string,
  184. onChange: PropTypes.func,
  185. onSearch: PropTypes.func,
  186. onSelect: PropTypes.func,
  187. onExpand: PropTypes.func,
  188. onChangeWithObject: PropTypes.bool,
  189. onBlur: PropTypes.func,
  190. onFocus: PropTypes.func,
  191. value: PropTypes.oneOfType([PropTypes.string, PropTypes.number, PropTypes.array, PropTypes.object]),
  192. expandedKeys: PropTypes.array,
  193. autoExpandParent: PropTypes.bool,
  194. showClear: PropTypes.bool,
  195. showSearchClear: PropTypes.bool,
  196. autoAdjustOverflow: PropTypes.bool,
  197. showFilteredOnly: PropTypes.bool,
  198. motionExpand: PropTypes.bool,
  199. emptyContent: PropTypes.node,
  200. leafOnly: PropTypes.bool,
  201. treeData: PropTypes.arrayOf(
  202. PropTypes.shape({
  203. key: PropTypes.string.isRequired,
  204. value: PropTypes.oneOfType([PropTypes.string, PropTypes.number]),
  205. label: PropTypes.any,
  206. })
  207. ),
  208. dropdownClassName: PropTypes.string,
  209. dropdownStyle: PropTypes.object,
  210. motion: PropTypes.bool,
  211. placeholder: PropTypes.string,
  212. maxTagCount: PropTypes.number,
  213. size: PropTypes.oneOf<TreeSelectProps['size']>(strings.SIZE_SET),
  214. className: PropTypes.string,
  215. style: PropTypes.object,
  216. treeNodeLabelProp: PropTypes.string,
  217. suffix: PropTypes.node,
  218. prefix: PropTypes.node,
  219. insetLabel: PropTypes.node,
  220. insetLabelId: PropTypes.string,
  221. zIndex: PropTypes.number,
  222. getPopupContainer: PropTypes.func,
  223. dropdownMatchSelectWidth: PropTypes.bool,
  224. validateStatus: PropTypes.oneOf(strings.STATUS),
  225. mouseEnterDelay: PropTypes.number,
  226. mouseLeaveDelay: PropTypes.number,
  227. triggerRender: PropTypes.func,
  228. stopPropagation: PropTypes.oneOfType([PropTypes.bool, PropTypes.string]),
  229. outerBottomSlot: PropTypes.node,
  230. outerTopSlot: PropTypes.node,
  231. onVisibleChange: PropTypes.func,
  232. expandAction: PropTypes.oneOf(['click' as const, 'doubleClick' as const, false as const]),
  233. searchPosition: PropTypes.oneOf([strings.SEARCH_POSITION_DROPDOWN, strings.SEARCH_POSITION_TRIGGER]),
  234. clickToHide: PropTypes.bool,
  235. renderLabel: PropTypes.func,
  236. renderFullLabel: PropTypes.func,
  237. labelEllipsis: PropTypes.bool,
  238. optionListStyle: PropTypes.object,
  239. searchRender: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
  240. renderSelectedItem: PropTypes.func,
  241. checkRelation: PropTypes.string,
  242. 'aria-label': PropTypes.string,
  243. showRestTagsPopover: PropTypes.bool,
  244. restTagsPopoverProps: PropTypes.object,
  245. };
  246. static defaultProps: Partial<TreeSelectProps> = {
  247. searchPosition: strings.SEARCH_POSITION_DROPDOWN,
  248. arrowIcon: <IconChevronDown />,
  249. autoExpandParent: false,
  250. autoAdjustOverflow: true,
  251. stopPropagation: true,
  252. motion: true,
  253. motionExpand: true,
  254. expandAll: false,
  255. zIndex: popoverNumbers.DEFAULT_Z_INDEX,
  256. disableStrictly: false,
  257. multiple: false,
  258. filterTreeNode: false,
  259. size: 'default' as const,
  260. treeNodeFilterProp: 'label' as const,
  261. onChangeWithObject: false,
  262. treeNodeLabelProp: 'label' as const,
  263. dropdownMatchSelectWidth: true,
  264. defaultOpen: false,
  265. showSearchClear: true,
  266. showClear: false,
  267. onVisibleChange: noop,
  268. expandAction: false,
  269. clickToHide: true,
  270. searchAutoFocus: false,
  271. checkRelation: 'related',
  272. 'aria-label': 'TreeSelect',
  273. showRestTagsPopover: false,
  274. restTagsPopoverProps: {},
  275. };
  276. inputRef: React.RefObject<typeof Input>;
  277. tagInputRef: React.RefObject<TagInput>;
  278. triggerRef: React.RefObject<HTMLDivElement>;
  279. optionsRef: React.RefObject<any>;
  280. clickOutsideHandler: any;
  281. _flattenNodes: TreeState['flattenNodes'];
  282. onNodeClick: any;
  283. onNodeDoubleClick: any;
  284. onMotionEnd: any;
  285. treeSelectID: string;
  286. context: ContextValue;
  287. constructor(props: TreeSelectProps) {
  288. super(props);
  289. this.state = {
  290. inputTriggerFocus: false,
  291. isOpen: false,
  292. isInput: false,
  293. rePosKey: key,
  294. dropdownMinWidth: null,
  295. inputValue: '',
  296. keyEntities: {},
  297. treeData: [],
  298. flattenNodes: [],
  299. selectedKeys: [],
  300. checkedKeys: new Set(),
  301. halfCheckedKeys: new Set(),
  302. realCheckedKeys: new Set([]),
  303. disabledKeys: new Set(),
  304. motionKeys: new Set([]),
  305. motionType: 'hide',
  306. expandedKeys: new Set(props.expandedKeys),
  307. filteredKeys: new Set(),
  308. filteredExpandedKeys: new Set(),
  309. filteredShownKeys: new Set(),
  310. prevProps: null,
  311. isHovering: false,
  312. cachedKeyValuePairs: {},
  313. loadedKeys: new Set(),
  314. loadingKeys: new Set(),
  315. };
  316. this.inputRef = React.createRef();
  317. this.tagInputRef = React.createRef();
  318. this.triggerRef = React.createRef();
  319. this.optionsRef = React.createRef();
  320. this.clickOutsideHandler = null;
  321. this.foundation = new TreeSelectFoundation(this.adapter);
  322. this.treeSelectID = Math.random().toString(36).slice(2);
  323. this.onMotionEnd = () => {
  324. this.adapter.rePositionDropdown();
  325. };
  326. }
  327. // eslint-disable-next-line max-lines-per-function
  328. static getDerivedStateFromProps(props: TreeSelectProps, prevState: TreeSelectState) {
  329. const { prevProps, rePosKey } = prevState;
  330. const needUpdate = (name: string) => (
  331. (!prevProps && name in props) ||
  332. (prevProps && !isEqual(prevProps[name], props[name]))
  333. );
  334. let treeData;
  335. const withObject = props.onChangeWithObject;
  336. let keyEntities = prevState.keyEntities || {};
  337. let valueEntities = prevState.cachedKeyValuePairs || {};
  338. const newState: Partial<TreeSelectState> = {
  339. prevProps: props,
  340. };
  341. // TreeNode
  342. if (needUpdate('treeData')) {
  343. treeData = props.treeData;
  344. newState.treeData = treeData;
  345. const entitiesMap = convertDataToEntities(treeData);
  346. newState.keyEntities = {
  347. ...entitiesMap.keyEntities,
  348. };
  349. keyEntities = newState.keyEntities;
  350. newState.cachedKeyValuePairs = { ...entitiesMap.valueEntities };
  351. valueEntities = newState.cachedKeyValuePairs;
  352. }
  353. // if treeData keys changes, we won't show animation
  354. if (
  355. treeData &&
  356. props.motion &&
  357. !isEqual(Object.keys(newState.keyEntities), Object.keys(prevState.keyEntities))
  358. ) {
  359. if (prevProps && props.motion) {
  360. newState.motionKeys = new Set([]);
  361. newState.motionType = null;
  362. }
  363. }
  364. const expandAllWhenDataChange = needUpdate('treeData') && props.expandAll;
  365. // expandedKeys
  366. if (needUpdate('expandedKeys') || (prevProps && needUpdate('autoExpandParent'))) {
  367. newState.expandedKeys = calcExpandedKeys(
  368. props.expandedKeys,
  369. keyEntities,
  370. props.autoExpandParent || !prevProps
  371. );
  372. // only show animation when treeData does not change
  373. if (prevProps && props.motion && !treeData) {
  374. const { motionKeys, motionType } = calcMotionKeys(
  375. prevState.expandedKeys,
  376. newState.expandedKeys,
  377. keyEntities
  378. );
  379. newState.motionKeys = new Set(motionKeys);
  380. newState.motionType = motionType;
  381. }
  382. } else if ((!prevProps && (props.defaultExpandAll || props.expandAll)) || expandAllWhenDataChange) {
  383. newState.expandedKeys = new Set(Object.keys(keyEntities));
  384. } else if (!prevProps && props.defaultExpandedKeys) {
  385. newState.expandedKeys = calcExpandedKeys(props.defaultExpandedKeys, keyEntities);
  386. } else if (!prevProps && props.defaultValue) {
  387. newState.expandedKeys = calcExpandedKeysForValues(
  388. normalizeValue(props.defaultValue, withObject),
  389. keyEntities,
  390. props.multiple,
  391. valueEntities
  392. );
  393. } else if (!prevProps && props.value) {
  394. newState.expandedKeys = calcExpandedKeysForValues(
  395. normalizeValue(props.value, withObject),
  396. keyEntities,
  397. props.multiple,
  398. valueEntities
  399. );
  400. }
  401. // flattenNodes
  402. if (treeData || needUpdate('expandedKeys')) {
  403. const flattenNodes = flattenTreeData(
  404. treeData || prevState.treeData,
  405. newState.expandedKeys || prevState.expandedKeys
  406. );
  407. newState.flattenNodes = flattenNodes;
  408. }
  409. // selectedKeys: single mode controlled
  410. const isMultiple = props.multiple;
  411. if (!isMultiple) {
  412. if (needUpdate('value')) {
  413. newState.selectedKeys = findKeysForValues(
  414. normalizeValue(props.value, withObject),
  415. valueEntities,
  416. isMultiple
  417. );
  418. } else if (!prevProps && props.defaultValue) {
  419. newState.selectedKeys = findKeysForValues(
  420. normalizeValue(props.defaultValue, withObject),
  421. valueEntities,
  422. isMultiple
  423. );
  424. } else if (treeData) {
  425. // If `treeData` changed, we also need check it
  426. if (props.value) {
  427. newState.selectedKeys = findKeysForValues(
  428. normalizeValue(props.value, withObject) || '',
  429. valueEntities,
  430. isMultiple
  431. );
  432. } else {
  433. newState.selectedKeys = updateKeys(prevState.selectedKeys, keyEntities);
  434. }
  435. }
  436. } else {
  437. // checkedKeys: multiple mode controlled || data changed
  438. let checkedKeyValues;
  439. if (needUpdate('value')) {
  440. checkedKeyValues = findKeysForValues(
  441. normalizeValue(props.value, withObject),
  442. valueEntities,
  443. isMultiple
  444. );
  445. } else if (!prevProps && props.defaultValue) {
  446. checkedKeyValues = findKeysForValues(
  447. normalizeValue(props.defaultValue, withObject),
  448. valueEntities,
  449. isMultiple
  450. );
  451. } else if (treeData) {
  452. // If `treeData` changed, we also need check it
  453. if (props.value) {
  454. checkedKeyValues = findKeysForValues(
  455. normalizeValue(props.value, withObject) || [],
  456. valueEntities,
  457. isMultiple
  458. );
  459. } else {
  460. checkedKeyValues = updateKeys(prevState.checkedKeys, keyEntities);
  461. }
  462. }
  463. if (checkedKeyValues) {
  464. if (props.checkRelation === 'unRelated') {
  465. newState.realCheckedKeys = new Set(checkedKeyValues);
  466. } else if (props.checkRelation === 'related') {
  467. const { checkedKeys, halfCheckedKeys } = calcCheckedKeys(checkedKeyValues, keyEntities);
  468. newState.checkedKeys = checkedKeys;
  469. newState.halfCheckedKeys = halfCheckedKeys;
  470. }
  471. }
  472. }
  473. // loadedKeys
  474. if (needUpdate('loadedKeys')) {
  475. newState.loadedKeys = new Set(props.loadedKeys);
  476. }
  477. // ================== rePosKey ==================
  478. if (needUpdate('treeData') || needUpdate('value')) {
  479. newState.rePosKey = rePosKey + 1;
  480. }
  481. // ================ disableStrictly =================
  482. if (treeData && props.disableStrictly && props.checkRelation === 'related') {
  483. newState.disabledKeys = calcDisabledKeys(keyEntities);
  484. }
  485. return newState;
  486. }
  487. get adapter(): TreeSelectAdapter<TreeSelectProps, TreeSelectState> {
  488. const filterAdapter: Pick<TreeSelectAdapter, 'updateInputValue'> = {
  489. updateInputValue: value => {
  490. this.setState({ inputValue: value });
  491. }
  492. };
  493. const treeSelectAdapter: Pick<TreeSelectAdapter,
  494. 'registerClickOutsideHandler'
  495. | 'unregisterClickOutsideHandler'
  496. | 'rePositionDropdown'
  497. > = {
  498. registerClickOutsideHandler: cb => {
  499. const clickOutsideHandler = (e: Event) => {
  500. const optionInstance = this.optionsRef && this.optionsRef.current as React.ReactInstance;
  501. const triggerDom = this.triggerRef && this.triggerRef.current;
  502. // eslint-disable-next-line
  503. const optionsDom = ReactDOM.findDOMNode(optionInstance);
  504. const target = e.target as Element;
  505. if (
  506. optionsDom &&
  507. (
  508. !optionsDom.contains(target) ||
  509. !optionsDom.contains(target.parentNode)
  510. ) &&
  511. triggerDom &&
  512. !triggerDom.contains(target)
  513. ) {
  514. cb(e);
  515. }
  516. };
  517. this.clickOutsideHandler = clickOutsideHandler;
  518. document.addEventListener('mousedown', clickOutsideHandler, false);
  519. },
  520. unregisterClickOutsideHandler: () => {
  521. document.removeEventListener('mousedown', this.clickOutsideHandler, false);
  522. this.clickOutsideHandler = null;
  523. },
  524. rePositionDropdown: () => {
  525. let { rePosKey } = this.state;
  526. rePosKey = rePosKey + 1;
  527. this.setState({ rePosKey });
  528. },
  529. };
  530. const treeAdapter: Pick<TreeSelectAdapter,
  531. 'updateState'
  532. | 'notifySelect'
  533. | 'notifySearch'
  534. | 'cacheFlattenNodes'
  535. | 'notifyLoad'
  536. > = {
  537. updateState: states => {
  538. this.setState({ ...states } as TreeSelectState);
  539. },
  540. notifySelect: ((selectKey, bool, node) => {
  541. this.props.onSelect && this.props.onSelect(selectKey, bool, node);
  542. }),
  543. notifySearch: (input, filteredExpandedKeys) => {
  544. this.props.onSearch && this.props.onSearch(input, filteredExpandedKeys);
  545. },
  546. cacheFlattenNodes: bool => {
  547. this._flattenNodes = bool ? cloneDeep(this.state.flattenNodes) : null;
  548. },
  549. notifyLoad: (newLoadedKeys, data) => {
  550. const { onLoad } = this.props;
  551. isFunction(onLoad) && onLoad(newLoadedKeys, data);
  552. }
  553. };
  554. return {
  555. ...super.adapter,
  556. ...filterAdapter,
  557. ...treeSelectAdapter,
  558. ...treeAdapter,
  559. updateLoadKeys: (data, resolve) => {
  560. this.setState(({ loadedKeys, loadingKeys }) =>
  561. this.foundation.handleNodeLoad(loadedKeys, loadingKeys, data, resolve));
  562. },
  563. updateState: states => {
  564. this.setState({ ...states } as TreeSelectState);
  565. },
  566. openMenu: () => {
  567. this.setState({ isOpen: true }, () => {
  568. this.props.onVisibleChange(true);
  569. });
  570. },
  571. closeMenu: cb => {
  572. this.setState({ isOpen: false }, () => {
  573. cb && cb();
  574. this.props.onVisibleChange(false);
  575. });
  576. },
  577. getTriggerWidth: () => {
  578. const el = this.triggerRef.current;
  579. return el && el.getBoundingClientRect().width;
  580. },
  581. setOptionWrapperWidth: width => {
  582. this.setState({ dropdownMinWidth: width });
  583. },
  584. notifyChange: (value, node, e) => {
  585. this.props.onChange && this.props.onChange(value, node, e);
  586. },
  587. notifyChangeWithObject: (node, e) => {
  588. this.props.onChange && this.props.onChange(node, e);
  589. },
  590. notifyExpand: (expandedKeys, { expanded: bool, node }) => {
  591. this.props.onExpand && this.props.onExpand([...expandedKeys], { expanded: bool, node });
  592. if (bool && this.props.loadData) {
  593. this.onNodeLoad(node);
  594. }
  595. },
  596. notifyFocus: (...v) => {
  597. this.props.onFocus && this.props.onFocus(...v);
  598. },
  599. notifyBlur: (...v) => {
  600. this.props.onBlur && this.props.onBlur(...v);
  601. },
  602. toggleHovering: bool => {
  603. this.setState({ isHovering: bool });
  604. },
  605. updateInputFocus: bool => { } // eslint-disable-line
  606. };
  607. }
  608. componentDidMount() {
  609. this.foundation.init();
  610. }
  611. componentWillUnmount() {
  612. this.foundation.destroy();
  613. }
  614. renderSuffix = () => {
  615. const { suffix }: any = this.props;
  616. const suffixWrapperCls = cls({
  617. [`${prefixcls}-suffix`]: true,
  618. [`${prefixcls}-suffix-text`]: suffix && isString(suffix),
  619. [`${prefixcls}-suffix-icon`]: isSemiIcon(suffix),
  620. });
  621. return (
  622. <div className={suffixWrapperCls} x-semi-prop="suffix">
  623. {suffix}
  624. </div>
  625. );
  626. };
  627. renderPrefix = () => {
  628. const { prefix, insetLabel, insetLabelId }: any = this.props;
  629. const labelNode = prefix || insetLabel;
  630. const prefixWrapperCls = cls({
  631. [`${prefixcls}-prefix`]: true,
  632. // to be doublechecked
  633. [`${prefixcls}-inset-label`]: insetLabel,
  634. [`${prefixcls}-prefix-text`]: labelNode && isString(labelNode),
  635. [`${prefixcls}-prefix-icon`]: isSemiIcon(labelNode),
  636. });
  637. return (
  638. <div className={prefixWrapperCls} id={insetLabelId} x-semi-prop="prefix,insetLabel">
  639. {labelNode}
  640. </div>
  641. );
  642. };
  643. renderContent = () => {
  644. const { dropdownMinWidth } = this.state;
  645. const { dropdownStyle, dropdownClassName } = this.props;
  646. const style = { minWidth: dropdownMinWidth, ...dropdownStyle };
  647. const popoverCls = cls(dropdownClassName, `${prefixcls}-popover`);
  648. return (
  649. <div className={popoverCls} style={style}>
  650. {this.renderTree()}
  651. </div>
  652. );
  653. };
  654. removeTag = (removedKey: TreeNodeData['key']) => {
  655. this.foundation.removeTag(removedKey);
  656. };
  657. handleClick = (e: React.MouseEvent) => {
  658. this.foundation.handleClick(e);
  659. };
  660. /* istanbul ignore next */
  661. handleSelectionEnterPress = (e: React.KeyboardEvent<HTMLDivElement>) => {
  662. this.foundation.handleSelectionEnterPress(e);
  663. };
  664. hasValue = (): boolean => {
  665. const { multiple, checkRelation } = this.props;
  666. const { realCheckedKeys, checkedKeys, selectedKeys } = this.state;
  667. let hasValue = false;
  668. if (multiple) {
  669. if (checkRelation === 'related') {
  670. hasValue = Boolean(checkedKeys.size);
  671. } else if (checkRelation === 'unRelated') {
  672. hasValue = Boolean(realCheckedKeys.size);
  673. }
  674. } else {
  675. hasValue = Boolean(selectedKeys.length);
  676. }
  677. return hasValue;
  678. }
  679. showClearBtn = () => {
  680. const { showClear, disabled, searchPosition } = this.props;
  681. const { inputValue, isOpen, isHovering } = this.state;
  682. const triggerSearchHasInputValue = searchPosition === strings.SEARCH_POSITION_TRIGGER && inputValue;
  683. return showClear && (this.hasValue() || triggerSearchHasInputValue) && !disabled && (isOpen || isHovering);
  684. };
  685. renderTagList = () => {
  686. const { checkedKeys, keyEntities, disabledKeys, realCheckedKeys } = this.state;
  687. const {
  688. treeNodeLabelProp,
  689. leafOnly,
  690. disabled,
  691. disableStrictly,
  692. size,
  693. checkRelation,
  694. renderSelectedItem: propRenderSelectedItem
  695. } = this.props;
  696. const renderSelectedItem = isFunction(propRenderSelectedItem) ?
  697. propRenderSelectedItem :
  698. (item: TreeNodeData) => ({
  699. isRenderInTag: true,
  700. content: get(item, treeNodeLabelProp, null)
  701. });
  702. let renderKeys = [];
  703. if (checkRelation === 'related') {
  704. renderKeys = normalizeKeyList([...checkedKeys], keyEntities, leafOnly);
  705. } else if (checkRelation === 'unRelated' && Object.keys(keyEntities).length > 0) {
  706. renderKeys = [...realCheckedKeys];
  707. }
  708. const tagList: Array<React.ReactNode> = [];
  709. // eslint-disable-next-line @typescript-eslint/no-shadow
  710. renderKeys.forEach((key: TreeNodeData['key']) => {
  711. const item = keyEntities[key].data;
  712. const onClose = (tagContent: any, e: React.MouseEvent) => {
  713. if (e && typeof e.preventDefault === 'function') {
  714. // make sure that tag will not hidden immediately in controlled mode
  715. e.preventDefault();
  716. }
  717. this.removeTag(key);
  718. };
  719. const { content, isRenderInTag } = (treeNodeLabelProp in item && item) ?
  720. (renderSelectedItem as RenderSelectedItemInMultiple)(item, { index: key, onClose }) :
  721. null;
  722. if (!content) {
  723. return;
  724. }
  725. const isDisabled = disabled || item.disabled || (disableStrictly && disabledKeys.has(item.key));
  726. const tag: Partial<TagProps> & React.Attributes = {
  727. closable: !isDisabled,
  728. color: 'white',
  729. visible: true,
  730. onClose,
  731. key,
  732. size: size === 'small' ? 'small' : 'large'
  733. };
  734. if (isRenderInTag) {
  735. // pass ReactNode list to tagList when using tagGroup custom mode
  736. tagList.push(<Tag {...tag}>{content}</Tag>);
  737. } else {
  738. tagList.push(content);
  739. }
  740. });
  741. return tagList;
  742. };
  743. /**
  744. * When single selection and the search box is on trigger, the items displayed in the rendered search box
  745. */
  746. renderSingleTriggerSearchItem = () => {
  747. const { placeholder, disabled } = this.props;
  748. const { inputTriggerFocus } = this.state;
  749. const renderText = this.foundation.getRenderTextInSingle();
  750. const spanCls = cls(`${prefixcls}-selection-TriggerSearchItem`, {
  751. [`${prefixcls}-selection-TriggerSearchItem-placeholder`]: (inputTriggerFocus || !renderText) && !disabled,
  752. [`${prefixcls}-selection-TriggerSearchItem-disabled`]: disabled,
  753. });
  754. return (
  755. <span className={spanCls}>
  756. {renderText ? renderText : placeholder}
  757. </span>
  758. );
  759. };
  760. /**
  761. * Single selection and the search box content rendered when the search box is on trigger
  762. */
  763. renderSingleTriggerSearch = () => {
  764. const { inputValue } = this.state;
  765. return (
  766. <>
  767. {!inputValue && this.renderSingleTriggerSearchItem()}
  768. {this.renderInput()}
  769. </>
  770. );
  771. };
  772. renderSelectContent = () => {
  773. const {
  774. multiple,
  775. placeholder,
  776. maxTagCount,
  777. searchPosition,
  778. filterTreeNode,
  779. showRestTagsPopover,
  780. restTagsPopoverProps
  781. } = this.props;
  782. const isTriggerPositionSearch = filterTreeNode && searchPosition === strings.SEARCH_POSITION_TRIGGER;
  783. // searchPosition = trigger
  784. if (isTriggerPositionSearch) {
  785. return multiple ? this.renderTagInput() : this.renderSingleTriggerSearch();
  786. }
  787. // searchPosition = dropdown and single seleciton
  788. if (!multiple || !this.hasValue()) {
  789. const renderText = this.foundation.getRenderTextInSingle();
  790. const spanCls = cls({
  791. [`${prefixcls}-selection-placeholder`]: !renderText,
  792. });
  793. return <span className={spanCls}>{renderText ? renderText : placeholder}</span>;
  794. }
  795. // searchPosition = dropdown and multiple seleciton
  796. const tagList = this.renderTagList();
  797. // mode=custom to return tagList directly
  798. return (
  799. <TagGroup<'custom'>
  800. maxTagCount={maxTagCount}
  801. tagList={tagList}
  802. size="large"
  803. mode="custom"
  804. showPopover={showRestTagsPopover}
  805. popoverProps={restTagsPopoverProps}
  806. />
  807. );
  808. };
  809. handleClear = (e: React.MouseEvent) => {
  810. e && e.stopPropagation();
  811. this.foundation.handleClear(e);
  812. };
  813. /* istanbul ignore next */
  814. handleClearEnterPress = (e: React.KeyboardEvent<HTMLDivElement>) => {
  815. e && e.stopPropagation();
  816. this.foundation.handleClearEnterPress(e);
  817. };
  818. handleMouseOver = (e: React.MouseEvent) => {
  819. this.foundation.toggleHoverState(true);
  820. };
  821. handleMouseLeave = (e: React.MouseEvent) => {
  822. this.foundation.toggleHoverState(false);
  823. };
  824. search = (value: string) => {
  825. const { isOpen } = this.state;
  826. if (!isOpen) {
  827. this.foundation.open();
  828. }
  829. this.foundation.handleInputChange(value);
  830. };
  831. close = () => {
  832. this.foundation.close(null);
  833. };
  834. renderArrow = () => {
  835. const showClearBtn = this.showClearBtn();
  836. const { arrowIcon } = this.props;
  837. if (showClearBtn) {
  838. return null;
  839. }
  840. return arrowIcon ? (
  841. <div className={cls(`${prefixcls}-arrow`)} x-semi-prop="arrowIcon">
  842. {arrowIcon}
  843. </div>
  844. ) : null;
  845. };
  846. renderClearBtn = () => {
  847. const showClearBtn = this.showClearBtn();
  848. const clearCls = cls(`${prefixcls}-clearbtn`);
  849. if (showClearBtn) {
  850. return (
  851. <div
  852. role='button'
  853. tabIndex={0}
  854. aria-label="Clear TreeSelect value"
  855. className={clearCls}
  856. onClick={this.handleClear}
  857. onKeyPress={this.handleClearEnterPress}
  858. >
  859. <IconClear />
  860. </div>
  861. );
  862. }
  863. return null;
  864. };
  865. renderSelection = () => {
  866. const {
  867. disabled,
  868. multiple,
  869. filterTreeNode,
  870. validateStatus,
  871. prefix,
  872. suffix,
  873. style,
  874. size,
  875. insetLabel,
  876. className,
  877. placeholder,
  878. showClear,
  879. leafOnly,
  880. searchPosition,
  881. triggerRender,
  882. } = this.props;
  883. const { isOpen, isInput, inputValue, selectedKeys, checkedKeys, keyEntities } = this.state;
  884. const filterable = Boolean(filterTreeNode);
  885. const useCustomTrigger = typeof triggerRender === 'function';
  886. const mouseEvent = showClear ?
  887. {
  888. onMouseEnter: (e: React.MouseEvent) => this.handleMouseOver(e),
  889. onMouseLeave: (e: React.MouseEvent) => this.handleMouseLeave(e),
  890. } :
  891. {};
  892. const isTriggerPositionSearch = searchPosition === strings.SEARCH_POSITION_TRIGGER && filterable;
  893. const isEmptyTriggerSearch = isTriggerPositionSearch && isEmpty(checkedKeys);
  894. const isValueTriggerSearch = isTriggerPositionSearch && !isEmpty(checkedKeys);
  895. const classNames = useCustomTrigger ?
  896. cls(className) :
  897. cls(
  898. prefixcls,
  899. {
  900. [`${prefixcls}-focus`]: isOpen && !isInput,
  901. [`${prefixcls}-disabled`]: disabled,
  902. [`${prefixcls}-single`]: !multiple,
  903. [`${prefixcls}-multiple`]: multiple,
  904. [`${prefixcls}-multiple-tagInput-empty`]: multiple && isEmptyTriggerSearch,
  905. [`${prefixcls}-multiple-tagInput-notEmpty`]: multiple && isValueTriggerSearch,
  906. [`${prefixcls}-filterable`]: filterable,
  907. [`${prefixcls}-error`]: validateStatus === 'error',
  908. [`${prefixcls}-warning`]: validateStatus === 'warning',
  909. [`${prefixcls}-small`]: size === 'small',
  910. [`${prefixcls}-large`]: size === 'large',
  911. [`${prefixcls}-with-prefix`]: prefix || insetLabel,
  912. [`${prefixcls}-with-suffix`]: suffix,
  913. [`${prefixcls}-with-suffix`]: suffix,
  914. },
  915. className
  916. );
  917. const triggerRenderKeys = multiple ? normalizeKeyList([...checkedKeys], keyEntities, leafOnly) : selectedKeys;
  918. const inner = useCustomTrigger ? (
  919. <Trigger
  920. inputValue={inputValue}
  921. // eslint-disable-next-line @typescript-eslint/no-shadow
  922. value={triggerRenderKeys.map((key: string) => get(keyEntities, [key, 'data']))}
  923. disabled={disabled}
  924. placeholder={placeholder}
  925. onClear={this.handleClear}
  926. componentName={'TreeSelect'}
  927. triggerRender={triggerRender}
  928. componentProps={{ ...this.props }}
  929. />
  930. ) : (
  931. [
  932. <Fragment key={'prefix'}>{prefix || insetLabel ? this.renderPrefix() : null}</Fragment>,
  933. <Fragment key={'selection'}>
  934. <div className={`${prefixcls}-selection`}>{this.renderSelectContent()}</div>
  935. </Fragment>,
  936. <Fragment key={'suffix'}>{suffix ? this.renderSuffix() : null}</Fragment>,
  937. <Fragment key={'clearBtn'}>
  938. {
  939. (showClear || (isTriggerPositionSearch && inputValue)) ?
  940. this.renderClearBtn() :
  941. null
  942. }
  943. </Fragment>,
  944. <Fragment key={'arrow'}>{this.renderArrow()}</Fragment>,
  945. ]
  946. );
  947. const tabIndex = disabled ? null : 0;
  948. /**
  949. * Reasons for disabling the a11y eslint rule:
  950. * The following attributes(aria-controls,aria-expanded) will be automatically added by Tooltip, no need to declare here
  951. */
  952. return (
  953. <div
  954. // eslint-disable-next-line jsx-a11y/role-has-required-aria-props
  955. role='combobox'
  956. aria-disabled={disabled}
  957. aria-haspopup="tree"
  958. tabIndex={tabIndex}
  959. className={classNames}
  960. style={style}
  961. ref={this.triggerRef}
  962. onClick={this.handleClick}
  963. onKeyPress={this.handleSelectionEnterPress}
  964. aria-invalid={this.props['aria-invalid']}
  965. aria-errormessage={this.props['aria-errormessage']}
  966. aria-label={this.props['aria-label']}
  967. aria-labelledby={this.props['aria-labelledby']}
  968. aria-describedby={this.props['aria-describedby']}
  969. aria-required={this.props['aria-required']}
  970. {...mouseEvent}
  971. >
  972. {inner}
  973. </div>
  974. );
  975. };
  976. // eslint-disable-next-line @typescript-eslint/no-shadow
  977. renderTagItem = (key: string, idx: number) => {
  978. const { keyEntities, disabledKeys } = this.state;
  979. const {
  980. size,
  981. leafOnly,
  982. disabled,
  983. disableStrictly,
  984. renderSelectedItem: propRenderSelectedItem,
  985. treeNodeLabelProp
  986. } = this.props;
  987. const keyList = normalizeKeyList([key], keyEntities, leafOnly);
  988. const nodes = keyList.map(i => keyEntities[i].data);
  989. const value = getValueOrKey(nodes);
  990. const tagCls = cls(`${prefixcls}-selection-tag`, {
  991. [`${prefixcls}-selection-tag-disabled`]: disabled,
  992. });
  993. const nodeHaveData = !isEmpty(nodes) && !isEmpty(nodes[0]);
  994. const isDisableStrictlyNode = disableStrictly && nodeHaveData && disabledKeys.has(nodes[0].key);
  995. const closable = nodeHaveData && !nodes[0].disabled && !disabled && !isDisableStrictlyNode;
  996. const onClose = (tagChildren: React.ReactNode, e: React.MouseEvent) => {
  997. // When value has not changed, prevent clicking tag closeBtn to close tag
  998. e.preventDefault();
  999. this.removeTag(key);
  1000. };
  1001. const tagProps: Partial<TagProps> & React.Attributes = {
  1002. size: size === 'small' ? 'small' : 'large',
  1003. key: `tag-${value}-${idx}`,
  1004. color: 'white',
  1005. className: tagCls,
  1006. closable,
  1007. onClose,
  1008. };
  1009. const item = nodes[0];
  1010. const renderSelectedItem = isFunction(propRenderSelectedItem) ? propRenderSelectedItem :
  1011. (selectedItem: TreeNodeData) => ({
  1012. isRenderInTag: true,
  1013. content: get(selectedItem, treeNodeLabelProp, null)
  1014. });
  1015. if (isFunction(renderSelectedItem)) {
  1016. const { content, isRenderInTag } = treeNodeLabelProp in item && item ?
  1017. (renderSelectedItem as RenderSelectedItemInMultiple)(item, { index: idx, onClose }) :
  1018. null;
  1019. if (isRenderInTag) {
  1020. return <Tag {...tagProps}>{content}</Tag>;
  1021. } else {
  1022. return content;
  1023. }
  1024. }
  1025. return (
  1026. <Tag {...tagProps}>
  1027. {value}
  1028. </Tag>
  1029. );
  1030. };
  1031. renderTagInput = () => {
  1032. const {
  1033. leafOnly,
  1034. disabled,
  1035. size,
  1036. searchAutoFocus,
  1037. placeholder,
  1038. maxTagCount,
  1039. checkRelation,
  1040. showRestTagsPopover,
  1041. restTagsPopoverProps
  1042. } = this.props;
  1043. const {
  1044. keyEntities,
  1045. checkedKeys,
  1046. inputValue,
  1047. realCheckedKeys,
  1048. } = this.state;
  1049. let keyList = [];
  1050. if (checkRelation === 'related') {
  1051. keyList = normalizeKeyList(checkedKeys, keyEntities, leafOnly);
  1052. } else if (checkRelation === 'unRelated') {
  1053. keyList = [...realCheckedKeys];
  1054. }
  1055. return (
  1056. <TagInput
  1057. maxTagCount={maxTagCount}
  1058. disabled={disabled}
  1059. onInputChange={v => this.search(v)}
  1060. ref={this.tagInputRef}
  1061. placeholder={placeholder}
  1062. value={keyList}
  1063. inputValue={inputValue}
  1064. size={size}
  1065. showRestTagsPopover={showRestTagsPopover}
  1066. restTagsPopoverProps={restTagsPopoverProps}
  1067. // eslint-disable-next-line jsx-a11y/no-autofocus
  1068. autoFocus={searchAutoFocus}
  1069. renderTagItem={(itemKey, index) => this.renderTagItem(itemKey, index)}
  1070. onRemove={itemKey => this.removeTag(itemKey)}
  1071. expandRestTagsOnClick={false}
  1072. />
  1073. );
  1074. };
  1075. // render Tree
  1076. renderInput = () => {
  1077. const {
  1078. searchPlaceholder,
  1079. searchRender,
  1080. showSearchClear,
  1081. searchPosition,
  1082. searchAutoFocus,
  1083. multiple,
  1084. disabled,
  1085. } = this.props;
  1086. const isDropdownPositionSearch = searchPosition === strings.SEARCH_POSITION_DROPDOWN;
  1087. const inputcls = cls({
  1088. [`${prefixTree}-input`]: isDropdownPositionSearch,
  1089. [`${prefixcls}-inputTrigger`]: !isDropdownPositionSearch
  1090. });
  1091. const { inputValue } = this.state;
  1092. const baseInputProps = {
  1093. value: inputValue,
  1094. className: inputcls,
  1095. onChange: (value: string) => this.search(value),
  1096. };
  1097. const inputDropdownProps = {
  1098. showClear: showSearchClear,
  1099. prefix: <IconSearch />,
  1100. };
  1101. const inputTriggerProps = {
  1102. onFocus: (e: React.FocusEvent) => this.foundation.handleInputTriggerFocus(),
  1103. onBlur: (e: React.FocusEvent) => this.foundation.handleInputTriggerBlur(),
  1104. disabled,
  1105. };
  1106. const realInputProps = isDropdownPositionSearch ? inputDropdownProps : inputTriggerProps;
  1107. const wrapperCls = cls({
  1108. [`${prefixTree}-search-wrapper`]: isDropdownPositionSearch,
  1109. [`${prefixcls}-triggerSingleSearch-wrapper`]: !isDropdownPositionSearch && !multiple,
  1110. });
  1111. const useCusSearch = typeof searchRender === 'function' || typeof searchRender === 'boolean';
  1112. if (useCusSearch && !searchRender) {
  1113. return null;
  1114. }
  1115. return (
  1116. <div className={wrapperCls}>
  1117. <LocaleConsumer componentName="TreeSelect">
  1118. {(locale: Locale['TreeSelect']) => {
  1119. const placeholder = isDropdownPositionSearch ?
  1120. searchPlaceholder || locale.searchPlaceholder :
  1121. '';
  1122. if (useCusSearch) {
  1123. return (searchRender as any)({ ...realInputProps, ...baseInputProps, placeholder });
  1124. }
  1125. return (
  1126. <Input
  1127. aria-label='Filter TreeSelect item'
  1128. ref={this.inputRef as any}
  1129. autofocus={searchAutoFocus}
  1130. placeholder={placeholder}
  1131. {...baseInputProps}
  1132. {...realInputProps}
  1133. />
  1134. );
  1135. }}
  1136. </LocaleConsumer>
  1137. </div>
  1138. );
  1139. };
  1140. renderEmpty = () => {
  1141. const { emptyContent } = this.props;
  1142. if (emptyContent) {
  1143. return <TreeNode empty emptyContent={this.props.emptyContent} />;
  1144. } else {
  1145. return (
  1146. <LocaleConsumer componentName="Tree">
  1147. {(locale: Locale['Tree']) => <TreeNode empty emptyContent={locale.emptyText} />}
  1148. </LocaleConsumer>
  1149. );
  1150. }
  1151. };
  1152. onNodeLoad = (data: TreeNodeData) => new Promise(resolve => this.foundation.setLoadKeys(data, resolve));
  1153. onNodeSelect = (e: React.MouseEvent | React.KeyboardEvent, treeNode: TreeNodeProps) => {
  1154. this.foundation.handleNodeSelect(e, treeNode);
  1155. };
  1156. onNodeCheck = (e: React.MouseEvent | React.KeyboardEvent, treeNode: TreeNodeProps) => {
  1157. this.foundation.handleNodeSelect(e, treeNode);
  1158. };
  1159. onNodeExpand = (e: React.MouseEvent | React.KeyboardEvent, treeNode: TreeNodeProps) => {
  1160. this.foundation.handleNodeExpand(e, treeNode);
  1161. };
  1162. getTreeNodeRequiredProps = () => {
  1163. const { expandedKeys, selectedKeys, checkedKeys, halfCheckedKeys, keyEntities, filteredKeys } = this.state;
  1164. return {
  1165. expandedKeys: expandedKeys || new Set(),
  1166. selectedKeys: selectedKeys || [],
  1167. checkedKeys: checkedKeys || new Set(),
  1168. halfCheckedKeys: halfCheckedKeys || new Set(),
  1169. filteredKeys: filteredKeys || new Set(),
  1170. keyEntities,
  1171. };
  1172. };
  1173. getTreeNodeKey = (treeNode: TreeNodeData) => {
  1174. const { data } = treeNode;
  1175. // eslint-disable-next-line @typescript-eslint/no-shadow
  1176. const { key }: { key: string } = data;
  1177. return key;
  1178. };
  1179. /* Event handler function after popover is closed */
  1180. handlePopoverClose = isVisible => {
  1181. const { filterTreeNode } = this.props;
  1182. if (isVisible === false && Boolean(filterTreeNode)) {
  1183. this.foundation.clearInput();
  1184. }
  1185. }
  1186. renderTreeNode = (treeNode: FlattenNode, ind: number, style: React.CSSProperties) => {
  1187. const { data } = treeNode;
  1188. // eslint-disable-next-line @typescript-eslint/no-shadow
  1189. const { key }: { key: string } = data;
  1190. const treeNodeProps = this.foundation.getTreeNodeProps(key);
  1191. if (!treeNodeProps) {
  1192. return null;
  1193. }
  1194. return <TreeNode {...treeNodeProps} {...data} key={key} data={data} style={style} />;
  1195. };
  1196. itemKey = (index: number, data: TreeNodeData) => {
  1197. // Find the item at the specified index.
  1198. const item = data[index];
  1199. // Return a value that uniquely identifies this item.
  1200. return item.key;
  1201. };
  1202. renderNodeList = () => {
  1203. const { flattenNodes, motionKeys, motionType, filteredKeys } = this.state;
  1204. const { direction } = this.context;
  1205. const { virtualize, motionExpand } = this.props;
  1206. const isExpandControlled = 'expandedKeys' in this.props;
  1207. if (!virtualize || isEmpty(virtualize)) {
  1208. return (
  1209. <NodeList
  1210. flattenNodes={flattenNodes}
  1211. flattenList={this._flattenNodes}
  1212. motionKeys={motionExpand ? motionKeys : new Set([])}
  1213. motionType={motionType}
  1214. // When motionKeys is empty, but filteredKeys is not empty (that is, the search hits), this situation should be distinguished from ordinary motionKeys
  1215. searchTargetIsDeep={
  1216. isExpandControlled &&
  1217. motionExpand &&
  1218. isEmpty(motionKeys) &&
  1219. !isEmpty(filteredKeys)
  1220. }
  1221. onMotionEnd={this.onMotionEnd}
  1222. renderTreeNode={this.renderTreeNode}
  1223. />
  1224. );
  1225. }
  1226. const option = ({ index, style, data }: OptionProps) => this.renderTreeNode(data[index], index, style);
  1227. return (
  1228. <AutoSizer defaultHeight={virtualize.height} defaultWidth={virtualize.width}>
  1229. {({ height, width }) => (
  1230. <VirtualList
  1231. itemCount={flattenNodes.length}
  1232. itemSize={virtualize.itemSize}
  1233. height={height}
  1234. width={width}
  1235. // @ts-ignore avoid strict check of itemKey
  1236. itemKey={this.itemKey as ListItemKeySelector<TreeNodeData>}
  1237. itemData={flattenNodes as any}
  1238. className={`${prefixTree}-virtual-list`}
  1239. style={{ direction }}
  1240. >
  1241. {option}
  1242. </VirtualList>
  1243. )}
  1244. </AutoSizer>
  1245. );
  1246. };
  1247. renderTree = () => {
  1248. const { keyEntities, motionKeys, motionType, inputValue, filteredKeys, flattenNodes, checkedKeys, realCheckedKeys } = this.state;
  1249. const {
  1250. loadData,
  1251. filterTreeNode,
  1252. disabled,
  1253. multiple,
  1254. showFilteredOnly,
  1255. motionExpand,
  1256. outerBottomSlot,
  1257. outerTopSlot,
  1258. expandAction,
  1259. labelEllipsis,
  1260. virtualize,
  1261. optionListStyle,
  1262. searchPosition,
  1263. renderLabel,
  1264. renderFullLabel,
  1265. checkRelation,
  1266. } = this.props;
  1267. const wrapperCls = cls(`${prefixTree}-wrapper`);
  1268. const listCls = cls(`${prefixTree}-option-list`, {
  1269. [`${prefixTree}-option-list-block`]: true,
  1270. });
  1271. const searchNoRes = Boolean(inputValue) && !filteredKeys.size;
  1272. const noData = isEmpty(flattenNodes) || (showFilteredOnly && searchNoRes);
  1273. const isDropdownPositionSearch = searchPosition === strings.SEARCH_POSITION_DROPDOWN;
  1274. return (
  1275. <TreeContext.Provider
  1276. value={{
  1277. loadData,
  1278. treeDisabled: disabled,
  1279. motion: motionExpand,
  1280. motionKeys,
  1281. motionType,
  1282. expandAction,
  1283. filterTreeNode,
  1284. keyEntities,
  1285. onNodeClick: this.onNodeClick,
  1286. onNodeDoubleClick: this.onNodeDoubleClick,
  1287. // tree node will call this function when treeNode is right clicked
  1288. onNodeRightClick: noop,
  1289. onNodeExpand: this.onNodeExpand,
  1290. onNodeSelect: this.onNodeSelect,
  1291. onNodeCheck: this.onNodeCheck,
  1292. renderTreeNode: this.renderTreeNode,
  1293. multiple,
  1294. showFilteredOnly,
  1295. isSearching: Boolean(inputValue),
  1296. renderLabel,
  1297. renderFullLabel,
  1298. labelEllipsis: typeof labelEllipsis === 'undefined' ? virtualize : labelEllipsis,
  1299. }}
  1300. >
  1301. <div className={wrapperCls}>
  1302. {outerTopSlot}
  1303. {
  1304. !outerTopSlot &&
  1305. filterTreeNode &&
  1306. isDropdownPositionSearch &&
  1307. this.renderInput()
  1308. }
  1309. <div className={listCls} role="tree" aria-multiselectable={multiple ? true : false} style={optionListStyle}>
  1310. { noData ? this.renderEmpty() : (multiple ?
  1311. (<CheckboxGroup value={Array.from(checkRelation === 'related' ? checkedKeys : realCheckedKeys)}>
  1312. {this.renderNodeList()}
  1313. </CheckboxGroup>) :
  1314. this.renderNodeList()
  1315. )}
  1316. </div>
  1317. {outerBottomSlot}
  1318. </div>
  1319. </TreeContext.Provider>
  1320. );
  1321. };
  1322. render() {
  1323. const content = this.renderContent();
  1324. const {
  1325. motion,
  1326. zIndex,
  1327. mouseLeaveDelay,
  1328. mouseEnterDelay,
  1329. autoAdjustOverflow,
  1330. stopPropagation,
  1331. getPopupContainer,
  1332. } = this.props;
  1333. const { isOpen, rePosKey } = this.state;
  1334. const selection = this.renderSelection();
  1335. const pos = 'bottomLeft';
  1336. return (
  1337. <Popover
  1338. stopPropagation={stopPropagation}
  1339. getPopupContainer={getPopupContainer}
  1340. zIndex={zIndex}
  1341. motion={motion}
  1342. ref={this.optionsRef}
  1343. content={content}
  1344. visible={isOpen}
  1345. trigger="custom"
  1346. rePosKey={rePosKey}
  1347. position={pos}
  1348. autoAdjustOverflow={autoAdjustOverflow}
  1349. mouseLeaveDelay={mouseLeaveDelay}
  1350. mouseEnterDelay={mouseEnterDelay}
  1351. onVisibleChange={this.handlePopoverClose}
  1352. >
  1353. {selection}
  1354. </Popover>
  1355. );
  1356. }
  1357. }
  1358. export default TreeSelect;