Merge pull request #1385 from hashicorp/SwiftEngineer/TF-11539

Support for Data Retention Policies

Author: SwiftEngineer

CHANGELOG.md

@@ -15,6 +15,9 @@ ENHANCEMENTS:
 * `r/tfe_oauth_client`: Add `agent_pool_id` as an optional argument to enable Private VCS support, by @roleesinhaHC [1255](https://github.com/hashicorp/terraform-provider-tfe/pull/1255)
 * `r/tfe_project`: Increase the Project name length from 36 to 40 characters @hs26gill [#1351](https://github.com/hashicorp/terraform-provider-tfe/pull/1351)
 
+FEATURES:
+* **New Resource**: `r/tfe_data_retention_policy` is a new resource for managing data retention policies for organizations and workspaces, by @SwiftEngineer [1385](https://github.com/hashicorp/terraform-provider-tfe/pull/1385)
+
 BUG FIXES:
 * `r/tfe_registry_module`: Prevents constant diff after a successful apply when `tags` and `tests_enabled` is not set by @Uk1288 [#1357](https://github.com/hashicorp/terraform-provider-tfe/pull/1357)
 

internal/provider/data_retention_policy.go

@@ -0,0 +1,81 @@
+package provider
+
+import (
+	"context"
+	"github.com/hashicorp/go-tfe"
+	"github.com/hashicorp/terraform-plugin-framework/attr"
+	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+	"github.com/hashicorp/terraform-plugin-framework/types/basetypes"
+	"math/big"
+)
+
+type modelTFEDataRetentionPolicy struct {
+	ID              types.String `tfsdk:"id"`
+	Organization    types.String `tfsdk:"organization"`
+	WorkspaceID     types.String `tfsdk:"workspace_id"`
+	DeleteOlderThan types.Object `tfsdk:"delete_older_than"`
+	DontDelete      types.Object `tfsdk:"dont_delete"`
+}
+
+type modelTFEDeleteOlderThan struct {
+	Days types.Number `tfsdk:"days"`
+}
+
+func (m modelTFEDeleteOlderThan) AttributeTypes() map[string]attr.Type {
+	return map[string]attr.Type{
+		"days": types.NumberType,
+	}
+}
+
+func DontDeleteEmptyObject() basetypes.ObjectValue {
+	object, diags := types.ObjectValue(map[string]attr.Type{}, map[string]attr.Value{})
+	if diags.HasError() {
+		panic(diags.Errors())
+	}
+	return object
+}
+
+func modelFromTFEDataRetentionPolicyDeleteOlder(ctx context.Context, model modelTFEDataRetentionPolicy, deleteOlder *tfe.DataRetentionPolicyDeleteOlder) (modelTFEDataRetentionPolicy, diag.Diagnostics) {
+	deleteOlderThan := modelTFEDeleteOlderThan{
+		Days: types.NumberValue(big.NewFloat(float64(deleteOlder.DeleteOlderThanNDays))),
+	}
+	deleteOlderThanObject, diags := types.ObjectValueFrom(ctx, deleteOlderThan.AttributeTypes(), deleteOlderThan)
+
+	organization := types.StringNull()
+	if model.WorkspaceID.IsNull() {
+		organization = model.Organization
+	}
+
+	return modelTFEDataRetentionPolicy{
+		ID:              types.StringValue(deleteOlder.ID),
+		Organization:    organization,
+		WorkspaceID:     model.WorkspaceID,
+		DeleteOlderThan: deleteOlderThanObject,
+		DontDelete:      types.ObjectNull(map[string]attr.Type{}),
+	}, diags
+}
+
+func modelFromTFEDataRetentionPolicyDontDelete(model modelTFEDataRetentionPolicy, dontDelete *tfe.DataRetentionPolicyDontDelete) modelTFEDataRetentionPolicy {
+	organization := types.StringNull()
+	if model.WorkspaceID.IsNull() {
+		organization = model.Organization
+	}
+
+	return modelTFEDataRetentionPolicy{
+		ID:              types.StringValue(dontDelete.ID),
+		Organization:    organization,
+		WorkspaceID:     model.WorkspaceID,
+		DeleteOlderThan: types.ObjectNull(modelTFEDeleteOlderThan{}.AttributeTypes()),
+		DontDelete:      DontDeleteEmptyObject(),
+	}
+}
+
+func modelFromTFEDataRetentionPolicyChoice(ctx context.Context, model modelTFEDataRetentionPolicy, choice *tfe.DataRetentionPolicyChoice) (modelTFEDataRetentionPolicy, diag.Diagnostics) {
+	if choice.DataRetentionPolicyDeleteOlder != nil {
+		return modelFromTFEDataRetentionPolicyDeleteOlder(ctx, model, choice.DataRetentionPolicyDeleteOlder)
+	}
+
+	var emptyDiag []diag.Diagnostic
+	return modelFromTFEDataRetentionPolicyDontDelete(model, choice.DataRetentionPolicyDontDelete), emptyDiag
+}

internal/provider/provider_next.go

@@ -140,6 +140,7 @@ func (p *frameworkProvider) Resources(ctx context.Context) []func() resource.Res
 		NewRegistryGPGKeyResource,
 		NewRegistryProviderResource,
 		NewResourceVariable,
+		NewDataRetentionPolicyResource,
 		NewResourceWorkspaceSettings,
 		NewSAMLSettingsResource,
 		NewTestVariableResource,