fix: Support multiple custom response body/header

Author: uyggnodoow

examples/CustomResponse/main.tf

@@ -22,7 +22,11 @@ module "wafv2" {
         response_code = 404
         response_header = [
           {
-            name  = "X-Custom-Response-Header-01"
+            name  = "X-Custom-Response-Header01"
+            value = "Not authorized"
+          },
+          {
+            name  = "X-Custom-Response-Header02"
             value = "Not authorized"
           }
         ]
@@ -38,11 +42,18 @@ module "wafv2" {
     }
   ]
 
-  custom_response_body = {
-    key          = "CustomResponseBody",
-    content      = "Not authorized",
-    content_type = "TEXT_PLAIN"
-  }
+  custom_response_body = [
+    {
+      key          = "CustomResponseBody1",
+      content      = "Not authorized1",
+      content_type = "TEXT_PLAIN"
+    },
+    {
+      key          = "CustomResponseBody2",
+      content      = "Not authorized2",
+      content_type = "TEXT_PLAIN"
+    }
+  ]
 
   visibility_config = {
     cloudwatch_metrics_enabled = false

main.tf

@@ -77,9 +77,11 @@ resource "aws_wafv2_web_acl" "this" {
 
                   dynamic "response_header" {
                     for_each = lookup(custom_response.value, "response_header", [])
+                    iterator = response_header
+
                     content {
-                      name  = lookup(response_header.value, "name")
-                      value = lookup(response_header.value, "value")
+                      name  = response_header.value.name
+                      value = response_header.value.value
                     }
                   }
                 }
@@ -12239,11 +12241,13 @@ resource "aws_wafv2_web_acl" "this" {
   }
 
   dynamic "custom_response_body" {
-    for_each = var.custom_response_body
+    for_each = var.custom_response_body == null ? [] : var.custom_response_body
+    iterator = custom_response_body
+
     content {
-      content      = var.custom_response_body.content
-      content_type = var.custom_response_body.content_type
-      key          = var.custom_response_body.key
+      content      = custom_response_body.value.content
+      content_type = custom_response_body.value.content_type
+      key          = custom_response_body.value.key
     }
   }
 

variables.tf

@@ -32,8 +32,12 @@ variable "visibility_config" {
 
 variable "custom_response_body" {
   description = "(Optional) Defines custom response bodies that can be referenced by custom_response actions."
-  type        = map(any)
-  default     = {}
+  type = list(object({
+    content      = string
+    content_type = string
+    key          = string
+  }))
+  default = []
 }
 
 variable "captcha_config" {