From 8db34eb3dd40feb00421cab4207b0ffbd14a083a Mon Sep 17 00:00:00 2001
From: arkon <eugcheung94@gmail.com>
Date: Mon, 10 Aug 2020 11:54:31 -0400
Subject: [PATCH] Allow annotating SourceFactory with @Nsfw to block all
 sources within it

---
 .../extension/util/ExtensionLoader.kt         | 24 +++++++++++++++----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/eu/kanade/tachiyomi/extension/util/ExtensionLoader.kt b/app/src/main/java/eu/kanade/tachiyomi/extension/util/ExtensionLoader.kt
index d7cdec5267..a2f51c7314 100644
--- a/app/src/main/java/eu/kanade/tachiyomi/extension/util/ExtensionLoader.kt
+++ b/app/src/main/java/eu/kanade/tachiyomi/extension/util/ExtensionLoader.kt
@@ -152,7 +152,13 @@ internal object ExtensionLoader {
                 try {
                     when (val obj = Class.forName(it, false, classLoader).newInstance()) {
                         is Source -> listOf(obj)
-                        is SourceFactory -> obj.createSources()
+                        is SourceFactory -> {
+                            if (isSourceNsfw(obj)) {
+                                emptyList()
+                            } else {
+                                obj.createSources()
+                            }
+                        }
                         else -> throw Exception("Unknown source class type! ${obj.javaClass}")
                     }
                 } catch (e: Throwable) {
@@ -160,7 +166,7 @@ internal object ExtensionLoader {
                     return LoadResult.Error(e)
                 }
             }
-            .filter { allowNsfwSource == PreferenceValues.NsfwAllowance.ALLOWED || !isSourceNsfw(it) }
+            .filter { !isSourceNsfw(it) }
 
         val langs = sources.filterIsInstance<CatalogueSource>()
             .map { it.lang }
@@ -202,11 +208,19 @@ internal object ExtensionLoader {
     }
 
     /**
-     * Checks whether a source is annotated with @Nsfw.
+     * Checks whether a Source or SourceFactory is annotated with @Nsfw.
      */
-    private fun isSourceNsfw(source: Source): Boolean {
+    private fun isSourceNsfw(clazz: Any): Boolean {
+        if (allowNsfwSource == PreferenceValues.NsfwAllowance.ALLOWED) {
+            return false
+        }
+
+        if (clazz !is Source && clazz !is SourceFactory) {
+            return false
+        }
+
         // Annotations are proxied, hence this janky way of checking for them
-        return source.javaClass.annotations
+        return clazz.javaClass.annotations
             .flatMap { it.javaClass.interfaces.map { it.simpleName } }
             .firstOrNull { it == Nsfw::class.java.simpleName } != null
     }