From 567dd0ea2c0f684a113a94a9bc37ea672f251c64 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carlos=20Cola=C3=A7o?= <coolapso@coolapso.tech>
Date: Tue, 2 Jul 2024 15:12:44 +0200
Subject: [PATCH] feat: Add price class

---
 README.md    | 1 +
 main.tf      | 1 +
 variables.tf | 6 ++++++
 3 files changed, 8 insertions(+)

diff --git a/README.md b/README.md
index 963d9a7..b43fa89 100644
--- a/README.md
+++ b/README.md
@@ -64,6 +64,7 @@ No modules.
 | <a name="input_geo_restriction_type"></a> [geo\_restriction\_type](#input\_geo\_restriction\_type) | whitelist/blacklist | `string` | `"none"` | no |
 | <a name="input_max_ttl"></a> [max\_ttl](#input\_max\_ttl) | Max cache ttl | `number` | `3600` | no |
 | <a name="input_min_ttl"></a> [min\_ttl](#input\_min\_ttl) | Min cache ttl | `number` | `0` | no |
+| <a name="input_price_class"></a> [price\_class](#input\_price\_class) | Price class for this distribution | `string` | `"PriceClass_All"` | no |
 | <a name="input_s3_bucket_id"></a> [s3\_bucket\_id](#input\_s3\_bucket\_id) | Name of s3 bucket | `string` | n/a | yes |
 | <a name="input_s3_objects"></a> [s3\_objects](#input\_s3\_objects) | The s3 onjects to allow access to ARN/objects | `list(any)` | n/a | yes |
 | <a name="input_s3_origin_id"></a> [s3\_origin\_id](#input\_s3\_origin\_id) | unique origin id | `string` | `"s3Website"` | no |
diff --git a/main.tf b/main.tf
index 4711aa4..988bade 100644
--- a/main.tf
+++ b/main.tf
@@ -49,6 +49,7 @@ resource "aws_cloudfront_distribution" "this" {
   is_ipv6_enabled     = var.enable_ipv6
   default_root_object = var.default_root_object
   aliases             = var.aliases
+  price_class         = var.price_class
 
 
   default_cache_behavior {
diff --git a/variables.tf b/variables.tf
index 3bc67a7..07018ae 100644
--- a/variables.tf
+++ b/variables.tf
@@ -65,6 +65,12 @@ variable "aliases" {
   default     = null
 }
 
+variable "price_class" {
+  description = "Price class for this distribution"
+  type        = string
+  default     = "PriceClass_All"
+}
+
 variable "forward_query_strings" {
   description = "Enables/disables query string forwarding"
   type        = bool