use serde::Deserialize;
#[derive(Debug, Clone, Deserialize)]
pub struct UpdateCheckConfig {
#[serde(default = "default_enabled")]
pub enabled: bool,
#[serde(default = "default_endpoint")]
pub endpoint: String,
#[serde(default = "default_timeout_ms")]
pub timeout_ms: u64,
}
fn default_enabled() -> bool {
true
}
fn default_endpoint() -> String {
"https://velesdb.com/api/check".to_string()
}
fn default_timeout_ms() -> u64 {
2000
}
impl Default for UpdateCheckConfig {
fn default() -> Self {
Self {
enabled: default_enabled(),
endpoint: default_endpoint(),
timeout_ms: default_timeout_ms(),
}
}
}
impl UpdateCheckConfig {
#[must_use]
pub fn is_enabled(&self) -> bool {
if let Ok(val) = std::env::var("VELESDB_NO_UPDATE_CHECK") {
if is_truthy(&val) {
return false;
}
}
if let Ok(val) = std::env::var("VELESDB_UPDATE_CHECK") {
return is_truthy(&val);
}
self.enabled
}
}
fn is_truthy(val: &str) -> bool {
!matches!(val.to_lowercase().as_str(), "0" | "false" | "no" | "off")
}
#[cfg(test)]
mod tests {
use super::*;
use serial_test::serial;
#[test]
#[serial(env)]
fn test_env_var_disables_update_check() {
std::env::set_var("VELESDB_NO_UPDATE_CHECK", "1");
let config = UpdateCheckConfig::default();
assert!(!config.is_enabled());
std::env::remove_var("VELESDB_NO_UPDATE_CHECK");
}
#[test]
#[serial(env)]
fn test_env_var_overrides_config() {
std::env::set_var("VELESDB_NO_UPDATE_CHECK", "1");
let config = UpdateCheckConfig {
enabled: true, ..Default::default()
};
assert!(!config.is_enabled());
std::env::remove_var("VELESDB_NO_UPDATE_CHECK");
}
#[test]
#[serial(env)]
fn test_config_disabled() {
std::env::remove_var("VELESDB_NO_UPDATE_CHECK");
std::env::remove_var("VELESDB_UPDATE_CHECK");
let config = UpdateCheckConfig {
enabled: false,
..Default::default()
};
assert!(!config.is_enabled());
}
#[test]
#[serial(env)]
fn test_default_enabled() {
std::env::remove_var("VELESDB_NO_UPDATE_CHECK");
std::env::remove_var("VELESDB_UPDATE_CHECK");
let config = UpdateCheckConfig::default();
assert!(config.is_enabled());
}
#[test]
fn test_default_endpoint() {
let config = UpdateCheckConfig::default();
assert_eq!(config.endpoint, "https://velesdb.com/api/check");
}
#[test]
fn test_default_timeout() {
let config = UpdateCheckConfig::default();
assert_eq!(config.timeout_ms, 2000);
}
}