webcomment/src/config.rs

113 lines
3.2 KiB
Rust
Raw Normal View History

2022-10-15 13:32:57 +00:00
use serde::{Deserialize, Serialize};
use std::{
net::{IpAddr, Ipv4Addr, SocketAddr},
path::Path,
};
const CONFIG_FILE: &str = "config.toml";
#[derive(Deserialize, Serialize)]
pub struct Config {
//#[serde(default = "Config::default_admin_emails")]
//pub admin_emails: Vec<String>,
#[serde(default = "Config::default_admin_passwords")]
pub admin_passwords: Vec<String>,
/// New or edited comments need admin's approval before being public
#[serde(default = "Config::default_comment_approve")]
pub comment_approve: bool,
/// Duration for which a comment can be edited
#[serde(default = "Config::default_comment_edit_timeout")]
pub comment_edit_timeout: u64,
#[serde(default = "Config::default_comment_author_max_len")]
pub comment_author_max_len: usize,
#[serde(default = "Config::default_comment_email_max_len")]
pub comment_email_max_len: usize,
#[serde(default = "Config::default_comment_text_max_len")]
pub comment_text_max_len: usize,
#[serde(default = "Config::default_cookies_https_only")]
pub cookies_https_only: bool,
#[serde(default = "Config::default_cookies_domain")]
pub cookies_domain: Option<String>,
#[serde(default = "Config::default_lang")]
pub lang: String,
#[serde(default = "Config::default_listen")]
pub listen: SocketAddr,
#[serde(default = "Config::default_root_url")]
pub root_url: String,
}
impl Config {
/*fn default_admin_emails() -> Vec<String> {
vec![]
}*/
fn default_admin_passwords() -> Vec<String> {
vec![]
}
fn default_comment_approve() -> bool {
true
}
fn default_comment_edit_timeout() -> u64 {
7 * 86400
}
fn default_comment_author_max_len() -> usize {
64
}
fn default_comment_email_max_len() -> usize {
128
}
fn default_comment_text_max_len() -> usize {
128 * 1024
}
fn default_cookies_https_only() -> bool {
false
}
fn default_cookies_domain() -> Option<String> {
None
}
fn default_lang() -> String {
"en_GB".into()
}
fn default_listen() -> SocketAddr {
SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 31720)
}
fn default_root_url() -> String {
"/".into()
}
}
impl Default for Config {
fn default() -> Self {
Self {
//admin_emails: Self::default_admin_emails(),
admin_passwords: Self::default_admin_passwords(),
comment_approve: Self::default_comment_approve(),
comment_edit_timeout: Self::default_comment_edit_timeout(),
comment_author_max_len: Self::default_comment_author_max_len(),
comment_email_max_len: Self::default_comment_email_max_len(),
comment_text_max_len: Self::default_comment_text_max_len(),
cookies_https_only: Self::default_cookies_https_only(),
cookies_domain: Self::default_cookies_domain(),
lang: Self::default_lang(),
listen: Self::default_listen(),
root_url: Self::default_root_url(),
}
}
}
pub fn read_config(dir: &Path) -> Config {
let path = dir.join(CONFIG_FILE);
if !path.is_file() {
let config = Config::default();
std::fs::write(path, toml_edit::easy::to_string_pretty(&config).unwrap())
.expect("Cannot write config file");
config
} else {
toml_edit::easy::from_str(
std::str::from_utf8(&std::fs::read(path).expect("Cannot read config file"))
.expect("Bad encoding in config file"),
)
.expect("Bad JSON in config file")
}
}