Added ability to redirect to a host that isn't the cf distribution

Author: robbytaylor

index.js

@@ -2,7 +2,7 @@ exports.handler = (event, context, callback) => {
     const request = event.Records[0].cf.request;
     const hostname = request.headers.host[0].value;
 
-    if ("${primary_hostname}" && hostname === "${primary_hostname}") {
+    if (hostname === "${redirect_to}") {
         return callback(null, request);
     }
 
@@ -12,7 +12,7 @@ exports.handler = (event, context, callback) => {
         headers: {
             location: [{
                 key: 'Location',
-                value: "https://${primary_hostname}" + request.uri + (request.querystring === "" ? "" : "?" + request.querystring),
+                value: "https://${redirect_to}" + request.uri + (request.querystring === "" ? "" : "?" + request.querystring),
             }],
         },
     }

lambda.tf

@@ -3,7 +3,7 @@ data template_file lambda {
   template = "${file("${path.module}/index.js")}"
 
   vars = {
-    primary_hostname = var.redirect_all ? "" : var.cloudfront_distribution
+    redirect_to = var.redirect_to ? var.redirect_to : var.cloudfront_distribution
   }
 }
 

variables.tf

@@ -58,8 +58,8 @@ variable lambda_function_name {
   description = "The name of the Lambda@Edge function for redirecting to the primary hostname"
 }
 
-variable redirect_all {
-  type        = bool
-  default     = false
-  description = "Whether all requests should be redirected, or just ones which don't match the CloudFront distribution primary host"
+variable redirect_to {
+  type        = string
+  default     = ""
+  description = "Where to redirect requests to, if not set the cloudfront distribution host will be used"
 }