colour 0.6.0

Macros for creating coloured console output.
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

use std::io::{self, Stderr, Stdout, Write};

use crossterm::{
    self,
    style::{Color, Print, ResetColor, SetForegroundColor},
    QueueableCommand,
};

pub type Colour = Color;

enum Terminal {
    StdOut(Stdout),
    StdErr(Stderr),
}

impl Write for Terminal {
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        match self {
            Terminal::StdOut(stdout) => stdout.write(buf),
            Terminal::StdErr(stderr) => stderr.write(buf),
        }
    }

    fn flush(&mut self) -> io::Result<()> {
        match self {
            Terminal::StdOut(stdout) => stdout.flush(),
            Terminal::StdErr(stderr) => stderr.flush(),
        }
    }
}

pub fn write(colour: Option<Colour>, message: &str, newline: bool) {
    if do_write(colour, message, newline, Terminal::StdOut(io::stdout())).is_err() {
        if newline {
            println!("{}", message);
        } else {
            print!("{}", message);
        }
    }
}

pub fn ewrite(colour: Option<Colour>, message: &str, newline: bool) {
    if do_write(colour, message, newline, Terminal::StdErr(io::stderr())).is_err() {
        if newline {
            eprintln!("{}", message);
        } else {
            eprint!("{}", message);
        }
    }
}

fn do_write(
    colour: Option<Colour>,
    message: &str,
    newline: bool,
    mut terminal: Terminal,
) -> crossterm::Result<()> {
    if let Some(colr) = colour {
        let _ = terminal.queue(SetForegroundColor(colr))?;
    }

    let _ = terminal.queue(Print(message))?.queue(ResetColor)?;

    if newline {
        let _ = terminal.queue(Print('\n'))?;
    }

    terminal.flush()?;

    Ok(())
}