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
 return q.db.GetPresetParametersByTemplateVersionID(ctx, args)
 }

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

 func (q *querier) GetPresetsAtFailureLimit(ctx context.Context, hardLimit int64) ([]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.
 return q.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)
 if err != nil {
 return nil, err
 }

 return q.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
 return parameters, nil
 }

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

 func (q *FakeQuerier) GetPresetsAtFailureLimit(ctx context.Context, hardLimit int64) ([]database.GetPresetsAtFailureLimitRow, error) {
 return nil, ErrUnimplemented
 }
 return presetParameters, nil
 }

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

 panic("not implemented")
 }

 func (q *FakeQuerier) InsertProvisionerJob(_ context.Context, arg database.InsertProvisionerJobParams) (database.ProvisionerJob, error) {
 if err := 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
 DROP TABLE template_version_preset_prebuild_schedules;

 -- Remove added columns from template_version_presets table
 ALTER TABLE 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
 ALTER TABLE template_version_presets
 ADD COLUMN autoscaling_enabled BOOLEAN DEFAULT false NOT NULL, -- Do we need it?
 ADD COLUMN autoscaling_timezone TEXT DEFAULT 'UTC' NOT NULL;

 -- New table for autoscaling schedules
 CREATE TABLE template_version_preset_prebuild_schedules (
    id UUID PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
    preset_id UUID NOT NULL,
    cron_expression TEXT NOT NULL,
    instances INTEGER NOT 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
		return q.db.GetPresetParametersByTemplateVersionID(ctx, args)
		}

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

		func (q *querier) GetPresetsAtFailureLimit(ctx context.Context, hardLimit int64) ([]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
		return q.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)
		if err != nil {
		return nil, err
		}

		return q.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,
		return parameters, nil
		}

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

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

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

		panic("not implemented")
		}

		func (q *FakeQuerier) InsertProvisionerJob(_ context.Context, arg database.InsertProvisionerJobParams) (database.ProvisionerJob, error) {
		if err := 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
		DROP TABLE template_version_preset_prebuild_schedules;

		-- Remove added columns from template_version_presets table
		ALTER TABLE 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
		ALTER TABLE template_version_presets
		ADD COLUMN autoscaling_enabled BOOLEAN DEFAULT false NOT NULL, -- Do we need it?
		ADD COLUMN autoscaling_timezone TEXT DEFAULT 'UTC' NOT NULL;

		-- New table for autoscaling schedules
		CREATE TABLE template_version_preset_prebuild_schedules (
		id UUID PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
		preset_id UUID NOT NULL,
		cron_expression TEXT NOT NULL,
		instances INTEGER NOT NULL,
		FOREIGN KEY (preset_id) REFERENCES template_version_presets (id) ON DELETE CASCADE
		);