matthiasbeyer/imag
1
0
Fork 0
Code Releases Activity

Make coloring optional via flag in ImagLogger

Browse source
This commit is contained in:
Matthias Beyer 2016-08-02 11:38:38 +02:00
parent d4fa5b7de3
commit 8a0200a183
1 changed files with 35 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
libimagrt/src/logger.rs
View file

@ -3,8 +3,13 @@ use std::io::stderr;
use log::{Log, LogLevel, LogRecord, LogMetadata}; use log::{Log, LogLevel, LogRecord, LogMetadata};
use ansi_term::Style;
use ansi_term::Colour;
use ansi_term::ANSIString;
pub struct ImagLogger { pub struct ImagLogger {
lvl: LogLevel, lvl: LogLevel,
color_enabled: bool,
} }
impl ImagLogger { impl ImagLogger {
@ -12,6 +17,28 @@ impl ImagLogger {
pub fn new(lvl: LogLevel) -> ImagLogger { pub fn new(lvl: LogLevel) -> ImagLogger {
ImagLogger { ImagLogger {
lvl: lvl, lvl: lvl,
color_enabled: true
}
}
pub fn with_color(mut self, b: bool) -> ImagLogger {
self.color_enabled = b;
self
}
fn style_or_not(&self, c: Style, s: String) -> ANSIString {
if self.color_enabled {
c.paint(s)
} else {
ANSIString::from(s)
}
}
fn color_or_not(&self, c: Colour, s: String) -> ANSIString {
if self.color_enabled {
c.paint(s)
} else {
ANSIString::from(s)
} }
} }
@ -33,22 +60,22 @@ impl Log for ImagLogger {
let loc = record.location(); let loc = record.location();
match record.metadata().level() { match record.metadata().level() {
LogLevel::Debug => { LogLevel::Debug => {
let lvl = Cyan.paint(format!("{}", record.level())); let lvl = self.color_or_not(Cyan, format!("{}", record.level()));
let file = Cyan.paint(format!("{}", loc.file())); let file = self.color_or_not(Cyan, format!("{}", loc.file()));
let ln = Cyan.paint(format!("{}", loc.line())); let ln = self.color_or_not(Cyan, format!("{}", loc.line()));
let args = Cyan.paint(format!("{}", record.args())); let args = self.color_or_not(Cyan, format!("{}", record.args()));
writeln!(stderr(), "[imag][{: <5}][{}][{: >5}]: {}", lvl, file, ln, args).ok(); writeln!(stderr(), "[imag][{: <5}][{}][{: >5}]: {}", lvl, file, ln, args).ok();
}, },
LogLevel::Warn | LogLevel::Error => { LogLevel::Warn | LogLevel::Error => {
let lvl = Red.blink().paint(format!("{}", record.level())); let lvl = self.style_or_not(Red.blink(), format!("{}", record.level()));
let args = Red.paint(format!("{}", record.args())); let args = self.color_or_not(Red, format!("{}", record.args()));
writeln!(stderr(), "[imag][{: <5}]: {}", lvl, args).ok(); writeln!(stderr(), "[imag][{: <5}]: {}", lvl, args).ok();
}, },
LogLevel::Info => { LogLevel::Info => {
let lvl = Yellow.paint(format!("{}", record.level())); let lvl = self.color_or_not(Yellow, format!("{}", record.level()));
let args = Yellow.paint(format!("{}", record.args())); let args = self.color_or_not(Yellow, format!("{}", record.args()));
writeln!(stderr(), "[imag][{: <5}]: {}", lvl, args).ok(); writeln!(stderr(), "[imag][{: <5}]: {}", lvl, args).ok();
}, },