mabel/src/main.rs

mod content;
mod error;
mod routes;
mod state;

use anyhow::Result;
use axum::{
    routing::{get, post},
    Router, Server,
};
use figment::{
    providers::{Env, Format, Serialized, Toml},
    Figment,
};
use serde::{Deserialize, Serialize};
use sqlx::postgres::PgPoolOptions;
use state::AppState;
use std::{net::SocketAddr, sync::Arc};

#[derive(Deserialize, Serialize)]
struct Config {
    bind: String,
    database_url: String,
}

impl Default for Config {
    fn default() -> Self {
        Config {
            bind: "127.0.0.1:3000".to_owned(),
            database_url: "postgres://artificiale:changeme@localhost/artificiale".to_owned(),
        }
    }
}

#[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 pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(config.database_url.as_str())
        .await?;

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

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

    let app = Router::new()
        .route("/pages/:site/:slug", get(routes::content::page))
        .route("/admin/bootstrap", post(routes::admin::bootstrap))
        .with_state(shared_state);

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