atagen

jelq

public

an http based roguelike rpg about jelqing

Star0
Watch0
159f0998263bdbc45fc96bfc40453947b3a44cc3
1
ssh://brewery@pls.lobotomise.me:33/atagen/jelq.git
src/config.rsRust51 lines1.5 KB
1
use std::{path::PathBuf, time::Duration};
2
3
use crate::game::oracle::OracleTemplate;
4
use crate::secret::Secret;
5
6
pub struct Config {
7
    pub addr: String,
8
    pub secret: Secret,
9
    pub max_runs: usize,
10
    pub idle_ttl: Duration,
11
    /// Where finished runs are kept. `None` keeps them in memory, which only tests want.
12
    pub db: Option<PathBuf>,
13
    pub(crate) oracle: OracleTemplate,
14
}
15
16
impl Default for Config {
17
    fn default() -> Self {
18
        Self {
19
            addr: "127.0.0.1:6969".into(),
20
            secret: Secret::random(),
21
            max_runs: 10_000,
22
            idle_ttl: Duration::from_secs(6 * 60 * 60),
23
            db: None,
24
            oracle: OracleTemplate::Rng,
25
        }
26
    }
27
}
28
29
impl Config {
30
    pub fn from_env() -> Self {
31
        let var = |name| std::env::var(name).ok();
32
        let defaults = Self::default();
33
        Self {
34
            addr: var("JELQD_ADDR").unwrap_or(defaults.addr),
35
            db: Some(
36
                var("JELQD_DB")
37
                    .unwrap_or_else(|| "jelqd.sqlite".into())
38
                    .into(),
39
            ),
40
            secret: var("JELQD_SECRET").map_or(defaults.secret, |s| Secret::new(s.into_bytes())),
41
            oracle: match var("JELQD_ORACLE").as_deref() {
42
                Some("clm") => OracleTemplate::Clm {
43
                    url: var("JELQD_CLM_URL")
44
                        .unwrap_or_else(|| "http://127.0.0.1:8700/v1/systemone".into()),
45
                },
46
                _ => defaults.oracle,
47
            },
48
            ..defaults
49
        }
50
    }
51
}