From 8bdc58a2c486fdd4c1e57d587c5e7c77e0d64dd1 Mon Sep 17 00:00:00 2001
From: Asaf Alima <asaf.alima@gmail.com>
Date: Wed, 6 Dec 2023 22:39:51 +0200
Subject: [PATCH] feat(cluster): add use_instance_metadata_hostname flag to
 cloud_provider block

---
 docs/resources/cluster.md                              |  1 +
 rancher2/schema_cluster_rke_config_cloud_provider.go   |  5 +++++
 .../structure_cluster_rke_config_cloud_provider.go     |  8 ++++++++
 ...ructure_cluster_rke_config_cloud_provider_z_test.go | 10 ++++++----
 4 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/docs/resources/cluster.md b/docs/resources/cluster.md
index 0a61c6af2..034e3e16e 100644
--- a/docs/resources/cluster.md
+++ b/docs/resources/cluster.md
@@ -745,6 +745,7 @@ The following attributes are exported:
 * `name` - (Optional) RKE Cloud Provider name (string)
 * `openstack_cloud_provider` - (Optional/Computed) RKE Openstack Cloud Provider config for Cloud Provider [rke-openstack-cloud-provider](https://rancher.com/docs/rke/latest/en/config-options/cloud-providers/openstack/) (list maxitems:1)
 * `vsphere_cloud_provider` - (Optional/Computed) RKE Vsphere Cloud Provider config for Cloud Provider [rke-vsphere-cloud-provider](https://rancher.com/docs/rke/latest/en/config-options/cloud-providers/vsphere/) Extra argument `name` is required on `virtual_center` configuration. (list maxitems:1)
+* `use_instance_metadata_hostname` - (Optional/Computed) (bool)
 
 ##### `aws_cloud_provider`
 
diff --git a/rancher2/schema_cluster_rke_config_cloud_provider.go b/rancher2/schema_cluster_rke_config_cloud_provider.go
index d2c257497..71f360367 100644
--- a/rancher2/schema_cluster_rke_config_cloud_provider.go
+++ b/rancher2/schema_cluster_rke_config_cloud_provider.go
@@ -49,6 +49,11 @@ func clusterRKEConfigCloudProviderFields() map[string]*schema.Schema {
 				Schema: clusterRKEConfigCloudProviderVsphereFields(),
 			},
 		},
+		"use_instance_metadata_hostname": {
+			Type:     schema.TypeBool,
+			Optional: true,
+			Computed: true,
+		},
 	}
 	return s
 }
diff --git a/rancher2/structure_cluster_rke_config_cloud_provider.go b/rancher2/structure_cluster_rke_config_cloud_provider.go
index 123e341f9..6b3d4d097 100644
--- a/rancher2/structure_cluster_rke_config_cloud_provider.go
+++ b/rancher2/structure_cluster_rke_config_cloud_provider.go
@@ -70,6 +70,10 @@ func flattenClusterRKEConfigCloudProvider(in *managementClient.CloudProvider, p
 		obj["vsphere_cloud_provider"] = vsphereProvider
 	}
 
+	if in.UseInstanceMetadataHostname != nil {
+		obj["use_instance_metadata_hostname"] = *in.UseInstanceMetadataHostname
+	}
+
 	return []interface{}{obj}, nil
 }
 
@@ -122,5 +126,9 @@ func expandClusterRKEConfigCloudProvider(p []interface{}) (*managementClient.Clo
 		obj.VsphereCloudProvider = vsphereProvider
 	}
 
+	if v, ok := in["use_instance_metadata_hostname"].(bool); ok {
+		obj.UseInstanceMetadataHostname = &v
+	}
+
 	return obj, nil
 }
diff --git a/rancher2/structure_cluster_rke_config_cloud_provider_z_test.go b/rancher2/structure_cluster_rke_config_cloud_provider_z_test.go
index f8a90274f..aeba2a2cc 100644
--- a/rancher2/structure_cluster_rke_config_cloud_provider_z_test.go
+++ b/rancher2/structure_cluster_rke_config_cloud_provider_z_test.go
@@ -50,13 +50,15 @@ func init() {
 		},
 	}
 	testClusterRKEConfigCloudProviderConf = &managementClient.CloudProvider{
-		CustomCloudProvider: "XXXXXXXXXXXX",
-		Name:                "test",
+		CustomCloudProvider:         "XXXXXXXXXXXX",
+		Name:                        "test",
+		UseInstanceMetadataHostname: newTrue(),
 	}
 	testClusterRKEConfigCloudProviderInterface = []interface{}{
 		map[string]interface{}{
-			"custom_cloud_provider": "XXXXXXXXXXXX",
-			"name":                  "test",
+			"custom_cloud_provider":          "XXXXXXXXXXXX",
+			"name":                           "test",
+			"use_instance_metadata_hostname": true,
 		},
 	}
 }