Merge pull request #45360 from hashicorp/f-s3-meridian

S3 Tables replication

Author: ewbankkit

.changelog/45360.txt

@@ -0,0 +1,7 @@
+```release-note:new-resource
+aws_s3tables_table_bucket_replication
+```
+
+```release-note:new-resource
+aws_s3tables_table_replication
+```

internal/service/s3tables/exports_test.go

@@ -4,17 +4,21 @@
 package s3tables
 
 var (
-	ResourceNamespace         = newNamespaceResource
-	ResourceTable             = newTableResource
-	ResourceTableBucket       = newTableBucketResource
-	ResourceTableBucketPolicy = newTableBucketPolicyResource
-	ResourceTablePolicy       = newTablePolicyResource
+	ResourceNamespace              = newNamespaceResource
+	ResourceTable                  = newTableResource
+	ResourceTableBucket            = newTableBucketResource
+	ResourceTableBucketPolicy      = newTableBucketPolicyResource
+	ResourceTableBucketReplication = newTableBucketReplicationResource
+	ResourceTablePolicy            = newTablePolicyResource
+	ResourceTableReplication       = newTableReplicationResource
 
-	FindNamespaceByTwoPartKey     = findNamespaceByTwoPartKey
-	FindTableByThreePartKey       = findTableByThreePartKey
-	FindTableBucketByARN          = findTableBucketByARN
-	FindTableBucketPolicyByARN    = findTableBucketPolicyByARN
-	FindTablePolicyByThreePartKey = findTablePolicyByThreePartKey
+	FindNamespaceByTwoPartKey       = findNamespaceByTwoPartKey
+	FindTableByThreePartKey         = findTableByThreePartKey
+	FindTableBucketByARN            = findTableBucketByARN
+	FindTableBucketPolicyByARN      = findTableBucketPolicyByARN
+	FindTableBucketReplicationByARN = findTableBucketReplicationByARN
+	FindTablePolicyByThreePartKey   = findTablePolicyByThreePartKey
+	FindTableReplicationByARN       = findTableReplicationByARN
 
 	TableIDFromTableARN = tableIDFromTableARN
 )

internal/service/s3tables/service_package_gen.go

@@ -0,0 +1,274 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package s3tables
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/s3tables"
+	awstypes "github.com/aws/aws-sdk-go-v2/service/s3tables/types"
+	"github.com/hashicorp/terraform-plugin-framework-validators/listvalidator"
+	"github.com/hashicorp/terraform-plugin-framework-validators/setvalidator"
+	"github.com/hashicorp/terraform-plugin-framework/resource"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier"
+	"github.com/hashicorp/terraform-plugin-framework/schema/validator"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/retry"
+	"github.com/hashicorp/terraform-provider-aws/internal/errs"
+	"github.com/hashicorp/terraform-provider-aws/internal/errs/fwdiag"
+	"github.com/hashicorp/terraform-provider-aws/internal/framework"
+	fwflex "github.com/hashicorp/terraform-provider-aws/internal/framework/flex"
+	fwtypes "github.com/hashicorp/terraform-provider-aws/internal/framework/types"
+	"github.com/hashicorp/terraform-provider-aws/internal/tfresource"
+	"github.com/hashicorp/terraform-provider-aws/names"
+)
+
+// @FrameworkResource("aws_s3tables_table_bucket_replication", name="Table Bucket Replication")
+// @ArnIdentity("table_bucket_arn")
+// @Testing(existsType="github.com/aws/aws-sdk-go-v2/service/s3tables;s3tables.GetTableBucketReplicationOutput")
+// @Testing(preCheck="testAccPreCheck")
+// @Testing(hasNoPreExistingResource=true)
+// @Testing(importIgnore="version_token")
+// @Testing(plannableImportAction="NoOp")
+func newTableBucketReplicationResource(_ context.Context) (resource.ResourceWithConfigure, error) {
+	return &tableBucketReplicationResource{}, nil
+}
+
+type tableBucketReplicationResource struct {
+	framework.ResourceWithModel[tableBucketReplicationResourceModel]
+	framework.WithImportByIdentity
+}
+
+func (r *tableBucketReplicationResource) Schema(ctx context.Context, request resource.SchemaRequest, response *resource.SchemaResponse) {
+	response.Schema = schema.Schema{
+		Attributes: map[string]schema.Attribute{
+			names.AttrRole: schema.StringAttribute{
+				CustomType: fwtypes.ARNType,
+				Required:   true,
+			},
+			"table_bucket_arn": schema.StringAttribute{
+				CustomType: fwtypes.ARNType,
+				Required:   true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
+			"version_token": schema.StringAttribute{
+				Computed: true,
+			},
+		},
+		Blocks: map[string]schema.Block{
+			names.AttrRule: schema.ListNestedBlock{
+				CustomType: fwtypes.NewListNestedObjectTypeOf[bucketReplicationRuleModel](ctx),
+				Validators: []validator.List{
+					listvalidator.SizeAtMost(1),
+				},
+				NestedObject: schema.NestedBlockObject{
+					Blocks: map[string]schema.Block{
+						names.AttrDestination: schema.SetNestedBlock{
+							CustomType: fwtypes.NewSetNestedObjectTypeOf[replicationDestinationModel](ctx),
+							Validators: []validator.Set{
+								setvalidator.SizeAtLeast(1),
+								setvalidator.SizeAtMost(5),
+								setvalidator.IsRequired(),
+							},
+							NestedObject: schema.NestedBlockObject{
+								Attributes: map[string]schema.Attribute{
+									"destination_table_bucket_arn": schema.StringAttribute{
+										CustomType: fwtypes.ARNType,
+										Required:   true,
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func (r *tableBucketReplicationResource) Create(ctx context.Context, request resource.CreateRequest, response *resource.CreateResponse) {
+	var data tableBucketReplicationResourceModel
+	response.Diagnostics.Append(request.Plan.Get(ctx, &data)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	conn := r.Meta().S3TablesClient(ctx)
+
+	var configuration awstypes.TableBucketReplicationConfiguration
+	response.Diagnostics.Append(fwflex.Expand(ctx, data, &configuration)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	tableBucketARN := fwflex.StringValueFromFramework(ctx, data.TableBucketARN)
+	input := s3tables.PutTableBucketReplicationInput{
+		Configuration:  &configuration,
+		TableBucketARN: aws.String(tableBucketARN),
+	}
+
+	output, err := conn.PutTableBucketReplication(ctx, &input)
+
+	if err != nil {
+		response.Diagnostics.AddError(fmt.Sprintf("creating S3 Tables Table Bucket Replication (%s)", tableBucketARN), err.Error())
+
+		return
+	}
+
+	// Set values for unknowns.
+	data.VersionToken = fwflex.StringToFramework(ctx, output.VersionToken)
+
+	response.Diagnostics.Append(response.State.Set(ctx, data)...)
+}
+
+func (r *tableBucketReplicationResource) Read(ctx context.Context, request resource.ReadRequest, response *resource.ReadResponse) {
+	var data tableBucketReplicationResourceModel
+	response.Diagnostics.Append(request.State.Get(ctx, &data)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	conn := r.Meta().S3TablesClient(ctx)
+
+	tableBucketARN := fwflex.StringValueFromFramework(ctx, data.TableBucketARN)
+	output, err := findTableBucketReplicationByARN(ctx, conn, tableBucketARN)
+
+	if tfresource.NotFound(err) {
+		response.Diagnostics.Append(fwdiag.NewResourceNotFoundWarningDiagnostic(err))
+		response.State.RemoveResource(ctx)
+
+		return
+	}
+
+	if err != nil {
+		response.Diagnostics.AddError(fmt.Sprintf("reading S3 Tables Table Bucket Replication (%s)", tableBucketARN), err.Error())
+
+		return
+	}
+
+	// Set attributes for import.
+	response.Diagnostics.Append(fwflex.Flatten(ctx, output.Configuration, &data)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	response.Diagnostics.Append(response.State.Set(ctx, &data)...)
+}
+
+func (r *tableBucketReplicationResource) Update(ctx context.Context, request resource.UpdateRequest, response *resource.UpdateResponse) {
+	var old, new tableBucketReplicationResourceModel
+	response.Diagnostics.Append(request.Plan.Get(ctx, &new)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+	response.Diagnostics.Append(request.State.Get(ctx, &old)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	conn := r.Meta().S3TablesClient(ctx)
+
+	var configuration awstypes.TableBucketReplicationConfiguration
+	response.Diagnostics.Append(fwflex.Expand(ctx, new, &configuration)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	tableBucketARN := fwflex.StringValueFromFramework(ctx, new.TableBucketARN)
+	input := s3tables.PutTableBucketReplicationInput{
+		Configuration:  &configuration,
+		TableBucketARN: aws.String(tableBucketARN),
+		VersionToken:   fwflex.StringFromFramework(ctx, old.VersionToken),
+	}
+
+	output, err := conn.PutTableBucketReplication(ctx, &input)
+
+	if err != nil {
+		response.Diagnostics.AddError(fmt.Sprintf("updating S3 Tables Table Bucket Replication (%s)", tableBucketARN), err.Error())
+
+		return
+	}
+
+	// Set values for unknowns.
+	new.VersionToken = fwflex.StringToFramework(ctx, output.VersionToken)
+
+	response.Diagnostics.Append(response.State.Set(ctx, new)...)
+}
+
+func (r *tableBucketReplicationResource) Delete(ctx context.Context, request resource.DeleteRequest, response *resource.DeleteResponse) {
+	var data tableBucketReplicationResourceModel
+	response.Diagnostics.Append(request.State.Get(ctx, &data)...)
+	if response.Diagnostics.HasError() {
+		return
+	}
+
+	conn := r.Meta().S3TablesClient(ctx)
+
+	tableBucketARN := fwflex.StringValueFromFramework(ctx, data.TableBucketARN)
+	input := s3tables.DeleteTableBucketReplicationInput{
+		TableBucketARN: aws.String(tableBucketARN),
+		VersionToken:   fwflex.StringFromFramework(ctx, data.VersionToken),
+	}
+	_, err := conn.DeleteTableBucketReplication(ctx, &input)
+
+	if errs.IsA[*awstypes.NotFoundException](err) {
+		return
+	}
+
+	if err != nil {
+		response.Diagnostics.AddError(fmt.Sprintf("deleting S3 Tables Table Replication (%s)", tableBucketARN), err.Error())
+
+		return
+	}
+}
+
+func findTableBucketReplicationByARN(ctx context.Context, conn *s3tables.Client, tableBucketARN string) (*s3tables.GetTableBucketReplicationOutput, error) {
+	input := s3tables.GetTableBucketReplicationInput{
+		TableBucketARN: aws.String(tableBucketARN),
+	}
+
+	return findTableBucketReplication(ctx, conn, &input)
+}
+
+func findTableBucketReplication(ctx context.Context, conn *s3tables.Client, input *s3tables.GetTableBucketReplicationInput) (*s3tables.GetTableBucketReplicationOutput, error) {
+	output, err := conn.GetTableBucketReplication(ctx, input)
+
+	if errs.IsA[*awstypes.NotFoundException](err) {
+		return nil, &retry.NotFoundError{
+			LastError: err,
+		}
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	if output == nil || output.Configuration == nil {
+		return nil, tfresource.NewEmptyResultError(input)
+	}
+
+	return output, nil
+}
+
+type tableBucketReplicationResourceModel struct {
+	framework.WithRegionModel
+	Role           fwtypes.ARN                                                 `tfsdk:"role"`
+	Rules          fwtypes.ListNestedObjectValueOf[bucketReplicationRuleModel] `tfsdk:"rule"`
+	TableBucketARN fwtypes.ARN                                                 `tfsdk:"table_bucket_arn"`
+	VersionToken   types.String                                                `tfsdk:"version_token"`
+}
+
+type bucketReplicationRuleModel struct {
+	Destinations fwtypes.SetNestedObjectValueOf[replicationDestinationModel] `tfsdk:"destination"`
+}
+
+type replicationDestinationModel struct {
+	DestinationTableBucketARN fwtypes.ARN `tfsdk:"destination_table_bucket_arn"`
+}