Search for security group identifiers instead of names if they start with sg- (#364)

* Accept security group identifiers

* Restyled by gofmt

* HasPrefix

Co-authored-by: Restyled.io <commits@restyled.io>
Co-authored-by: Daniel Barnes <dabarnes2b@gmail.com>

Author: 3 people

iterative/aws/provider.go

@@ -178,16 +178,24 @@ func ResourceMachineCreate(ctx context.Context, d *schema.ResourceData, m interf
 		}
 	}
 
+	sgFilters := []types.Filter{}
+	if strings.HasPrefix(securityGroup, "sg-") {
+		sgFilters = append(sgFilters, types.Filter{
+			Name:   aws.String("group-id"),
+			Values: []string{securityGroup},
+		})
+	} else {
+		sgFilters = append(sgFilters, types.Filter{
+			Name: aws.String("group-name"),
+			Values: []string{
+				securityGroup,
+				strings.Title(securityGroup),
+				strings.ToUpper(securityGroup)},
+		})
+	}
+
 	sgDesc, err := svc.DescribeSecurityGroups(ctx, &ec2.DescribeSecurityGroupsInput{
-		Filters: []types.Filter{
-			{
-				Name: aws.String("group-name"),
-				Values: []string{
-					securityGroup,
-					strings.Title(securityGroup),
-					strings.ToUpper(securityGroup)},
-			},
-		},
+		Filters: sgFilters,
 	})
 	if err != nil {
 		return decodeAWSError(region, err)