vtex proxy: replace only text/html and add referer for AviseMe path (#…

…864) * check content type before replace * add referer * fix: transfer referer logic to vtex loader * add comment --------- Co-authored-by: Gustavo Martins Vasconcellos <gustavo@carcara.tech>
deco-cx · Sep 24, 2024 · 877e238 · 877e238
1 parent ac67a2f
commit 877e238
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 12 deletions.
diff --git a/vtex/loaders/proxy.ts b/vtex/loaders/proxy.ts
@@ -59,18 +59,38 @@ const buildProxyRoutes = (
     const urlToProxy = `https://${hostname}`;
     const hostToUse = hostname;
 
-    const routeFromPath = (pathTemplate: string): Route => ({
-      pathTemplate,
-      handler: {
-        value: {
-          __resolveType: "website/handlers/proxy.ts",
-          url: urlToProxy,
-          host: hostToUse,
-          includeScriptsToHead,
-          removeDirtyCookies: true,
+    const routeFromPath = (pathTemplate: string): Route => {
+      const handlerValue = {
+        __resolveType: "website/handlers/proxy.ts",
+        url: urlToProxy,
+        host: hostToUse,
+        includeScriptsToHead,
+        removeDirtyCookies: true,
+      };
+      // we have this check because we need to add
+      // the referer header to the AviseMe route
+      if (pathTemplate.includes("AviseMe")) {
+        return {
+          pathTemplate,
+          handler: {
+            value: {
+              ...handlerValue,
+              customHeaders: [{
+                key: "referer",
+                value: urlToProxy,
+              }],
+            },
+          },
+        };
+      }
+
+      return ({
+        pathTemplate,
+        handler: {
+          value: handlerValue,
         },
-      },
-    });
+      });
+    };
     const routesFromPaths = [...PATHS_TO_PROXY, ...extraPaths].map(
       routeFromPath,
     );

diff --git a/website/handlers/proxy.ts b/website/handlers/proxy.ts
@@ -180,7 +180,7 @@ export default function Proxy({
       }
     }
     let text: undefined | string = undefined;
-    if (replaces && replaces.length > 0) {
+    if (replaces && replaces.length > 0 && contentType?.includes("text/html")) {
       if (response.ok) {
         text = await new Response(newBody).text();
         replaces.forEach(({ from, to }) => {