restless-web 0.1.1

Back-end framework inspired by express.js
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
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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155

use once_cell::sync::Lazy;
use std::collections::HashMap;
use tokio::io::AsyncWriteExt;
use tokio::net::tcp::WriteHalf;

#[derive(Debug)]
pub struct Res {
    pub outcome: String,
    status: usize,
    headers: HashMap<String, String>,
}

// NOTE: Reference:
// <https://developer.mozilla.org/en-US/docs/Web/HTTP/Status>
static STATUS_TITLES: Lazy<HashMap<usize, &'static str>> = Lazy::new(|| {
    HashMap::from([
        // Information responses
        (100, "Continue"),
        (101, "Switching Protocol"),
        (102, "Processing"),
        (103, "Early Hints"), // WARN: Experimental
        // Successful responses
        (200, "OK"),
        (201, "Created"),
        (202, "Accepted"),
        (203, "Non-Authoritative Information"),
        (204, "No Content"),
        (205, "Reset Content"),
        (206, "Partial Content"),
        (207, "Multi-Status Status"), // NOTE: WebDAV
        (208, "Already Reported"),
        (226, "IM Used"),
        // Redirection messages
        (300, "Multiple Choices"),
        (301, "Moved Permanently"),
        (302, "Found"),
        (303, "See Other"),
        (304, "Not Modified"),
        (305, "Use Proxy"), // WARN: Deprecated
        (306, "unused"),
        (307, "Temporary Redirect"),
        (308, "Permanent Redirect"),
        // Client error responses
        (400, "Bad Request"),
        (401, "Unauthorized"),
        (402, "Payment Required"), // NOTE: Experimental
        (403, "Forbidden"),
        (404, "Not Found"),
        (405, "Method Not Allowed"),
        (406, "Not Acceptable"),
        (407, "Proxy Authentication Required"),
        (408, "Request Timeout"),
        (409, "Conflict"),
        (410, "Gone"),
        (411, "Length Required"),
        (412, "Precondition Failed"),
        (413, "Payload Too Large"),
        (414, "URI Too Long"),
        (415, "Unsupported Media Type"),
        (416, "Range Not Satisfiable"),
        (417, "Expectation Failed"),
        (418, "I'm a teapot"),
        (421, "Misdirected Request"),
        (422, "Unprocessable Content"), // NOTE: WebDAV
        (423, "Locked"),                // NOTE: WebDAV
        (424, "Failed Dependency"),     // NOTE: WebDAV
        (425, "Too Early"),             // NOTE: Experimental
        (426, "Upgrade Required"),
        (428, "Precondition Required"),
        (429, "Too Many Requests"),
        (431, "Request Header Fields Too Large"),
        (451, "Unavailable For Legal Reasons"),
        // Server error responses
        (500, "Internal Server Error"),
        (501, "Not Implemented"),
        (502, "Bad Gateway"),
        (503, "Service Unavailable"),
        (504, "Gateway Timeout"),
        (505, "HTTP Version Not Supported"),
        (506, "Variant Also Negotiates"),
        (507, "Insufficient Storage"), // NOTE: WebDAV
        (508, "Loop Detected"),        // NOTE: WebDAV
        (510, "Not Extended"),
        (511, "Network Authentication Required"),
    ])
});

impl<'a> Res {
    pub fn new() -> Res {
        Res {
            outcome: String::new(),
            status: 200,
            headers: HashMap::new(),
        }
    }

    pub fn status(mut self, status: usize) -> Res {
        self.status = status;

        self
    }

    pub fn set(&'a mut self, header_key: &'a str, header_value: &'a str) -> &'a mut Res {
        self.headers
            .insert(header_key.parse().unwrap(), header_value.parse().unwrap());

        self
    }

    pub fn get(&self, header_key: &str) -> Option<&str> {
        let header_value = self.headers.get(header_key)?;

        Some(header_value)
    }

    fn status_title(&'a self) -> Option<&'static str> {
        let title = *STATUS_TITLES.get(&self.status)?;

        Some(title)
    }

    pub fn send(mut self, outcome: &str) -> Res {
        self.outcome.push_str(outcome);
        self
    }

    pub async fn send_outcome(self, mut stream: WriteHalf<'_>) {
        let formatted_headers = self.format_headers();
        let title = self.status_title().expect("Wrong status code");

        let raw_res = format!(
            "HTTP/1.1 {} {}\r\n{}\r\n{}",
            self.status, title, formatted_headers, self.outcome
        );

        stream.write_all(raw_res.as_bytes()).await.unwrap();
        stream.flush().await.unwrap();
    }

    fn format_headers(&'a self) -> String {
        let mut formatted_headers = String::new();

        for (key, value) in &self.headers {
            formatted_headers += &format!("{key}: {value}\r\n");
        }

        formatted_headers
    }
}

impl Default for Res {
    fn default() -> Self {
        Self::new()
    }
}