From 930a0ea73b74921d687f3389f8dfb99f8fda8cea Mon Sep 17 00:00:00 2001
From: mo8it <mo8it@proton.me>
Date: Thu, 17 Oct 2024 16:00:10 +0200
Subject: [PATCH] list: Highlight search match in exercise names

---
 src/list/state.rs | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/list/state.rs b/src/list/state.rs
index 5bdbca77..53fe07c4 100644
--- a/src/list/state.rs
+++ b/src/list/state.rs
@@ -105,6 +105,28 @@ impl<'a> ListState<'a> {
         );
     }
 
+    fn draw_exericse_name(&self, writer: &mut MaxLenWriter, exercise: &Exercise) -> io::Result<()> {
+        if !self.search_query.is_empty() {
+            if let Some((pre_highlight, highlight, post_highlight)) = exercise
+                .name
+                .find(&self.search_query)
+                .and_then(|ind| exercise.name.split_at_checked(ind))
+                .and_then(|(pre_highlight, rest)| {
+                    rest.split_at_checked(self.search_query.len())
+                        .map(|x| (pre_highlight, x.0, x.1))
+                })
+            {
+                writer.write_str(pre_highlight)?;
+                writer.stdout.queue(SetForegroundColor(Color::Magenta))?;
+                writer.write_str(highlight)?;
+                writer.stdout.queue(ResetColor)?;
+                return writer.write_str(post_highlight);
+            }
+        }
+
+        writer.write_str(exercise.name)
+    }
+
     fn draw_rows(
         &self,
         stdout: &mut StdoutLock,
@@ -147,10 +169,10 @@ impl<'a> ListState<'a> {
                 writer.stdout.queue(SetForegroundColor(Color::Yellow))?;
                 writer.write_ascii(b"PENDING  ")?;
             }
-
             writer.stdout.queue(SetForegroundColor(Color::Reset))?;
 
-            writer.write_str(exercise.name)?;
+            self.draw_exericse_name(&mut writer, exercise)?;
+
             writer.write_ascii(&self.name_col_padding[exercise.name.len()..])?;
 
             // The list links aren't shown correctly in VS Code on Windows.