movies-diary/crates/adapters/poster-sync/src/lib.rs

use std::{sync::Arc, time::Duration};

use async_trait::async_trait;
use domain::{
    errors::DomainError,
    events::DomainEvent,
    ports::{
        EventHandler, EventPublisher, MetadataClient, MovieRepository, ObjectStorage,
        PosterFetcherClient,
    },
    value_objects::{ExternalMetadataId, MovieId, PosterPath},
};

pub struct PosterSyncHandler {
    movie_repository: Arc<dyn MovieRepository>,
    metadata_client: Arc<dyn MetadataClient>,
    poster_fetcher: Arc<dyn PosterFetcherClient>,
    object_storage: Arc<dyn ObjectStorage>,
    event_publisher: Arc<dyn EventPublisher>,
    max_retries: u32,
}

impl PosterSyncHandler {
    pub fn new(
        movie_repository: Arc<dyn MovieRepository>,
        metadata_client: Arc<dyn MetadataClient>,
        poster_fetcher: Arc<dyn PosterFetcherClient>,
        object_storage: Arc<dyn ObjectStorage>,
        event_publisher: Arc<dyn EventPublisher>,
        max_retries: u32,
    ) -> Self {
        Self {
            movie_repository,
            metadata_client,
            poster_fetcher,
            object_storage,
            event_publisher,
            max_retries,
        }
    }

    async fn sync(
        &self,
        movie_id: MovieId,
        external_metadata_id: ExternalMetadataId,
    ) -> Result<(), DomainError> {
        let mut movie = match self.movie_repository.get_movie_by_id(&movie_id).await? {
            Some(m) => m,
            None => {
                tracing::warn!("Sync cancelled: Movie {} not found", movie_id.value());
                return Err(DomainError::NotFound("Movie not found".into()));
            }
        };

        let poster_url = match self
            .metadata_client
            .get_poster_url(&external_metadata_id)
            .await
        {
            Ok(Some(url)) => url,
            Ok(None) => return Ok(()),
            Err(e) => {
                tracing::warn!("Failed to find poster URL: {:?}", e);
                return Err(e);
            }
        };

        let image_bytes = self.poster_fetcher.fetch_poster_bytes(&poster_url).await?;
        let stored_path = self
            .object_storage
            .store(&movie_id.value().to_string(), &image_bytes)
            .await?;
        if let Err(e) = self
            .event_publisher
            .publish(&DomainEvent::ImageStored {
                key: stored_path.clone(),
            })
            .await
        {
            tracing::warn!("failed to emit ImageStored for {stored_path}: {e}");
        }
        let poster_path = PosterPath::new(stored_path)?;

        movie.update_poster(poster_path);
        self.movie_repository.upsert_movie(&movie).await
    }
}

#[async_trait]
impl EventHandler for PosterSyncHandler {
    async fn handle(&self, event: &DomainEvent) -> Result<(), DomainError> {
        let (movie_id, external_metadata_id) = match event {
            DomainEvent::MovieDiscovered {
                movie_id,
                external_metadata_id,
            } => (movie_id.value(), external_metadata_id.value().to_owned()),
            DomainEvent::MovieEnrichmentRequested {
                movie_id,
                external_metadata_id,
            } => {
                // Only sync poster if the movie doesn't have one yet
                let already_has_poster = self
                    .movie_repository
                    .get_movie_by_id(&MovieId::from_uuid(movie_id.value()))
                    .await?
                    .map(|m| m.poster_path().is_some())
                    .unwrap_or(false);
                if already_has_poster {
                    return Ok(());
                }
                (movie_id.value(), external_metadata_id.clone())
            }
            _ => return Ok(()),
        };

        let movie_id = MovieId::from_uuid(movie_id);
        let external_metadata_id = ExternalMetadataId::new(external_metadata_id)?;

        let mut last_err: Option<DomainError> = None;
        for attempt in 0..=self.max_retries {
            match self
                .sync(movie_id.clone(), external_metadata_id.clone())
                .await
            {
                Ok(()) => return Ok(()),
                Err(e) => {
                    if attempt < self.max_retries {
                        let delay = Duration::from_secs(2u64.pow(attempt));
                        tracing::warn!(
                            attempt = attempt + 1,
                            max_attempts = self.max_retries + 1,
                            delay_secs = delay.as_secs(),
                            "poster sync failed, retrying: {e}"
                        );
                        tokio::time::sleep(delay).await;
                    }
                    last_err = Some(e);
                }
            }
        }

        let err = last_err.expect("loop runs at least once");
        tracing::error!(
            attempts = self.max_retries + 1,
            "poster sync failed after all attempts: {err}"
        );
        Err(err)
    }
}