various: simplify code; work on security and performance

Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I9a5114addcab5fbff430ab2b919b83466a6a6964

crates/pinakes-core/src/enrichment/mod.rs

@@ -0,0 +1,66 @@
+//! Metadata enrichment from external sources.
+
+pub mod lastfm;
+pub mod musicbrainz;
+pub mod tmdb;
+
+use chrono::{DateTime, Utc};
+use serde::{Deserialize, Serialize};
+use uuid::Uuid;
+
+use crate::error::Result;
+use crate::model::{MediaId, MediaItem};
+
+/// Externally-sourced metadata for a media item.
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ExternalMetadata {
+    pub id: Uuid,
+    pub media_id: MediaId,
+    pub source: EnrichmentSourceType,
+    pub external_id: Option<String>,
+    pub metadata_json: String,
+    pub confidence: f64,
+    pub last_updated: DateTime<Utc>,
+}
+
+/// Supported enrichment data sources.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
+pub enum EnrichmentSourceType {
+    #[serde(rename = "musicbrainz")]
+    MusicBrainz,
+    #[serde(rename = "tmdb")]
+    Tmdb,
+    #[serde(rename = "lastfm")]
+    LastFm,
+}
+
+impl std::fmt::Display for EnrichmentSourceType {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        let s = match self {
+            Self::MusicBrainz => "musicbrainz",
+            Self::Tmdb => "tmdb",
+            Self::LastFm => "lastfm",
+        };
+        write!(f, "{s}")
+    }
+}
+
+impl std::str::FromStr for EnrichmentSourceType {
+    type Err = String;
+
+    fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
+        match s {
+            "musicbrainz" => Ok(Self::MusicBrainz),
+            "tmdb" => Ok(Self::Tmdb),
+            "lastfm" => Ok(Self::LastFm),
+            _ => Err(format!("unknown enrichment source: {s}")),
+        }
+    }
+}
+
+/// Trait for metadata enrichment providers.
+#[async_trait::async_trait]
+pub trait MetadataEnricher: Send + Sync {
+    fn source(&self) -> EnrichmentSourceType;
+    async fn enrich(&self, item: &MediaItem) -> Result<Option<ExternalMetadata>>;
+}