feat(router): global cache pre-route

1 files changed, 41 insertions, 0 deletions

diff --git a/src/modules/router/module.rs b/src/modules/router/module.rs
index 76083b8..a51812b 100644
--- a/src/modules/router/module.rs
+++ b/src/modules/router/module.rs
@@ -16,8 +16,47 @@
 // Copyright (C) 2022-2022 Fuwn <[email protected]>
 // SPDX-License-Identifier: GPL-3.0-only
 
+use std::sync::{LazyLock, Mutex};
+
+use tokio::time::Instant;
 use windmark::context::HookContext;
 
+use crate::route::{CACHE_RATE, ROUTES};
+
+fn cache(context: &windmark::context::HookContext, response: &str) {
+  static LAST_CACHED: LazyLock<Mutex<Instant>> =
+    LazyLock::new(|| Mutex::new(Instant::now()));
+
+  if (*LAST_CACHED.lock().unwrap()).elapsed()
+    >= std::time::Duration::from_secs(CACHE_RATE)
+    || (*ROUTES.lock().unwrap())
+      .get(context.url.path())
+      .is_some_and(|r| r.text_cache.is_empty())
+  {
+    (*LAST_CACHED.lock().unwrap()) = Instant::now();
+
+    if let Some(route) = (*ROUTES.lock().unwrap()).get_mut(context.url.path()) {
+      route.text_cache = response.to_string();
+      info!("cache set for {}", context.url.path());
+    } else {
+      warn!(
+        "cache could not be set for {} as it is not being tracked",
+        context.url.path()
+      );
+    }
+
+    trace!("recache for {}", {
+      let path = context.url.path();
+
+      if path.is_empty() {
+        "/"
+      } else {
+        path
+      }
+    });
+  }
+}
+
 pub fn module(router: &mut windmark::router::Router) {
   router.set_pre_route_callback(move |context: HookContext| {
     info!(
@@ -36,6 +75,8 @@ pub fn module(router: &mut windmark::router::Router) {
         context.url.to_string(),
       );
 
+      cache(&context, &response.content);
+
       if let Some(language) =
         windmark::utilities::queries_from_url(&context.url).get("translate")
       {