WIP

mtojek · mtojek · commit f4e34a730c80 · 2024-09-09T15:40:25.000+02:00
diff --git a/coderd/notifications/reports/generator.go b/coderd/notifications/reports/generator.go
@@ -2,23 +2,26 @@ package reports
 
 import (
 	"context"
+	"database/sql"
 	"io"
 	"time"
 
 	"cdr.dev/slog"
+	"golang.org/x/xerrors"
 
 	"github.com/coder/coder/v2/coderd/database"
 	"github.com/coder/coder/v2/coderd/database/dbauthz"
 	"github.com/coder/coder/v2/coderd/database/dbtime"
 	"github.com/coder/coder/v2/coderd/notifications"
+	"github.com/coder/coder/v2/codersdk"
 	"github.com/coder/quartz"
 )
 
 const (
 	delay = 5 * time.Minute
 )
 
-func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, _ notifications.Enqueuer, clk quartz.Clock) io.Closer {
+func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, enqueur notifications.Enqueuer, clk quartz.Clock) io.Closer {
 	closed := make(chan struct{})
 
 	ctx, cancelFunc := context.WithCancel(ctx)
@@ -41,22 +44,11 @@ func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Sto
 				return nil
 			}
 
-			// TODO Report - workspace_builds_failed:
-			//
-			// 1. Fetch template admins.
-			// 2. Fetch templates.
-			// 3. For every template:
-			//    1. Fetch failed builds.
-			//    2. If failed builds == 0, continue.
-			//    3. Render the report.
-			//    4. Fetch template RW users.
-			//    5. For user := range template admins + RW users:
-			//       1. Check if report is enabled for the person.
-			//       2. Check `report_generator_log`.
-			//       3. If sent recently, continue
-			//       4. Send notification
-			//       5. Upsert into `report_generator_log`.
-			// 4. clean stale `report_generator_log` entries
+			err = reportFailedWorkspaceBuilds(ctx, logger, db, enqueur, clk)
+			if err != nil {
+				logger.Debug(ctx, "unable to report failed workspace builds")
+				return err
+			}
 
 			logger.Info(ctx, "report generator finished", slog.F("duration", clk.Since(start)))
 
@@ -98,3 +90,41 @@ func (i *reportGenerator) Close() error {
 	<-i.closed
 	return nil
 }
+
+func reportFailedWorkspaceBuilds(ctx context.Context, logger slog.Logger, db database.Store, _ notifications.Enqueuer, clk quartz.Clock) error {
+	const frequencyDays = 7
+
+	templateAdmins, err := db.GetUsers(ctx, database.GetUsersParams{
+		RbacRole: []string{codersdk.RoleTemplateAdmin},
+	})
+	if err != nil {
+		return xerrors.Errorf("unable to fetch template admins: %w", err)
+	}
+
+	templates, err := db.GetTemplatesWithFilter(ctx, database.GetTemplatesWithFilterParams{
+		Deleted:    false,
+		Deprecated: sql.NullBool{Bool: false, Valid: true},
+	})
+	if err != nil {
+		return xerrors.Errorf("unable to fetch active templates: %w", err)
+	}
+
+	for _, template := range templates {
+		//    1. Fetch failed builds.
+		//    2. If failed builds == 0, continue.
+		//    3. Render the report.
+		//    4. Fetch template RW users.
+		//    5. For user := range template admins + RW users:
+		//       1. Check if report is enabled for the person.
+		//       2. Check `report_generator_log`.
+		//       3. If sent recently, continue
+		//       4. Send notification
+		//       5. Upsert into `report_generator_log`.
+	}
+
+	err = db.DeleteOldReportGeneratorLogs(ctx, frequencyDays)
+	if err != nil {
+		return xerrors.Errorf("unable to delete old report generator logs: %w", err)
+	}
+	return nil
+}
