eric/pmomusic
1
0
Fork 0
Code Issues 5 Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Refactoring de l'API rest des musicsources

Browse Source
This commit is contained in:
Eric Coissac
2025-10-17 12:45:18 +02:00
parent 1f263c2295
commit 823617c5f3
14 changed files with 908 additions and 272 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pmomediaserver/Cargo.toml
View File

@@ -6,7 +6,7 @@ edition = "2024"
[dependencies]
pmoupnp = { path = "../pmoupnp" }
pmodidl = { path = "../pmodidl" }
pmosource = { path = "../pmosource" }
pmosource = { path = "../pmosource", features = ["server"] }
pmoserver = { path = "../pmoserver" }
once_cell = "1.20"

27
pmomediaserver/src/content_handler.rs
View File

@@ -10,7 +10,7 @@
//! - **Search** : Recherche dans les sources qui le supportent
//! - **Update ID** : Suivi des changements pour les notifications UPnP
use crate::server_ext::get_source_registry;
use pmosource::api::{list_all_sources, get_source as get_source_from_registry};
use pmodidl::{Container, DIDLLite};
use pmosource::{BrowseResult, MusicSource};
use std::sync::Arc;
@@ -102,10 +102,8 @@ impl ContentHandler {
Ok((didl, 1, 1, 0))
} else {
// Essayer de trouver l'objet dans les sources
let registry = get_source_registry().await;
// Vérifier si c'est un container racine d'une source
if let Some(source) = registry.get(object_id).await {
if let Some(source) = get_source_from_registry(object_id).await {
let container = source
.root_container()
.await
@@ -116,7 +114,7 @@ impl ContentHandler {
}
// Sinon, chercher dans les sources
for source in registry.list_all().await {
for source in list_all_sources().await {
if let Ok(result) = source.browse(object_id).await {
// L'objet a été trouvé, retourner ses métadonnées
match result {
@@ -165,17 +163,15 @@ impl ContentHandler {
return self.browse_root(starting_index, requested_count).await;
}
let registry = get_source_registry().await;
// Vérifier si c'est le container racine d'une source
if let Some(source) = registry.get(object_id).await {
if let Some(source) = get_source_from_registry(object_id).await {
return self
.browse_source_root(source, starting_index, requested_count)
.await;
}
// Sinon, chercher dans les sources
for source in registry.list_all().await {
for source in list_all_sources().await {
if let Ok(result) = source.browse(object_id).await {
return self
.browse_result_to_didl(result, source, starting_index, requested_count)
@@ -192,8 +188,7 @@ impl ContentHandler {
starting_index: u32,
requested_count: u32,
) -> Result<(String, u32, u32, u32), String> {
let registry = get_source_registry().await;
let sources = registry.list_all().await;
let sources = list_all_sources().await;
let mut containers = Vec::new();
for source in sources.iter() {
@@ -299,8 +294,8 @@ impl ContentHandler {
/// Construit le container racine du MediaServer
async fn build_root_container(&self) -> Container {
let registry = get_source_registry().await;
let child_count = registry.count().await;
let sources = list_all_sources().await;
let child_count = sources.len();
Container {
id: "0".to_string(),
@@ -335,12 +330,11 @@ impl ContentHandler {
"ContentDirectory::Search"
);
let registry = get_source_registry().await;
let mut all_containers = Vec::new();
let mut all_items = Vec::new();
// Rechercher dans toutes les sources qui supportent la recherche
for source in registry.list_all().await {
for source in list_all_sources().await {
if source.capabilities().supports_search {
if let Ok(result) = source.search(search_criteria).await {
match result {
@@ -375,8 +369,7 @@ impl ContentHandler {
/// Retourne le system update ID global
pub async fn get_system_update_id(&self) -> u32 {
let registry = get_source_registry().await;
let sources = registry.list_all().await;
let sources = list_all_sources().await;
// Combiner les update IDs de toutes les sources
let mut combined_id = 0u32;

5
pmomediaserver/src/lib.rs
View File

@@ -76,11 +76,14 @@ pub mod sources_api;
pub use device::MEDIA_SERVER;
pub use source_registry::SourceRegistry;
pub use server_ext::{MediaServerExt, get_source_registry};
pub use server_ext::{MediaServerExt, get_source_registry, MusicSourceExt};
pub use content_handler::ContentHandler;
pub use sources::{SourcesExt, SourceInitError};
// L'API des sources est maintenant dans pmosource
// Pour des raisons de compatibilité, on réexporte ici
#[cfg(feature = "api")]
#[deprecated(since = "0.2.0", note = "Use pmosource::api directly")]
pub use sources_api::{sources_api_router, SourcesApiDoc};
// Re-export sources when features are enabled

264
pmomediaserver/src/server_ext.rs
View File

@@ -2,277 +2,77 @@
//!
//! Ce module fournit un trait d'extension pour `pmoserver::Server` permettant
//! d'enregistrer facilement des sources musicales et de configurer le MediaServer.
//!
//! **Note**: Ce module réexporte `MusicSourceExt` de `pmosource` et ajoute des
//! méthodes spécifiques au MediaServer UPnP.
use crate::source_registry::SourceRegistry;
use async_trait::async_trait;
use pmosource::MusicSource;
use pmoserver::Server;
use std::sync::Arc;
use tokio::sync::OnceCell;
/// Extension pour le registre de sources au niveau global
///
/// Ce registre est partagé par toutes les instances du serveur et permet
/// d'accéder aux sources musicales depuis n'importe où dans l'application.
static GLOBAL_REGISTRY: OnceCell<SourceRegistry> = OnceCell::const_new();
// Réexporter le trait de base de pmosource
pub use pmosource::MusicSourceExt;
/// Initialise le registre global
///
/// Cette fonction est appelée automatiquement lors de la première utilisation.
async fn init_global_registry() -> &'static SourceRegistry {
GLOBAL_REGISTRY
.get_or_init(|| async { SourceRegistry::new() })
.await
}
/// Récupère le registre global de sources
/// Récupère le registre global de sources (délègue à pmosource)
///
/// # Examples
///
/// ```ignore
/// use pmomediaserver::server_ext::get_source_registry;
///
/// let registry = get_source_registry().await;
/// if let Some(source) = registry.get("qobuz").await {
/// // Utiliser la source
/// }
/// let sources = pmosource::api::list_all_sources().await;
/// ```
pub async fn get_source_registry() -> &'static SourceRegistry {
init_global_registry().await
#[deprecated(since = "0.2.0", note = "Use pmosource::api::list_all_sources() directly")]
pub async fn get_source_registry() -> Vec<Arc<dyn MusicSource>> {
pmosource::api::list_all_sources().await
}
/// Trait d'extension pour le serveur permettant l'enregistrement de sources musicales
/// Trait d'extension pour le serveur MediaServer UPnP
///
/// Ce trait ajoute des méthodes pratiques à `Server` pour enregistrer des sources
/// musicales et les rendre disponibles via le MediaServer.
/// Ce trait ajoute des méthodes spécifiques au MediaServer UPnP.
/// Pour l'enregistrement de sources, utilisez le trait `MusicSourceExt` de `pmosource`.
///
/// # Examples
/// **Note**: Ce trait est maintenant deprecated. Utilisez directement `MusicSourceExt`
/// de `pmosource` pour l'enregistrement et la gestion des sources.
///
/// # Migration
///
/// Ancien code :
/// ```ignore
/// use pmomediaserver::server_ext::MediaServerExt;
/// use pmoserver::ServerBuilder;
/// server.register_music_source(source).await;
/// ```
///
/// let mut server = ServerBuilder::new_configured().build();
///
/// // Enregistrer une source
/// let qobuz = Arc::new(QobuzSource::new());
/// server.register_music_source(qobuz).await;
///
/// // Lister toutes les sources
/// let sources = server.list_music_sources().await;
/// Nouveau code :
/// ```ignore
/// use pmosource::MusicSourceExt;
/// server.register_music_source(source).await;
/// ```
#[async_trait]
pub trait MediaServerExt {
/// Enregistre une source musicale dans le MediaServer
///
/// La source devient immédiatement disponible via le service ContentDirectory
/// et peut être parcourue par les clients UPnP.
///
/// # Arguments
///
/// * `source` - La source musicale à enregistrer (Arc<dyn MusicSource>)
///
/// # Examples
///
/// ```ignore
/// let qobuz = Arc::new(QobuzSource::new(credentials));
/// server.register_music_source(qobuz).await;
/// ```
async fn register_music_source(&mut self, source: Arc<dyn MusicSource>);
/// Récupère une source musicale par son ID
///
/// # Arguments
///
/// * `id` - L'ID unique de la source
///
/// # Returns
///
/// Un `Arc` vers la source si elle existe, ou `None`.
///
/// # Examples
///
/// ```ignore
/// if let Some(source) = server.get_music_source("qobuz").await {
/// println!("Found: {}", source.name());
/// }
/// ```
async fn get_music_source(&self, id: &str) -> Option<Arc<dyn MusicSource>>;
/// Liste toutes les sources musicales enregistrées
///
/// # Returns
///
/// Un vecteur contenant toutes les sources enregistrées.
///
/// # Examples
///
/// ```ignore
/// let sources = server.list_music_sources().await;
/// for source in sources {
/// println!("- {} ({})", source.name(), source.id());
/// }
/// ```
async fn list_music_sources(&self) -> Vec<Arc<dyn MusicSource>>;
/// Compte le nombre de sources musicales enregistrées
///
/// # Returns
///
/// Le nombre total de sources.
///
/// # Examples
///
/// ```ignore
/// let count = server.count_music_sources().await;
/// println!("Total sources: {}", count);
/// ```
async fn count_music_sources(&self) -> usize;
/// Supprime une source musicale du registre
///
/// # Arguments
///
/// * `id` - L'ID de la source à supprimer
///
/// # Returns
///
/// `true` si la source a été supprimée, `false` si elle n'existait pas.
///
/// # Examples
///
/// ```ignore
/// if server.remove_music_source("old-radio").await {
/// println!("Source removed");
/// }
/// ```
async fn remove_music_source(&mut self, id: &str) -> bool;
async fn count_music_sources(&self) -> usize {
pmosource::api::list_all_sources().await.len()
}
}
#[async_trait]
impl MediaServerExt for Server {
async fn register_music_source(&mut self, source: Arc<dyn MusicSource>) {
let registry = get_source_registry().await;
tracing::info!(
source_id = %source.id(),
source_name = %source.name(),
"Registering music source to MediaServer"
);
registry.register(source).await;
}
async fn get_music_source(&self, id: &str) -> Option<Arc<dyn MusicSource>> {
let registry = get_source_registry().await;
registry.get(id).await
}
async fn list_music_sources(&self) -> Vec<Arc<dyn MusicSource>> {
let registry = get_source_registry().await;
registry.list_all().await
}
async fn count_music_sources(&self) -> usize {
let registry = get_source_registry().await;
registry.count().await
}
async fn remove_music_source(&mut self, id: &str) -> bool {
let registry = get_source_registry().await;
registry.remove(id).await
}
// Implementation par défaut fournie dans le trait
}
#[cfg(test)]
mod tests {
use super::*;
use pmosource::{MusicSource, Result, BrowseResult};
use pmodidl::{Container, Item};
use std::time::SystemTime;
#[derive(Debug)]
struct DummySource {
id: String,
name: String,
}
impl DummySource {
fn new(id: &str, name: &str) -> Self {
Self {
id: id.to_string(),
name: name.to_string(),
}
}
}
#[async_trait::async_trait]
impl MusicSource for DummySource {
fn name(&self) -> &str {
&self.name
}
fn id(&self) -> &str {
&self.id
}
fn default_image(&self) -> &[u8] {
&[]
}
async fn root_container(&self) -> Result<Container> {
Ok(Container {
id: self.id.clone(),
parent_id: "0".to_string(),
restricted: Some("1".to_string()),
child_count: Some("0".to_string()),
title: self.name.clone(),
class: "object.container".to_string(),
containers: vec![],
items: vec![],
})
}
async fn browse(&self, _object_id: &str) -> Result<BrowseResult> {
Ok(BrowseResult::Items(vec![]))
}
async fn resolve_uri(&self, object_id: &str) -> Result<String> {
Ok(format!("http://example.com/{}", object_id))
}
fn supports_fifo(&self) -> bool {
false
}
async fn append_track(&self, _track: Item) -> Result<()> {
Err(pmosource::MusicSourceError::FifoNotSupported)
}
async fn remove_oldest(&self) -> Result<Option<Item>> {
Err(pmosource::MusicSourceError::FifoNotSupported)
}
async fn update_id(&self) -> u32 {
0
}
async fn last_change(&self) -> Option<SystemTime> {
None
}
async fn get_items(&self, _offset: usize, _count: usize) -> Result<Vec<Item>> {
Ok(vec![])
}
}
#[tokio::test]
async fn test_global_registry_singleton() {
// Vérifier que le registre global est bien un singleton
let registry1 = get_source_registry().await;
let registry2 = get_source_registry().await;
// Les deux références devraient pointer vers le même registre
assert!(std::ptr::eq(registry1, registry2));
#[test]
fn test_trait_exists() {
// Ce test vérifie simplement que le module compile
// Les tests fonctionnels sont maintenant dans pmosource
}
}

2
pmomediaserver/src/sources.rs
View File

@@ -3,7 +3,7 @@
//! Ce module fournit des helpers pour créer et enregistrer facilement des sources
//! musicales préconfigurées à partir de la configuration système.
use crate::server_ext::MediaServerExt;
use pmosource::MusicSourceExt;
use pmoserver::Server;
use std::sync::Arc;

22
pmomediaserver/src/sources_api.rs
View File

@@ -12,7 +12,8 @@
//! - `POST /sources/qobuz` - Enregistrer Qobuz (feature "qobuz")
//! - `DELETE /sources/:id` - Désenregistrer une source
use crate::server_ext::get_source_registry;
// Utiliser les fonctions du registre de pmosource
use pmosource::api::{list_all_sources, get_source as get_source_from_registry, register_source, unregister_source as unregister_source_from_registry};
use axum::{
extract::Path,
http::StatusCode,
@@ -102,8 +103,7 @@ pub struct ErrorResponse {
tag = "sources"
)]
async fn list_sources() -> impl IntoResponse {
let registry = get_source_registry().await;
let sources = registry.list_all().await;
let sources = list_all_sources().await;
let source_infos: Vec<SourceInfo> = sources
.iter()
@@ -147,9 +147,7 @@ async fn list_sources() -> impl IntoResponse {
tag = "sources"
)]
async fn get_source(Path(id): Path<String>) -> impl IntoResponse {
let registry = get_source_registry().await;
match registry.get(&id).await {
match get_source_from_registry(&id).await {
Some(source) => {
let caps = source.capabilities();
let info = SourceInfo {
@@ -192,8 +190,6 @@ async fn get_source(Path(id): Path<String>) -> impl IntoResponse {
async fn register_qobuz(Json(creds): Json<QobuzCredentials>) -> impl IntoResponse {
use pmoqobuz::{QobuzClient, QobuzSource};
let registry = get_source_registry().await;
// Créer le client selon les credentials fournis
let client_result = if let (Some(username), Some(password)) = (creds.username, creds.password) {
QobuzClient::new(&username, &password).await
@@ -223,7 +219,7 @@ async fn register_qobuz(Json(creds): Json<QobuzCredentials>) -> impl IntoRespons
let source = Arc::new(QobuzSource::new(client, &base_url));
let source_id = source.as_ref().id().to_string();
registry.register(source).await;
register_source(source).await;
(
StatusCode::CREATED,
@@ -252,8 +248,6 @@ async fn register_qobuz(Json(creds): Json<QobuzCredentials>) -> impl IntoRespons
async fn register_paradise(Json(params): Json<ParadiseParams>) -> impl IntoResponse {
use pmoparadise::{RadioParadiseClient, RadioParadiseSource};
let registry = get_source_registry().await;
// Créer le client (Radio Paradise ne nécessite pas d'auth)
let client = match RadioParadiseClient::new().await {
Ok(c) => c,
@@ -282,7 +276,7 @@ async fn register_paradise(Json(params): Json<ParadiseParams>) -> impl IntoRespo
let source_id = source.as_ref().id().to_string();
registry.register(source).await;
register_source(source).await;
(
StatusCode::CREATED,
@@ -310,9 +304,7 @@ async fn register_paradise(Json(params): Json<ParadiseParams>) -> impl IntoRespo
tag = "sources"
)]
async fn unregister_source(Path(id): Path<String>) -> impl IntoResponse {
let registry = get_source_registry().await;
if registry.remove(&id).await {
if unregister_source_from_registry(&id).await {
(
StatusCode::OK,
Json(serde_json::json!({