Added Saving to file

amigin · amigin · commit eb0f6f3deada · 2022-11-09T18:37:13.000+02:00
diff --git a/src/http_ok_result.rs b/src/http_ok_result.rs
@@ -18,6 +18,11 @@ pub enum HttpOutput {
         permanent: bool,
     },
 
+    File {
+        file_name: String,
+        content: Vec<u8>,
+    },
+
     Raw(Response<Body>),
 }
 
@@ -62,6 +67,19 @@ impl HttpOutput {
             HttpOutput::Raw(_) => {
                 panic!("Raw response can not be turned into Http Fail result")
             }
+            HttpOutput::File { file_name, content } => HttpFailResult {
+                content_type: if let Some(ct) =
+                    WebContentType::detect_by_extension(file_name.as_str())
+                {
+                    ct
+                } else {
+                    WebContentType::Text
+                },
+                status_code,
+                content,
+                write_telemetry,
+                write_to_log: false,
+            },
         };
 
         Err(result)
@@ -116,6 +134,11 @@ impl HttpOutput {
                 }
             }
 
+            Self::File {
+                file_name: _,
+                content: _,
+            } => 200,
+
             HttpOutput::Raw(body) => body.status().as_u16(),
         }
     }
@@ -196,6 +219,20 @@ impl Into<Response<Body>> for HttpOkResult {
                 .unwrap(),
 
             HttpOutput::Raw(body) => body,
+            HttpOutput::File { file_name, content } => {
+                let builder = Response::builder().header(
+                    "content-disposition",
+                    format!(
+                        "attachment; filename=\"{file_name}\"; filename*=UTF-8''{file_name}",
+                        file_name = file_name
+                    ),
+                );
+
+                builder
+                    .status(status_code)
+                    .body(Body::from(content))
+                    .unwrap()
+            }
         };
     }
 }
diff --git a/src/web_content_type.rs b/src/web_content_type.rs
@@ -7,6 +7,7 @@ pub enum WebContentType {
     JavaScript,
     Json,
     Text,
+    Raw(String),
 }
 
 impl WebContentType {
@@ -19,6 +20,7 @@ impl WebContentType {
             WebContentType::Text => "text/plain; charset=utf-8",
             WebContentType::Png => "image/png",
             WebContentType::Svg => "image/svg+xml",
+            WebContentType::Raw(content_type) => content_type.as_str(),
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,7 @@ pub enum WebContentType {`
`7`	`7`	`JavaScript,`
`8`	`8`	`Json,`
`9`	`9`	`Text,`
	`10`	`+ Raw(String),`
`10`	`11`	`}`
`11`	`12`
`12`	`13`	`impl WebContentType {`
`@@ -19,6 +20,7 @@ impl WebContentType {`
`19`	`20`	`WebContentType::Text => "text/plain; charset=utf-8",`
`20`	`21`	`WebContentType::Png => "image/png",`
`21`	`22`	`WebContentType::Svg => "image/svg+xml",`
	`23`	`+ WebContentType::Raw(content_type) => content_type.as_str(),`
`22`	`24`	`}`
`23`	`25`	`}`
`24`	`26`