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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
use serde::{Deserialize, Serialize};
use figment::{
providers::{Format, Toml},
value::{Dict, Map},
Error, Figment, Metadata, Profile, Provider,
};
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct Config {
pub user: User,
pub server: Server,
pub settings: Settings,
}
impl From<Config> for irc::client::prelude::Config {
fn from(input: Config) -> Self {
Self {
nickname: Some(input.user.nickname),
username: Some(input.user.username),
realname: Some(input.user.realname),
nick_password: input.user.password,
server: Some(input.server.hostname),
port: Some(input.server.port),
use_tls: Some(input.server.tls),
channels: input.server.channels,
..irc::client::prelude::Config::default()
}
}
}
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct User {
pub nickname: String,
pub username: String,
#[serde(default)]
pub password: Option<String>,
pub realname: String,
}
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct Server {
pub hostname: String,
#[serde(default = "default_port")]
pub port: u16,
#[serde(default = "default_tls")]
pub tls: bool,
#[serde(default)]
pub sasl: bool,
#[serde(default)]
pub channels: Vec<String>,
}
const fn default_port() -> u16 {
6697
}
const fn default_tls() -> bool {
true
}
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Deserialize, Serialize)]
pub struct Settings {
#[serde(default = "default_prefix")]
pub prefix: char,
// pub wa_api_key: String,
}
const fn default_prefix() -> char {
':'
}
impl Config {
// Allow the configuration to be extracted from any `Provider`.
pub fn from<T: Provider>(provider: T) -> Result<Config, Error> {
Figment::from(provider).extract()
}
// Provide a default provider, a `Figment`.
pub fn figment() -> Figment {
use figment::providers::Env;
let figment = Figment::new();
#[cfg(debug_assertions)]
const PROFILE: &str = "debug";
#[cfg(not(debug_assertions))]
const PROFILE: &str = "release";
figment
.merge(Toml::file("config.toml").nested())
.merge(Toml::file("config.debug.toml").nested())
.merge(Env::prefixed("CATINATOR_").split('_'))
.select(PROFILE)
}
}
// Make `Config` a provider itself for composability.
impl Provider for Config {
fn metadata(&self) -> Metadata {
Metadata::named("Library Config")
}
fn data(&self) -> Result<Map<Profile, Dict>, Error> {
figment::providers::Serialized::defaults(self).data()
}
fn profile(&self) -> Option<Profile> {
Some(Profile::Default)
}
}
|
