chore: add setting of headers for bucket response

jetbridge · Dec 5, 2022 · 45dda8f · 45dda8f
1 parent e88ff05
commit 45dda8f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/assets/lambda/ImageOptimization/index.ts b/assets/lambda/ImageOptimization/index.ts
@@ -61,7 +61,14 @@ const requestHandler =
         throw new Error(`Could not fetch image ${trimmedKey} from bucket.`)
       }
       pipeRes(response.Body, res)
-
+      // Respect the bucket file's content-type and cache-control
+      // nextImageOptimizer will use this to set the results.maxAge
+      if (response.ContentType) {
+        res.setHeader('Content-Type', response.ContentType)
+      }
+      if (response.CacheControl) {
+        res.setHeader('Cache-Control', response.CacheControl)
+      }
     }
   }