Broke the code some more and created some helper functions for parsing a
config tree.
This commit is contained in:
parent
78d5bafae3
commit
c47b511e04
2 changed files with 55 additions and 20 deletions
|
@ -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,17 +86,8 @@ 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)
|
||||||
}
|
}
|
||||||
|
|
|
@ -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 => (),
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue