First commit

Author: luke-owen-crowdhandler

.gitignore

@@ -0,0 +1,12 @@
+# package directories
+node_modules
+jspm_packages
+
+# Serverless directories
+.serverless
+.webpack
+
+# custom
+dummy-payloads
+dist/*.json
+.DS_Store

dist/originResponse.zip

@@ -0,0 +1,123 @@
+"use strict";
+
+const helpers = require("./helpers/misc");
+const http_helpers = require("./helpers/http");
+
+require("source-map-support").install();
+
+module.exports.originResponse = async (event) => {
+  let APIDomain;
+  let crowdhandlerResponseID;
+  let crowdhandlerToken;
+  let publicKey;
+  let requestTime;
+
+  // Infer useful information from the request event and save it
+  let request = event.Records[0].cf.request;
+  let response = event.Records[0].cf.response;
+  const requestHeaders = request.headers;
+  const uri = request.uri;
+
+  // Don't try and queue static assets
+  let fileExtension = uri.split(".").pop();
+
+  // No need to execute anymore of this script if the request is for a static file.
+  const creativeAssetExtensions = helpers.creativeAssetExtensions;
+  if (creativeAssetExtensions.indexOf(fileExtension) !== -1) {
+    console.log("Static file detected");
+    return response;
+  }
+
+  // Environment Setup
+  (function () {
+    // Guard against the crowdhandlerResponseID value coming back null from the API
+    try {
+      if (requestHeaders["x-ch-responseID"]) {
+        crowdhandlerResponseID = requestHeaders["x-ch-responseID"][0].value;
+      }
+    } catch (error) {
+      console.error(error);
+    }
+
+    // Handle lack of valid token i.e. partial API responses for unprotected rooms.
+    try {
+      //Make sure we don't set invalid values
+      const validToken = /(.*\d+.*)/
+
+      if (requestHeaders["x-ch-crowdhandler-token"] && validToken.test(requestHeaders["x-ch-crowdhandler-token"][0].value) === true) {
+        crowdhandlerToken = requestHeaders["x-ch-crowdhandler-token"][0].value;
+      }
+    } catch (error) {
+      console.error(error);
+      return response;
+    }
+
+    APIDomain = requestHeaders["x-ch-api-endpoint"][0].value;
+    publicKey = requestHeaders["x-ch-public-key"][0].value;
+    requestTime = requestHeaders["x-ch-request-time"][0].value;
+  })();
+
+  // If we don't have a valid crowdhandlerToken by this stage return response.
+  if (!crowdhandlerToken) {
+    return response;
+  }
+
+  // Requested Domain
+  const host = requestHeaders.host[0].value;
+
+  // Set cookies
+  (function () {
+    if (response.headers["set-cookie"]) {
+      response.headers["set-cookie"].push({
+        key: "Set-Cookie",
+        value: `crowdhandler=${crowdhandlerToken}; path=/; Secure; HttpOnly`,
+      });
+      response.headers["set-cookie"].push({
+        key: "Set-Cookie",
+        value: `crowdhandler_integration=cloudfront; path=/; Secure;`,
+      });
+    } else {
+      response.headers["set-cookie"] = [
+        {
+          key: "Set-Cookie",
+          value: `crowdhandler=${crowdhandlerToken}; path=/; Secure; HttpOnly`,
+        },
+      ];
+      response.headers["set-cookie"].push({
+        key: "Set-Cookie",
+        value: `crowdhandler_integration=cloudfront; path=/; Secure;`,
+      });
+    }
+  })();
+
+  const totalLoadTime = Date.now() - requestTime;
+
+  async function sendPageLoadTime() {
+    let response;
+    try {
+      response = await http_helpers.httpPUT(
+        {
+          hostname: APIDomain,
+          port: 443,
+          path: `/v1/responses/${crowdhandlerResponseID}`,
+          method: "PUT",
+          headers: {
+            "x-api-key": publicKey,
+            "Content-Type": "application/json",
+          },
+        },
+        JSON.stringify({
+          httpCode: 200,
+          sampleRate: 100,
+          time: totalLoadTime,
+        })
+      );
+    } catch (error) {
+      console.error(error);
+    }
+  }
+
+  await sendPageLoadTime();
+
+  return response;
+};

dist/viewerOverride.zip

@@ -0,0 +1,66 @@
+"use strict";
+
+const helpers = require("./helpers/misc");
+const http_helpers = require("./helpers/http");
+
+require("source-map-support").install();
+
+module.exports.viewerOverride = async (event) => {
+  let request = event.Records[0].cf.request;
+  let queryString = request.querystring;
+  // Trim /ch/
+  let uri = request.uri.substring(4);
+
+  // If we have a URI after trimming it means we have been sent a slug that can be used.
+  let fallbackPath = `/?${queryString}`;
+  let templateDomain = "wait.crowdhandler.com";
+  let templatePath;
+  if (uri) {
+    templatePath = `/${uri}`;
+  } else {
+    templatePath = fallbackPath;
+  }
+
+  // Template fetch with retry mechanic.
+  let fetchCounter = 0;
+  async function fetchTemplate(retry) {
+    let headers = {
+      "Content-Type": "text/html",
+    };
+    let response;
+    fetchCounter++;
+
+    if (fetchCounter === 3) {
+      templatePath = fallbackPath;
+    }
+
+    try {
+      response = await http_helpers.httpGET({
+        headers: headers,
+        hostname: templateDomain,
+        method: "GET",
+        path: templatePath,
+        port: 443,
+      });
+      return response;
+    } catch (error) {
+      console.log(error);
+      response = null;
+      return response;
+    }
+  }
+
+  let templateBody;
+  while (!templateBody && fetchCounter < 4) {
+    templateBody = await fetchTemplate();
+  }
+
+  if (templateBody) {
+    let healthyResponse = http_helpers.http200Response(templateBody);
+    return healthyResponse;
+    // Handle failure to retrieve template
+  } else {
+    let errorResponse = http_helpers.error404Response();
+    throw errorResponse;
+  }
+};