feat(datastore): Add support for maintenance window (#19)

* feat(datastore): Add support for maintenance window

* feat(datastore): fix issues and make it work

* follow existing pattern

* remove computed true

* remove build

* set to null otherwise

Author: Pothulapati

internal/provider/datastore.go

@@ -158,6 +158,24 @@ func (r *datastoreResource) Schema(_ context.Context, _ resource.SchemaRequest,
 					},
 				},
 			},
+			"maintenance_window": schema.SingleNestedAttribute{
+				MarkdownDescription: "The maintenance window configuration for the datastore.",
+				Optional:            true,
+				Attributes: map[string]schema.Attribute{
+					"weekday": schema.Int64Attribute{
+						MarkdownDescription: "The day of the week to start the maintenance window. 0-6, 0 is Sunday.",
+						Optional:            true,
+					},
+					"hour": schema.Int64Attribute{
+						MarkdownDescription: "The hour of the day to start the maintenance window. 0-23.",
+						Optional:            true,
+					},
+					"duration_hours": schema.Int64Attribute{
+						MarkdownDescription: "DurationHours is the duration of the maintenance window in hours. 0 means maintenance is always allowed.",
+						Optional:            true,
+					},
+				},
+			},
 		},
 	}
 }

internal/resource_model/datastore.go

@@ -12,16 +12,17 @@ import (
 
 // Datastore maps the resource schema data.
 type Datastore struct {
-	ID             types.String      `tfsdk:"id"`
-	Name           types.String      `tfsdk:"name"`
-	NetworkId      types.String      `tfsdk:"network_id"`
-	Location       DatastoreLocation `tfsdk:"location"`
-	Tier           DatastoreTier     `tfsdk:"tier"`
-	Dragonfly      types.Object      `tfsdk:"dragonfly"`
-	CreatedAt      types.Int64       `tfsdk:"created_at"`
-	Password       types.String      `tfsdk:"password"`
-	Addr           types.String      `tfsdk:"addr"`
-	DisablePassKey types.Bool        `tfsdk:"disable_pass_key"`
+	ID                types.String      `tfsdk:"id"`
+	Name              types.String      `tfsdk:"name"`
+	NetworkId         types.String      `tfsdk:"network_id"`
+	Location          DatastoreLocation `tfsdk:"location"`
+	Tier              DatastoreTier     `tfsdk:"tier"`
+	Dragonfly         types.Object      `tfsdk:"dragonfly"`
+	CreatedAt         types.Int64       `tfsdk:"created_at"`
+	Password          types.String      `tfsdk:"password"`
+	Addr              types.String      `tfsdk:"addr"`
+	DisablePassKey    types.Bool        `tfsdk:"disable_pass_key"`
+	MaintenanceWindow types.Object      `tfsdk:"maintenance_window"`
 }
 
 type DatastoreLocation struct {
@@ -50,6 +51,24 @@ func (d *Datastore) FromConfig(ctx context.Context, in *dfcloud.Datastore) {
 	d.Tier.PerformanceTier = types.StringValue(string(in.Config.Tier.PerformanceTier))
 	d.Tier.Replicas = types.Int64Value(int64(*in.Config.Tier.Replicas))
 
+	if in.Config.MaintenanceWindow.DurationHours != nil || in.Config.MaintenanceWindow.Hour != nil || in.Config.MaintenanceWindow.Weekday != nil {
+		d.MaintenanceWindow = types.ObjectValueMust(map[string]attr.Type{
+			"weekday":        types.Int64Type,
+			"hour":           types.Int64Type,
+			"duration_hours": types.Int64Type,
+		}, map[string]attr.Value{
+			"weekday":        types.Int64Value(int64(lo.FromPtr(in.Config.MaintenanceWindow.Weekday))),
+			"hour":           types.Int64Value(int64(lo.FromPtr(in.Config.MaintenanceWindow.Hour))),
+			"duration_hours": types.Int64Value(int64(lo.FromPtr(in.Config.MaintenanceWindow.DurationHours))),
+		})
+	} else {
+		d.MaintenanceWindow = types.ObjectNull(map[string]attr.Type{
+			"weekday":        types.Int64Type,
+			"hour":           types.Int64Type,
+			"duration_hours": types.Int64Type,
+		})
+	}
+
 	aclRules, _ := types.ListValueFrom(ctx, types.StringType, in.Config.Dragonfly.AclRules)
 	d.Dragonfly = types.ObjectValueMust(map[string]attr.Type{
 		"cache_mode": types.BoolType,
@@ -135,6 +154,18 @@ func IntoDatastoreConfig(in Datastore) *dfcloud.Datastore {
 		datastore.Config.Dragonfly.AclRules = &rules
 	}
 
+	if in.MaintenanceWindow.Attributes()["weekday"] != nil {
+		datastore.Config.MaintenanceWindow.Weekday = lo.ToPtr(int(in.MaintenanceWindow.Attributes()["weekday"].(types.Int64).ValueInt64()))
+	}
+
+	if in.MaintenanceWindow.Attributes()["hour"] != nil {
+		datastore.Config.MaintenanceWindow.Hour = lo.ToPtr(int(in.MaintenanceWindow.Attributes()["hour"].(types.Int64).ValueInt64()))
+	}
+
+	if in.MaintenanceWindow.Attributes()["duration_hours"] != nil {
+		datastore.Config.MaintenanceWindow.DurationHours = lo.ToPtr(int(in.MaintenanceWindow.Attributes()["duration_hours"].(types.Int64).ValueInt64()))
+	}
+
 	return datastore
 }
 

internal/sdk/datastore.go

@@ -75,9 +75,21 @@ type DatastoreConfig struct {
 
 	Restore RestoreBackup `json:"restore"`
 
+	MaintenanceWindow MaintenanceWindow `json:"maintenance_window,omitempty"`
+
 	DisablePasskey bool `json:"disable_passkey"`
 }
 
+type MaintenanceWindow struct {
+	// Weekday is the day of the week to start the maintenance window. 0-6, 0 is Sunday.
+	Weekday *int `json:"weekday"`
+	// Hour is the hour of the day to start the maintenance window. 0-23.
+	Hour *int `json:"hour"`
+	// DurationHours is the duration of the maintenance window in hours.
+	// 0 means maintenance is always allowed.
+	DurationHours *int `json:"duration_hours"`
+}
+
 type RestoreBackup struct {
 	// Backup contains the ID of the backup to restore.
 	BackupId string `json:"backup_id"`