From ad53adef63e93b319450d52c0b6cfae402fd6414 Mon Sep 17 00:00:00 2001
From: Felix Peters <felix.peters@breuninger.de>
Date: Thu, 11 Apr 2024 21:17:37 +0200
Subject: [PATCH] feat: add api_versions and kube_version to helm block

---
 docs/data-sources/overlay.md                        | 4 ++++
 kustomize/data_source_kustomization_overlay.go      | 9 +++++++++
 kustomize/data_source_kustomization_overlay_test.go | 5 +++++
 3 files changed, 18 insertions(+)

diff --git a/docs/data-sources/overlay.md b/docs/data-sources/overlay.md
index 3a4778d..05a4ad2 100644
--- a/docs/data-sources/overlay.md
+++ b/docs/data-sources/overlay.md
@@ -630,6 +630,8 @@ Must enable helm support via [kustomize_options](#kustomize_options---optional)
 - `values_file` specify a file with helm values to use for templating. Not specifying this uses the in-chart values file, if one exists.
 - `values_inline` specify helm values inline from terraform, as a string
 - `values_merge` merge strategy if both `values_file` and `values_inline` are specified. Can be one of `override`, `replace`, `merge`. (default: `override`)
+- `api_versions` List of Kubernetes api versions used for Capabilities.APIVersions.
+- `kube_version` Allows specifying a custom kubernetes version to use when templating.
 
 #### Example
 
@@ -642,6 +644,8 @@ data "kustomization_overlay" "minecraft" {
     release_name = "moria"
     include_crds = false
     skip_tests = false
+		kube_version = "1.29.0"
+		api_versions = ["monitoring.coreos.com/v1"]
     values_inline = <<VALUES
       minecraftServer:
         eula: true
diff --git a/kustomize/data_source_kustomization_overlay.go b/kustomize/data_source_kustomization_overlay.go
index be6f085..c69804b 100644
--- a/kustomize/data_source_kustomization_overlay.go
+++ b/kustomize/data_source_kustomization_overlay.go
@@ -505,6 +505,15 @@ func dataSourceKustomizationOverlay() *schema.Resource {
 							Type:     schema.TypeString,
 							Optional: true,
 						},
+						"api_versions": {
+							Type:     schema.TypeList,
+							Optional: true,
+							Elem:     &schema.Schema{Type: schema.TypeString},
+						},
+						"kube_version": {
+							Type:     schema.TypeString,
+							Optional: true,
+						},
 					},
 				},
 			},
diff --git a/kustomize/data_source_kustomization_overlay_test.go b/kustomize/data_source_kustomization_overlay_test.go
index f0ef5eb..d3fa041 100644
--- a/kustomize/data_source_kustomization_overlay_test.go
+++ b/kustomize/data_source_kustomization_overlay_test.go
@@ -1185,6 +1185,9 @@ func TestDataSourceKustomizationOverlay_helm_charts_attr(t *testing.T) {
 					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "ids.#", "3"),
 					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "ids_prio.#", "3"),
 					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "manifests.%", "3"),
+					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "helm_charts.0.kube_version", "1.29.0"),
+					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "helm_charts.0.api_versions.#", "1"),
+					resource.TestCheckResourceAttr("data.kustomization_overlay.test", "helm_charts.0.api_versions.0", "monitoring.coreos.com/v1"),
 				),
 			},
 		},
@@ -1203,6 +1206,8 @@ data "kustomization_overlay" "test" {
 		version = "0.0.1"
 		namespace = "not-used"
 		skip_tests = true
+		kube_version = "1.29.0"
+		api_versions = ["monitoring.coreos.com/v1"]
 	}
 
 	namespace = "test-basic"