Merge pull request #5 from Crowdhandler/improvement/set-cookie-on-redirect

Stickier cookies to improve UX when origin is intermittently struggli…

Author: luke-owen-crowdhandler

dist/originOverride.zip

@@ -229,7 +229,7 @@ module.exports.viewerRequest = async (event) => {
     case true: {
       // redirect
       console.log("redirecting...");
-      return http_helpers.redirect302Response(redirectLocation);
+      return http_helpers.redirect302Response(redirectLocation, result.token);
       break;
     }
     case false: {

dist/viewerRequest.zip

@@ -135,19 +135,42 @@ export const http200Response = function (content) {
 };
 
 // Generic 302 Response
-export const redirect302Response = function (redirect_location) {
-  const response = {
-    status: "302",
-    statusDescription: "Found",
-    headers: {
-      location: [
-        {
-          key: "Location",
-          value: redirect_location,
-        },
-      ],
-    },
-  };
+export const redirect302Response = function (redirect_location, token) {
+  let response;
+
+  if (token) {
+    response = {
+      status: "302",
+      statusDescription: "Found",
+      headers: {
+        location: [
+          {
+            key: "Location",
+            value: redirect_location,
+          },
+        ],
+        "set-cookie": [
+          {
+            key: "Set-Cookie",
+            value: `crowdhandler=${token}; path=/; Secure; HttpOnly`,
+          },
+        ],
+      },
+    };
+  } else {
+    response = {
+      status: "302",
+      statusDescription: "Found",
+      headers: {
+        location: [
+          {
+            key: "Location",
+            value: redirect_location,
+          },
+        ],
+      },
+    };
+  }
   return response;
 };