1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
pub use anyhow::Result;

pub trait ConfigParser {
    fn parse(&mut self, value: toml::Value) -> Result<()>;
}

pub use config_parser_derive::ConfigParse;

#[macro_export]
macro_rules! config_parser_impl {
    ($($t:ty),+) => {
        $(
            impl ConfigParser for $t {
                fn parse(&mut self, value: toml::Value) -> Result<()> {
                    *self = value.try_into::<$t>()?;
                    Ok(())
                }
            }
        )*
    };
}

impl<'de, T> ConfigParser for Option<T>
where
    T: serde::de::Deserialize<'de>,
{
    fn parse(&mut self, value: toml::Value) -> Result<()> {
        if let Ok(value) = value.try_into::<T>() {
            *self = Some(value);
        }
        Ok(())
    }
}

impl<'de, T> ConfigParser for Vec<T>
where
    T: serde::de::Deserialize<'de>,
{
    fn parse(&mut self, value: toml::Value) -> Result<()> {
        if let toml::Value::Array(array) = value {
            let result: Result<Vec<_>> =
                array.into_iter().map(|e| Ok(e.try_into::<T>()?)).collect();
            match result {
                Err(err) => Err(err),
                Ok(value) => {
                    *self = value;
                    Ok(())
                }
            }
        } else {
            Err(anyhow::anyhow!(
                "config parsing error: expect a TOML::Array, receive {:?}",
                value
            ))
        }
    }
}

config_parser_impl!(
    String, usize, u128, u64, u32, u16, u8, isize, i128, i64, i32, i16, i8, f64, f32, bool, char
);