Replace custom logging with log crate and custom format
Signed-off-by: trivernis <trivernis@protonmail.com>axtloss/rework-partitioning
parent
f9b775eff1
commit
f57e73e0c6
GPG Key ID:
DFFFCC2C7A02DB45
@ -0,0 +1,69 @@
|
|||||||
|
use crate::uwu;
|
||||||
|
use flexi_logger::{style, DeferredNow, LogSpecification, Logger};
|
||||||
|
use lazy_static::lazy_static;
|
||||||
|
use log::{Level, LevelFilter};
|
||||||
|
use std::env;
|
||||||
|
use std::io::Write;
|
||||||
|
|
||||||
|
lazy_static! {
|
||||||
|
static ref UWU: bool = env::var("JADE_UWU").map(|v| v == "true").unwrap_or(false);
|
||||||
|
static ref UWU_DEBUG: bool = env::var("JADE_UWU_DEBUG")
|
||||||
|
.map(|v| v == "true")
|
||||||
|
.unwrap_or(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn init(verbosity: usize) {
|
||||||
|
let log_specification = match verbosity {
|
||||||
|
0 => LogSpecification::builder()
|
||||||
|
.default(LevelFilter::Info)
|
||||||
|
.build(),
|
||||||
|
1 => LogSpecification::builder()
|
||||||
|
.default(LevelFilter::Debug)
|
||||||
|
.build(),
|
||||||
|
_ => LogSpecification::builder()
|
||||||
|
.default(LevelFilter::Trace)
|
||||||
|
.build(),
|
||||||
|
};
|
||||||
|
Logger::with(log_specification)
|
||||||
|
.format(format_log_entry)
|
||||||
|
.start()
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Formats a log entry with color
|
||||||
|
fn format_log_entry(
|
||||||
|
w: &mut dyn Write,
|
||||||
|
now: &mut DeferredNow,
|
||||||
|
record: &log::Record,
|
||||||
|
) -> std::io::Result<()> {
|
||||||
|
let msg = record.args().to_string();
|
||||||
|
let level = record.level();
|
||||||
|
let msg = apply_uwu(level, msg);
|
||||||
|
write!(
|
||||||
|
w,
|
||||||
|
"[ {} ] {} {}",
|
||||||
|
style(level).paint(level.to_string()),
|
||||||
|
now.now(),
|
||||||
|
msg
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Applies uwu if the required environment variables are set
|
||||||
|
fn apply_uwu(level: Level, msg: String) -> String {
|
||||||
|
match level {
|
||||||
|
Level::Error | Level::Warn | Level::Info => {
|
||||||
|
if *UWU {
|
||||||
|
uwu!(msg)
|
||||||
|
} else {
|
||||||
|
msg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Level::Debug | Level::Trace => {
|
||||||
|
if *UWU_DEBUG {
|
||||||
|
uwu!(msg)
|
||||||
|
} else {
|
||||||
|
msg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue