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

// Copyright 2016  Jonas me
// See the 'AUTHORS' file at the top-level directory for a full list of authors.
//
// 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/.

//! Write messages at the terminal with severity level.

use std::fmt;
use std::io::{self, Write};

extern crate ansi_term;
use ansi_term::Colour;

#[derive(Debug, PartialEq, Eq, Clone, Copy)]
/// Indicate the verbosity.
/// At the command line interfaces, it is usually indicated with a specific
/// argument.
pub enum Verbosity {
    Normal,
    /// Argument '-q'
    Quiet,
    /// Argument '-v'
    Verbose,
    /// Argument '-vv'
    Verbose2,
}

/// Severity level.
pub struct Level {
    verbosity: Verbosity,
}

impl Level {
    pub fn new(v: Verbosity) -> Level {
        Level {
            verbosity: v,
        }
    }

    /// Write an information message, where the first word is coloured at green.
    pub fn info<T: fmt::Display>(&self, message: T) {
        match self.verbosity {
            Verbosity::Quiet => (),
            _ => {
                let msg = message.to_string();
                //let _msg = msg.trim_left();
                // Get the position where starts the first white space.
                let i = msg.find(char::is_whitespace).unwrap_or(msg.len());

                println!("    {}{}", Colour::Green.paint(&msg[..i]), &msg[i..]);
            }
        }
    }

    /// Write a warning message.
    pub fn warn<T: fmt::Display>(&self, message: T) {
        match self.verbosity {
            Verbosity::Quiet => (),
            _ => println!("    {} {}", Colour::Yellow.paint("warning:"), message),
        }
    }

    /// Write an error message to stderr.
    pub fn error<T: fmt::Display>(&self, message: T) {
        writeln!(io::stderr(), "    {} {}", Colour::Red.paint("error:"), message);
    }

    /// Write a debug message to stderr.
    pub fn debug<T: fmt::Display>(&self, message: T) {
        writeln!(io::stderr(), "    {} {}", Colour::Cyan.paint("debug:"), message);
    }
}