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
|
pub mod args;
use serde::{Deserialize, Serialize};
use std::{collections::BTreeMap, ops::Deref};
use figment::{
providers::{Format, Toml},
value::{Dict, Map},
Error, Figment, Metadata, Profile, Provider,
};
use anyhow::{Context, Result};
/// Configuration for the Bot
#[derive(Clone, Debug, Deserialize, Serialize)]
// pub struct Config();
// TODO make forge optional
pub struct Config {
#[serde(flatten)]
config: BTreeMap<String, ForgeConfig>,
}
impl Deref for Config {
type Target = BTreeMap<String, ForgeConfig>;
fn deref(&self) -> &Self::Target {
&self.config
}
}
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() -> Result<Figment> {
use figment::providers::Env;
let dirs = xdg::BaseDirectories::with_prefix(env!("CARGO_PKG_NAME")).unwrap();
Ok(Figment::from(Toml::file(
dirs.place_config_file("config.toml")
.context("failed to create config directory")?,
))
.merge(Toml::file(
dirs.place_config_file("config.yaml")
.context("failed to create config directory")?,
))
.merge(Env::prefixed("GTREE_")))
}
}
// 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()
}
}
#[derive(Clone, Debug, Deserialize, Serialize)]
#[serde(tag = "type")]
pub enum ForgeConfig {
#[serde(alias = "gitlab")]
Gitlab(crate::forge::gitlab::config::Gitlab),
}
pub trait ForgeConfigTrait {
fn root(&self) -> &str;
}
impl Deref for ForgeConfig {
type Target = dyn ForgeConfigTrait;
fn deref(&self) -> &Self::Target {
match self {
ForgeConfig::Gitlab(conf) => conf,
}
}
}
|
