rust_web_server/entry_point/
mod.rs

1#[cfg(test)]
2mod tests;
3pub mod command_line_args;
4pub mod config_file;
5pub mod environment_variables;
6
7
8use std::{env};
9use crate::virtual_host::VirtualHostConfig;
10
11use crate::entry_point::command_line_args::{override_environment_variables_from_command_line_args};
12use crate::entry_point::config_file::override_environment_variables_from_config;
13use crate::entry_point::environment_variables::read_system_environment_variables;
14
15#[derive(PartialEq, Eq, Clone, Debug)]
16pub struct Config {}
17
18impl Config {
19    pub const RWS_CONFIG_IP: &'static str = "RWS_CONFIG_IP";
20    /// Default is `0.0.0.0` so the server is reachable inside containers and K8s pods.
21    /// For local development you can override to `127.0.0.1` via env var or config file.
22    pub const RWS_CONFIG_IP_DEFAULT_VALUE: &'static str = "0.0.0.0";
23
24    /// Log format: `"combined"` (default, Combined Log Format) or `"json"` (structured JSON).
25    pub const RWS_CONFIG_LOG_FORMAT: &'static str = "RWS_CONFIG_LOG_FORMAT";
26    pub const RWS_CONFIG_LOG_FORMAT_DEFAULT_VALUE: &'static str = "json";
27
28    pub const RWS_CONFIG_PORT: &'static str = "RWS_CONFIG_PORT";
29    pub const RWS_CONFIG_PORT_DEFAULT_VALUE: &'static str = "7878";
30
31    pub const RWS_CONFIG_THREAD_COUNT: &'static str = "RWS_CONFIG_THREAD_COUNT";
32    pub const RWS_CONFIG_THREAD_COUNT_DEFAULT_VALUE: &'static str = "200";
33
34    pub const RWS_CONFIG_CORS_ALLOW_ALL: &'static str = "RWS_CONFIG_CORS_ALLOW_ALL";
35    pub const RWS_CONFIG_CORS_ALLOW_ALL_DEFAULT_VALUE: &'static str = "true";
36
37    pub const RWS_CONFIG_CORS_ALLOW_ORIGINS: &'static str = "RWS_CONFIG_CORS_ALLOW_ORIGINS";
38    pub const RWS_CONFIG_CORS_ALLOW_ORIGINS_DEFAULT_VALUE: &'static str = "";
39
40    pub const RWS_CONFIG_CORS_ALLOW_CREDENTIALS: &'static str = "RWS_CONFIG_CORS_ALLOW_CREDENTIALS";
41    pub const RWS_CONFIG_CORS_ALLOW_CREDENTIALS_DEFAULT_VALUE: &'static str = "";
42
43    pub const RWS_CONFIG_CORS_ALLOW_HEADERS: &'static str = "RWS_CONFIG_CORS_ALLOW_HEADERS";
44    pub const RWS_CONFIG_CORS_ALLOW_HEADERS_DEFAULT_VALUE: &'static str = "";
45
46    pub const RWS_CONFIG_CORS_ALLOW_METHODS: &'static str = "RWS_CONFIG_CORS_ALLOW_METHODS";
47    pub const RWS_CONFIG_CORS_ALLOW_METHODS_DEFAULT_VALUE: &'static str = "";
48
49    pub const RWS_CONFIG_CORS_EXPOSE_HEADERS: &'static str = "RWS_CONFIG_CORS_EXPOSE_HEADERS";
50    pub const RWS_CONFIG_CORS_EXPOSE_HEADERS_DEFAULT_VALUE: &'static str = "";
51
52    pub const RWS_CONFIG_CORS_MAX_AGE: &'static str = "RWS_CONFIG_CORS_MAX_AGE";
53    pub const RWS_CONFIG_CORS_MAX_AGE_DEFAULT_VALUE: &'static str = "86400";
54
55    pub const RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES: &'static str = "RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES";
56    pub const RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES_DEFAULT_VALUE: &'static str = "10000";
57
58    pub const RWS_CONFIG_TLS_CERT_FILE: &'static str = "RWS_CONFIG_TLS_CERT_FILE";
59    pub const RWS_CONFIG_TLS_CERT_FILE_DEFAULT_VALUE: &'static str = "";
60
61    pub const RWS_CONFIG_TLS_KEY_FILE: &'static str = "RWS_CONFIG_TLS_KEY_FILE";
62    pub const RWS_CONFIG_TLS_KEY_FILE_DEFAULT_VALUE: &'static str = "";
63
64    /// Path to a PEM-encoded CA certificate used to verify client certificates (mTLS).
65    /// When set, the TLS handshake requires a valid client certificate signed by this CA.
66    /// Connections without a valid cert are rejected at the TLS layer (before any HTTP processing).
67    pub const RWS_CONFIG_TLS_CLIENT_CA_FILE: &'static str = "RWS_CONFIG_TLS_CLIENT_CA_FILE";
68    pub const RWS_CONFIG_TLS_CLIENT_CA_FILE_DEFAULT_VALUE: &'static str = "";
69
70    /// Directory containing Tera HTML templates (default: `"templates"`). Requires `tera` feature.
71    pub const RWS_CONFIG_TEMPLATE_DIR: &'static str = "RWS_CONFIG_TEMPLATE_DIR";
72    pub const RWS_CONFIG_TEMPLATE_DIR_DEFAULT_VALUE: &'static str = "templates";
73
74    /// When non-empty, a plain-HTTP listener on this port redirects all requests to HTTPS.
75    /// Set to e.g. `"80"` when running on standard ports. Requires TLS to be configured.
76    pub const RWS_CONFIG_HTTP_REDIRECT_PORT: &'static str = "RWS_CONFIG_HTTP_REDIRECT_PORT";
77    pub const RWS_CONFIG_HTTP_REDIRECT_PORT_DEFAULT_VALUE: &'static str = "";
78
79    // ── ACME (Automatic Certificate Management Environment) ───────────────────
80
81    /// Comma-separated list of domain names to obtain a certificate for.
82    /// Setting this activates ACME at startup. Example: `"example.com,www.example.com"`
83    pub const RWS_CONFIG_ACME_DOMAINS: &'static str = "RWS_CONFIG_ACME_DOMAINS";
84    pub const RWS_CONFIG_ACME_DOMAINS_DEFAULT_VALUE: &'static str = "";
85
86    /// Contact email sent to the CA. Recommended but not required.
87    pub const RWS_CONFIG_ACME_EMAIL: &'static str = "RWS_CONFIG_ACME_EMAIL";
88    pub const RWS_CONFIG_ACME_EMAIL_DEFAULT_VALUE: &'static str = "";
89
90    /// Set to `"true"` to use the Let's Encrypt staging environment (for testing).
91    pub const RWS_CONFIG_ACME_STAGING: &'static str = "RWS_CONFIG_ACME_STAGING";
92    pub const RWS_CONFIG_ACME_STAGING_DEFAULT_VALUE: &'static str = "false";
93
94    /// Custom ACME directory URL. Defaults to Let's Encrypt production.
95    pub const RWS_CONFIG_ACME_DIRECTORY: &'static str = "RWS_CONFIG_ACME_DIRECTORY";
96    pub const RWS_CONFIG_ACME_DIRECTORY_DEFAULT_VALUE: &'static str = "";
97
98    /// Where to write the provisioned certificate chain (PEM). Defaults to `RWS_CONFIG_TLS_CERT_FILE`.
99    pub const RWS_CONFIG_ACME_CERT_PATH: &'static str = "RWS_CONFIG_ACME_CERT_PATH";
100    pub const RWS_CONFIG_ACME_CERT_PATH_DEFAULT_VALUE: &'static str = "";
101
102    /// Where to write the certificate's private key (PEM). Defaults to `RWS_CONFIG_TLS_KEY_FILE`.
103    pub const RWS_CONFIG_ACME_KEY_PATH: &'static str = "RWS_CONFIG_ACME_KEY_PATH";
104    pub const RWS_CONFIG_ACME_KEY_PATH_DEFAULT_VALUE: &'static str = "";
105
106    /// Port for the temporary HTTP-01 challenge server (default 80).
107    /// Must be reachable from the internet on port 80. Not used with DNS-01.
108    pub const RWS_CONFIG_ACME_CHALLENGE_PORT: &'static str = "RWS_CONFIG_ACME_CHALLENGE_PORT";
109    pub const RWS_CONFIG_ACME_CHALLENGE_PORT_DEFAULT_VALUE: &'static str = "80";
110
111    /// Renew when fewer than this many days remain on the certificate (default 30).
112    pub const RWS_CONFIG_ACME_RENEW_BEFORE_DAYS: &'static str = "RWS_CONFIG_ACME_RENEW_BEFORE_DAYS";
113    pub const RWS_CONFIG_ACME_RENEW_BEFORE_DAYS_DEFAULT_VALUE: &'static str = "30";
114
115    /// Path to persist the ACME account key between restarts (default `acme_account.key`).
116    pub const RWS_CONFIG_ACME_ACCOUNT_KEY_PATH: &'static str = "RWS_CONFIG_ACME_ACCOUNT_KEY_PATH";
117    pub const RWS_CONFIG_ACME_ACCOUNT_KEY_PATH_DEFAULT_VALUE: &'static str = "acme_account.key";
118
119
120    pub const RWS_DEFAULT_IP: &'static str = "127.0.0.1";
121    pub const RWS_DEFAULT_PORT: &'static i32 = &7878;
122    pub const RWS_DEFAULT_THREAD_COUNT: &'static i32 = &200;
123    pub const RWS_DEFAULT_REQUEST_ALLOCATION_SIZE_IN_BYTES: &'static i64 = &10000;
124
125
126}
127
128pub fn bootstrap() {
129    read_system_environment_variables();
130    override_environment_variables_from_config(None);
131    override_environment_variables_from_command_line_args();
132}
133
134pub fn set_default_values() {
135    println!("  Initializing default values");
136
137    let is_var_set = env::var(Config::RWS_CONFIG_IP).is_ok();
138    if !is_var_set {
139        env::set_var(Config::RWS_CONFIG_IP, Config::RWS_CONFIG_IP_DEFAULT_VALUE);
140        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_IP, Config::RWS_CONFIG_IP_DEFAULT_VALUE);
141    } else {
142        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_IP);
143    }
144
145
146    let is_var_set = env::var(Config::RWS_CONFIG_PORT).is_ok();
147    if !is_var_set {
148        env::set_var(Config::RWS_CONFIG_PORT, Config::RWS_CONFIG_PORT_DEFAULT_VALUE);
149        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_PORT, Config::RWS_CONFIG_PORT_DEFAULT_VALUE);
150    } else {
151        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_PORT);
152    }
153
154    let is_var_set = env::var(Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES).is_ok();
155    if !is_var_set {
156        env::set_var(Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES, Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES_DEFAULT_VALUE);
157        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES, Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES_DEFAULT_VALUE);
158    } else {
159        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES);
160    }
161
162
163    let is_var_set = env::var(Config::RWS_CONFIG_THREAD_COUNT).is_ok();
164    if !is_var_set {
165        env::set_var(Config::RWS_CONFIG_THREAD_COUNT, Config::RWS_CONFIG_THREAD_COUNT_DEFAULT_VALUE);
166        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_THREAD_COUNT, Config::RWS_CONFIG_THREAD_COUNT_DEFAULT_VALUE);
167    } else {
168        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_THREAD_COUNT);
169    }
170
171    let is_var_set = env::var(Config::RWS_CONFIG_CORS_ALLOW_ALL).is_ok();
172    if !is_var_set {
173        env::set_var(Config::RWS_CONFIG_CORS_ALLOW_ALL, Config::RWS_CONFIG_CORS_ALLOW_ALL_DEFAULT_VALUE);
174        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_ALLOW_ALL, Config::RWS_CONFIG_CORS_ALLOW_ALL_DEFAULT_VALUE);
175    } else {
176        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_ALLOW_ALL);
177    }
178
179
180    let is_var_set = env::var(Config::RWS_CONFIG_CORS_ALLOW_ORIGINS).is_ok();
181    if !is_var_set {
182        env::set_var(Config::RWS_CONFIG_CORS_ALLOW_ORIGINS, Config::RWS_CONFIG_CORS_ALLOW_ORIGINS_DEFAULT_VALUE);
183        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_ALLOW_ORIGINS, Config::RWS_CONFIG_CORS_ALLOW_ORIGINS_DEFAULT_VALUE);
184    } else {
185        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_ALLOW_ORIGINS);
186    }
187
188    let is_var_set = env::var(Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS).is_ok();
189    if !is_var_set {
190        env::set_var(Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS, Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS_DEFAULT_VALUE);
191        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS, Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS_DEFAULT_VALUE);
192    } else {
193        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_ALLOW_CREDENTIALS);
194    }
195
196    let is_var_set = env::var(Config::RWS_CONFIG_CORS_ALLOW_HEADERS).is_ok();
197    if !is_var_set {
198        env::set_var(Config::RWS_CONFIG_CORS_ALLOW_HEADERS, Config::RWS_CONFIG_CORS_ALLOW_HEADERS_DEFAULT_VALUE);
199        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_ALLOW_HEADERS, Config::RWS_CONFIG_CORS_ALLOW_HEADERS_DEFAULT_VALUE);
200    } else {
201        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_ALLOW_HEADERS);
202    }
203
204
205    let is_var_set = env::var(Config::RWS_CONFIG_CORS_ALLOW_METHODS).is_ok();
206    if !is_var_set {
207        env::set_var(Config::RWS_CONFIG_CORS_ALLOW_METHODS, Config::RWS_CONFIG_CORS_ALLOW_METHODS_DEFAULT_VALUE);
208        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_ALLOW_METHODS, Config::RWS_CONFIG_CORS_ALLOW_METHODS_DEFAULT_VALUE);
209    } else {
210        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_ALLOW_METHODS);
211    }
212
213    let is_var_set = env::var(Config::RWS_CONFIG_CORS_EXPOSE_HEADERS).is_ok();
214    if !is_var_set {
215        env::set_var(Config::RWS_CONFIG_CORS_EXPOSE_HEADERS, Config::RWS_CONFIG_CORS_EXPOSE_HEADERS_DEFAULT_VALUE);
216        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_EXPOSE_HEADERS, Config::RWS_CONFIG_CORS_EXPOSE_HEADERS_DEFAULT_VALUE);
217    } else {
218        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_EXPOSE_HEADERS);
219    }
220
221    let is_var_set = env::var(Config::RWS_CONFIG_CORS_MAX_AGE).is_ok();
222    if !is_var_set {
223        env::set_var(Config::RWS_CONFIG_CORS_MAX_AGE, Config::RWS_CONFIG_CORS_MAX_AGE_DEFAULT_VALUE);
224        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_CORS_MAX_AGE, Config::RWS_CONFIG_CORS_MAX_AGE_DEFAULT_VALUE);
225    } else {
226        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_CORS_MAX_AGE);
227    }
228
229
230    let is_var_set = env::var(Config::RWS_CONFIG_TLS_CERT_FILE).is_ok();
231    if !is_var_set {
232        env::set_var(Config::RWS_CONFIG_TLS_CERT_FILE, Config::RWS_CONFIG_TLS_CERT_FILE_DEFAULT_VALUE);
233        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_TLS_CERT_FILE, Config::RWS_CONFIG_TLS_CERT_FILE_DEFAULT_VALUE);
234    } else {
235        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_TLS_CERT_FILE);
236    }
237
238    let is_var_set = env::var(Config::RWS_CONFIG_TLS_KEY_FILE).is_ok();
239    if !is_var_set {
240        env::set_var(Config::RWS_CONFIG_TLS_KEY_FILE, Config::RWS_CONFIG_TLS_KEY_FILE_DEFAULT_VALUE);
241        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_TLS_KEY_FILE, Config::RWS_CONFIG_TLS_KEY_FILE_DEFAULT_VALUE);
242    } else {
243        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_TLS_KEY_FILE);
244    }
245
246    let is_var_set = env::var(Config::RWS_CONFIG_HTTP_REDIRECT_PORT).is_ok();
247    if !is_var_set {
248        env::set_var(Config::RWS_CONFIG_HTTP_REDIRECT_PORT, Config::RWS_CONFIG_HTTP_REDIRECT_PORT_DEFAULT_VALUE);
249        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_HTTP_REDIRECT_PORT, Config::RWS_CONFIG_HTTP_REDIRECT_PORT_DEFAULT_VALUE);
250    } else {
251        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_HTTP_REDIRECT_PORT);
252    }
253
254    let is_var_set = env::var(Config::RWS_CONFIG_LOG_FORMAT).is_ok();
255    if !is_var_set {
256        env::set_var(Config::RWS_CONFIG_LOG_FORMAT, Config::RWS_CONFIG_LOG_FORMAT_DEFAULT_VALUE);
257        println!("    Default value  for '{}' is '{}'", Config::RWS_CONFIG_LOG_FORMAT, Config::RWS_CONFIG_LOG_FORMAT_DEFAULT_VALUE);
258    } else {
259        println!("    There is an environment variable  for '{}', default value won't be set", Config::RWS_CONFIG_LOG_FORMAT);
260    }
261
262    println!("  End of initializing default values\n");
263}
264
265
266pub fn get_ip_port_thread_count() -> (String, i32, i32) {
267    let mut ip : String = Config::RWS_CONFIG_IP_DEFAULT_VALUE.to_string();
268    let mut port: i32 = *Config::RWS_DEFAULT_PORT;
269    let mut thread_count: i32 = *Config::RWS_DEFAULT_THREAD_COUNT;
270
271    let boxed_ip = env::var(Config::RWS_CONFIG_IP);
272    if boxed_ip.is_ok() {
273        ip = boxed_ip.unwrap()
274    }
275
276    let boxed_port = env::var(Config::RWS_CONFIG_PORT);
277    if boxed_port.is_ok() {
278        let _port = boxed_port.unwrap();
279        let boxed_parse = _port.parse::<i32>();
280        if boxed_parse.is_ok() {
281            port = boxed_parse.unwrap();
282        } else {
283            eprintln!("unable to parse port value, expected number, got {}, variable: {}",
284                      _port, Config::RWS_CONFIG_PORT);
285        }
286    } else {
287        eprintln!("unable to parse port value, variable: {}", Config::RWS_CONFIG_PORT);
288    }
289
290    let boxed_thread_count = env::var(Config::RWS_CONFIG_THREAD_COUNT);
291    if boxed_thread_count.is_ok() {
292        let _thread_count = boxed_thread_count.unwrap();
293        let boxed_parse = _thread_count.parse();
294        if boxed_parse.is_ok() {
295            thread_count = boxed_parse.unwrap()
296        } else {
297            eprintln!("unable to parse thread count value, expected number, got {}, variable: {}",
298                      thread_count, Config::RWS_CONFIG_THREAD_COUNT);
299        }
300
301    } else {
302        eprintln!("unable to parse thread count value, variable: {}", Config::RWS_CONFIG_THREAD_COUNT);
303    }
304
305    (ip, port, thread_count)
306}
307
308/// Read all `[[virtual_host]]` entries from config / env vars.
309///
310/// Each entry must have `RWS_CONFIG_VIRTUAL_HOST_{N}_DOMAIN` set; reading
311/// stops at the first missing index.  `cert_file` and `key_file` default to
312/// empty string if omitted.
313pub fn get_virtual_hosts() -> Vec<VirtualHostConfig> {
314    let mut hosts = Vec::new();
315    let mut i = 0usize;
316    loop {
317        match env::var(format!("RWS_CONFIG_VIRTUAL_HOST_{}_DOMAIN", i)) {
318            Err(_) => break,
319            Ok(domain) => {
320                let cert_file = env::var(format!("RWS_CONFIG_VIRTUAL_HOST_{}_CERT_FILE", i))
321                    .unwrap_or_default();
322                let key_file = env::var(format!("RWS_CONFIG_VIRTUAL_HOST_{}_KEY_FILE", i))
323                    .unwrap_or_default();
324                hosts.push(VirtualHostConfig { domain, cert_file, key_file });
325                i += 1;
326            }
327        }
328    }
329    hosts
330}
331
332pub fn get_request_allocation_size() -> i64 {
333    let mut request_allocation_size: i64 = *Config::RWS_DEFAULT_REQUEST_ALLOCATION_SIZE_IN_BYTES;
334
335    let boxed_port = env::var(Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES);
336    if boxed_port.is_ok() {
337        let _request_allocation_size = boxed_port.unwrap();
338        let boxed_parse = _request_allocation_size.parse::<i64>();
339        if boxed_parse.is_ok() {
340            request_allocation_size = boxed_parse.unwrap();
341        } else {
342            eprintln!("unable to parse port value, expected number, got {}, variable: {}",
343                      _request_allocation_size, Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES);
344        }
345    } else {
346        eprintln!("unable to parse request allocation size value, variable: {}", Config::RWS_CONFIG_REQUEST_ALLOCATION_SIZE_IN_BYTES);
347    }
348
349
350    request_allocation_size
351}
352
rust_web_server/entry_point/mod.rs

rust_web_server/entry_point/
mod.rs
