//! Music source implementation for Qobuz //! //! This module implements the [`pmosource::MusicSource`] trait for Qobuz, //! providing a complete music catalog browsing and searching experience. use crate::client::QobuzClient; use crate::didl::ToDIDL; use crate::lazy_provider::QobuzLazyProvider; use crate::models::Track; use pmoaudiocache::{AudioMetadata, Cache as AudioCache}; use pmocovers::Cache as CoverCache; use pmodidl::{Container, Item}; use pmosource::SourceCacheManager; use pmosource::{async_trait, BrowseResult, MusicSource, MusicSourceError, Result}; use serde_json::json; use std::sync::Arc; use std::time::{Duration, SystemTime}; /// TTL pour les playlists d'albums (7 jours) const ALBUM_PLAYLIST_TTL: Duration = Duration::from_secs(7 * 24 * 3600); /// Default image for Qobuz (300x300 WebP, embedded in binary) const DEFAULT_IMAGE: &[u8] = include_bytes!("../assets/default.webp"); /// Qobuz music source with full MusicSource trait implementation /// /// This struct combines a [`QobuzClient`] for API access with browsing and /// navigation capabilities, implementing the complete [`MusicSource`] trait. /// /// # Features /// /// - **Catalog Navigation**: Browse albums, artists, playlists, favorites /// - **Search**: Full-text search across the Qobuz catalog /// - **URI Resolution**: Resolves track streaming URIs with authentication /// - **DIDL-Lite Export**: Converts albums, tracks, and playlists to UPnP formats /// - **Caching**: Integrated with QobuzClient's cache for performance /// /// # Architecture /// /// Unlike streaming sources like Radio Paradise, Qobuz is a catalog-based source: /// - Root container has multiple sub-containers (Albums, Artists, Favorites, etc.) /// - No FIFO support (it's a static catalog, not a dynamic stream) /// - Hierarchical browsing: Root → Category → Albums → Tracks /// /// # Examples /// /// ```no_run /// use std::sync::Arc; /// use pmoaudiocache::cache as audio_cache; /// use pmocovers::cache as cover_cache; /// use pmoqobuz::{QobuzSource, QobuzClient}; /// use pmosource::MusicSource; /// /// #[tokio::main] /// async fn main() -> Result<(), Box> { /// let client = QobuzClient::from_config().await?; /// let cover_cache = Arc::new(cover_cache::new_cache("/tmp/qobuz_covers", 256)?); /// let audio_cache = Arc::new(audio_cache::new_cache("/tmp/qobuz_audio", 64)?); /// let source = QobuzSource::new(client, cover_cache, audio_cache, "http://localhost:8080"); /// /// println!("Source: {}", source.name()); /// println!("Supports FIFO: {}", source.supports_fifo()); /// /// // Browse root container /// let root = source.root_container().await?; /// println!("Root: {} with {} children", root.title, root.child_count.unwrap_or_default()); /// /// Ok(()) /// } /// ``` #[derive(Clone)] pub struct QobuzSource { inner: Arc, } struct QobuzSourceInner { /// Qobuz API client client: Arc, /// Cache manager (centralisé) cache_manager: SourceCacheManager, /// Base URL for streaming server (e.g., "http://192.168.0.138:8080") base_url: String, /// Update tracking update_counter: tokio::sync::RwLock, last_change: tokio::sync::RwLock, } impl std::fmt::Debug for QobuzSource { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { f.debug_struct("QobuzSource").finish() } } impl QobuzSource { /// Create a new Qobuz source from the cache registry /// /// This is the recommended way to create a source when using the UPnP server. /// The caches are automatically retrieved from the global registry. /// /// # Arguments /// /// * `client` - Authenticated Qobuz API client /// * `base_url` - Base URL for streaming server (e.g., "http://192.168.0.138:8080") /// /// # Errors /// /// Returns an error if the caches are not initialized in the registry #[cfg(feature = "server")] pub fn from_registry(client: QobuzClient, base_url: impl Into) -> Result { let cache_manager = SourceCacheManager::from_registry("qobuz".to_string())?; let client = Arc::new(client); cache_manager.register_lazy_provider(Arc::new(QobuzLazyProvider::new(client.clone()))); Ok(Self { inner: Arc::new(QobuzSourceInner { client, cache_manager, base_url: base_url.into(), update_counter: tokio::sync::RwLock::new(0), last_change: tokio::sync::RwLock::new(SystemTime::now()), }), }) } /// Create a new Qobuz source with explicit caches (for tests) /// /// # Arguments /// /// * `client` - Authenticated Qobuz API client /// * `cover_cache` - Cover image cache (required) /// * `audio_cache` - Audio cache (required) /// * `base_url` - Base URL for streaming server (e.g., "http://192.168.0.138:8080") pub fn new( client: QobuzClient, cover_cache: Arc, audio_cache: Arc, base_url: impl Into, ) -> Self { let cache_manager = SourceCacheManager::new("qobuz".to_string(), cover_cache, audio_cache); let client = Arc::new(client); cache_manager.register_lazy_provider(Arc::new(QobuzLazyProvider::new(client.clone()))); Self { inner: Arc::new(QobuzSourceInner { client, cache_manager, base_url: base_url.into(), update_counter: tokio::sync::RwLock::new(0), last_change: tokio::sync::RwLock::new(SystemTime::now()), }), } } /// Get the Qobuz client pub fn client(&self) -> &QobuzClient { &self.inner.client } /// Add a track from Qobuz with caching /// /// This method downloads and caches both cover art and audio data. pub async fn add_track(&self, track: &Track) -> Result { let track_id = format!("qobuz://track/{}", track.id); // Get streaming URL let stream_url = self .inner .client .get_stream_url(&track.id) .await .map_err(|e| MusicSourceError::UriResolutionError(e.to_string()))?; // 1. Cache cover via manager let cached_cover_pk = if let Some(ref album) = track.album { if let Some(ref image_url) = album.image { self.inner.cache_manager.cache_cover(image_url).await.ok() } else { None } } else { None }; // 2. Prepare rich metadata from Qobuz track let metadata = AudioMetadata { title: Some(track.title.clone()), artist: track.performer.as_ref().map(|p| p.name.clone()), album: track.album.as_ref().map(|a| a.title.clone()), duration_secs: Some(track.duration as u64), year: track.album.as_ref().and_then(|a| { a.release_date .as_ref() .and_then(|d| d.split('-').next()?.parse().ok()) }), track_number: Some(track.track_number), track_total: track.album.as_ref().and_then(|a| a.tracks_count), disc_number: Some(track.media_number), disc_total: None, genre: track.album.as_ref().and_then(|a| { if !a.genres.is_empty() { Some(a.genres.join(", ")) } else { None } }), sample_rate: track.sample_rate, channels: track.channels, bitrate: None, conversion: None, }; // 3. Cache audio via manager let cached_audio_pk = self .inner .cache_manager .cache_audio(&stream_url, Some(metadata)) .await .ok(); if let (Some(ref audio_pk), Some(ref cover_pk)) = (&cached_audio_pk, &cached_cover_pk) { let _ = self.inner .cache_manager .set_audio_metadata(audio_pk, "cover_pk", json!(cover_pk)); } // 4. Store metadata self.inner .cache_manager .update_metadata( track_id.clone(), pmosource::TrackMetadata { original_uri: stream_url, cached_audio_pk, cached_cover_pk, }, ) .await; Ok(track_id) } /// Add track with lazy audio caching (cover eager, audio lazy) /// /// This method caches cover art immediately (small, needed for UI) but /// defers audio download until the track is actually played. /// /// # Arguments /// /// * `track` - The Qobuz track to add /// /// # Returns /// /// `(track_id, lazy_pk)` where `track_id` is the logical Qobuz URI and /// `lazy_pk` the cache identifier stored in pmocache. pub async fn add_track_lazy(&self, track: &Track) -> Result<(String, String)> { let track_id = format!("qobuz://track/{}", track.id); let lazy_pk = format!("QOBUZ:{}", track.id); // Get streaming URL (for metadata fallback) let stream_url = self .inner .client .get_stream_url(&track.id) .await .map_err(|e| MusicSourceError::UriResolutionError(e.to_string()))?; // 1. Cache cover EAGERLY (small, UI needs it) let cached_cover_pk = if let Some(ref album) = track.album { if let Some(ref image_url) = album.image { self.inner.cache_manager.cache_cover(image_url).await.ok() } else { None } } else { None }; // 2. Prepare rich metadata from Qobuz track let metadata = AudioMetadata { title: Some(track.title.clone()), artist: track.performer.as_ref().map(|p| p.name.clone()), album: track.album.as_ref().map(|a| a.title.clone()), duration_secs: Some(track.duration as u64), year: track.album.as_ref().and_then(|a| { a.release_date .as_ref() .and_then(|d| d.split('-').next()?.parse().ok()) }), track_number: Some(track.track_number), track_total: track.album.as_ref().and_then(|a| a.tracks_count), disc_number: Some(track.media_number), disc_total: None, genre: track.album.as_ref().and_then(|a| { if !a.genres.is_empty() { Some(a.genres.join(", ")) } else { None } }), sample_rate: track.sample_rate, channels: track.channels, bitrate: None, conversion: None, }; // 3. Cache audio LAZILY avec un provider let cached_audio_pk = self .inner .cache_manager .cache_audio_lazy_with_provider( &lazy_pk, Some(metadata.clone()), cached_cover_pk.clone(), ) .await .map_err(|e| { MusicSourceError::CacheError(format!( "Failed to register lazy track {}: {}", track.title, e )) })?; if let Some(ref cover_pk) = cached_cover_pk { let _ = self.inner.cache_manager.set_audio_metadata( &cached_audio_pk, "cover_pk", json!(cover_pk), ); } // Stocker le track_id Qobuz pour reconstruction DIDL ultérieure let _ = self.inner.cache_manager.set_audio_metadata( &cached_audio_pk, "qobuz_track_id", json!(track.id), ); // 4. Store metadata self.inner .cache_manager .update_metadata( track_id.clone(), pmosource::TrackMetadata { original_uri: stream_url, cached_audio_pk: Some(cached_audio_pk.clone()), cached_cover_pk, }, ) .await; Ok((track_id, cached_audio_pk)) } /// Load full album into pmoplaylist with lazy audio /// /// This method fetches all tracks from a Qobuz album and adds them to a playlist /// with lazy audio loading. Covers are downloaded eagerly, audio lazily. /// /// # Arguments /// /// * `playlist_id` - ID of the target playlist /// * `album_id` - Qobuz album ID /// /// # Returns /// /// Number of tracks successfully added pub async fn add_album_to_playlist(&self, playlist_id: &str, album_id: &str) -> Result { use tracing::{debug, info, warn}; // 1. Get tracks from Qobuz (goes through rate limiter) let tracks = self .inner .client .get_album_tracks(album_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; if tracks.is_empty() { return Ok(0); } info!( "Adding album {} ({} tracks) to playlist {} with lazy audio", album_id, tracks.len(), playlist_id ); // 2. Add each track lazily + collect lazy PKs let mut lazy_pks = Vec::with_capacity(tracks.len()); for (i, track) in tracks.iter().enumerate() { match self.add_track_lazy(track).await { Ok((_track_id, lazy_pk)) => { debug!( "Track {}/{}: {} (lazy pk {})", i + 1, tracks.len(), track.title, &lazy_pk ); lazy_pks.push(lazy_pk); } Err(e) => { warn!("Failed to add track {} ({}): {}", i + 1, track.title, e); // Continue with other tracks } } } // 3. Batch insert into playlist (single DB transaction) let playlist_manager = pmoplaylist::PlaylistManager(); let writer = playlist_manager .get_persistent_write_handle(playlist_id.to_string()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; writer .set_role(pmoplaylist::PlaylistRole::Album) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; writer .push_lazy_batch(lazy_pks.clone()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; // 4. Enable lazy mode with lookahead of 2 tracks playlist_manager.enable_lazy_mode(playlist_id, 2); info!( "Album {} added: {}/{} tracks", album_id, lazy_pks.len(), tracks.len() ); Ok(lazy_pks.len()) } /// Load Qobuz playlist into pmoplaylist with lazy audio /// /// This method fetches all tracks from a Qobuz playlist and adds them to a pmoplaylist /// with lazy audio loading. Covers are downloaded eagerly, audio lazily. /// /// # Arguments /// /// * `playlist_id` - ID of the target pmoplaylist /// * `qobuz_playlist_id` - Qobuz playlist ID /// /// # Returns /// /// Number of tracks successfully added pub async fn add_qobuz_playlist_to_playlist( &self, playlist_id: &str, qobuz_playlist_id: &str, ) -> Result { use tracing::{debug, info, warn}; // 1. Get tracks from Qobuz playlist (goes through rate limiter) let tracks = self .inner .client .get_playlist_tracks(qobuz_playlist_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; if tracks.is_empty() { return Ok(0); } info!( "Adding Qobuz playlist {} ({} tracks) to pmoplaylist {} with lazy audio", qobuz_playlist_id, tracks.len(), playlist_id ); // 2. Add each track lazily + collect lazy PKs let mut lazy_pks = Vec::with_capacity(tracks.len()); for (i, track) in tracks.iter().enumerate() { match self.add_track_lazy(track).await { Ok((_track_id, lazy_pk)) => { debug!( "Track {}/{}: {} (lazy pk {})", i + 1, tracks.len(), track.title, &lazy_pk ); lazy_pks.push(lazy_pk); } Err(e) => { warn!("Failed to add track {} ({}): {}", i + 1, track.title, e); // Continue with other tracks } } } // 3. Batch insert into playlist (single DB transaction) let playlist_manager = pmoplaylist::PlaylistManager(); let writer = playlist_manager .get_persistent_write_handle(playlist_id.to_string()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; writer .push_lazy_batch(lazy_pks.clone()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; // 4. Enable lazy mode with lookahead of 2 tracks playlist_manager.enable_lazy_mode(playlist_id, 2); info!( "Qobuz playlist {} added: {}/{} tracks", qobuz_playlist_id, lazy_pks.len(), tracks.len() ); Ok(lazy_pks.len()) } /// Vérifie si une playlist d'album existe et est valide (non expirée ET non vide) async fn is_album_playlist_valid(&self, playlist_id: &str) -> Result { let playlist_manager = pmoplaylist::PlaylistManager(); if !playlist_manager.exists(playlist_id).await { return Ok(false); } // Vérifier l'âge match playlist_manager.get_playlist_age(playlist_id).await { Ok(Some(age)) if age < ALBUM_PLAYLIST_TTL => { // Playlist non expirée, vérifier qu'elle contient des tracks match playlist_manager.get_read_handle(playlist_id).await { Ok(reader) => { let count = reader.remaining().await.unwrap_or(0); Ok(count > 0) // Valide seulement si non vide } Err(_) => Ok(false), } } _ => Ok(false), } } /// Adapte les Items d'une playlist pour correspondre au schéma UPnP Qobuz async fn adapt_playlist_items_to_qobuz( &self, items: Vec, album_id: &str, ) -> Result> { use tracing::warn; let parent_id = format!("qobuz:album:{}", album_id); let mut adapted = Vec::with_capacity(items.len()); for mut item in items { // Extraire cache_pk depuis l'URL du resource let cache_pk = if let Some(resource) = item.resources.first() { resource.url .strip_prefix("/audio/flac/") .map(|s| s.to_string()) } else { None }; if let Some(pk) = cache_pk { // Récupérer track_id depuis metadata if let Ok(Some(track_id_value)) = self.inner.cache_manager.get_audio_metadata(&pk, "qobuz_track_id") { if let Some(track_id) = track_id_value.as_str() { item.id = format!("qobuz:track:{}", track_id); } else { warn!("qobuz_track_id not a string for {}", pk); } } else { warn!("No qobuz_track_id metadata for {}", pk); } // Convertir URL relative en URL absolue // From: /audio/flac/QOBUZ:123 // To: http://192.168.0.138:8080/audio/flac/QOBUZ:123 if let Some(resource) = item.resources.first_mut() { if resource.url.starts_with('/') { resource.url = format!("{}{}", self.inner.base_url, resource.url); } } } item.parent_id = parent_id.clone(); adapted.push(item); } Ok(adapted) } /// Récupère ou crée une playlist lazy pour un album async fn get_or_create_album_playlist_items( &self, album_id: &str, limit: usize, ) -> Result> { use tracing::{debug, info}; let playlist_id = format!("qobuz-album-{}", album_id); let playlist_manager = pmoplaylist::PlaylistManager(); // Vérifier validité (existe ET non expirée) let is_valid = self.is_album_playlist_valid(&playlist_id).await?; if is_valid { debug!("Album playlist {} found and valid", playlist_id); let reader = playlist_manager .get_read_handle(&playlist_id) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; let items = reader .to_items(limit) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; return self.adapt_playlist_items_to_qobuz(items, album_id).await; } // Playlist invalide/inexistante : (re)créer info!("Album playlist {} creating/refreshing", playlist_id); // 1. Métadonnées album let album = self .inner .client .get_album(album_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; // 2. Cache cover let cover_pk = if let Some(ref image_url) = album.image { self.inner .cache_manager .cache_cover(image_url) .await .ok() } else { None }; // 3. Créer ou récupérer playlist let writer = if playlist_manager.exists(&playlist_id).await { let writer = playlist_manager .get_persistent_write_handle(playlist_id.clone()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; writer .flush() .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; writer } else { playlist_manager .create_persistent_playlist_with_role( playlist_id.clone(), pmoplaylist::PlaylistRole::Album, ) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))? }; // 4. Métadonnées playlist writer .set_title(album.title.clone()) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; if let Some(pk) = cover_pk { writer .set_cover_pk(Some(pk)) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; } // IMPORTANT: Libérer le write lock avant d'appeler add_album_to_playlist drop(writer); // 5. Ajouter tracks (réutilise add_album_to_playlist existant) self.add_album_to_playlist(&playlist_id, album_id).await?; // 6. Récupérer items let reader = playlist_manager .get_read_handle(&playlist_id) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; let items = reader .to_items(limit) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; // 7. Adapter IDs self.adapt_playlist_items_to_qobuz(items, album_id).await } /// Increment update counter (called on catalog changes) async fn increment_update_id(&self) { let mut counter = self.inner.update_counter.write().await; *counter = counter.wrapping_add(1); let mut last = self.inner.last_change.write().await; *last = SystemTime::now(); } /// Construit le container Discover Catalog fn build_discover_catalog_container(&self) -> Container { Container { id: "qobuz:discover".to_string(), parent_id: "qobuz".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Discover Catalog".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Discover Genres fn build_discover_genres_container(&self) -> Container { Container { id: "qobuz:genres".to_string(), parent_id: "qobuz".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Discover Genres".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Favourites (My Music) fn build_favourites_container(&self) -> Container { Container { id: "qobuz:favorites".to_string(), parent_id: "qobuz".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "My Music".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Albums favoris fn build_favourite_albums_container(&self) -> Container { Container { id: "qobuz:favorites:albums".to_string(), parent_id: "qobuz:favorites".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Albums".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Tracks favoris fn build_favourite_tracks_container(&self) -> Container { Container { id: "qobuz:favorites:tracks".to_string(), parent_id: "qobuz:favorites".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Tracks".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Artists favoris fn build_favourite_artists_container(&self) -> Container { Container { id: "qobuz:favorites:artists".to_string(), parent_id: "qobuz:favorites".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Artists".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Construit le container Playlists favoris fn build_favourite_playlists_container(&self) -> Container { Container { id: "qobuz:favorites:playlists".to_string(), parent_id: "qobuz:favorites".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Playlists".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Browse Favourites - retourne 4 sous-containers async fn browse_favourites(&self) -> Result { let containers = vec![ self.build_favourite_albums_container(), self.build_favourite_tracks_container(), self.build_favourite_artists_container(), self.build_favourite_playlists_container(), ]; Ok(BrowseResult::Containers(containers)) } /// Browse Favourite Albums async fn browse_favourite_albums(&self) -> Result { let albums = self .inner .client .get_favorite_albums() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container("qobuz:favorites:albums").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Favourite Tracks async fn browse_favourite_tracks(&self) -> Result { let tracks = self .inner .client .get_favorite_tracks() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let items: Vec = tracks .into_iter() .filter_map(|track| track.to_didl_item("qobuz:favorites:tracks").ok()) .collect(); Ok(BrowseResult::Items(items)) } /// Browse Favourite Artists async fn browse_favourite_artists(&self) -> Result { let artists = self .inner .client .get_favorite_artists() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = artists .into_iter() .filter_map(|artist| { // Créer un container pour chaque artiste Some(Container { id: format!("qobuz:artist:{}", artist.id), parent_id: "qobuz:favorites:artists".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: artist.name.clone(), class: "object.container".to_string(), containers: vec![], items: vec![], }) }) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Favourite Playlists async fn browse_favourite_playlists(&self) -> Result { let playlists = self .inner .client .get_user_playlists() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = playlists .into_iter() .filter_map(|playlist| playlist.to_didl_container("qobuz:favorites:playlists").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } // ===== DISCOVER CATALOG ===== /// Browse Discover Catalog - retourne 5 containers principaux + 10 playlists par tag async fn browse_discover_catalog(&self) -> Result { use crate::models::PlaylistTag; let mut containers = vec![ self.build_discover_playlists_container(), self.build_discover_albums_ideal_container(), self.build_discover_albums_qobuzissime_container(), self.build_discover_albums_new_container(), self.build_discover_artists_container(), ]; // Ajouter les 10 tags de playlists for tag in PlaylistTag::all() { containers.push(Container { id: format!("qobuz:discover:playlists:{}", tag.api_id()), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: tag.display_name().to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], }); } Ok(BrowseResult::Containers(containers)) } fn build_discover_playlists_container(&self) -> Container { Container { id: "qobuz:discover:playlists".to_string(), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Playlists".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_discover_albums_ideal_container(&self) -> Container { Container { id: "qobuz:discover:albums:ideal".to_string(), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Albums (Ideal Discography)".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_discover_albums_qobuzissime_container(&self) -> Container { Container { id: "qobuz:discover:albums:qobuzissime".to_string(), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Albums (Qobuzissime)".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_discover_albums_new_container(&self) -> Container { Container { id: "qobuz:discover:albums:new".to_string(), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Albums (New Releases)".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_discover_artists_container(&self) -> Container { Container { id: "qobuz:discover:artists".to_string(), parent_id: "qobuz:discover".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Artists".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Browse Discover Playlists (all featured playlists) async fn browse_discover_playlists(&self) -> Result { let playlists = self .inner .client .get_featured_playlists(None, None) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = playlists .into_iter() .filter_map(|playlist| playlist.to_didl_container("qobuz:discover:playlists").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Discover Albums (Ideal Discography) async fn browse_discover_albums_ideal(&self) -> Result { let albums = self .inner .client .get_featured_albums(None, "ideal-discography") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container("qobuz:discover:albums:ideal").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Discover Albums (Qobuzissime) async fn browse_discover_albums_qobuzissime(&self) -> Result { let albums = self .inner .client .get_featured_albums(None, "qobuzissims") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container("qobuz:discover:albums:qobuzissime").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Discover Albums (New Releases) async fn browse_discover_albums_new(&self) -> Result { let albums = self .inner .client .get_featured_albums(None, "new-releases") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container("qobuz:discover:albums:new").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Discover Artists (Featured Artists) async fn browse_discover_artists(&self) -> Result { let artists = self .inner .client .get_featured_artists(None, None, None) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = artists .into_iter() .map(|artist| Container { id: format!("qobuz:artist:{}", artist.id), parent_id: "qobuz:discover:artists".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: artist.name.clone(), class: "object.container".to_string(), containers: vec![], items: vec![], }) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Discover Playlists by tag async fn browse_discover_playlists_tag(&self, tag: &str) -> Result { let playlists = self .inner .client .get_featured_playlists(None, Some(tag)) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:discover:playlists:{}", tag); let containers: Vec = playlists .into_iter() .filter_map(|playlist| playlist.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } // ===== DISCOVER GENRES ===== /// Browse Discover Genres - liste des genres async fn browse_discover_genres(&self) -> Result { let genres = self .inner .client .get_genres() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = genres .into_iter() .filter_map(|genre| { // Filtrer les genres sans ID genre.id.map(|id| Container { id: format!("qobuz:genre:{}", id), parent_id: "qobuz:genres".to_string(), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: genre.name.clone(), class: "object.container".to_string(), containers: vec![], items: vec![], }) }) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse un genre spécifique - retourne 6 sous-containers async fn browse_genre(&self, genre_id: &str) -> Result { let containers = vec![ self.build_genre_new_releases_container(genre_id), self.build_genre_ideal_discography_container(genre_id), self.build_genre_qobuzissime_container(genre_id), self.build_genre_editor_picks_container(genre_id), self.build_genre_press_awards_container(genre_id), self.build_genre_playlists_container(genre_id), ]; Ok(BrowseResult::Containers(containers)) } fn build_genre_new_releases_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:new-releases", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "New Releases".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_genre_ideal_discography_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:ideal", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Ideal Discography".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_genre_qobuzissime_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:qobuzissime", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Qobuzissime".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_genre_editor_picks_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:editor-picks", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Editor Picks".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_genre_press_awards_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:press-awards", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Press Awards".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } fn build_genre_playlists_container(&self, genre_id: &str) -> Container { Container { id: format!("qobuz:genre:{}:playlists", genre_id), parent_id: format!("qobuz:genre:{}", genre_id), restricted: Some("1".to_string()), child_count: None, searchable: Some("1".to_string()), title: "Qobuz Playlists".to_string(), class: "object.container".to_string(), containers: vec![], items: vec![], } } /// Browse Genre New Releases async fn browse_genre_new_releases(&self, genre_id: &str) -> Result { let albums = self .inner .client .get_featured_albums(Some(genre_id), "new-releases") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:new-releases", genre_id); let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Genre Ideal Discography async fn browse_genre_ideal_discography(&self, genre_id: &str) -> Result { let albums = self .inner .client .get_featured_albums(Some(genre_id), "ideal-discography") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:ideal", genre_id); let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Genre Qobuzissime async fn browse_genre_qobuzissime(&self, genre_id: &str) -> Result { let albums = self .inner .client .get_featured_albums(Some(genre_id), "qobuzissims") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:qobuzissime", genre_id); let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Genre Editor Picks async fn browse_genre_editor_picks(&self, genre_id: &str) -> Result { let albums = self .inner .client .get_featured_albums(Some(genre_id), "editor-picks") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:editor-picks", genre_id); let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Genre Press Awards async fn browse_genre_press_awards(&self, genre_id: &str) -> Result { let albums = self .inner .client .get_featured_albums(Some(genre_id), "press-awards") .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:press-awards", genre_id); let containers: Vec = albums .into_iter() .filter_map(|album| album.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Browse Genre Playlists async fn browse_genre_playlists(&self, genre_id: &str) -> Result { let playlists = self .inner .client .get_featured_playlists(Some(genre_id), None) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let parent_id = format!("qobuz:genre:{}:playlists", genre_id); let containers: Vec = playlists .into_iter() .filter_map(|playlist| playlist.to_didl_container(&parent_id).ok()) .collect(); Ok(BrowseResult::Containers(containers)) } /// Parse object_id to determine what to browse /// /// Object IDs follow these patterns: /// - "qobuz" or "0" → Root container /// - "qobuz:discover" → Discover Catalog /// - "qobuz:genres" → Discover Genres /// - "qobuz:favorites" → My Music /// - "qobuz:album:{id}" → Tracks in album /// - "qobuz:playlist:{id}" → Tracks in playlist /// - etc. fn parse_object_id(&self, object_id: &str) -> ObjectIdType { if object_id == "qobuz" || object_id == "0" { return ObjectIdType::Root; } let parts: Vec<&str> = object_id.split(':').collect(); match parts.as_slice() { // Discover Catalog ["qobuz", "discover"] => ObjectIdType::DiscoverCatalog, ["qobuz", "discover", "playlists"] => ObjectIdType::DiscoverPlaylists, ["qobuz", "discover", "albums", "ideal"] => ObjectIdType::DiscoverAlbumsIdeal, ["qobuz", "discover", "albums", "qobuzissime"] => ObjectIdType::DiscoverAlbumsQobuzissime, ["qobuz", "discover", "albums", "new"] => ObjectIdType::DiscoverAlbumsNew, ["qobuz", "discover", "artists"] => ObjectIdType::DiscoverArtists, ["qobuz", "discover", "playlists", tag] => ObjectIdType::DiscoverPlaylistsByTag(tag.to_string()), // Discover Genres ["qobuz", "genres"] => ObjectIdType::DiscoverGenres, ["qobuz", "genre", id] => ObjectIdType::GenreRoot(id.to_string()), ["qobuz", "genre", id, "new-releases"] => ObjectIdType::GenreNewReleases(id.to_string()), ["qobuz", "genre", id, "ideal"] => ObjectIdType::GenreIdealDiscography(id.to_string()), ["qobuz", "genre", id, "qobuzissime"] => ObjectIdType::GenreQobuzissime(id.to_string()), ["qobuz", "genre", id, "editor-picks"] => ObjectIdType::GenreEditorPicks(id.to_string()), ["qobuz", "genre", id, "press-awards"] => ObjectIdType::GenrePressAwards(id.to_string()), ["qobuz", "genre", id, "playlists"] => ObjectIdType::GenrePlaylists(id.to_string()), // Favourites ["qobuz", "favorites"] => ObjectIdType::Favourites, ["qobuz", "favorites", "albums"] => ObjectIdType::FavouriteAlbums, ["qobuz", "favorites", "tracks"] => ObjectIdType::FavouriteTracks, ["qobuz", "favorites", "artists"] => ObjectIdType::FavouriteArtists, ["qobuz", "favorites", "playlists"] => ObjectIdType::FavouritePlaylists, // Items (existant) ["qobuz", "album", id] => ObjectIdType::Album(id.to_string()), ["qobuz", "playlist", id] => ObjectIdType::Playlist(id.to_string()), ["qobuz", "artist", id] => ObjectIdType::Artist(id.to_string()), ["qobuz", "track", id] => ObjectIdType::Track(id.to_string()), _ => ObjectIdType::Unknown, } } } #[derive(Debug)] enum ObjectIdType { Root, // Discover Catalog DiscoverCatalog, DiscoverPlaylists, DiscoverAlbumsIdeal, DiscoverAlbumsQobuzissime, DiscoverAlbumsNew, DiscoverArtists, DiscoverPlaylistsByTag(String), // tag // Discover Genres DiscoverGenres, GenreRoot(String), // genre_id GenreNewReleases(String), // genre_id GenreIdealDiscography(String), // genre_id GenreQobuzissime(String), // genre_id GenreEditorPicks(String), // genre_id GenrePressAwards(String), // genre_id GenrePlaylists(String), // genre_id // Favourites Favourites, FavouriteAlbums, FavouriteTracks, FavouriteArtists, FavouritePlaylists, // Items (existant) Album(String), Playlist(String), Artist(String), Track(String), Unknown, } #[async_trait] impl MusicSource for QobuzSource { fn name(&self) -> &str { "Qobuz" } fn id(&self) -> &str { "qobuz" } fn default_image(&self) -> &[u8] { DEFAULT_IMAGE } async fn root_container(&self) -> Result { // Create the root container with 3 main branches Ok(Container { id: "qobuz".to_string(), parent_id: "0".to_string(), restricted: Some("1".to_string()), child_count: Some("3".to_string()), searchable: Some("1".to_string()), title: "Qobuz".to_string(), class: "object.container".to_string(), containers: vec![ self.build_discover_catalog_container(), self.build_discover_genres_container(), self.build_favourites_container(), ], items: vec![], }) } async fn browse(&self, object_id: &str) -> Result { match self.parse_object_id(object_id) { ObjectIdType::Root => { // Return the root container's children let root = self.root_container().await?; Ok(BrowseResult::Containers(root.containers)) } // Discover Catalog ObjectIdType::DiscoverCatalog => self.browse_discover_catalog().await, ObjectIdType::DiscoverPlaylists => self.browse_discover_playlists().await, ObjectIdType::DiscoverAlbumsIdeal => self.browse_discover_albums_ideal().await, ObjectIdType::DiscoverAlbumsQobuzissime => self.browse_discover_albums_qobuzissime().await, ObjectIdType::DiscoverAlbumsNew => self.browse_discover_albums_new().await, ObjectIdType::DiscoverArtists => self.browse_discover_artists().await, ObjectIdType::DiscoverPlaylistsByTag(tag) => self.browse_discover_playlists_tag(&tag).await, // Discover Genres ObjectIdType::DiscoverGenres => self.browse_discover_genres().await, ObjectIdType::GenreRoot(id) => self.browse_genre(&id).await, ObjectIdType::GenreNewReleases(id) => self.browse_genre_new_releases(&id).await, ObjectIdType::GenreIdealDiscography(id) => self.browse_genre_ideal_discography(&id).await, ObjectIdType::GenreQobuzissime(id) => self.browse_genre_qobuzissime(&id).await, ObjectIdType::GenreEditorPicks(id) => self.browse_genre_editor_picks(&id).await, ObjectIdType::GenrePressAwards(id) => self.browse_genre_press_awards(&id).await, ObjectIdType::GenrePlaylists(id) => self.browse_genre_playlists(&id).await, // Favourites ObjectIdType::Favourites => self.browse_favourites().await, ObjectIdType::FavouriteAlbums => self.browse_favourite_albums().await, ObjectIdType::FavouriteTracks => self.browse_favourite_tracks().await, ObjectIdType::FavouriteArtists => self.browse_favourite_artists().await, ObjectIdType::FavouritePlaylists => self.browse_favourite_playlists().await, // Items (existant) ObjectIdType::Album(album_id) => { let items = self .get_or_create_album_playlist_items(&album_id, usize::MAX) .await?; Ok(BrowseResult::Items(items)) } ObjectIdType::Playlist(playlist_id) => { // Get tracks in playlist let tracks = self .inner .client .get_playlist_tracks(&playlist_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let items: Vec = tracks .into_iter() .filter_map(|track| { track .to_didl_item(&format!("qobuz:playlist:{}", playlist_id)) .ok() }) .collect(); Ok(BrowseResult::Items(items)) } ObjectIdType::Artist(artist_id) => { // Get albums by artist let albums = self .inner .client .get_artist_albums(&artist_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .filter_map(|album| { album .to_didl_container(&format!("qobuz:artist:{}", artist_id)) .ok() }) .collect(); Ok(BrowseResult::Containers(containers)) } ObjectIdType::Track(_) => { // Track object_ids ne sont pas browsables, retourner une erreur Err(MusicSourceError::NotSupported( "Tracks are not browsable containers".to_string(), )) } ObjectIdType::Unknown => Err(MusicSourceError::ObjectNotFound(object_id.to_string())), } } async fn resolve_uri(&self, object_id: &str) -> Result { // Try cache manager first if let Ok(uri) = self.inner.cache_manager.resolve_uri(object_id).await { return Ok(uri); } // If not cached, extract track ID and get streaming URL from Qobuz let track_id = object_id .strip_prefix("qobuz://track/") .unwrap_or(object_id); self.inner .client .get_stream_url(track_id) .await .map_err(|e| MusicSourceError::UriResolutionError(e.to_string())) } fn supports_fifo(&self) -> bool { // Qobuz is a catalog, not a dynamic stream false } async fn append_track(&self, _track: Item) -> Result<()> { Err(MusicSourceError::FifoNotSupported) } async fn remove_oldest(&self) -> Result> { Err(MusicSourceError::FifoNotSupported) } async fn update_id(&self) -> u32 { *self.inner.update_counter.read().await } async fn last_change(&self) -> Option { Some(*self.inner.last_change.read().await) } async fn get_items(&self, offset: usize, count: usize) -> Result> { // For Qobuz, "get_items" returns favorite tracks with pagination let all_tracks = self .inner .client .get_favorite_tracks() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let items: Vec = all_tracks .into_iter() .skip(offset) .take(count) .filter_map(|track| track.to_didl_item("qobuz:favorites").ok()) .collect(); Ok(items) } async fn search(&self, query: &str) -> Result { // Search across Qobuz catalog let results = self .inner .client .search(query, None) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; // Convert albums to containers and tracks to items let containers: Vec = results .albums .into_iter() .filter_map(|album| album.to_didl_container("qobuz").ok()) .collect(); let items: Vec = results .tracks .into_iter() .filter_map(|track| track.to_didl_item("qobuz").ok()) .collect(); if !containers.is_empty() || !items.is_empty() { Ok(BrowseResult::Mixed { containers, items }) } else { Ok(BrowseResult::Items(vec![])) } } // ============= Extended Features Implementation ============= fn capabilities(&self) -> pmosource::SourceCapabilities { pmosource::SourceCapabilities { supports_fifo: false, supports_search: true, supports_favorites: true, supports_playlists: true, supports_user_content: false, supports_high_res_audio: true, max_sample_rate: Some(192_000), // Qobuz supports up to 192kHz supports_multiple_formats: true, supports_advanced_search: false, // TODO: Qobuz API supports it, not yet implemented supports_pagination: true, } } async fn get_available_formats(&self, object_id: &str) -> Result> { use pmosource::AudioFormat; // Extract track ID from object_id let track_id = if let Some(id) = object_id.strip_prefix("qobuz://track/") { id } else { object_id }; // Get track details from Qobuz let track = self .inner .client .get_track(track_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; // Qobuz provides multiple formats based on subscription let mut formats = vec![]; // MP3 320 (format_id 5) - available to all formats.push(AudioFormat { format_id: "mp3-320".to_string(), mime_type: "audio/mpeg".to_string(), sample_rate: Some(44100), bit_depth: None, bitrate: Some(320), channels: Some(2), }); // FLAC 16/44.1 (format_id 6) - CD quality formats.push(AudioFormat { format_id: "flac-16-44".to_string(), mime_type: "audio/flac".to_string(), sample_rate: Some(44100), bit_depth: Some(16), bitrate: None, channels: Some(2), }); // Hi-Res formats (if available for this track) if let Some(sample_rate) = track.sample_rate { if sample_rate > 44100 { // FLAC 24-bit Hi-Res let bit_depth = track.bit_depth.map(|d| d as u8).or(Some(24)); formats.push(AudioFormat { format_id: format!("flac-{}-{}", bit_depth.unwrap_or(24), sample_rate / 1000), mime_type: "audio/flac".to_string(), sample_rate: Some(sample_rate), bit_depth, bitrate: None, channels: track.channels, }); } } Ok(formats) } async fn get_cache_status(&self, object_id: &str) -> Result { self.inner.cache_manager.get_cache_status(object_id).await } async fn cache_item(&self, object_id: &str) -> Result { // Extract track ID let track_id = object_id .strip_prefix("qobuz://track/") .unwrap_or(object_id); // Get track details from Qobuz let track = self .inner .client .get_track(track_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; // Add track to cache (via manager) let cached_id = self.add_track(&track).await?; // Return the cache status self.get_cache_status(&cached_id).await } async fn add_favorite(&self, object_id: &str) -> Result<()> { // Parse object_id to determine type let parts: Vec<&str> = object_id.split(':').collect(); match parts.as_slice() { ["qobuz", "album", id] | ["qobuz://album", id] => { self.inner .client .add_favorite_album(id) .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; } ["qobuz", "track", id] | ["qobuz://track", id] => { self.inner .client .add_favorite_track(id) .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; } _ => { return Err(MusicSourceError::NotSupported( "Favorites only supported for albums and tracks".to_string(), )); } } self.increment_update_id().await; Ok(()) } async fn remove_favorite(&self, object_id: &str) -> Result<()> { // Parse object_id to determine type let parts: Vec<&str> = object_id.split(':').collect(); match parts.as_slice() { ["qobuz", "album", id] | ["qobuz://album", id] => { self.inner .client .remove_favorite_album(id) .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; } ["qobuz", "track", id] | ["qobuz://track", id] => { self.inner .client .remove_favorite_track(id) .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; } _ => { return Err(MusicSourceError::NotSupported( "Favorites only supported for albums and tracks".to_string(), )); } } self.increment_update_id().await; Ok(()) } async fn is_favorite(&self, object_id: &str) -> Result { // Parse object_id to determine type let parts: Vec<&str> = object_id.split(':').collect(); match parts.as_slice() { ["qobuz", "album", id] | ["qobuz://album", id] => { let favorites = self .inner .client .get_favorite_albums() .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; Ok(favorites.iter().any(|album| album.id == *id)) } ["qobuz", "track", id] | ["qobuz://track", id] => { let favorites = self .inner .client .get_favorite_tracks() .await .map_err(|e| MusicSourceError::FavoritesError(e.to_string()))?; Ok(favorites.iter().any(|track| track.id == *id)) } _ => Err(MusicSourceError::NotSupported( "Favorites only supported for albums and tracks".to_string(), )), } } async fn get_user_playlists(&self) -> Result> { let playlists = self .inner .client .get_user_playlists() .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; let containers: Vec = playlists .into_iter() .filter_map(|playlist| playlist.to_didl_container("qobuz").ok()) .collect(); Ok(containers) } async fn add_to_playlist(&self, playlist_id: &str, item_id: &str) -> Result<()> { // Extract track ID from item_id let track_id = if let Some(id) = item_id.strip_prefix("qobuz://track/") { id } else if let Some(id) = item_id.strip_prefix("qobuz:track:") { id } else { item_id }; self.inner .client .add_to_playlist(playlist_id, track_id) .await .map_err(|e| MusicSourceError::PlaylistError(e.to_string()))?; self.increment_update_id().await; Ok(()) } async fn get_item_count(&self, object_id: &str) -> Result { match self.parse_object_id(object_id) { ObjectIdType::Album(album_id) => { let album = self .inner .client .get_album(&album_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; Ok(album.tracks_count.unwrap_or(0) as usize) } ObjectIdType::Playlist(playlist_id) => { let playlist = self .inner .client .get_playlist(&playlist_id) .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; Ok(playlist.tracks_count.unwrap_or(0) as usize) } _ => { // Fall back to default implementation let result = self.browse(object_id).await?; Ok(result.count()) } } } async fn browse_paginated( &self, object_id: &str, offset: usize, limit: usize, ) -> Result { match self.parse_object_id(object_id) { ObjectIdType::Album(album_id) => { let all_items = self .get_or_create_album_playlist_items(&album_id, usize::MAX) .await?; let items: Vec = all_items .into_iter() .skip(offset) .take(limit) .collect(); Ok(BrowseResult::Items(items)) } ObjectIdType::FavouriteAlbums => { let albums = self .inner .client .get_favorite_albums() .await .map_err(|e| MusicSourceError::BrowseError(e.to_string()))?; let containers: Vec = albums .into_iter() .skip(offset) .take(limit) .filter_map(|album| album.to_didl_container("qobuz:favorites:albums").ok()) .collect(); Ok(BrowseResult::Containers(containers)) } _ => { // Fall back to default implementation self.browse(object_id).await } } } async fn statistics(&self) -> Result { let mut stats = pmosource::SourceStatistics::default(); // Try to get favorite counts if let Ok(albums) = self.inner.client.get_favorite_albums().await { stats.total_containers = Some(albums.len()); } if let Ok(tracks) = self.inner.client.get_favorite_tracks().await { stats.total_items = Some(tracks.len()); } // Get cache statistics from manager let cache_stats = self.inner.cache_manager.statistics().await; stats.cached_items = Some(cache_stats.cached_tracks); Ok(stats) } } #[cfg(test)] mod tests { use super::*; #[test] fn test_default_image_present() { assert!(DEFAULT_IMAGE.len() > 0, "Default image should not be empty"); // Check WebP magic bytes (RIFF...WEBP) assert!( DEFAULT_IMAGE.len() >= 12, "Image too small to be valid WebP" ); assert_eq!(&DEFAULT_IMAGE[0..4], b"RIFF", "Missing RIFF header"); assert_eq!(&DEFAULT_IMAGE[8..12], b"WEBP", "Missing WEBP signature"); } // Note: We can't easily test parse_object_id without creating a real client // which requires authentication. The parsing logic is simple enough that // it's covered by integration tests. }