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
112
113
114

use std::time::Duration;

use feature_probe_server_sdk::Url;
use serde::Deserialize;
use thiserror::Error;

#[derive(Debug, Error, Deserialize)]
pub enum FPServerError {
    #[error("not found {0}")]
    NotFound(String),
    #[error("user base64 decode error")]
    UserDecodeError,
    #[error("config error: {0}")]
    ConfigError(String),
}

#[derive(Debug, Clone)]
pub struct ServerConfig {
    pub toggles_url: Url,
    pub events_url: Url,
    pub keys_url: Option<Url>,
    pub refresh_interval: Duration,
    pub server_sdk_key: Option<String>,
    pub client_sdk_key: Option<String>,
    pub server_port: u16,
}

impl ServerConfig {
    pub fn try_parse(config: config::Config) -> Result<ServerConfig, FPServerError> {
        let toggles_url = match config.get_string("toggles_url") {
            Err(_) => {
                return Err(FPServerError::ConfigError(
                    "NOT SET FP_TOGGLES_URL".to_owned(),
                ))
            }
            Ok(url) => match Url::parse(&url) {
                Err(e) => {
                    return Err(FPServerError::ConfigError(format!(
                        "INVALID FP_TOGGLES_URL: {}",
                        e,
                    )))
                }
                Ok(u) => u,
            },
        };

        let events_url = match config.get_string("events_url") {
            Err(_) => {
                return Err(FPServerError::ConfigError(
                    "NOT SET FP_EVENTS_URL".to_owned(),
                ))
            }
            Ok(url) => match Url::parse(&url) {
                Err(e) => {
                    return Err(FPServerError::ConfigError(format!(
                        "INVALID FP_EVENTS_URL: {}",
                        e,
                    )))
                }
                Ok(u) => u,
            },
        };
        let client_sdk_key = config.get_string("client_sdk_key").ok();
        let server_sdk_key = config.get_string("server_sdk_key").ok();

        let keys_url = match config.get_string("keys_url") {
            Ok(url) => match Url::parse(&url) {
                Err(e) => {
                    return Err(FPServerError::ConfigError(format!(
                        "INVALID FP_KEYS_URL: {}",
                        e,
                    )))
                }
                Ok(u) => Some(u),
            },
            Err(_) => {
                if client_sdk_key.is_none() {
                    return Err(FPServerError::ConfigError(
                        "NOT SET FP_CLIENT_SDK_KEY".to_owned(),
                    ));
                }
                if server_sdk_key.is_none() {
                    return Err(FPServerError::ConfigError(
                        "NOT SET FP_SERVER_SDK_KEY".to_owned(),
                    ));
                }
                None
            }
        };

        let refresh_interval = match config.get_int("refresh_seconds") {
            Err(_) => {
                return Err(FPServerError::ConfigError(
                    "NOT SET FP_REFRESH_SECONDS".to_owned(),
                ))
            }
            Ok(interval) => Duration::from_secs(interval as u64),
        };
        let server_port = match config.get_int("server_port") {
            Err(_) => 9000, // default port
            Ok(server_port) => server_port as u16,
        };

        Ok(ServerConfig {
            toggles_url,
            events_url,
            keys_url,
            refresh_interval,
            client_sdk_key,
            server_sdk_key,
            server_port,
        })
    }
}