data source/service_name: Add data source for service names

Author: YakDriver

aws/data_source_aws_service_name.go

@@ -0,0 +1,76 @@
+package aws
+
+import (
+	"fmt"
+	"sort"
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws/endpoints"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func dataSourceAwsServiceName() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceAwsServiceNameRead,
+
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:     schema.TypeString,
+				Computed: true,
+				Optional: true,
+			},
+
+			"region": {
+				Type:     schema.TypeString,
+				Computed: true,
+				Optional: true,
+			},
+
+			"service": {
+				Type:     schema.TypeString,
+				Computed: true,
+				Optional: true,
+			},
+
+			"service_prefix": {
+				Type:     schema.TypeString,
+				Computed: true,
+				Optional: true,
+			},
+		},
+	}
+}
+
+func dataSourceAwsServiceNameRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*AWSClient)
+
+	if v, ok := d.GetOk("name"); ok {
+		serviceParts := strings.Split(v.(string), ".")
+		if len(serviceParts) < 4 {
+			return fmt.Errorf("service names must have at least 4 parts (%s has %d)", v.(string), len(serviceParts))
+		}
+
+		d.Set("service", serviceParts[len(serviceParts)-1])
+		d.Set("region", serviceParts[len(serviceParts)-2])
+		d.Set("service_prefix", strings.Join(serviceParts[0:len(serviceParts)-2], "."))
+	}
+
+	if _, ok := d.GetOk("region"); !ok {
+		d.Set("region", client.region)
+	}
+
+	if _, ok := d.GetOk("service"); !ok {
+		d.Set("service", endpoints.Ec2ServiceID)
+	}
+
+	if _, ok := d.GetOk("service_prefix"); !ok {
+		dnsParts := strings.Split(meta.(*AWSClient).dnsSuffix, ".")
+		sort.Sort(sort.Reverse(sort.StringSlice(dnsParts)))
+		d.Set("service_prefix", strings.Join(dnsParts, "."))
+	}
+
+	d.Set("name", fmt.Sprintf("%s.%s.%s", d.Get("service_prefix").(string), d.Get("region").(string), d.Get("service").(string)))
+	d.SetId(d.Get("name").(string))
+
+	return nil
+}

aws/data_source_aws_service_name_test.go

@@ -0,0 +1,96 @@
+package aws
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/aws/aws-sdk-go/aws/endpoints"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func TestAccAWSServiceName_basic(t *testing.T) {
+	dataSourceName := "data.aws_service_name.default"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCheckAwsServiceNameConfig_basic(),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(dataSourceName, "region", testAccGetRegion()),
+					resource.TestCheckResourceAttr(dataSourceName, "service", endpoints.Ec2ServiceID),
+					resource.TestCheckResourceAttr(dataSourceName, "name", fmt.Sprintf("%s.%s.%s", "com.amazonaws", testAccGetRegion(), endpoints.Ec2ServiceID)),
+					resource.TestCheckResourceAttr(dataSourceName, "service_prefix", "com.amazonaws"),
+				),
+			},
+		},
+	})
+}
+
+func TestAccAWSServiceName_byServiceName(t *testing.T) {
+	dataSourceName := "data.aws_service_name.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCheckAwsServiceNameConfig_byServiceName(),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(dataSourceName, "region", endpoints.ApNortheast1RegionID),
+					resource.TestCheckResourceAttr(dataSourceName, "service", endpoints.S3ServiceID),
+					resource.TestCheckResourceAttr(dataSourceName, "name", fmt.Sprintf("%s.%s.%s", "cn.com.amazonaws", endpoints.ApNortheast1RegionID, endpoints.S3ServiceID)),
+					resource.TestCheckResourceAttr(dataSourceName, "service_prefix", "cn.com.amazonaws"),
+				),
+			},
+		},
+	})
+}
+
+func TestAccAWSServiceName_byPart(t *testing.T) {
+	dataSourceName := "data.aws_service_name.test"
+
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCheckAwsServiceNameConfig_byPart(),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr(dataSourceName, "region", testAccGetRegion()),
+					resource.TestCheckResourceAttr(dataSourceName, "service", endpoints.Ec2ServiceID),
+					resource.TestCheckResourceAttr(dataSourceName, "name", fmt.Sprintf("%s.%s.%s", "com.amazonaws", testAccGetRegion(), endpoints.Ec2ServiceID)),
+					resource.TestCheckResourceAttr(dataSourceName, "service_prefix", "com.amazonaws"),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckAwsServiceNameConfig_basic() string {
+	return fmt.Sprintf(`
+data "aws_service_name" "default" {}
+`)
+}
+
+func testAccCheckAwsServiceNameConfig_byServiceName() string {
+	// lintignore:AWSAT003
+	return fmt.Sprintf(`
+data "aws_service_name" "test" {
+  name = "cn.com.amazonaws.ap-northeast-1.s3"
+}
+`)
+}
+
+func testAccCheckAwsServiceNameConfig_byPart() string {
+	return fmt.Sprintf(`
+data "aws_region" "current" {}
+
+data "aws_service_name" "test" {
+  service        = "ec2"
+  region         = data.aws_region.current.name
+  service_prefix = "com.amazonaws"
+}
+`)
+}

aws/provider.go

@@ -340,6 +340,7 @@ func Provider() *schema.Provider {
 			"aws_secretsmanager_secret":                      dataSourceAwsSecretsManagerSecret(),
 			"aws_secretsmanager_secret_rotation":             dataSourceAwsSecretsManagerSecretRotation(),
 			"aws_secretsmanager_secret_version":              dataSourceAwsSecretsManagerSecretVersion(),
+			"aws_service_name":                               dataSourceAwsServiceName(),
 			"aws_servicequotas_service":                      dataSourceAwsServiceQuotasService(),
 			"aws_servicequotas_service_quota":                dataSourceAwsServiceQuotasServiceQuota(),
 			"aws_sfn_activity":                               dataSourceAwsSfnActivity(),

website/docs/d/service_name.html.markdown

@@ -0,0 +1,41 @@
+---
+subcategory: ""
+layout: "aws"
+page_title: "AWS: aws_service_name"
+description: |-
+  Construct AWS service names
+---
+
+# Data Source: aws_service_name
+
+Use this data source to compose AWS service names.
+
+## Example Usage
+
+```hcl
+resource "aws_vpc" "test" {
+  cidr_block = "10.0.0.0/16"
+}
+
+data "aws_service_name" "s3" {
+  service = "s3"
+}
+
+resource "aws_vpc_endpoint" "test" {
+  vpc_id       = aws_vpc.test.id
+  service_name = data.aws_service_name.s3.id
+}
+```
+
+## Argument Reference
+
+~> **Note:** If the `name` argument is provided, `region`, `service`, and `service_prefix` values are ignored.
+
+* `name` - Name of the service (e.g. `com.amazonaws.us-west-2.s3`)
+* `region` - Region of the service (e.g. `us-west-2`, `ap-northeast-1`).
+* `service` - Service (e.g. `s3`, `ec2`). Defaults to `ec2`.
+* `service_prefix` - Prefix of the service (e.g. `com.amazonaws` in AWS Commercial, `cn.com.amazonaws` in AWS China).
+
+## Attributes Reference
+
+Besides the arguments above, no attributes are exported.