azure windows sku filter g2 sku

Author: lilyLuLiu

pkg/provider/azure/action/windows/windows.go

@@ -31,6 +31,11 @@ import (
 	"github.com/redhat-developer/mapt/pkg/util/logging"
 	resourcesUtil "github.com/redhat-developer/mapt/pkg/util/resources"
 	"golang.org/x/exp/slices"
+
+	armcompute "github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/compute/armcompute/v6"
+	"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
+	"context"
+	"os"
 )
 
 //go:embed rhqp-ci-setup.ps1
@@ -149,6 +154,10 @@ func (r *WindowsRequest) deployer(ctx *pulumi.Context) error {
 		AdminPasswd:     adminPasswd,
 		SpotPrice:       spotPrice,
 	}
+	err = SkuG2Support(&vmr, *location)
+	if err != nil {
+		return err
+	}
 	vm, err := vmr.Create(ctx)
 	if err != nil {
 		return err
@@ -371,3 +380,58 @@ func (r *WindowsRequest) profilesAsParams() string {
 		})
 	return strings.Join(pp, " ")
 }
+
+func SkuG2Support(r *virtualmachine.VirtualMachineRequest, location string) error {
+	cred, err := azidentity.NewDefaultAzureCredential(nil)
+	if err != nil {
+		return err
+	}
+	subscriptionId := os.Getenv("AZURE_SUBSCRIPTION_ID")
+
+	clientFactory, err := armcompute.NewClientFactory(subscriptionId, cred, nil)
+	if err != nil {
+		return err
+	}
+	imagesClient := clientFactory.NewVirtualMachineImagesClient()
+	if ! verify_g2(imagesClient, location, r.Publisher, r.Offer, r.Sku) {
+		finalSKU, err := get_g2_sku(imagesClient, location, r.Publisher, r.Offer, r.Sku)
+		if err == nil && finalSKU != "" {
+			if verify_g2(imagesClient, location, r.Publisher, r.Offer, finalSKU) {
+				r.Sku = finalSKU
+				fmt.Printf("%s is g1, Using SKU %s\n", r.Sku, finalSKU)
+				return nil
+			}
+		}
+	} else {
+		return nil
+	} 
+	return fmt.Errorf("the SKU %s is not support for G2", r.Sku)
+}
+
+func verify_g2(imagesClient *armcompute.VirtualMachineImagesClient, location string, publisher string, offer string, sku string) bool {
+	// List available image versions
+	resp, err := imagesClient.List(context.Background(),location, publisher, offer, sku, nil)
+	if err != nil {
+		return false
+	} 
+
+	image := resp.VirtualMachineImageResourceArray[0]
+	version := *image.Name
+	resps, _ := imagesClient.Get(context.Background(),location, publisher, offer, sku, version, nil)
+	info := resps.VirtualMachineImage
+	generation := *info.Properties.HyperVGeneration
+	return generation == "V2"
+}
+
+func get_g2_sku(imagesClient *armcompute.VirtualMachineImagesClient, location string, publisher string, offer string, originSKU string) (string, error) {
+	resp, err := imagesClient.ListSKUs(context.Background(),location, publisher, offer, nil)
+	if err != nil {
+		return "", err
+	}
+	for _, sku := range resp.VirtualMachineImageResourceArray {
+		if strings.HasPrefix(*sku.Name, originSKU+"-") {
+			return *sku.Name, nil
+		}
+	}
+	return "", nil
+}