Merge pull request #99 from upsuper/default_level_style

Move default level style to a public function

Author: KodrAus

src/fmt.rs

@@ -42,6 +42,7 @@ use std::error::Error;
 use std::cell::RefCell;
 use std::time::SystemTime;
 
+use log::Level;
 use termcolor::{self, ColorSpec, ColorChoice, Buffer, BufferWriter, WriteColor};
 use atty;
 use humantime::{format_rfc3339_seconds, format_rfc3339_nanos};
@@ -449,6 +450,19 @@ impl Formatter {
         }
     }
 
+    /// Get the default [`Style`] for the given level.
+    pub fn default_level_style(&self, level: Level) -> Style {
+        let mut level_style = self.style();
+        match level {
+            Level::Trace => level_style.set_color(Color::White),
+            Level::Debug => level_style.set_color(Color::Blue),
+            Level::Info => level_style.set_color(Color::Green),
+            Level::Warn => level_style.set_color(Color::Yellow),
+            Level::Error => level_style.set_color(Color::Red).set_bold(true),
+        };
+        level_style
+    }
+
     /// Get a [`Timestamp`] for the current date and time in UTC.
     ///
     /// # Examples

src/lib.rs

@@ -226,7 +226,7 @@ use std::io;
 use std::mem;
 use std::cell::RefCell;
 
-use log::{Log, LevelFilter, Level, Record, SetLoggerError, Metadata};
+use log::{Log, LevelFilter, Record, SetLoggerError, Metadata};
 
 pub mod filter;
 pub mod fmt;
@@ -320,16 +320,7 @@ impl Format {
             Box::new(move |buf, record| {
                 let write_level = if self.default_format_level {
                     let level = record.level();
-                    let mut level_style = buf.style();
-
-                    match level {
-                        Level::Trace => level_style.set_color(Color::White),
-                        Level::Debug => level_style.set_color(Color::Blue),
-                        Level::Info => level_style.set_color(Color::Green),
-                        Level::Warn => level_style.set_color(Color::Yellow),
-                        Level::Error => level_style.set_color(Color::Red).set_bold(true),
-                    };
-
+                    let level_style = buf.default_level_style(level);
                     write!(buf, "{:>5} ", level_style.value(level))
                 } else {
                     Ok(())