Tree mode: file & directory are created in selected dir

1 files changed, 13 insertions, 1 deletions

diff --git a/src/tree.rs b/src/tree.rs
index cc7605d..57609b6 100644
--- a/src/tree.rs
+++ b/src/tree.rs
@@ -1,6 +1,6 @@
 use std::path::Path;
 
-use anyhow::Result;
+use anyhow::{Context, Result};
 use tuikit::attr::Attr;
 use users::UsersCache;
 
@@ -574,6 +574,18 @@ impl Tree {
 
         visited.node.position.clone()
     }
+
+    pub fn directory_of_selected(&self) -> Result<&std::path::Path> {
+        let fileinfo = &self.current_node.fileinfo;
+
+        match fileinfo.file_kind {
+            FileKind::Directory => Ok(&self.current_node.fileinfo.path),
+            _ => Ok(fileinfo
+                .path
+                .parent()
+                .context("selected file should have a parent")?),
+        }
+    }
 }
 
 fn first_prefix(mut prefix: String) -> String {