feat(youtube/comments): support creator replies

TeamNewPipe · Oct 9, 2023 · 34b05a0 · 34b05a0
1 parent 0821f09
commit 34b05a0
Show file tree

Hide file tree

Showing 5 changed files with 79 additions and 0 deletions.
diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItem.java b/extractor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItem.java
@@ -31,6 +31,7 @@ public class CommentsInfoItem extends InfoItem {
     @Nullable
     private Page replies;
     private boolean isChannelOwner;
+    private boolean creatorReply;
 
     public static final int NO_LIKE_COUNT = -1;
     public static final int NO_STREAM_POSITION = -1;
@@ -182,4 +183,13 @@ public boolean isChannelOwner() {
         return isChannelOwner;
     }
 
+
+    public void setCreatorReply(final boolean creatorReply) {
+        this.creatorReply = creatorReply;
+    }
+
+    public boolean hasCreatorReply() {
+        return creatorReply;
+    }
+
 }
diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItemExtractor.java b/extractor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItemExtractor.java
@@ -141,4 +141,11 @@ default Page getReplies() throws ParsingException {
     default boolean isChannelOwner() throws ParsingException {
         return false;
     }
+
+    /**
+     * Whether the comment was replied to by the creator.
+     */
+    default boolean hasCreatorReply() throws ParsingException {
+        return false;
+    }
 }
diff --git a/...actor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItemsCollector.java b/...actor/src/main/java/org/schabi/newpipe/extractor/comments/CommentsInfoItemsCollector.java
@@ -108,6 +108,13 @@ public CommentsInfoItem extract(final CommentsInfoItemExtractor extractor)
         }
 
 
+        try {
+            resultItem.setCreatorReply(extractor.hasCreatorReply());
+        } catch (final Exception e) {
+            addError(e);
+        }
+
+
         return resultItem;
     }
 

diff --git a/...chabi/newpipe/extractor/services/youtube/extractors/YoutubeCommentsInfoItemExtractor.java b/...chabi/newpipe/extractor/services/youtube/extractors/YoutubeCommentsInfoItemExtractor.java
@@ -283,4 +283,16 @@ public boolean isChannelOwner() throws ParsingException {
         return getCommentRenderer().getBoolean("authorIsChannelOwner");
     }
 
+
+    @Override
+    public boolean hasCreatorReply() throws ParsingException {
+        try {
+            final JsonObject commentRepliesRenderer = JsonUtils.getObject(json,
+                    "replies.commentRepliesRenderer");
+            return commentRepliesRenderer.has("viewRepliesCreatorThumbnail");
+        } catch (final Exception e) {
+            return false;
+        }
+    }
+
 }
diff --git a/...test/java/org/schabi/newpipe/extractor/services/youtube/YoutubeCommentsExtractorTest.java b/...test/java/org/schabi/newpipe/extractor/services/youtube/YoutubeCommentsExtractorTest.java
@@ -395,6 +395,49 @@ void testGetCommentsAllData() throws IOException, ExtractionException {
     }
 
 
+    public static class CreatorReply {
+        private final static String url = "https://www.youtube.com/watch?v=bem4adjGKjE";
+        private static YoutubeCommentsExtractor extractor;
+
+        @BeforeAll
+        public static void setUp() throws Exception {
+            YoutubeTestsUtils.ensureStateless();
+            NewPipe.init(DownloaderFactory.getDownloader(RESOURCE_PATH + "creatorReply"));
+            extractor = (YoutubeCommentsExtractor) YouTube
+                    .getCommentsExtractor(url);
+            extractor.fetchPage();
+        }
+
+        @Test
+        void testGetCommentsAllData() throws IOException, ExtractionException {
+            final InfoItemsPage<CommentsInfoItem> comments = extractor.getInitialPage();
+
+            DefaultTests.defaultTestListOfItems(YouTube, comments.getItems(), comments.getErrors());
+
+            boolean creatorReply = false;
+
+            for (final CommentsInfoItem c : comments.getItems()) {
+                assertFalse(Utils.isBlank(c.getUploaderUrl()));
+                assertFalse(Utils.isBlank(c.getUploaderName()));
+                YoutubeTestsUtils.testImages(c.getUploaderAvatars());
+                assertFalse(Utils.isBlank(c.getCommentId()));
+                assertFalse(Utils.isBlank(c.getName()));
+                assertFalse(Utils.isBlank(c.getTextualUploadDate()));
+                assertNotNull(c.getUploadDate());
+                YoutubeTestsUtils.testImages(c.getThumbnails());
+                assertFalse(Utils.isBlank(c.getUrl()));
+                assertTrue(c.getLikeCount() >= 0);
+                assertFalse(Utils.isBlank(c.getCommentText().getContent()));
+                if (c.hasCreatorReply()) {
+                    creatorReply = true;
+                }
+            }
+            assertTrue(creatorReply, "No comments was replied to by creator");
+
+        }
+    }
+
+
     public static class FormattingTest {
 
         private final static String url = "https://www.youtube.com/watch?v=zYpyS2HaZHM";