prole 1.3.1

Prole is a collection of re-usable methods in Rust. Feel free to use them for your own work.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use std::env;
use std::io::Write;

use chrono::Local;
use env_logger::Builder;
use env_logger::fmt::Color;
use log::Level;
use log::LevelFilter;

/// Initialises an env_logger with the specified [LevelFilter].
///
/// ```
/// use log::LevelFilter;
/// use prole::util::logging::init_logger;
///
/// init_logger(LevelFilter::Info);
/// ```
pub fn init_logger(level: LevelFilter) {
    // Initialise the logger at the default level of info
    if env::var("RUST_LOG").is_err() {
        env::set_var("RUST_LOG", level.as_str())
    }
    // env_logger::init();
    Builder::from_default_env()
        .format(|buf, record| {
            let color = match record.level() {
                Level::Error => Color::Red,
                Level::Warn => Color::Yellow,
                Level::Info => Color::Green,
                Level::Debug => Color::Magenta,
                Level::Trace => Color::Cyan,
            };
            let mut style = buf.style();
            style.set_color(color).set_bold(true);

            writeln!(buf,
                     "[{}] {} - {}",
                     Local::now().format("%Y-%m-%d %H:%M:%S"),
                     style.value(record.level()),
                     record.args()
            )
        })
        .filter(None, LevelFilter::Info)
        .init();
}