From af49b3c48777dfb002401cc7680766312ae96a6b Mon Sep 17 00:00:00 2001
From: wb9688 <wb9688@users.noreply.github.com>
Date: Fri, 21 Feb 2020 20:08:34 +0100
Subject: [PATCH] Fix live streams in extractVideoPreviewInfo()

---
 .../youtube/extractors/YoutubeStreamExtractor.java | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamExtractor.java b/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamExtractor.java
index 802a28f3b..53f1585d2 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamExtractor.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamExtractor.java
@@ -1072,6 +1072,11 @@ public class YoutubeStreamExtractor extends StreamExtractor {
         return new YoutubeStreamInfoItemExtractor(videoInfo, timeAgoParser) {
             @Override
             public StreamType getStreamType() {
+                try {
+                    if (videoInfo.getArray("badges").getObject(0).getObject("metadataBadgeRenderer").getString("label").equals("LIVE NOW")) {
+                        return StreamType.LIVE_STREAM;
+                    }
+                } catch (Exception ignored) {}
                 return StreamType.VIDEO_STREAM;
             }
 
@@ -1103,6 +1108,7 @@ public class YoutubeStreamExtractor extends StreamExtractor {
             @Override
             public long getDuration() throws ParsingException {
                 try {
+                    if (getStreamType() == StreamType.LIVE_STREAM) return -1;
                     return YoutubeParsingHelper.parseDurationString(videoInfo.getObject("lengthText").getString("simpleText"));
                 } catch (Exception e) {
                     throw new ParsingException("Could not get duration", e);
@@ -1139,7 +1145,13 @@ public class YoutubeStreamExtractor extends StreamExtractor {
             @Override
             public long getViewCount() throws ParsingException {
                 try {
-                    String viewCount = videoInfo.getObject("viewCountText").getString("simpleText");
+                    String viewCount;
+                    if (getStreamType() == StreamType.LIVE_STREAM)  {
+                        viewCount = videoInfo.getObject("viewCountText")
+                                .getArray("runs").getObject(0).getString("text");
+                    } else {
+                        viewCount = videoInfo.getObject("viewCountText").getString("simpleText");
+                    }
                     if (viewCount.equals("Recommended for you")) return -1;
                     return Long.parseLong(Utils.removeNonDigitCharacters(viewCount));
                 } catch (Exception e) {