sark 0.1.0

Simple Asynchronous Rust webKit
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

use bytes::BytesMut;
use http::{StatusCode, HeaderMap};

#[derive(Clone)]
pub struct Response {
    status: StatusCode,
    headers: HeaderMap,
    body: BytesMut,
}

impl Response {
    pub fn new(status: StatusCode) -> Self {
        Self {
            status,
            headers: HeaderMap::new(),
            body: BytesMut::new(),
        }
    }

    pub fn ok() -> Self { Self::new(StatusCode::OK) }
    pub fn not_found() -> Self { Self::new(StatusCode::NOT_FOUND) }
    pub fn status(&self) -> StatusCode { self.status }
    pub fn set_status(&mut self, status: StatusCode) { self.status = status; }
    pub fn headers(&self) -> &HeaderMap { &self.headers }
    pub fn headers_mut(&mut self) -> &mut HeaderMap { &mut self.headers }
    pub fn body(&self) -> &BytesMut { &self.body }
    pub fn body_mut(&mut self) -> &mut BytesMut { &mut self.body }
    
    pub fn set_body(&mut self, body: impl Into<BytesMut>) {
        self.body = body.into();
    }
    
    pub fn set_body_str(&mut self, body: &str) -> &mut Self {
        self.body = BytesMut::from(body.as_bytes());
        self
    }
    
    pub fn body_str(&self) -> Option<&str> {
        std::str::from_utf8(self.body.as_ref()).ok()
    }
}