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

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

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,7 +288,10 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
public int getCommentsCount() throws ExtractionException {
assertPageFetched();
if (commentsCount == ITEM_COUNT_UNKNOWN) {
if (commentsDisabled) {
return -1;
}
final JsonObject countText = ajaxJson
.getArray("onResponseReceivedEndpoints").getObject(0)
.getObject("reloadContinuationItemsCommand")
@ -298,14 +300,11 @@ public class YoutubeCommentsExtractor extends CommentsExtractor {
.getObject("countText");
try {
commentsCount = Integer.parseInt(
return Integer.parseInt(
Utils.removeNonDigitCharacters(getTextFromObject(countText))
);
} catch (final Exception e) {
throw new ExtractionException("Unable to get comments count", e);
}
}
return commentsCount;
}
}