ftnet_utils/
http.rs

1#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
2pub struct Request {
3    pub uri: String,
4    pub method: String,
5    pub headers: Vec<(String, Vec<u8>)>,
6}
7
8impl From<hyper::http::request::Parts> for Request {
9    fn from(r: hyper::http::request::Parts) -> Self {
10        let mut headers = vec![];
11        for (k, v) in r.headers {
12            let k = match k {
13                Some(v) => v.to_string(),
14                None => continue,
15            };
16            headers.push((k, v.as_bytes().to_vec()));
17        }
18
19        Request {
20            uri: r.uri.to_string(),
21            method: r.method.to_string(),
22            headers,
23        }
24    }
25}
26
27#[derive(serde::Deserialize, serde::Serialize, Debug)]
28pub struct Response {
29    pub status: u16,
30    pub headers: Vec<(String, Vec<u8>)>,
31}
32
33pub type ProxyResponse =
34    hyper::Response<http_body_util::combinators::BoxBody<hyper::body::Bytes, hyper::Error>>;
35pub type ProxyResult = eyre::Result<ProxyResponse>;
36
37#[allow(dead_code)]
38pub fn server_error_(s: String) -> ProxyResponse {
39    bytes_to_resp(s.into_bytes(), hyper::StatusCode::INTERNAL_SERVER_ERROR)
40}
41
42#[allow(dead_code)]
43pub fn not_found_(m: String) -> ProxyResponse {
44    bytes_to_resp(m.into_bytes(), hyper::StatusCode::NOT_FOUND)
45}
46
47pub fn bad_request_(m: String) -> ProxyResponse {
48    bytes_to_resp(m.into_bytes(), hyper::StatusCode::BAD_REQUEST)
49}
50
51#[macro_export]
52macro_rules! server_error {
53    ($($t:tt)*) => {{
54        $crate::http::server_error_(format!($($t)*))
55    }};
56}
57
58#[macro_export]
59macro_rules! not_found {
60    ($($t:tt)*) => {{
61        $crate::http::not_found_(format!($($t)*))
62    }};
63}
64
65#[macro_export]
66macro_rules! bad_request {
67    ($($t:tt)*) => {{
68        $crate::http::bad_request_(format!($($t)*))
69    }};
70}
71
72#[allow(dead_code)]
73pub fn redirect<S: AsRef<str>>(url: S) -> ProxyResponse {
74    let mut r = bytes_to_resp(vec![], hyper::StatusCode::PERMANENT_REDIRECT);
75    *r.headers_mut().get_mut(hyper::header::LOCATION).unwrap() = url.as_ref().parse().unwrap();
76    r
77}
78
79pub fn bytes_to_resp(bytes: Vec<u8>, status: hyper::StatusCode) -> ProxyResponse {
80    use http_body_util::BodyExt;
81
82    let mut r = hyper::Response::new(
83        http_body_util::Full::new(hyper::body::Bytes::from(bytes))
84            .map_err(|e| match e {})
85            .boxed(),
86    );
87    *r.status_mut() = status;
88    r
89}
ftnet_utils/http.rs

ftnet_utils/
http.rs
