From 243f781cb417742e9ffdf271695d2071f2577b72 Mon Sep 17 00:00:00 2001
From: Krishan Thisera <krishan.thisera@aligent.com.au>
Date: Tue, 28 Mar 2023 13:02:21 +1030
Subject: [PATCH] JBH-615: Set is a bot header in prerender proxy repo

---
 .../lib/handlers/prerender-check.ts           | 27 ++++++++++++-------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/packages/prerender-proxy/lib/handlers/prerender-check.ts b/packages/prerender-proxy/lib/handlers/prerender-check.ts
index 587230db..6e289df6 100644
--- 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;