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
117
118
119
120
121
122
123
124
125
126
127
|
use colored::*;
use colored_json::prelude::*;
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);
}
}
fn _format_body(&self, body: String) -> String {
if self.raw_body {
return body;
}
let formatted = format_json(body.as_str(), Indentation::Default);
match self.color {
true => formatted.to_colored_json_auto().unwrap_or(formatted),
_ => formatted,
}
}
pub fn body(&self, body: String) {
if !self.body {
return;
}
println!(
"{}",
match self.raw_body {
true => body,
_ => self._format_body(body),
}
);
}
}
|
