diff options
| author | salmonik <salmonik@noreply.codeberg.org> | 2026-06-06 19:07:09 +0200 |
|---|---|---|
| committer | salmonik <salmonik@noreply.codeberg.org> | 2026-06-06 19:07:09 +0200 |
| commit | 385c2a38eba689154b46d515f6a3a010c34d3119 (patch) | |
| tree | 8916681bafc9822ef371f949a176cf22307d2a2e /src | |
| parent | add verbosity to `rsp scan` (diff) | |
| parent | delete not needed md file (diff) | |
| download | rsplayer-385c2a38eba689154b46d515f6a3a010c34d3119.tar.gz rsplayer-385c2a38eba689154b46d515f6a3a010c34d3119.zip | |
Merge pull request 'autocompletion' (#1) from autocompletion into main
Reviewed-on: https://codeberg.org/salmonik/RSPlayer/pulls/1
Diffstat (limited to 'src')
| -rw-r--r-- | src/cli.rs | 41 | ||||
| -rw-r--r-- | src/daemon.rs | 296 | ||||
| -rw-r--r-- | src/database.rs | 305 | ||||
| -rw-r--r-- | src/queue.rs | 14 | ||||
| -rw-r--r-- | src/scrobbler.rs | 22 |
5 files changed, 542 insertions, 136 deletions
| @@ -92,6 +92,13 @@ pub enum Commands { | |||
| 92 | #[arg(value_name = "SHELL")] | 92 | #[arg(value_name = "SHELL")] |
| 93 | shell: Option<String>, | 93 | shell: Option<String>, |
| 94 | }, | 94 | }, |
| 95 | #[clap(hide = true)] | ||
| 96 | /// Complete artists, albums, or tracks from database | ||
| 97 | DbComplete { | ||
| 98 | /// Query string (e.g. \"J\", \"Artist/\", \"Artist/Album/\") | ||
| 99 | #[arg(value_name = "QUERY")] | ||
| 100 | query: String, | ||
| 101 | }, | ||
| 95 | } | 102 | } |
| 96 | 103 | ||
| 97 | #[derive(Subcommand)] | 104 | #[derive(Subcommand)] |
| @@ -235,11 +242,43 @@ impl Cli { | |||
| 235 | } | 242 | } |
| 236 | } | 243 | } |
| 237 | } | 244 | } |
| 245 | Commands::DbComplete { query } => { | ||
| 246 | // Unescape shell escapes in the input query | ||
| 247 | let query = { | ||
| 248 | let mut unescaped = String::with_capacity(query.len()); | ||
| 249 | let mut chars = query.chars(); | ||
| 250 | while let Some(c) = chars.next() { | ||
| 251 | if c == '\\' { | ||
| 252 | if let Some(next) = chars.next() { | ||
| 253 | unescaped.push(next); | ||
| 254 | } | ||
| 255 | } else { | ||
| 256 | unescaped.push(c); | ||
| 257 | } | ||
| 258 | } | ||
| 259 | unescaped | ||
| 260 | }; | ||
| 261 | |||
| 262 | let config = match Config::load() { | ||
| 263 | Ok(c) => c, | ||
| 264 | Err(_) => return Ok(()), | ||
| 265 | }; | ||
| 266 | let db = match Database::open(&config) { | ||
| 267 | Ok(d) => d, | ||
| 268 | Err(_) => return Ok(()), | ||
| 269 | }; | ||
| 270 | |||
| 271 | let results = db.complete(&query).unwrap_or_default(); | ||
| 272 | |||
| 273 | for r in &results { | ||
| 274 | println!("{r}"); | ||
| 275 | } | ||
| 276 | } | ||
| 238 | Commands::Completions { shell } => { | 277 | Commands::Completions { shell } => { |
| 239 | let full = Cli::command(); | 278 | let full = Cli::command(); |
| 240 | let subcommands: Vec<Command> = full | 279 | let subcommands: Vec<Command> = full |
| 241 | .get_subcommands() | 280 | .get_subcommands() |
| 242 | .filter(|s| s.get_name() != "completions") | 281 | .filter(|s| s.get_name() != "completions" && s.get_name() != "db-complete") |
| 243 | .cloned() | 282 | .cloned() |
| 244 | .collect(); | 283 | .collect(); |
| 245 | let cmd_name = full.get_name().to_string(); | 284 | let cmd_name = full.get_name().to_string(); |
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}; | |||
| 18 | use std::thread; | 18 | use std::thread; |
| 19 | use std::time::{Duration, Instant, SystemTime}; | 19 | use std::time::{Duration, Instant, SystemTime}; |
| 20 | 20 | ||
| 21 | struct ScrobbleState { | 21 | pub struct ScrobbleState { |
| 22 | track: Option<Track>, | 22 | pub track: Option<Track>, |
| 23 | start_time: Option<Instant>, | 23 | pub start_time: Option<Instant>, |
| 24 | timestamp: Option<u64>, | 24 | pub timestamp: Option<u64>, |
| 25 | } | ||
| 26 | |||
| 27 | impl Default for ScrobbleState { | ||
| 28 | fn default() -> Self { | ||
| 29 | Self::new() | ||
| 30 | } | ||
| 25 | } | 31 | } |
| 26 | 32 | ||
| 27 | impl ScrobbleState { | 33 | impl ScrobbleState { |
| 28 | fn new() -> Self { | 34 | pub fn new() -> Self { |
| 29 | ScrobbleState { | 35 | ScrobbleState { |
| 30 | track: None, | 36 | track: None, |
| 31 | start_time: None, | 37 | start_time: None, |
| @@ -33,7 +39,7 @@ impl ScrobbleState { | |||
| 33 | } | 39 | } |
| 34 | } | 40 | } |
| 35 | 41 | ||
| 36 | fn start(&mut self, track: Track) { | 42 | pub fn start(&mut self, track: Track) { |
| 37 | if track.is_stream { | 43 | if track.is_stream { |
| 38 | crate::debug_eprintln(format_args!( | 44 | crate::debug_eprintln(format_args!( |
| 39 | "scrobble: not tracking stream '{}'", | 45 | "scrobble: not tracking stream '{}'", |
| @@ -57,7 +63,7 @@ impl ScrobbleState { | |||
| 57 | self.timestamp = Some(ts); | 63 | self.timestamp = Some(ts); |
| 58 | } | 64 | } |
| 59 | 65 | ||
| 60 | fn should_scrobble(&self) -> bool { | 66 | pub fn should_scrobble(&self) -> bool { |
| 61 | let (Some(track), Some(start)) = (&self.track, self.start_time) else { | 67 | let (Some(track), Some(start)) = (&self.track, self.start_time) else { |
| 62 | crate::debug_eprintln(format_args!( | 68 | crate::debug_eprintln(format_args!( |
| 63 | "scrobble: should_scrobble false — no track or start_time" | 69 | "scrobble: should_scrobble false — no track or start_time" |
| @@ -78,7 +84,7 @@ impl ScrobbleState { | |||
| 78 | result | 84 | result |
| 79 | } | 85 | } |
| 80 | 86 | ||
| 81 | fn should_scrobble_with_params(elapsed: u64, duration_secs: Option<f64>) -> bool { | 87 | pub fn should_scrobble_with_params(elapsed: u64, duration_secs: Option<f64>) -> bool { |
| 82 | if elapsed < 30 { | 88 | if elapsed < 30 { |
| 83 | return false; | 89 | return false; |
| 84 | } | 90 | } |
| @@ -89,86 +95,13 @@ impl ScrobbleState { | |||
| 89 | elapsed >= threshold | 95 | elapsed >= threshold |
| 90 | } | 96 | } |
| 91 | 97 | ||
| 92 | fn clear(&mut self) { | 98 | pub fn clear(&mut self) { |
| 93 | self.track = None; | 99 | self.track = None; |
| 94 | self.start_time = None; | 100 | self.start_time = None; |
| 95 | self.timestamp = None; | 101 | self.timestamp = None; |
| 96 | } | 102 | } |
| 97 | } | 103 | } |
| 98 | 104 | ||
| 99 | #[cfg(test)] | ||
| 100 | mod tests { | ||
| 101 | use super::*; | ||
| 102 | |||
| 103 | #[test] | ||
| 104 | fn test_scrobble_state_new() { | ||
| 105 | let s = ScrobbleState::new(); | ||
| 106 | assert!(s.track.is_none()); | ||
| 107 | assert!(s.start_time.is_none()); | ||
| 108 | assert!(s.timestamp.is_none()); | ||
| 109 | } | ||
| 110 | |||
| 111 | #[test] | ||
| 112 | fn test_scrobble_state_clear() { | ||
| 113 | let mut s = ScrobbleState::new(); | ||
| 114 | let t = Track { | ||
| 115 | path: PathBuf::from("/test/song.mp3"), | ||
| 116 | title: Some("Test".into()), | ||
| 117 | artist: Some("Artist".into()), | ||
| 118 | album: None, | ||
| 119 | duration_secs: Some(200.0), | ||
| 120 | is_stream: false, | ||
| 121 | }; | ||
| 122 | s.start(t); | ||
| 123 | assert!(s.track.is_some()); | ||
| 124 | s.clear(); | ||
| 125 | assert!(s.track.is_none()); | ||
| 126 | assert!(s.start_time.is_none()); | ||
| 127 | assert!(s.timestamp.is_none()); | ||
| 128 | } | ||
| 129 | |||
| 130 | #[test] | ||
| 131 | fn test_should_scrobble_without_track_returns_false() { | ||
| 132 | assert!(!ScrobbleState::new().should_scrobble()); | ||
| 133 | } | ||
| 134 | |||
| 135 | #[test] | ||
| 136 | fn test_should_scrobble_threshold_below_30_seconds() { | ||
| 137 | assert!(!ScrobbleState::should_scrobble_with_params(29, Some(200.0))); | ||
| 138 | } | ||
| 139 | |||
| 140 | #[test] | ||
| 141 | fn test_should_scrobble_threshold_half_duration() { | ||
| 142 | assert!(ScrobbleState::should_scrobble_with_params(100, Some(200.0))); | ||
| 143 | } | ||
| 144 | |||
| 145 | #[test] | ||
| 146 | fn test_should_scrobble_threshold_below_half_duration() { | ||
| 147 | assert!(!ScrobbleState::should_scrobble_with_params(80, Some(200.0))); | ||
| 148 | } | ||
| 149 | |||
| 150 | #[test] | ||
| 151 | fn test_should_scrobble_threshold_capped_at_4min() { | ||
| 152 | assert!(ScrobbleState::should_scrobble_with_params(240, Some(600.0))); | ||
| 153 | assert!(!ScrobbleState::should_scrobble_with_params( | ||
| 154 | 239, | ||
| 155 | Some(600.0) | ||
| 156 | )); | ||
| 157 | } | ||
| 158 | |||
| 159 | #[test] | ||
| 160 | fn test_should_scrobble_no_duration_requires_4min() { | ||
| 161 | assert!(ScrobbleState::should_scrobble_with_params(240, None)); | ||
| 162 | assert!(!ScrobbleState::should_scrobble_with_params(239, None)); | ||
| 163 | } | ||
| 164 | |||
| 165 | #[test] | ||
| 166 | fn test_should_scrobble_short_song() { | ||
| 167 | assert!(ScrobbleState::should_scrobble_with_params(30, Some(60.0))); | ||
| 168 | assert!(!ScrobbleState::should_scrobble_with_params(15, Some(60.0))); | ||
| 169 | } | ||
| 170 | } | ||
| 171 | |||
| 172 | fn try_scrobble_current(scrobbler: &RwLock<Scrobbler>, state: &mut ScrobbleState) { | 105 | fn try_scrobble_current(scrobbler: &RwLock<Scrobbler>, state: &mut ScrobbleState) { |
| 173 | if let Some(ref track) = state.track | 106 | if let Some(ref track) = state.track |
| 174 | && track.is_stream | 107 | && track.is_stream |
| @@ -747,30 +680,40 @@ fn process_command( | |||
| 747 | } else { | 680 | } else { |
| 748 | match Config::load() { | 681 | match Config::load() { |
| 749 | Ok(config) => match crate::database::Database::open(&config) { | 682 | Ok(config) => match crate::database::Database::open(&config) { |
| 750 | Ok(db) => match db.search_tracks(query_str) { | 683 | Ok(db) => { |
| 751 | Ok(tracks) if !tracks.is_empty() => { | 684 | let tracks = match db.resolve_path(query_str) { |
| 752 | let db_track = &tracks[0]; | 685 | Ok(Some(t)) => t, |
| 753 | let album_tracks = db | 686 | Ok(None) => match db.search_tracks(query_str) { |
| 754 | .get_album_tracks_after( | 687 | Ok(t) => t, |
| 755 | db_track.album_id, | 688 | Err(e) => return format!("Database error: {}\n", e), |
| 756 | db_track.track_number, | 689 | }, |
| 757 | &db_track.path, | 690 | Err(e) => return format!("Database error: {}\n", e), |
| 758 | ) | 691 | }; |
| 759 | .unwrap_or_default(); | 692 | if tracks.is_empty() { |
| 760 | let display = format!( | ||
| 761 | "{} - {}", | ||
| 762 | db_track.artist_name.as_deref().unwrap_or("unknown"), | ||
| 763 | db_track.title, | ||
| 764 | ); | ||
| 765 | let paths: Vec<PathBuf> = | ||
| 766 | album_tracks.into_iter().map(|t| t.path).collect(); | ||
| 767 | (db_track.path.clone(), paths, display) | ||
| 768 | } | ||
| 769 | Ok(_) => { | ||
| 770 | return format!("No tracks found matching '{}'\n", query_str); | 693 | return format!("No tracks found matching '{}'\n", query_str); |
| 771 | } | 694 | } |
| 772 | Err(e) => return format!("Database error: {}\n", e), | 695 | let db_track = &tracks[0]; |
| 773 | }, | 696 | let album_paths: Vec<PathBuf> = if tracks.len() > 1 { |
| 697 | // resolve_path returned full album — use remaining tracks | ||
| 698 | tracks[1..].iter().map(|t| t.path.clone()).collect() | ||
| 699 | } else { | ||
| 700 | db.get_album_tracks_after( | ||
| 701 | db_track.album_id, | ||
| 702 | db_track.track_number, | ||
| 703 | &db_track.path, | ||
| 704 | ) | ||
| 705 | .unwrap_or_default() | ||
| 706 | .into_iter() | ||
| 707 | .map(|t| t.path) | ||
| 708 | .collect() | ||
| 709 | }; | ||
| 710 | let display = format!( | ||
| 711 | "{} - {}", | ||
| 712 | db_track.artist_name.as_deref().unwrap_or("unknown"), | ||
| 713 | db_track.title, | ||
| 714 | ); | ||
| 715 | (db_track.path.clone(), album_paths, display) | ||
| 716 | } | ||
| 774 | Err(e) => return format!("Database error: {}\n", e), | 717 | Err(e) => return format!("Database error: {}\n", e), |
| 775 | }, | 718 | }, |
| 776 | Err(e) => return format!("Config error: {}\n", e), | 719 | Err(e) => return format!("Config error: {}\n", e), |
| @@ -862,6 +805,24 @@ fn process_command( | |||
| 862 | } else { | 805 | } else { |
| 863 | path_str.to_string() | 806 | path_str.to_string() |
| 864 | }; | 807 | }; |
| 808 | // Strip trailing slash so autocompleted Artist/Album/ resolves cleanly | ||
| 809 | let path_str = path_str.trim_end_matches('/').to_string(); | ||
| 810 | |||
| 811 | // Unescape backslash sequences (e.g., \! → !) so shell escaping works | ||
| 812 | let path_str = { | ||
| 813 | let mut unescaped = String::with_capacity(path_str.len()); | ||
| 814 | let mut chars = path_str.chars(); | ||
| 815 | while let Some(c) = chars.next() { | ||
| 816 | if c == '\\' { | ||
| 817 | if let Some(next) = chars.next() { | ||
| 818 | unescaped.push(next); | ||
| 819 | } | ||
| 820 | } else { | ||
| 821 | unescaped.push(c); | ||
| 822 | } | ||
| 823 | } | ||
| 824 | unescaped | ||
| 825 | }; | ||
| 865 | 826 | ||
| 866 | if path_str.is_empty() { | 827 | if path_str.is_empty() { |
| 867 | return "Usage: play [now|next|last] <path>\n".to_string(); | 828 | return "Usage: play [now|next|last] <path>\n".to_string(); |
| @@ -1052,7 +1013,132 @@ fn process_command( | |||
| 1052 | _ => format!("Error: No audio files found in '{}'\n", path_str), | 1013 | _ => format!("Error: No audio files found in '{}'\n", path_str), |
| 1053 | } | 1014 | } |
| 1054 | } else { | 1015 | } else { |
| 1055 | format!("File not found: {}\n", path_str) | 1016 | match Config::load() { |
| 1017 | Ok(config) => match crate::database::Database::open(&config) { | ||
| 1018 | Ok(db) => { | ||
| 1019 | // Collect tracks: try hierarchical path first, fallback to search | ||
| 1020 | let tracks = match db.resolve_path(&path_str) { | ||
| 1021 | Ok(Some(t)) => t, | ||
| 1022 | Ok(None) => { | ||
| 1023 | let query = path_str | ||
| 1024 | .rsplit_once('/') | ||
| 1025 | .and_then(|(_, after)| { | ||
| 1026 | (!after.is_empty()).then_some(after) | ||
| 1027 | }) | ||
| 1028 | .unwrap_or(&path_str); | ||
| 1029 | match db.search_tracks(query) { | ||
| 1030 | Ok(t) => t, | ||
| 1031 | Err(e) => return format!("Database error: {}\n", e), | ||
| 1032 | } | ||
| 1033 | } | ||
| 1034 | Err(e) => return format!("Database error: {}\n", e), | ||
| 1035 | }; | ||
| 1036 | |||
| 1037 | if tracks.is_empty() { | ||
| 1038 | return format!("No tracks found matching '{}'\n", path_str); | ||
| 1039 | } | ||
| 1040 | |||
| 1041 | let is_album_play = tracks.len() > 1; | ||
| 1042 | if let Ok(mut q) = queue.lock() { | ||
| 1043 | let was_empty = q.is_empty(); | ||
| 1044 | match position { | ||
| 1045 | "now" => { | ||
| 1046 | if let Ok(mut s) = scrobble_state.lock() { | ||
| 1047 | try_scrobble_current(scrobbler, &mut s); | ||
| 1048 | } | ||
| 1049 | for t in tracks.iter().rev() { | ||
| 1050 | q.insert_after_current(t.path.clone()); | ||
| 1051 | } | ||
| 1052 | if !was_empty { | ||
| 1053 | let new_pos = q.position() + 1; | ||
| 1054 | q.set_position(new_pos); | ||
| 1055 | } | ||
| 1056 | if let Some(track) = q.track_at(q.position()) { | ||
| 1057 | audio.play(track.path.clone()); | ||
| 1058 | send_now_playing(scrobbler, track); | ||
| 1059 | if let Ok(mut s) = scrobble_state.lock() { | ||
| 1060 | s.start(track.clone()); | ||
| 1061 | } | ||
| 1062 | let fmt = Config::load() | ||
| 1063 | .ok() | ||
| 1064 | .and_then(|c| c.display_format.clone()); | ||
| 1065 | notification::notify_track(track, fmt.as_deref()); | ||
| 1066 | } | ||
| 1067 | if is_album_play { | ||
| 1068 | format!( | ||
| 1069 | "Playing now: {} tracks from {}\n", | ||
| 1070 | tracks.len(), | ||
| 1071 | path_str | ||
| 1072 | ) | ||
| 1073 | } else { | ||
| 1074 | format!("Playing now: {}\n", path_str) | ||
| 1075 | } | ||
| 1076 | } | ||
| 1077 | "next" => { | ||
| 1078 | for t in tracks.iter().rev() { | ||
| 1079 | q.insert_after_current(t.path.clone()); | ||
| 1080 | } | ||
| 1081 | if is_album_play { | ||
| 1082 | format!( | ||
| 1083 | "Queued next: {} tracks from {}\n", | ||
| 1084 | tracks.len(), | ||
| 1085 | path_str | ||
| 1086 | ) | ||
| 1087 | } else { | ||
| 1088 | format!("Queued next: {}\n", path_str) | ||
| 1089 | } | ||
| 1090 | } | ||
| 1091 | _ => { | ||
| 1092 | let was_playing = audio.is_playing(); | ||
| 1093 | for t in &tracks { | ||
| 1094 | q.add_track(t.path.clone()); | ||
| 1095 | } | ||
| 1096 | if was_empty && !was_playing { | ||
| 1097 | if is_album_play { | ||
| 1098 | q.shuffle_initial(); | ||
| 1099 | } | ||
| 1100 | if let Some(track) = q.current() { | ||
| 1101 | audio.play(track.path.clone()); | ||
| 1102 | send_now_playing(scrobbler, track); | ||
| 1103 | if let Ok(mut s) = scrobble_state.lock() { | ||
| 1104 | s.start(track.clone()); | ||
| 1105 | } | ||
| 1106 | let fmt = Config::load() | ||
| 1107 | .ok() | ||
| 1108 | .and_then(|c| c.display_format.clone()); | ||
| 1109 | notification::notify_track( | ||
| 1110 | track, | ||
| 1111 | fmt.as_deref(), | ||
| 1112 | ); | ||
| 1113 | } | ||
| 1114 | if is_album_play { | ||
| 1115 | format!( | ||
| 1116 | "Playing: {} tracks from {}\n", | ||
| 1117 | tracks.len(), | ||
| 1118 | path_str | ||
| 1119 | ) | ||
| 1120 | } else { | ||
| 1121 | format!("Playing: {}\n", path_str) | ||
| 1122 | } | ||
| 1123 | } else if is_album_play { | ||
| 1124 | format!( | ||
| 1125 | "Queued {} tracks from: {}\n", | ||
| 1126 | tracks.len(), | ||
| 1127 | path_str | ||
| 1128 | ) | ||
| 1129 | } else { | ||
| 1130 | format!("Queued: {}\n", path_str) | ||
| 1131 | } | ||
| 1132 | } | ||
| 1133 | } | ||
| 1134 | } else { | ||
| 1135 | "Error accessing queue\n".to_string() | ||
| 1136 | } | ||
| 1137 | } | ||
| 1138 | Err(e) => format!("Database error: {}\n", e), | ||
| 1139 | }, | ||
| 1140 | Err(e) => format!("Config error: {}\n", e), | ||
| 1141 | } | ||
| 1056 | } | 1142 | } |
| 1057 | } | 1143 | } |
| 1058 | } | 1144 | } |
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 { | |||
| 45 | 45 | ||
| 46 | #[allow(dead_code)] | 46 | #[allow(dead_code)] |
| 47 | impl Database { | 47 | impl Database { |
| 48 | pub fn conn(&self) -> &Connection { | ||
| 49 | &self.conn | ||
| 50 | } | ||
| 51 | |||
| 48 | pub fn open(_config: &crate::config::Config) -> SqlResult<Self> { | 52 | pub fn open(_config: &crate::config::Config) -> SqlResult<Self> { |
| 49 | let proj_dirs = ProjectDirs::from("com", "rsplayer", "rsplayer").ok_or_else(|| { | 53 | let proj_dirs = ProjectDirs::from("com", "rsplayer", "rsplayer").ok_or_else(|| { |
| 50 | rusqlite::Error::InvalidPath("Failed to get project directories".into()) | 54 | rusqlite::Error::InvalidPath("Failed to get project directories".into()) |
| @@ -59,7 +63,29 @@ impl Database { | |||
| 59 | fs::create_dir_all(&data_dir).ok(); | 63 | fs::create_dir_all(&data_dir).ok(); |
| 60 | let db_path = data_dir.join("library.db"); | 64 | let db_path = data_dir.join("library.db"); |
| 61 | 65 | ||
| 62 | let conn = Connection::open(&db_path)?; | 66 | let debug_log = std::env::temp_dir().join("rsp-completion-debug.log"); |
| 67 | macro_rules! debug_log { | ||
| 68 | ($($arg:tt)*) => { | ||
| 69 | let _ = std::fs::OpenOptions::new() | ||
| 70 | .create(true) | ||
| 71 | .append(true) | ||
| 72 | .open(&debug_log) | ||
| 73 | .and_then(|mut f| { | ||
| 74 | use std::io::Write; | ||
| 75 | writeln!(f, $($arg)*) | ||
| 76 | }); | ||
| 77 | }; | ||
| 78 | } | ||
| 79 | debug_log!("[Database::open] db_path: {:?}", db_path); | ||
| 80 | debug_log!("[Database::open] db exists: {}", db_path.exists()); | ||
| 81 | |||
| 82 | let conn = match Connection::open(&db_path) { | ||
| 83 | Ok(c) => c, | ||
| 84 | Err(e) => { | ||
| 85 | debug_log!("[Database::open] Connection::open failed: {e}"); | ||
| 86 | return Err(e); | ||
| 87 | } | ||
| 88 | }; | ||
| 63 | let db = Database { conn }; | 89 | let db = Database { conn }; |
| 64 | db.init_schema()?; | 90 | db.init_schema()?; |
| 65 | Ok(db) | 91 | Ok(db) |
| @@ -279,6 +305,261 @@ impl Database { | |||
| 279 | self.get_album_tracks_after(album_id, track_number, &path) | 305 | self.get_album_tracks_after(album_id, track_number, &path) |
| 280 | } | 306 | } |
| 281 | 307 | ||
| 308 | pub fn complete(&self, query: &str) -> SqlResult<Vec<String>> { | ||
| 309 | let mut results: Vec<String> = Vec::new(); | ||
| 310 | |||
| 311 | if query.is_empty() { | ||
| 312 | let mut stmt = self | ||
| 313 | .conn | ||
| 314 | .prepare("SELECT name FROM artists ORDER BY name LIMIT 25")?; | ||
| 315 | let rows = stmt.query_map([], |row| row.get::<_, String>(0))?; | ||
| 316 | for name in rows.flatten() { | ||
| 317 | results.push(format!("{name}/")); | ||
| 318 | } | ||
| 319 | return Ok(results); | ||
| 320 | } | ||
| 321 | |||
| 322 | let has_trailing_slash = query.ends_with('/'); | ||
| 323 | |||
| 324 | if has_trailing_slash { | ||
| 325 | let prefix = query.trim_end_matches('/'); | ||
| 326 | let segments: Vec<&str> = prefix.split('/').filter(|s| !s.is_empty()).collect(); | ||
| 327 | |||
| 328 | match segments.len() { | ||
| 329 | 1 => { | ||
| 330 | let artist = segments[0]; | ||
| 331 | let mut stmt = self.conn.prepare( | ||
| 332 | "SELECT a.title FROM albums a | ||
| 333 | JOIN artists ar ON a.artist_id = ar.id | ||
| 334 | WHERE ar.name LIKE ?1 | ||
| 335 | ORDER BY a.title LIMIT 25", | ||
| 336 | )?; | ||
| 337 | let rows = stmt.query_map([artist], |row| row.get::<_, String>(0))?; | ||
| 338 | for album in rows.flatten() { | ||
| 339 | results.push(format!("{artist}/{album}/")); | ||
| 340 | } | ||
| 341 | } | ||
| 342 | 2 => { | ||
| 343 | let artist = segments[0]; | ||
| 344 | let album = segments[1]; | ||
| 345 | let mut stmt = self.conn.prepare( | ||
| 346 | "SELECT t.title FROM tracks t | ||
| 347 | JOIN albums a ON t.album_id = a.id | ||
| 348 | JOIN artists ar ON a.artist_id = ar.id | ||
| 349 | WHERE ar.name LIKE ?1 AND a.title LIKE ?2 | ||
| 350 | ORDER BY t.track_number, t.title LIMIT 25", | ||
| 351 | )?; | ||
| 352 | let rows = stmt.query_map(rusqlite::params![artist, album], |row| { | ||
| 353 | row.get::<_, String>(0) | ||
| 354 | })?; | ||
| 355 | for track in rows.flatten() { | ||
| 356 | results.push(format!("{artist}/{album}/{track}")); | ||
| 357 | } | ||
| 358 | } | ||
| 359 | _ => {} | ||
| 360 | } | ||
| 361 | } else { | ||
| 362 | let segments: Vec<&str> = query.split('/').collect(); | ||
| 363 | |||
| 364 | match segments.len() { | ||
| 365 | 1 => { | ||
| 366 | let prefix = segments[0]; | ||
| 367 | let like_pattern = format!("%{prefix}%"); | ||
| 368 | let prefix_pattern = format!("{prefix}%"); | ||
| 369 | |||
| 370 | let mut stmt = self.conn.prepare( | ||
| 371 | "SELECT name FROM artists | ||
| 372 | WHERE name LIKE ?1 | ||
| 373 | ORDER BY | ||
| 374 | CASE WHEN name LIKE ?2 THEN 0 ELSE 1 END, | ||
| 375 | name | ||
| 376 | LIMIT 25", | ||
| 377 | )?; | ||
| 378 | let rows = stmt | ||
| 379 | .query_map(rusqlite::params![&like_pattern, &prefix_pattern], |row| { | ||
| 380 | row.get::<_, String>(0) | ||
| 381 | })?; | ||
| 382 | for name in rows.flatten() { | ||
| 383 | results.push(format!("{name}/")); | ||
| 384 | } | ||
| 385 | } | ||
| 386 | 2 => { | ||
| 387 | let artist = segments[0]; | ||
| 388 | let album_prefix = segments[1]; | ||
| 389 | if artist.is_empty() { | ||
| 390 | return Ok(results); | ||
| 391 | } | ||
| 392 | |||
| 393 | let artist_pattern = format!("%{artist}%"); | ||
| 394 | let artist_prefix = format!("{artist}%"); | ||
| 395 | let album_pattern = format!("%{album_prefix}%"); | ||
| 396 | let album_prefix_pattern = format!("{album_prefix}%"); | ||
| 397 | |||
| 398 | let mut stmt = self.conn.prepare( | ||
| 399 | "SELECT ar.name, a.title FROM albums a | ||
| 400 | JOIN artists ar ON a.artist_id = ar.id | ||
| 401 | WHERE ar.name LIKE ?1 AND a.title LIKE ?2 | ||
| 402 | ORDER BY | ||
| 403 | CASE WHEN ar.name LIKE ?3 THEN 0 ELSE 1 END, | ||
| 404 | CASE WHEN a.title LIKE ?4 THEN 0 ELSE 1 END, | ||
| 405 | ar.name, a.title | ||
| 406 | LIMIT 25", | ||
| 407 | )?; | ||
| 408 | let rows = stmt.query_map( | ||
| 409 | rusqlite::params![ | ||
| 410 | &artist_pattern, | ||
| 411 | &album_pattern, | ||
| 412 | &artist_prefix, | ||
| 413 | &album_prefix_pattern, | ||
| 414 | ], | ||
| 415 | |row| Ok((row.get::<_, String>(0)?, row.get::<_, String>(1)?)), | ||
| 416 | )?; | ||
| 417 | for (name, album) in rows.flatten() { | ||
| 418 | results.push(format!("{name}/{album}/")); | ||
| 419 | } | ||
| 420 | } | ||
| 421 | 3 => { | ||
| 422 | let artist = segments[0]; | ||
| 423 | let album = segments[1]; | ||
| 424 | let track_prefix = segments[2]; | ||
| 425 | if artist.is_empty() || album.is_empty() { | ||
| 426 | return Ok(results); | ||
| 427 | } | ||
| 428 | |||
| 429 | let artist_pattern = format!("%{artist}%"); | ||
| 430 | let artist_prefix = format!("{artist}%"); | ||
| 431 | let album_pattern = format!("%{album}%"); | ||
| 432 | let album_prefix = format!("{album}%"); | ||
| 433 | let track_pattern = format!("%{track_prefix}%"); | ||
| 434 | let track_prefix_pattern = format!("{track_prefix}%"); | ||
| 435 | |||
| 436 | let mut stmt = self.conn.prepare( | ||
| 437 | "SELECT ar.name, a.title, t.title FROM tracks t | ||
| 438 | JOIN albums a ON t.album_id = a.id | ||
| 439 | JOIN artists ar ON a.artist_id = ar.id | ||
| 440 | WHERE ar.name LIKE ?1 AND a.title LIKE ?2 AND t.title LIKE ?3 | ||
| 441 | ORDER BY | ||
| 442 | CASE WHEN ar.name LIKE ?4 THEN 0 ELSE 1 END, | ||
| 443 | CASE WHEN a.title LIKE ?5 THEN 0 ELSE 1 END, | ||
| 444 | CASE WHEN t.title LIKE ?6 THEN 0 ELSE 1 END, | ||
| 445 | t.track_number, t.title | ||
| 446 | LIMIT 25", | ||
| 447 | )?; | ||
| 448 | let rows = stmt.query_map( | ||
| 449 | rusqlite::params![ | ||
| 450 | &artist_pattern, | ||
| 451 | &album_pattern, | ||
| 452 | &track_pattern, | ||
| 453 | &artist_prefix, | ||
| 454 | &album_prefix, | ||
| 455 | &track_prefix_pattern, | ||
| 456 | ], | ||
| 457 | |row| { | ||
| 458 | Ok(( | ||
| 459 | row.get::<_, String>(0)?, | ||
| 460 | row.get::<_, String>(1)?, | ||
| 461 | row.get::<_, String>(2)?, | ||
| 462 | )) | ||
| 463 | }, | ||
| 464 | )?; | ||
| 465 | for (name, album_title, track) in rows.flatten() { | ||
| 466 | results.push(format!("{name}/{album_title}/{track}")); | ||
| 467 | } | ||
| 468 | } | ||
| 469 | _ => {} | ||
| 470 | } | ||
| 471 | } | ||
| 472 | |||
| 473 | results.truncate(25); | ||
| 474 | |||
| 475 | Ok(results) | ||
| 476 | } | ||
| 477 | |||
| 478 | #[allow(dead_code)] | ||
| 479 | pub fn open_in_memory() -> SqlResult<Database> { | ||
| 480 | let conn = Connection::open_in_memory()?; | ||
| 481 | let db = Database { conn }; | ||
| 482 | db.init_schema()?; | ||
| 483 | Ok(db) | ||
| 484 | } | ||
| 485 | |||
| 486 | /// Find all tracks in an album by case-insensitive artist and album match. | ||
| 487 | pub fn find_tracks_by_album(&self, artist: &str, album: &str) -> SqlResult<Vec<Track>> { | ||
| 488 | let mut stmt = self.conn.prepare( | ||
| 489 | "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title | ||
| 490 | FROM tracks t | ||
| 491 | JOIN albums a ON t.album_id = a.id | ||
| 492 | JOIN artists ar ON a.artist_id = ar.id | ||
| 493 | WHERE ar.name COLLATE NOCASE = ?1 AND a.title COLLATE NOCASE = ?2 | ||
| 494 | ORDER BY t.track_number", | ||
| 495 | )?; | ||
| 496 | let rows = stmt.query_map(rusqlite::params![artist, album], |row| { | ||
| 497 | Ok(Track { | ||
| 498 | id: row.get(0)?, | ||
| 499 | album_id: row.get(1)?, | ||
| 500 | title: row.get(2)?, | ||
| 501 | path: PathBuf::from(row.get::<_, String>(3)?), | ||
| 502 | duration: row.get(4)?, | ||
| 503 | track_number: row.get(5)?, | ||
| 504 | artist_name: Some(row.get(6)?), | ||
| 505 | album_title: Some(row.get(7)?), | ||
| 506 | }) | ||
| 507 | })?; | ||
| 508 | rows.collect() | ||
| 509 | } | ||
| 510 | |||
| 511 | /// Find a single track by exact artist, exact album, and track title prefix. | ||
| 512 | /// Returns None if no match found. | ||
| 513 | pub fn find_track_by_parts( | ||
| 514 | &self, | ||
| 515 | artist: &str, | ||
| 516 | album: &str, | ||
| 517 | track_prefix: &str, | ||
| 518 | ) -> SqlResult<Option<Track>> { | ||
| 519 | let pattern = format!("{}%", track_prefix); | ||
| 520 | let mut stmt = self.conn.prepare( | ||
| 521 | "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title | ||
| 522 | FROM tracks t | ||
| 523 | JOIN albums a ON t.album_id = a.id | ||
| 524 | JOIN artists ar ON a.artist_id = ar.id | ||
| 525 | WHERE ar.name COLLATE NOCASE = ?1 AND a.title COLLATE NOCASE = ?2 AND t.title LIKE ?3 | ||
| 526 | ORDER BY t.track_number LIMIT 1", | ||
| 527 | )?; | ||
| 528 | let mut rows = stmt.query_map(rusqlite::params![artist, album, pattern], |row| { | ||
| 529 | Ok(Track { | ||
| 530 | id: row.get(0)?, | ||
| 531 | album_id: row.get(1)?, | ||
| 532 | title: row.get(2)?, | ||
| 533 | path: PathBuf::from(row.get::<_, String>(3)?), | ||
| 534 | duration: row.get(4)?, | ||
| 535 | track_number: row.get(5)?, | ||
| 536 | artist_name: Some(row.get(6)?), | ||
| 537 | album_title: Some(row.get(7)?), | ||
| 538 | }) | ||
| 539 | })?; | ||
| 540 | match rows.next() { | ||
| 541 | Some(Ok(t)) => Ok(Some(t)), | ||
| 542 | Some(Err(e)) => Err(e), | ||
| 543 | None => Ok(None), | ||
| 544 | } | ||
| 545 | } | ||
| 546 | |||
| 547 | /// Resolve a hierarchical path like "Artist/Album" or "Artist/Album/Track" | ||
| 548 | /// to DB tracks. Returns None if the path isn't hierarchical (single part | ||
| 549 | /// or 4+ parts) — the caller should fall back to search_tracks(). | ||
| 550 | pub fn resolve_path(&self, path: &str) -> SqlResult<Option<Vec<Track>>> { | ||
| 551 | let path = path.strip_suffix('/').unwrap_or(path); | ||
| 552 | let parts: Vec<&str> = path.splitn(3, '/').collect(); | ||
| 553 | match parts.len() { | ||
| 554 | 2 => Ok(Some(self.find_tracks_by_album(parts[0], parts[1])?)), | ||
| 555 | 3 => match self.find_track_by_parts(parts[0], parts[1], parts[2])? { | ||
| 556 | Some(t) => Ok(Some(vec![t])), | ||
| 557 | None => Ok(Some(vec![])), | ||
| 558 | }, | ||
| 559 | _ => Ok(None), | ||
| 560 | } | ||
| 561 | } | ||
| 562 | |||
| 282 | pub fn search_tracks(&self, query: &str) -> SqlResult<Vec<Track>> { | 563 | pub fn search_tracks(&self, query: &str) -> SqlResult<Vec<Track>> { |
| 283 | let mut stmt = self.conn.prepare( | 564 | let mut stmt = self.conn.prepare( |
| 284 | "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title | 565 | "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 | |||
| 336 | .and_then(|s| s.to_str()) | 617 | .and_then(|s| s.to_str()) |
| 337 | .unwrap_or("Unknown"); | 618 | .unwrap_or("Unknown"); |
| 338 | 619 | ||
| 339 | ( | 620 | let fs_artist = grandparent.to_string(); |
| 340 | grandparent.to_string(), | 621 | let fs_album = parent.to_string(); |
| 341 | parent.to_string(), | 622 | let fs_title = file_stem.to_string(); |
| 342 | file_stem.to_string(), | 623 | |
| 343 | None, | 624 | if let Some(audio_meta) = crate::queue::extract_metadata(path) { |
| 344 | None, | 625 | ( |
| 345 | ) | 626 | audio_meta.artist.unwrap_or(fs_artist), |
| 627 | audio_meta.album.unwrap_or(fs_album), | ||
| 628 | audio_meta.title.unwrap_or(fs_title), | ||
| 629 | audio_meta.track_number, | ||
| 630 | audio_meta.duration_secs.map(|d| d as u64), | ||
| 631 | ) | ||
| 632 | } else { | ||
| 633 | (fs_artist, fs_album, fs_title, None, None) | ||
| 634 | } | ||
| 346 | } | 635 | } |
diff --git a/src/queue.rs b/src/queue.rs index 3c1c6fa..7675ddd 100644 --- a/src/queue.rs +++ b/src/queue.rs | |||
| @@ -21,6 +21,7 @@ pub struct AudioMetadata { | |||
| 21 | pub title: Option<String>, | 21 | pub title: Option<String>, |
| 22 | pub artist: Option<String>, | 22 | pub artist: Option<String>, |
| 23 | pub album: Option<String>, | 23 | pub album: Option<String>, |
| 24 | pub track_number: Option<i32>, | ||
| 24 | pub duration_secs: Option<f64>, | 25 | pub duration_secs: Option<f64>, |
| 25 | } | 26 | } |
| 26 | 27 | ||
| @@ -64,16 +65,27 @@ pub fn extract_metadata(path: &Path) -> Option<AudioMetadata> { | |||
| 64 | Some(StandardTag::Artist(value)) => { | 65 | Some(StandardTag::Artist(value)) => { |
| 65 | metadata.artist = Some(value.to_string()); | 66 | metadata.artist = Some(value.to_string()); |
| 66 | } | 67 | } |
| 68 | Some(StandardTag::AlbumArtist(value)) => { | ||
| 69 | metadata.artist = Some(value.to_string()); | ||
| 70 | } | ||
| 67 | Some(StandardTag::Album(value)) => { | 71 | Some(StandardTag::Album(value)) => { |
| 68 | metadata.album = Some(value.to_string()); | 72 | metadata.album = Some(value.to_string()); |
| 69 | } | 73 | } |
| 74 | Some(StandardTag::TrackNumber(value)) => { | ||
| 75 | metadata.track_number = value.to_string().parse().ok(); | ||
| 76 | } | ||
| 70 | _ => {} | 77 | _ => {} |
| 71 | } | 78 | } |
| 72 | } | 79 | } |
| 73 | } | 80 | } |
| 74 | } | 81 | } |
| 75 | 82 | ||
| 76 | if metadata.title.is_some() || metadata.artist.is_some() || metadata.album.is_some() { | 83 | if metadata.title.is_some() |
| 84 | || metadata.artist.is_some() | ||
| 85 | || metadata.album.is_some() | ||
| 86 | || metadata.track_number.is_some() | ||
| 87 | || metadata.duration_secs.is_some() | ||
| 88 | { | ||
| 77 | Some(metadata) | 89 | Some(metadata) |
| 78 | } else { | 90 | } else { |
| 79 | None | 91 | None |
diff --git a/src/scrobbler.rs b/src/scrobbler.rs index e133669..40d3ce6 100644 --- a/src/scrobbler.rs +++ b/src/scrobbler.rs | |||
| @@ -162,7 +162,7 @@ impl Scrobbler { | |||
| 162 | } | 162 | } |
| 163 | } | 163 | } |
| 164 | 164 | ||
| 165 | fn sign_lastfm(params: &BTreeMap<&str, &str>, secret: &str) -> String { | 165 | pub fn sign_lastfm(params: &BTreeMap<&str, &str>, secret: &str) -> String { |
| 166 | let mut concat = String::new(); | 166 | let mut concat = String::new(); |
| 167 | for (key, value) in params { | 167 | for (key, value) in params { |
| 168 | concat.push_str(key); | 168 | concat.push_str(key); |
| @@ -333,23 +333,3 @@ impl Scrobbler { | |||
| 333 | Ok(()) | 333 | Ok(()) |
| 334 | } | 334 | } |
| 335 | } | 335 | } |
| 336 | |||
| 337 | #[cfg(test)] | ||
| 338 | mod tests { | ||
| 339 | use super::*; | ||
| 340 | |||
| 341 | #[test] | ||
| 342 | fn test_sign_lastfm() { | ||
| 343 | let params = BTreeMap::from([ | ||
| 344 | ("api_key", "abc123"), | ||
| 345 | ("method", "track.scrobble"), | ||
| 346 | ("sk", "session_key"), | ||
| 347 | ("track", "Test Track"), | ||
| 348 | ("artist", "Test Artist"), | ||
| 349 | ("timestamp", "1234567890"), | ||
| 350 | ]); | ||
| 351 | let sig = Scrobbler::sign_lastfm(¶ms, "sekrit"); | ||
| 352 | assert_eq!(sig.len(), 32); | ||
| 353 | assert!(sig.chars().all(|c| c.is_ascii_hexdigit())); | ||
| 354 | } | ||
| 355 | } | ||
