1 files changed, 29 insertions, 2 deletions
diff --git a/kittybox-rs/src/media/mod.rs b/kittybox-rs/src/media/mod.rs
index 0ce2ec9..ce08704 100644
--- a/kittybox-rs/src/media/mod.rs
+++ b/kittybox-rs/src/media/mod.rs
@@ -1,6 +1,10 @@
+use std::convert::TryFrom;
+
 use axum::{
     extract::{Extension, Host, multipart::Multipart, Path},
-    response::{IntoResponse, Response}, headers::HeaderValue,
+    response::{IntoResponse, Response},
+    headers::{Header, HeaderValue, IfNoneMatch, HeaderMapExt},
+    TypedHeader,
 };
 use kittybox_util::error::{MicropubError, ErrorType};
 use kittybox_indieauth::Scope;
@@ -65,6 +69,7 @@ pub(crate) async fn upload<S: MediaStore, A: AuthBackend>(
 pub(crate) async fn serve<S: MediaStore>(
     Host(host): Host,
     Path(path): Path<String>,
+    if_none_match: Option<TypedHeader<IfNoneMatch>>,
     Extension(blobstore): Extension<S>
 ) -> Response {
     use axum::http::StatusCode;
@@ -72,6 +77,23 @@ pub(crate) async fn serve<S: MediaStore>(
     match blobstore.read_streaming(&host, path.as_str()).await {
         Ok((metadata, stream)) => {
             tracing::debug!("Metadata: {:?}", metadata);
+
+            let etag = if let Some(etag) = metadata.etag {
+                let etag = format!("\"{}\"", etag).parse::<axum::headers::ETag>().unwrap();
+
+                if let Some(TypedHeader(if_none_match)) = if_none_match {
+                    tracing::debug!("If-None-Match: {:?}", if_none_match);
+                    // If-None-Match is a negative precondition that
+                    // returns 304 when it doesn't match because it
+                    // only matches when file is different
+                    if !if_none_match.precondition_passes(&etag) {
+                        return StatusCode::NOT_MODIFIED.into_response()
+                    }
+                }
+
+                Some(etag)
+            } else { None };
+
             let mut r = Response::builder();
             {
                 let headers = r.headers_mut().unwrap();
@@ -89,8 +111,13 @@ pub(crate) async fn serve<S: MediaStore>(
                         HeaderValue::from_str(&length.to_string()).unwrap()
                     );
                 }
+                if let Some(etag) = etag {
+                    headers.typed_insert(etag);
+                }
             }
-            r.body(axum::body::StreamBody::new(stream)).unwrap().into_response()
+            r.body(axum::body::StreamBody::new(stream))
+                .unwrap()
+                .into_response()
         },
         Err(err) => match err.kind() {
             ErrorKind::NotFound => {