Add support do edgelambda on site-redirect

grupoboticario · Jun 28, 2021 · 20f4506 · 20f4506
1 parent 7444498
commit 20f4506
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/site-redirect/main.tf b/site-redirect/main.tf
@@ -115,6 +115,14 @@ resource "aws_cloudfront_distribution" "website_cdn" {
     allowed_methods = ["GET", "HEAD", "DELETE", "OPTIONS", "PATCH", "POST", "PUT"]
     cached_methods  = ["GET", "HEAD"]
 
+    dynamic "lambda_function_association" {
+      for_each = var.enable_lambda_sec_headers == null ? [] : var.enable_lambda_sec_headers
+      content {
+        event_type = lambda_function_association.value.event_type
+        lambda_arn = lambda_function_association.value.lambda_arn
+      }
+    }
+
     forwarded_values {
       query_string = var.forward-query-string
 

diff --git a/site-redirect/variables.tf b/site-redirect/variables.tf
@@ -45,4 +45,14 @@ variable "forward-query-string" {
   type        = bool
   description = "Forward the query string to the origin"
   default     = true
+}
+
+variable "enable_lambda_sec_headers" {
+  type = list(object({
+    event_type = string
+    lambda_arn = string
+  }))
+  default = null
+
+  description = "Specifies the lambda function of security headers"
 }