feat: add more custom error response codes

Author: ryanoolala

README.md

@@ -67,6 +67,62 @@ module "cloudfront" {
 - Only works with HTML files and requires proper Content-Type headers
 - For Angular applications, also injects `ngCspNonce` attribute on the `<app-root>` element when script nonces are enabled
 
+### Custom Error Responses
+
+This module supports configurable custom error responses for CloudFront distributions. This allows you to customize how different HTTP error codes are handled, such as serving custom error pages or routing all errors to your SPA's index.html file.
+
+By default, the module is configured for Single Page Applications (SPAs) and will route common error codes (400, 403, 404, 500, 503) to `/index.html` with a 200 response code.
+
+**Example Configuration for SPA:**
+```hcl
+module "cloudfront" {
+  source = "..."
+  
+  # Default configuration routes all errors to index.html for SPAs
+  # No additional configuration needed
+}
+```
+
+**Example Configuration for Custom Error Pages:**
+```hcl
+module "cloudfront" {
+  source = "..."
+  
+  custom_error_responses = [
+    {
+      error_code         = 404
+      response_code      = 404
+      response_page_path = "/error/404.html"
+    },
+    {
+      error_code         = 500
+      response_code      = 500
+      response_page_path = "/error/500.html"
+    }
+  ]
+}
+```
+
+**Example Configuration for Mixed Approach:**
+```hcl
+module "cloudfront" {
+  source = "..."
+  
+  custom_error_responses = [
+    {
+      error_code         = 403
+      response_code      = 200
+      response_page_path = "/index.html"
+    },
+    {
+      error_code         = 404
+      response_code      = 404
+      response_page_path = "/error/404.html"
+    }
+  ]
+}
+```
+
 ```
 locals {
   default-src = join(" ", ["default-src", "'none'"])

cloudfront.tf

@@ -119,11 +119,14 @@ resource "aws_cloudfront_distribution" "web_dist" {
     }
   }
 
-  # SPA
-  custom_error_response {
-    error_code         = 403
-    response_code      = 200
-    response_page_path = "/index.html"
+  # Custom error responses
+  dynamic "custom_error_response" {
+    for_each = var.custom_error_responses
+    content {
+      error_code         = custom_error_response.value.error_code
+      response_code      = custom_error_response.value.response_code
+      response_page_path = custom_error_response.value.response_page_path
+    }
   }
 
   dynamic "logging_config" {
@@ -246,6 +249,16 @@ resource "aws_cloudfront_distribution" "web_redirect" {
     }
   }
 
+  # Custom error responses
+  dynamic "custom_error_response" {
+    for_each = var.custom_error_responses
+    content {
+      error_code         = custom_error_response.value.error_code
+      response_code      = custom_error_response.value.response_code
+      response_page_path = custom_error_response.value.response_page_path
+    }
+  }
+
   default_cache_behavior {
     allowed_methods  = ["GET", "HEAD", "OPTIONS"]
     cached_methods   = ["GET", "HEAD", "OPTIONS"]

variables.tf

@@ -127,3 +127,39 @@ variable "enable_compression" {
   default     = true
   type        = bool
 }
+
+variable "custom_error_responses" {
+  description = "Custom error responses for CloudFront distribution. For SPA applications, set response_code to 200 and response_page_path to /index.html"
+  type = list(object({
+    error_code         = number
+    response_code      = number
+    response_page_path = string
+  }))
+  default = [
+    {
+      error_code         = 400
+      response_code      = 200
+      response_page_path = "/index.html"
+    },
+    {
+      error_code         = 403
+      response_code      = 200
+      response_page_path = "/index.html"
+    },
+    {
+      error_code         = 404
+      response_code      = 200
+      response_page_path = "/index.html"
+    },
+    {
+      error_code         = 500
+      response_code      = 200
+      response_page_path = "/index.html"
+    },
+    {
+      error_code         = 503
+      response_code      = 200
+      response_page_path = "/index.html"
+    }
+  ]
+}