Jun 20, 2025 · Jun 23, 2025 · Jun 23, 2025 · Jun 23, 2025 · Jun 23, 2025
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
 return q.db.GetTemplateParameterInsights(ctx, arg)
 }

 func (q *querier) GetTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.GetTemplatePrebuildNotificationCooldownParams) (database.TemplatePrebuildNotificationCooldown, error) {
 _, err := q.GetTemplateByID(ctx, arg.TemplateID)
 if err != nil {
 return database.TemplatePrebuildNotificationCooldown{}, xerrors.Errorf("failed to get template by id: %w", err)
 }
 return q.db.GetTemplatePrebuildNotificationCooldown(ctx, arg)
 }

 func (q *querier) GetTemplatePresetsWithPrebuilds(ctx context.Context, templateID uuid.NullUUID) ([]database.GetTemplatePresetsWithPrebuildsRow, error) {
 // GetTemplatePresetsWithPrebuilds retrieves template versions with configured presets and prebuilds.
 // Presets and prebuilds are part of the template, so if you can access templates - you can access them as well.
 return q.db.UpsertTelemetryItem(ctx, arg)
 }

 func (q *querier) UpsertTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.UpsertTemplatePrebuildNotificationCooldownParams) error {
 tmpl, err := q.db.GetTemplateByID(ctx, arg.TemplateID)
 if err != nil {
 return xerrors.Errorf("verify template by id: %w", err)
 }
 if err := q.authorizeContext(ctx, policy.ActionCreate, tmpl); err != nil {
 return err
 }
 return q.db.UpsertTemplatePrebuildNotificationCooldown(ctx, arg)
 }

 func (q *querier) UpsertTemplateUsageStats(ctx context.Context) error {
 if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil {
 return err
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
 check.Args(req).
 Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate)
 }))
 s.Run("GetTemplatePrebuildNotificationCooldown", 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{
 OrganizationID: org.ID,
 CreatedBy:      user.ID,
 })
 arg := database.GetTemplatePrebuildNotificationCooldownParams{
 TemplateID:       template.ID,
 NotificationType: "admin",
 }
 check.Args(arg).
 Asserts(template, policy.ActionRead).
 Returns(database.TemplatePrebuildNotificationCooldown{}).
 ErrorsWithPG(sql.ErrNoRows).
 ErrorsWithInMemDB(dbmem.ErrUnimplemented)
 }))
 s.Run("UpsertTemplatePrebuildNotificationCooldown", 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{
 OrganizationID: org.ID,
 CreatedBy:      user.ID,
 })
 arg := database.UpsertTemplatePrebuildNotificationCooldownParams{
 TemplateID:       template.ID,
 NotificationType: "admin",
 }
 check.Args(arg).
 Asserts(template, policy.ActionCreate).
 Returns().
 ErrorsWithInMemDB(dbmem.ErrUnimplemented)
 }))
 }

 func (s *MethodTestSuite) TestOAuth2ProviderApps() {
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
 return rows, nil
 }

 func (q *FakeQuerier) GetTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.GetTemplatePrebuildNotificationCooldownParams) (database.TemplatePrebuildNotificationCooldown, error) {
 err := validateDatabaseType(arg)
 if err != nil {
 return database.TemplatePrebuildNotificationCooldown{}, err
 }

 return database.TemplatePrebuildNotificationCooldown{}, ErrUnimplemented
 }

 func (*FakeQuerier) GetTemplatePresetsWithPrebuilds(_ context.Context, _ uuid.NullUUID) ([]database.GetTemplatePresetsWithPrebuildsRow, error) {
 return nil, ErrUnimplemented
 }
 return nil
 }

 func (q *FakeQuerier) UpsertTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.UpsertTemplatePrebuildNotificationCooldownParams) error {
 err := validateDatabaseType(arg)
 if err != nil {
 return err
 }

 return ErrUnimplemented
 }

 func (q *FakeQuerier) UpsertTemplateUsageStats(ctx context.Context) error {
 q.mutex.Lock()
 defer q.mutex.Unlock()
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/migrations/000341_template_prebuild_notification_cooldowns.down.sql b/coderd/database/migrations/000341_template_prebuild_notification_cooldowns.down.sql
 DROP TABLE IF EXISTS template_prebuild_notification_cooldowns;
diff --git a/coderd/database/migrations/000341_template_prebuild_notification_cooldowns.up.sql b/coderd/database/migrations/000341_template_prebuild_notification_cooldowns.up.sql
 CREATE TABLE template_prebuild_notification_cooldowns (
 template_id UUID NOT NULL,
 notification_type TEXT NOT NULL,
 last_notification_sent TIMESTAMPTZ NOT NULL,
 PRIMARY KEY (template_id, notification_type)
 );

 COMMENT ON TABLE template_prebuild_notification_cooldowns IS 'Tracks when prebuild failure notifications were last sent to prevent notification noise';
 COMMENT ON COLUMN template_prebuild_notification_cooldowns.notification_type IS 'Type of notification: admin or author';
diff --git a/...abase/migrations/testdata/fixtures/000341_template_prebuild_notification_cooldowns.up.sql b/...abase/migrations/testdata/fixtures/000341_template_prebuild_notification_cooldowns.up.sql
 INSERT INTO
 template_prebuild_notification_cooldowns (
 template_id,
 notification_type,
 last_notification_sent
 )
 VALUES (
 '6b298946-7a4f-47ac-9158-b03b08740a41',
 'admin',
 '2025-01-01 00:00:00+00'
 );
diff --git a/coderd/database/models.go b/coderd/database/models.go
diff --git a/coderd/database/querier.go b/coderd/database/querier.go
diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go
diff --git a/coderd/database/queries/prebuilds.sql b/coderd/database/queries/prebuilds.sql
 WHERE NOT t.deleted AND wpb.build_number = 1
 GROUP BY t.name, tvp.name, o.name
 ORDER BY t.name, tvp.name, o.name;

 -- name: GetTemplatePrebuildNotificationCooldown :one
 SELECT * FROM template_prebuild_notification_cooldowns
 WHERE template_id = $1 AND notification_type = $2;

 -- name: UpsertTemplatePrebuildNotificationCooldown :exec
 INSERT INTO template_prebuild_notification_cooldowns (
 template_id,
 notification_type,
 last_notification_sent
 ) VALUES (
 $1, $2, $3
 ) ON CONFLICT (template_id, notification_type) DO UPDATE SET
 last_notification_sent = $3;
diff --git a/coderd/database/unique_constraint.go b/coderd/database/unique_constraint.go
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2665,6 +2665,14 @@ func (q *querier) GetTemplateParameterInsights(ctx context.Context, arg database
		return q.db.GetTemplateParameterInsights(ctx, arg)
		}

		func (q *querier) GetTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.GetTemplatePrebuildNotificationCooldownParams) (database.TemplatePrebuildNotificationCooldown, error) {
		_, err := q.GetTemplateByID(ctx, arg.TemplateID)
		if err != nil {
		return database.TemplatePrebuildNotificationCooldown{}, xerrors.Errorf("failed to get template by id: %w", err)
		}
		return q.db.GetTemplatePrebuildNotificationCooldown(ctx, arg)
		}

		func (q *querier) GetTemplatePresetsWithPrebuilds(ctx context.Context, templateID uuid.NullUUID) ([]database.GetTemplatePresetsWithPrebuildsRow, error) {
		// GetTemplatePresetsWithPrebuilds retrieves template versions with configured presets and prebuilds.
		// Presets and prebuilds are part of the template, so if you can access templates - you can access them as well.
Expand DownExpand Up		@@ -5136,6 +5144,17 @@ func (q *querier) UpsertTelemetryItem(ctx context.Context, arg database.UpsertTe
		return q.db.UpsertTelemetryItem(ctx, arg)
		}

		func (q *querier) UpsertTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.UpsertTemplatePrebuildNotificationCooldownParams) error {
		tmpl, err := q.db.GetTemplateByID(ctx, arg.TemplateID)
		if err != nil {
		return xerrors.Errorf("verify template by id: %w", err)
		}
		if err := q.authorizeContext(ctx, policy.ActionCreate, tmpl); err != nil {
		return err
		}
		return q.db.UpsertTemplatePrebuildNotificationCooldown(ctx, arg)
		}

		func (q *querier) UpsertTemplateUsageStats(ctx context.Context) error {
		if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceSystem); err != nil {
		return err
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -5087,6 +5087,39 @@ func (s *MethodTestSuite) TestPrebuilds() {
		check.Args(req).
		Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate)
		}))
		s.Run("GetTemplatePrebuildNotificationCooldown", 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{
		OrganizationID: org.ID,
		CreatedBy: user.ID,
		})
		arg := database.GetTemplatePrebuildNotificationCooldownParams{
		TemplateID: template.ID,
		NotificationType: "admin",
		}
		check.Args(arg).
		Asserts(template, policy.ActionRead).
		Returns(database.TemplatePrebuildNotificationCooldown{}).
		ErrorsWithPG(sql.ErrNoRows).
		ErrorsWithInMemDB(dbmem.ErrUnimplemented)
		}))
		s.Run("UpsertTemplatePrebuildNotificationCooldown", 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{
		OrganizationID: org.ID,
		CreatedBy: user.ID,
		})
		arg := database.UpsertTemplatePrebuildNotificationCooldownParams{
		TemplateID: template.ID,
		NotificationType: "admin",
		}
		check.Args(arg).
		Asserts(template, policy.ActionCreate).
		Returns().
		ErrorsWithInMemDB(dbmem.ErrUnimplemented)
		}))
		}

		func (s *MethodTestSuite) TestOAuth2ProviderApps() {
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -6178,6 +6178,15 @@ func (q *FakeQuerier) GetTemplateParameterInsights(ctx context.Context, arg data
		return rows, nil
		}

		func (q *FakeQuerier) GetTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.GetTemplatePrebuildNotificationCooldownParams) (database.TemplatePrebuildNotificationCooldown, error) {
		err := validateDatabaseType(arg)
		if err != nil {
		return database.TemplatePrebuildNotificationCooldown{}, err
		}

		return database.TemplatePrebuildNotificationCooldown{}, ErrUnimplemented
		}

		func (*FakeQuerier) GetTemplatePresetsWithPrebuilds(_ context.Context, _ uuid.NullUUID) ([]database.GetTemplatePresetsWithPrebuildsRow, error) {
		return nil, ErrUnimplemented
		}
Expand DownExpand Up		@@ -12507,6 +12516,15 @@ func (q *FakeQuerier) UpsertTelemetryItem(_ context.Context, arg database.Upsert
		return nil
		}

		func (q *FakeQuerier) UpsertTemplatePrebuildNotificationCooldown(ctx context.Context, arg database.UpsertTemplatePrebuildNotificationCooldownParams) error {
		err := validateDatabaseType(arg)
		if err != nil {
		return err
		}

		return ErrUnimplemented
		}

		func (q *FakeQuerier) UpsertTemplateUsageStats(ctx context.Context) error {
		q.mutex.Lock()
		defer q.mutex.Unlock()
Expand Down
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		DROP TABLE IF EXISTS template_prebuild_notification_cooldowns;
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,9 @@
		CREATE TABLE template_prebuild_notification_cooldowns (
		template_id UUID NOT NULL,
		notification_type TEXT NOT NULL,
		last_notification_sent TIMESTAMPTZ NOT NULL,
		PRIMARY KEY (template_id, notification_type)
		);

		COMMENT ON TABLE template_prebuild_notification_cooldowns IS 'Tracks when prebuild failure notifications were last sent to prevent notification noise';
		COMMENT ON COLUMN template_prebuild_notification_cooldowns.notification_type IS 'Type of notification: admin or author';
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,11 @@
		INSERT INTO
		template_prebuild_notification_cooldowns (
		template_id,
		notification_type,
		last_notification_sent
		)
		VALUES (
		'6b298946-7a4f-47ac-9158-b03b08740a41',
		'admin',
		'2025-01-01 00:00:00+00'
		);
Original file line number	Diff line number	Diff line change
Expand Up		@@ -188,3 +188,17 @@ INNER JOIN organizations o ON o.id = w.organization_id
		WHERE NOT t.deleted AND wpb.build_number = 1
		GROUP BY t.name, tvp.name, o.name
		ORDER BY t.name, tvp.name, o.name;

		-- name: GetTemplatePrebuildNotificationCooldown :one
		SELECT * FROM template_prebuild_notification_cooldowns
		WHERE template_id = $1 AND notification_type = $2;

		-- name: UpsertTemplatePrebuildNotificationCooldown :exec
		INSERT INTO template_prebuild_notification_cooldowns (
		template_id,
		notification_type,
		last_notification_sent
		) VALUES (
		$1, $2, $3
		) ON CONFLICT (template_id, notification_type) DO UPDATE SET
		last_notification_sent = $3;