CypherpunkSamurai/NewPipeExtractor
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Don't cache comments count and return early on page fetch if no token.

Browse source
This commit is contained in:
Kavin 2022-12-08 12:10:33 +00:00
parent 2974dfaa48
commit 98a90fd9c8
No known key found for this signature in database
GPG key ID: 49451E4482CC5BCD
1 changed files with 21 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeCommentsExtractor.java
View file

@ -33,12 +33,7 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
/**
* Whether comments are disabled on video.
*/
private boolean commentsDisabled = true;
/**
* The total number of comments on video.
*/
private int commentsCount = (int) ITEM_COUNT_UNKNOWN;
private boolean commentsDisabled;
/**
* The second ajax <b>/next</b> response.
@ -268,6 +263,10 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
final String initialToken =
findInitialCommentsToken(getJsonPostResponse("next", body, localization));
if (initialToken == null) {
return;
}
// @formatter:off
final byte[] ajaxBody = JsonWriter.string(
prepareDesktopJsonBuilder(localization, getExtractorContentCountry())
@ -289,23 +288,23 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
public int getCommentsCount() throws ExtractionException {
assertPageFetched();
if (commentsCount == ITEM_COUNT_UNKNOWN) {
final JsonObject countText = ajaxJson
.getArray("onResponseReceivedEndpoints").getObject(0)
.getObject("reloadContinuationItemsCommand")
.getArray("continuationItems").getObject(0)
.getObject("commentsHeaderRenderer")
.getObject("countText");
try {
commentsCount = Integer.parseInt(
Utils.removeNonDigitCharacters(getTextFromObject(countText))
);
} catch (final Exception e) {
throw new ExtractionException("Unable to get comments count", e);
}
if (commentsDisabled) {
return -1;
}
return commentsCount;
final JsonObject countText = ajaxJson
.getArray("onResponseReceivedEndpoints").getObject(0)
.getObject("reloadContinuationItemsCommand")
.getArray("continuationItems").getObject(0)
.getObject("commentsHeaderRenderer")
.getObject("countText");
try {
return Integer.parseInt(
Utils.removeNonDigitCharacters(getTextFromObject(countText))
);
} catch (final Exception e) {
throw new ExtractionException("Unable to get comments count", e);
}
}
}