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
128
129
130
131
132
133
134
135
136
137
138
139
140

use std::{collections::HashMap, mem};

#[derive(Clone, Debug, Default)]
pub struct Wildcard<T> {
    wildcard: T,
}

impl<T> Wildcard<T> {
    pub fn get_wildcard(&self) -> &T {
        &self.wildcard
    }
}

/// Struct containing data on a single request.
///
/// parsed_body which is a Option<String> that can contain the body as a String
///
/// body is used when the body of the request is not a String
#[derive(Clone, Debug, Default)]
pub struct Request {
    raw_headers: Vec<String>,
    status_line: Vec<String>,
    body: Vec<u8>,
    wildcard: Option<String>,
    is_http2: bool,
}

#[derive(Clone, Debug)]
pub enum BodyType {
    ASCII(String),
    Bytes(Vec<u8>),
}

impl Request {
    pub fn new(
        raw_body: &[u8],
        raw_headers: Vec<String>,
        status_line: Vec<String>,
        wildcard: Option<String>,
    ) -> Request {
        Request {
            body: raw_body.to_vec(),
            raw_headers,
            status_line,
            wildcard,
            is_http2: false,
        }
    }

    pub(crate) fn set_wildcard(&mut self, w: Option<String>) -> &Self {
        self.wildcard = w;
        self
    }

    /// Get request body as bytes
    pub fn get_raw_body(&self) -> &[u8] {
        &self.body
    }

    /// Get request body as a string
    pub fn get_parsed_body(&self) -> Option<&str> {
        if let Ok(s) = std::str::from_utf8(&self.body) {
            Some(s)
        } else {
            None
        }
    }

    /// Get request headers in a HashMap
    pub fn get_headers(&self) -> HashMap<&str, &str> {
        #[cfg(feature = "log")]
        log::trace!("Headers: {:#?}", self.raw_headers);
        self.raw_headers
            .iter()
            .map(|i| i.split(": "))
            .map(|mut i| (i.next().unwrap(), i.next().unwrap()))
            .collect::<HashMap<&str, &str>>()
    }

    /// Get status line of request
    pub fn get_status_line<'a>(&'a self) -> &'a [String] {
        &*self.status_line
    }

    pub fn get_wildcard(&self) -> Option<&String> {
        self.wildcard.as_ref()
    }

    pub fn get_http2(&self) -> bool {
        self.is_http2
    }

    pub fn set_http2(mut self, w: bool) -> Self {
        self.is_http2 = w;
        self
    }
}

impl<'a> From<&'a mut Request> for Wildcard<&'a str> {
    fn from(value: &'a mut Request) -> Self {
        Wildcard {
            wildcard: value.wildcard.as_ref().unwrap(),
        }
    }
}

impl<'a> From<&'a mut Request> for Wildcard<&'a [u8]> {
    fn from(value: &'a mut Request) -> Self {
        Wildcard {
            wildcard: value.wildcard.as_ref().unwrap().as_bytes(),
        }
    }
}

impl<'a> From<&'a mut Request> for &'a [u8] {
    fn from(value: &'a mut Request) -> Self {
        value.get_raw_body()
    }
}

impl<'a> From<&'a mut Request> for Option<&'a str> {
    fn from(value: &'a mut Request) -> Self {
        value.get_parsed_body()
    }
}

impl From<&mut Request> for Request {
    fn from(value: &mut Request) -> Self {
        mem::take(value)
    }
}
use thiserror::Error;

#[derive(Error, Debug)]
pub enum RequestError {
    #[error("failed to parse status line")]
    StatusLineErr,
    #[error("failed to parse headers")]
    HeadersErr,
}