Merge pull request #60 from namada-net/tiago/add-cache-control-headers

sug0 · web-flow · commit 79d1b7316607 · 2025-09-08T09:36:06.000Z
add cache control headers
diff --git a/webserver/src/app.rs b/webserver/src/app.rs
@@ -20,6 +20,7 @@ use crate::appstate::AppState;
 use crate::config::AppConfig;
 use crate::handler;
 use crate::state::common::CommonState;
+use crate::utils::cache_forever;
 
 lazy_static! {
     static ref HTTP_TIMEOUT: u64 = 60;
@@ -43,17 +44,17 @@ impl ApplicationServer {
             Router::new()
                 .route(
                     "/commitment-tree",
-                    get(handler::tree::get_commitment_tree),
+                    get(cache_forever(handler::tree::get_commitment_tree)),
                 )
                 .route(
                     "/witness-map",
-                    get(handler::witness_map::get_witness_map),
+                    get(cache_forever(handler::witness_map::get_witness_map)),
                 )
                 .route(
                     "/notes-index",
-                    get(handler::notes_index::get_notes_index),
+                    get(cache_forever(handler::notes_index::get_notes_index)),
                 )
-                .route("/tx", get(handler::tx::get_tx))
+                .route("/tx", get(cache_forever(handler::tx::get_tx)))
                 .route("/height", get(handler::namada_state::get_latest_height))
                 .route(
                     "/block-index",
diff --git a/webserver/src/utils/mod.rs b/webserver/src/utils/mod.rs
@@ -1,6 +1,51 @@
+use std::future::Future;
+use std::pin::Pin;
+
+use axum::body::Body;
+use axum::handler::Handler;
+use axum::http::header::{CACHE_CONTROL, HeaderValue};
+use axum::http::request::Request;
+use axum::response::Response;
+
 pub mod sql {
     use diesel::expression::functions::define_sql_function;
     use diesel::sql_types::Integer;
 
     define_sql_function!(fn abs(x: Integer) -> Integer);
 }
+
+pub const fn cache_forever<H>(handler: H) -> CacheControlMiddleware<H> {
+    CacheControlMiddleware::new(handler)
+}
+
+#[derive(Debug, Clone)]
+pub struct CacheControlMiddleware<H> {
+    next_handler: H,
+}
+
+impl<H> CacheControlMiddleware<H> {
+    const CACHE_CONTROL_VALUE: &str = "public, max-age=31536000, immutable";
+
+    pub const fn new(next_handler: H) -> Self {
+        Self { next_handler }
+    }
+}
+
+impl<T, S, H> Handler<T, S> for CacheControlMiddleware<H>
+where
+    H: Handler<T, S>,
+    S: Send + 'static,
+{
+    type Future = Pin<Box<dyn Future<Output = Response> + Send + 'static>>;
+
+    fn call(self, req: Request<Body>, state: S) -> Self::Future {
+        Box::pin(async move {
+            let mut response = self.next_handler.call(req, state).await;
+            response.headers_mut().insert(
+                CACHE_CONTROL,
+                HeaderValue::from_static(Self::CACHE_CONTROL_VALUE),
+            );
+            response
+        })
+    }
+}
