Broke the code some more and created some helper functions for parsing a

config tree.
This commit is contained in:
Thelie 2021-05-14 20:55:38 +02:00
parent 78d5bafae3
commit c47b511e04
2 changed files with 55 additions and 20 deletions

View file

@ -20,7 +20,35 @@ use std::{
io::Read, io::Read,
fs::read_dir, fs::read_dir,
}; };
use toml::value::Value; use toml::{
value::Value,
map::Map,
};
fn parse_toml_to_table(path: PathBuf)
-> Result<Map<String, Value>, Box<dyn std::error::Error>> {
// I once read that this multiple let style
// is the "rusty" way to do things.
// Personally, I find this specific instance to be somewhat sketchy.
let mut config = String::new();
std::fs::File::open(path)?
.read_to_string(&mut config)?;
let mut config = config.parse::<Value>()?;
// The important thing here is that config is a mut Table in the end.
Ok(config.as_table_mut().unwrap().clone())
}
fn merge_config_maps(mut receiver: Map<String, Value>, donor: Map<String, Value>)
-> Result<Map<String, Value>, Box<dyn std::error::Error>> {
for key in donor.keys() {
if receiver.contains_key(&key.clone()) {
receiver[&key.clone()] = donor[key].clone();
} else {
receiver.insert(key.clone(), donor[key].clone());
}
}
Ok(receiver)
}
fn parse_server_dir(mut config: Value, server_config_path: PathBuf) fn parse_server_dir(mut config: Value, server_config_path: PathBuf)
-> Result<Value, Box<dyn std::error::Error>> { -> Result<Value, Box<dyn std::error::Error>> {
@ -29,9 +57,22 @@ fn parse_server_dir(mut config: Value, server_config_path: PathBuf)
for entry in read_dir(server_config_path)? { for entry in read_dir(server_config_path)? {
let entry = entry?; let entry = entry?;
match entry.file_name().to_str().ok_or("Could not read file name")? { match entry.file_name().to_str().ok_or("Could not read file name")? {
"config.toml" => (), "config.toml" => {
"clients.toml" => (), let server_config = parse_toml_to_table(entry.path())?;
"channels.toml" => (), config = merge_config_maps(config, server_config)?;
},
"clients.toml" => {
let client_config = parse_toml_to_table(entry.path())?;
config = merge_config_maps(config, client_config)?;
},
"channels.toml" => {
let channels = Value::from(parse_toml_to_table(entry.path())?);
if config.contains_key(&"channels".to_owned()) {
config[&"channels".to_owned()] = channels;
} else {
config.insert("channels".to_owned(), channels);
}
},
_ => (), _ => (),
} }
} }
@ -45,16 +86,7 @@ fn parse_server_dir(mut config: Value, server_config_path: PathBuf)
// TODO: Document config file options once they are stable. // TODO: Document config file options once they are stable.
pub fn get_main_config(config_path: PathBuf) pub fn get_main_config(config_path: PathBuf)
-> Result<Value, Box<dyn std::error::Error>> { -> Result<Value, Box<dyn std::error::Error>> {
let mut config = parse_toml_to_table(config_path)?;
// I once read that this multiple let style
// is the "rusty" way to do things.
// Personally, I find this specific instance to be somewhat sketchy.
let mut config = String::new();
std::fs::File::open(config_path)?
.read_to_string(&mut config)?;
let mut config = config.parse::<Value>()?;
// The important thing here is that config is a mut Table in the end.
let mut config = config.as_table_mut().unwrap().clone();
if config.get("port") == None { if config.get("port") == None {
config.insert(String::from("port"), Value::from("6697")); config.insert(String::from("port"), Value::from("6697"));
@ -73,16 +105,17 @@ pub fn get_main_config(config_path: PathBuf)
Ok(Value::from(config)) Ok(Value::from(config))
} }
pub fn get_server_config(mut config: Value, config_path: PathBuf) pub fn get_server_configs(config: Value, config_path: PathBuf)
-> Result<Value, Box<dyn std::error::Error>> { -> Result<Vec<Value>, Box<dyn std::error::Error>> {
let mut config_vec = vec![];
if config_path.is_dir() { if config_path.is_dir() {
for entry in read_dir(config_path)? { for entry in read_dir(config_path)? {
let entry = entry?; let entry = entry?;
let server_path = entry.path(); let server_path = entry.path();
if server_path.is_dir(){ if server_path.is_dir(){
config = parse_server_dir(config, server_path)?; config_vec.push(parse_server_dir(config.clone(), server_path)?);
} }
} }
} }
Ok(config) Ok(config_vec)
} }

View file

@ -152,7 +152,7 @@ fn main() {
} }
} }
let mut config; let config;
match config_path { match config_path {
Some(p) => config = config::get_main_config(PathBuf::from(p)) Some(p) => config = config::get_main_config(PathBuf::from(p))
@ -161,8 +161,10 @@ fn main() {
.expect("Could not get default config in /etc/Mention2Mail/default.toml"), .expect("Could not get default config in /etc/Mention2Mail/default.toml"),
} }
let mut server_configs;
match server_conf_path { match server_conf_path {
Some(p) => config = config::get_server_config(config, PathBuf::from(p)) Some(p) => server_configs = config::get_server_configs(config, PathBuf::from(p))
.expect("Could not get server config."), .expect("Could not get server config."),
None => (), None => (),
} }