Jon Häggblad
df03daf2cc
* socks5: remove pub mod not needed anymore * all: dedup parse_validators and move to common * rustfmt
102 lines
3.2 KiB
Rust
102 lines
3.2 KiB
Rust
// Copyright 2020 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use handlebars::Handlebars;
|
|
use serde::de::DeserializeOwned;
|
|
use serde::Serialize;
|
|
#[cfg(unix)]
|
|
use std::os::unix::fs::PermissionsExt;
|
|
use std::path::PathBuf;
|
|
use std::{fs, io};
|
|
|
|
pub mod defaults;
|
|
|
|
pub trait NymConfig: Default + Serialize + DeserializeOwned {
|
|
fn template() -> &'static str;
|
|
|
|
fn config_file_name() -> String {
|
|
"config.toml".to_string()
|
|
}
|
|
|
|
fn default_root_directory() -> PathBuf;
|
|
|
|
// default, most probable, implementations; can be easily overridden where required
|
|
fn default_config_directory(id: Option<&str>) -> PathBuf {
|
|
if let Some(id) = id {
|
|
Self::default_root_directory().join(id).join("config")
|
|
} else {
|
|
Self::default_root_directory().join("config")
|
|
}
|
|
}
|
|
|
|
fn default_data_directory(id: Option<&str>) -> PathBuf {
|
|
if let Some(id) = id {
|
|
Self::default_root_directory().join(id).join("data")
|
|
} else {
|
|
Self::default_root_directory().join("data")
|
|
}
|
|
}
|
|
|
|
fn default_config_file_path(id: Option<&str>) -> PathBuf {
|
|
Self::default_config_directory(id).join(Self::config_file_name())
|
|
}
|
|
|
|
fn root_directory(&self) -> PathBuf;
|
|
fn config_directory(&self) -> PathBuf;
|
|
fn data_directory(&self) -> PathBuf;
|
|
|
|
fn save_to_file(&self, custom_location: Option<PathBuf>) -> io::Result<()> {
|
|
let reg = Handlebars::new();
|
|
// it's whoever is implementing the trait responsibility to make sure you can execute your own template on your data
|
|
let templated_config = reg.render_template(Self::template(), self).unwrap();
|
|
|
|
// make sure the whole directory structure actually exists
|
|
match custom_location.clone() {
|
|
Some(loc) => {
|
|
if let Some(parent_dir) = loc.parent() {
|
|
fs::create_dir_all(parent_dir)
|
|
} else {
|
|
Ok(())
|
|
}
|
|
}
|
|
None => fs::create_dir_all(self.config_directory()),
|
|
}?;
|
|
|
|
let location = custom_location
|
|
.unwrap_or_else(|| self.config_directory().join(Self::config_file_name()));
|
|
|
|
cfg_if::cfg_if! {
|
|
if #[cfg(unix)] {
|
|
fs::write(location.clone(), templated_config)?;
|
|
let mut perms = fs::metadata(location.clone())?.permissions();
|
|
perms.set_mode(0o600);
|
|
fs::set_permissions(location, perms)?;
|
|
} else {
|
|
fs::write(location, templated_config)?;
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn load_from_file(id: Option<&str>) -> io::Result<Self> {
|
|
let file = Self::default_config_file_path(id);
|
|
log::trace!("Loading from file: {:#?}", file);
|
|
let config_contents = fs::read_to_string(file)?;
|
|
|
|
toml::from_str(&config_contents)
|
|
.map_err(|toml_err| io::Error::new(io::ErrorKind::Other, toml_err))
|
|
}
|
|
}
|
|
|
|
pub fn parse_validators(raw: &str) -> Vec<url::Url> {
|
|
raw.split(',')
|
|
.map(|raw_validator| {
|
|
raw_validator
|
|
.trim()
|
|
.parse()
|
|
.expect("one of the provided validator api urls is invalid")
|
|
})
|
|
.collect()
|
|
}
|