@douyinfe/semi-foundation 2.24.2 → 2.25.0-alpha.0

package/lib/cjs/treeSelect/foundation.js

@@ -4,39 +4,34 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.default = void 0;
-
 var _difference2 = _interopRequireDefault(require("lodash/difference"));
-
 var _isEmpty2 = _interopRequireDefault(require("lodash/isEmpty"));
-
-var _cloneDeep2 = _interopRequireDefault(require("lodash/cloneDeep"));
-
 var _isString2 = _interopRequireDefault(require("lodash/isString"));
-
 var _isUndefined2 = _interopRequireDefault(require("lodash/isUndefined"));
-
 var _get2 = _interopRequireDefault(require("lodash/get"));
-
 var _isFunction2 = _interopRequireDefault(require("lodash/isFunction"));
-
 var _isNumber2 = _interopRequireDefault(require("lodash/isNumber"));
-
 var _constants = require("../treeSelect/constants");
-
 var _foundation = _interopRequireDefault(require("../base/foundation"));
-
 var _treeUtil = require("../tree/treeUtil");
-
 var _isEnterPress = _interopRequireDefault(require("../utils/isEnterPress"));
-
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-// eslint-disable-next-line max-len
 class TreeSelectFoundation extends _foundation.default {
   constructor(adapter) {
     super(Object.assign({}, adapter));
+    this._registerClickOutsideHandler = e => {
+      this._adapter.registerClickOutsideHandler(e => {
+        this.handlerTriggerBlur(e);
+        this.close(e);
+      });
+    };
+    this.clearInputValue = () => {
+      const {
+        inputValue
+      } = this.getStates();
+      inputValue && this._adapter.updateInputValue('');
+    };
   }
-
   init() {
     const {
       searchAutoFocus,
@@ -45,28 +40,25 @@ class TreeSelectFoundation extends _foundation.default {
     } = this.getProps();
     const triggerSearch = searchPosition === _constants.strings.SEARCH_POSITION_TRIGGER && filterTreeNode;
     const triggerSearchAutoFocus = searchAutoFocus && triggerSearch;
-
     this._setDropdownWidth();
-
     const isOpen = (this.getProp('defaultOpen') || triggerSearchAutoFocus) && !this._isDisabled();
-
     if (isOpen) {
       this.open();
     }
+    if (triggerSearchAutoFocus) {
+      this.handleTriggerFocus(null);
+    }
   }
-
   destroy() {
     // Ensure that event monitoring will be uninstalled, and the user may not trigger closePanel
     this._adapter.unregisterClickOutsideHandler();
   }
-
   _setDropdownWidth() {
     const {
       style,
       dropdownMatchSelectWidth
     } = this.getProps();
     let width;
-
     if (dropdownMatchSelectWidth) {
       if (style && (0, _isNumber2.default)(style.width)) {
         width = style.width;
@@ -75,36 +67,28 @@ class TreeSelectFoundation extends _foundation.default {
       } else {
         width = this._adapter.getTriggerWidth();
       }
-
       this._adapter.setOptionWrapperWidth(width);
     }
   }
-
   _isMultiple() {
     return this.getProp('multiple');
   }
-
   _isAnimated() {
     return this.getProp('motionExpand');
   }
-
   _isDisabled() {
     let treeNode = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
     return this.getProp('disabled') || treeNode.disabled;
   }
-
   _isExpandControlled() {
     return this.getProp('expandedKeys');
   }
-
   _isSelectToClose() {
     return !this.getProp('expandAction');
   }
-
   _isLoadControlled() {
     return this.getProp('loadedKeys');
   }
-
   _showFilteredOnly() {
     const {
       inputValue
@@ -114,15 +98,46 @@ class TreeSelectFoundation extends _foundation.default {
     } = this.getProps();
     return Boolean(inputValue) && showFilteredOnly;
   }
-
-  getCopyFromState(items) {
-    const res = {};
-    (0, _treeUtil.normalizedArr)(items).forEach(key => {
-      res[key] = (0, _cloneDeep2.default)(this.getState(key));
+  findDataForValue(findValue) {
+    const {
+      value,
+      defaultValue,
+      keyMaps
+    } = this.getProps();
+    const realValueName = (0, _get2.default)(keyMaps, 'value', 'value');
+    const realKeyName = (0, _get2.default)(keyMaps, 'key', 'key');
+    let valueArr = [];
+    if (value) {
+      valueArr = Array.isArray(value) ? value : [value];
+    } else if (defaultValue) {
+      valueArr = Array.isArray(defaultValue) ? defaultValue : [defaultValue];
+    }
+    return valueArr.find(item => {
+      return item[realValueName] === findValue || item[realKeyName] === findValue;
     });
-    return res;
   }
-
+  constructDataForValue(value) {
+    const {
+      treeNodeLabelProp,
+      keyMaps
+    } = this.getProps();
+    const keyName = (0, _get2.default)(keyMaps, 'key', 'key');
+    const labelName = (0, _get2.default)(keyMaps, 'label', treeNodeLabelProp);
+    return {
+      [keyName]: value,
+      [labelName]: value
+    };
+  }
+  getDataForKeyNotInKeyEntities(value) {
+    const {
+      onChangeWithObject
+    } = this.getProps();
+    if (onChangeWithObject) {
+      return this.findDataForValue(value);
+    } else {
+      return this.constructDataForValue(value);
+    }
+  }
   getTreeNodeProps(key) {
     const {
       expandedKeys = new Set([]),
@@ -144,15 +159,12 @@ class TreeSelectFoundation extends _foundation.default {
     } = this.getProps();
     const entity = keyEntities[key];
     const notExist = !entity;
-
     if (notExist) {
       return null;
-    } // if checkRelation is invalid, the checked status of node will be false
-
-
+    }
+    // if checkRelation is invalid, the checked status of node will be false
     let realChecked = false;
     let realHalfChecked = false;
-
     if (checkRelation === 'related') {
       realChecked = checkedKeys.has(key);
       realHalfChecked = halfCheckedKeys.has(key);
@@ -160,11 +172,10 @@ class TreeSelectFoundation extends _foundation.default {
       realChecked = realCheckedKeys.has(key);
       realHalfChecked = false;
     }
-
     const isSearching = Boolean(inputValue);
     const treeNodeProps = {
       eventKey: key,
-      expanded: isSearching && !this._isExpandControlled() ? filteredExpandedKeys.has(key) : expandedKeys.has(key),
+      expanded: isSearching ? filteredExpandedKeys.has(key) : expandedKeys.has(key),
       selected: selectedKeys.includes(key),
       checked: realChecked,
       halfChecked: realHalfChecked,
@@ -176,14 +187,11 @@ class TreeSelectFoundation extends _foundation.default {
       loading: loadingKeys.has(key) && !loadedKeys.has(key),
       loaded: loadedKeys.has(key)
     };
-
     if (this.getProp('disableStrictly') && disabledKeys.has(key)) {
       treeNodeProps.disabled = true;
     }
-
     return treeNodeProps;
   }
-
   handleNodeLoad(loadedKeys, loadingKeys, data, resolve) {
     const {
       loadData
@@ -191,32 +199,24 @@ class TreeSelectFoundation extends _foundation.default {
     const {
       key
     } = data;
-
     if (!loadData || loadedKeys.has(key) || loadingKeys.has(key)) {
       return {};
     }
-
     loadData(data).then(() => {
-      const {
-        loadedKeys: prevLoadedKeys,
-        loadingKeys: prevLoadingKeys
-      } = this.getCopyFromState(['loadedKeys', 'loadingKeys']);
+      const prevLoadedKeys = new Set(this.getState('loadedKeys'));
+      const prevLoadingKeys = new Set(this.getState('loadingKeys'));
       const newLoadedKeys = prevLoadedKeys.add(key);
       const newLoadingKeys = new Set([...prevLoadingKeys]);
       newLoadingKeys.delete(key);
-
       this._adapter.notifyLoad(newLoadedKeys, data);
-
       if (!this._isLoadControlled()) {
         this._adapter.updateState({
           loadedKeys: newLoadedKeys
         });
       }
-
       this._adapter.setState({
         loadingKeys: newLoadingKeys
       });
-
       resolve();
     });
     return {
@@ -224,138 +224,124 @@ class TreeSelectFoundation extends _foundation.default {
     };
   }
   /* istanbul ignore next */
-
-
   focusInput(bool) {
     this._adapter.updateInputFocus(bool);
   }
-
   _notifyMultipleChange(key, e) {
     const {
       keyEntities
     } = this.getStates();
     const {
       leafOnly,
-      checkRelation
+      checkRelation,
+      keyMaps
     } = this.getProps();
     let keyList = [];
-
     if (checkRelation === 'related') {
-      keyList = (0, _treeUtil.normalizeKeyList)(key, keyEntities, leafOnly);
+      keyList = (0, _treeUtil.normalizeKeyList)(key, keyEntities, leafOnly, true);
     } else if (checkRelation === 'unRelated') {
       keyList = key;
     }
-
-    const nodes = keyList.map(i => keyEntities[i].data);
-
+    const nodes = keyList.map(key => keyEntities[key] && keyEntities[key].key === key ? keyEntities[key].data : this.getDataForKeyNotInKeyEntities(key));
     if (this.getProp('onChangeWithObject')) {
       this._adapter.notifyChangeWithObject(nodes, e);
     } else {
-      const value = (0, _treeUtil.getValueOrKey)(nodes);
-
+      const value = (0, _treeUtil.getValueOrKey)(nodes, keyMaps);
       this._adapter.notifyChange(value, nodes, e);
     }
   }
-
   _notifyChange(key, e) {
     const {
       keyEntities
     } = this.getStates();
-
+    const {
+      keyMaps
+    } = this.getProps();
     if (this._isMultiple() && Array.isArray(key)) {
       this._notifyMultipleChange(key, e);
     } else {
       const nodes = (0, _isUndefined2.default)(key) ? key : keyEntities[key].data;
-      const value = (0, _isUndefined2.default)(key) ? key : (0, _treeUtil.getValueOrKey)(nodes);
-
+      const value = (0, _isUndefined2.default)(key) ? key : (0, _treeUtil.getValueOrKey)(nodes, keyMaps);
       if (this.getProp('onChangeWithObject')) {
         this._adapter.notifyChangeWithObject(nodes, e);
       } else {
         this._adapter.notifyChange(value, nodes, e);
       }
     }
-  } // Scenes that may trigger focus:
+  }
+  // Scenes that may trigger focus:
   //  1、click selection
-
-
   _notifyFocus(e) {
     this._adapter.notifyFocus(e);
-  } // Scenes that may trigger blur
+  }
+  handleTriggerFocus(e) {
+    this._adapter.updateIsFocus(true);
+    this._notifyFocus(e);
+    this._registerClickOutsideHandler(e);
+  }
+  // Scenes that may trigger blur
   //  1、clickOutSide
   //  2、click option / press enter, and then select complete（when multiple is false
   //  3、press esc when dropdown list open
-
-
   _notifyBlur(e) {
     this._adapter.notifyBlur(e);
   }
-
+  handlerTriggerBlur(e) {
+    this._adapter.updateIsFocus(false);
+    this._notifyBlur(e);
+    this._adapter.unregisterClickOutsideHandler();
+  }
   toggleHoverState(bool) {
     this._adapter.toggleHovering(bool);
   }
-
   open() {
     this._adapter.openMenu();
-
     this._setDropdownWidth();
-
-    this._adapter.registerClickOutsideHandler(e => {
-      this.close(e);
-    });
   }
-
   close(e) {
     this._adapter.closeMenu();
-
-    this._adapter.unregisterClickOutsideHandler();
-
-    this._notifyBlur(e);
-
     if (this.getProp('motionExpand')) {
       this._adapter.updateState({
         motionKeys: new Set([])
       });
     }
   }
-
   handleClick(e) {
     const isDisabled = this._isDisabled();
-
     const {
       isOpen,
-      inputValue
+      inputValue,
+      isFocus
     } = this.getStates();
     const {
-      searchPosition
+      searchPosition,
+      clickTriggerToHide
     } = this.getProps();
-
     if (isDisabled) {
       return;
-    } else if (!isOpen) {
-      this.open();
-
-      this._notifyFocus(e);
-    } else if (isOpen) {
-      if (searchPosition === 'trigger' && inputValue) {
-        return;
+    } else {
+      if (!isFocus) {
+        this.handleTriggerFocus(e);
+      }
+      if (isOpen) {
+        if (searchPosition === 'trigger' && inputValue) {
+          return;
+        }
+        clickTriggerToHide && this.close(e);
+      } else {
+        this.open();
       }
-
-      this.close(e);
     }
   }
   /**
    * A11y: simulate selection click
    */
-
   /* istanbul ignore next */
-
-
   handleSelectionEnterPress(e) {
     if ((0, _isEnterPress.default)(e)) {
       this.handleClick(e);
     }
   }
-
   handleClear(e) {
     const {
       searchPosition,
@@ -365,28 +351,21 @@ class TreeSelectFoundation extends _foundation.default {
       inputValue,
       selectedKeys
     } = this.getStates();
-
     const isMultiple = this._isMultiple();
-
     const isControlled = this._isControlledComponent();
-
     const value = isMultiple ? [] : undefined;
-
     this._notifyChange(value, e);
-
     if (!isControlled) {
       // reposition dropdown when selected values change
       this._adapter.rePositionDropdown();
-
       this._adapter.updateState({
         selectedKeys: [],
         checkedKeys: new Set(),
         halfCheckedKeys: new Set(),
         realCheckedKeys: new Set([])
       });
-    } // When triggerSearch, clicking the clear button will trigger to clear Input
-
-
+    }
+    // When triggerSearch, clicking the clear button will trigger to clear Input
     if (filterTreeNode && searchPosition === _constants.strings.SEARCH_POSITION_TRIGGER) {
       if (inputValue !== '') {
         if ((0, _isEmpty2.default)(selectedKeys)) {
@@ -396,47 +375,41 @@ class TreeSelectFoundation extends _foundation.default {
         }
       }
     }
+    this._adapter.notifyClear(e);
   }
   /**
    * A11y: simulate clear button click
    */
-
   /* istanbul ignore next */
-
-
   handleClearEnterPress(e) {
     if ((0, _isEnterPress.default)(e)) {
       this.handleClear(e);
     }
   }
-
   removeTag(eventKey) {
     const {
       disableStrictly,
-      checkRelation
+      checkRelation,
+      keyMaps
     } = this.getProps();
     const {
       keyEntities,
       disabledKeys,
       realCheckedKeys
     } = this.getStates();
-    const item = keyEntities[eventKey].data;
-
-    if (item.disabled || disableStrictly && disabledKeys.has(eventKey)) {
+    const item = keyEntities[eventKey] && keyEntities[eventKey].key === eventKey ? keyEntities[eventKey].data : this.getDataForKeyNotInKeyEntities(eventKey);
+    const disabledName = (0, _get2.default)(keyMaps, 'disabled', 'disabled');
+    if (item[disabledName] || disableStrictly && disabledKeys.has(eventKey)) {
       return;
     }
-
     if (checkRelation === 'unRelated') {
       const newRealCheckedKeys = new Set(realCheckedKeys);
       newRealCheckedKeys.delete(eventKey);
-
       this._notifyChange([...newRealCheckedKeys], null);
-
       if (!this._isControlledComponent()) {
         this._adapter.updateState({
           realCheckedKeys: newRealCheckedKeys
         });
-
         this._adapter.rePositionDropdown();
       }
     } else if (checkRelation === 'related') {
@@ -444,25 +417,19 @@ class TreeSelectFoundation extends _foundation.default {
         checkedKeys,
         halfCheckedKeys
       } = this.calcCheckedKeys(eventKey, false);
-
       this._notifyChange([...checkedKeys], null);
-
       if (!this._isControlledComponent()) {
         this._adapter.updateState({
           checkedKeys,
           halfCheckedKeys
         });
-
         this._adapter.rePositionDropdown();
       }
     }
-
-    this._adapter.notifySelect(eventKey, false, item); // reposition dropdown when selected values change
-
-
+    this._adapter.notifySelect(eventKey, false, item);
+    // reposition dropdown when selected values change
     this._adapter.rePositionDropdown();
   }
-
   clearInput() {
     const {
       flattenNodes,
@@ -471,17 +438,17 @@ class TreeSelectFoundation extends _foundation.default {
       keyEntities,
       treeData
     } = this.getStates();
+    const {
+      keyMaps
+    } = this.getProps();
     const newExpandedKeys = new Set(expandedKeys);
-
     const isExpandControlled = this._isExpandControlled();
-
     const expandedOptsKeys = (0, _treeUtil.findAncestorKeys)(selectedKeys, keyEntities);
     expandedOptsKeys.forEach(item => newExpandedKeys.add(item));
-    const newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, newExpandedKeys);
-
+    const newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, newExpandedKeys, keyMaps);
     this._adapter.updateState({
-      expandedKeys: isExpandControlled ? expandedKeys : newExpandedKeys,
-      flattenNodes: isExpandControlled ? flattenNodes : newFlattenNodes,
+      expandedKeys: newExpandedKeys,
+      flattenNodes: newFlattenNodes,
       inputValue: '',
       motionKeys: new Set([]),
       filteredKeys: new Set([]),
@@ -489,11 +456,9 @@ class TreeSelectFoundation extends _foundation.default {
       filteredShownKeys: new Set([])
     });
   }
-
   handleInputChange(sugInput) {
     // Input is used as controlled component
     this._adapter.updateInputValue(sugInput);
-
     const {
       flattenNodes,
       expandedKeys,
@@ -504,63 +469,55 @@ class TreeSelectFoundation extends _foundation.default {
     const {
       showFilteredOnly,
       filterTreeNode,
-      treeNodeFilterProp
+      treeNodeFilterProp,
+      keyMaps
     } = this.getProps();
+    const realFilterProp = treeNodeFilterProp !== 'label' ? treeNodeFilterProp : (0, _get2.default)(keyMaps, 'label', 'label');
     const newExpandedKeys = new Set(expandedKeys);
     let filteredOptsKeys = [];
     let expandedOptsKeys = [];
     let newFlattenNodes = [];
     let filteredShownKeys = new Set([]);
-
     if (!sugInput) {
       expandedOptsKeys = (0, _treeUtil.findAncestorKeys)(selectedKeys, keyEntities);
       expandedOptsKeys.forEach(item => newExpandedKeys.add(item));
-      newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, newExpandedKeys);
+      newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, newExpandedKeys, keyMaps);
     } else {
       filteredOptsKeys = Object.values(keyEntities).filter(item => {
         const {
           data
         } = item;
-        return (0, _treeUtil.filter)(sugInput, data, filterTreeNode, treeNodeFilterProp);
+        return (0, _treeUtil.filter)(sugInput, data, filterTreeNode, realFilterProp);
       }).map(item => item.key);
       expandedOptsKeys = (0, _treeUtil.findAncestorKeys)(filteredOptsKeys, keyEntities, false);
       const shownChildKeys = (0, _treeUtil.findDescendantKeys)(filteredOptsKeys, keyEntities, true);
       filteredShownKeys = new Set([...shownChildKeys, ...expandedOptsKeys]);
-      newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, new Set(expandedOptsKeys), showFilteredOnly && filteredShownKeys);
+      newFlattenNodes = (0, _treeUtil.flattenTreeData)(treeData, new Set(expandedOptsKeys), keyMaps, showFilteredOnly && filteredShownKeys);
     }
-
     const newFilteredExpandedKeys = new Set(expandedOptsKeys);
-
     this._adapter.notifySearch(sugInput, Array.from(newFilteredExpandedKeys));
-
     this._adapter.updateState({
-      expandedKeys: this._isExpandControlled() ? expandedKeys : newExpandedKeys,
-      flattenNodes: this._isExpandControlled() ? flattenNodes : newFlattenNodes,
+      expandedKeys: newExpandedKeys,
+      flattenNodes: newFlattenNodes,
       motionKeys: new Set([]),
       filteredKeys: new Set(filteredOptsKeys),
       filteredExpandedKeys: newFilteredExpandedKeys,
       filteredShownKeys
     });
   }
-
   handleNodeSelect(e, treeNode) {
     const isDisabled = this._isDisabled(treeNode);
-
     if (isDisabled) {
       return;
     }
-
     if (!this._isMultiple()) {
       this.handleSingleSelect(e, treeNode);
     } else {
       this.handleMultipleSelect(e, treeNode);
     }
   }
-
   handleSingleSelect(e, treeNode) {
-    let {
-      selectedKeys
-    } = this.getCopyFromState('selectedKeys');
+    let selectedKeys = [...this.getState('selectedKeys')];
     const {
       clickToHide
     } = this.getProps();
@@ -569,42 +526,33 @@ class TreeSelectFoundation extends _foundation.default {
       eventKey,
       data
     } = treeNode;
-
     this._adapter.notifySelect(eventKey, true, data);
-
     if (!selectedKeys.includes(eventKey) && !selected) {
       selectedKeys = [eventKey];
-
       this._notifyChange(eventKey, e);
-
       if (!this._isControlledComponent()) {
         this._adapter.updateState({
           selectedKeys
         });
       }
     }
-
     if (clickToHide && (this._isSelectToClose() || !data.children)) {
       this.close(e);
+      this.handlerTriggerBlur(e);
     }
   }
-
   calcCheckedKeys(eventKey, targetStatus) {
     const {
       keyEntities
     } = this.getStates();
-    const {
-      checkedKeys,
-      halfCheckedKeys
-    } = this.getCopyFromState(['checkedKeys', 'halfCheckedKeys']);
-
+    const checkedKeys = new Set(this.getState('checkedKeys'));
+    const halfCheckedKeys = new Set(this.getState('halfCheckedKeys'));
     if (targetStatus) {
       return (0, _treeUtil.calcCheckedKeysForChecked)(eventKey, keyEntities, checkedKeys, halfCheckedKeys);
     } else {
       return (0, _treeUtil.calcCheckedKeysForUnchecked)(eventKey, keyEntities, checkedKeys, halfCheckedKeys);
     }
   }
-
   handleMultipleSelect(e, treeNode) {
     const {
       searchPosition,
@@ -620,30 +568,24 @@ class TreeSelectFoundation extends _foundation.default {
       eventKey,
       data
     } = treeNode;
-
     if (checkRelation === 'related') {
       const targetStatus = disableStrictly ? this.calcCheckedStatus(!checked, eventKey) : !checked;
       const {
         checkedKeys,
         halfCheckedKeys
       } = disableStrictly ? this.calcNonDisabledCheckedKeys(eventKey, targetStatus) : this.calcCheckedKeys(eventKey, targetStatus);
-
       this._adapter.notifySelect(eventKey, targetStatus, data);
-
       this._notifyChange([...checkedKeys], e);
-
       if (!this._isControlledComponent()) {
         this._adapter.updateState({
           checkedKeys,
           halfCheckedKeys
         });
-
         this._adapter.rePositionDropdown();
       }
     } else if (checkRelation === 'unRelated') {
       const newRealCheckedKeys = new Set(realCheckedKeys);
       let targetStatus;
-
       if (realCheckedKeys.has(eventKey)) {
         newRealCheckedKeys.delete(eventKey);
         targetStatus = false;
@@ -651,52 +593,40 @@ class TreeSelectFoundation extends _foundation.default {
         newRealCheckedKeys.add(eventKey);
         targetStatus = true;
       }
-
       this._adapter.notifySelect(eventKey, targetStatus, data);
-
       this._notifyChange([...newRealCheckedKeys], e);
-
       if (!this._isControlledComponent()) {
         this._adapter.updateState({
           realCheckedKeys: newRealCheckedKeys
         });
-
         this._adapter.rePositionDropdown();
       }
     }
-
     if (searchPosition === _constants.strings.SEARCH_POSITION_TRIGGER && inputValue !== '') {
       this._adapter.updateState({
         inputValue: ''
       });
     }
   }
-
   calcNonDisabledCheckedKeys(eventKey, targetStatus) {
     const {
       keyEntities,
       disabledKeys
     } = this.getStates();
-    const {
-      checkedKeys
-    } = this.getCopyFromState(['checkedKeys']);
+    const checkedKeys = new Set(this.getState('checkedKeys'));
     const descendantKeys = (0, _treeUtil.normalizeKeyList)((0, _treeUtil.findDescendantKeys)([eventKey], keyEntities, false), keyEntities, true);
     const hasDisabled = descendantKeys.some(key => disabledKeys.has(key));
-
     if (!hasDisabled) {
       return this.calcCheckedKeys(eventKey, targetStatus);
     }
-
     const nonDisabled = descendantKeys.filter(key => !disabledKeys.has(key));
-    const newCheckedKeys = targetStatus ? [...nonDisabled, ...checkedKeys] : (0, _difference2.default)((0, _treeUtil.normalizeKeyList)([...checkedKeys], keyEntities, true), nonDisabled);
+    const newCheckedKeys = targetStatus ? [...nonDisabled, ...checkedKeys] : (0, _difference2.default)((0, _treeUtil.normalizeKeyList)([...checkedKeys], keyEntities, true, true), nonDisabled);
     return (0, _treeUtil.calcCheckedKeys)(newCheckedKeys, keyEntities);
   }
-
   calcCheckedStatus(targetStatus, eventKey) {
     if (!targetStatus) {
       return targetStatus;
     }
-
     const {
       checkedKeys,
       keyEntities,
@@ -704,49 +634,41 @@ class TreeSelectFoundation extends _foundation.default {
     } = this.getStates();
     const descendantKeys = (0, _treeUtil.normalizeKeyList)((0, _treeUtil.findDescendantKeys)([eventKey], keyEntities, false), keyEntities, true);
     const hasDisabled = descendantKeys.some(key => disabledKeys.has(key));
-
     if (!hasDisabled) {
       return targetStatus;
     }
-
     const nonDisabledKeys = descendantKeys.filter(key => !disabledKeys.has(key));
     const allChecked = nonDisabledKeys.every(key => checkedKeys.has(key));
     return !allChecked;
   }
-
   handleNodeExpandInSearch(e, treeNode) {
     const {
       treeData,
       filteredShownKeys,
-      keyEntities
+      keyEntities,
+      keyMaps
     } = this.getStates();
-
-    const showFilteredOnly = this._showFilteredOnly(); // clone otherwise will be modified unexpectedly
-
-
-    const {
-      filteredExpandedKeys
-    } = this.getCopyFromState('filteredExpandedKeys');
+    const showFilteredOnly = this._showFilteredOnly();
+    // clone otherwise will be modified unexpectedly
+    const filteredExpandedKeys = new Set(this.getState('filteredExpandedKeys'));
     let motionType = 'show';
     const {
       eventKey,
       expanded,
       data
-    } = treeNode; // debugger;
-
+    } = treeNode;
+    // debugger;
     if (!expanded) {
       filteredExpandedKeys.add(eventKey);
     } else if (filteredExpandedKeys.has(eventKey)) {
       filteredExpandedKeys.delete(eventKey);
       motionType = 'hide';
-    } // cache prev flattenNodes, otherwise the calculation will remove hidden items
-
-
+    }
+    // cache prev flattenNodes, otherwise the calculation will remove hidden items
     this._adapter.cacheFlattenNodes(motionType === 'hide' && this._isAnimated());
-
     if (!this._isExpandControlled()) {
       // debugger;
-      const flattenNodes = (0, _treeUtil.flattenTreeData)(treeData, filteredExpandedKeys, showFilteredOnly && filteredShownKeys);
+      const flattenNodes = (0, _treeUtil.flattenTreeData)(treeData, filteredExpandedKeys, keyMaps, showFilteredOnly && filteredShownKeys);
       const motionKeys = this._isAnimated() ? (0, _treeUtil.getMotionKeys)(eventKey, filteredExpandedKeys, keyEntities) : [];
       const newState = {
         filteredExpandedKeys,
@@ -754,64 +676,53 @@ class TreeSelectFoundation extends _foundation.default {
         motionKeys: new Set(motionKeys),
         motionType
       };
-
       this._adapter.updateState(newState);
     }
-
     this._adapter.notifyExpand(filteredExpandedKeys, {
       expanded: !expanded,
       node: data
     });
   }
-
   handleNodeExpand(e, treeNode) {
     // debugger;
     const {
-      loadData
+      loadData,
+      keyMaps
     } = this.getProps();
     const {
       inputValue,
       keyEntities
     } = this.getStates();
     const isSearching = Boolean(inputValue);
-
     if (!loadData && (!treeNode.children || !treeNode.children.length)) {
       return;
     }
-
     const isExpandControlled = this._isExpandControlled();
-
-    if (isSearching && !isExpandControlled) {
+    if (isSearching) {
       this.handleNodeExpandInSearch(e, treeNode);
       return;
     }
-
     const {
       treeData
-    } = this.getStates(); // clone otherwise will be modified unexpectedly
-
-    const {
-      expandedKeys
-    } = this.getCopyFromState('expandedKeys');
+    } = this.getStates();
+    // clone otherwise will be modified unexpectedly
+    const expandedKeys = new Set(this.getState('expandedKeys'));
     let motionType = 'show';
     const {
       eventKey,
       expanded,
       data
     } = treeNode;
-
     if (!expanded) {
       expandedKeys.add(eventKey);
     } else if (expandedKeys.has(eventKey)) {
       expandedKeys.delete(eventKey);
       motionType = 'hide';
     }
-
     this._adapter.cacheFlattenNodes(motionType === 'hide' && this._isAnimated());
-
     if (!isExpandControlled) {
       // debugger;
-      const flattenNodes = (0, _treeUtil.flattenTreeData)(treeData, expandedKeys);
+      const flattenNodes = (0, _treeUtil.flattenTreeData)(treeData, expandedKeys, keyMaps);
       const motionKeys = this._isAnimated() ? (0, _treeUtil.getMotionKeys)(eventKey, expandedKeys, keyEntities) : [];
       const newState = {
         expandedKeys,
@@ -819,10 +730,8 @@ class TreeSelectFoundation extends _foundation.default {
         motionKeys: new Set(motionKeys),
         motionType
       };
-
       this._adapter.updateState(newState);
     }
-
     this._adapter.notifyExpand(expandedKeys, {
       expanded: !expanded,
       node: data
@@ -831,27 +740,29 @@ class TreeSelectFoundation extends _foundation.default {
   /**
    * The selected items that need to be displayed in the search box when obtaining a single selection
    */
-
-
   getRenderTextInSingle() {
     const {
       renderSelectedItem: propRenderSelectedItem,
-      treeNodeLabelProp
+      treeNodeLabelProp,
+      keyMaps
     } = this.getProps();
     const {
       selectedKeys,
       keyEntities
     } = this.getStates();
-    const renderSelectedItem = (0, _isFunction2.default)(propRenderSelectedItem) ? propRenderSelectedItem : item => (0, _get2.default)(item, treeNodeLabelProp, null);
-    const item = selectedKeys.length && keyEntities[selectedKeys[0]] ? keyEntities[selectedKeys[0]].data : undefined;
-    const renderText = item && treeNodeLabelProp in item ? renderSelectedItem(item) : null;
+    const realLabelName = (0, _get2.default)(keyMaps, 'label', treeNodeLabelProp);
+    const renderSelectedItem = (0, _isFunction2.default)(propRenderSelectedItem) ? propRenderSelectedItem : item => (0, _get2.default)(item, realLabelName, null);
+    let item;
+    if (selectedKeys.length) {
+      const key = selectedKeys[0];
+      item = keyEntities[key] && keyEntities[key].key === key ? keyEntities[key].data : this.getDataForKeyNotInKeyEntities(key);
+    }
+    const renderText = item ? renderSelectedItem(item) : null;
     return renderText;
   }
   /**
    * When the search box is on the trigger, the blur event handling method
    */
-
-
   handleInputTriggerBlur() {
     this._adapter.updateState({
       inputTriggerFocus: false
@@ -860,20 +771,14 @@ class TreeSelectFoundation extends _foundation.default {
   /**
    * When the search box is on the trigger, the focus event processing method
    */
-
-
   handleInputTriggerFocus() {
     this.clearInput();
-
     this._adapter.updateState({
       inputTriggerFocus: true
     });
   }
-
   setLoadKeys(data, resolve) {
     this._adapter.updateLoadKeys(data, resolve);
   }
-
 }
-
 exports.default = TreeSelectFoundation;