80 lines
2.7 KiB
Rust
80 lines
2.7 KiB
Rust
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<ServerConfig> = LazyLock::new(ServerConfig::default);
|
|
pub static ONE_TIME_PASSWORDS: LazyLock<RwLock<Vec<OneTimePassword>>> =
|
|
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<Postgres>,
|
|
}
|