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

use ansi_term::Colour;
use log::{Level, LevelFilter, Metadata, Record};

pub struct RustOut;

impl log::Log for RustOut {
    fn enabled(&self, metadata: &Metadata) -> bool {
        metadata.level() <= Level::Trace
    }

    fn log(&self, record: &Record) {
        if self.enabled(record.metadata()) {
            let level_str = match record.level() {
                Level::Error => Colour::Red.bold().paint("ERROR"),
                Level::Warn => Colour::Yellow.bold().paint("WARN"),
                Level::Info => Colour::Green.bold().paint("INFO"),
                Level::Debug => Colour::Blue.bold().paint("DEBUG"),
                Level::Trace => Colour::Purple.bold().paint("TRACE"),
            };

            let message = format!("{} → {}", level_str, record.args());

            println!("{}", message);
        }
    }

    fn flush(&self) {}
}

/// Initializes the Rustout logger with the desired log level filter.
pub fn init_logger(log_level: LevelFilter) {
    log::set_logger(&RustOut).unwrap();
    log::set_max_level(log_level);
}

pub fn main() {
    init_logger(LevelFilter::Trace);
    log::error!("This is an error message");
    log::warn!("This is a warning message");
    log::info!("This is an info message");
    log::debug!("This is a debug message");
    log::trace!("This is a trace message");
}