Go through all badges when checking if a StreamInfoItem is a live stream

TeamNewPipe · Mar 1, 2020 · 3525223 · 3525223
1 parent 830b7d3
commit 3525223
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.../schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java b/.../schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java
@@ -55,9 +55,13 @@ public YoutubeStreamInfoItemExtractor(JsonObject videoInfoItem, @Nullable TimeAg
     @Override
     public StreamType getStreamType() {
         try {
-            if (videoInfo.getArray("badges").getObject(0).getObject("metadataBadgeRenderer").getString("label").equals("LIVE NOW")) {
-                return StreamType.LIVE_STREAM;
+            JsonArray badges = videoInfo.getArray("badges");
+            for (Object badge : badges) {
+                if (((JsonObject) badge).getObject("metadataBadgeRenderer").getString("label").equals("LIVE NOW")) {
+                    return StreamType.LIVE_STREAM;
+                }
             }
+
         } catch (Exception ignored) {}
         return StreamType.VIDEO_STREAM;
     }