101 lines
4.0 KiB
Java
101 lines
4.0 KiB
Java
package org.schabi.newpipe.extractor.comments;
|
|
|
|
import org.schabi.newpipe.extractor.ListExtractor.InfoItemsPage;
|
|
import org.schabi.newpipe.extractor.ListInfo;
|
|
import org.schabi.newpipe.extractor.NewPipe;
|
|
import org.schabi.newpipe.extractor.Page;
|
|
import org.schabi.newpipe.extractor.StreamingService;
|
|
import org.schabi.newpipe.extractor.exceptions.ExtractionException;
|
|
import org.schabi.newpipe.extractor.linkhandler.ListLinkHandler;
|
|
import org.schabi.newpipe.extractor.utils.ExtractorHelper;
|
|
|
|
import java.io.IOException;
|
|
|
|
public class CommentsInfo extends ListInfo<CommentsInfoItem> {
|
|
|
|
private CommentsInfo(
|
|
final int serviceId,
|
|
final ListLinkHandler listUrlIdHandler,
|
|
final String name) {
|
|
super(serviceId, listUrlIdHandler, name);
|
|
}
|
|
|
|
public static CommentsInfo getInfo(final String url) throws IOException, ExtractionException {
|
|
return getInfo(NewPipe.getServiceByUrl(url), url);
|
|
}
|
|
|
|
public static CommentsInfo getInfo(final StreamingService serviceByUrl, final String url)
|
|
throws ExtractionException, IOException {
|
|
return getInfo(serviceByUrl.getCommentsExtractor(url));
|
|
}
|
|
|
|
public static CommentsInfo getInfo(final CommentsExtractor commentsExtractor)
|
|
throws IOException, ExtractionException {
|
|
// for services which do not have a comments extractor
|
|
if (commentsExtractor == null) {
|
|
return null;
|
|
}
|
|
|
|
commentsExtractor.fetchPage();
|
|
|
|
final String name = commentsExtractor.getName();
|
|
final int serviceId = commentsExtractor.getServiceId();
|
|
final ListLinkHandler listUrlIdHandler = commentsExtractor.getLinkHandler();
|
|
|
|
final CommentsInfo commentsInfo = new CommentsInfo(serviceId, listUrlIdHandler, name);
|
|
commentsInfo.setCommentsExtractor(commentsExtractor);
|
|
final InfoItemsPage<CommentsInfoItem> initialCommentsPage =
|
|
ExtractorHelper.getItemsPageOrLogError(commentsInfo, commentsExtractor);
|
|
commentsInfo.setCommentsDisabled(commentsExtractor.isCommentsDisabled());
|
|
commentsInfo.setRelatedItems(initialCommentsPage.getItems());
|
|
commentsInfo.setNextPage(initialCommentsPage.getNextPage());
|
|
|
|
return commentsInfo;
|
|
}
|
|
|
|
public static InfoItemsPage<CommentsInfoItem> getMoreItems(
|
|
final CommentsInfo commentsInfo,
|
|
final Page page) throws ExtractionException, IOException {
|
|
return getMoreItems(NewPipe.getService(commentsInfo.getServiceId()), commentsInfo, page);
|
|
}
|
|
|
|
public static InfoItemsPage<CommentsInfoItem> getMoreItems(
|
|
final StreamingService service,
|
|
final CommentsInfo commentsInfo,
|
|
final Page page) throws IOException, ExtractionException {
|
|
if (null == commentsInfo.getCommentsExtractor()) {
|
|
commentsInfo.setCommentsExtractor(service.getCommentsExtractor(commentsInfo.getUrl()));
|
|
commentsInfo.getCommentsExtractor().fetchPage();
|
|
}
|
|
return commentsInfo.getCommentsExtractor().getPage(page);
|
|
}
|
|
|
|
private transient CommentsExtractor commentsExtractor;
|
|
private boolean commentsDisabled = false;
|
|
|
|
public CommentsExtractor getCommentsExtractor() {
|
|
return commentsExtractor;
|
|
}
|
|
|
|
public void setCommentsExtractor(final CommentsExtractor commentsExtractor) {
|
|
this.commentsExtractor = commentsExtractor;
|
|
}
|
|
|
|
/**
|
|
* @apiNote Warning: This method is experimental and may get removed in a future release.
|
|
* @return <code>true</code> if the comments are disabled otherwise <code>false</code> (default)
|
|
* @see CommentsExtractor#isCommentsDisabled()
|
|
*/
|
|
public boolean isCommentsDisabled() {
|
|
return commentsDisabled;
|
|
}
|
|
|
|
/**
|
|
* @apiNote Warning: This method is experimental and may get removed in a future release.
|
|
* @param commentsDisabled <code>true</code> if the comments are disabled otherwise <code>false</code>
|
|
*/
|
|
public void setCommentsDisabled(final boolean commentsDisabled) {
|
|
this.commentsDisabled = commentsDisabled;
|
|
}
|
|
}
|