pub mod database; pub mod error; pub mod feature; pub mod mail; pub mod routing; pub mod server; pub mod utils; use std::sync::LazyLock; use feature::auth::OneTimePassword; use sqlx::{Pool, Postgres}; use tokio::sync::RwLock; use utils::naive_toml_parser; pub static SERVER_CONFIG: LazyLock = LazyLock::new(ServerConfig::default); pub static ONE_TIME_PASSWORDS: LazyLock>> = LazyLock::new(OneTimePassword::init); const DATABASE_CONFIG_FILE_LOCATION: &str = "./configs/database_config.toml"; const SERVER_CONFIG_FILE_LOCATION: &str = "./configs/server_config.toml"; #[derive(Debug)] pub struct DatabaseConfig { pub address: String, pub username: String, pub password: String, pub database: String, pub backend: String, pub connection_pool_size: u32, } impl Default for DatabaseConfig { fn default() -> Self { let (header, mut database_configs) = naive_toml_parser(DATABASE_CONFIG_FILE_LOCATION); if header == "[database_config]" { Self { address: database_configs.pop_front().unwrap().parse().unwrap(), username: database_configs.pop_front().unwrap().parse().unwrap(), password: database_configs.pop_front().unwrap().parse().unwrap(), database: database_configs.pop_front().unwrap().parse().unwrap(), backend: database_configs.pop_front().unwrap().parse().unwrap(), connection_pool_size: database_configs.pop_front().unwrap().parse().unwrap(), } } else { panic!("Database Config File Must Include [database_config] at the First Line") } } } #[derive(Debug)] pub struct ServerConfig { pub address: String, pub otp_time_limit: usize, pub login_token_time_limit: usize, pub concurrency_limit: usize, } impl Default for ServerConfig { fn default() -> Self { let (header, mut server_configs) = naive_toml_parser(SERVER_CONFIG_FILE_LOCATION); let value_or_max = |value: String| value.parse().map_or(usize::MAX, |value| value); if header == "[server_config]" { Self { address: server_configs.pop_front().unwrap().parse().unwrap(), otp_time_limit: value_or_max(server_configs.pop_front().unwrap()), login_token_time_limit: value_or_max(server_configs.pop_front().unwrap()), concurrency_limit: value_or_max(server_configs.pop_front().unwrap()), } } else { panic!("Server Config File Must Include [server_config] at the First Line") } } } #[derive(Debug, Clone)] pub struct AppState { pub database_connection: Pool, }