mabel/src/main.rs

mod auth;
mod content;
mod http;
mod roles;
mod routes;
mod state;

use crate::{auth::http::refresh_sessions, state::Config};
use anyhow::Result;
use axum::{middleware, Server};
use figment::{
    providers::{Env, Format, Serialized, Toml},
    Figment,
};
use sqlx::postgres::PgPoolOptions;
use state::AppState;
use std::{net::SocketAddr, sync::Arc};

#[tokio::main]
async fn main() -> Result<()> {
    tracing_subscriber::fmt::init();

    let config: Config = Figment::from(Serialized::defaults(Config::default()))
        .merge(Toml::file("mabel.toml"))
        .merge(Env::prefixed("MABEL_"))
        .extract()?;
    let addr: SocketAddr = config.bind.parse()?;

    let database = PgPoolOptions::new()
        .max_connections(5)
        .connect(config.database_url.as_str())
        .await?;

    sqlx::migrate!().run(&database).await?;

    // Sanity check config
    config.sanity_check();

    let shared_state = Arc::new(AppState { database, config });

    let app = routes::create_router()
        .route_layer(middleware::from_fn_with_state(
            shared_state.clone(),
            refresh_sessions,
        ))
        .with_state(shared_state);

    tracing::debug!("listening on {}", addr);
    Server::bind(&addr).serve(app.into_make_service()).await?;
    Ok(())
}