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/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