krustie 0.3.0

Krustie is a simple and easy-to-use backend framework.
Documentation 
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

use std::fmt::{Display, Formatter, Result as fResult};

use super::http_method::HttpMethod;

#[derive(Clone)]
pub(crate) struct RequestLine {
    method: HttpMethod,
    uri: String,
    version: String,
    path_array: Vec<String>,
}

impl RequestLine {
    pub(super) fn new(
        method: &str,
        uri: &str,
        version: &str,
    ) -> Result<Self, ParseRequestLineError> {
        match HttpMethod::try_from(method) {
            Ok(method) => Ok(Self {
                method,
                uri: uri.to_string(),
                version: version.to_string(),
                path_array: Self::uri_parser(uri),
            }),
            Err(_) => {
                return Err(ParseRequestLineError);
            }
        }
    }

    fn uri_parser(uri: &str) -> Vec<String> {
        if uri.starts_with('/') {
            uri.split('/').skip(1).map(|str| str.to_string()).collect()
        } else if uri.starts_with("http") {
            uri.split('/').skip(3).map(|str| str.to_string()).collect()
        } else {
            uri.split('/').skip(1).map(|str| str.to_string()).collect()
        }
    }

    pub(super) fn get_method(&self) -> &HttpMethod {
        &self.method
    }

    pub(super) fn get_path_array(&self) -> &Vec<String> {
        &self.path_array
    }

    pub(super) fn get_version(&self) -> &String {
        &self.version
    }

    pub(super) fn get_path(&self) -> &str {
        &self.uri.as_str()
    }

    pub(crate) fn set_method(&mut self, method: HttpMethod) {
        self.method = method;
    }

    pub(crate) fn set_uri(&mut self, uri: &str) {
        self.uri = String::from(uri);
        self.path_array = Self::uri_parser(uri);
    }

    pub(crate) fn set_version(&mut self, version: &str) {
        self.version = String::from(version.trim());
    }
}

impl Display for RequestLine {
    fn fmt(&self, f: &mut Formatter<'_>) -> fResult {
        write!(f, "{} {} {}", self.method, self.uri, self.version)
    }
}

impl TryFrom<&str> for RequestLine {
    type Error = ParseRequestLineError;
    fn try_from(request_line: &str) -> Result<Self, Self::Error> {
        let request_line: Vec<&str> = request_line.split(' ').collect();

        if request_line.len() != 3
            || !HttpMethod::is_valid(request_line[0])
            || !request_line[2].starts_with("HTTP/")
        {
            return Err(ParseRequestLineError);
        }

        let (method, uri, version) = (
            request_line[0].trim(),
            request_line[1].trim(),
            request_line[2].trim(),
        );

        match Self::new(method, uri, version) {
            Ok(request_line) => Ok(request_line),
            Err(_) => Err(ParseRequestLineError),
        }
    }
}

#[derive(Debug, Clone)]
pub(crate) struct ParseRequestLineError;

impl Display for ParseRequestLineError {
    fn fmt(&self, f: &mut Formatter<'_>) -> fResult {
        write!(f, "invalid request line")
    }
}