[Youtube] Mark members-only videos

YouTube inserts members-only videos (i.e., videos that require channel
membership to watch) into the Videos tab. Because the extractor unable
to distinguish between them and "normal" videos, they may appear in
subscriptions feeds.
This enables the extractor to check if videos require membership,
allowing clients to filter them.

Ref: TeamNewPipe/NewPipe#12040
Ref: TeamNewPipe/NewPipe#12011

Author: FineFindus

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;
     }