azure: add --azure-hyper-v-generation flag

Add a new flag to allow specifying the Hyper-V generation (V1 or V2)
when creating Azure images. This enables support for both Gen1 and Gen2
image creation.

Author: marmijo

mantle/cmd/kola/kola.go

@@ -308,6 +308,7 @@ func writeProps() error {
 		Location         string `json:"location"`
 		Size             string `json:"size"`
 		AvailabilityZone string `json:"availability_zone"`
+		HyperVGeneration string `json:"hyper_v_generation"`
 	}
 	type DO struct {
 		Region string `json:"region"`
@@ -364,6 +365,7 @@ func writeProps() error {
 			Location:         kola.AzureOptions.Location,
 			Size:             kola.AzureOptions.Size,
 			AvailabilityZone: kola.AzureOptions.AvailabilityZone,
+			HyperVGeneration: kola.AzureOptions.HyperVGeneration,
 		},
 		DO: DO{
 			Region: kola.DOOptions.Region,

mantle/cmd/kola/options.go

@@ -101,6 +101,7 @@ func init() {
 	sv(&kola.AzureOptions.Location, "azure-location", "westus", "Azure location (default \"westus\"")
 	sv(&kola.AzureOptions.Size, "azure-size", "Standard_D2_v2", "Azure machine size (default \"Standard_D2_v2\")")
 	sv(&kola.AzureOptions.AvailabilityZone, "azure-availability-zone", "1", "Azure Availability Zone (default \"1\")")
+	sv(&kola.AzureOptions.HyperVGeneration, "azure-hyper-v-generation", "V1", "Azure Hyper-V Generation (default \"V1\")")
 
 	// do-specific options
 	sv(&kola.DOOptions.ConfigPath, "do-config-file", "", "DigitalOcean config file (default \"~/"+auth.DOConfigPath+"\")")

mantle/cmd/ore/azure/azure.go

@@ -34,6 +34,7 @@ var (
 
 	azureCredentials string
 	azureLocation    string
+	azureHyperVGen   string
 
 	api *azure.API
 )
@@ -44,6 +45,7 @@ func init() {
 	sv := Azure.PersistentFlags().StringVar
 	sv(&azureCredentials, "azure-credentials", "", "Azure credentials file location (default \"~/"+auth.AzureCredentialsPath+"\")")
 	sv(&azureLocation, "azure-location", "westus", "Azure location (default \"westus\")")
+	sv(&azureHyperVGen, "azure-hyper-v-generation", "V1", "Azure Hypervisor Generation")
 }
 
 func preauth(cmd *cobra.Command, args []string) error {
@@ -52,6 +54,7 @@ func preauth(cmd *cobra.Command, args []string) error {
 	a, err := azure.New(&azure.Options{
 		AzureCredentials: azureCredentials,
 		Location:         azureLocation,
+		HyperVGeneration: azureHyperVGen,
 	})
 	if err != nil {
 		plog.Fatalf("Failed to create Azure API: %v", err)

mantle/platform/api/azure/image.go

@@ -28,7 +28,7 @@ func (a *API) CreateImage(name, resourceGroup, blobURI string) (armcompute.Image
 		Name:     &name,
 		Location: &a.opts.Location,
 		Properties: &armcompute.ImageProperties{
-			HyperVGeneration: to.Ptr(armcompute.HyperVGenerationTypesV1),
+			HyperVGeneration: to.Ptr(armcompute.HyperVGenerationTypes(a.opts.HyperVGeneration)),
 			StorageProfile: &armcompute.ImageStorageProfile{
 				OSDisk: &armcompute.ImageOSDisk{
 					OSType:  to.Ptr(armcompute.OperatingSystemTypesLinux),

mantle/platform/api/azure/options.go

@@ -33,6 +33,7 @@ type Options struct {
 	Size             string
 	Location         string
 	AvailabilityZone string
+	HyperVGeneration string
 
 	SubscriptionName string
 	SubscriptionID   string