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
use std::fmt::{self, Write};
use bytes::BytesMut;
pub struct Response {
pub response: Vec<u8>,
pub status_message: StatusMessage,
pub header_raw: BytesMut,
}
pub enum StatusMessage {
Ok,
Custom(u32, String),
}
impl Response {
pub fn new() -> Response {
Response {
response: Vec::new(),
status_message: StatusMessage::Ok,
header_raw: BytesMut::new(),
}
}
pub fn status_code(&mut self, code: u32, message: &str) -> &mut Response {
self.status_message = StatusMessage::Custom(code, message.to_string());
self
}
pub fn header(&mut self, name: &str, val: &str) -> &mut Response {
let header_string = templatify! { "" ; name ; ": " ; val ; "\r\n" };
self.header_raw.extend_from_slice(header_string.as_bytes());
self
}
pub fn body(&mut self, s: &str) -> &mut Response {
self.response = s.as_bytes().to_vec();
self
}
pub fn body_bytes(&mut self, b: &[u8]) -> &mut Response {
self.response = b.to_vec();
self
}
pub fn body_bytes_from_vec(&mut self, b: Vec<u8>) -> &mut Response {
self.response = b;
self
}
}
pub fn encode(msg: &Response, buf: &mut BytesMut) {
let length = msg.response.len();
let now = crate::core::date::now();
write!(
FastWrite(buf),
"\
HTTP/1.1 {}\r\n\
Content-Length: {}\r\n\
Date: {}\r\n\
",
msg.status_message,
length,
now
)
.unwrap();
buf.extend_from_slice(&msg.header_raw);
buf.extend_from_slice(b"\r\n");
buf.extend_from_slice(msg.response.as_slice());
}
impl Default for Response {
fn default() -> Response {
Response::new()
}
}
struct FastWrite<'a>(&'a mut BytesMut);
impl<'a> fmt::Write for FastWrite<'a> {
fn write_str(&mut self, s: &str) -> fmt::Result {
(*self.0).extend_from_slice(s.as_bytes());
Ok(())
}
fn write_fmt(&mut self, args: fmt::Arguments) -> fmt::Result {
fmt::write(self, args)
}
}
impl fmt::Display for StatusMessage {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match *self {
StatusMessage::Ok => f.pad("200 OK"),
StatusMessage::Custom(c, ref s) => write!(f, "{} {}", c, s),
}
}
}