69 lines
2.3 KiB
JavaScript
69 lines
2.3 KiB
JavaScript
// src/middleware/cache/index.ts
|
|
var cache = (options) => {
|
|
if (!globalThis.caches) {
|
|
console.log("Cache Middleware is not enabled because caches is not defined.");
|
|
return async (_c, next) => await next();
|
|
}
|
|
if (options.wait === void 0) {
|
|
options.wait = false;
|
|
}
|
|
const cacheControlDirectives = options.cacheControl?.split(",").map((directive) => directive.toLowerCase());
|
|
const varyDirectives = Array.isArray(options.vary) ? options.vary : options.vary?.split(",").map((directive) => directive.trim());
|
|
if (options.vary?.includes("*")) {
|
|
throw new Error(
|
|
'Middleware vary configuration cannot include "*", as it disallows effective caching.'
|
|
);
|
|
}
|
|
const addHeader = (c) => {
|
|
if (cacheControlDirectives) {
|
|
const existingDirectives = c.res.headers.get("Cache-Control")?.split(",").map((d) => d.trim().split("=", 1)[0]) ?? [];
|
|
for (const directive of cacheControlDirectives) {
|
|
let [name, value] = directive.trim().split("=", 2);
|
|
name = name.toLowerCase();
|
|
if (!existingDirectives.includes(name)) {
|
|
c.header("Cache-Control", `${name}${value ? `=${value}` : ""}`, { append: true });
|
|
}
|
|
}
|
|
}
|
|
if (varyDirectives) {
|
|
const existingDirectives = c.res.headers.get("Vary")?.split(",").map((d) => d.trim()) ?? [];
|
|
const vary = Array.from(
|
|
new Set(
|
|
[...existingDirectives, ...varyDirectives].map((directive) => directive.toLowerCase())
|
|
)
|
|
).sort();
|
|
if (vary.includes("*")) {
|
|
c.header("Vary", "*");
|
|
} else {
|
|
c.header("Vary", vary.join(", "));
|
|
}
|
|
}
|
|
};
|
|
return async function cache2(c, next) {
|
|
let key = c.req.url;
|
|
if (options.keyGenerator) {
|
|
key = await options.keyGenerator(c);
|
|
}
|
|
const cacheName = typeof options.cacheName === "function" ? await options.cacheName(c) : options.cacheName;
|
|
const cache3 = await caches.open(cacheName);
|
|
const response = await cache3.match(key);
|
|
if (response) {
|
|
return new Response(response.body, response);
|
|
}
|
|
await next();
|
|
if (!c.res.ok) {
|
|
return;
|
|
}
|
|
addHeader(c);
|
|
const res = c.res.clone();
|
|
if (options.wait) {
|
|
await cache3.put(key, res);
|
|
} else {
|
|
c.executionCtx.waitUntil(cache3.put(key, res));
|
|
}
|
|
};
|
|
};
|
|
export {
|
|
cache
|
|
};
|