Improved HTTP headers

1 files changed, 14 insertions, 0 deletions

diff --git a/src/archive.rs b/src/archive.rs
index 0f09005..1b31d08 100644
--- a/src/archive.rs
+++ b/src/archive.rs
@@ -1,3 +1,4 @@
+use actix_web::http::ContentEncoding;
 use bytes::Bytes;
 use libflate::gzip::Encoder;
 use serde::Deserialize;
@@ -29,6 +30,19 @@ impl CompressionMethod {
         }
         .to_string()
     }
+
+    pub fn content_type(&self) -> String {
+        match &self {
+            CompressionMethod::TarGz => "application/gzip",
+        }
+        .to_string()
+    }
+
+    pub fn content_encoding(&self) -> ContentEncoding {
+        match &self {
+            CompressionMethod::TarGz => ContentEncoding::Gzip,
+        }
+    }
 }
 
 /// Possible errors