commit

backend/internal/photos/store.go

@@ -123,5 +123,16 @@ func (s *Store) Delete(ctx context.Context, id string) error {
 }
 
 func (s *Store) SetCover(ctx context.Context, albumID, photoID string) {
-	_, _ = s.pool.Exec(ctx, `UPDATE albums SET cover_photo_id = $2 WHERE id = $1`, albumID, photoID)
+	_, _ = s.pool.Exec(ctx, `UPDATE albums SET cover_photo_id = $2 WHERE id = $1 AND cover_photo_id IS NULL`, albumID, photoID)
+}
+
+func (s *Store) ClearCover(ctx context.Context, albumID, photoID string) {
+	_, _ = s.pool.Exec(ctx, `UPDATE albums SET cover_photo_id = NULL WHERE id = $1 AND cover_photo_id = $2`, albumID, photoID)
+}
+
+func (s *Store) PhotoInAlbum(ctx context.Context, photoID, albumID string) (bool, error) {
+	var exists bool
+	err := s.pool.QueryRow(ctx,
+		`SELECT EXISTS(SELECT 1 FROM photos WHERE id = $1 AND album_id = $2)`, photoID, albumID).Scan(&exists)
+	return exists, err
 }