From adc72f2d875c15cd9a4f895f36c9fd7599eb0c00 Mon Sep 17 00:00:00 2001
From: Denys Rybalka <denys.rybalka@tuta.io>
Date: Thu, 3 Oct 2024 10:20:43 +0200
Subject: [PATCH] Distinguish dirs visually in file_browser

---
 helix-term/src/ui/mod.rs    | 16 ++++++++--------
 helix-term/src/ui/picker.rs | 10 ++++++++--
 2 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/helix-term/src/ui/mod.rs b/helix-term/src/ui/mod.rs
index 62c9fb912..2c105aba5 100644
--- a/helix-term/src/ui/mod.rs
+++ b/helix-term/src/ui/mod.rs
@@ -273,10 +273,12 @@ pub fn file_browser(root: PathBuf) -> Result<FilePicker, std::io::Error> {
     let columns = [PickerColumn::new(
         "path",
         |item: &PathBuf, root: &PathBuf| {
-            item.strip_prefix(root)
-                .unwrap_or(item)
-                .to_string_lossy()
-                .into()
+            let name = item.strip_prefix(root).unwrap_or(item).to_string_lossy();
+            if item.is_dir() {
+                format!("{}/", name).into()
+            } else {
+                name.into()
+            }
         },
     )];
     let picker = Picker::new(
@@ -316,11 +318,10 @@ fn directory_content(path: &Path) -> Result<Vec<PathBuf>, std::io::Error> {
     let mut dirs = Vec::new();
     let mut files = Vec::new();
     for entry in std::fs::read_dir(path)?.flatten() {
-        let entry_path = entry.path();
         if entry.path().is_dir() {
-            dirs.push(entry_path);
+            dirs.push(entry.path());
         } else {
-            files.push(entry_path);
+            files.push(entry.path());
         }
     }
     dirs.sort();
@@ -328,7 +329,6 @@ fn directory_content(path: &Path) -> Result<Vec<PathBuf>, std::io::Error> {
 
     let mut content = Vec::new();
     if path.parent().is_some() {
-        log::warn!("{}", path.to_string_lossy());
         content.insert(0, path.join(".."));
     }
     content.extend(dirs);
diff --git a/helix-term/src/ui/picker.rs b/helix-term/src/ui/picker.rs
index 766bbf357..d6ee760f5 100644
--- a/helix-term/src/ui/picker.rs
+++ b/helix-term/src/ui/picker.rs
@@ -614,8 +614,14 @@ impl<T: 'static + Send + Sync, D: 'static + Send + Sync> Picker<T, D> {
                             let files = super::directory_content(&path)?;
                             let file_names: Vec<_> = files
                                 .iter()
-                                .filter_map(|file| file.file_name())
-                                .map(|name| name.to_string_lossy().into_owned())
+                                .filter_map(|file| {
+                                    let name = file.file_name()?.to_string_lossy();
+                                    if file.is_dir() {
+                                        Some(format!("{}/", name))
+                                    } else {
+                                        Some(name.into_owned())
+                                    }
+                                })
                                 .collect();
                             Ok(CachedPreview::Directory(file_names))
                         } else if metadata.is_file() {