Fix cache response on redirect

Author: lorenzofiamingo

Sources/CachedAsyncImage/CachedAsyncImage.swift

@@ -354,6 +354,8 @@ private extension CachedAsyncImage {
     private func remoteImage(from request: URLRequest, session: URLSession) async throws -> (Image, URLSessionTaskMetrics) {
         let (data, _, metrics) = try await session.data(for: request)
         if metrics.redirectCount > 0, let lastResponse = metrics.transactionMetrics.last?.response {
+            let requests = metrics.transactionMetrics.map { $0.request }
+            requests.forEach(session.configuration.urlCache!.removeCachedResponse)
             let lastCachedResponse = CachedURLResponse(response: lastResponse, data: data)
             session.configuration.urlCache!.storeCachedResponse(lastCachedResponse, for: request)
         }
@@ -402,3 +404,23 @@ private extension URLSession {
         return (data, response, controller.metrics!)
     }
 }
+
+extension URLCache {
+    
+    func cachedResponse(for request: URLRequest, followRedirects: Bool) -> CachedURLResponse? {
+        if
+            followRedirects,
+            let response = cachedResponse(for: request)?.response as? HTTPURLResponse,
+            [301, 302, 303, 307, 308].contains(response.statusCode),
+            let redirectedURLString = response.allHeaderFields["Location"] as? String,
+            let redirectedURL = URL(string: redirectedURLString)
+        {
+            var redirectedRequest = request
+            redirectedRequest.url = redirectedURL
+            let redirectedResponse = cachedResponse(for: redirectedRequest, followRedirects: followRedirects)
+            return redirectedResponse
+        } else {
+            return cachedResponse(for: request)
+        }
+    }
+}