Add condition to create function

cloudfront.tf

@@ -71,8 +71,10 @@ module "cdn" {
 }
 
 resource "aws_cloudfront_function" "viewer_request" {
+  count   = var.associate_function ? 1 : 0
   name    = var.default_index_function_name
   runtime = "cloudfront-js-1.0"
   publish = true
   code    = var.override_default_index_function_code == "" ? templatefile("${path.module}/templates/viewer-request-default.js", { default_root_object = var.default_root_object }) : var.override_default_index_function_code
 }
+

locals.tf

@@ -1,9 +1,8 @@
 locals {
-  default_function_association = length(var.default_cache_behavior) == 0 ? {
-    function_association = {
-      viewer-request = {
-        function_arn = aws_cloudfront_function.viewer_request.arn
-      }
+  function_association = (var.associate_function && length(var.default_cache_behavior) == 0) ? {
+    viewer-request = {
+      function_arn = aws_cloudfront_function.viewer_request.arn
     }
   } : {}
 }
+

variables.tf

@@ -283,6 +283,12 @@ variable "web_acl_id" {
   default     = ""
 }
 
+variable "associate_function" {
+  description = "If the CloudFront function should be associated with the default cache behavior."
+  type        = bool
+  default     = false
+}
+
 variable "default_index_function_name" {
   description = "Name of the CloudFront Function to create for index page redirection"
   type        = string