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
use std::str::FromStr;
use std::{collections::HashMap, fmt};
use futures::AsyncReadExt;
use http_types::convert::DeserializeOwned;
use http_types::headers::{HeaderName, HeaderValue, HeaderValues};
use http_types::{Method, Url};
#[derive(Debug, Clone)]
pub struct Request {
pub url: Url,
pub method: Method,
pub headers: HashMap<HeaderName, HeaderValues>,
pub body: Vec<u8>,
}
impl fmt::Display for Request {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
writeln!(f, "{} {}", self.method, self.url)?;
for (name, values) in &self.headers {
let values = values
.iter()
.map(|value| format!("{}", value))
.collect::<Vec<_>>();
let values = values.join(",");
writeln!(f, "{}: {}", name, values)?;
}
writeln!(f, "{}", String::from_utf8_lossy(&self.body))
}
}
impl Request {
pub fn body_json<T: DeserializeOwned>(&self) -> Result<T, serde_json::Error> {
serde_json::from_slice(&self.body)
}
pub async fn from(mut request: http_types::Request) -> Request {
let method = request.method();
let url = request.url().to_owned();
let mut headers = HashMap::new();
for (header_name, header_values) in &request {
headers.insert(header_name.to_owned(), header_values.to_owned());
}
let mut body: Vec<u8> = vec![];
request
.take_body()
.into_reader()
.read_to_end(&mut body)
.await
.expect("Failed to read body");
Self {
url,
method,
headers,
body,
}
}
pub(crate) async fn from_hyper(request: hyper::Request<hyper::Body>) -> Request {
let (parts, body) = request.into_parts();
let method = parts.method.into();
let url = match parts.uri.authority() {
Some(_) => parts.uri.to_string(),
None => format!("http://localhost{}", parts.uri),
}
.parse()
.unwrap();
let mut headers = HashMap::new();
for (name, value) in parts.headers {
if let Some(name) = name {
let name = name.as_str().as_bytes().to_owned();
let name = HeaderName::from_bytes(name).unwrap();
let value = value.as_bytes().to_owned();
let value = HeaderValue::from_bytes(value).unwrap();
let value_parts = value.as_str().split(',');
let value_parts = value_parts
.map(|it| it.trim())
.filter_map(|it| HeaderValue::from_str(it).ok());
headers.insert(name, value_parts.collect());
}
}
let body = hyper::body::to_bytes(body)
.await
.expect("Failed to read request body.")
.to_vec();
Self {
url,
method,
headers,
body,
}
}
}