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
|
use colored::*;
use jsonformat::{format_json, Indentation};
use std::fmt::Display;
pub struct BarbOutput {
request: bool,
req_headers: bool,
headers: bool,
body: bool,
raw_body: bool,
color: bool,
}
impl BarbOutput {
fn _print_header<T, I>(&self, name: T, value: I)
where T: Display, I: Display{
println!("{}: {}", name, value);
}
pub fn new(request: bool, req_headers: bool, headers: bool, body: bool, raw_body: bool, color: bool) -> BarbOutput {
BarbOutput {
request,
req_headers,
headers,
body,
raw_body,
color,
}
}
pub fn req(&self, method: String, url: String) {
if !self.request {
return;
}
if self.color {
println!("{} {}", method.purple(), url.purple());
} else {
println!("{} {}", method, url);
}
}
pub fn req_hdr(&self, name: String, value: String) {
if !self.req_headers {
return;
}
if self.color {
self._print_header(name.yellow(), value);
} else {
self._print_header(name, value);
}
}
pub fn end_req(&self) {
if self.req_headers || self.request {
println!("");
}
}
pub fn resp_hdr(&self, name: String, value: &str) {
if !self.headers {
return;
}
if self.color {
self._print_header(name.blue(), value);
} else {
self._print_header(name, value);
}
}
pub fn end_resp_hdr(&self) {
if self.headers {
println!("");
}
}
pub fn status(&self, code: u16, text: &str) {
if !self.headers {
return;
}
if self.color {
if code >= 400 {
println!("{} {}", code.to_string().red(), text.red());
} else {
println!("{} {}", code.to_string().green(), text);
}
} else {
println!("{} {}", code, text);
}
}
pub fn body(&self, body: String) {
if !self.body {
return;
}
println!(
"{}",
match self.raw_body {
true => body,
false => format_json(body.as_str(), Indentation::Default),
}
);
}
}
|
