Merge pull request #34672 from hashicorp/f-codeguruprofiler_profiling…

…_group

[new resource/datasource]: aws_codeguruprofiler_profiling_group

.changelog/34672.txt

@@ -0,0 +1,7 @@
+```release-note:new-resource
+aws_codeguruprofiler_profiling_group
+```
+
+```release-note:new-data-source
+aws_codeguruprofiler_profiling_group
+```

internal/service/codeguruprofiler/exports_test.go

@@ -0,0 +1,11 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package codeguruprofiler
+
+// Exports for use in tests only.
+var (
+	ResourceProfilingGroup = newResourceProfilingGroup
+
+	FindProfilingGroupByName = findProfilingGroupByName
+)

internal/service/codeguruprofiler/profiling_group.go

@@ -0,0 +1,288 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package codeguruprofiler
+
+import (
+	"context"
+	"errors"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/codeguruprofiler"
+	awstypes "github.com/aws/aws-sdk-go-v2/service/codeguruprofiler/types"
+	"github.com/hashicorp/terraform-plugin-framework-validators/listvalidator"
+	"github.com/hashicorp/terraform-plugin-framework/path"
+	"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/id"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/retry"
+	"github.com/hashicorp/terraform-provider-aws/internal/create"
+	"github.com/hashicorp/terraform-provider-aws/internal/errs"
+	"github.com/hashicorp/terraform-provider-aws/internal/framework"
+	"github.com/hashicorp/terraform-provider-aws/internal/framework/flex"
+	fwtypes "github.com/hashicorp/terraform-provider-aws/internal/framework/types"
+	tftags "github.com/hashicorp/terraform-provider-aws/internal/tags"
+	"github.com/hashicorp/terraform-provider-aws/internal/tfresource"
+	"github.com/hashicorp/terraform-provider-aws/names"
+)
+
+// @FrameworkResource(name="Profiling Group")
+// @Tags(identifierAttribute="arn")
+func newResourceProfilingGroup(_ context.Context) (resource.ResourceWithConfigure, error) {
+	r := &resourceProfilingGroup{}
+
+	return r, nil
+}
+
+const (
+	ResNameProfilingGroup = "Profiling Group"
+)
+
+type resourceProfilingGroup struct {
+	framework.ResourceWithConfigure
+}
+
+func (r *resourceProfilingGroup) Metadata(_ context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
+	resp.TypeName = "aws_codeguruprofiler_profiling_group"
+}
+
+func (r *resourceProfilingGroup) Schema(ctx context.Context, req resource.SchemaRequest, resp *resource.SchemaResponse) {
+	computePlatform := fwtypes.StringEnumType[awstypes.ComputePlatform]()
+
+	resp.Schema = schema.Schema{
+		Attributes: map[string]schema.Attribute{
+			"arn": framework.ARNAttributeComputedOnly(),
+			"compute_platform": schema.StringAttribute{
+				CustomType: computePlatform,
+				Optional:   true,
+				Computed:   true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+					stringplanmodifier.UseStateForUnknown(),
+				},
+			},
+			"id": framework.IDAttribute(),
+			"name": schema.StringAttribute{
+				Required: true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
+			names.AttrTags:    tftags.TagsAttribute(),
+			names.AttrTagsAll: tftags.TagsAttributeComputedOnly(),
+		},
+		Blocks: map[string]schema.Block{
+			"agent_orchestration_config": schema.ListNestedBlock{
+				CustomType: fwtypes.NewListNestedObjectTypeOf[agentOrchestrationConfig](ctx),
+				Validators: []validator.List{
+					listvalidator.SizeAtMost(1),
+					listvalidator.IsRequired(),
+				},
+				NestedObject: schema.NestedBlockObject{
+					Attributes: map[string]schema.Attribute{
+						"profiling_enabled": schema.BoolAttribute{
+							Required: true,
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func (r *resourceProfilingGroup) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
+	conn := r.Meta().CodeGuruProfilerClient(ctx)
+
+	var plan resourceProfilingGroupData
+	resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	in := &codeguruprofiler.CreateProfilingGroupInput{}
+	resp.Diagnostics.Append(flex.Expand(ctx, plan, in)...)
+
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	in.ProfilingGroupName = flex.StringFromFramework(ctx, plan.Name)
+	in.ClientToken = aws.String(id.UniqueId())
+	in.Tags = getTagsIn(ctx)
+
+	out, err := conn.CreateProfilingGroup(ctx, in)
+	if err != nil {
+		resp.Diagnostics.AddError(
+			create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionCreating, ResNameProfilingGroup, plan.Name.ValueString(), err),
+			err.Error(),
+		)
+		return
+	}
+	if out == nil || out.ProfilingGroup == nil {
+		resp.Diagnostics.AddError(
+			create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionCreating, ResNameProfilingGroup, plan.Name.ValueString(), nil),
+			errors.New("empty output").Error(),
+		)
+		return
+	}
+
+	state := plan
+
+	resp.Diagnostics.Append(flex.Flatten(ctx, out.ProfilingGroup, &state)...)
+
+	state.ID = flex.StringToFramework(ctx, out.ProfilingGroup.Name)
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+}
+
+func (r *resourceProfilingGroup) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) {
+	conn := r.Meta().CodeGuruProfilerClient(ctx)
+
+	var state resourceProfilingGroupData
+	resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	out, err := findProfilingGroupByName(ctx, conn, state.ID.ValueString())
+	if tfresource.NotFound(err) {
+		resp.State.RemoveResource(ctx)
+		return
+	}
+	if err != nil {
+		resp.Diagnostics.AddError(
+			create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionSetting, ResNameProfilingGroup, state.ID.ValueString(), err),
+			err.Error(),
+		)
+		return
+	}
+
+	resp.Diagnostics.Append(flex.Flatten(ctx, out, &state)...)
+
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	setTagsOut(ctx, out.Tags)
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &state)...)
+}
+
+func (r *resourceProfilingGroup) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) {
+	conn := r.Meta().CodeGuruProfilerClient(ctx)
+
+	var plan, state resourceProfilingGroupData
+	resp.Diagnostics.Append(req.Plan.Get(ctx, &plan)...)
+	resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	if !plan.AgentOrchestrationConfig.Equal(state.AgentOrchestrationConfig) {
+		in := &codeguruprofiler.UpdateProfilingGroupInput{}
+		resp.Diagnostics.Append(flex.Expand(ctx, plan, in)...)
+
+		if resp.Diagnostics.HasError() {
+			return
+		}
+
+		in.ProfilingGroupName = flex.StringFromFramework(ctx, state.ID)
+		out, err := conn.UpdateProfilingGroup(ctx, in)
+		if err != nil {
+			resp.Diagnostics.AddError(
+				create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionUpdating, ResNameProfilingGroup, plan.ID.String(), err),
+				err.Error(),
+			)
+			return
+		}
+
+		if out == nil || out.ProfilingGroup == nil {
+			resp.Diagnostics.AddError(
+				create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionUpdating, ResNameProfilingGroup, plan.ID.String(), nil),
+				errors.New("empty output").Error(),
+			)
+			return
+		}
+	}
+
+	resp.Diagnostics.Append(resp.State.Set(ctx, &plan)...)
+}
+
+func (r *resourceProfilingGroup) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
+	conn := r.Meta().CodeGuruProfilerClient(ctx)
+
+	var state resourceProfilingGroupData
+	resp.Diagnostics.Append(req.State.Get(ctx, &state)...)
+	if resp.Diagnostics.HasError() {
+		return
+	}
+
+	in := &codeguruprofiler.DeleteProfilingGroupInput{
+		ProfilingGroupName: aws.String(state.ID.ValueString()),
+	}
+
+	_, err := conn.DeleteProfilingGroup(ctx, in)
+
+	if errs.IsA[*awstypes.ResourceNotFoundException](err) {
+		return
+	}
+
+	if err != nil {
+		resp.Diagnostics.AddError(
+			create.ProblemStandardMessage(names.CodeGuruProfiler, create.ErrActionDeleting, ResNameProfilingGroup, state.ID.String(), err),
+			err.Error(),
+		)
+		return
+	}
+}
+
+func (r *resourceProfilingGroup) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
+	resource.ImportStatePassthroughID(ctx, path.Root("id"), req, resp)
+}
+
+func (r *resourceProfilingGroup) ModifyPlan(ctx context.Context, request resource.ModifyPlanRequest, response *resource.ModifyPlanResponse) {
+	r.SetTagsAll(ctx, request, response)
+}
+
+func findProfilingGroupByName(ctx context.Context, conn *codeguruprofiler.Client, name string) (*awstypes.ProfilingGroupDescription, error) {
+	in := &codeguruprofiler.DescribeProfilingGroupInput{
+		ProfilingGroupName: aws.String(name),
+	}
+
+	out, err := conn.DescribeProfilingGroup(ctx, in)
+	if errs.IsA[*awstypes.ResourceNotFoundException](err) {
+		return nil, &retry.NotFoundError{
+			LastError:   err,
+			LastRequest: in,
+		}
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	if out == nil || out.ProfilingGroup == nil {
+		return nil, tfresource.NewEmptyResultError(in)
+	}
+
+	return out.ProfilingGroup, nil
+}
+
+type resourceProfilingGroupData struct {
+	ARN                      types.String                                              `tfsdk:"arn"`
+	AgentOrchestrationConfig fwtypes.ListNestedObjectValueOf[agentOrchestrationConfig] `tfsdk:"agent_orchestration_config"`
+	ComputePlatform          fwtypes.StringEnum[awstypes.ComputePlatform]              `tfsdk:"compute_platform"`
+	ID                       types.String                                              `tfsdk:"id"`
+	Name                     types.String                                              `tfsdk:"name"`
+	Tags                     types.Map                                                 `tfsdk:"tags"`
+	TagsAll                  types.Map                                                 `tfsdk:"tags_all"`
+}
+
+type agentOrchestrationConfig struct {
+	ProfilingEnabled types.Bool `tfsdk:"profiling_enabled"`
+}