tezx/middleware/cacheControl.js

TezX is a high-performance, lightweight JavaScript framework designed for speed, scalability, and flexibility. It enables efficient routing, middleware management, and static file serving with minimal configuration. Fully compatible with Node.js, Deno, an

import { GlobalConfig } from "../core/config.js";
export const cacheControl = (options) => {
    const { defaultSettings, useWeakETag = false, rules = [], logEvent = (event, ctx, error) => {
        if (event === "error") {
            GlobalConfig.debugging.error(`[CACHE] ${event.toUpperCase()}: ${error?.message}`);
        }
        else {
            GlobalConfig.debugging.success(`[CACHE] ${event.toUpperCase()} for ${ctx.method} ${ctx.pathname}`);
        }
    }, } = options;
    return async function cacheControl(ctx, next) {
        if (!["GET", "HEAD"].includes(ctx.method)) {
            return await next();
        }
        try {
            await next();
            const matchedRule = rules.find((rule) => rule.condition(ctx)) || null;
            const { maxAge, scope, enableETag, vary } = matchedRule || defaultSettings;
            const cacheControlValue = `${scope}, max-age=${maxAge}`;
            ctx.header("Cache-Control", cacheControlValue);
            const expiresDate = new Date(Date.now() + maxAge * 1000).toUTCString();
            ctx.header("Expires", expiresDate);
            if (vary?.length) {
                ctx.header("Vary", vary.join(", "));
            }
            if (enableETag) {
                const responseBody = typeof ctx.resBody === "string"
                    ? ctx.resBody
                    : JSON.stringify(ctx.resBody ?? "");
                const etag = await generateETag(responseBody, useWeakETag);
                const ifNoneMatch = ctx.req.headers.get("if-none-match");
                if (ifNoneMatch === etag) {
                    ctx.setStatus = 304;
                    ctx.body = null;
                    logEvent("cached", ctx);
                    return;
                }
                ctx.header("ETag", etag);
            }
            logEvent("cached", ctx);
        }
        catch (error) {
            logEvent("error", ctx, error);
            ctx.setStatus = 500;
            ctx.body = { error: "Failed to set cache headers." };
        }
    };
};
const generateETag = async (content, weak = false) => {
    const crypto = await import("node:crypto");
    const hash = crypto.createHash("md5").update(content).digest("hex");
    return weak ? `W/"${hash}"` : `"${hash}"`;
};