Merge pull request #768 from skade/cache-header

Set cache headers on all statics

Author: skade

src/caching.rs

@@ -0,0 +1,40 @@
+use rocket::http::hyper::header::{CacheControl, CacheDirective};
+use rocket::request::Request;
+use rocket::response::{self, Responder};
+
+pub struct Cached<R> {
+    inner: R,
+    directives: Vec<CacheDirective>,
+}
+
+impl<'r, R> Responder<'r> for Cached<R>
+where
+    R: Responder<'r>,
+{
+    fn respond_to(self, req: &Request) -> response::Result<'r> {
+        let Cached { inner, directives } = self;
+        inner.respond_to(req).map(|mut res| {
+            res.set_header(CacheControl(directives));
+            res
+        })
+    }
+}
+
+pub trait Caching
+where
+    Self: Sized,
+{
+    fn cached(self, directives: Vec<CacheDirective>) -> Cached<Self>;
+}
+
+impl<'r, R> Caching for R
+where
+    R: Responder<'r>,
+{
+    fn cached(self, directives: Vec<CacheDirective>) -> Cached<Self> {
+        Cached {
+            inner: self,
+            directives: directives,
+        }
+    }
+}

src/main.rs

@@ -23,6 +23,7 @@ extern crate regex;
 extern crate handlebars;
 
 mod cache;
+mod caching;
 mod category;
 mod headers;
 mod i18n;
@@ -56,7 +57,9 @@ use sass_rs::{compile_file, Options};
 
 use category::Category;
 
+use caching::{Cached, Caching};
 use i18n::{I18NHelper, SupportedLocale, TeamHelper};
+use rocket::http::hyper::header::CacheDirective;
 
 lazy_static! {
     static ref ASSETS: AssetFiles = {
@@ -137,13 +140,17 @@ fn components_locale(locale: SupportedLocale, _file: PathBuf) -> Template {
 }
 
 #[get("/logos/<file..>", rank = 1)]
-fn logos(file: PathBuf) -> Option<NamedFile> {
-    NamedFile::open(Path::new("static/logos").join(file)).ok()
+fn logos(file: PathBuf) -> Option<Cached<NamedFile>> {
+    NamedFile::open(Path::new("static/logos").join(file))
+        .ok()
+        .map(|file| file.cached(vec![CacheDirective::MaxAge(3600)]))
 }
 
 #[get("/static/<file..>", rank = 1)]
-fn files(file: PathBuf) -> Option<NamedFile> {
-    NamedFile::open(Path::new("static/").join(file)).ok()
+fn files(file: PathBuf) -> Option<Cached<NamedFile>> {
+    NamedFile::open(Path::new("static/").join(file))
+        .ok()
+        .map(|file| file.cached(vec![CacheDirective::MaxAge(3600)]))
 }
 
 #[get("/")]