tree select root

2 files changed, 16 insertions, 2 deletions

diff --git a/src/tab.rs b/src/tab.rs
index 72e5ef2..ff111ad 100644
--- a/src/tab.rs
+++ b/src/tab.rs
@@ -533,8 +533,10 @@ impl Tab {
     /// Fold every child node in the tree.
     /// Recursively explore the tree and fold every node. Reset the display.
     pub fn tree_go_to_root(&mut self) -> Result<()> {
-        self.directory.tree.reset_required_height();
-        self.tree_select_root()
+        // self.directory.tree.reset_required_height();
+        // self.tree_select_root()
+        self.tree.select_root();
+        Ok(())
     }
 
     /// Select the first child of the current node and reset the display.
diff --git a/src/trees.rs b/src/trees.rs
index ddf54e7..e19cec1 100644
--- a/src/trees.rs
+++ b/src/trees.rs
@@ -285,6 +285,18 @@ impl FileSystem {
         }
     }
 
+    pub fn select_root(&mut self) {
+        let Some(selected_node) = self.nodes.get_mut(&self.selected) else {
+            unreachable!("selected path should be in node")
+        };
+        selected_node.unselect();
+        let Some(root_node) = self.nodes.get_mut(&self.root_path) else {
+            unreachable!("root path should be in nodes")
+        };
+        root_node.select();
+        self.selected = self.root_path.to_owned();
+    }
+
     /// Fold selected node
     pub fn toggle_fold(&mut self) {
         if let Some(node) = self.nodes.get_mut(&self.selected) {