skantravels/rust-backend/src/main.rs

mod auth;

use actix_web::{get, Responder, HttpResponse, HttpServer, App, web};
use actix_web::middleware;
use std::sync::{Mutex, Arc};
use crate::auth::AuthorizedUser;

mod embedded {
    use refinery::embed_migrations;
    embed_migrations!("./migrations");
}

struct AppData {
    database: Arc<Mutex<rusqlite::Connection>>,
}

#[get("/hc")]
async fn healthcheck(data: web::Data<AppData>) -> impl Responder {
    let db = data.database.lock().unwrap();

    match db.pragma_query(None, "integrity_check", |_| Ok(())) {
        Ok(_) => HttpResponse::Ok().body("OK"),
        Err(_) => HttpResponse::InternalServerError().body("Error"),
    }

}

#[get("/authorized")]
async fn authorized(_: AuthorizedUser) -> impl Responder {
    HttpResponse::Ok().body("Authorized")
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    let port = std::env::var("RUST_BACKEND_PORT")
        .ok()
        .and_then(|port| port.parse::<u16>().ok())
        .unwrap_or(8080);

    let database_path = std::env::var("RUST_BACKEND_DB")
        .unwrap_or("database.sqlite3".to_string());

    println!("Opening database: {}", database_path);

    let mut conn = rusqlite::Connection::open(database_path.clone()).unwrap();

    embedded::migrations::runner().run(&mut conn).unwrap();

    println!("Starting web server at port {}", port);

    HttpServer::new(|| {
        let database_path = std::env::var("RUST_BACKEND_DB")
            .unwrap_or("database.sqlite3".to_string());

        let conn = Arc::new(Mutex::new(rusqlite::Connection::open(database_path).unwrap()));

        App::new()
            .app_data(web::Data::new(AppData {
                database: conn,
            }))
            .service(healthcheck)
            .service(authorized)
    })
    .bind(("0.0.0.0", port))?
    .run()
    .await
}