Spaces:
Paused
Paused
| use std::sync::Arc; | |
| use crate::error::AppError; | |
| use crate::services::retry::{self, TRAKT_RETRY}; | |
| use serde::Deserialize; | |
| use zeroize::Zeroizing; | |
| pub struct TraktClient { | |
| client_id: Arc<Zeroizing<String>>, | |
| http: reqwest::Client, | |
| } | |
| pub struct TraktRatingsResponse { | |
| pub rating: f64, | |
| pub votes: u64, | |
| } | |
| impl TraktClient { | |
| pub fn new(client_id: String, http: reqwest::Client) -> Self { | |
| Self { | |
| client_id: Arc::new(Zeroizing::new(client_id)), | |
| http, | |
| } | |
| } | |
| /// Build a GET request with the required Trakt API headers. | |
| /// | |
| /// Trakt's API requires `Content-Type: application/json` even on GET | |
| /// requests (no body). This is a quirk of their API spec. | |
| fn request(&self, url: &str) -> reqwest::RequestBuilder { | |
| self.http | |
| .get(url) | |
| .header("Content-Type", "application/json") | |
| .header("trakt-api-version", "2") | |
| .header("trakt-api-key", self.client_id.as_str()) | |
| } | |
| /// Fetch a ratings endpoint. Returns `Ok(None)` when Trakt responds 404 | |
| /// (the title/episode simply isn't catalogued), which is an expected, | |
| /// non-error outcome — only genuine failures surface as `Err`. | |
| async fn get_rating(&self, url: &str) -> Result<Option<TraktRatingsResponse>, AppError> { | |
| let resp = retry::send_with_retry(&TRAKT_RETRY, || self.request(url).send()).await?; | |
| if resp.status() == reqwest::StatusCode::NOT_FOUND { | |
| return Ok(None); | |
| } | |
| let resp = resp.error_for_status()?; | |
| Ok(Some(resp.json().await?)) | |
| } | |
| pub async fn get_movie_rating( | |
| &self, | |
| imdb_id: &str, | |
| ) -> Result<Option<TraktRatingsResponse>, AppError> { | |
| self.get_rating(&format!("https://api.trakt.tv/movies/{imdb_id}/ratings")) | |
| .await | |
| } | |
| pub async fn get_show_rating( | |
| &self, | |
| imdb_id: &str, | |
| ) -> Result<Option<TraktRatingsResponse>, AppError> { | |
| self.get_rating(&format!("https://api.trakt.tv/shows/{imdb_id}/ratings")) | |
| .await | |
| } | |
| pub async fn get_episode_rating( | |
| &self, | |
| show_id: &str, | |
| season: u32, | |
| episode: u32, | |
| ) -> Result<Option<TraktRatingsResponse>, AppError> { | |
| self.get_rating(&format!( | |
| "https://api.trakt.tv/shows/{show_id}/seasons/{season}/episodes/{episode}/ratings" | |
| )) | |
| .await | |
| } | |
| } | |