added extractor components for ReturnYouTubeDislike

Author: polymorphicshade

extractor/src/main/java/org/schabi/newpipe/extractor/returnyoutubedislike/ReturnYouTubeDislikeApiSettings.java

@@ -0,0 +1,9 @@
+package org.schabi.newpipe.extractor.returnyoutubedislike;
+
+public class ReturnYouTubeDislikeApiSettings {
+    public String apiUrl;
+
+    // TODO: add more if needed
+    //
+    //
+}

extractor/src/main/java/org/schabi/newpipe/extractor/returnyoutubedislike/ReturnYouTubeDislikeExtractorHelper.java

@@ -0,0 +1,53 @@
+package org.schabi.newpipe.extractor.returnyoutubedislike;
+
+import com.grack.nanojson.JsonObject;
+import com.grack.nanojson.JsonParser;
+import com.grack.nanojson.JsonParserException;
+
+import org.schabi.newpipe.extractor.NewPipe;
+import org.schabi.newpipe.extractor.exceptions.ReCaptchaException;
+import org.schabi.newpipe.extractor.stream.StreamInfo;
+
+import java.io.IOException;
+
+public final class ReturnYouTubeDislikeExtractorHelper {
+
+    private ReturnYouTubeDislikeExtractorHelper() {
+    }
+
+    @SuppressWarnings("CheckStyle")
+    public static ReturnYouTubeDislikeInfo getInfo(
+            final StreamInfo streamInfo,
+            final ReturnYouTubeDislikeApiSettings apiSettings) {
+        if (!streamInfo.getUrl().startsWith("https://www.youtube.com")
+                || apiSettings.apiUrl == null
+                || apiSettings.apiUrl.isEmpty()) {
+            return null;
+        }
+
+        final String url = apiSettings.apiUrl + "votes?videoId=" + streamInfo.getId();
+
+        JsonObject response = null;
+
+        try {
+            final String responseBody =
+                    NewPipe.getDownloader().get(url).responseBody();
+
+            response = JsonParser.object().from(responseBody);
+        } catch (ReCaptchaException | IOException | JsonParserException e) {
+            // ignored
+        }
+
+        if (response == null) {
+            return null;
+        }
+
+        final int likes = response.getInt("likes", 0);
+        final int dislikes = response.getInt("dislikes", 0);
+        final double rating = response.getDouble("rating", 0);
+        final int viewCount = response.getInt("viewCount", 0);
+        final boolean deleted = response.getBoolean("deleted", false);
+
+        return new ReturnYouTubeDislikeInfo(likes, dislikes, rating, viewCount, deleted);
+    }
+}

extractor/src/main/java/org/schabi/newpipe/extractor/returnyoutubedislike/ReturnYouTubeDislikeInfo.java

@@ -0,0 +1,20 @@
+package org.schabi.newpipe.extractor.returnyoutubedislike;
+
+import java.io.Serializable;
+
+public class ReturnYouTubeDislikeInfo implements Serializable {
+    public long likes;
+    public long dislikes;
+    public double rating;
+    public long viewCount;
+    public boolean deleted;
+
+    public ReturnYouTubeDislikeInfo(final long likes, final long dislikes, final double rating,
+                                    final long viewCount, final boolean deleted) {
+        this.likes = likes;
+        this.dislikes = dislikes;
+        this.rating = rating;
+        this.viewCount = viewCount;
+        this.deleted = deleted;
+    }
+}

extractor/src/main/java/org/schabi/newpipe/extractor/stream/StreamInfo.java

@@ -33,6 +33,9 @@
 import org.schabi.newpipe.extractor.sponsorblock.SponsorBlockApiSettings;
 import org.schabi.newpipe.extractor.sponsorblock.SponsorBlockExtractorHelper;
 import org.schabi.newpipe.extractor.sponsorblock.SponsorBlockSegment;
+import org.schabi.newpipe.extractor.returnyoutubedislike.ReturnYouTubeDislikeApiSettings;
+import org.schabi.newpipe.extractor.returnyoutubedislike.ReturnYouTubeDislikeExtractorHelper;
+import org.schabi.newpipe.extractor.returnyoutubedislike.ReturnYouTubeDislikeInfo;
 import org.schabi.newpipe.extractor.utils.ExtractorHelper;
 
 import java.io.IOException;
@@ -71,22 +74,25 @@ public StreamInfo(final int serviceId,
 
     public static StreamInfo getInfo(
             final String url,
-            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings)
+            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings,
+            @Nullable final ReturnYouTubeDislikeApiSettings returnYouTubeDislikeApiSettings)
             throws IOException, ExtractionException {
-        return getInfo(NewPipe.getServiceByUrl(url), url, sponsorBlockApiSettings);
+        return getInfo(NewPipe.getServiceByUrl(url), url, sponsorBlockApiSettings, returnYouTubeDislikeApiSettings);
     }
 
     public static StreamInfo getInfo(
             @Nonnull final StreamingService service,
             final String url,
-            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings)
+            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings,
+            @Nullable final ReturnYouTubeDislikeApiSettings returnYouTubeDislikeApiSettings)
             throws IOException, ExtractionException {
-        return getInfo(service.getStreamExtractor(url), sponsorBlockApiSettings);
+        return getInfo(service.getStreamExtractor(url), sponsorBlockApiSettings, returnYouTubeDislikeApiSettings);
     }
 
     public static StreamInfo getInfo(
             @Nonnull final StreamExtractor extractor,
-            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings)
+            @Nullable final SponsorBlockApiSettings sponsorBlockApiSettings,
+            @Nullable final ReturnYouTubeDislikeApiSettings returnYouTubeDislikeApiSettings)
             throws ExtractionException, IOException {
         extractor.fetchPage();
         final StreamInfo streamInfo;
@@ -101,6 +107,13 @@ public static StreamInfo getInfo(
                 streamInfo.setSponsorBlockSegments(sponsorBlockSegments);
             }
 
+            if (returnYouTubeDislikeApiSettings != null) {
+                final ReturnYouTubeDislikeInfo rydInfo  =
+                        ReturnYouTubeDislikeExtractorHelper.getInfo(
+                                streamInfo, returnYouTubeDislikeApiSettings);
+                streamInfo.setReturnYouTubeDislikeInfo(rydInfo);
+            }
+
             return streamInfo;
 
         } catch (final ExtractionException e) {
@@ -403,6 +416,7 @@ private static void extractOptionalData(final StreamInfo streamInfo,
     private List<MetaInfo> metaInfo = List.of();
     private boolean shortFormContent = false;
     private List<SponsorBlockSegment> sponsorBlockSegments = new ArrayList<>();
+    @Nullable private ReturnYouTubeDislikeInfo rydInfo;
 
     /**
      * Preview frames, e.g. for the storyboard / seekbar thumbnail preview
@@ -780,4 +794,13 @@ public void removeSponsorBlockSegment(final String uuid) {
             removeSponsorBlockSegment(target);
         }
     }
+
+    @Nullable
+    public ReturnYouTubeDislikeInfo getRydInfo() {
+        return rydInfo;
+    }
+
+    public void setReturnYouTubeDislikeInfo(final @Nullable ReturnYouTubeDislikeInfo rydInfo) {
+        this.rydInfo = rydInfo;
+    }
 }