nocturne/backend/server/src/modules/mod.rs

pub mod blog;

use std::{env, num::ParseIntError, thread};

use crate::{
    constants::config::server::{default_value, var_name},
    scheduler::{tasks::CronTask, Scheduler},
};
use macros::error_panic;

use axum::Router;
use futures::executor::block_on;
use tokio::net::TcpListener;
use tracing::{info, warn};

// todo: merge `get_server_router` & `get_mount_point`, make Option
pub trait Module {
    fn get_server_router(&self) -> Router;

    fn get_mount_point(&self) -> &'static str;

    fn get_cron_tasks(&self) -> Vec<Box<dyn CronTask>>;
}

pub struct ModuleTree {
    pub modules: Vec<Box<dyn Module>>,
}

impl ModuleTree {
    pub fn new() -> Self {
        ModuleTree { modules: vec![] }
    }

    pub fn add_module(&mut self, module: impl Module + 'static) -> &mut Self {
        self.modules.push(Box::new(module));
        self
    }

    pub async fn blocking_serve(&self) {
        let scheduler = Scheduler::new();
        let mut app = Router::new();

        let port: u32 = match env::var(var_name::PORT) {
            Ok(value) => value.parse().unwrap_or_else(|e: ParseIntError| {
                warn!(
                    "failed to parse `{}`, using default port: {}",
                    var_name::PORT,
                    e.to_string()
                );
                default_value::PORT
            }),
            Err(_) => default_value::PORT,
        };

        let listener = match TcpListener::bind(format!("{}:{}", default_value::HOST, port)).await {
            Ok(listener) => {
                info!("server started on {}:{}", default_value::HOST, port);
                listener
            }
            Err(e) => {
                error_panic!(
                    "failed to start server on {}:{}: {}",
                    default_value::HOST,
                    port,
                    e.to_string()
                );
            }
        };

        for module in &self.modules {
            let _ = scheduler.insert_list(module.get_cron_tasks()).await;
            app = app.nest(module.get_mount_point(), module.get_server_router());
        }

        let _ = thread::spawn(move || {
            let fut = scheduler.start();
            block_on(fut).unwrap()
        })
        .join();

        axum::serve(listener, app).await.unwrap_or_else(|e| {
            error_panic!("failed to start axum: {}", e.to_string());
        });
    }
}