#![allow(unused)]
#![allow(dead_code)]

use axum_htmx::AutoVaryLayer;
use compendium::config::Config;
use compendium::static_assets::Assets;
use compendium::{config, router, AppState, Error, Result, Tx};
use minijinja::{Environment, Value};
use sqlx::postgres::PgPoolOptions;
use sqlx::PgPool;
use std::env;
use std::sync::Arc;
use tower_http::trace::TraceLayer;
use tracing::info;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

#[tokio::main]
async fn main() -> Result<()> {
    let config = Config::new("./config.toml".into())?;
    let assets = Assets::new(&config)?;

    // Logs
    tracing_subscriber::registry()
        .with(
            tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| {
                format!("{}=debug,tower_http=debug", env!("CARGO_CRATE_NAME")).into()
            }),
        )
        .with(tracing_subscriber::fmt::layer())
        .init();

    info!("Connecting to database {}", config.db);

    let pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(&config.db.generate_db_string(false)?)
        .await?;

    sqlx::migrate!("./migrations").run(&pool).await?;

    let (tx_state, tx_layer) = Tx::setup(pool);

    let app = router::new(&config)
        .layer(TraceLayer::new_for_http().on_request(()))
        .layer(tx_layer)
        .layer(AutoVaryLayer)
        .with_state(AppState::new(assets, tx_state));

    // Add hot reload only on dev mode
    #[cfg(debug_assertions)]
    let app = app.layer(tower_livereload::LiveReloadLayer::new());

    let listener = tokio::net::TcpListener::bind(config.addr).await?;
    info!("listening on {}", listener.local_addr()?);

    axum::serve(listener, app).await?;
    Ok(())
}