isVerified and isOfficialArtist (#22)

* Added isVerified, isOfficialArtist and fixed broken Jest Tests * Added new fields to ReadMe
FreeTubeApp · Jun 25, 2021 · e84c9e8 · e84c9e8
1 parent d7b19dd
commit e84c9e8
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 16 deletions.
diff --git a/README.md b/README.md
@@ -73,6 +73,8 @@ The data is returned as a list of objects (seen below).
     isOwner: Boolean, // If the video channel made the comment
     isHearted: Boolean, // If the video channel hearted the comment
     isPinned: Boolean, // If the video channel pinned the comment
+    isVerified: Boolean,
+    isOfficialArtist: Boolean,
     hasOwnerReplied: Boolean, // If the video channel replied to the comment
     replyToken: String // The continuation token needed for getCommentReplies()
   }],
@@ -120,6 +122,8 @@ The data is returned as a list of objects (seen below).
     isOwner: Boolean, // If the video channel made the comment
     isHearted: Boolean, // If the video channel hearted the comment
     isPinned: false,
+    isVerified: Boolean,
+    isOfficialArtist: Boolean,
     hasOwnerReplied: false,
     replyToken: null
   }],

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
-  "name": "yt-comment-scraper",
-  "version": "5.1.0",
+  "name": "@freetube/yt-comment-scraper",
+  "version": "5.2.1",
   "description": "Scrapes the comments of any YouTube video without YouTube API access. Uses the default YouTube Ajax calls to get the comment data.",
   "main": "index.js",
   "scripts": {

diff --git a/src/htmlParser.js b/src/htmlParser.js
@@ -24,6 +24,9 @@ class HtmlParser {
       const heartBadge = comment.actionButtons.commentActionButtonsRenderer.creatorHeart
       const isOwner = comment.authorIsChannelOwner
       const isPinned = comment.pinnedCommentBadge ? true : false
+      const isVerified = ('authorCommentBadge' in comment && comment.authorCommentBadge.authorCommentBadgeRenderer.icon.iconType === "CHECK_CIRCLE_THICK")
+      const isOfficialArtist = ('authorCommentBadge' in comment && comment.authorCommentBadge.authorCommentBadgeRenderer.icon.iconType === "OFFICIAL_ARTIST_BADGE")
+
 
       if (typeof heartBadge !== 'undefined') {
         isHearted = heartBadge.creatorHeartRenderer.isHearted
@@ -53,7 +56,9 @@ class HtmlParser {
         hasOwnerReplied: false,
         time: publishedText,
         edited: isEdited,
-        replyToken: null
+        replyToken: null,
+        isVerified: isVerified,
+        isOfficialArtist: isOfficialArtist
       }
 
       if (comment.replyCount > 0) {

diff --git a/test/scraper.test.js b/test/scraper.test.js
@@ -6,21 +6,21 @@ describe('Standalone Mode: Comment Testing', () => {
 
     test('Scrape top video comments of first page', () => {
         const parameters = {videoId: 'oBLQmE-nG60', setCookie: true, sortByNewest: false};
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             expect(data.comments).toHaveLength(20);
         });
     });
 
     test('Scrape newest video comments of first page', () => {
         const parameters = {videoId: 'oBLQmE-nG60', setCookie: true, sortByNewest: true};
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             expect(data.comments).toHaveLength(20);
         });
     });
 
     test('Scrape newest video comments second page', () => {
         let parameters = {videoId: 'oBLQmE-nG60', setCookie: true, sortByNewest: true, continuation: null};
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             parameters["continuation"] = data.continuation;
             ytcm.getComments(parameters).then((data) => {
                 expect(data.comments).not.toHaveLength(0);
@@ -31,15 +31,14 @@ describe('Standalone Mode: Comment Testing', () => {
     test('Scrape top replies of first page', () => {
         const parameters = {videoId: 'oBLQmE-nG60', setCookie: true, sortByNewest: false};
         // This test first gets comments of the video with above Id
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             for (let i = 0; i < data.comments.length; i++) {
                 // The test searches for a comment which does have at least 1 reply and then ask for the replies
                 if (data.comments[i].numReplies > 0) {
                     const replyParameters = {videoId: 'oBLQmE-nG60', replyToken: data.comments[i].replyToken, setCookie: true};
-                    ytcm.getCommentReplies(replyParameters).then((replyData) => {
+                    return ytcm.getCommentReplies(replyParameters).then((replyData) => {
                         expect(replyData.comments).toHaveLength(10);
                     });
-                    break
                 }
             }
         });
@@ -48,26 +47,25 @@ describe('Standalone Mode: Comment Testing', () => {
     test('Scrape second batch of top replies of first page', () => {
         const parameters = {videoId: 'oBLQmE-nG60', setCookie: true, sortByNewest: false};
         // This test first gets comments of the video with above Id
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             for (let i = 0; i < data.comments.length; i++) {
                 // The test searches for a comment which does have at least 1 reply and then ask for the replies
                 if (data.comments[i].numReplies > 0) {
                     let replyParameters = {videoId: 'oBLQmE-nG60', replyToken: data.comments[i].replyToken, setCookie: true};
-                    ytcm.getCommentReplies(replyParameters).then((replyData) => {
+                    return ytcm.getCommentReplies(replyParameters).then((replyData) => {
                         replyParameters.replyToken = replyData.continuation;
-                        ytcm.getCommentReplies(replyParameters).then((replyData) => {
+                        return ytcm.getCommentReplies(replyParameters).then((replyData) => {
                             expect(replyData.comments).not.toHaveLength(0);
                         })
                     });
-                    break
                 }
             }
         });
     });
 
     test('Scrape video without comments', () => {
         const parameters = {videoId: 'Bj-3M-KqZsI', setCookie: true, sortByNewest: false};
-        ytcm.getComments(parameters).then((data) => {
+        return ytcm.getComments(parameters).then((data) => {
             expect(data.comments).toHaveLength(0);
         });
     });