@douyinfe/semi-ui 2.55.5 → 2.56.0-alpha.0

package/dist/umd/semi-ui.js

@@ -90389,6 +90389,7 @@ class TreeSelectFoundation extends foundation {
     } = this.getProps();
     const realFilterProp = treeNodeFilterProp !== 'label' ? treeNodeFilterProp : get_default()(keyMaps, 'label', 'label');
     const newExpandedKeys = new Set(expandedKeys);
+    let filteredNodes = [];
     let filteredOptsKeys = [];
     let expandedOptsKeys = [];
     let newFlattenNodes = [];
@@ -90398,19 +90399,21 @@ class TreeSelectFoundation extends foundation {
       expandedOptsKeys.forEach(item => newExpandedKeys.add(item));
       newFlattenNodes = flattenTreeData(treeData, newExpandedKeys, keyMaps);
     } else {
-      filteredOptsKeys = Object.values(keyEntities).filter(item => {
+      const filteredOpts = Object.values(keyEntities).filter(item => {
         const {
           data
         } = item;
         return filter(sugInput, data, filterTreeNode, realFilterProp);
-      }).map(item => item.key);
+      });
+      filteredNodes = filteredOpts.map(item => item.data);
+      filteredOptsKeys = filteredOpts.map(item => item.key);
       expandedOptsKeys = findAncestorKeys(filteredOptsKeys, keyEntities, false);
       const shownChildKeys = findDescendantKeys(filteredOptsKeys, keyEntities, true);
       filteredShownKeys = new Set([...shownChildKeys, ...expandedOptsKeys]);
       newFlattenNodes = flattenTreeData(treeData, new Set(expandedOptsKeys), keyMaps, showFilteredOnly && filteredShownKeys);
     }
     const newFilteredExpandedKeys = new Set(expandedOptsKeys);
-    this._adapter.notifySearch(sugInput, Array.from(newFilteredExpandedKeys));
+    this._adapter.notifySearch(sugInput, Array.from(newFilteredExpandedKeys), filteredNodes);
     this._adapter.updateState({
       expandedKeys: newExpandedKeys,
       flattenNodes: newFlattenNodes,
@@ -91834,8 +91837,8 @@ class TreeSelect extends BaseComponent {
       notifySelect: (selectKey, bool, node) => {
         this.props.onSelect && this.props.onSelect(selectKey, bool, node);
       },
-      notifySearch: (input, filteredExpandedKeys) => {
-        this.props.onSearch && this.props.onSearch(input, filteredExpandedKeys);
+      notifySearch: (input, filteredExpandedKeys, filteredNodes) => {
+        this.props.onSearch && this.props.onSearch(input, filteredExpandedKeys, filteredNodes);
       },
       cacheFlattenNodes: bool => {
         this.setState({