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

@@ -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
}
}