May 29, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 4, 2025
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
 returnq.db.GetPresetParametersByTemplateVersionID(ctx,args)
 }

 func (q*querier)GetPresetPrebuildSchedules(ctx context.Context) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
 iferr:=q.authorizeContext(ctx,policy.ActionRead,rbac.ResourceTemplate.All());err!=nil {
 returnnil,err
 }
 returnq.db.GetPresetPrebuildSchedules(ctx)
 }

 func (q*querier)GetPresetsAtFailureLimit(ctx context.Context,hardLimitint64) ([]database.GetPresetsAtFailureLimitRow,error) {
 // GetPresetsAtFailureLimit returns a list of template version presets that have reached the hard failure limit.
 // Request the same authorization permissions as GetPresetsBackoff, since the methods are similar.
 returnq.db.InsertPresetParameters(ctx,arg)
 }

 func (q*querier)InsertPresetPrebuildSchedule(ctx context.Context,arg database.InsertPresetPrebuildScheduleParams) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
 err:=q.authorizeContext(ctx,policy.ActionUpdate,rbac.ResourceTemplate)
 iferr!=nil {
 returnnil,err
 }

 returnq.db.InsertPresetPrebuildSchedule(ctx,arg)
 }

 func (q*querier)InsertProvisionerJob(ctx context.Context,arg database.InsertProvisionerJobParams) (database.ProvisionerJob,error) {
 // TODO: Remove this once we have a proper rbac check for provisioner jobs.
 // Details in https://github.com/coder/coder/issues/16160
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
 }
 check.Args(insertPresetParametersParams).Asserts(rbac.ResourceTemplate,policy.ActionUpdate)
 }))
 s.Run("InsertPresetPrebuildSchedule",s.Subtest(func(db database.Store,check*expects) {
 org:=dbgen.Organization(s.T(),db, database.Organization{})
 user:=dbgen.User(s.T(),db, database.User{})
 template:=dbgen.Template(s.T(),db, database.Template{
 CreatedBy:user.ID,
 OrganizationID:org.ID,
 })
 templateVersion:=dbgen.TemplateVersion(s.T(),db, database.TemplateVersion{
 TemplateID:     uuid.NullUUID{UUID:template.ID,Valid:true},
 OrganizationID:org.ID,
 CreatedBy:user.ID,
 })
 insertPresetParams:= database.InsertPresetParams{
 TemplateVersionID:templateVersion.ID,
 Name:"test",
 }
 preset:=dbgen.Preset(s.T(),db,insertPresetParams)
 insertPresetPrebuildScheduleParams:= database.InsertPresetPrebuildScheduleParams{
 PresetID:preset.ID,
 }
 check.Args(insertPresetPrebuildScheduleParams).Asserts(rbac.ResourceTemplate,policy.ActionUpdate)
 }))
 s.Run("DeleteOrganizationMember",s.Subtest(func(db database.Store,check*expects) {
 o:=dbgen.Organization(s.T(),db, database.Organization{})
 u:=dbgen.User(s.T(),db, database.User{})
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
 returnparameters,nil
 }

 func (q*FakeQuerier)GetPresetPrebuildSchedules(ctx context.Context) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
 panic("not implemented")
 }

 func (q*FakeQuerier)GetPresetsAtFailureLimit(ctx context.Context,hardLimitint64) ([]database.GetPresetsAtFailureLimitRow,error) {
 returnnil,ErrUnimplemented
 }
 returnpresetParameters,nil
 }

 func (q*FakeQuerier)InsertPresetPrebuildSchedule(ctx context.Context,arg database.InsertPresetPrebuildScheduleParams) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
 err:=validateDatabaseType(arg)
 iferr!=nil {
 returnnil,err
 }

 panic("not implemented")
 }

 func (q*FakeQuerier)InsertProvisionerJob(_ context.Context,arg database.InsertProvisionerJobParams) (database.ProvisionerJob,error) {
 iferr:=validateDatabaseType(arg);err!=nil {
 return database.ProvisionerJob{},err
diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go
diff --git a/coderd/database/dbmock/dbmock.go b/coderd/database/dbmock/dbmock.go
diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/foreign_key_constraint.go b/coderd/database/foreign_key_constraint.go
diff --git a/coderd/database/migrations/000334_add_autoscaling_to_presets.down.sql b/coderd/database/migrations/000334_add_autoscaling_to_presets.down.sql
 -- Drop the autoscaling schedules table
 DROPTABLE template_version_preset_prebuild_schedules;

 -- Remove added columns from template_version_presets table
 ALTERTABLE template_version_presets
 DROP COLUMN autoscaling_enabled,
 DROP COLUMN autoscaling_timezone;
diff --git a/coderd/database/migrations/000334_add_autoscaling_to_presets.up.sql b/coderd/database/migrations/000334_add_autoscaling_to_presets.up.sql
 -- Add new columns to template_version_presets table
 ALTERTABLE template_version_presets
 ADD COLUMN autoscaling_enabledBOOLEAN DEFAULT falseNOT NULL,-- Do we need it?
 ADD COLUMN autoscaling_timezoneTEXT DEFAULT'UTC'NOT NULL;

 -- New table for autoscaling schedules
 CREATETABLEtemplate_version_preset_prebuild_schedules (
    id UUIDPRIMARY KEY DEFAULT gen_random_uuid()NOT NULL,
    preset_id UUIDNOT NULL,
    cron_expressionTEXTNOT NULL,
    instancesINTEGERNOT NULL,
 FOREIGN KEY (preset_id)REFERENCES template_version_presets (id)ON DELETE CASCADE
 );
diff --git a/coderd/database/models.go b/coderd/database/models.go
diff --git a/coderd/database/querier.go b/coderd/database/querier.go
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2267,6 +2267,13 @@ func (q *querier) GetPresetParametersByTemplateVersionID(ctx context.Context, ar
		returnq.db.GetPresetParametersByTemplateVersionID(ctx,args)
		}

		func (q*querier)GetPresetPrebuildSchedules(ctx context.Context) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
		iferr:=q.authorizeContext(ctx,policy.ActionRead,rbac.ResourceTemplate.All());err!=nil {
		returnnil,err
		}
		returnq.db.GetPresetPrebuildSchedules(ctx)
		}

		func (q*querier)GetPresetsAtFailureLimit(ctx context.Context,hardLimitint64) ([]database.GetPresetsAtFailureLimitRow,error) {
		// GetPresetsAtFailureLimit returns a list of template version presets that have reached the hard failure limit.
		// Request the same authorization permissions as GetPresetsBackoff, since the methods are similar.
Expand DownExpand Up		@@ -3618,6 +3625,15 @@ func (q *querier) InsertPresetParameters(ctx context.Context, arg database.Inser
		returnq.db.InsertPresetParameters(ctx,arg)
		}

		func (q*querier)InsertPresetPrebuildSchedule(ctx context.Context,arg database.InsertPresetPrebuildScheduleParams) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
		err:=q.authorizeContext(ctx,policy.ActionUpdate,rbac.ResourceTemplate)
		iferr!=nil {
		returnnil,err
		}

		returnq.db.InsertPresetPrebuildSchedule(ctx,arg)
		}

		func (q*querier)InsertProvisionerJob(ctx context.Context,arg database.InsertProvisionerJobParams) (database.ProvisionerJob,error) {
		// TODO: Remove this once we have a proper rbac check for provisioner jobs.
		// Details in https://github.com/coder/coder/issues/16160
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -979,6 +979,28 @@ func (s *MethodTestSuite) TestOrganization() {
		}
		check.Args(insertPresetParametersParams).Asserts(rbac.ResourceTemplate,policy.ActionUpdate)
		}))
		s.Run("InsertPresetPrebuildSchedule",s.Subtest(func(db database.Store,check*expects) {
		org:=dbgen.Organization(s.T(),db, database.Organization{})
		user:=dbgen.User(s.T(),db, database.User{})
		template:=dbgen.Template(s.T(),db, database.Template{
		CreatedBy:user.ID,
		OrganizationID:org.ID,
		})
		templateVersion:=dbgen.TemplateVersion(s.T(),db, database.TemplateVersion{
		TemplateID: uuid.NullUUID{UUID:template.ID,Valid:true},
		OrganizationID:org.ID,
		CreatedBy:user.ID,
		})
		insertPresetParams:= database.InsertPresetParams{
		TemplateVersionID:templateVersion.ID,
		Name:"test",
		}
		preset:=dbgen.Preset(s.T(),db,insertPresetParams)
		insertPresetPrebuildScheduleParams:= database.InsertPresetPrebuildScheduleParams{
		PresetID:preset.ID,
		}
		check.Args(insertPresetPrebuildScheduleParams).Asserts(rbac.ResourceTemplate,policy.ActionUpdate)
		}))
		s.Run("DeleteOrganizationMember",s.Subtest(func(db database.Store,check*expects) {
		o:=dbgen.Organization(s.T(),db, database.Organization{})
		u:=dbgen.User(s.T(),db, database.User{})
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -4369,6 +4369,10 @@ func (q *FakeQuerier) GetPresetParametersByTemplateVersionID(_ context.Context,
		returnparameters,nil
		}

		func (q*FakeQuerier)GetPresetPrebuildSchedules(ctx context.Context) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
		panic("not implemented")
		}

		func (q*FakeQuerier)GetPresetsAtFailureLimit(ctx context.Context,hardLimitint64) ([]database.GetPresetsAtFailureLimitRow,error) {
		returnnil,ErrUnimplemented
		}
Expand DownExpand Up		@@ -9156,6 +9160,15 @@ func (q *FakeQuerier) InsertPresetParameters(_ context.Context, arg database.Ins
		returnpresetParameters,nil
		}

		func (q*FakeQuerier)InsertPresetPrebuildSchedule(ctx context.Context,arg database.InsertPresetPrebuildScheduleParams) ([]database.TemplateVersionPresetPrebuildSchedule,error) {
		err:=validateDatabaseType(arg)
		iferr!=nil {
		returnnil,err
		}

		panic("not implemented")
		}

		func (q*FakeQuerier)InsertProvisionerJob(_ context.Context,arg database.InsertProvisionerJobParams) (database.ProvisionerJob,error) {
		iferr:=validateDatabaseType(arg);err!=nil {
		return database.ProvisionerJob{},err
Expand Down
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,7 @@
		-- Drop the autoscaling schedules table
		DROPTABLE template_version_preset_prebuild_schedules;

		-- Remove added columns from template_version_presets table
		ALTERTABLE template_version_presets
		DROP COLUMN autoscaling_enabled,
		DROP COLUMN autoscaling_timezone;
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,13 @@
		-- Add new columns to template_version_presets table
		ALTERTABLE template_version_presets
		ADD COLUMN autoscaling_enabledBOOLEAN DEFAULT falseNOT NULL,-- Do we need it?
		ADD COLUMN autoscaling_timezoneTEXT DEFAULT'UTC'NOT NULL;

		-- New table for autoscaling schedules
		CREATETABLEtemplate_version_preset_prebuild_schedules (
		id UUIDPRIMARY KEY DEFAULT gen_random_uuid()NOT NULL,
		preset_id UUIDNOT NULL,
		cron_expressionTEXTNOT NULL,
		instancesINTEGERNOT NULL,
		FOREIGN KEY (preset_id)REFERENCES template_version_presets (id)ON DELETE CASCADE
		);