Merge 37641fb into 92698cb

dowjones · Jun 18, 2020 · fc8f0a4 · fc8f0a4
2 parents 92698cb + 37641fb
commit fc8f0a4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -151,7 +151,7 @@ class DropdownTreeSelect extends Component {
     const searchModeOn = value.length > 0
 
     this.setState({
-      tree,
+      tree: searchModeOn ? tree : this.treeManager.restoreNodes(),
       searchModeOn,
       allNodesHidden,
     })

diff --git a/src/tree-manager/index.js b/src/tree-manager/index.js
@@ -21,6 +21,9 @@ class TreeManager {
     })
     this.tree = list
     this.defaultValues = defaultValues
+    this.defaultHidden = Array.from(list)
+      .filter(item => item[1].hide)
+      .map(item => item[0])
     this.showPartialState = !this.hierarchical && showPartiallySelected
     this.searchMaps = new Map()
 
@@ -122,7 +125,7 @@ class TreeManager {
 
   restoreNodes() {
     this.tree.forEach(node => {
-      node.hide = false
+      node.hide = this.defaultHidden.includes(node._id) ? true : false
     })
 
     return this.tree