axum_responses 0.4.6

A Simple way to use Axum responses
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

use axum::{http::StatusCode, response::IntoResponse};

use crate::http::HttpResponse;

pub type FileResult = Result<FileResponse, HttpResponse>;

#[derive(Default, Debug)]
pub struct FileResponse {
    bytes: Option<Vec<u8>>,
    filename: String,
    content_type: String,
    disposition: ContentDisposition,
}

#[derive(Debug, Default)]
pub enum ContentDisposition {
    Inline,
    #[default]
    Attachment,
}

impl FileResponse {
    pub fn builder() -> FileResponse {
        FileResponse::default()
    }

    pub fn bytes(mut self, bytes: Vec<u8>) -> Self {
        self.bytes = Some(bytes);
        self
    }

    pub fn filename(mut self, filename: &str) -> Self {
        self.filename = filename.to_string();
        self
    }

    pub fn disposition(mut self, disposition: ContentDisposition) -> Self {
        self.disposition = disposition;
        self
    }

    pub fn content_type(mut self, content_type: &str) -> Self {
        self.content_type = content_type.to_string();
        self
    }

    #[doc(hidden)]
    pub fn build(self) -> impl IntoResponse {
        let bytes = self.bytes.unwrap_or_default();
        let disposition = match self.disposition {
            ContentDisposition::Inline => "inline",
            ContentDisposition::Attachment => "attachment",
        };

        let filename = if self.filename.is_empty() {
            "file"
        } else {
            &self.filename
        };

        (
            StatusCode::OK,
            [
                ("Content-type", self.content_type.as_str()),
                (
                    "Content-disposition",
                    &format!("{disposition}; filename=\"{filename}\""),
                ),
            ],
            bytes,
        )
            .into_response()
    }
}

impl IntoResponse for FileResponse {
    fn into_response(self) -> axum::response::Response {
        self.build().into_response()
    }
}