lemmy/lemmy_utils/src/settings.rs

174 lines
4.6 KiB
Rust
Raw Normal View History

2019-12-15 16:40:55 +00:00
use config::{Config, ConfigError, Environment, File};
use serde::Deserialize;
use std::{env, fs, io::Error, net::IpAddr, sync::RwLock};
2019-12-15 16:40:55 +00:00
static CONFIG_FILE_DEFAULTS: &str = "config/defaults.hjson";
static CONFIG_FILE: &str = "config/config.hjson";
2019-12-15 16:40:55 +00:00
#[derive(Debug, Deserialize, Clone)]
2019-12-15 16:40:55 +00:00
pub struct Settings {
pub setup: Option<Setup>,
pub database: DatabaseConfig,
2019-12-15 16:40:55 +00:00
pub hostname: String,
pub bind: IpAddr,
pub port: u16,
pub jwt_secret: String,
pub pictrs_url: String,
2019-12-15 16:40:55 +00:00
pub rate_limit: RateLimitConfig,
pub email: Option<EmailConfig>,
pub federation: FederationConfig,
pub captcha: CaptchaConfig,
2019-12-15 16:40:55 +00:00
}
#[derive(Debug, Deserialize, Clone)]
pub struct Setup {
pub admin_username: String,
pub admin_password: String,
pub admin_email: Option<String>,
pub site_name: String,
}
#[derive(Debug, Deserialize, Clone)]
2019-12-15 16:40:55 +00:00
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,
pub image: i32,
pub image_per_second: i32,
2019-12-15 16:40:55 +00:00
}
#[derive(Debug, Deserialize, Clone)]
2019-12-15 16:40:55 +00:00
pub struct EmailConfig {
pub smtp_server: String,
2020-01-31 11:03:26 +00:00
pub smtp_login: Option<String>,
pub smtp_password: Option<String>,
2019-12-15 16:40:55 +00:00
pub smtp_from_address: String,
2020-01-31 11:03:26 +00:00
pub use_tls: bool,
2019-12-15 16:40:55 +00:00
}
#[derive(Debug, Deserialize, Clone)]
pub struct CaptchaConfig {
pub enabled: bool,
pub difficulty: String, // easy, medium, or hard
}
#[derive(Debug, Deserialize, Clone)]
pub struct DatabaseConfig {
2019-12-15 16:40:55 +00:00
pub user: String,
pub password: String,
pub host: String,
pub port: i32,
pub database: String,
2019-12-16 12:25:47 +00:00
pub pool_size: u32,
2019-12-15 16:40:55 +00:00
}
#[derive(Debug, Deserialize, Clone)]
pub struct FederationConfig {
2020-03-18 21:09:00 +00:00
pub enabled: bool,
pub tls_enabled: bool,
pub allowed_instances: String,
pub blocked_instances: String,
2020-03-18 21:09:00 +00:00
}
2019-12-15 16:40:55 +00:00
lazy_static! {
static ref SETTINGS: RwLock<Settings> = RwLock::new(match Settings::init() {
Ok(c) => c,
Err(e) => panic!("{}", e),
});
2019-12-15 16:40:55 +00:00
}
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 (optional). Finally, values from the environment (with prefix LEMMY) are
/// added to the config.
///
/// Note: The env var `LEMMY_DATABASE_URL` is parsed in
/// `lemmy_db/src/lib.rs::get_database_url_from_env()`
2019-12-15 16:40:55 +00:00
fn init() -> Result<Self, ConfigError> {
let mut s = Config::new();
s.merge(File::with_name(&Self::get_config_defaults_location()))?;
2019-12-15 16:40:55 +00:00
s.merge(File::with_name(CONFIG_FILE).required(false))?;
2019-12-15 16:40:55 +00:00
// Add in settings from the environment (with a prefix of LEMMY)
// Eg.. `LEMMY_DEBUG=1 ./target/app` would set the `debug` key
// Note: we need to use double underscore here, because otherwise variables containing
// underscore cant be set from environmnet.
// https://github.com/mehcode/config-rs/issues/73
s.merge(Environment::with_prefix("LEMMY").separator("__"))?;
2019-12-15 16:40:55 +00:00
s.try_into()
2019-12-15 16:40:55 +00:00
}
/// Returns the config as a struct.
pub fn get() -> Self {
SETTINGS.read().unwrap().to_owned()
2019-12-15 16:40:55 +00:00
}
pub fn get_database_url(&self) -> String {
format!(
"postgres://{}:{}@{}:{}/{}",
self.database.user,
self.database.password,
self.database.host,
self.database.port,
self.database.database
)
2019-12-15 16:40:55 +00:00
}
pub fn get_config_defaults_location() -> String {
env::var("LEMMY_CONFIG_LOCATION").unwrap_or_else(|_| CONFIG_FILE_DEFAULTS.to_string())
}
pub fn read_config_file() -> Result<String, Error> {
fs::read_to_string(CONFIG_FILE)
}
pub fn get_allowed_instances(&self) -> Vec<String> {
let mut allowed_instances: Vec<String> = self
.federation
.allowed_instances
.split(',')
.map(|d| d.to_string())
.collect();
// The defaults.hjson config always returns a [""]
allowed_instances.retain(|d| !d.eq(""));
allowed_instances
}
pub fn get_blocked_instances(&self) -> Vec<String> {
let mut blocked_instances: Vec<String> = self
.federation
.blocked_instances
.split(',')
.map(|d| d.to_string())
.collect();
// The defaults.hjson config always returns a [""]
blocked_instances.retain(|d| !d.eq(""));
blocked_instances
}
pub fn save_config_file(data: &str) -> Result<String, Error> {
fs::write(CONFIG_FILE, data)?;
// Reload the new settings
// From https://stackoverflow.com/questions/29654927/how-do-i-assign-a-string-to-a-mutable-static-variable/47181804#47181804
let mut new_settings = SETTINGS.write().unwrap();
*new_settings = match Settings::init() {
Ok(c) => c,
Err(e) => panic!("{}", e),
};
Self::read_config_file()
}
2019-12-15 16:40:55 +00:00
}