use crate::Loggify;
use log;
use log::{Level, SetLoggerError};
pub struct LogBuilder {
exclude: Vec<String>,
level: Level,
time_format: String,
log_target: bool,
color: bool
}
impl LogBuilder {
pub fn new() -> Self {
Self::default()
}
pub fn add_exclude(mut self, name: String) -> Self {
self.exclude.push(name);
self
}
pub fn set_level(mut self, level: Level) -> Self {
self.level = level;
self
}
pub fn set_time_format(mut self, format: String) -> Self {
self.time_format = format;
self
}
pub fn set_log_target(mut self, state: bool) -> Self {
self.log_target = state;
self
}
pub fn disable_color(mut self) -> Self {
self.color = false;
self
}
pub fn build(self) -> Result<(), SetLoggerError> {
let logger = Loggify {
level: self.level,
exclude: self.exclude,
time_format: self.time_format,
log_target: self.log_target,
color: self.color
};
log::set_boxed_logger(Box::new(logger))?;
log::set_max_level(self.level.to_level_filter());
Ok(())
}
}
impl Default for LogBuilder {
fn default() -> Self {
Self {
level: Level::Info,
exclude: Vec::new(),
time_format: String::from("%d.%m.%Y %H:%M:%S"),
log_target: false,
color: std::env::var("LOGGIFY_COLOR")
.map(|x| x.parse().unwrap_or(true))
.unwrap_or(true)
}
}
}