fix yandex music null tracks

topi314 · Aug 28, 2023 · dec190c · dec190c
1 parent 3f26e2e
commit dec190c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/main/src/main/java/com/github/topi314/lavasrc/yandexmusic/YandexMusicSourceManager.java b/main/src/main/java/com/github/topi314/lavasrc/yandexmusic/YandexMusicSourceManager.java
@@ -107,7 +107,11 @@ private AudioItem getAlbum(String id) throws IOException {
 		var tracks = new ArrayList<AudioTrack>();
 		for (var volume : json.get("result").get("volumes").values()) {
 			for (var track : volume.values()) {
-				tracks.add(this.parseTrack(track));
+				var parsedTrack = this.parseTrack(track);
+				if (parsedTrack != null) {
+					tracks.add(parsedTrack);
+				}
+				tracks.add(parsedTrack);
 			}
 		}
 		if (tracks.isEmpty()) {
@@ -150,7 +154,10 @@ private AudioItem getPlaylist(String userString, String id) throws IOException {
 		}
 		var tracks = new ArrayList<AudioTrack>();
 		for (var track : json.get("result").get("tracks").values()) {
-			tracks.add(this.parseTrack(track.get("track")));
+			var parsedTrack = this.parseTrack(track);
+			if (parsedTrack != null) {
+				tracks.add(parsedTrack);
+			}
 		}
 		if (tracks.isEmpty()) {
 			return AudioReference.NO_TRACK;