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
|
use bytesize::ByteSize;
use figment::{
error::Result,
providers::{Env, Format, Serialized, Toml},
value::{Dict, Map},
Figment, Metadata, Profile, Provider,
};
use serde::{Deserialize, Serialize};
use std::path::PathBuf;
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Config {
pub quota: ByteSize,
pub pool_name: String,
pub mountpoint: PathBuf,
}
impl Default for Config {
fn default() -> Self {
Config {
quota: ByteSize::gb(16),
pool_name: String::from("pool"),
mountpoint: PathBuf::from("/svr"),
}
}
}
impl Config {
pub fn from<T: Provider>(provider: T) -> Result<Config> {
Figment::from(provider).extract()
}
pub fn figment() -> Figment {
Figment::from(Config::default())
.merge(Toml::file(Env::var_or("ZFS_CONFIG", "ZFS.toml")).nested())
.merge(Env::prefixed("ZFS_").global())
}
}
impl Provider for Config {
fn metadata(&self) -> Metadata {
Metadata::named("ZFS Config")
}
fn data(&self) -> Result<Map<Profile, Dict>> {
Serialized::defaults(Config::default()).data()
}
}
|
