From 0ca4f09dbc7981077d90dd7131a77972113ad193 Mon Sep 17 00:00:00 2001
From: afonsofrancof <afonso@francof.net>
Date: Tue, 30 Jul 2024 21:52:12 +0100
Subject: [PATCH] Merged with #82, still need testing

---
 src/list.rs | 62 ++++++++++++++++++++++++++++++++---------------------
 src/main.rs | 24 +++++++++++++++++++--
 2 files changed, 59 insertions(+), 27 deletions(-)

diff --git a/src/list.rs b/src/list.rs
index 60564d83..3d7453d2 100644
--- a/src/list.rs
+++ b/src/list.rs
@@ -9,6 +9,7 @@ use ratatui::{
     Frame,
 };
 
+#[derive(Clone)]
 struct ListNode {
     name: &'static str,
     command: Command,
@@ -161,7 +162,8 @@ impl CustomList {
             self.filtered_items
                 .iter()
                 .map(|node| {
-                    Line::from(format!("{}  {}", state.theme.cmd_icon, node.value().name)).style(state.theme.cmd_color)
+                    Line::from(format!("{}  {}", state.theme.cmd_icon, node.name))
+                        .style(state.theme.cmd_color)
                 })
                 .collect()
         };
@@ -313,6 +315,7 @@ impl CustomList {
         } else {
             self.filtered_items.len()
         };
+
         let curr_selection = self.list_state.selected().unwrap();
         if self.at_root() {
             self.list_state
@@ -378,35 +381,44 @@ impl CustomList {
     ///
     /// Returns `Some(command)` when command is selected, othervise we returns `None`
     fn handle_enter(&mut self) -> Option<Command> {
-        // Get the current node (current directory)
-        let curr = self
-            .inner_tree
-            .get(*self.visit_stack.last().unwrap())
-            .unwrap();
+        // Get the selected index
         let selected = self.list_state.selected().unwrap();
 
-        // if we are not at the root, and the first element is selected,
-        // we can be sure it's '..', so we go up the directory
-        if !self.at_root() && selected == 0 {
-            self.visit_stack.pop();
-            self.list_state.select(Some(0));
-            return None;
-        }
+        if self.filter_query.is_empty() {
+            // No filter query, use the regular tree navigation
+            let curr = self
+                .inner_tree
+                .get(*self.visit_stack.last().unwrap())
+                .unwrap();
 
-        for (mut idx, node) in curr.children().enumerate() {
-            // at this point, we know that we are not on the .. item, and our indexes of the items never had ..
-            // item. so to balance it out, in case the selection index contains .., se add 1 to our node index
-            if !self.at_root() {
-                idx += 1;
+            // if we are not at the root, and the first element is selected,
+            // we can be sure it's '..', so we go up the directory
+            if !self.at_root() && selected == 0 {
+                self.visit_stack.pop();
+                self.list_state.select(Some(0));
+                return None;
             }
-            if idx == selected {
-                if node.has_children() {
-                    self.visit_stack.push(node.id());
-                    self.list_state.select(Some(0));
-                    return None;
-                } else {
-                    return Some(node.value().command.clone());
+
+            for (mut idx, node) in curr.children().enumerate() {
+                // at this point, we know that we are not on the .. item, and our indexes of the items never had ..
+                // item. so to balance it out, in case the selection index contains .., se add 1 to our node index
+                if !self.at_root() {
+                    idx += 1;
                 }
+                if idx == selected {
+                    if node.has_children() {
+                        self.visit_stack.push(node.id());
+                        self.list_state.select(Some(0));
+                        return None;
+                    } else {
+                        return Some(node.value().command.clone());
+                    }
+                }
+            }
+        } else {
+            // Filter query is active, use the filtered items
+            if let Some(filtered_node) = self.filtered_items.get(selected) {
+                return Some(filtered_node.command.clone());
             }
         }
         None
diff --git a/src/main.rs b/src/main.rs
index 350c4584..c1122acb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -117,7 +117,7 @@ fn run<B: Backend>(terminal: &mut Terminal<B>, state: &AppState) -> io::Result<(
                 //Render the search bar (First chunk of the screen)
                 frame.render_widget(search_bar, chunks[0]);
                 //Render the command list (Second chunk of the screen)
-                custom_list.draw(frame, chunks[1], search_input.clone(),state);
+                custom_list.draw(frame, chunks[1], search_input.clone(), state);
 
                 if let Some(ref mut command) = &mut command_opt {
                     command.draw(frame, state);
@@ -145,7 +145,27 @@ fn run<B: Backend>(terminal: &mut Terminal<B>, state: &AppState) -> io::Result<(
                 if key.code == KeyCode::Char('q') {
                     return Ok(());
                 }
-                if let Some(cmd) = custom_list.handle_key(key, state) {
+                //Activate search mode if the forward slash key gets pressed
+                if key.code == KeyCode::Char('/') {
+                    // Enter search mode
+                    in_search_mode = true;
+                    continue;
+                }
+                //Insert user input into the search bar
+                if in_search_mode {
+                    match key.code {
+                        KeyCode::Char(c) => search_input.push(c),
+                        KeyCode::Backspace => {
+                            search_input.pop();
+                        }
+                        KeyCode::Esc => {
+                            search_input = String::new();
+                            in_search_mode = false
+                        }
+                        KeyCode::Enter => in_search_mode = false,
+                        _ => {}
+                    }
+                } else if let Some(cmd) = custom_list.handle_key(key, state) {
                     command_opt = Some(RunningCommand::new(cmd, state));
                 }
             }