Fix for requested changes.

TeamNewPipe · Feb 11, 2022 · 0779c33 · 0779c33
1 parent df708a7
commit 0779c33
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/...va/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeCommentsExtractor.java b/...va/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeCommentsExtractor.java
@@ -45,6 +45,9 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
      */
     @SuppressWarnings("OptionalUsedAsFieldOrParameterType")
     private Optional<Boolean> optCommentsDisabled = Optional.empty();
+    /**
+     * The second ajax <b>/next</b> response.
+     */
     private JsonObject ajaxJson;
 
     public YoutubeCommentsExtractor(
@@ -260,14 +263,17 @@ public boolean isCommentsDisabled() throws ExtractionException {
 
     @Override
     public int getCommentsCount() throws ExtractionException {
-        final JsonObject commentsHeaderRenderer = ajaxJson
+        final JsonObject countText = ajaxJson
                 .getArray("onResponseReceivedEndpoints").getObject(0)
                 .getObject("reloadContinuationItemsCommand")
                 .getArray("continuationItems").getObject(0)
-                .getObject("commentsHeaderRenderer");
-
-        final String text = getTextFromObject(commentsHeaderRenderer.getObject("countText"));
+                .getObject("commentsHeaderRenderer")
+                .getObject("countText");
 
-        return Integer.parseInt(Utils.removeNonDigitCharacters(text));
+        try {
+            return Integer.parseInt(Utils.removeNonDigitCharacters(getTextFromObject(countText)));
+        } catch (final Exception e) {
+            throw new ExtractionException("Unable to get comments count", e);
+        }
     }
 }