Correcting AMI ID being sent from the UI

Signed-off-by: Sudarshan <asudarshan@vmware.com>
vmware-tanzu · Sep 2, 2021 · c5748ce · c5748ce
1 parent c26663a
commit c5748ce
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/pkg/v1/tkg/tkgconfigproviders/aws.go b/pkg/v1/tkg/tkgconfigproviders/aws.go
@@ -121,8 +121,17 @@ func (c *client) NewAWSConfig(params *models.AWSRegionalClusterParams, encodedCr
 
 	amiID := ""
 
-	if val, ok := bomConfiguration.AMI[params.AwsAccountParams.Region]; ok {
-		amiID = val[0].ID
+	if amis, ok := bomConfiguration.AMI[params.AwsAccountParams.Region]; ok {
+		if params.Os != nil && params.Os.OsInfo != nil {
+			for _, ami := range amis {
+				if ami.OSInfo.Name == params.Os.OsInfo.Name {
+					amiID = ami.ID
+					break
+				}
+			}
+		} else {
+			amiID = amis[0].ID
+		}
 	}
 
 	if amiID == "" {