li7y/src/main.rs

// SPDX-FileCopyrightText: 2024 Simon Bruder <simon@sbruder.de>
//
// SPDX-License-Identifier: AGPL-3.0-or-later

use std::env;

use actix_web::{web, App, HttpServer};
use diesel::prelude::*;
use diesel::r2d2;
use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
use log::{debug, info};

pub const MIGRATIONS: EmbeddedMigrations = embed_migrations!();

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    env_logger::init();

    let manager = r2d2::ConnectionManager::<PgConnection>::new(
        env::var("DATABASE_URL").expect("DATABASE_URL must be set"),
    );
    let pool = r2d2::Pool::builder()
        .build(manager)
        .expect("database URL must be valid");

    pool.get()
        .expect("couldn’t get db connection from pool")
        .run_pending_migrations(MIGRATIONS)
        .expect("failed to run migrations");

    let address = env::var("LISTEN_ADDRESS").unwrap_or("::1".to_string());
    let port = env::var("LISTEN_PORT").map_or(8080, |s| {
        s.parse::<u16>().expect("failed to parse LISTEN_PORT")
    });
    let static_root = env::var("STATIC_ROOT").unwrap_or("static".to_string());

    info!("Starting on {address}:{port} with static files from {static_root}");
    debug!("Serving static files from {static_root}");

    HttpServer::new(move || {
        App::new()
            .app_data(web::Data::new(pool.clone()))
            .service(web::scope("/api/v1").configure(li7y::api::v1::config))
            .service(actix_files::Files::new("/static", &static_root))
    })
    .bind((address, port))?
    .run()
    .await
}