1 files changed, 191 insertions, 105 deletions

diff --git a/src/daemon.rs b/src/daemon.rs
index c794495..5ad6654 100644
--- a/src/daemon.rs
+++ b/src/daemon.rs
@@ -18,14 +18,20 @@ use std::sync::{Arc, Mutex, RwLock};
 use std::thread;
 use std::time::{Duration, Instant, SystemTime};
 
-struct ScrobbleState {
-    track: Option<Track>,
-    start_time: Option<Instant>,
-    timestamp: Option<u64>,
+pub struct ScrobbleState {
+    pub track: Option<Track>,
+    pub start_time: Option<Instant>,
+    pub timestamp: Option<u64>,
+}
+
+impl Default for ScrobbleState {
+    fn default() -> Self {
+        Self::new()
+    }
 }
 
 impl ScrobbleState {
-    fn new() -> Self {
+    pub fn new() -> Self {
         ScrobbleState {
             track: None,
             start_time: None,
@@ -33,7 +39,7 @@ impl ScrobbleState {
         }
     }
 
-    fn start(&mut self, track: Track) {
+    pub fn start(&mut self, track: Track) {
         if track.is_stream {
             crate::debug_eprintln(format_args!(
                 "scrobble: not tracking stream '{}'",
@@ -57,7 +63,7 @@ impl ScrobbleState {
         self.timestamp = Some(ts);
     }
 
-    fn should_scrobble(&self) -> bool {
+    pub fn should_scrobble(&self) -> bool {
         let (Some(track), Some(start)) = (&self.track, self.start_time) else {
             crate::debug_eprintln(format_args!(
                 "scrobble: should_scrobble false — no track or start_time"
@@ -78,7 +84,7 @@ impl ScrobbleState {
         result
     }
 
-    fn should_scrobble_with_params(elapsed: u64, duration_secs: Option<f64>) -> bool {
+    pub fn should_scrobble_with_params(elapsed: u64, duration_secs: Option<f64>) -> bool {
         if elapsed < 30 {
             return false;
         }
@@ -89,86 +95,13 @@ impl ScrobbleState {
         elapsed >= threshold
     }
 
-    fn clear(&mut self) {
+    pub fn clear(&mut self) {
         self.track = None;
         self.start_time = None;
         self.timestamp = None;
     }
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn test_scrobble_state_new() {
-        let s = ScrobbleState::new();
-        assert!(s.track.is_none());
-        assert!(s.start_time.is_none());
-        assert!(s.timestamp.is_none());
-    }
-
-    #[test]
-    fn test_scrobble_state_clear() {
-        let mut s = ScrobbleState::new();
-        let t = Track {
-            path: PathBuf::from("/test/song.mp3"),
-            title: Some("Test".into()),
-            artist: Some("Artist".into()),
-            album: None,
-            duration_secs: Some(200.0),
-            is_stream: false,
-        };
-        s.start(t);
-        assert!(s.track.is_some());
-        s.clear();
-        assert!(s.track.is_none());
-        assert!(s.start_time.is_none());
-        assert!(s.timestamp.is_none());
-    }
-
-    #[test]
-    fn test_should_scrobble_without_track_returns_false() {
-        assert!(!ScrobbleState::new().should_scrobble());
-    }
-
-    #[test]
-    fn test_should_scrobble_threshold_below_30_seconds() {
-        assert!(!ScrobbleState::should_scrobble_with_params(29, Some(200.0)));
-    }
-
-    #[test]
-    fn test_should_scrobble_threshold_half_duration() {
-        assert!(ScrobbleState::should_scrobble_with_params(100, Some(200.0)));
-    }
-
-    #[test]
-    fn test_should_scrobble_threshold_below_half_duration() {
-        assert!(!ScrobbleState::should_scrobble_with_params(80, Some(200.0)));
-    }
-
-    #[test]
-    fn test_should_scrobble_threshold_capped_at_4min() {
-        assert!(ScrobbleState::should_scrobble_with_params(240, Some(600.0)));
-        assert!(!ScrobbleState::should_scrobble_with_params(
-            239,
-            Some(600.0)
-        ));
-    }
-
-    #[test]
-    fn test_should_scrobble_no_duration_requires_4min() {
-        assert!(ScrobbleState::should_scrobble_with_params(240, None));
-        assert!(!ScrobbleState::should_scrobble_with_params(239, None));
-    }
-
-    #[test]
-    fn test_should_scrobble_short_song() {
-        assert!(ScrobbleState::should_scrobble_with_params(30, Some(60.0)));
-        assert!(!ScrobbleState::should_scrobble_with_params(15, Some(60.0)));
-    }
-}
-
 fn try_scrobble_current(scrobbler: &RwLock<Scrobbler>, state: &mut ScrobbleState) {
     if let Some(ref track) = state.track
         && track.is_stream
@@ -747,30 +680,40 @@ fn process_command(
                 } else {
                     match Config::load() {
                         Ok(config) => match crate::database::Database::open(&config) {
-                            Ok(db) => match db.search_tracks(query_str) {
-                                Ok(tracks) if !tracks.is_empty() => {
-                                    let db_track = &tracks[0];
-                                    let album_tracks = db
-                                        .get_album_tracks_after(
-                                            db_track.album_id,
-                                            db_track.track_number,
-                                            &db_track.path,
-                                        )
-                                        .unwrap_or_default();
-                                    let display = format!(
-                                        "{} - {}",
-                                        db_track.artist_name.as_deref().unwrap_or("unknown"),
-                                        db_track.title,
-                                    );
-                                    let paths: Vec<PathBuf> =
-                                        album_tracks.into_iter().map(|t| t.path).collect();
-                                    (db_track.path.clone(), paths, display)
-                                }
-                                Ok(_) => {
+                            Ok(db) => {
+                                let tracks = match db.resolve_path(query_str) {
+                                    Ok(Some(t)) => t,
+                                    Ok(None) => match db.search_tracks(query_str) {
+                                        Ok(t) => t,
+                                        Err(e) => return format!("Database error: {}\n", e),
+                                    },
+                                    Err(e) => return format!("Database error: {}\n", e),
+                                };
+                                if tracks.is_empty() {
                                     return format!("No tracks found matching '{}'\n", query_str);
                                 }
-                                Err(e) => return format!("Database error: {}\n", e),
-                            },
+                                let db_track = &tracks[0];
+                                let album_paths: Vec<PathBuf> = if tracks.len() > 1 {
+                                    // resolve_path returned full album — use remaining tracks
+                                    tracks[1..].iter().map(|t| t.path.clone()).collect()
+                                } else {
+                                    db.get_album_tracks_after(
+                                        db_track.album_id,
+                                        db_track.track_number,
+                                        &db_track.path,
+                                    )
+                                    .unwrap_or_default()
+                                    .into_iter()
+                                    .map(|t| t.path)
+                                    .collect()
+                                };
+                                let display = format!(
+                                    "{} - {}",
+                                    db_track.artist_name.as_deref().unwrap_or("unknown"),
+                                    db_track.title,
+                                );
+                                (db_track.path.clone(), album_paths, display)
+                            }
                             Err(e) => return format!("Database error: {}\n", e),
                         },
                         Err(e) => return format!("Config error: {}\n", e),
@@ -862,6 +805,24 @@ fn process_command(
             } else {
                 path_str.to_string()
             };
+            // Strip trailing slash so autocompleted Artist/Album/ resolves cleanly
+            let path_str = path_str.trim_end_matches('/').to_string();
+
+            // Unescape backslash sequences (e.g., \! → !) so shell escaping works
+            let path_str = {
+                let mut unescaped = String::with_capacity(path_str.len());
+                let mut chars = path_str.chars();
+                while let Some(c) = chars.next() {
+                    if c == '\\' {
+                        if let Some(next) = chars.next() {
+                            unescaped.push(next);
+                        }
+                    } else {
+                        unescaped.push(c);
+                    }
+                }
+                unescaped
+            };
 
             if path_str.is_empty() {
                 return "Usage: play [now|next|last] <path>\n".to_string();
@@ -1052,7 +1013,132 @@ fn process_command(
                         _ => format!("Error: No audio files found in '{}'\n", path_str),
                     }
                 } else {
-                    format!("File not found: {}\n", path_str)
+                    match Config::load() {
+                        Ok(config) => match crate::database::Database::open(&config) {
+                            Ok(db) => {
+                                // Collect tracks: try hierarchical path first, fallback to search
+                                let tracks = match db.resolve_path(&path_str) {
+                                    Ok(Some(t)) => t,
+                                    Ok(None) => {
+                                        let query = path_str
+                                            .rsplit_once('/')
+                                            .and_then(|(_, after)| {
+                                                (!after.is_empty()).then_some(after)
+                                            })
+                                            .unwrap_or(&path_str);
+                                        match db.search_tracks(query) {
+                                            Ok(t) => t,
+                                            Err(e) => return format!("Database error: {}\n", e),
+                                        }
+                                    }
+                                    Err(e) => return format!("Database error: {}\n", e),
+                                };
+
+                                if tracks.is_empty() {
+                                    return format!("No tracks found matching '{}'\n", path_str);
+                                }
+
+                                let is_album_play = tracks.len() > 1;
+                                if let Ok(mut q) = queue.lock() {
+                                    let was_empty = q.is_empty();
+                                    match position {
+                                        "now" => {
+                                            if let Ok(mut s) = scrobble_state.lock() {
+                                                try_scrobble_current(scrobbler, &mut s);
+                                            }
+                                            for t in tracks.iter().rev() {
+                                                q.insert_after_current(t.path.clone());
+                                            }
+                                            if !was_empty {
+                                                let new_pos = q.position() + 1;
+                                                q.set_position(new_pos);
+                                            }
+                                            if let Some(track) = q.track_at(q.position()) {
+                                                audio.play(track.path.clone());
+                                                send_now_playing(scrobbler, track);
+                                                if let Ok(mut s) = scrobble_state.lock() {
+                                                    s.start(track.clone());
+                                                }
+                                                let fmt = Config::load()
+                                                    .ok()
+                                                    .and_then(|c| c.display_format.clone());
+                                                notification::notify_track(track, fmt.as_deref());
+                                            }
+                                            if is_album_play {
+                                                format!(
+                                                    "Playing now: {} tracks from {}\n",
+                                                    tracks.len(),
+                                                    path_str
+                                                )
+                                            } else {
+                                                format!("Playing now: {}\n", path_str)
+                                            }
+                                        }
+                                        "next" => {
+                                            for t in tracks.iter().rev() {
+                                                q.insert_after_current(t.path.clone());
+                                            }
+                                            if is_album_play {
+                                                format!(
+                                                    "Queued next: {} tracks from {}\n",
+                                                    tracks.len(),
+                                                    path_str
+                                                )
+                                            } else {
+                                                format!("Queued next: {}\n", path_str)
+                                            }
+                                        }
+                                        _ => {
+                                            let was_playing = audio.is_playing();
+                                            for t in &tracks {
+                                                q.add_track(t.path.clone());
+                                            }
+                                            if was_empty && !was_playing {
+                                                if is_album_play {
+                                                    q.shuffle_initial();
+                                                }
+                                                if let Some(track) = q.current() {
+                                                    audio.play(track.path.clone());
+                                                    send_now_playing(scrobbler, track);
+                                                    if let Ok(mut s) = scrobble_state.lock() {
+                                                        s.start(track.clone());
+                                                    }
+                                                    let fmt = Config::load()
+                                                        .ok()
+                                                        .and_then(|c| c.display_format.clone());
+                                                    notification::notify_track(
+                                                        track,
+                                                        fmt.as_deref(),
+                                                    );
+                                                }
+                                                if is_album_play {
+                                                    format!(
+                                                        "Playing: {} tracks from {}\n",
+                                                        tracks.len(),
+                                                        path_str
+                                                    )
+                                                } else {
+                                                    format!("Playing: {}\n", path_str)
+                                                }
+                                            } else if is_album_play {
+                                                format!(
+                                                    "Queued {} tracks from: {}\n",
+                                                    tracks.len(),
+                                                    path_str
+                                                )
+                                            } else {
+                                                format!("Queued: {}\n", path_str)
+                                            }
+                                        }
+                                    }
+                                } else {
+                                    "Error accessing queue\n".to_string()
+                                }
+                            }
+                            Err(e) => format!("Database error: {}\n", e),
+                        },
+                        Err(e) => format!("Config error: {}\n", e),
+                    }
                 }
             }
         }