openws-server 0.1.2

WebSocket server with room-based messaging, private messages, HTTP broadcast, and configurable limits
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
115
116

use serde::Deserialize;

#[derive(Debug, Default, Deserialize, Clone)]
pub struct ServerSection {
    #[serde(default = "default_host")]
    pub host: String,
    #[serde(default = "default_port")]
    pub port: u16,
}

#[derive(Debug, Default, Deserialize, Clone)]
pub struct LimitsSection {
    #[serde(default = "default_max_clients")]
    pub max_clients: usize,
    #[serde(default = "default_max_rooms")]
    pub max_rooms: usize,
    #[serde(default = "default_max_message_size")]
    pub max_message_size: usize,
    #[serde(default = "default_max_username_len")]
    pub max_username_len: usize,
}

#[derive(Debug, Default, Deserialize, Clone)]
pub struct HeartbeatSection {
    #[serde(default = "default_ping_interval")]
    pub ping_interval_secs: u64,
    #[serde(default = "default_pong_timeout")]
    pub pong_timeout_secs: u64,
}

#[derive(Debug, Default, Deserialize, Clone)]
pub struct BroadcastSection {
    #[serde(default)]
    pub api_key: String,
}

#[derive(Debug, Deserialize, Clone)]
pub struct AppConfig {
    #[serde(default)]
    pub server: ServerSection,
    #[serde(default)]
    pub limits: LimitsSection,
    #[serde(default)]
    pub heartbeat: HeartbeatSection,
    #[serde(default)]
    pub broadcast: BroadcastSection,
}

fn default_host() -> String {
    "0.0.0.0".into()
}
fn default_port() -> u16 {
    3000
}
fn default_max_clients() -> usize {
    1000
}
fn default_max_rooms() -> usize {
    100
}
fn default_max_message_size() -> usize {
    8192
}
fn default_max_username_len() -> usize {
    32
}
fn default_ping_interval() -> u64 {
    30
}
fn default_pong_timeout() -> u64 {
    10
}

pub fn load_config() -> Result<AppConfig, config::ConfigError> {
    let c = config::Config::builder()
        .add_source(config::File::with_name("openws").required(false))
        .add_source(
            config::Environment::with_prefix("OPENWS")
                .separator("_")
                .try_parsing(true),
        )
        .build()?;

    let mut cfg: AppConfig = c.try_deserialize()?;

    if let Ok(v) = std::env::var("OPENWS_PORT") {
        if let Ok(p) = v.parse() {
            cfg.server.port = p;
        }
    }
    if let Ok(v) = std::env::var("OPENWS_MAX_CLIENTS") {
        if let Ok(n) = v.parse() {
            cfg.limits.max_clients = n;
        }
    }
    if let Ok(v) = std::env::var("OPENWS_PING_INTERVAL") {
        if let Ok(s) = v.parse() {
            cfg.heartbeat.ping_interval_secs = s;
        }
    }
    if let Ok(v) = std::env::var("OPENWS_API_KEY") {
        cfg.broadcast.api_key = v;
    }

    Ok(cfg)
}

pub fn mask_api_key(key: &str) -> String {
    if key.is_empty() {
        return "(none)".into();
    }
    if key.len() <= 4 {
        return "****".into();
    }
    format!("{}****", &key[..4])
}