sest cache control header

1 files changed, 14 insertions, 2 deletions

diff --git a/src/filters/files.rs b/src/filters/files.rs
index 59de2f3..0b313e2 100644
--- a/src/filters/files.rs
+++ b/src/filters/files.rs
@@ -1,7 +1,13 @@
-use crate::{config::FileserverConfig, ServiceError};
+use crate::{
+    config::{CacheConfig, FileserverConfig},
+    ServiceError,
+};
 use bytes::{Bytes, BytesMut};
 use futures_util::{future, future::Either, ready, stream, FutureExt, Stream, StreamExt};
-use headers::{AcceptRanges, ContentLength, ContentRange, ContentType, HeaderMapExt, LastModified};
+use headers::{
+    AcceptRanges, CacheControl, ContentLength, ContentRange, ContentType, HeaderMapExt,
+    LastModified,
+};
 use http_body_util::{combinators::BoxBody, BodyExt, StreamBody};
 use humansize::FormatSizeOptions;
 use hyper::{
@@ -139,6 +145,12 @@ pub async fn serve_files(
     let mime = mime_guess::from_path(path).first_or_octet_stream();
     r.headers_mut().typed_insert(ContentType::from(mime));
 
+    r.headers_mut().typed_insert(match config.cache {
+        CacheConfig::Public => CacheControl::new().with_public(),
+        CacheConfig::Private => CacheControl::new().with_private(),
+        CacheConfig::NoStore => CacheControl::new().with_no_store(),
+    });
+
     // if config.etag {
     //     r.headers_mut().typed_insert(etag);
     // }