From bccb1a491fa364d7c8cbea856694180116cdebc8 Mon Sep 17 00:00:00 2001
From: Koitharu <nvasya95@gmail.com>
Date: Fri, 8 Jul 2022 11:38:56 +0300
Subject: [PATCH] [YaoiLib] New parser

---
 .../koitharu/kotatsu/parsers/MangaLoaderContext.kt |  5 ++++-
 .../parsers/site/{ => rulib}/HentaiLibParser.kt    |  2 +-
 .../parsers/site/{ => rulib}/MangaLibParser.kt     |  2 +-
 .../kotatsu/parsers/site/rulib/YaoiLibParser.kt    | 14 ++++++++++++++
 4 files changed, 20 insertions(+), 3 deletions(-)
 rename src/main/kotlin/org/koitharu/kotatsu/parsers/site/{ => rulib}/HentaiLibParser.kt (91%)
 rename src/main/kotlin/org/koitharu/kotatsu/parsers/site/{ => rulib}/MangaLibParser.kt (99%)
 create mode 100644 src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/YaoiLibParser.kt

diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
index bdd704f1..61e582d1 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/MangaLoaderContext.kt
@@ -142,7 +142,10 @@ abstract class MangaLoaderContext {
 
 	private fun Response.ensureSuccess() = apply {
 		val exception: Exception? = when (code) { // Catch some error codes, not all
-			in 500..599 -> HttpStatusException(message, code, request.url.toString())
+			404,
+			in 500..599,
+			-> HttpStatusException(message, code, request.url.toString())
+
 			else -> null
 		}
 		if (exception != null) {
diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/HentaiLibParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/HentaiLibParser.kt
similarity index 91%
rename from src/main/kotlin/org/koitharu/kotatsu/parsers/site/HentaiLibParser.kt
rename to src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/HentaiLibParser.kt
index fee4e450..84febec9 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/HentaiLibParser.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/HentaiLibParser.kt
@@ -1,4 +1,4 @@
-package org.koitharu.kotatsu.parsers.site
+package org.koitharu.kotatsu.parsers.site.rulib
 
 import org.jsoup.nodes.Document
 import org.koitharu.kotatsu.parsers.MangaLoaderContext
diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/MangaLibParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/MangaLibParser.kt
similarity index 99%
rename from src/main/kotlin/org/koitharu/kotatsu/parsers/site/MangaLibParser.kt
rename to src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/MangaLibParser.kt
index 89d2926f..9da969e2 100644
--- a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/MangaLibParser.kt
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/MangaLibParser.kt
@@ -1,4 +1,4 @@
-package org.koitharu.kotatsu.parsers.site
+package org.koitharu.kotatsu.parsers.site.rulib
 
 import androidx.collection.ArraySet
 import org.json.JSONArray
diff --git a/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/YaoiLibParser.kt b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/YaoiLibParser.kt
new file mode 100644
index 00000000..e7f3b08e
--- /dev/null
+++ b/src/main/kotlin/org/koitharu/kotatsu/parsers/site/rulib/YaoiLibParser.kt
@@ -0,0 +1,14 @@
+package org.koitharu.kotatsu.parsers.site.rulib
+
+import org.jsoup.nodes.Document
+import org.koitharu.kotatsu.parsers.MangaLoaderContext
+import org.koitharu.kotatsu.parsers.MangaSourceParser
+import org.koitharu.kotatsu.parsers.config.ConfigKey
+import org.koitharu.kotatsu.parsers.model.MangaSource
+
+@MangaSourceParser("YAOILIB", "YaoiLib", "ru")
+internal class YaoiLibParser(context: MangaLoaderContext) : MangaLibParser(context, MangaSource.YAOILIB) {
+
+	override val configKeyDomain = ConfigKey.Domain("yaoilib.me", null)
+	override fun isNsfw(doc: Document) = true
+}
\ No newline at end of file