add partition generator, and run gofmt (#2076)

aws · isaiahvita · Jul 31, 2023 · Mar 20, 2023 · Mar 29, 2023 · Apr 5, 2023
commit f4381d479f8fca554edbee20479aef5f37d4063f
@@ -9,8 +9,8 @@ type ARN struct {
 	Partition  string
 	Service    string
 	Region     string
-	AccountId  string
-	ResourceId OptionalStringSlice
+	AccountID  string
+	ResourceID OptionalStringSlice
 }
 
 const (
@@ -54,8 +54,8 @@ func ParseARN(input string) *ARN {
 		Partition:  sections[sectionPartition],
 		Service:    sections[sectionService],
 		Region:     sections[sectionRegion],
-		AccountId:  sections[sectionAccountID],
-		ResourceId: splitResource(sections[sectionResource]),
+		AccountID:  sections[sectionAccountID],
+		ResourceID: splitResource(sections[sectionResource]),
 	}
 }
 

@@ -1,21 +1,21 @@
 package awsrulesfn
 
 import (
-	"testing"
 	"github.com/google/go-cmp/cmp"
+	"testing"
 )
 
 func TestParseARN(t *testing.T) {
 	cases := []struct {
-		input     string
-		expect    *ARN
+		input  string
+		expect *ARN
 	}{
 		{
-			input:     "invalid",
+			input:  "invalid",
 			expect: nil,
 		},
 		{
-			input:     "arn:nope",
+			input:  "arn:nope",
 			expect: nil,
 		},
 		{
@@ -24,8 +24,8 @@ func TestParseARN(t *testing.T) {
 				Partition:  "aws",
 				Service:    "ecr",
 				Region:     "us-west-2",
-				AccountId:  "123456789012",
-				ResourceId: []string{"repository", "foo", "bar"},
+				AccountID:  "123456789012",
+				ResourceID: []string{"repository", "foo", "bar"},
 			},
 		},
 		{
@@ -34,8 +34,8 @@ func TestParseARN(t *testing.T) {
 				Partition:  "aws",
 				Service:    "elasticbeanstalk",
 				Region:     "us-east-1",
-				AccountId:  "123456789012",
-				ResourceId: []string{"environment", "My App", "MyEnvironment"},
+				AccountID:  "123456789012",
+				ResourceID: []string{"environment", "My App", "MyEnvironment"},
 			},
 		},
 		{
@@ -44,8 +44,8 @@ func TestParseARN(t *testing.T) {
 				Partition:  "aws",
 				Service:    "iam",
 				Region:     "",
-				AccountId:  "123456789012",
-				ResourceId: []string{"user", "David"},
+				AccountID:  "123456789012",
+				ResourceID: []string{"user", "David"},
 			},
 		},
 		{
@@ -54,8 +54,8 @@ func TestParseARN(t *testing.T) {
 				Partition:  "aws",
 				Service:    "rds",
 				Region:     "eu-west-1",
-				AccountId:  "123456789012",
-				ResourceId: []string{"db", "mysql-db"},
+				AccountID:  "123456789012",
+				ResourceID: []string{"db", "mysql-db"},
 			},
 		},
 		{
@@ -64,8 +64,8 @@ func TestParseARN(t *testing.T) {
 				Partition:  "aws",
 				Service:    "s3",
 				Region:     "",
-				AccountId:  "",
-				ResourceId: []string{"my_corporate_bucket", "exampleobject.png"},
+				AccountID:  "",
+				ResourceID: []string{"my_corporate_bucket", "exampleobject.png"},
 			},
 		},
 	}

@@ -1,4 +1,3 @@
 // Package awsrulesfn provides AWS focused endpoint rule functions for
 // evaluating endpoint resolution rules.
-
 package awsrulesfn
@@ -0,0 +1,7 @@
+//go:build codegen
+// +build codegen
+
+package awsrulesfn
+
+//go:generate go run -tags codegen ./internal/partition/codegen.go -model partitions.json -output partitions.go
+//go:generate gofmt -w -s .
@@ -1,10 +1,9 @@
 package awsrulesfn
 
 import (
+	smithyhttp "github.com/aws/smithy-go/transport/http"
 	"net/netip"
 	"strings"
-	smithyhttp "github.com/aws/smithy-go/transport/http"
-
 )
 
 // IsVirtualHostableS3Bucket returns if the input is a DNS compatible bucket

@@ -15,7 +15,7 @@ func TestIsVirtualHostableS3Bucket(t *testing.T) {
 			expect: true,
 		},
 		"single label no split too short": {
-			input:     "a",
+			input:  "a",
 			expect: false,
 		},
 		"single label with split": {
@@ -24,7 +24,7 @@ func TestIsVirtualHostableS3Bucket(t *testing.T) {
 			expect:          true,
 		},
 		"multiple labels no split": {
-			input:     "abc.123-",
+			input:  "abc.123-",
 			expect: false,
 		},
 		"multiple labels with split": {
@@ -35,32 +35,32 @@ func TestIsVirtualHostableS3Bucket(t *testing.T) {
 		"multiple labels with split invalid label": {
 			input:           "abc.123-...",
 			allowSubDomains: true,
-			expect: false,
+			expect:          false,
 		},
 		"max length host label": {
 			input:  "012345678901234567890123456789012345678901234567890123456789123",
 			expect: true,
 		},
 		"too large host label": {
-			input:     "0123456789012345678901234567890123456789012345678901234567891234",
+			input:  "0123456789012345678901234567890123456789012345678901234567891234",
 			expect: false,
 		},
 		"too small host label": {
-			input:     "",
+			input:  "",
 			expect: false,
 		},
 		"lower case only": {
-			input:     "AbC",
+			input:  "AbC",
 			expect: false,
 		},
 		"like IP address": {
-			input:     "127.111.222.123",
+			input:  "127.111.222.123",
 			expect: false,
 		},
 		"multiple labels like IP address": {
 			input:           "127.111.222.123",
 			allowSubDomains: true,
-			expect: false,
+			expect:          false,
 		},
 	}
 

@@ -0,0 +1,73 @@
+package awsrulesfn
+
+import "regexp"
+
+// Partition provides the metadata describing an AWS partition.
+type Partition struct {
+	ID            string                     `json:"id"`
+	Regions       map[string]PartitionConfig `json:"regions"`
+	RegionRegex   string                     `json:"regionRegex"`
+	DefaultConfig PartitionConfig            `json:"outputs"`
+}
+
+// PartitionConfig provides the endpoint metadata for an AWS region or partition.
+type PartitionConfig struct {
+	Name               *string `json:"name"`
+	DNSSuffix          *string `json:"dnsSuffix"`
+	DualStackDNSSuffix *string `json:"dualStackDnsSuffix"`
+	SupportsFIPS       *bool   `json:"supportsFIPS"`
+	SupportsDualStack  *bool   `json:"supportsDualStack"`
+}
+
+const defaultPartition = "aws"
+
+// GetPartition returns an AWS [Partition] for the region provided. If the
+// partition cannot be determined nil will be returned.
+func GetPartition(region string) *PartitionConfig {
+	return getPartition(partitions, region)
+}
+
+func getPartition(partitions []Partition, region string) *PartitionConfig {
+	for _, partition := range partitions {
+		if v, ok := partition.Regions[region]; ok {
+			v = mergePartition(v, partition.DefaultConfig)
+			return &v
+		}
+	}
+
+	for _, partition := range partitions {
+		regionRegex := regexp.MustCompile(partition.RegionRegex)
+		if regionRegex.MatchString(region) {
+			v := partition.DefaultConfig
+			return &v
+		}
+	}
+
+	for _, partition := range partitions {
+		if partition.ID == defaultPartition {
+			v := partition.DefaultConfig
+			return &v
+		}
+	}
+
+	return nil
+}
+
+func mergePartition(into PartitionConfig, from PartitionConfig) PartitionConfig {
+	if into.Name == nil {
+		into.Name = from.Name
+	}
+	if into.DNSSuffix == nil {
+		into.DNSSuffix = from.DNSSuffix
+	}
+	if into.DualStackDNSSuffix == nil {
+		into.DualStackDNSSuffix = from.DualStackDNSSuffix
+	}
+	if into.SupportsFIPS == nil {
+		into.SupportsFIPS = from.SupportsFIPS
+	}
+	if into.SupportsDualStack == nil {
+		into.SupportsDualStack = from.SupportsDualStack
+	}
+	return into
+}