From b8e18fa94774c1018e3f12bc2e99b9b7389d1dd7 Mon Sep 17 00:00:00 2001
From: Olamide <Olamide@thoughtbot.com>
Date: Wed, 14 Aug 2024 18:42:56 +0100
Subject: [PATCH] Enable user data selection by node group name

---
 aws/cluster/main.tf      | 2 +-
 aws/cluster/variables.tf | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/aws/cluster/main.tf b/aws/cluster/main.tf
index 183635e..d39ecbc 100644
--- a/aws/cluster/main.tf
+++ b/aws/cluster/main.tf
@@ -48,7 +48,7 @@ module "node_groups" {
   role           = module.node_role.instance
   subnets        = values(data.aws_subnet.private)
   tags           = var.tags
-  user_data      = var.user_data
+  user_data      = lookup(var.user_data, each.key, null)
 
   depends_on = [module.node_role]
 }
diff --git a/aws/cluster/variables.tf b/aws/cluster/variables.tf
index c6ea101..04a4c44 100644
--- a/aws/cluster/variables.tf
+++ b/aws/cluster/variables.tf
@@ -43,7 +43,7 @@ variable "tags" {
 }
 
 variable "user_data" {
-  type        = string
+  type        = map(string)
   description = "Optional user data script for the launch template"
-  default     = null # Default to an empty string if no user data is provided
+  default     = {} # Default to an empty string if no user data is provided
 }