JBH-615: Set is a bot header in prerender proxy repo

aligent · Mar 28, 2023 · 243f781 · 243f781
1 parent d397a25
commit 243f781
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/packages/prerender-proxy/lib/handlers/prerender-check.ts b/packages/prerender-proxy/lib/handlers/prerender-check.ts
@@ -10,17 +10,24 @@ export const handler = async (event: CloudFrontRequestEvent): Promise<CloudFront
     // If the request is from a bot, is not a file and is not from prerender
     // then set the x-request-prerender header so the origin-request lambda function
     // alters the origin to prerender.io
-    if (!IS_FILE.test(request.uri) 
-        && IS_BOT.test(request.headers['user-agent'][0].value)
-        && !request.headers['x-prerender']) {
-        request.headers['x-request-prerender'] = [
-            {
-                key: 'x-request-prerender',
-                value: 'true'
-            }
-        ];
+    if (IS_BOT.test(request.headers['user-agent'][0].value)) {
+      request.headers["x-is-a-bot"] = [
+        {
+          key: "x-is-a-bot",
+          value: "true",
+        },
+      ];
+      if (!IS_FILE.test(request.uri) 
+          && !request.headers['x-prerender']) {
+          request.headers['x-request-prerender'] = [
+              {
+                  key: 'x-request-prerender',
+                  value: 'true'
+              }
+          ];
 
-        request.headers['x-prerender-host'] = [{ key: 'X-Prerender-Host', value: request.headers.host[0].value}];
+          request.headers['x-prerender-host'] = [{ key: 'X-Prerender-Host', value: request.headers.host[0].value}];
+      }
     }
 
     return request;