Merge pull request #90 from abdelaziz-mahdy/add-cache-headers

feat(cache): add Cache-Control middleware and update .env.example

2 files changed, 17 insertions, 0 deletions

diff --git a/src/config/cacheControlMiddleware.ts b/src/config/cacheControlMiddleware.ts
new file mode 100644
index 0000000..47035ed
--- /dev/null
+++ b/src/config/cacheControlMiddleware.ts
@@ -0,0 +1,15 @@
+import type { MiddlewareHandler } from "hono";
+
+// Define middleware to add Cache-Control header
+const cacheControlMiddleware: MiddlewareHandler = async (c, next) => {
+  const sMaxAge = process.env.ANIWATCH_API_S_MAXAGE || "60";
+  const staleWhileRevalidate = process.env.ANIWATCH_API_STALE_WHILE_REVALIDATE || "30";
+  c.header(
+    "Cache-Control",
+    `s-maxage=${sMaxAge}, stale-while-revalidate=${staleWhileRevalidate}`
+  );
+
+  await next();
+};
+
+export default cacheControlMiddleware;
diff --git a/src/server.ts b/src/server.ts
index 41a64c5..85d3fbc 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -5,6 +5,7 @@ import corsConfig from "./config/cors.js";
 import { ratelimit } from "./config/ratelimit.js";
 
 import { hianimeRouter } from "./routes/hianime.js";
+import cacheControlMiddleware from "./config/cacheControlMiddleware.js";
 
 import { Hono } from "hono";
 import { logger } from "hono/logger";
@@ -26,6 +27,7 @@ const app = new Hono<{ Variables: AniwatchAPIVariables }>();
 
 app.use(logger());
 app.use(corsConfig);
+app.use(cacheControlMiddleware);
 
 // CAUTION: For personal deployments, "refrain" from having an env
 // named "ANIWATCH_API_HOSTNAME". You may face rate limitting