121 lines
2.8 KiB
Rust
121 lines
2.8 KiB
Rust
|
use anyhow::Result;
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
use std::convert::TryFrom;
|
||
|
|
|
||
|
|
use crate::util::is_yaml_empty;
|
||
|
|
|
||
|
|
use super::ParsedConfig;
|
||
|
|
|
||
|
|
#[derive(Debug, Serialize, Deserialize, Clone)]
|
||
|
|
pub(crate) struct YAMLConfig {
|
||
|
|
#[serde(default)]
|
||
|
|
pub label: Option<String>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub includes: Option<Vec<String>>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub excludes: Option<Vec<String>>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub extra_includes: Option<Vec<String>>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub extra_excludes: Option<Vec<String>>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub use_standard_includes: Option<bool>,
|
||
|
|
|
||
|
|
// Filters
|
||
|
|
#[serde(default)]
|
||
|
|
pub filter_title: Option<String>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub filter_class: Option<String>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub filter_exec: Option<String>,
|
||
|
|
|
||
|
|
#[serde(default)]
|
||
|
|
pub filter_os: Option<String>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl YAMLConfig {
|
||
|
|
pub fn parse_from_str(yaml: &str) -> Result<Self> {
|
||
|
|
// Because an empty string is not valid YAML but we want to support it anyway
|
||
|
|
if is_yaml_empty(yaml) {
|
||
|
|
return Ok(serde_yaml::from_str(
|
||
|
|
"arbitrary_field_that_will_not_block_the_parser: true",
|
||
|
|
)?);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(serde_yaml::from_str(yaml)?)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl TryFrom<YAMLConfig> for ParsedConfig {
|
||
|
|
type Error = anyhow::Error;
|
||
|
|
|
||
|
|
fn try_from(yaml_config: YAMLConfig) -> Result<Self, Self::Error> {
|
||
|
|
Ok(Self {
|
||
|
|
label: yaml_config.label,
|
||
|
|
|
||
|
|
use_standard_includes: yaml_config.use_standard_includes,
|
||
|
|
includes: yaml_config.includes,
|
||
|
|
extra_includes: yaml_config.extra_includes,
|
||
|
|
excludes: yaml_config.excludes,
|
||
|
|
extra_excludes: yaml_config.extra_excludes,
|
||
|
|
|
||
|
|
filter_class: yaml_config.filter_class,
|
||
|
|
filter_exec: yaml_config.filter_exec,
|
||
|
|
filter_os: yaml_config.filter_os,
|
||
|
|
filter_title: yaml_config.filter_title,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
use std::convert::TryInto;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn conversion_to_parsed_config_works_correctly() {
|
||
|
|
let config = YAMLConfig::parse_from_str(
|
||
|
|
r#"
|
||
|
|
label: "test"
|
||
|
|
|
||
|
|
use_standard_includes: true
|
||
|
|
includes: ["test1"]
|
||
|
|
extra_includes: ["test2"]
|
||
|
|
excludes: ["test3"]
|
||
|
|
extra_excludes: ["test4"]
|
||
|
|
|
||
|
|
filter_class: "test5"
|
||
|
|
filter_exec: "test6"
|
||
|
|
filter_os: "test7"
|
||
|
|
filter_title: "test8"
|
||
|
|
"#,
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
let parsed_config: ParsedConfig = config.try_into().unwrap();
|
||
|
|
|
||
|
|
assert_eq!(
|
||
|
|
parsed_config,
|
||
|
|
ParsedConfig {
|
||
|
|
label: Some("test".to_string()),
|
||
|
|
use_standard_includes: Some(true),
|
||
|
|
includes: Some(vec!["test1".to_string()]),
|
||
|
|
extra_includes: Some(vec!["test2".to_string()]),
|
||
|
|
excludes: Some(vec!["test3".to_string()]),
|
||
|
|
extra_excludes: Some(vec!["test4".to_string()]),
|
||
|
|
|
||
|
|
filter_class: Some("test5".to_string()),
|
||
|
|
filter_exec: Some("test6".to_string()),
|
||
|
|
filter_os: Some("test7".to_string()),
|
||
|
|
filter_title: Some("test8".to_string()),
|
||
|
|
}
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|