KUBE-422: [AKS] adding support of AzureLinux

castai/resource_node_configuration.go

@@ -39,6 +39,7 @@ const (
 	FieldNodeConfigurationKOPS             = "kops"
 	FieldNodeConfigurationGKE              = "gke"
 	FieldNodeConfigurationEKSTargetGroup   = "target_group"
+	FieldNodeConfigurationAKSImageFamily   = "aks_image_family"
 	FieldNodeConfigurationEKSImageFamily   = "eks_image_family"
 )
 
@@ -47,6 +48,10 @@ const (
 	eksImageFamilyAL2023       = "al2023"
 	eksImageFamilyBottlerocket = "bottlerocket"
 )
+const (
+	aksImageFamilyUbuntu     = "ubuntu"
+	aksImageFamilyAzureLinux = "azure-linux"
+)
 
 func resourceNodeConfiguration() *schema.Resource {
 	return &schema.Resource{
@@ -296,6 +301,20 @@ func resourceNodeConfiguration() *schema.Resource {
 							Description:      "Type of managed os disk attached to the node. (See [disk types](https://learn.microsoft.com/en-us/azure/virtual-machines/disks-types)). One of: standard, standard-ssd, premium-ssd (ultra and premium-ssd-v2 are not supported for os disk)",
 							ValidateDiagFunc: validation.ToDiagFunc(validation.StringInSlice([]string{"standard", "standard-ssd", "premium-ssd"}, false)),
 						},
+						FieldNodeConfigurationAKSImageFamily: {
+							Type:     schema.TypeString,
+							Optional: true,
+							Description: fmt.Sprintf(
+								"Image OS Family to use when provisioning node in AKS. "+
+									"If both image and family are provided, the system will use provided image and provisioning logic for given family. "+
+									"If only image family is provided, the system will attempt to resolve the latest image from that family based on kubernetes version and node architecture. "+
+									"If image family is omitted, a default family (based on cloud provider) will be used. "+
+									"See Cast.ai documentation for details. Possible values: (%v)", strings.Join([]string{aksImageFamilyUbuntu, aksImageFamilyAzureLinux}, ",")),
+							ValidateDiagFunc: validation.ToDiagFunc(validation.StringInSlice([]string{aksImageFamilyUbuntu, aksImageFamilyAzureLinux}, true)),
+							DiffSuppressFunc: func(k, oldValue, newValue string, d *schema.ResourceData) bool {
+								return strings.EqualFold(oldValue, newValue)
+							},
+						},
 					},
 				},
 			},
@@ -851,6 +870,10 @@ func toAKSSConfig(obj map[string]interface{}) *sdk.NodeconfigV1AKSConfig {
 		out.OsDiskType = toAKSOSDiskType(v)
 	}
 
+	if v, ok := obj[FieldNodeConfigurationAKSImageFamily].(string); ok {
+		out.ImageFamily = toAKSImageFamily(v)
+	}
+
 	return out
 }
 
@@ -871,6 +894,21 @@ func toAKSOSDiskType(v string) *sdk.NodeconfigV1AKSConfigOsDiskType {
 	}
 }
 
+func toAKSImageFamily(v string) *sdk.NodeconfigV1AKSConfigImageFamily {
+	if v == "" {
+		return nil
+	}
+
+	switch strings.ToLower(v) {
+	case aksImageFamilyUbuntu:
+		return lo.ToPtr(sdk.NodeconfigV1AKSConfigImageFamilyFAMILYUBUNTU)
+	case aksImageFamilyAzureLinux:
+		return lo.ToPtr(sdk.NodeconfigV1AKSConfigImageFamilyFAMILYAZURELINUX)
+	default:
+		return nil
+	}
+}
+
 func flattenAKSConfig(config *sdk.NodeconfigV1AKSConfig) []map[string]interface{} {
 	if config == nil {
 		return nil
@@ -884,6 +922,10 @@ func flattenAKSConfig(config *sdk.NodeconfigV1AKSConfig) []map[string]interface{
 		m["os_disk_type"] = fromAKSDiskType(config.OsDiskType)
 	}
 
+	if v := config.ImageFamily; v != nil {
+		m[FieldNodeConfigurationAKSImageFamily] = fromAKSImageFamily(*v)
+	}
+
 	return []map[string]interface{}{m}
 }
 
@@ -903,6 +945,16 @@ func fromAKSDiskType(osDiskType *sdk.NodeconfigV1AKSConfigOsDiskType) string {
 	}
 }
 
+func fromAKSImageFamily(family sdk.NodeconfigV1AKSConfigImageFamily) string {
+	switch family {
+	case sdk.NodeconfigV1AKSConfigImageFamilyFAMILYAZURELINUX, sdk.NodeconfigV1AKSConfigImageFamilyFamilyAzureLinux:
+		return aksImageFamilyAzureLinux
+	case sdk.NodeconfigV1AKSConfigImageFamilyFAMILYUBUNTU, sdk.NodeconfigV1AKSConfigImageFamilyFamilyUbuntu:
+		return aksImageFamilyUbuntu
+	default:
+		return ""
+	}
+}
 func toGKEConfig(obj map[string]interface{}) *sdk.NodeconfigV1GKEConfig {
 	if obj == nil {
 		return nil

castai/resource_node_configuration_aks_test.go

@@ -29,6 +29,7 @@ func TestAccResourceNodeConfiguration_aks(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "disk_cpu_ratio", "35"),
 					resource.TestCheckResourceAttr(resourceName, "min_disk_size", "122"),
 					resource.TestCheckResourceAttr(resourceName, "aks.0.max_pods_per_node", "31"),
+					resource.TestCheckResourceAttr(resourceName, "aks.0.aks_image_family", "ubuntu"),
 					resource.TestCheckResourceAttr(resourceName, "eks.#", "0"),
 					resource.TestCheckResourceAttr(resourceName, "kops.#", "0"),
 					resource.TestCheckResourceAttr(resourceName, "gke.#", "0"),
@@ -41,6 +42,7 @@ func TestAccResourceNodeConfiguration_aks(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "disk_cpu_ratio", "0"),
 					resource.TestCheckResourceAttr(resourceName, "min_disk_size", "121"),
 					resource.TestCheckResourceAttr(resourceName, "aks.0.max_pods_per_node", "32"),
+					resource.TestCheckResourceAttr(resourceName, "aks.0.aks_image_family", "azure-linux"),
 					resource.TestCheckResourceAttr(resourceName, "eks.#", "0"),
 					resource.TestCheckResourceAttr(resourceName, "kops.#", "0"),
 					resource.TestCheckResourceAttr(resourceName, "gke.#", "0"),
@@ -73,6 +75,7 @@ resource "castai_node_configuration" "test" {
   }
   aks {
 	max_pods_per_node = 31
+    aks_image_family = "ubuntu"
   }
 }
 
@@ -96,6 +99,7 @@ resource "castai_node_configuration" "test" {
   }
   aks {
 	max_pods_per_node = 32
+    aks_image_family = "azure-linux"
   }
 }
 `, rgName, rName))