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;