Support CRUD operations for PITR configs, PITR restore, clone (#151)

Support CRUD operations for PITR configs, PITR restore and clone

Author: harshitha-yb

go.mod

@@ -2,15 +2,15 @@ module github.com/yugabyte/terraform-provider-ybm
 
 go 1.22.7
 
-toolchain go1.23.1
+toolchain go1.23.4
 
 require (
 	github.com/golang/mock v1.6.0
 	github.com/hashicorp/terraform-plugin-framework v0.10.0
 	github.com/hashicorp/terraform-plugin-framework-validators v0.4.0
 	github.com/hashicorp/terraform-plugin-log v0.9.0
 	github.com/sethvargo/go-retry v0.2.3
-	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250111091824-e051c979c3dd
+	github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250204064923-ec656335987b
 )
 
 require github.com/stretchr/testify v1.8.2 // indirect

go.sum

@@ -169,8 +169,8 @@ github.com/vmihailenco/msgpack/v5 v5.3.5 h1:5gO0H1iULLWGhs2H5tbAHIZTV8/cYafcFOr9
 github.com/vmihailenco/msgpack/v5 v5.3.5/go.mod h1:7xyJ9e+0+9SaZT0Wt1RGleJXzli6Q/V5KbhBonMG9jc=
 github.com/vmihailenco/tagparser/v2 v2.0.0 h1:y09buUbR+b5aycVFQs/g70pqKVZNBmxwAhO7/IwNM9g=
 github.com/vmihailenco/tagparser/v2 v2.0.0/go.mod h1:Wri+At7QHww0WTrCBeu4J6bNtoV6mEfg5OIWRZA9qds=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250111091824-e051c979c3dd h1:vuMsPasq6Uzoy5fBJ0HJz7/jiCuLhKRj+LOyNVolqCk=
-github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250111091824-e051c979c3dd/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250204064923-ec656335987b h1:yvhcGfFhG06y3I6KOJBogTkCwZasKZiroRjpVbXYa1s=
+github.com/yugabyte/yugabytedb-managed-go-client-internal v0.0.0-20250204064923-ec656335987b/go.mod h1:5vW0xIzIZw+1djkiWKx0qqNmqbRBSf4mjc4qw8lIMik=
 github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=

managed/fflags/feature_flags.go

@@ -15,11 +15,17 @@ type FeatureFlag string
 const (
 	CONNECTION_POOLING FeatureFlag = "CONNECTION_POOLING"
 	DR                 FeatureFlag = "DR"
+	PITR               FeatureFlag = "PITR"
+	PITR_RESTORE       FeatureFlag = "PITR_RESTORE"
+	PITR_CLONE         FeatureFlag = "PITR_CLONE"
 )
 
 var flagEnabled = map[FeatureFlag]bool{
 	CONNECTION_POOLING: false,
 	DR:                 false,
+	PITR:               false,
+	PITR_RESTORE:       false,
+	PITR_CLONE:         false,
 }
 
 func (f FeatureFlag) String() string {

managed/models.go

@@ -402,3 +402,40 @@ type DrConfig struct {
 	TargetClusterId types.String   `tfsdk:"target_cluster_id"`
 	Databases       []types.String `tfsdk:"databases"`
 }
+
+type PitrConfig struct {
+	AccountId                  types.String `tfsdk:"account_id"`
+	ProjectId                  types.String `tfsdk:"project_id"`
+	ClusterId                  types.String `tfsdk:"cluster_id"`
+	PitrConfigId               types.String `tfsdk:"pitr_config_id"`
+	NamespaceId                types.String `tfsdk:"namespace_id"`
+	NamespaceName              types.String `tfsdk:"namespace_name"`
+	NamespaceType              types.String `tfsdk:"namespace_type"`
+	RetentionPeriodInDays      types.Int64  `tfsdk:"retention_period_in_days"`
+	State                      types.String `tfsdk:"state"`
+	EarliestRecoveryTimeMillis types.Int64  `tfsdk:"earliest_recovery_time_millis"`
+	LatestRecoveryTimeMillis   types.Int64  `tfsdk:"latest_recovery_time_millis"`
+}
+
+type PitrRestore struct {
+	AccountId       types.String `tfsdk:"account_id"`
+	ProjectId       types.String `tfsdk:"project_id"`
+	ClusterId       types.String `tfsdk:"cluster_id"`
+	PitrConfigId    types.String `tfsdk:"pitr_config_id"`
+	PitrRestoreId   types.String `tfsdk:"pitr_restore_id"`
+	RestoreAtMillis types.Int64  `tfsdk:"restore_at_millis"`
+	State           types.String `tfsdk:"state"`
+}
+
+type PitrClone struct {
+	AccountId         types.String `tfsdk:"account_id"`
+	ProjectId         types.String `tfsdk:"project_id"`
+	ClusterId         types.String `tfsdk:"cluster_id"`
+	CloneNamespaceId  types.String `tfsdk:"clone_namespace_id"`
+	CloneAs           types.String `tfsdk:"clone_as"`
+	SourceNamespaceId types.String `tfsdk:"source_namespace_id"`
+	NamespaceName     types.String `tfsdk:"namespace_name"`
+	NamespaceType     types.String `tfsdk:"namespace_type"`
+	CloneAtMillis     types.Int64  `tfsdk:"clone_at_millis"`
+	State             types.String `tfsdk:"state"`
+}

managed/provider.go

@@ -166,6 +166,21 @@ func (p *provider) GetResources(_ context.Context) (map[string]tfsdk.ResourceTyp
 		resources["ybm_dr_config"] = resourceDrConfigType{}
 	}
 
+	// Add PITR config resource only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.PITR) {
+		resources["ybm_pitr_config"] = resourcePitrConfigType{}
+	}
+
+	// Add PITR restore resource only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.PITR_RESTORE) {
+		resources["ybm_pitr_restore"] = resourcePitrRestoreType{}
+	}
+
+	// Add PITR clone resource only if the feature flag is enabled
+	if fflags.IsFeatureFlagEnabled(fflags.PITR_CLONE) {
+		resources["ybm_pitr_clone"] = resourcePitrCloneType{}
+	}
+
 	return resources, nil
 }