Add support for vm boot diagnostic profile

kubernetes/machine-class.yaml

@@ -54,6 +54,9 @@ providerSpec:
     machineSet: 
       id: <string>
       Kind: <string>
+    diagnosticsProfile:
+      enabled: false
+    # storageURI: <string>
   resourceGroup: <resource-group-name>
   subnetInfo:
     # vnetResourceGroup: <vnet-resource-group-name>

pkg/azure/api/providerspec.go

@@ -104,6 +104,9 @@ type AzureVirtualMachineProperties struct {
 	// 3. Only `Flexible` variant of VMSS is currently supported. It is strongly recommended that consumers turn-off any
 	// autoscaling capabilities as it interferes with the lifecycle management of MCM and auto-scaling capabilities offered by Cluster-Autoscaler.
 	VirtualMachineScaleSet *AzureSubResource `json:"virtualMachineScaleSet,omitempty"`
+	// DiagnosticsProfile specifies if boot metrics are enabled and where they are stored
+	// For additional information see: [https://learn.microsoft.com/en-us/azure/virtual-machines/boot-diagnostics]
+	DiagnosticsProfile *AzureDiagnosticsProfile `json:"diagnosticsProfile,omitempty"`
 	// Deprecated: Use either AvailabilitySet or VirtualMachineScaleSet instead
 	MachineSet *AzureMachineSetConfig `json:"machineSet,omitempty"`
 }
@@ -266,3 +269,12 @@ type AzureSubnetInfo struct {
 	// SubnetName is the name of the subnet which is unique within a resource group.
 	SubnetName string `json:"subnetName,omitempty"`
 }
+
+// AzureDiagnosticsProfile specifies boot diagnostic options
+type AzureDiagnosticsProfile struct {
+	// Enabled configures boot diagnostics to be stored or not
+	Enabled bool `json:"enabled,omitempty"`
+	// StorageURI is the URI of the storage account to use for storing console output and screenshot.
+	// If not specified azure managed storage will be used.
+	StorageURI *string `json:"storageURI,omitempty"`
+}

pkg/azure/provider/helpers/driver.go

@@ -620,6 +620,7 @@ func createVMCreationParams(providerSpec api.AzureProviderSpec, imageRef armcomp
 			},
 			AvailabilitySet:        getAvailabilitySet(providerSpec.Properties.AvailabilitySet),
 			VirtualMachineScaleSet: getVirtualMachineScaleSet(providerSpec.Properties.VirtualMachineScaleSet),
+			DiagnosticsProfile:     getDiagnosticsProfile(providerSpec.Properties.DiagnosticsProfile),
 		},
 		Tags:     vmTags,
 		Zones:    getZonesFromProviderSpec(providerSpec),
@@ -727,3 +728,15 @@ func getZonesFromProviderSpec(spec api.AzureProviderSpec) []*string {
 	}
 	return zones
 }
+
+func getDiagnosticsProfile(profile *api.AzureDiagnosticsProfile) *armcompute.DiagnosticsProfile {
+	if profile == nil {
+		return nil
+	}
+	return &armcompute.DiagnosticsProfile{
+		BootDiagnostics: &armcompute.BootDiagnostics{
+			Enabled:    to.Ptr(profile.Enabled),
+			StorageURI: profile.StorageURI,
+		},
+	}
+}