From 1a15c0e750dc95fa0a92cc1c988e781e66631fa7 Mon Sep 17 00:00:00 2001
From: B0pol <bopol@e.email>
Date: Thu, 23 Jan 2020 04:42:54 +0100
Subject: [PATCH] agelimit now returns 18 if the video is marked as nsfw, 0
 otherwise

+ created getBoolean method in JsonUtils.java
---
 .../peertube/extractors/PeertubeStreamExtractor.java   |  8 ++++++--
 .../org/schabi/newpipe/extractor/utils/JsonUtils.java  | 10 ++++++++++
 .../peertube/PeertubeStreamExtractorDefaultTest.java   |  8 ++++++++
 3 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/services/peertube/extractors/PeertubeStreamExtractor.java b/extractor/src/main/java/org/schabi/newpipe/extractor/services/peertube/extractors/PeertubeStreamExtractor.java
index 0de72a94..b841d9fd 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/services/peertube/extractors/PeertubeStreamExtractor.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/services/peertube/extractors/PeertubeStreamExtractor.java
@@ -92,7 +92,12 @@ public class PeertubeStreamExtractor extends StreamExtractor {
 
     @Override
     public int getAgeLimit() throws ParsingException {
-        return NO_AGE_LIMIT;
+        boolean isNSFW = JsonUtils.getBoolean(json, "nsfw");
+        if (isNSFW) {
+            return 18;
+        } else {
+            return NO_AGE_LIMIT;
+        }
     }
 
     @Override
@@ -352,7 +357,6 @@ public class PeertubeStreamExtractor extends StreamExtractor {
         return baseUrl + "/videos/watch/" + getId();
     }
 
-    //TODO: change privacy, category, licence by getting ID, therefore we will be able to translate it
     @Override
     public String getHost() throws ParsingException {
         return JsonUtils.getString(json, "account.host");
diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/utils/JsonUtils.java b/extractor/src/main/java/org/schabi/newpipe/extractor/utils/JsonUtils.java
index e5d7bb62..ebd251d7 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/utils/JsonUtils.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/utils/JsonUtils.java
@@ -37,6 +37,16 @@ public class JsonUtils {
             throw new ParsingException("Unable to get " + path);
         }
     }
+
+    @Nonnull
+    public static Boolean getBoolean(@Nonnull JsonObject object, @Nonnull String path) throws ParsingException{
+        Object value = getValue(object, path);
+        if(value instanceof Boolean) {
+            return (Boolean) value;
+        }else {
+            throw new ParsingException("Unable to get " + path);
+        }
+    }
     
     @Nonnull
     public static Number getNumber(@Nonnull JsonObject object, @Nonnull String path) throws ParsingException{
diff --git a/extractor/src/test/java/org/schabi/newpipe/extractor/services/peertube/PeertubeStreamExtractorDefaultTest.java b/extractor/src/test/java/org/schabi/newpipe/extractor/services/peertube/PeertubeStreamExtractorDefaultTest.java
index 2a4378a2..4166b192 100644
--- a/extractor/src/test/java/org/schabi/newpipe/extractor/services/peertube/PeertubeStreamExtractorDefaultTest.java
+++ b/extractor/src/test/java/org/schabi/newpipe/extractor/services/peertube/PeertubeStreamExtractorDefaultTest.java
@@ -139,4 +139,12 @@ public class PeertubeStreamExtractorDefaultTest {
     public void testGetSubtitlesList() throws IOException, ExtractionException {
         assertFalse(extractor.getSubtitlesDefault().isEmpty());
     }
+
+    @Test
+    public void testGetAgeLimit() throws ExtractionException, IOException {
+        assertEquals(0, extractor.getAgeLimit());
+        PeertubeStreamExtractor ageLimit = (PeertubeStreamExtractor) PeerTube.getStreamExtractor("https://peertube.co.uk/videos/watch/6762bb04-cad5-407b-81ee-c18eac4715a7");
+        ageLimit.fetchPage();
+        assertEquals(18, ageLimit.getAgeLimit());
+    }
 }