hono
Version:
Web framework built on Web Standards
103 lines (102 loc) • 3.83 kB
JavaScript
// src/middleware/cache/index.ts
var defaultCacheableStatusCodes = [200];
var shouldSkipCacheControl = (cacheControl) => !!cacheControl && /(?:^|,\s*)(?:private|no-(?:store|cache))(?:\s*(?:=|,|$))/i.test(cacheControl);
var parseVaryDirectives = (vary) => {
if (vary == null) {
return [];
}
return (Array.isArray(vary) ? vary : vary.split(",")).map((directive) => directive.trim().toLowerCase()).filter(Boolean);
};
var shouldSkipCache = (res, optionsVaryDirectives, responseVary) => responseVary.length && (!optionsVaryDirectives || responseVary.some((name) => !optionsVaryDirectives.has(name))) || shouldSkipCacheControl(res.headers.get("Cache-Control")) || res.headers.has("Set-Cookie");
var cache = (options) => {
if (!globalThis.caches) {
if (options.onCacheNotAvailable === false) {
} else if (options.onCacheNotAvailable) {
options.onCacheNotAvailable();
} else {
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 optionsVaryList = parseVaryDirectives(options.vary);
const varyDirectives = optionsVaryList.length ? new Set(optionsVaryList) : void 0;
if (varyDirectives?.has("*")) {
throw new Error(
'Middleware vary configuration cannot include "*", as it disallows effective caching.'
);
}
const cacheableStatusCodes = new Set(
options.cacheableStatusCodes ?? defaultCacheableStatusCodes
);
const addHeader = (c, responseVary) => {
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) {
if (responseVary.length === 0) {
c.header("Vary", Array.from(varyDirectives).join(", "));
} else {
const merged = new Set(varyDirectives);
for (const directive of responseVary) {
merged.add(directive);
}
if (merged.has("*")) {
c.header("Vary", "*");
} else {
c.header("Vary", Array.from(merged).join(", "));
}
}
}
};
return async function cache2(c, next) {
if (c.req.method !== "GET" || c.req.raw.headers.has("Authorization")) {
await next();
return;
}
let key = c.req.url;
if (options.keyGenerator) {
key = await options.keyGenerator(c);
}
if (varyDirectives) {
for (const directive of varyDirectives) {
const value = c.req.raw.headers.get(directive) ?? "";
key += `::${directive}=${encodeURIComponent(value)}`;
}
}
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 (!cacheableStatusCodes.has(c.res.status)) {
return;
}
const responseVary = parseVaryDirectives(c.res.headers.get("Vary"));
addHeader(c, responseVary);
if (shouldSkipCache(c.res, varyDirectives, responseVary)) {
return;
}
const res = c.res.clone();
if (options.wait) {
await cache3.put(key, res);
} else {
c.executionCtx.waitUntil(cache3.put(key, res));
}
};
};
export {
cache
};