81 lines
2.1 KiB
Rust
81 lines
2.1 KiB
Rust
use std::io;
|
|
use std::path::Path;
|
|
|
|
pub mod env;
|
|
pub mod hierarchical;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub enum ItemStatus {
|
|
Present,
|
|
MissingFromActive,
|
|
MissingFromTemplate,
|
|
Modified,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub enum ValueType {
|
|
String,
|
|
Number,
|
|
Bool,
|
|
Null,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct ConfigItem {
|
|
pub key: String,
|
|
pub path: String,
|
|
pub value: Option<String>,
|
|
pub template_value: Option<String>,
|
|
pub default_value: Option<String>,
|
|
pub depth: usize,
|
|
pub is_group: bool,
|
|
pub status: ItemStatus,
|
|
pub value_type: ValueType,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub enum FormatType {
|
|
Env,
|
|
Json,
|
|
Yaml,
|
|
Toml,
|
|
}
|
|
|
|
pub trait FormatHandler {
|
|
fn parse(&self, path: &Path) -> io::Result<Vec<ConfigItem>>;
|
|
fn merge(&self, path: &Path, vars: &mut Vec<ConfigItem>) -> io::Result<()>;
|
|
fn write(&self, path: &Path, vars: &[ConfigItem]) -> io::Result<()>;
|
|
}
|
|
|
|
pub fn detect_format(path: &Path, override_format: Option<String>) -> FormatType {
|
|
if let Some(fmt) = override_format {
|
|
match fmt.to_lowercase().as_str() {
|
|
"env" => return FormatType::Env,
|
|
"json" => return FormatType::Json,
|
|
"yaml" | "yml" => return FormatType::Yaml,
|
|
"toml" => return FormatType::Toml,
|
|
_ => {}
|
|
}
|
|
}
|
|
|
|
let file_name = path.file_name().unwrap_or_default().to_string_lossy();
|
|
if file_name.ends_with(".json") {
|
|
FormatType::Json
|
|
} else if file_name.ends_with(".yaml") || file_name.ends_with(".yml") {
|
|
FormatType::Yaml
|
|
} else if file_name.ends_with(".toml") {
|
|
FormatType::Toml
|
|
} else {
|
|
FormatType::Env
|
|
}
|
|
}
|
|
|
|
pub fn get_handler(format: FormatType) -> Box<dyn FormatHandler> {
|
|
match format {
|
|
FormatType::Env => Box::new(env::EnvHandler),
|
|
FormatType::Json => Box::new(hierarchical::HierarchicalHandler::new(FormatType::Json)),
|
|
FormatType::Yaml => Box::new(hierarchical::HierarchicalHandler::new(FormatType::Yaml)),
|
|
FormatType::Toml => Box::new(hierarchical::HierarchicalHandler::new(FormatType::Toml)),
|
|
}
|
|
}
|