controller/emgauwa-lib/src/utils.rs

48 lines
1.2 KiB
Rust
Raw Normal View History

2023-11-29 13:27:46 +00:00
use std::str::FromStr;
2023-11-30 00:43:56 +00:00
use chrono::Datelike;
use log::LevelFilter;
use simple_logger::SimpleLogger;
2023-12-05 00:42:19 +00:00
use crate::errors::EmgauwaError;
2023-11-30 00:43:56 +00:00
use crate::types::Weekday;
2023-12-05 00:42:19 +00:00
pub fn load_settings<T>(config_name: &str, env_prefix: &str) -> Result<T, EmgauwaError>
where
2023-11-27 11:49:40 +00:00
for<'de> T: serde::Deserialize<'de>,
{
2023-11-27 11:49:40 +00:00
let default_file = config::File::with_name(&format!("emgauwa-{}", config_name)).required(false);
config::Config::builder()
.add_source(default_file)
.add_source(
config::Environment::with_prefix(&format!("EMGAUWA_{}", env_prefix))
.prefix_separator("__")
.separator("__"),
)
2023-12-05 00:42:19 +00:00
.build()?
.try_deserialize::<T>()
2023-12-05 00:42:19 +00:00
.map_err(EmgauwaError::from)
2023-11-27 11:49:40 +00:00
}
2023-12-05 00:42:19 +00:00
pub fn init_logging(level: &str) -> Result<(), EmgauwaError> {
let log_level: LevelFilter = LevelFilter::from_str(level)
2023-12-05 00:51:06 +00:00
.map_err(|_| EmgauwaError::Other(format!("Invalid log level: {}", level)))?;
log::trace!("Log level set to {:?}", log_level);
SimpleLogger::new()
.with_level(log_level)
.init()
2023-12-05 00:51:06 +00:00
.map_err(|err| EmgauwaError::Other(format!("Failed to initialize logger: {}", err)))?;
2023-12-05 00:42:19 +00:00
Ok(())
}
2023-11-30 00:43:56 +00:00
pub fn get_weekday() -> Weekday {
(chrono::offset::Local::now()
.date_naive()
.weekday()
.number_from_monday()
- 1) as Weekday
}