Merge pull request 'autocompletion' (#1) from autocompletion into main

Reviewed-on: https://codeberg.org/salmonik/RSPlayer/pulls/1

1 files changed, 297 insertions, 8 deletions

diff --git a/src/database.rs b/src/database.rs
index 7fd1797..e6f6f6e 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -45,6 +45,10 @@ pub struct Database {
 
 #[allow(dead_code)]
 impl Database {
+    pub fn conn(&self) -> &Connection {
+        &self.conn
+    }
+
     pub fn open(_config: &crate::config::Config) -> SqlResult<Self> {
         let proj_dirs = ProjectDirs::from("com", "rsplayer", "rsplayer").ok_or_else(|| {
             rusqlite::Error::InvalidPath("Failed to get project directories".into())
@@ -59,7 +63,29 @@ impl Database {
         fs::create_dir_all(&data_dir).ok();
         let db_path = data_dir.join("library.db");
 
-        let conn = Connection::open(&db_path)?;
+        let debug_log = std::env::temp_dir().join("rsp-completion-debug.log");
+        macro_rules! debug_log {
+            ($($arg:tt)*) => {
+                let _ = std::fs::OpenOptions::new()
+                    .create(true)
+                    .append(true)
+                    .open(&debug_log)
+                    .and_then(|mut f| {
+                        use std::io::Write;
+                        writeln!(f, $($arg)*)
+                    });
+            };
+        }
+        debug_log!("[Database::open] db_path: {:?}", db_path);
+        debug_log!("[Database::open] db exists: {}", db_path.exists());
+
+        let conn = match Connection::open(&db_path) {
+            Ok(c) => c,
+            Err(e) => {
+                debug_log!("[Database::open] Connection::open failed: {e}");
+                return Err(e);
+            }
+        };
         let db = Database { conn };
         db.init_schema()?;
         Ok(db)
@@ -279,6 +305,261 @@ impl Database {
         self.get_album_tracks_after(album_id, track_number, &path)
     }
 
+    pub fn complete(&self, query: &str) -> SqlResult<Vec<String>> {
+        let mut results: Vec<String> = Vec::new();
+
+        if query.is_empty() {
+            let mut stmt = self
+                .conn
+                .prepare("SELECT name FROM artists ORDER BY name LIMIT 25")?;
+            let rows = stmt.query_map([], |row| row.get::<_, String>(0))?;
+            for name in rows.flatten() {
+                results.push(format!("{name}/"));
+            }
+            return Ok(results);
+        }
+
+        let has_trailing_slash = query.ends_with('/');
+
+        if has_trailing_slash {
+            let prefix = query.trim_end_matches('/');
+            let segments: Vec<&str> = prefix.split('/').filter(|s| !s.is_empty()).collect();
+
+            match segments.len() {
+                1 => {
+                    let artist = segments[0];
+                    let mut stmt = self.conn.prepare(
+                        "SELECT a.title FROM albums a
+                         JOIN artists ar ON a.artist_id = ar.id
+                         WHERE ar.name LIKE ?1
+                         ORDER BY a.title LIMIT 25",
+                    )?;
+                    let rows = stmt.query_map([artist], |row| row.get::<_, String>(0))?;
+                    for album in rows.flatten() {
+                        results.push(format!("{artist}/{album}/"));
+                    }
+                }
+                2 => {
+                    let artist = segments[0];
+                    let album = segments[1];
+                    let mut stmt = self.conn.prepare(
+                        "SELECT t.title FROM tracks t
+                         JOIN albums a ON t.album_id = a.id
+                         JOIN artists ar ON a.artist_id = ar.id
+                         WHERE ar.name LIKE ?1 AND a.title LIKE ?2
+                         ORDER BY t.track_number, t.title LIMIT 25",
+                    )?;
+                    let rows = stmt.query_map(rusqlite::params![artist, album], |row| {
+                        row.get::<_, String>(0)
+                    })?;
+                    for track in rows.flatten() {
+                        results.push(format!("{artist}/{album}/{track}"));
+                    }
+                }
+                _ => {}
+            }
+        } else {
+            let segments: Vec<&str> = query.split('/').collect();
+
+            match segments.len() {
+                1 => {
+                    let prefix = segments[0];
+                    let like_pattern = format!("%{prefix}%");
+                    let prefix_pattern = format!("{prefix}%");
+
+                    let mut stmt = self.conn.prepare(
+                        "SELECT name FROM artists
+                         WHERE name LIKE ?1
+                         ORDER BY
+                           CASE WHEN name LIKE ?2 THEN 0 ELSE 1 END,
+                           name
+                         LIMIT 25",
+                    )?;
+                    let rows = stmt
+                        .query_map(rusqlite::params![&like_pattern, &prefix_pattern], |row| {
+                            row.get::<_, String>(0)
+                        })?;
+                    for name in rows.flatten() {
+                        results.push(format!("{name}/"));
+                    }
+                }
+                2 => {
+                    let artist = segments[0];
+                    let album_prefix = segments[1];
+                    if artist.is_empty() {
+                        return Ok(results);
+                    }
+
+                    let artist_pattern = format!("%{artist}%");
+                    let artist_prefix = format!("{artist}%");
+                    let album_pattern = format!("%{album_prefix}%");
+                    let album_prefix_pattern = format!("{album_prefix}%");
+
+                    let mut stmt = self.conn.prepare(
+                        "SELECT ar.name, a.title FROM albums a
+                         JOIN artists ar ON a.artist_id = ar.id
+                         WHERE ar.name LIKE ?1 AND a.title LIKE ?2
+                         ORDER BY
+                           CASE WHEN ar.name LIKE ?3 THEN 0 ELSE 1 END,
+                           CASE WHEN a.title LIKE ?4 THEN 0 ELSE 1 END,
+                           ar.name, a.title
+                         LIMIT 25",
+                    )?;
+                    let rows = stmt.query_map(
+                        rusqlite::params![
+                            &artist_pattern,
+                            &album_pattern,
+                            &artist_prefix,
+                            &album_prefix_pattern,
+                        ],
+                        |row| Ok((row.get::<_, String>(0)?, row.get::<_, String>(1)?)),
+                    )?;
+                    for (name, album) in rows.flatten() {
+                        results.push(format!("{name}/{album}/"));
+                    }
+                }
+                3 => {
+                    let artist = segments[0];
+                    let album = segments[1];
+                    let track_prefix = segments[2];
+                    if artist.is_empty() || album.is_empty() {
+                        return Ok(results);
+                    }
+
+                    let artist_pattern = format!("%{artist}%");
+                    let artist_prefix = format!("{artist}%");
+                    let album_pattern = format!("%{album}%");
+                    let album_prefix = format!("{album}%");
+                    let track_pattern = format!("%{track_prefix}%");
+                    let track_prefix_pattern = format!("{track_prefix}%");
+
+                    let mut stmt = self.conn.prepare(
+                        "SELECT ar.name, a.title, t.title FROM tracks t
+                         JOIN albums a ON t.album_id = a.id
+                         JOIN artists ar ON a.artist_id = ar.id
+                         WHERE ar.name LIKE ?1 AND a.title LIKE ?2 AND t.title LIKE ?3
+                         ORDER BY
+                           CASE WHEN ar.name LIKE ?4 THEN 0 ELSE 1 END,
+                           CASE WHEN a.title LIKE ?5 THEN 0 ELSE 1 END,
+                           CASE WHEN t.title LIKE ?6 THEN 0 ELSE 1 END,
+                           t.track_number, t.title
+                         LIMIT 25",
+                    )?;
+                    let rows = stmt.query_map(
+                        rusqlite::params![
+                            &artist_pattern,
+                            &album_pattern,
+                            &track_pattern,
+                            &artist_prefix,
+                            &album_prefix,
+                            &track_prefix_pattern,
+                        ],
+                        |row| {
+                            Ok((
+                                row.get::<_, String>(0)?,
+                                row.get::<_, String>(1)?,
+                                row.get::<_, String>(2)?,
+                            ))
+                        },
+                    )?;
+                    for (name, album_title, track) in rows.flatten() {
+                        results.push(format!("{name}/{album_title}/{track}"));
+                    }
+                }
+                _ => {}
+            }
+        }
+
+        results.truncate(25);
+
+        Ok(results)
+    }
+
+    #[allow(dead_code)]
+    pub fn open_in_memory() -> SqlResult<Database> {
+        let conn = Connection::open_in_memory()?;
+        let db = Database { conn };
+        db.init_schema()?;
+        Ok(db)
+    }
+
+    /// Find all tracks in an album by case-insensitive artist and album match.
+    pub fn find_tracks_by_album(&self, artist: &str, album: &str) -> SqlResult<Vec<Track>> {
+        let mut stmt = self.conn.prepare(
+            "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title
+             FROM tracks t
+             JOIN albums a ON t.album_id = a.id
+             JOIN artists ar ON a.artist_id = ar.id
+             WHERE ar.name COLLATE NOCASE = ?1 AND a.title COLLATE NOCASE = ?2
+             ORDER BY t.track_number",
+        )?;
+        let rows = stmt.query_map(rusqlite::params![artist, album], |row| {
+            Ok(Track {
+                id: row.get(0)?,
+                album_id: row.get(1)?,
+                title: row.get(2)?,
+                path: PathBuf::from(row.get::<_, String>(3)?),
+                duration: row.get(4)?,
+                track_number: row.get(5)?,
+                artist_name: Some(row.get(6)?),
+                album_title: Some(row.get(7)?),
+            })
+        })?;
+        rows.collect()
+    }
+
+    /// Find a single track by exact artist, exact album, and track title prefix.
+    /// Returns None if no match found.
+    pub fn find_track_by_parts(
+        &self,
+        artist: &str,
+        album: &str,
+        track_prefix: &str,
+    ) -> SqlResult<Option<Track>> {
+        let pattern = format!("{}%", track_prefix);
+        let mut stmt = self.conn.prepare(
+            "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title
+             FROM tracks t
+             JOIN albums a ON t.album_id = a.id
+             JOIN artists ar ON a.artist_id = ar.id
+             WHERE ar.name COLLATE NOCASE = ?1 AND a.title COLLATE NOCASE = ?2 AND t.title LIKE ?3
+             ORDER BY t.track_number LIMIT 1",
+        )?;
+        let mut rows = stmt.query_map(rusqlite::params![artist, album, pattern], |row| {
+            Ok(Track {
+                id: row.get(0)?,
+                album_id: row.get(1)?,
+                title: row.get(2)?,
+                path: PathBuf::from(row.get::<_, String>(3)?),
+                duration: row.get(4)?,
+                track_number: row.get(5)?,
+                artist_name: Some(row.get(6)?),
+                album_title: Some(row.get(7)?),
+            })
+        })?;
+        match rows.next() {
+            Some(Ok(t)) => Ok(Some(t)),
+            Some(Err(e)) => Err(e),
+            None => Ok(None),
+        }
+    }
+
+    /// Resolve a hierarchical path like "Artist/Album" or "Artist/Album/Track"
+    /// to DB tracks. Returns None if the path isn't hierarchical (single part
+    /// or 4+ parts) — the caller should fall back to search_tracks().
+    pub fn resolve_path(&self, path: &str) -> SqlResult<Option<Vec<Track>>> {
+        let path = path.strip_suffix('/').unwrap_or(path);
+        let parts: Vec<&str> = path.splitn(3, '/').collect();
+        match parts.len() {
+            2 => Ok(Some(self.find_tracks_by_album(parts[0], parts[1])?)),
+            3 => match self.find_track_by_parts(parts[0], parts[1], parts[2])? {
+                Some(t) => Ok(Some(vec![t])),
+                None => Ok(Some(vec![])),
+            },
+            _ => Ok(None),
+        }
+    }
+
     pub fn search_tracks(&self, query: &str) -> SqlResult<Vec<Track>> {
         let mut stmt = self.conn.prepare(
             "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title
@@ -336,11 +617,19 @@ pub fn extract_metadata(path: &Path) -> (String, String, String, Option<i32>, Op
         .and_then(|s| s.to_str())
         .unwrap_or("Unknown");
 
-    (
-        grandparent.to_string(),
-        parent.to_string(),
-        file_stem.to_string(),
-        None,
-        None,
-    )
+    let fs_artist = grandparent.to_string();
+    let fs_album = parent.to_string();
+    let fs_title = file_stem.to_string();
+
+    if let Some(audio_meta) = crate::queue::extract_metadata(path) {
+        (
+            audio_meta.artist.unwrap_or(fs_artist),
+            audio_meta.album.unwrap_or(fs_album),
+            audio_meta.title.unwrap_or(fs_title),
+            audio_meta.track_number,
+            audio_meta.duration_secs.map(|d| d as u64),
+        )
+    } else {
+        (fs_artist, fs_album, fs_title, None, None)
+    }
 }