extract whether the comment is pinned

This commit is contained in:
bopol 2021-01-22 23:59:22 +01:00
parent 69f155d292
commit 66e4eb2f96
7 changed files with 74 additions and 0 deletions

View File

@ -17,6 +17,7 @@ public class CommentsInfoItem extends InfoItem {
private DateWrapper uploadDate; private DateWrapper uploadDate;
private int likeCount; private int likeCount;
private boolean heartedByUploader; private boolean heartedByUploader;
private boolean pinned;
public CommentsInfoItem(int serviceId, String url, String name) { public CommentsInfoItem(int serviceId, String url, String name) {
super(InfoType.COMMENT, serviceId, url, name); super(InfoType.COMMENT, serviceId, url, name);
@ -94,4 +95,12 @@ public class CommentsInfoItem extends InfoItem {
public boolean getHeartedByUploader() { public boolean getHeartedByUploader() {
return this.heartedByUploader; return this.heartedByUploader;
} }
public boolean getPinned() {
return pinned;
}
public void setPinned(boolean pinned) {
this.pinned = pinned;
}
} }

View File

@ -11,6 +11,7 @@ public interface CommentsInfoItemExtractor extends InfoItemExtractor {
/** /**
* Return the like count of the comment, or -1 if it's unavailable * Return the like count of the comment, or -1 if it's unavailable
*
* @see StreamExtractor#getLikeCount() * @see StreamExtractor#getLikeCount()
*/ */
int getLikeCount() throws ParsingException; int getLikeCount() throws ParsingException;
@ -22,12 +23,14 @@ public interface CommentsInfoItemExtractor extends InfoItemExtractor {
/** /**
* The upload date given by the service, unmodified * The upload date given by the service, unmodified
*
* @see StreamExtractor#getTextualUploadDate() * @see StreamExtractor#getTextualUploadDate()
*/ */
String getTextualUploadDate() throws ParsingException; String getTextualUploadDate() throws ParsingException;
/** /**
* The upload date wrapped with DateWrapper class * The upload date wrapped with DateWrapper class
*
* @see StreamExtractor#getUploadDate() * @see StreamExtractor#getUploadDate()
*/ */
@Nullable @Nullable
@ -45,4 +48,9 @@ public interface CommentsInfoItemExtractor extends InfoItemExtractor {
* Whether the comment has been hearted by the uploader * Whether the comment has been hearted by the uploader
*/ */
boolean getHeartedByUploader() throws ParsingException; boolean getHeartedByUploader() throws ParsingException;
/**
* Whether the comment is pinned
*/
boolean getPinned() throws ParsingException;
} }

View File

@ -76,6 +76,12 @@ public class CommentsInfoItemsCollector extends InfoItemsCollector<CommentsInfoI
addError(e); addError(e);
} }
try {
resultItem.setPinned(extractor.getPinned());
} catch (Exception e) {
addError(e);
}
return resultItem; return resultItem;
} }

View File

@ -93,6 +93,11 @@ public class PeertubeCommentsInfoItemExtractor implements CommentsInfoItemExtrac
return false; return false;
} }
@Override
public boolean getPinned() throws ParsingException {
return false;
}
@Override @Override
public String getUploaderName() throws ParsingException { public String getUploaderName() throws ParsingException {
return JsonUtils.getString(item, "account.name") + "@" + JsonUtils.getString(item, "account.host"); return JsonUtils.getString(item, "account.name") + "@" + JsonUtils.getString(item, "account.host");

View File

@ -44,6 +44,11 @@ public class SoundcloudCommentsInfoItemExtractor implements CommentsInfoItemExtr
return false; return false;
} }
@Override
public boolean getPinned() throws ParsingException {
return false;
}
@Override @Override
public String getUploaderUrl() { public String getUploaderUrl() {
return json.getObject("user").getString("permalink_url"); return json.getObject("user").getString("permalink_url");

View File

@ -120,6 +120,11 @@ public class YoutubeCommentsInfoItemExtractor implements CommentsInfoItemExtract
return json.has("creatorHeart"); return json.has("creatorHeart");
} }
@Override
public boolean getPinned() {
return json.has("pinnedCommentBadge");
}
@Override @Override
public String getUploaderName() throws ParsingException { public String getUploaderName() throws ParsingException {
try { try {

View File

@ -189,4 +189,40 @@ public class YoutubeCommentsExtractorTest {
} }
} }
public static class Pinned {
private final static String url = "https://www.youtube.com/watch?v=bjFtFMilb34";
private static YoutubeCommentsExtractor extractor;
@BeforeClass
public static void setUp() throws Exception {
NewPipe.init(DownloaderTestImpl.getInstance());
extractor = (YoutubeCommentsExtractor) YouTube
.getCommentsExtractor(url);
extractor.fetchPage();
}
@Test
public void testGetCommentsAllData() throws IOException, ExtractionException {
final InfoItemsPage<CommentsInfoItem> comments = extractor.getInitialPage();
DefaultTests.defaultTestListOfItems(YouTube, comments.getItems(), comments.getErrors());
for (CommentsInfoItem c : comments.getItems()) {
assertFalse(Utils.isBlank(c.getUploaderUrl()));
assertFalse(Utils.isBlank(c.getUploaderName()));
assertFalse(Utils.isBlank(c.getUploaderAvatarUrl()));
assertFalse(Utils.isBlank(c.getCommentId()));
assertFalse(Utils.isBlank(c.getName()));
assertFalse(Utils.isBlank(c.getTextualUploadDate()));
assertNotNull(c.getUploadDate());
assertFalse(Utils.isBlank(c.getThumbnailUrl()));
assertFalse(Utils.isBlank(c.getUrl()));
assertFalse(c.getLikeCount() < 0);
assertFalse(Utils.isBlank(c.getCommentText()));
}
assertTrue("First comment isn't pinned", comments.getItems().get(0).getPinned());
}
}
} }