fix: Keep search tree intact upon selection

Closes #190

Author: mrchief

docs/bundle.js

@@ -129,7 +129,8 @@ class DropdownTreeSelect extends Component {
 
   onNodeToggle = id => {
     this.treeManager.toggleNodeExpandState(id)
-    this.setState({ tree: this.treeManager.tree })
+    const tree = this.state.searchModeOn ? this.treeManager.matchTree : this.treeManager.tree
+    this.setState({ tree })
     typeof this.props.onNodeToggle === 'function' && this.props.onNodeToggle(this.treeManager.getNodeById(id))
   }
 
@@ -143,8 +144,9 @@ class DropdownTreeSelect extends Component {
       tags = this.treeManager.getTags()
     }
 
+    const tree = this.state.searchModeOn ? this.treeManager.matchTree : this.treeManager.tree
     const nextState = {
-      tree: this.treeManager.tree,
+      tree,
       tags,
       showDropdown
     }

src/index.js

@@ -80,6 +80,11 @@ class TreeManager {
     })
 
     const allNodesHidden = matches.length === 0
+
+    // we store a local reference so that components can use it in subsequent renders
+    // this is the least intrusive way of fixing #190
+    this.matchTree = matchTree
+
     return { allNodesHidden, tree: matchTree }
   }