hitman 0.12.1

A command line tool for hitting API endpoints
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
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

use log::{
    set_boxed_logger, set_max_level, Level, LevelFilter, Log, Metadata, Record,
    SetLoggerError,
};
use std::{
    io::{self, IsTerminal, Write},
    ops::{Deref, DerefMut},
};
use termcolor::{Color, ColorChoice, ColorSpec, StandardStream, WriteColor};

struct Logger {
    level: Level,
    color: ColorChoice,
}

/// Applies colors based on line prefices such as <, > or #,
/// while level doesn't affect the color.
impl Log for Logger {
    fn enabled(&self, metadata: &Metadata) -> bool {
        metadata.level() <= self.level
    }

    fn log(&self, record: &Record) {
        if self.enabled(record.metadata()) {
            let mut stream = ScopedColorStream::new(self.color);
            let msg = format!("{}", record.args());
            if msg.starts_with('<') {
                stream
                    .set_color(ColorSpec::new().set_fg(Some(Color::Cyan)))
                    .ok();
            } else if msg.starts_with('>') {
                stream
                    .set_color(ColorSpec::new().set_fg(Some(Color::Blue)))
                    .ok();
            } else if msg.starts_with('#') {
                stream
                    .set_color(ColorSpec::new().set_fg(Some(Color::Yellow)))
                    .ok();
            }
            writeln!(&mut stream, "{}", record.args()).unwrap_or_else(|_| {
                eprintln!("{}", record.args());
            });
        }
    }

    fn flush(&self) {
        io::stderr().flush().ok();
    }
}

pub fn init(
    verbose: bool,
    quiet: bool,
    is_flurry: bool,
) -> Result<(), SetLoggerError> {
    let logger = Logger {
        level: match (verbose, quiet, is_flurry) {
            (_, true, _) => Level::Error,
            (_, _, true) => Level::Warn,
            (true, _, _) => Level::Debug,
            _ => Level::Info,
        },
        color: if io::stderr().is_terminal() {
            ColorChoice::Auto
        } else {
            ColorChoice::Never
        },
    };

    set_boxed_logger(Box::new(logger))?;
    set_max_level(LevelFilter::Info);

    Ok(())
}

/// Wrapper that automatically resets the terminal color
struct ScopedColorStream {
    stream: StandardStream,
}

impl ScopedColorStream {
    fn new(color: ColorChoice) -> Self {
        Self {
            stream: StandardStream::stderr(color),
        }
    }
}

impl Drop for ScopedColorStream {
    fn drop(&mut self) {
        self.stream.reset().ok();
    }
}

impl Deref for ScopedColorStream {
    type Target = StandardStream;

    fn deref(&self) -> &Self::Target {
        &self.stream
    }
}

impl DerefMut for ScopedColorStream {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.stream
    }
}