feat: add TMDB metadata provider, prefer over OMDB when TMDB_API_KEY is set

This commit is contained in:
2026-05-10 02:30:53 +02:00
parent ebf74a59fd
commit 9c11ac2bcc
3 changed files with 193 additions and 3 deletions

View File

@@ -7,6 +7,7 @@ use domain::{
}; };
mod omdb; mod omdb;
mod tmdb;
pub(crate) struct ProviderMovie { pub(crate) struct ProviderMovie {
pub imdb_id: ExternalMetadataId, pub imdb_id: ExternalMetadataId,
@@ -31,6 +32,12 @@ impl MetadataClientImpl {
provider: Box::new(omdb::OmdbProvider::new(api_key)), provider: Box::new(omdb::OmdbProvider::new(api_key)),
} }
} }
pub fn new_tmdb(api_key: String) -> Self {
Self {
provider: Box::new(tmdb::TmdbProvider::new(api_key)),
}
}
} }
#[async_trait] #[async_trait]

View File

@@ -0,0 +1,179 @@
use async_trait::async_trait;
use domain::{
errors::DomainError,
ports::MetadataSearchCriteria,
value_objects::{ExternalMetadataId, MovieTitle, PosterUrl, ReleaseYear},
};
use serde::Deserialize;
use crate::{MetadataProvider, ProviderMovie};
pub(crate) struct TmdbProvider {
client: reqwest::Client,
api_key: String,
}
impl TmdbProvider {
pub(crate) fn new(api_key: String) -> Self {
Self {
client: reqwest::Client::new(),
api_key,
}
}
fn base(&self, path: &str) -> String {
format!("https://api.themoviedb.org/3{}", path)
}
fn poster_url(&self, path: &str) -> Option<PosterUrl> {
if path.is_empty() || path == "null" {
return None;
}
PosterUrl::new(format!("https://image.tmdb.org/t/p/w500{}", path)).ok()
}
async fn get<T: for<'de> Deserialize<'de>>(
&self,
url: &str,
extra: &[(&str, &str)],
) -> Result<T, DomainError> {
let mut req = self
.client
.get(url)
.query(&[("api_key", self.api_key.as_str())]);
for (k, v) in extra {
req = req.query(&[(k, v)]);
}
req.send()
.await
.map_err(|e| DomainError::InfrastructureError(e.to_string()))?
.error_for_status()
.map_err(|e| DomainError::InfrastructureError(e.to_string()))?
.json::<T>()
.await
.map_err(|e| DomainError::InfrastructureError(e.to_string()))
}
async fn fetch_details(&self, tmdb_id: u64) -> Result<ProviderMovie, DomainError> {
#[derive(Deserialize)]
struct CrewMember {
job: String,
name: String,
}
#[derive(Deserialize)]
struct Credits {
crew: Vec<CrewMember>,
}
#[derive(Deserialize)]
struct Details {
imdb_id: Option<String>,
title: String,
release_date: String, // "YYYY-MM-DD"
poster_path: Option<String>,
credits: Credits,
}
let url = self.base(&format!("/movie/{}", tmdb_id));
let d: Details = self
.get(&url, &[("append_to_response", "credits")])
.await?;
let year: u16 = d
.release_date
.split('-')
.next()
.and_then(|y| y.parse().ok())
.ok_or_else(|| {
DomainError::InfrastructureError(format!(
"Unparseable release_date: {}",
d.release_date
))
})?;
// Prefer IMDB ID; fall back to "tmdb:{id}" so the record is still usable.
let raw_id = d
.imdb_id
.filter(|s| !s.is_empty())
.unwrap_or_else(|| format!("tmdb:{}", tmdb_id));
let imdb_id = ExternalMetadataId::new(raw_id)
.map_err(|e| DomainError::InfrastructureError(e.to_string()))?;
let title =
MovieTitle::new(d.title).map_err(|e| DomainError::InfrastructureError(e.to_string()))?;
let release_year =
ReleaseYear::new(year).map_err(|e| DomainError::InfrastructureError(e.to_string()))?;
let director = d
.credits
.crew
.into_iter()
.find(|c| c.job == "Director")
.map(|c| c.name);
let poster_url = d
.poster_path
.as_deref()
.and_then(|p| self.poster_url(p));
Ok(ProviderMovie {
imdb_id,
title,
release_year,
director,
poster_url,
})
}
}
#[async_trait]
impl MetadataProvider for TmdbProvider {
async fn fetch(&self, criteria: &MetadataSearchCriteria) -> Result<ProviderMovie, DomainError> {
let tmdb_id: u64 = match criteria {
MetadataSearchCriteria::ImdbId(id) => {
#[derive(Deserialize)]
struct FindResult {
id: u64,
}
#[derive(Deserialize)]
struct FindResponse {
movie_results: Vec<FindResult>,
}
let url = self.base(&format!("/find/{}", id.value()));
let resp: FindResponse =
self.get(&url, &[("external_source", "imdb_id")]).await?;
resp.movie_results
.into_iter()
.next()
.ok_or_else(|| DomainError::NotFound(format!("TMDB: no movie for {}", id.value())))?
.id
}
MetadataSearchCriteria::Title { title, year } => {
#[derive(Deserialize)]
struct SearchResult {
id: u64,
}
#[derive(Deserialize)]
struct SearchResponse {
results: Vec<SearchResult>,
}
let url = self.base("/search/movie");
let mut extra = vec![("query", title.value())];
let year_str;
if let Some(y) = year {
year_str = y.value().to_string();
extra.push(("year", year_str.as_str()));
}
let resp: SearchResponse = self.get(&url, &extra).await?;
resp.results
.into_iter()
.next()
.ok_or_else(|| {
DomainError::NotFound(format!("TMDB: no results for '{}'", title.value()))
})?
.id
}
};
self.fetch_details(tmdb_id).await
}
}

View File

@@ -71,13 +71,17 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
let auth_config = AuthConfig::from_env()?; let auth_config = AuthConfig::from_env()?;
let storage_config = StorageConfig::from_env()?; let storage_config = StorageConfig::from_env()?;
let app_config = AppConfig::from_env(); let app_config = AppConfig::from_env();
let omdb_api_key = std::env::var("OMDB_API_KEY").context("OMDB_API_KEY must be set")?;
let database_url = std::env::var("DATABASE_URL").context("DATABASE_URL must be set")?; let database_url = std::env::var("DATABASE_URL").context("DATABASE_URL must be set")?;
let backend = std::env::var("DATABASE_BACKEND").unwrap_or_else(|_| "sqlite".to_string()); let backend = std::env::var("DATABASE_BACKEND").unwrap_or_else(|_| "sqlite".to_string());
let metadata_client: Arc<dyn MetadataClient> = let metadata_client: Arc<dyn MetadataClient> =
Arc::new(MetadataClientImpl::new_omdb(omdb_api_key)); if let Ok(tmdb_key) = std::env::var("TMDB_API_KEY") {
Arc::new(MetadataClientImpl::new_tmdb(tmdb_key))
} else {
let omdb_key = std::env::var("OMDB_API_KEY")
.context("Either TMDB_API_KEY or OMDB_API_KEY must be set")?;
Arc::new(MetadataClientImpl::new_omdb(omdb_key))
};
let poster_fetcher: Arc<dyn PosterFetcherClient> = let poster_fetcher: Arc<dyn PosterFetcherClient> =
Arc::new(ReqwestPosterFetcher::new(PosterFetcherConfig::from_env())?); Arc::new(ReqwestPosterFetcher::new(PosterFetcherConfig::from_env())?);
let poster_storage: Arc<dyn PosterStorage> = let poster_storage: Arc<dyn PosterStorage> =