27 lines
713 B
Rust
27 lines
713 B
Rust
|
use envconfig::Envconfig;
|
||
|
|
|
||
|
|
pub enum CuddleFetchPolicy {
|
||
|
|
Always,
|
||
|
|
Once,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Envconfig, Clone)]
|
||
|
|
pub struct CuddleConfig {
|
||
|
|
#[envconfig(from = "CUDDLE_FETCH_POLICY", default = "once")]
|
||
|
|
fetch_policy: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl CuddleConfig {
|
||
|
|
pub fn from_env() -> anyhow::Result<Self> {
|
||
|
|
CuddleConfig::init_from_env().map_err(|e| anyhow::Error::from(e))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn get_fetch_policy(&self) -> anyhow::Result<CuddleFetchPolicy> {
|
||
|
|
match self.fetch_policy.clone().to_lowercase().as_str() {
|
||
|
|
"always" => Ok(CuddleFetchPolicy::Always),
|
||
|
|
"once" => Ok(CuddleFetchPolicy::Once),
|
||
|
|
_ => Err(anyhow::anyhow!("could not parse fetch policy")),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|