Added leading slash to Location URL response (#1411)

aligent · Dec 2, 2024 · 0c37465 · 0c37465
1 parent 0e3437c
commit 0c37465
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/packages/prerender-fargate/lib/prerender/server.js b/packages/prerender-fargate/lib/prerender/server.js
@@ -238,11 +238,16 @@ server.use({
       }
 
       let headerMatch = headerMatchRegex.exec(head);
-      while (headerMatch) {
-        s3Metadata.location =
-          headerMatch[1].toLowerCase() == "location"
-            ? he.decode(headerMatch[2] || headerMatch[4])
-            : "";
+        while (headerMatch) {
+            const decoded = he.decode(headerMatch[2] || headerMatch[4])
+            if (headerMatch[1].toLowerCase() == "location") {
+                s3Metadata.location = decoded
+                if (!decoded.startsWith('http') && !decoded.startsWith('/')) {
+                    s3Metadata.location = '/' + s3Metadata.location
+                }
+            } else {
+                s3Metadata.location = ""
+            }
         res.setHeader(headerMatch[1] || headerMatch[3], s3Metadata.location);
         req.prerender.content = req.prerender.content
           .toString()