added creed

th0rn0 · Sep 27, 2024 · 558ec2c · 558ec2c
1 parent 539cd9e
commit 558ec2c
Showing 1 changed file with 25 additions and 3 deletions.
diff --git a/api/helpers.go b/api/helpers.go
@@ -44,20 +44,42 @@ func getNextSongExcludeURI(excludeUri spotify.URI) (Track, error) {
 
 func getNextSongRandom() (Track, error) {
 	var track Track
-	if err := db.Raw("SELECT * FROM tracks ORDER BY random()").First(&track).Error; err != nil {
+	track, err := getNextCreedSong()
+	if err != nil {
+		if err := db.Raw("SELECT * FROM tracks ORDER BY random()").First(&track).Error; err != nil {
+			return track, err
+		}
+	}
+	return track, nil
+}
+
+func getNextCreedSong() (Track, error) {
+	var track Track
+	if err := db.Raw("SELECT * FROM tracks WHERE artist == 'Creed' ORDER BY random()").First(&track).Error; err != nil {
 		return track, err
 	}
 	return track, nil
 }
 
-func getNextSongRandomExcludeURI(excludeUri spotify.URI) (Track, error) {
+func getNextCreedSongExcludeURI(excludeUri spotify.URI) (Track, error) {
 	var track Track
-	if err := db.Raw("SELECT * FROM tracks WHERE uri != ? ORDER BY random()", excludeUri).First(&track).Error; err != nil {
+	if err := db.Raw("SELECT * FROM tracks WHERE artist == 'Creed' AND  uri != ? ORDER BY random()", excludeUri).First(&track).Error; err != nil {
 		return track, err
 	}
 	return track, nil
 }
 
+func getNextSongRandomExcludeURI(excludeUri spotify.URI) (Track, error) {
+	var track Track
+	track, err := getNextCreedSongExcludeURI(excludeUri)
+	if err != nil {
+		if err := db.Raw("SELECT * FROM tracks WHERE uri != ? ORDER BY random()", excludeUri).First(&track).Error; err != nil {
+			return track, err
+		}
+	}
+	return track, nil
+}
+
 func getNextSongByVotes() (Track, error) {
 	var track Track
 	if err := db.Raw("SELECT * FROM tracks WHERE votes = ( SELECT MAX(votes) FROM tracks )").First(&track).Error; err != nil {