Merge pull request #138 from planetscale/seed-data

merging of #132

Author: joemiller

docs/resources/branch.md

@@ -34,6 +34,7 @@ resource "planetscale_branch" "example" {
 ### Optional
 
 - `production` (Boolean) Whether or not the branch is a production branch.
+- `seed_data` (String) Seed data using the Data Branching® feature. Valid value is `last_successful_backup`
 
 ### Read-Only
 

internal/provider/branch_resource.go

@@ -37,7 +37,7 @@ type branchResource struct {
 
 // branchResourceModelV* and branchSchemaV* are defined in branch_resource_migration.go
 
-func branchResourceFromClient(ctx context.Context, branch *planetscale.Branch, organization, database types.String, diags diag.Diagnostics) *branchResourceModelV1 {
+func branchResourceFromClient(ctx context.Context, branch *planetscale.Branch, organization, database types.String, seedData types.String, diags diag.Diagnostics) *branchResourceModelV1 {
 	if branch == nil {
 		return nil
 	}
@@ -50,6 +50,7 @@ func branchResourceFromClient(ctx context.Context, branch *planetscale.Branch, o
 	return &branchResourceModelV1{
 		Organization: organization,
 		Database:     database,
+		SeedData:     seedData,
 
 		Actor:                       actor,
 		Region:                      region,
@@ -108,6 +109,7 @@ func (r *branchResource) Create(ctx context.Context, req resource.CreateRequest,
 	org := data.Organization
 	database := data.Database
 	name := data.Name
+	seedData := data.SeedData
 
 	if org.IsNull() || org.IsUnknown() || org.ValueString() == "" {
 		resp.Diagnostics.AddAttributeError(path.Root("organization"), "organization is required", "an organization must be provided and cannot be empty")
@@ -131,6 +133,7 @@ func (r *branchResource) Create(ctx context.Context, req resource.CreateRequest,
 	createReq := planetscale.CreateBranchReq{
 		Name:         name.ValueString(),
 		ParentBranch: *parentBranch,
+		SeedData:     seedData.ValueStringPointer(),
 	}
 	if !data.RestoredFromBranch.IsNull() && !data.RestoredFromBranch.IsUnknown() {
 		var rfb restoredFromBranchResource
@@ -199,7 +202,7 @@ func (r *branchResource) Create(ctx context.Context, req resource.CreateRequest,
 		branch = res.Branch
 	}
 
-	data = branchResourceFromClient(ctx, &branch, data.Organization, data.Database, resp.Diagnostics)
+	data = branchResourceFromClient(ctx, &branch, data.Organization, data.Database, data.SeedData, resp.Diagnostics)
 	if resp.Diagnostics.HasError() {
 		return
 	}
@@ -244,7 +247,7 @@ func (r *branchResource) Read(ctx context.Context, req resource.ReadRequest, res
 		return
 	}
 
-	data = branchResourceFromClient(ctx, &res.Branch, data.Organization, data.Database, resp.Diagnostics)
+	data = branchResourceFromClient(ctx, &res.Branch, data.Organization, data.Database, data.SeedData, resp.Diagnostics)
 	if resp.Diagnostics.HasError() {
 		return
 	}
@@ -303,7 +306,7 @@ func (r *branchResource) Update(ctx context.Context, req resource.UpdateRequest,
 			branch = res.Branch
 		}
 	}
-	data = branchResourceFromClient(ctx, &branch, data.Organization, data.Database, resp.Diagnostics)
+	data = branchResourceFromClient(ctx, &branch, data.Organization, data.Database, data.SeedData, resp.Diagnostics)
 	if resp.Diagnostics.HasError() {
 		return
 	}

internal/provider/branch_resource_migration.go

@@ -38,6 +38,7 @@ type branchResourceModelV1 struct {
 	SchemaLastUpdatedAt         types.String  `tfsdk:"schema_last_updated_at"`
 	ShardCount                  types.Float64 `tfsdk:"shard_count"`
 	Sharded                     types.Bool    `tfsdk:"sharded"`
+	SeedData                    types.String  `tfsdk:"seed_data"`
 	UpdatedAt                   types.String  `tfsdk:"updated_at"`
 }
 
@@ -141,6 +142,13 @@ func branchSchemaV1() *schema.Schema {
 				Description: "When the branch was last updated.",
 				Computed:    true,
 			},
+			"seed_data": schema.StringAttribute{
+				Description: "Seed data using the Data Branching® feature. Valid value is `last_successful_backup`",
+				Optional:    true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
 		},
 	}
 }
@@ -170,6 +178,7 @@ type branchResourceModelV0 struct {
 	ShardCount                  types.Float64 `tfsdk:"shard_count"`
 	Sharded                     types.Bool    `tfsdk:"sharded"`
 	UpdatedAt                   types.String  `tfsdk:"updated_at"`
+	SeedData                    types.String  `tfsdk:"seed_data"`
 }
 
 // branchSchemaV0 defines the schema for version 0.
@@ -271,6 +280,13 @@ func branchSchemaV0() *schema.Schema {
 				Description: "When the branch was last updated.",
 				Computed:    true,
 			},
+			"seed_data": schema.StringAttribute{
+				Description: "Seed data using the Data Branching® feature. Valid value is `last_successful_backup`",
+				Optional:    true,
+				PlanModifiers: []planmodifier.String{
+					stringplanmodifier.RequiresReplace(),
+				},
+			},
 		},
 	}
 }
@@ -304,6 +320,7 @@ func upgradeBranchStateV0toCurrent(ctx context.Context, req resource.UpgradeStat
 		ShardCount:                  priorStateData.ShardCount,
 		Sharded:                     priorStateData.Sharded,
 		UpdatedAt:                   priorStateData.UpdatedAt,
+		SeedData:                    priorStateData.SeedData,
 	}
 
 	resp.Diagnostics.Append(resp.State.Set(ctx, upgradedStateData)...)

internal/provider/branch_resource_migration_test.go

@@ -42,6 +42,7 @@ func TestUpgradeBranchStateV0toCurrent(t *testing.T) {
 					"shard_count":                    tftypes.NewValue(tftypes.Number, 1),
 					"sharded":                        tftypes.NewValue(tftypes.Bool, false),
 					"updated_at":                     tftypes.NewValue(tftypes.String, "2023-01-01T03:00:00Z"),
+					"seed_data":                      tftypes.NewValue(tftypes.String, "last_successful_backup"),
 				}),
 			Schema: branchSchemaV0(),
 		},
@@ -78,6 +79,7 @@ func TestUpgradeBranchStateV0toCurrent(t *testing.T) {
 					"shard_count":                    tftypes.NewValue(tftypes.Number, 1),
 					"sharded":                        tftypes.NewValue(tftypes.Bool, false),
 					"updated_at":                     tftypes.NewValue(tftypes.String, "2023-01-01T03:00:00Z"),
+					"seed_data":                      tftypes.NewValue(tftypes.String, "last_successful_backup"),
 				}),
 			Schema: branchSchemaV1(),
 		},