0.2.64 update

server/internal/store/my_shows.go

@@ -18,9 +18,21 @@ type UserShow struct {
 type NotificationPreferences struct {
 	Enabled          bool `json:"enabled"`
 	ShowReturnAlerts bool `json:"showReturnAlerts"`
+	SonarrAlerts     bool `json:"sonarrAlerts"`
+	RadarrAlerts     bool `json:"radarrAlerts"`
+	UpdateAlerts     bool `json:"updateAlerts"`
+	LibraryAlerts    bool `json:"libraryAlerts"`
+	SystemAlerts     bool `json:"systemAlerts"`
 	LeadDays         int  `json:"leadDays"`
 }
 
+func DefaultNotificationPreferences() NotificationPreferences {
+	return NotificationPreferences{
+		Enabled: true, ShowReturnAlerts: true, SonarrAlerts: true, RadarrAlerts: true,
+		UpdateAlerts: true, LibraryAlerts: true, SystemAlerts: true, LeadDays: 7,
+	}
+}
+
 type UserNotification struct {
 	ID        int64      `json:"id"`
 	Kind      string     `json:"kind"`
@@ -195,11 +207,13 @@ func (s *Store) RecordSonarrSeriesStatuses(
 }
 
 func (s *Store) NotificationPreferences(ctx context.Context, userID string) (NotificationPreferences, error) {
-	prefs := NotificationPreferences{Enabled: true, ShowReturnAlerts: true, LeadDays: 7}
+	prefs := DefaultNotificationPreferences()
 	err := s.pool.QueryRow(ctx, `
-		SELECT enabled, show_return_alerts, lead_days
+		SELECT enabled, show_return_alerts, sonarr_alerts, radarr_alerts,
+		       update_alerts, library_alerts, system_alerts, lead_days
 		FROM user_notification_preferences WHERE emby_user_id = $1`, userID).
-		Scan(&prefs.Enabled, &prefs.ShowReturnAlerts, &prefs.LeadDays)
+		Scan(&prefs.Enabled, &prefs.ShowReturnAlerts, &prefs.SonarrAlerts, &prefs.RadarrAlerts,
+			&prefs.UpdateAlerts, &prefs.LibraryAlerts, &prefs.SystemAlerts, &prefs.LeadDays)
 	if err != nil && !isNoRows(err) {
 		return prefs, fmt.Errorf("store: notification preferences: %w", err)
 	}
@@ -217,17 +231,47 @@ func (s *Store) SetNotificationPreferences(
 	}
 	_, err := s.pool.Exec(ctx, `
 		INSERT INTO user_notification_preferences
-			(emby_user_id, enabled, show_return_alerts, lead_days)
-		VALUES ($1, $2, $3, $4)
+			(emby_user_id, enabled, show_return_alerts, sonarr_alerts, radarr_alerts,
+			 update_alerts, library_alerts, system_alerts, lead_days)
+		VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
 		ON CONFLICT (emby_user_id) DO UPDATE SET
 			enabled = EXCLUDED.enabled,
 			show_return_alerts = EXCLUDED.show_return_alerts,
+			sonarr_alerts = EXCLUDED.sonarr_alerts,
+			radarr_alerts = EXCLUDED.radarr_alerts,
+			update_alerts = EXCLUDED.update_alerts,
+			library_alerts = EXCLUDED.library_alerts,
+			system_alerts = EXCLUDED.system_alerts,
 			lead_days = EXCLUDED.lead_days,
 			updated_at = now()`,
-		userID, prefs.Enabled, prefs.ShowReturnAlerts, prefs.LeadDays)
+		userID, prefs.Enabled, prefs.ShowReturnAlerts, prefs.SonarrAlerts, prefs.RadarrAlerts,
+		prefs.UpdateAlerts, prefs.LibraryAlerts, prefs.SystemAlerts, prefs.LeadDays)
 	return err
 }
 
+func (s *Store) AllNotificationPreferences(ctx context.Context) (map[string]NotificationPreferences, error) {
+	rows, err := s.pool.Query(ctx, `
+		SELECT emby_user_id, enabled, show_return_alerts, sonarr_alerts, radarr_alerts,
+		       update_alerts, library_alerts, system_alerts, lead_days
+		FROM user_notification_preferences`)
+	if err != nil {
+		return nil, fmt.Errorf("store: list notification preferences: %w", err)
+	}
+	defer rows.Close()
+	result := map[string]NotificationPreferences{}
+	for rows.Next() {
+		prefs := DefaultNotificationPreferences()
+		var userID string
+		if err := rows.Scan(&userID, &prefs.Enabled, &prefs.ShowReturnAlerts, &prefs.SonarrAlerts,
+			&prefs.RadarrAlerts, &prefs.UpdateAlerts, &prefs.LibraryAlerts, &prefs.SystemAlerts,
+			&prefs.LeadDays); err != nil {
+			return nil, fmt.Errorf("store: scan notification preferences: %w", err)
+		}
+		result[userID] = prefs
+	}
+	return result, rows.Err()
+}
+
 func (s *Store) UpsertNotification(
 	ctx context.Context, userID, sourceKey, kind, itemID, title, message string, eventAt *time.Time,
 ) error {