FIX: filename in first line

1 files changed, 32 insertions, 28 deletions

diff --git a/src/tree.rs b/src/tree.rs
index e437bdd..08de5e7 100644
--- a/src/tree.rs
+++ b/src/tree.rs
@@ -155,26 +155,6 @@ impl Tree {
         }
     }
 
-    fn increment_required_height(&mut self) {
-        if self.required_height < usize::MAX {
-            self.required_height += 1
-        }
-    }
-
-    fn decrement_required_height(&mut self) {
-        if self.required_height > Self::DEFAULT_REQUIRED_HEIGHT {
-            self.required_height -= 1
-        }
-    }
-
-    fn set_required_height_to_max(&mut self) {
-        self.required_height = usize::MAX
-    }
-
-    fn reset_required_height(&mut self) {
-        self.required_height = Self::DEFAULT_REQUIRED_HEIGHT
-    }
-
     pub fn empty() -> Self {
         Self {
             root_path: PathBuf::default(),
@@ -193,6 +173,18 @@ impl Tree {
         self.nodes.get(&self.selected)
     }
 
+    pub fn directory_of_selected(&self) -> Option<&Path> {
+        if self.selected.is_dir() && !self.selected.is_symlink() {
+            Some(self.selected.as_path())
+        } else {
+            self.selected.parent()
+        }
+    }
+
+    pub fn selected_path_relative_to_root(&self) -> Result<&Path> {
+        Ok(self.selected.strip_prefix(&self.root_path)?)
+    }
+
     /// Select next sibling or the next sibling of the parent
     pub fn select_next(&mut self) -> Result<()> {
         log::info!("select_next START {sel}", sel = self.selected.display());
@@ -360,6 +352,26 @@ impl Tree {
         self.set_required_height_to_max()
     }
 
+    fn increment_required_height(&mut self) {
+        if self.required_height < usize::MAX {
+            self.required_height += 1
+        }
+    }
+
+    fn decrement_required_height(&mut self) {
+        if self.required_height > Self::DEFAULT_REQUIRED_HEIGHT {
+            self.required_height -= 1
+        }
+    }
+
+    fn set_required_height_to_max(&mut self) {
+        self.required_height = usize::MAX
+    }
+
+    fn reset_required_height(&mut self) {
+        self.required_height = Self::DEFAULT_REQUIRED_HEIGHT
+    }
+
     /// Fold selected node
     pub fn toggle_fold(&mut self) {
         if let Some(node) = self.nodes.get_mut(&self.selected) {
@@ -379,14 +391,6 @@ impl Tree {
         }
     }
 
-    pub fn directory_of_selected(&self) -> Option<&Path> {
-        if self.selected.is_dir() && !self.selected.is_symlink() {
-            Some(self.selected.as_path())
-        } else {
-            self.selected.parent()
-        }
-    }
-
     // FIX: can only find the first match and nothing else
     pub fn search_first_match(&mut self, pattern: &str) {
         let initial_selected = self.selected.to_owned();