feat: add search command

Author: howenyap

Cargo.lock

@@ -23,5 +23,6 @@ directories = "6.0.0"
 jsonc-parser = { version = "0.32.1", features = ["cst", "serde", "serde_json"] }
 serde = { version = "1.0.228", features = ["derive"] }
 similar = "2.7.0"
+strsim = "0.11.1"
 thiserror = "2.0.18"
 toml = "1.0.7"

Cargo.toml

@@ -27,6 +27,8 @@ Please create an issue or PR if you'd like to add support for other targets.
 # Usage
 - `hue current` prints out the current set theme
 - `hue list` prints out available themes
+- `hue search <query>` prints the closest matching themes
+- `hue search <query> --limit <n>` limits the number of results
 - `hue reset` removes `~/.config/hue`, it will be lazily reinitiliased by other commands
 - `hue set <theme>` sets a logical theme and synchronises it across targets
 - `hue set <theme> --dry-run` shows a diff of the update without applying any changes

README.md

@@ -17,6 +17,8 @@ pub enum Command {
     Config,
     #[command(about = "List available themes")]
     List,
+    #[command(about = "Search available themes")]
+    Search(SearchArgs),
     #[command(about = "Show current theme")]
     Current,
     #[command(about = "Resets the config for Hue")]
@@ -31,3 +33,14 @@ pub struct SetArgs {
     #[arg(long, help = "Preview changes")]
     pub dry_run: bool,
 }
+
+#[derive(Args)]
+pub struct SearchArgs {
+    pub query: String,
+    #[arg(
+        long,
+        default_value_t = 10,
+        help = "Maximum number of matches to print"
+    )]
+    pub limit: usize,
+}

src/cli.rs

@@ -1,4 +1,5 @@
 use clap::Parser;
+use strsim::jaro_winkler;
 
 pub mod catalog;
 pub mod cli;
@@ -8,7 +9,7 @@ pub mod target;
 pub mod ui;
 
 use catalog::load_catalog;
-use cli::{Cli, Command, SetArgs};
+use cli::{Cli, Command, SearchArgs, SetArgs};
 use config::{Paths, load_config, reset_config, save_config};
 use error::{Error, Result};
 use target::set_theme;
@@ -21,6 +22,7 @@ pub fn run() -> Result<()> {
     match cli.command {
         Command::Config => show_config_dir(&paths),
         Command::List => list_themes(&paths),
+        Command::Search(args) => search_themes(&paths, args),
         Command::Current => show_current_theme(&paths),
         Command::Reset => handle_reset(&paths),
         Command::Set(args) => handle_set_theme(&paths, args),
@@ -39,12 +41,72 @@ fn list_themes(paths: &Paths) -> Result<()> {
 
     println!(
         "{}",
-        render_theme_table(config.current_theme.as_deref(), &catalog)
+        render_theme_table(
+            config.current_theme.as_deref(),
+            catalog
+                .iter()
+                .map(|(name, mapping)| (name.as_str(), mapping)),
+        )
     );
 
     Ok(())
 }
 
+struct ThemeSearchMatch<'a> {
+    theme_name: String,
+    mapping: &'a catalog::ThemeMapping,
+    score: f64,
+}
+
+fn search_themes(paths: &Paths, args: SearchArgs) -> Result<()> {
+    let SearchArgs { query, limit } = args;
+
+    let config = load_config(paths)?;
+    let catalog = load_catalog(paths)?;
+    let query = query.trim().to_lowercase();
+
+    let matches = {
+        let mut matches = catalog
+            .iter()
+            .filter_map(|(theme, mapping)| {
+                const MIN_SEARCH_SCORE: f64 = 0.70;
+                let score = jaro_winkler(&query, &theme.trim().to_lowercase());
+
+                (score >= MIN_SEARCH_SCORE).then(|| ThemeSearchMatch {
+                    theme_name: theme.clone(),
+                    score,
+                    mapping,
+                })
+            })
+            .collect::<Vec<_>>();
+
+        matches.sort_by(|left, right| {
+            right
+                .score
+                .total_cmp(&left.score)
+                .then_with(|| left.theme_name.cmp(&right.theme_name))
+        });
+
+        matches.truncate(limit.max(1));
+        matches
+    };
+
+    if matches.is_empty() {
+        println!("No matches found for `{query}`.");
+    } else {
+        let table = render_theme_table(
+            config.current_theme.as_deref(),
+            matches
+                .iter()
+                .map(|item| (item.theme_name.as_str(), item.mapping)),
+        );
+
+        println!("{table}",);
+    }
+
+    Ok(())
+}
+
 fn show_current_theme(paths: &Paths) -> Result<()> {
     let config = load_config(paths)?;
     let current = config

src/lib.rs

@@ -1,4 +1,3 @@
-use std::collections::BTreeMap;
 use std::iter::once;
 use std::path::Path;
 
@@ -9,10 +8,10 @@ use similar::TextDiff;
 use crate::catalog::ThemeMapping;
 use crate::target::Target;
 
-pub fn render_theme_table(
-    current_theme: Option<&str>,
-    catalog: &BTreeMap<String, ThemeMapping>,
-) -> String {
+pub fn render_theme_table<'a, I>(current_theme: Option<&str>, themes: I) -> String
+where
+    I: IntoIterator<Item = (&'a str, &'a ThemeMapping)>,
+{
     let mut table = DisplayTable::new();
 
     table
@@ -24,11 +23,11 @@ pub fn render_theme_table(
                 .collect::<Vec<_>>(),
         );
 
-    for (name, mapping) in catalog {
-        let theme_name = if current_theme == Some(name.as_str()) {
+    for (name, mapping) in themes {
+        let theme_name = if current_theme == Some(name) {
             format!("{name} (current)")
         } else {
-            name.clone()
+            name.to_string()
         };
 
         table.add_row(