fix: handle more cache directives

Author: ronag

lib/interceptor/cache.js

@@ -42,20 +42,26 @@ class CacheHandler extends DecoratorHandler {
       return super.onHeaders(statusCode, headers, resume)
     }
 
-    const cacheControl = parseCacheControl(headers['cache-control'])
+    const cacheControl = parseCacheControl(headers['cache-control']) ?? {}
+
+    if (this.#key.headers.authorization && !cacheControl.public) {
+      return super.onHeaders(statusCode, headers, resume)
+    }
+
+    if (cacheControl.private || cacheControl['no-store']) {
+      return super.onHeaders(statusCode, headers, resume)
+    }
+
     if (
-      !cacheControl ||
-      !cacheControl.public ||
-      cacheControl.private ||
-      cacheControl['no-store'] ||
-      // TODO (fix): Support all cache control directives...
-      // cacheControl['no-transform'] ||
-      cacheControl['no-cache'] ||
-      cacheControl['must-understand'] ||
+      cacheControl['no-transform'] ||
       cacheControl['must-revalidate'] ||
-      cacheControl['proxy-revalidate']
+      cacheControl['proxy-revalidate'] ||
+      cacheControl['must-understand'] ||
+      cacheControl['stale-while-revalidate'] ||
+      cacheControl['stale-if-error'] ||
+      cacheControl['no-cache']
     ) {
-      // Not cacheble...
+      // TODO (fix): Support all cache control directives...
       return super.onHeaders(statusCode, headers, resume)
     }
 
@@ -134,16 +140,17 @@ export default () => (dispatch) => (opts, handler) => {
     return dispatch(opts, handler)
   }
 
-  if (opts.headers?.['cache-control'] || opts.headers?.authorization) {
+  const cacheControl = parseCacheControl(opts?.headers['cache-control']) ?? {}
+
+  if (
+    cacheControl['max-age'] ||
+    cacheControl['max-stale'] ||
+    cacheControl['min-fresh'] ||
+    cacheControl['no-cache'] ||
+    cacheControl['no-transform'] ||
+    cacheControl['stale-if-error']
+  ) {
     // TODO (fix): Support all cache control directives...
-    // const cacheControl = cacheControlParser.parse(opts.headers['cache-control'])
-    // cacheControl['no-cache']
-    // cacheControl['no-store']
-    // cacheControl['max-age']
-    // cacheControl['max-stale']
-    // cacheControl['min-fresh']
-    // cacheControl['no-transform']
-    // cacheControl['only-if-cached']
     return dispatch(opts, handler)
   }
 
@@ -152,10 +159,12 @@ export default () => (dispatch) => (opts, handler) => {
 
   const store = opts.cache.store ?? DEFAULT_STORE
   const entry = store.get(opts)
-  if (!entry) {
+  if (!entry && !cacheControl['only-if-cached']) {
     return dispatch(
       opts,
-      new CacheHandler(undici.util.cache.makeCacheKey(opts), { store, handler }),
+      cacheControl['no-store']
+        ? handler
+        : new CacheHandler(undici.util.cache.makeCacheKey(opts), { store, handler }),
     )
   }
 
@@ -174,7 +183,7 @@ export default () => (dispatch) => (opts, handler) => {
     }
   }
 
-  const { statusCode, headers } = entry
+  const { statusCode, headers } = entry ?? { statusCode: 504, headers: {} }
   try {
     handler.onConnect(abort)
     if (aborted) {