1 files changed, 40 insertions, 1 deletions

diff --git a/src/cli.rs b/src/cli.rs
index f5fdd2c..e4bb484 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -92,6 +92,13 @@ pub enum Commands {
         #[arg(value_name = "SHELL")]
         shell: Option<String>,
     },
+    #[clap(hide = true)]
+    /// Complete artists, albums, or tracks from database
+    DbComplete {
+        /// Query string (e.g. \"J\", \"Artist/\", \"Artist/Album/\")
+        #[arg(value_name = "QUERY")]
+        query: String,
+    },
 }
 
 #[derive(Subcommand)]
@@ -235,11 +242,43 @@ impl Cli {
                     }
                 }
             }
+            Commands::DbComplete { query } => {
+                // Unescape shell escapes in the input query
+                let query = {
+                    let mut unescaped = String::with_capacity(query.len());
+                    let mut chars = query.chars();
+                    while let Some(c) = chars.next() {
+                        if c == '\\' {
+                            if let Some(next) = chars.next() {
+                                unescaped.push(next);
+                            }
+                        } else {
+                            unescaped.push(c);
+                        }
+                    }
+                    unescaped
+                };
+
+                let config = match Config::load() {
+                    Ok(c) => c,
+                    Err(_) => return Ok(()),
+                };
+                let db = match Database::open(&config) {
+                    Ok(d) => d,
+                    Err(_) => return Ok(()),
+                };
+
+                let results = db.complete(&query).unwrap_or_default();
+
+                for r in &results {
+                    println!("{r}");
+                }
+            }
             Commands::Completions { shell } => {
                 let full = Cli::command();
                 let subcommands: Vec<Command> = full
                     .get_subcommands()
-                    .filter(|s| s.get_name() != "completions")
+                    .filter(|s| s.get_name() != "completions" && s.get_name() != "db-complete")
                     .cloned()
                     .collect();
                 let cmd_name = full.get_name().to_string();