coder · evgeniy-scherbina · May 29, 2025 · Jun 13, 2025 · Jun 13, 2025 · Jun 13, 2025
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
@@ -2282,6 +2282,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.
@@ -3633,6 +3640,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 database.TemplateVersionPresetPrebuildSchedule{}, 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
@@ -979,6 +979,29 @@ 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,
+		})
+		preset := dbgen.Preset(s.T(), db, database.InsertPresetParams{
+			TemplateVersionID: templateVersion.ID,
+			Name:              "test",
+		})
+		arg := database.InsertPresetPrebuildScheduleParams{
+			PresetID: preset.ID,
+		}
+		check.Args(arg).
+			Asserts(rbac.ResourceTemplate, policy.ActionUpdate).
+			ErrorsWithInMemDB(dbmem.ErrUnimplemented)
+	}))
 	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{})
@@ -4913,6 +4936,12 @@ func (s *MethodTestSuite) TestPrebuilds() {
 			Asserts(template.RBACObject(), policy.ActionRead).
 			Returns(insertedParameters)
 	}))
+	s.Run("GetPresetPrebuildSchedules", s.Subtest(func(db database.Store, check *expects) {
+		check.Args().
+			Asserts(rbac.ResourceTemplate.All(), policy.ActionRead).
+			Returns([]database.TemplateVersionPresetPrebuildSchedule{}).
+			ErrorsWithInMemDB(dbmem.ErrUnimplemented)
+	}))
 	s.Run("GetPresetsByTemplateVersionID", s.Subtest(func(db database.Store, check *expects) {
 		ctx := context.Background()
 		org := dbgen.Organization(s.T(), db, database.Organization{})

diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -395,6 +395,7 @@ func (t TemplateVersionBuilder) Do() TemplateVersionResponse {
 			CreatedAt:           version.CreatedAt,
 			DesiredInstances:    preset.DesiredInstances,
 			InvalidateAfterSecs: preset.InvalidateAfterSecs,
+			AutoscalingTimezone: preset.AutoscalingTimezone,
 		})
 	}
 

diff --git a/coderd/database/dbgen/dbgen.go b/coderd/database/dbgen/dbgen.go
@@ -1259,11 +1259,22 @@ func Preset(t testing.TB, db database.Store, seed database.InsertPresetParams) d
 		CreatedAt:           takeFirst(seed.CreatedAt, dbtime.Now()),
 		DesiredInstances:    seed.DesiredInstances,
 		InvalidateAfterSecs: seed.InvalidateAfterSecs,
+		AutoscalingTimezone: seed.AutoscalingTimezone,
 	})
 	require.NoError(t, err, "insert preset")
 	return preset
 }
 
+func PresetPrebuildSchedule(t testing.TB, db database.Store, seed database.InsertPresetPrebuildScheduleParams) database.TemplateVersionPresetPrebuildSchedule {
+	schedule, err := db.InsertPresetPrebuildSchedule(genCtx, database.InsertPresetPrebuildScheduleParams{
+		PresetID:       seed.PresetID,
+		CronExpression: seed.CronExpression,
+		Instances:      seed.Instances,
+	})
+	require.NoError(t, err, "insert preset")
-	require.NoError(t, err, "insert preset")
+	require.NoError(t, err, "insert preset prebuild schedule")
-	require.NoError(t, err, "insert preset")
+	require.NoError(t, err, "insert preset prebuild schedule")
+	return schedule
+}
+
 func PresetParameter(t testing.TB, db database.Store, seed database.InsertPresetParametersParams) []database.TemplateVersionPresetParameter {
 	parameters, err := db.InsertPresetParameters(genCtx, database.InsertPresetParametersParams{
 		TemplateVersionPresetID: takeFirst(seed.TemplateVersionPresetID, uuid.New()),

diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -4369,6 +4369,10 @@ func (q *FakeQuerier) GetPresetParametersByTemplateVersionID(_ context.Context,
 	return parameters, nil
 }
 
+func (q *FakeQuerier) GetPresetPrebuildSchedules(ctx context.Context) ([]database.TemplateVersionPresetPrebuildSchedule, error) {
+	return nil, ErrUnimplemented
+}
+
 func (q *FakeQuerier) GetPresetsAtFailureLimit(ctx context.Context, hardLimit int64) ([]database.GetPresetsAtFailureLimitRow, error) {
 	return nil, ErrUnimplemented
 }
@@ -9164,6 +9168,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 database.TemplateVersionPresetPrebuildSchedule{}, err
+	}
+
+	return database.TemplateVersionPresetPrebuildSchedule{}, ErrUnimplemented
+}
+
 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/000336_add_autoscaling_to_presets.down.sql b/coderd/database/migrations/000336_add_autoscaling_to_presets.down.sql
@@ -0,0 +1,6 @@
+-- Drop the autoscaling schedules table
+DROP TABLE template_version_preset_prebuild_schedules;
+
+-- Remove autoscaling_timezone column from template_version_presets table
+ALTER TABLE template_version_presets
+DROP COLUMN autoscaling_timezone;
diff --git a/coderd/database/migrations/000336_add_autoscaling_to_presets.up.sql b/coderd/database/migrations/000336_add_autoscaling_to_presets.up.sql
@@ -0,0 +1,12 @@
+-- Add autoscaling_timezone column to template_version_presets table
+ALTER TABLE template_version_presets
+ADD COLUMN autoscaling_timezone TEXT DEFAULT 'UTC' NOT NULL;
+
+-- Add 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/migrations/testdata/fixtures/000336_add_autoscaling_to_presets.up.sql b/coderd/database/migrations/testdata/fixtures/000336_add_autoscaling_to_presets.up.sql
@@ -0,0 +1,13 @@
+INSERT INTO
+	template_version_preset_prebuild_schedules (
+		id,
+		preset_id,
+		cron_expression,
+		instances
+	)
+	VALUES (
+		'e387cac1-9bf1-4fb6-8a34-db8cfb750dd0',
+		'28b42cc0-c4fe-4907-a0fe-e4d20f1e9bfe',
+		'* 8-18 * * 1-5',
+		1
+	);
diff --git a/coderd/database/models.go b/coderd/database/models.go
diff --git a/coderd/database/querier.go b/coderd/database/querier.go