Custom cache

// Define configurable cache duration in seconds (default: 30 days)
const CACHE_DURATION_SECONDS = 30 * 24 * 60 * 60;

// Define which parts of the request to include in the cache key
const USE_PATH = true; // Include path in the cache key
const USE_QUERY_STRING = true; // Include query string in the cache key
const INCLUDE_HEADERS = ["User-Agent"]; // Headers to include in the cache key

export default {
    async fetch(request, env, ctx) {
        // Generate a custom cache key based on user preferences
        const cacheKey = createCacheKey(request);
        console.log(`Retrieving cache for: ${cacheKey.url}.`)

        // Access the default Cache API
        const cache = caches.default;

        // Attempt to retrieve the cached response
        let response = await cache.match(cacheKey);

        if (!response) {
            // Cache miss: Fetch the asset from the origin
            console.log(`Cache miss for: ${cacheKey.url}. Fetching from origin...`);
            response = await fetch(request);

            // Wrap the origin response for caching
            response = new Response(response.body, response);

            // Set Cache-Control headers to define the TTL
            response.headers.set("Cache-Control", `s-maxage=${CACHE_DURATION_SECONDS}`);
            response.headers.set("x-snippets-cache", "stored");

            // Store the response in the cache
            await cache.put(cacheKey, response.clone());
        } else {
            // Cache hit: Return the cached response
            console.log(`Cache hit for: ${cacheKey.url}.`);
            response = new Response(response.body, response);
            response.headers.set("x-snippets-cache", "hit");

            // Optionally check if the cache should expire based on age
            const ageHeader = response.headers.get("Age");
            if (ageHeader && parseInt(ageHeader, 10) > CACHE_DURATION_SECONDS) {
                console.log(`Cache expired for: ${cacheKey.url}. Deleting cached response...`);
                await cache.delete(cacheKey);
                response.headers.set("x-snippets-cache", "deleted");
            }
        }

        // Return the response to the client
        return response;
    },
};

/**
 * Function to create a custom cache key based on request properties
 * @param {Request} request - The incoming request object
 * @returns {Request} - A valid cache key based on the URL
 */
function createCacheKey(request) {
    const url = new URL(request.url); // Use the request's base URL
    const cacheKey = new URL(url.origin); // Start with the origin (scheme + hostname)

    // Optionally include the path
    if (USE_PATH) {
        cacheKey.pathname = url.pathname;
    }

    // Optionally include the query string
    if (USE_QUERY_STRING) {
        cacheKey.search = url.search;
    }

    // Optionally include specific headers
    if (INCLUDE_HEADERS.length > 0) {
        const headerParts = INCLUDE_HEADERS.map(header => `${header}=${request.headers.get(header) || ""}`).join("&");
        cacheKey.searchParams.append("headers", headerParts);
    }

    // Return the constructed URL as the cache key
    return new Request(cacheKey.toString(), {
        method: "GET"
    });
}