core/emgauwa-core/src/main.rs

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_server = 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_server.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
}