105 lines
2.8 KiB
Rust
105 lines
2.8 KiB
Rust
|
extern crate lazy_static;
|
||
|
|
use config::{Config, ConfigError, Environment, File};
|
||
|
|
use serde::Deserialize;
|
||
|
|
use std::env;
|
||
|
|
use std::net::IpAddr;
|
||
|
|
|
||
|
|
static CONFIG_FILE_DEFAULTS: &str = "config/defaults.hjson";
|
||
|
|
static CONFIG_FILE_COSTUMIZED: &str = "config/custom.hjson";
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
pub struct Settings {
|
||
|
|
pub database: Database,
|
||
|
|
pub hostname: String,
|
||
|
|
pub bind: IpAddr,
|
||
|
|
pub port: u16,
|
||
|
|
pub jwt_secret: String,
|
||
|
|
pub rate_limit: RateLimitConfig,
|
||
|
|
pub email: Option<EmailConfig>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
pub struct RateLimitConfig {
|
||
|
|
pub message: i32,
|
||
|
|
pub message_per_second: i32,
|
||
|
|
pub post: i32,
|
||
|
|
pub post_per_second: i32,
|
||
|
|
pub register: i32,
|
||
|
|
pub register_per_second: i32,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
pub struct EmailConfig {
|
||
|
|
pub smtp_server: String,
|
||
|
|
pub smtp_login: String,
|
||
|
|
pub smtp_password: String,
|
||
|
|
pub smtp_from_address: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
pub struct Database {
|
||
|
|
pub user: String,
|
||
|
|
pub password: String,
|
||
|
|
pub host: String,
|
||
|
|
pub port: i32,
|
||
|
|
pub database: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
lazy_static! {
|
||
|
|
static ref SETTINGS: Settings = {
|
||
|
|
return match Settings::init() {
|
||
|
|
Ok(c) => c,
|
||
|
|
Err(e) => panic!("{}", e),
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Settings {
|
||
|
|
|
||
|
|
/// Reads config from the files and environment.
|
||
|
|
/// First, defaults are loaded from CONFIG_FILE_DEFAULTS, then these values can be overwritten
|
||
|
|
/// from CONFIG_FILE_COSTUMIZED (optional). Finally, values from the environment
|
||
|
|
/// (with prefix LEMMY) are added to the config.
|
||
|
|
fn init() -> Result<Self, ConfigError> {
|
||
|
|
let mut s = Config::new();
|
||
|
|
|
||
|
|
// Start off by merging in the "default" configuration file
|
||
|
|
s.merge(File::with_name(CONFIG_FILE_DEFAULTS))?;
|
||
|
|
|
||
|
|
// TODO: we could also automatically load dev/prod configs based on environment
|
||
|
|
// https://github.com/mehcode/config-rs/blob/master/examples/hierarchical-env/src/settings.rs#L49
|
||
|
|
s.merge(File::with_name(CONFIG_FILE_COSTUMIZED).required(false))?;
|
||
|
|
|
||
|
|
// Add in settings from the environment (with a prefix of LEMMY)
|
||
|
|
// Eg.. `LEMMY_DEBUG=1 ./target/app` would set the `debug` key
|
||
|
|
s.merge(Environment::with_prefix("LEMMY"))?;
|
||
|
|
|
||
|
|
return s.try_into();
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Returns the config as a struct.
|
||
|
|
pub fn get() -> &'static Self {
|
||
|
|
&SETTINGS
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Returns the postgres connection url. If LEMMY_DATABASE_URL is set, that is used,
|
||
|
|
/// otherwise the connection url is generated from the config.
|
||
|
|
pub fn get_database_url(&self) -> String {
|
||
|
|
match env::var("LEMMY_DATABASE_URL") {
|
||
|
|
Ok(url) => url,
|
||
|
|
Err(_) => format!(
|
||
|
|
"postgres://{}:{}@{}:{}/{}",
|
||
|
|
self.database.user,
|
||
|
|
self.database.password,
|
||
|
|
self.database.host,
|
||
|
|
self.database.port,
|
||
|
|
self.database.database
|
||
|
|
),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn api_endpoint(&self) -> String {
|
||
|
|
format!("{}/api/v1", self.hostname)
|
||
|
|
}
|
||
|
|
}
|