[Youtube] Mark members-only videos

extractor/src/main/java/org/schabi/newpipe/extractor/services/youtube/extractors/YoutubeStreamInfoItemExtractor.java

@@ -470,4 +470,18 @@ public boolean isShortFormContent() throws ParsingException {
             throw new ParsingException("Could not determine if this is short-form content", e);
         }
     }
+
+    @Nonnull
+    @Override
+    public boolean requiresMembership() throws ParsingException {
+        final JsonArray badges = videoInfo.getArray("badges");
+        for (final Object badge : badges) {
+            if (((JsonObject) badge).getObject("metadataBadgeRenderer")
+                    .getString("style", "").equals("BADGE_STYLE_TYPE_MEMBERS_ONLY")) {
+                return true;
+            }
+        }
+        return false;
+    }
+
 }

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

@@ -581,6 +581,16 @@ public boolean isShortFormContent() throws ParsingException {
         return false;
     }
 
+    /**
+     * Whether the stream is only available to channel members.
+     *
+     * @return whether the stream is only available to channel members.
+     * @throws ParsingException if there is an error in the extraction
+     */
+    public boolean requiresMembership() throws ParsingException {
+        return false;
+    }
+
     public enum Privacy {
         PUBLIC,
         UNLISTED,

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

@@ -330,6 +330,11 @@ private static void extractOptionalData(final StreamInfo streamInfo,
         } catch (final Exception e) {
             streamInfo.addError(e);
         }
+        try {
+            streamInfo.setRequiresMembership(extractor.requiresMembership());
+        } catch (final Exception e) {
+            streamInfo.addError(e);
+        }
 
         streamInfo.setRelatedItems(ExtractorHelper.getRelatedItemsOrLogError(streamInfo,
                 extractor));
@@ -381,6 +386,7 @@ private static void extractOptionalData(final StreamInfo streamInfo,
     private List<StreamSegment> streamSegments = List.of();
     private List<MetaInfo> metaInfo = List.of();
     private boolean shortFormContent = false;
+    private boolean membersOnly = false;
 
     /**
      * Preview frames, e.g. for the storyboard / seekbar thumbnail preview
@@ -727,4 +733,12 @@ public boolean isShortFormContent() {
     public void setShortFormContent(final boolean isShortFormContent) {
         this.shortFormContent = isShortFormContent;
     }
+
+    public boolean requiresMembership() {
+        return membersOnly;
+    }
+
+    public void setRequiresMembership(final boolean requiresMembership) {
+        this.membersOnly = requiresMembership;
+    }
 }

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

@@ -47,6 +47,7 @@ public class StreamInfoItem extends InfoItem {
     private List<Image> uploaderAvatars = List.of();
     private boolean uploaderVerified = false;
     private boolean shortFormContent = false;
+    private boolean requiresMembership = false;
 
     public StreamInfoItem(final int serviceId,
                           final String url,
@@ -143,6 +144,14 @@ public void setShortFormContent(final boolean shortFormContent) {
         this.shortFormContent = shortFormContent;
     }
 
+    public boolean requiresMembership() {
+        return requiresMembership;
+    }
+
+    public void setRequiresMembership(final boolean requiresMembership) {
+        this.requiresMembership = requiresMembership;
+    }
+
     @Override
     public String toString() {
         return "StreamInfoItem{"

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

@@ -146,4 +146,14 @@ default String getShortDescription() throws ParsingException {
     default boolean isShortFormContent() throws ParsingException {
         return false;
     }
+
+    /**
+     * Whether the stream is only available to channel members.
+     *
+     * @return whether the stream is only available to channel members.
+     * @throws ParsingException if there is an error in the extraction
+     */
+    default boolean requiresMembership() throws ParsingException {
+        return false;
+    }
 }

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

@@ -103,6 +103,11 @@ public StreamInfoItem extract(final StreamInfoItemExtractor extractor) throws Pa
         } catch (final Exception e) {
             addError(e);
         }
+        try {
+            resultItem.setRequiresMembership(extractor.requiresMembership());
+        } catch (final Exception e) {
+            addError(e);
+        }
 
         return resultItem;
     }