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
99

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) 2019, Olof Kraigher olof.kraigher@gmail.com

use std::path::Path;

#[derive(Debug, PartialEq)]
pub enum MessageType {
    Error,
    Warning,
    Info,
    Log,
}

#[must_use]
#[derive(Debug, PartialEq)]
pub struct Message {
    pub message_type: MessageType,
    pub message: String,
}

impl Message {
    pub fn log(message: impl Into<String>) -> Message {
        Message {
            message_type: MessageType::Log,
            message: message.into(),
        }
    }

    pub fn info(message: impl Into<String>) -> Message {
        Message {
            message_type: MessageType::Info,
            message: message.into(),
        }
    }

    pub fn warning(message: impl Into<String>) -> Message {
        Message {
            message_type: MessageType::Warning,
            message: message.into(),
        }
    }

    pub fn error(message: impl Into<String>) -> Message {
        Message {
            message_type: MessageType::Error,
            message: message.into(),
        }
    }

    pub fn file_error(message: impl Into<String>, file_name: &Path) -> Message {
        Message {
            message_type: MessageType::Error,
            message: format!(
                "{} (In file {})",
                message.into(),
                file_name.to_string_lossy()
            ),
        }
    }
}

impl std::fmt::Display for Message {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}: {}", self.message_type.as_ref(), self.message)
    }
}

impl AsRef<str> for MessageType {
    fn as_ref(&self) -> &str {
        match self {
            Self::Error => "error",
            Self::Warning => "warning",
            Self::Info => "info",
            Self::Log => "log",
        }
    }
}

pub trait MessageHandler {
    fn push(&mut self, message: Message);
}

impl MessageHandler for Vec<Message> {
    fn push(&mut self, message: Message) {
        self.push(message)
    }
}

#[derive(Default)]
pub struct MessagePrinter {}

impl MessageHandler for MessagePrinter {
    fn push(&mut self, message: Message) {
        println!("{}", message);
    }
}