di 0.1.2

Dependency injection container.
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

/*!

Pretty printing to terminal output, with colors.

*/

use term;
use super::ErrorWriter;

/// Pretty printer output for terminal.
pub struct PrettyTerminalOutput {
    color_error: Option<term::color::Color>,
    color_success: Option<term::color::Color>,
    color_definition: Option<term::color::Color>,
    color_module: Option<term::color::Color>,
    color_typename: Option<term::color::Color>,
    color_number: Option<term::color::Color>,
    color_operator: Option<term::color::Color>,
    color_layout: Option<term::color::Color>,

    color: Option<term::color::Color>,

    t: Box<term::Terminal<term::WriterWrapper>+Send>,
}

impl PrettyTerminalOutput {
    pub fn new() -> PrettyTerminalOutput {
        PrettyTerminalOutput {
            color_error: Some(term::color::BRIGHT_RED),
            color_success: Some(term::color::GREEN),
            color_definition: Some(term::color::BRIGHT_YELLOW),
            color_module: Some(term::color::BRIGHT_BLUE),
            color_typename: Some(term::color::CYAN),
            color_number: Some(term::color::MAGENTA),
            color_operator: Some(term::color::BRIGHT_WHITE),
            color_layout: Some(term::color::YELLOW),
            color: None,
            t: term::stdout().unwrap(),
        }
    }

    pub fn set_color(&mut self, color: Option<term::color::Color>) {
        if self.color != color {
            self.color = color;
            if let Some(c) = color {
                self.t.fg(c).unwrap();
            } else {
                self.t.reset().unwrap();
            }
        }
    }
}

impl ErrorWriter for PrettyTerminalOutput {

    fn error(&mut self, m: &str) {
        let new_color = self.color_error.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn success(&mut self, m: &str) {
        let new_color = self.color_success.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn definition(&mut self, m: &str) {
        let new_color = self.color_definition.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn module(&mut self, m: &str) {
        let new_color = self.color_module.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn typename(&mut self, m: &str) {
        let new_color = self.color_typename.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn number(&mut self, m: &str) {
        let new_color = self.color_number.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn operator(&mut self, m: &str) {
        let new_color = self.color_operator.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn layout(&mut self, m: &str) {
        let new_color = self.color_layout.clone();
        self.set_color(new_color);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn text(&mut self, m: &str) {
        self.set_color(None);
        (write!(self.t, "{}", m)).unwrap();
    }

    fn eol(&mut self) {
        (writeln!(self.t, "")).unwrap();
    }

    fn flush(&mut self) {
        self.t.flush().unwrap();
    }
}