mincat 0.1.0

Web 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

use http::{header, HeaderValue, StatusCode};
use mincat_core::response::{IntoResponse, IntoResponseParts, Response};

pub struct Redirect {
    status_code: StatusCode,
    location: HeaderValue,
}

impl Redirect {
    pub fn sse_other(uri: &str) -> Self {
        Self::with_status_code(StatusCode::SEE_OTHER, uri)
    }

    pub fn temporary(uri: &str) -> Self {
        Self::with_status_code(StatusCode::TEMPORARY_REDIRECT, uri)
    }

    pub fn permanent(uri: &str) -> Self {
        Self::with_status_code(StatusCode::PERMANENT_REDIRECT, uri)
    }

    fn with_status_code(status_code: StatusCode, uri: &str) -> Self {
        if !status_code.is_redirection() {
            panic!("status code can't redirection")
        }

        Self {
            status_code,
            location: HeaderValue::try_from(uri).expect("uri isn't a valid header value"),
        }
    }
}

impl IntoResponse for Redirect {
    fn into_response(self) -> Response {
        (self.status_code, [(header::LOCATION, self.location)]).into_response()
    }
}

impl IntoResponseParts for Redirect {
    fn into_response_parts(self, mut response: Response) -> Response {
        *response.status_mut() = self.status_code;
        response
            .headers_mut()
            .extend([(header::LOCATION, self.location)]);
        response
    }
}