From 3c1707f7ebbd478bf83345b8d9c3670b1f9248b5 Mon Sep 17 00:00:00 2001
From: Adam Pasyniuk <adam@cast.ai>
Date: Fri, 15 Dec 2023 13:21:38 +0100
Subject: [PATCH] Impl flatternASKConfig for os_disk_type

---
 castai/resource_node_configuration.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/castai/resource_node_configuration.go b/castai/resource_node_configuration.go
index f66ad07d..e54dd30b 100644
--- a/castai/resource_node_configuration.go
+++ b/castai/resource_node_configuration.go
@@ -706,9 +706,29 @@ func flattenAKSConfig(config *sdk.NodeconfigV1AKSConfig) []map[string]interface{
 		m["max_pods_per_node"] = *config.MaxPodsPerNode
 	}
 
+	if v := config.MaxPodsPerNode; v != nil {
+		m["os_disk_type"] = fromAKSDiskType(config.OsDiskType)
+	}
+
 	return []map[string]interface{}{m}
 }
 
+func fromAKSDiskType(osDiskType *sdk.NodeconfigV1AKSConfigOsDiskType) string {
+	if osDiskType == nil {
+		return ""
+	}
+	switch *osDiskType {
+	case sdk.OSDISKTYPESTANDARD:
+		return "standard"
+	case sdk.OSDISKTYPESTANDARDSSD:
+		return "standard-ssd"
+	case sdk.OSDISKTYPEPREMIUMSSD:
+		return "premium-ssd"
+	default:
+		return ""
+	}
+}
+
 func toGKEConfig(obj map[string]interface{}) *sdk.NodeconfigV1GKEConfig {
 	if obj == nil {
 		return nil