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

use colored::*;

#[derive(Debug, Eq, PartialEq)]
pub enum MessageLevel {
    Error,
    Warning,
    Info,
    Command,
}

#[derive(Debug)]
pub struct Message {
    level: MessageLevel,
    text: String,
}

impl Message {
    pub fn new(level: MessageLevel, text: &str) -> Message {
        Message {
            level,
            text: String::from(text),
        }
    }

    pub fn show(&self) {
        let msg = match self.level {
            MessageLevel::Error => format!("{} {}", "[ERROR]".red().bold(), self.text.red()),
            MessageLevel::Warning => format!("{} {}", "[WARN]".yellow().bold(), self.text.yellow()),
            MessageLevel::Info => format!("[ {} ]", self.text.normal())
                .cyan()
                .bold()
                .reversed()
                .to_string(),
            MessageLevel::Command => format!("$ {}", self.text.normal()).cyan().to_string(),
        };
        println!("{}", msg);
    }
}

pub trait Messageable {
    fn error_message(&self) -> Message;
    fn warning_message(&self) -> Message;
    fn info_message(&self) -> Message;
    fn command_message(&self) -> Message;
}

impl Messageable for &str {
    fn error_message(&self) -> Message {
        Message::new(MessageLevel::Error, self)
    }

    fn warning_message(&self) -> Message {
        Message::new(MessageLevel::Warning, self)
    }

    fn info_message(&self) -> Message {
        Message::new(MessageLevel::Info, self)
    }

    fn command_message(&self) -> Message {
        Message::new(MessageLevel::Command, self)
    }
}

impl Messageable for String {
    fn error_message(&self) -> Message {
        Message::new(MessageLevel::Error, &self)
    }

    fn warning_message(&self) -> Message {
        Message::new(MessageLevel::Warning, &self)
    }

    fn info_message(&self) -> Message {
        Message::new(MessageLevel::Info, &self)
    }

    fn command_message(&self) -> Message {
        Message::new(MessageLevel::Command, &self)
    }
}