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
use anyhow::Error;
use once_cell::sync::OnceCell;
use std::env;
use std::str::FromStr;
pub struct ConfigPatch<T> {
pre: &'static str,
post: OnceCell<T>,
}
impl<T> ConfigPatch<T> {
pub const fn new(pre: &'static str) -> Self {
Self {
pre,
post: OnceCell::new(),
}
}
pub const fn var(&self) -> &'static str {
self.pre
}
pub fn env_var(&self) -> Result<T, Error>
where
T: FromStr,
{
let s = env::var(self.pre)?;
let value = s.parse().map_err(|_err| {
Error::msg(format!("Can't parse {} variable from '{}'.", self.pre, s))
})?;
Ok(value)
}
pub fn offer(&self, value: T) {
if let Err(_err) = self.post.set(value) {
log::error!("Config value {} already overriden.", self.pre);
}
}
pub fn get<F, D>(&self, value: F, default: D) -> T
where
T: FromStr + Clone,
F: Fn() -> Option<T>,
D: Fn() -> T,
{
self.env_var()
.map_err(|err| {
log::error!("Default value for {} will be used: {}", self.pre, err);
})
.ok()
.or_else(value)
.or_else(|| self.post.get().cloned())
.unwrap_or_else(default)
}
}