use std::net::TcpListener; use actix::Actor; use actix_cors::Cors; use actix_web::middleware::TrailingSlash; use actix_web::{middleware, web, App, HttpServer}; use emgauwa_lib::db::DbController; use emgauwa_lib::errors::EmgauwaError; use emgauwa_lib::utils::init_logging; use crate::app_state::AppState; use crate::utils::drop_privileges; mod app_state; mod handlers; mod settings; mod utils; #[actix_web::main] async fn main() -> Result<(), std::io::Error> { let settings = settings::init()?; init_logging(&settings.logging.level)?; let listener = TcpListener::bind(format!("{}:{}", settings.host, settings.port))?; drop_privileges(&settings)?; let pool = emgauwa_lib::db::init(&settings.database).await?; let mut conn = pool.acquire().await.map_err(EmgauwaError::from)?; DbController::all_inactive(&mut conn) .await .map_err(EmgauwaError::from)?; conn.close().await.map_err(EmgauwaError::from)?; let app_state = AppState::new(pool.clone()).start(); log::info!("Starting server on {}:{}", settings.host, settings.port); HttpServer::new(move || { let cors = Cors::default().allow_any_method().allow_any_header(); let origins = settings.origins.clone(); let cors = match settings.origins.is_empty() { true => cors.allow_any_origin(), false => cors.allowed_origin_fn(move |origin, _req_head| { origins.contains(&origin.to_str().unwrap_or_default().to_string()) }), }; App::new() .wrap(cors) .wrap(middleware::Logger::default()) .wrap(middleware::NormalizePath::new(TrailingSlash::Trim)) .app_data(web::JsonConfig::default().error_handler(handlers::json_error_handler)) .app_data(web::Data::new(pool.clone())) .app_data(web::Data::new(app_state.clone())) .service(handlers::v1::controllers::index) .service(handlers::v1::controllers::show) .service(handlers::v1::controllers::update) .service(handlers::v1::controllers::delete) .service(handlers::v1::relays::index) .service(handlers::v1::relays::tagged) .service(handlers::v1::relays::index_for_controller) .service(handlers::v1::relays::show_for_controller) .service(handlers::v1::relays::update_for_controller) .service(handlers::v1::schedules::index) .service(handlers::v1::schedules::tagged) .service(handlers::v1::schedules::show) .service(handlers::v1::schedules::add) .service(handlers::v1::schedules::add_list) .service(handlers::v1::schedules::update) .service(handlers::v1::schedules::delete) .service(handlers::v1::tags::index) .service(handlers::v1::ws::ws_controllers) }) .listen(listener)? .run() .await }