cfg_rs/source/
yaml.rs

1//! Yaml config source.
2use yaml_rust2::YamlLoader;
3
4use super::{memory::ConfigSourceBuilder, ConfigSourceAdaptor, ConfigSourceParser};
5use crate::ConfigError;
6
7impl ConfigSourceAdaptor for yaml_rust2::Yaml {
8    fn convert_source(self, source: &mut ConfigSourceBuilder<'_>) -> Result<(), ConfigError> {
9        match self {
10            yaml_rust2::Yaml::Real(v) => source.insert(v),
11            yaml_rust2::Yaml::Integer(v) => source.insert(v),
12            yaml_rust2::Yaml::String(v) => source.insert(v),
13            yaml_rust2::Yaml::Boolean(v) => source.insert(v),
14            yaml_rust2::Yaml::Array(v) => source.insert_array(v)?,
15            yaml_rust2::Yaml::Hash(v) => source.insert_map(
16                v.into_iter()
17                    .filter_map(|(k, v)| k.as_str().map(|k| (k.to_string(), v))),
18            )?,
19            _ => {}
20        }
21        Ok(())
22    }
23}
24
25impl ConfigSourceAdaptor for Yaml {
26    fn convert_source(self, builder: &mut ConfigSourceBuilder<'_>) -> Result<(), ConfigError> {
27        for y in self.0 {
28            y.convert_source(builder)?;
29        }
30        Ok(())
31    }
32}
33
34impl ConfigSourceParser for Yaml {
35    type Adaptor = Yaml;
36    fn parse_source(content: &str) -> Result<Self::Adaptor, ConfigError> {
37        Ok(Yaml(
38            YamlLoader::load_from_str(content).map_err(ConfigError::from_cause)?,
39        ))
40    }
41
42    fn file_extensions() -> Vec<&'static str> {
43        vec!["yaml", "yml"]
44    }
45}
46
47/// Yaml source.
48#[allow(missing_debug_implementations)]
49pub struct Yaml(Vec<yaml_rust2::Yaml>);
50
51#[cfg_attr(coverage_nightly, coverage(off))]
52#[cfg(test)]
53mod test {
54    use super::*;
55    use crate::{source::inline_source, test::source_test_suit};
56
57    #[test]
58    #[allow(unused_qualifications)]
59    fn inline_test() -> Result<(), ConfigError> {
60        source_test_suit(inline_source!("../../app.yaml")?)
61    }
62}
cfg_rs/source/yaml.rs

cfg_rs/source/
yaml.rs
