2020-10-10 01:01:12 +00:00
|
|
|
|
//! 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 {
|
2020-10-10 18:49:46 +00:00
|
|
|
|
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"),
|
2020-10-10 01:01:12 +00:00
|
|
|
|
}
|
|
|
|
|
}
|