mirror of https://github.com/Llewellynvdm/exa.git
67 lines
1.6 KiB
Rust
67 lines
1.6 KiB
Rust
//! Debug error logging.
|
||
|
||
use std::ffi::OsStr;
|
||
|
||
use ansi_term::{Colour, ANSIString};
|
||
|
||
|
||
/// Sets the internal logger, changing the log level based on the value of an
|
||
/// environment variable.
|
||
pub fn configure<T: AsRef<OsStr>>(ev: Option<T>) {
|
||
let ev = match ev {
|
||
Some(v) => v,
|
||
None => return,
|
||
};
|
||
|
||
let env_var = ev.as_ref();
|
||
if env_var.is_empty() {
|
||
return;
|
||
}
|
||
|
||
if env_var == "trace" {
|
||
log::set_max_level(log::LevelFilter::Trace);
|
||
}
|
||
else {
|
||
log::set_max_level(log::LevelFilter::Debug);
|
||
}
|
||
|
||
let result = log::set_logger(GLOBAL_LOGGER);
|
||
if let Err(e) = result {
|
||
eprintln!("Failed to initialise logger: {}", e);
|
||
}
|
||
}
|
||
|
||
|
||
#[derive(Debug)]
|
||
struct Logger;
|
||
|
||
const GLOBAL_LOGGER: &Logger = &Logger;
|
||
|
||
impl log::Log for Logger {
|
||
fn enabled(&self, _: &log::Metadata<'_>) -> bool {
|
||
true // no need to filter after using ‘set_max_level’.
|
||
}
|
||
|
||
fn log(&self, record: &log::Record<'_>) {
|
||
let open = Colour::Fixed(243).paint("[");
|
||
let level = level(record.level());
|
||
let close = Colour::Fixed(243).paint("]");
|
||
|
||
eprintln!("{}{} {}{} {}", open, level, record.target(), close, record.args());
|
||
}
|
||
|
||
fn flush(&self) {
|
||
// no need to flush with ‘eprintln!’.
|
||
}
|
||
}
|
||
|
||
fn level(level: log::Level) -> ANSIString<'static> {
|
||
match level {
|
||
log::Level::Error => Colour::Red.paint("ERROR"),
|
||
log::Level::Warn => Colour::Yellow.paint("WARN"),
|
||
log::Level::Info => Colour::Cyan.paint("INFO"),
|
||
log::Level::Debug => Colour::Blue.paint("DEBUG"),
|
||
log::Level::Trace => Colour::Fixed(245).paint("TRACE"),
|
||
}
|
||
}
|