saleor-apps-rs/sitemap-generator/src/main.rs

#![allow(
    non_upper_case_globals,
    clippy::large_enum_variant,
    clippy::upper_case_acronyms
)]
#![feature(let_chains)]
#![deny(clippy::unwrap_used, clippy::expect_used)]
mod app;
mod queries;
mod routes;

use saleor_app_sdk::{
    config::Config,
    manifest::{cargo_info, AppManifestBuilder, AppPermission},
    webhooks::{AsyncWebhookEventType, WebhookManifestBuilder},
    SaleorApp,
};
use std::sync::Arc;
use tokio::sync::Mutex;
use tracing::{debug, info};

use crate::{
    app::{trace_to_std, AppState, SitemapConfig, XmlCache},
    queries::event_subjects_updated::EVENTS_QUERY,
    routes::{create_routes, register::regenerate},
};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let config = Config::load()?;
    trace_to_std(&config)?;
    let sitemap_config = SitemapConfig::load()?;
    debug!("Creating configs...");

    let saleor_app = SaleorApp::new(&config)?;

    debug!("Creating saleor App...");

    let app_manifest = AppManifestBuilder::new(&config, cargo_info!())
        .add_permissions(vec![
            AppPermission::ManageProducts,
            AppPermission::ManagePages,
        ])
        .add_webhook(
            WebhookManifestBuilder::new(&config)
                .set_query(EVENTS_QUERY)
                .add_async_events(vec![
                    AsyncWebhookEventType::ProductCreated,
                    AsyncWebhookEventType::ProductUpdated,
                    AsyncWebhookEventType::ProductDeleted,
                    AsyncWebhookEventType::CategoryCreated,
                    AsyncWebhookEventType::CategoryUpdated,
                    AsyncWebhookEventType::CategoryDeleted,
                    AsyncWebhookEventType::PageCreated,
                    AsyncWebhookEventType::PageUpdated,
                    AsyncWebhookEventType::PageDeleted,
                    AsyncWebhookEventType::CollectionCreated,
                    AsyncWebhookEventType::CollectionUpdated,
                    AsyncWebhookEventType::CollectionDeleted,
                ])
                .build(),
        )
        .build();
    debug!("Created AppManifest...");
    let app_state = AppState {
        sitemap_config,
        xml_cache: Arc::new(Mutex::new(XmlCache::new(
            &config.apl_url,
            &config.app_api_base_url,
        )?)),
        manifest: app_manifest,
        config: config.clone(),
        saleor_app: Arc::new(Mutex::new(saleor_app)),
    };
    debug!("Created AppState...");

    {
        // either clear the cache, regenerate or both from command args
        let mut pargs = pico_args::Arguments::from_env();

        if let Some(for_url) = pargs.opt_value_from_str::<_, String>("--for-url")? {
            if pargs.contains("--cache-clear") {
                let xml_cache = app_state.xml_cache.lock().await;
                xml_cache.delete_all(&for_url).await?;
                debug!("Cleared Xml Cache for {for_url}");
            }

            if pargs.contains("--cache-regenerate") {
                regenerate(app_state.clone(), for_url).await?;
            }
            std::process::exit(0)
        }
    }

    let app = create_routes(app_state);
    let listener = tokio::net::TcpListener::bind(
        "0.0.0.0:".to_owned()
            + config
                .app_api_base_url
                .split(':')
                .collect::<Vec<_>>()
                .get(2)
                .unwrap_or(&"3000"),
    )
    .await?;
    info!("listening on {}", listener.local_addr()?);
    match axum::serve(listener, app).await {
        Ok(o) => Ok(o),
        Err(e) => anyhow::bail!(e),
    }
}