mahogany 0.1.0

[BETA] Logging framework for Rust
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use chrono::Utc;
use std::{
    fs,
    io::Write,
    sync::mpsc::{channel, Receiver, Sender},
    thread,
};
use crate::{LogLevel, Loggable, LogLocation};

#[derive(Clone)]
pub struct Logger<T: LogLevel> {
    sender: Sender<LogTransmit<T>>,
}
impl<T: LogLevel> Logger<T> {
    pub fn new(output: LogLocation) -> Logger<T> {
        let (sender, receiver) = channel();

        let output_path = output.resolve();

        LoggerTool {
            receiver,
            output_path,
        }
        .start();

        Logger::<T> { sender }
    }

    pub fn log(&self, level: T, message: &dyn Loggable) {
        self.sender
            .send(LogTransmit::Log(LogInstance {
                level,
                message: message.as_log(),
            }))
            .unwrap();
    }
}

struct LoggerTool<T: LogLevel> {
    receiver: Receiver<LogTransmit<T>>,
    output_path: String,
}
impl<T: LogLevel> LoggerTool<T> {
    fn start(self) {
        thread::spawn(move || {
            match fs::OpenOptions::new()
                
                .append(true)
                .create(true)
                .open(&self.output_path)
            {
                Ok(mut file) => {
                    while let Ok(message) = self.receiver.recv() {
                        match message {
                            LogTransmit::Close => {
                                self.write_msg(&mut file, "SYSTEM", "Abort message recieved".to_string());
                                break;
                            },
                            LogTransmit::Log(log) => {
                                self.write_msg(&mut file, log.level.level_string(), log.message);
                            }
                        };
                    }
                },
                Err(e) => {
                    println!("[CRITICAL] Failed to open log file: {}", e);
                }
            }
        });
    }

    fn write_msg(&self, file: &mut fs::File, level: &'static str, message: String) {
        let msg = format!(
            "{} [{}]: {}\n",
            self.current_ts(),
            level,
            message
        );

        if let Err(e) = file.write(msg.as_bytes()) {
            println!("[CRITICAL] Failed to write to log file: {}", e)
        }
    }

    fn current_ts(&self) -> String {
        Utc::now().format("%Y-%m-%d %H:%M:%S").to_string()
    }
}

pub enum LogTransmit<T: LogLevel> {
    Log(LogInstance<T>),
    Close,
}

pub struct LogInstance<T: LogLevel> {
    pub level: T,
    pub message: String,
}