chore[tui]: fuzzy search column names (#4729)

Signed-off-by: Joe Isaacs <[email protected]>

Author: joseph-isaacs

Cargo.lock

@@ -106,6 +106,7 @@ fastlanes = "0.2.2"
 flatbuffers = "25.2.10"
 fsst-rs = "0.5.2"
 futures = { version = "0.3.31", default-features = false }
+fuzzy-matcher = "0.3"
 glob = "0.3.2"
 goldenfile = "1"
 half = { version = "2.6", features = ["std", "num-traits"] }

Cargo.toml

@@ -20,6 +20,7 @@ crossterm = { workspace = true }
 env_logger = { version = "0.11" }
 flatbuffers = { workspace = true }
 futures = { workspace = true, features = ["executor"] }
+fuzzy-matcher = { workspace = true }
 humansize = { workspace = true }
 indicatif = { workspace = true, features = ["futures"] }
 itertools = { workspace = true }

vortex-tui/Cargo.toml

@@ -1,7 +1,10 @@
 // SPDX-License-Identifier: Apache-2.0
 // SPDX-FileCopyrightText: Copyright the Vortex contributors
 
+use fuzzy_matcher::FuzzyMatcher;
+use fuzzy_matcher::skim::SkimMatcherV2;
 use humansize::{DECIMAL, make_format};
+use itertools::Itertools;
 use ratatui::buffer::Buffer;
 use ratatui::layout::{Constraint, Direction, Layout, Rect};
 use ratatui::style::{Color, Style, Stylize};
@@ -218,44 +221,71 @@ fn render_children_list(app: &mut AppState, area: Rect, buf: &mut Buffer) {
     let layout = app.cursor.layout();
 
     if layout.nchildren() > 0 {
-        let filter: Vec<bool> = layout
-            .child_names()
-            .map(|name| {
-                if search_filter.is_empty() {
-                    true
-                } else {
-                    name.contains(&search_filter)
-                }
-            })
-            .collect();
-
-        let list_items: Vec<String> = layout
-            .child_names()
-            .zip(filter.iter())
-            .filter_map(|(name, keep)| keep.then_some(name.to_string()))
-            .collect();
+        if search_filter.is_empty() {
+            // No search filter, show all items
+            let list_items = layout
+                .child_names()
+                .map(|name| name.to_string())
+                .collect_vec();
+
+            app.filter = None;
+            render_child_list_items(app, area, buf, list_items);
+        } else {
+            // Use fuzzy matching to rank and filter results
+            let matcher = SkimMatcherV2::default();
+
+            // Collect scored matches
+            let mut scored_matches = layout
+                .child_names()
+                .enumerate()
+                .filter_map(|(idx, name)| {
+                    matcher
+                        .fuzzy_match(&name, &search_filter)
+                        .map(|score| (idx, name.to_string(), score))
+                })
+                .collect_vec();
+
+            // Sort by score (higher is better)
+            scored_matches.sort_by(|a, b| b.2.cmp(&a.2));
+
+            // Create filter based on fuzzy matches
+            let mut filter = vec![false; layout.nchildren()];
+            let list_items = scored_matches
+                .iter()
+                .map(|(idx, name, _score)| {
+                    filter[*idx] = true;
+                    name.clone()
+                })
+                .collect_vec();
 
-        if !app.search_filter.is_empty() {
             app.filter = Some(filter);
+            render_child_list_items(app, area, buf, list_items);
         }
+    }
+}
 
-        let container = Block::new()
-            .title("Child Layouts")
-            .borders(Borders::ALL)
-            .border_type(BorderType::Rounded)
-            .border_style(Style::default().fg(Color::DarkGray));
+fn render_child_list_items(
+    app: &mut AppState,
+    area: Rect,
+    buf: &mut Buffer,
+    list_items: Vec<String>,
+) {
+    let container = Block::new()
+        .title("Child Layouts")
+        .borders(Borders::ALL)
+        .border_type(BorderType::Rounded)
+        .border_style(Style::default().fg(Color::DarkGray));
 
-        let inner_area = container.inner(area);
+    let inner_area = container.inner(area);
 
-        container.render(area, buf);
+    container.render(area, buf);
 
-        // Render the List view.
-        // TODO: add state so we can scroll
-        StatefulWidget::render(
-            List::new(list_items).highlight_style(Style::default().black().on_white().bold()),
-            inner_area,
-            buf,
-            &mut app.layouts_list_state,
-        );
-    }
+    // Render the List view.
+    // TODO: add state so we can scroll
+    StatefulWidget::render(
+        List::new(list_items).highlight_style(Style::default().black().on_white().bold()),
+        inner_area,
+        buf,
+        &mut app.layouts_list_state,
+    );
 }