Refactor file and tree rendering

Author: igrek51

src/app.rs

@@ -1,7 +1,6 @@
 use anyhow::Result;
 use ratatui::widgets::ListState;
 use signal_hook::{consts::SIGINT, consts::SIGTERM, iterator::Signals};
-use std::fs::File;
 use std::sync::mpsc;
 use std::thread;
 

src/filesystem.rs

@@ -12,19 +12,6 @@ pub struct FileNode {
     pub is_directory: bool,
 }
 
-impl FileNode {
-    pub fn display_name(&self) -> String {
-        let mut display = self.name.clone();
-        if self.is_symlink {
-            display = format!("{}@", display)
-        }
-        if self.file_type == FileType::Directory {
-            display = format!("{}/", display)
-        }
-        display
-    }
-}
-
 #[derive(Debug, PartialEq, Eq, Clone, Copy)]
 pub enum FileType {
     Regular,   // regular file or symlink to regular file

src/tree.rs

@@ -1,13 +1,36 @@
+use ratatui::{
+    style::{Style, Stylize},
+    text::{Line, Span},
+    widgets::ListItem,
+};
+
 use crate::filesystem::FileNode;
 
 #[derive(Debug, Clone)]
 pub struct TreeNode {
-    pub display_name: String,
     pub indent: i32,
     pub relevance: i32,
     pub file_node: FileNode,
 }
 
+impl TreeNode {
+    pub fn render_list_item(&self) -> ListItem {
+        let display = self.file_node.name.clone();
+        let mut suffix = String::new();
+        let mut style = Style::default();
+        if self.file_node.is_symlink {
+            suffix = format!("{suffix}@");
+            style = Style::default().fg(ratatui::style::Color::LightCyan).bold();
+        }
+        if self.file_node.is_directory {
+            suffix = format!("{suffix}/");
+            style = Style::default().fg(ratatui::style::Color::LightBlue).bold();
+        }
+
+        Line::from(vec![Span::styled(display, style), Span::raw(suffix)]).into()
+    }
+}
+
 pub fn evaluate_relevance(text: &str, words: &Vec<String>) -> i32 {
     if words.is_empty() {
         return 0;
@@ -35,7 +58,6 @@ pub fn render_tree_nodes(child_nodes: &Vec<FileNode>, filter_text: &str) -> Vec<
     let mut current_tree_nodes: Vec<TreeNode> = child_nodes
         .iter()
         .map(|it: &FileNode| TreeNode {
-            display_name: it.display_name(),
             indent: 0,
             relevance: evaluate_relevance(it.lowercase_name.as_str(), &filter_words),
             file_node: it.clone(),

src/ui.rs

@@ -49,7 +49,8 @@ fn render_dir_tree(app: &mut App, frame: &mut Frame, area: Rect) {
     let list_items: Vec<ListItem> = app
         .child_tree_nodes
         .iter()
-        .map(|it: &TreeNode| ListItem::new(it.display_name.clone()))
+        .map(|it: &TreeNode| it.render_list_item())
+        // .map(|it: &TreeNode| ListItem::new(it.display_name()))
         .collect();
 
     let title = app.get_current_string_path();