[YouTube] Fix search subscriber count extraction with channel handles

Author: Theta-Dev
ShareASmile · Sep 2, 2023 · 0e15037 · 0e15037
1 parent 9bf0069
commit 0e15037
Showing 1 changed file with 25 additions and 2 deletions.
diff --git a/...schabi/newpipe/extractor/services/youtube/extractors/YoutubeChannelInfoItemExtractor.java b/...schabi/newpipe/extractor/services/youtube/extractors/YoutubeChannelInfoItemExtractor.java
@@ -34,9 +34,26 @@
 
 public class YoutubeChannelInfoItemExtractor implements ChannelInfoItemExtractor {
     private final JsonObject channelInfoItem;
+    /**
+     * New layout:
+     * "subscriberCountText": Channel handle
+     * "videoCountText": Subscriber count
+     */
+    private final boolean withHandle;
 
     public YoutubeChannelInfoItemExtractor(final JsonObject channelInfoItem) {
         this.channelInfoItem = channelInfoItem;
+
+        boolean wHandle = false;
+        try {
+            final String subscriberCountText = getTextFromObject(
+                    channelInfoItem.getObject("subscriberCountText"));
+            if (subscriberCountText != null) {
+                wHandle = subscriberCountText.startsWith("@");
+            }
+        } catch (final ParsingException ignored) {
+        }
+        this.withHandle = wHandle;
     }
 
     @Override
@@ -78,6 +95,11 @@ public long getSubscriberCount() throws ParsingException {
                 return -1;
             }
 
+            if (withHandle) {
+                return Utils.mixedNumberWordToLong(getTextFromObject(
+                        channelInfoItem.getObject("videoCountText")));
+            }
+
             return Utils.mixedNumberWordToLong(getTextFromObject(
                     channelInfoItem.getObject("subscriberCountText")));
         } catch (final Exception e) {
@@ -88,8 +110,9 @@ public long getSubscriberCount() throws ParsingException {
     @Override
     public long getStreamCount() throws ParsingException {
         try {
-            if (!channelInfoItem.has("videoCountText")) {
-                // Video count is not available, channel probably has no public uploads.
+            if (withHandle || !channelInfoItem.has("videoCountText")) {
+                // Video count is not available, either the channel has no public uploads
+                // or YouTube displays the channel handle instead.
                 return ListExtractor.ITEM_COUNT_UNKNOWN;
             }