From 33505e9f7cebf72fddfd8739d03ca23744217eef Mon Sep 17 00:00:00 2001
From: Matt Petro <matt.petro@atomicjolt.com>
Date: Fri, 27 Sep 2024 10:47:04 -0400
Subject: [PATCH] Add support for al2023 self-managed nodes

To use the new nodes, add this to the worker configuration
  platform = 'al2023'
---
 locals.tf                     | 4 ++++
 templates/userdata_al2023.tpl | 9 +++++++++
 2 files changed, 13 insertions(+)
 create mode 100644 templates/userdata_al2023.tpl

diff --git a/locals.tf b/locals.tf
index 8e1b54e49c..a4f5ae77e2 100644
--- a/locals.tf
+++ b/locals.tf
@@ -8,6 +8,7 @@ locals {
   cluster_auth_base64               = coalescelist(aws_eks_cluster.this[*].certificate_authority[0].data, [""])[0]
   cluster_oidc_issuer_url           = flatten(concat(aws_eks_cluster.this[*].identity[*].oidc[0].issuer, [""]))[0]
   cluster_primary_security_group_id = coalescelist(aws_eks_cluster.this[*].vpc_config[0].cluster_security_group_id, [""])[0]
+  cluster_service_cidr              = coalescelist(aws_eks_cluster.this[*].kubernetes_network_config.service_ipv4_cidr, [""])[0]
 
   cluster_security_group_id = var.cluster_create_security_group ? join("", aws_security_group.cluster.*.id) : var.cluster_security_group_id
   cluster_iam_role_name     = var.manage_cluster_iam_resources ? join("", aws_iam_role.cluster.*.name) : var.cluster_iam_role_name
@@ -180,6 +181,8 @@ locals {
         "userdata_template_file",
         lookup(var.worker_groups[index], "platform", local.workers_group_defaults["platform"]) == "windows"
         ? "${path.module}/templates/userdata_windows.tpl"
+        : lookup(var.worker_groups[index], "platform", local.workers_group_defaults["platform"]) == "al2023"
+        ? "${path.module}/templates/userdata_al2023.tpl"
         : "${path.module}/templates/userdata.sh.tpl"
       ),
       merge({
@@ -187,6 +190,7 @@ locals {
         cluster_name        = local.cluster_name
         endpoint            = local.cluster_endpoint
         cluster_auth_base64 = local.cluster_auth_base64
+        cluster_service_cidr = local.cluster_service_cidr
         pre_userdata = lookup(
           var.worker_groups[index],
           "pre_userdata",
diff --git a/templates/userdata_al2023.tpl b/templates/userdata_al2023.tpl
new file mode 100644
index 0000000000..665628e78b
--- /dev/null
+++ b/templates/userdata_al2023.tpl
@@ -0,0 +1,9 @@
+---
+apiVersion: node.eks.aws/v1alpha1
+kind: NodeConfig
+spec:
+  cluster:
+    name: ${cluster_name}
+    apiServerEndpoint: ${endpoint}
+    certificateAuthority: ${cluster_auth_base64}
+    cidr: ${cluster_service_cidr}