use config::{ConfigError, FileStoredFormat, Format, Map, Value, ValueKind};
use std::error::Error;
#[derive(Debug, Clone)]
pub struct MyTomlFile;
impl Format for MyTomlFile {
fn parse(&self, uri: Option<&String>, text: &str) -> Result<Map<String, Value>, Box<dyn Error + Send + Sync>> {
let value = from_toml_value(uri, &toml::from_str(text)?);
match value.kind {
ValueKind::Table(map) => Ok(map),
_ => Err(ConfigError::Message("The config is not a table".to_string())),
}
.map_err(|err| Box::new(err) as Box<dyn Error + Send + Sync>)
}
}
impl FileStoredFormat for MyTomlFile {
fn file_extensions(&self) -> &'static [&'static str] {
&["toml"]
}
}
fn from_toml_value(uri: Option<&String>, value: &toml::Value) -> Value {
match *value {
toml::Value::String(ref value) => {
if value.to_string().to_lowercase() == "auto" {
Value::new(uri, ValueKind::Nil)
} else {
Value::new(uri, value.to_string())
}
}
toml::Value::Float(value) => Value::new(uri, value),
toml::Value::Integer(value) => Value::new(uri, value),
toml::Value::Boolean(value) => Value::new(uri, value),
toml::Value::Table(ref table) => {
let mut m = Map::new();
for (key, value) in table {
m.insert(key.clone(), from_toml_value(uri, value));
}
Value::new(uri, m)
}
toml::Value::Array(ref array) => {
let mut l = Vec::new();
for value in array {
l.push(from_toml_value(uri, value));
}
Value::new(uri, l)
}
toml::Value::Datetime(ref datetime) => Value::new(uri, datetime.to_string()),
}
}