movies-diary/crates/worker/src/main.rs

mod db;
mod event_bus;

use std::sync::Arc;

use anyhow::Context;
use application::{config::AppConfig, context::AppContext, worker::WorkerService};
use export::ExportAdapter;
use importer::ImporterDocumentParser;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

use domain::ports::{DiaryExporter, DocumentParser, EventHandler, PeriodicJob};

#[cfg(not(any(feature = "sqlite", feature = "postgres")))]
compile_error!("At least one database backend must be enabled. Use --features sqlite or --features postgres");

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    dotenvy::dotenv().ok();
    init_tracing();

    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 app_config = AppConfig::from_env();

    let (auth_service, password_hasher) = auth::create()?;
    let metadata_client = metadata::create()?;
    let poster_fetcher = poster_fetcher::create()?;
    let image_storage = image_storage::create()?;

    let (repos, db_pool) = db::connect(&database_url, &backend).await?;
    let (event_publisher_arc, consumer_arc) = event_bus::create(&db_pool).await?;

    // Clone refs federation handler needs before ctx consumes them.
    #[cfg(feature = "federation")]
    let (fed_movie_repo, fed_review_repo, fed_diary_repo, fed_user_repo, base_url, allow_registration) = (
        Arc::clone(&repos.movie),
        Arc::clone(&repos.review),
        Arc::clone(&repos.diary),
        Arc::clone(&repos.user),
        app_config.base_url.clone(),
        app_config.allow_registration,
    );

    let ctx = AppContext {
        movie_repository:       repos.movie,
        review_repository:      repos.review,
        diary_repository:       repos.diary,
        diary_exporter:         Arc::new(ExportAdapter) as Arc<dyn DiaryExporter>,
        document_parser:        Arc::new(ImporterDocumentParser) as Arc<dyn DocumentParser>,
        stats_repository:       repos.stats,
        metadata_client,
        poster_fetcher,
        image_storage,
        event_publisher:        event_publisher_arc,
        auth_service,
        password_hasher,
        user_repository:        repos.user,
        import_session_repository: repos.import_session,
        import_profile_repository: repos.import_profile,
        movie_profile_repository:  repos.movie_profile,
        config: app_config,
    };

    // ── Enrichment ────────────────────────────────────────────────────────────

    let enrichment_handler: Option<Arc<dyn EventHandler>> =
        match tmdb_enrichment::TmdbEnrichmentClient::from_env() {
            Ok(client) => {
                tracing::info!("TMDb enrichment enabled");
                Some(Arc::new(tmdb_enrichment::EnrichmentHandler {
                    enrichment_client: Arc::new(client),
                    profile_repo: Arc::clone(&ctx.movie_profile_repository),
                }))
            }
            Err(e) => {
                tracing::warn!("TMDb enrichment disabled: {e}");
                None
            }
        };

    // ── Periodic jobs ─────────────────────────────────────────────────────────

    let periodic_jobs: Vec<Arc<dyn PeriodicJob>> = vec![
        Arc::new(application::jobs::ImportSessionCleanupJob::new(ctx.clone())),
        Arc::new(application::jobs::EnrichmentStalenessJob::new(ctx.clone())),
    ];

    for job in periodic_jobs {
        tokio::spawn(async move {
            let mut tick = tokio::time::interval(job.interval());
            loop {
                tick.tick().await;
                if let Err(e) = job.run().await {
                    tracing::error!("periodic job failed: {e}");
                }
            }
        });
    }

    // ── Event handlers ────────────────────────────────────────────────────────

    let handlers: Vec<Arc<dyn EventHandler>> = {
        let poster = Arc::new(poster_sync::PosterSyncHandler::new(
            Arc::clone(&ctx.movie_repository),
            Arc::clone(&ctx.metadata_client),
            Arc::clone(&ctx.poster_fetcher),
            Arc::clone(&ctx.image_storage),
            3,
        )) as Arc<dyn EventHandler>;

        let cleanup = Arc::new(image_storage::ImageCleanupHandler::new(
            Arc::clone(&ctx.image_storage),
        )) as Arc<dyn EventHandler>;

        #[cfg(not(feature = "federation"))]
        {
            let mut h = vec![poster, cleanup];
            if let Some(e) = enrichment_handler { h.push(e); }
            h
        }

        #[cfg(feature = "federation")]
        {
            let (federation_repo, _social_query, review_store) = match &db_pool {
                #[cfg(feature = "sqlite-federation")]
                db::DbPool::Sqlite(pool) => sqlite_federation::wire(pool.clone()),
                #[cfg(feature = "postgres-federation")]
                db::DbPool::Postgres(pool) => postgres_federation::wire(pool.clone()),
            };

            let ap = activitypub::wire(
                federation_repo,
                review_store,
                fed_user_repo,
                fed_movie_repo,
                fed_review_repo,
                fed_diary_repo,
                base_url,
                allow_registration,
            ).await?.event_handler;

            tracing::info!("federation event handler registered");
            let mut h = vec![poster, cleanup, ap];
            if let Some(e) = enrichment_handler { h.push(e); }
            h
        }
    };

    // ── Run ───────────────────────────────────────────────────────────────────

    let worker = WorkerService::new(consumer_arc, handlers);
    tracing::info!("worker started");
    worker.run().await;
    tracing::info!("worker stopped");

    Ok(())
}

fn init_tracing() {
    let filter = std::env::var("RUST_LOG")
        .unwrap_or_else(|_| "worker=info,application=info".to_string());
    tracing_subscriber::registry()
        .with(tracing_subscriber::EnvFilter::new(filter))
        .with(tracing_subscriber::fmt::layer())
        .init();
}