From 5da79f54a072841e59bb41db6f68bccc5a6a1638 Mon Sep 17 00:00:00 2001
From: h-hyuuga <83582211+h-hyuuga@users.noreply.github.com>
Date: Mon, 10 May 2021 07:02:24 -0400
Subject: [PATCH] Fix wpmangareader filtering + Dynamically Fetch Genres
 (#6935)

---
 .../multisrc/wpmangareader/WPMangaReader.kt   | 279 +++++++++---------
 .../wpmangareader/WPMangaReaderGenerator.kt   |   2 +-
 2 files changed, 136 insertions(+), 145 deletions(-)

diff --git a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReader.kt b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReader.kt
index 32779c5e0..8743bf52f 100644
--- a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReader.kt
+++ b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReader.kt
@@ -1,13 +1,17 @@
 package eu.kanade.tachiyomi.multisrc.wpmangareader
 
+import android.util.Log
 import eu.kanade.tachiyomi.network.GET
 import eu.kanade.tachiyomi.source.model.Filter
 import eu.kanade.tachiyomi.source.model.FilterList
+import eu.kanade.tachiyomi.source.model.MangasPage
 import eu.kanade.tachiyomi.source.model.Page
 import eu.kanade.tachiyomi.source.model.SChapter
 import eu.kanade.tachiyomi.source.model.SManga
 import eu.kanade.tachiyomi.source.online.ParsedHttpSource
 import eu.kanade.tachiyomi.util.asJsoup
+import okhttp3.HttpUrl
+import okhttp3.HttpUrl.Companion.toHttpUrlOrNull
 import okhttp3.OkHttpClient
 import okhttp3.Request
 import okhttp3.Response
@@ -30,45 +34,58 @@ abstract class WPMangaReader(
     override val client: OkHttpClient = network.cloudflareClient
 
     // popular
-    override fun popularMangaSelector() = ".utao .uta .imgu, .listupd .bs .bsx, .listo .bs .bsx"
+    override fun popularMangaRequest(page: Int) = searchMangaRequest(page, "", FilterList(OrderByFilter(5)))
+    override fun popularMangaParse(response: Response) = searchMangaParse(response)
 
-    override fun popularMangaRequest(page: Int) = GET("$baseUrl$mangaUrlDirectory/?page=$page&order=popular", headers)
+    override fun popularMangaFromElement(element: Element) = throw UnsupportedOperationException("Not used")
+    override fun popularMangaSelector() = throw UnsupportedOperationException("Not used")
+    override fun popularMangaNextPageSelector() = throw UnsupportedOperationException("Not used")
 
-    override fun popularMangaFromElement(element: Element) = SManga.create().apply {
+
+    // latest
+    override fun latestUpdatesRequest(page: Int) = searchMangaRequest(page, "", FilterList(OrderByFilter(3)))
+    override fun latestUpdatesParse(response: Response) = searchMangaParse(response)
+
+    override fun latestUpdatesSelector() = throw UnsupportedOperationException("Not used")
+    override fun latestUpdatesFromElement(element: Element) = throw UnsupportedOperationException("Not used")
+    override fun latestUpdatesNextPageSelector() = throw UnsupportedOperationException("Not used")
+
+    // search
+    override fun searchMangaSelector() = ".utao .uta .imgu, .listupd .bs .bsx, .listo .bs .bsx"
+
+    override fun searchMangaRequest(page: Int, query: String, filters: FilterList): Request {
+        val url = "$baseUrl".toHttpUrlOrNull()!!.newBuilder()
+        if (query.isNotEmpty()) {
+            url.addPathSegments("page/$page").addQueryParameter("s", query)
+        } else {
+            url.addPathSegment(mangaUrlDirectory.substring(1)).addQueryParameter("page", "$page")
+            filters.forEach { filter ->
+                when (filter) {
+                    is UrlEncoded -> filter.encode(url)
+                }
+            }
+        }
+        return GET("$url")
+    }
+
+    override fun searchMangaParse(response: Response): MangasPage {
+        if (genrelist == null)
+            genrelist = parseGenres(response.asJsoup(response.peekBody(Long.MAX_VALUE).string()))
+        return super.searchMangaParse(response)
+    }
+
+    private fun parseGenres(document: Document): List<LabeledValue>? {
+        return document.selectFirst("ul.c4")?.select("li")?.map { li ->
+            LabeledValue(li.selectFirst("label").text(), li.selectFirst("input[type=checkbox]").`val`())
+        }
+    }
+
+    override fun searchMangaFromElement(element: Element) = SManga.create().apply {
         thumbnail_url = element.select("img").attr("abs:src")
         title = element.select("a").attr("title")
         setUrlWithoutDomain(element.select("a").attr("href"))
     }
-
-    override fun popularMangaNextPageSelector() = "div.pagination .next, div.hpage .r"
-
-    // latest
-    override fun latestUpdatesSelector() = popularMangaSelector()
-
-    override fun latestUpdatesRequest(page: Int) = GET("$baseUrl$mangaUrlDirectory/?page=$page&order=update", headers)
-
-    override fun latestUpdatesFromElement(element: Element) = popularMangaFromElement(element)
-
-    override fun latestUpdatesNextPageSelector() = popularMangaNextPageSelector()
-
-    // search
-    override fun searchMangaSelector() = popularMangaSelector()
-
-    override fun searchMangaRequest(page: Int, query: String, filters: FilterList): Request {
-        val filters = if (filters.isEmpty()) getFilterList() else filters
-        val genre = filters.findInstance<GenreList>()?.toUriPart()
-        val order = filters.findInstance<OrderByFilter>()?.toUriPart()
-
-        return when {
-            order!!.isNotEmpty() -> GET("$baseUrl$mangaUrlDirectory/?page=$page&order=$order")
-            genre!!.isNotEmpty() -> GET("$baseUrl/genres/$genre/page/$page/?s=$query")
-            else -> GET("$baseUrl/page/$page/?s=$query")
-        }
-    }
-
-    override fun searchMangaFromElement(element: Element) = popularMangaFromElement(element)
-
-    override fun searchMangaNextPageSelector() = popularMangaNextPageSelector()
+    override fun searchMangaNextPageSelector() =  "div.pagination .next, div.hpage .r"
 
     // manga details
     override fun mangaDetailsParse(document: Document) = SManga.create().apply {
@@ -174,117 +191,91 @@ abstract class WPMangaReader(
 
     override fun imageUrlParse(document: Document): String = throw UnsupportedOperationException("Not Used")
 
-    // filters
-    override fun getFilterList() = FilterList(
-        Filter.Header("Order by filter cannot be used with others"),
-        OrderByFilter(),
-        Filter.Separator(),
-        GenreList()
-    )
-
-    private class OrderByFilter : UriPartFilter(
-        "Order By",
-        arrayOf(
-            Pair("", "<select>"),
-            Pair("title", "A-Z"),
-            Pair("update", "Latest Update"),
-            Pair("latest", "Latest Added")
-        )
-    )
-
-    private class GenreList : UriPartFilter(
-        "Select Genre",
-        arrayOf(
-            Pair("", "<select>"),
-            Pair("4-koma", "4-Koma"),
-            Pair("action", "Action"),
-            Pair("adaptation", "Adaptation"),
-            Pair("adult", "Adult"),
-            Pair("adventure", "Adventure"),
-            Pair("animal", "Animal"),
-            Pair("animals", "Animals"),
-            Pair("anthology", "Anthology"),
-            Pair("apocalypto", "Apocalypto"),
-            Pair("comedy", "Comedy"),
-            Pair("comic", "Comic"),
-            Pair("cooking", "Cooking"),
-            Pair("crime", "Crime"),
-            Pair("demons", "Demons"),
-            Pair("doujinshi", "Doujinshi"),
-            Pair("drama", "Drama"),
-            Pair("ecchi", "Ecchi"),
-            Pair("fantasi", "Fantasi"),
-            Pair("fantasy", "Fantasy"),
-            Pair("game", "Game"),
-            Pair("gender-bender", "Gender Bender"),
-            Pair("genderswap", "Genderswap"),
-            Pair("drama", "Drama"),
-            Pair("gore", "Gore"),
-            Pair("harem", "Harem"),
-            Pair("hentai", "Hentai"),
-            Pair("historical", "Historical"),
-            Pair("horor", "Horor"),
-            Pair("horror", "Horror"),
-            Pair("isekai", "Isekai"),
-            Pair("josei", "Josei"),
-            Pair("kingdom", "kingdom"),
-            Pair("magic", "Magic"),
-            Pair("manga", "Manga"),
-            Pair("manhua", "Manhua"),
-            Pair("manhwa", "Manhwa"),
-            Pair("martial-art", "Martial Art"),
-            Pair("martial-arts", "Martial Arts"),
-            Pair("mature", "Mature"),
-            Pair("mecha", "Mecha"),
-            Pair("medical", "Medical"),
-            Pair("military", "Military"),
-            Pair("modern", "Modern"),
-            Pair("monster", "Monster"),
-            Pair("monster-girls", "Monster Girls"),
-            Pair("music", "Music"),
-            Pair("mystery", "Mystery"),
-            Pair("oneshot", "Oneshot"),
-            Pair("post-apocalyptic", "Post-Apocalyptic"),
-            Pair("project", "Project"),
-            Pair("psychological", "Psychological"),
-            Pair("reincarnation", "Reincarnation"),
-            Pair("romance", "Romance"),
-            Pair("romancem", "Romancem"),
-            Pair("samurai", "Samurai"),
-            Pair("school", "School"),
-            Pair("school-life", "School Life"),
-            Pair("sci-fi", "Sci-Fi"),
-            Pair("seinen", "Seinen"),
-            Pair("shoujo", "Shoujo"),
-            Pair("shoujo-ai", "Shoujo Ai"),
-            Pair("shounen", "Shounen"),
-            Pair("shounen-ai", "Shounen Ai"),
-            Pair("slice-of-life", "Slice of Life"),
-            Pair("smut", "Smut"),
-            Pair("sports", "Sports"),
-            Pair("style-ancient", "Style ancient"),
-            Pair("super-power", "Super Power"),
-            Pair("superhero", "Superhero"),
-            Pair("supernatural", "Supernatural"),
-            Pair("survival", "Survival"),
-            Pair("survive", "Survive"),
-            Pair("thriller", "Thriller"),
-            Pair("time-travel", "Time Travel"),
-            Pair("tragedy", "Tragedy"),
-            Pair("urban", "Urban"),
-            Pair("vampire", "Vampire"),
-            Pair("video-games", "Video Games"),
-            Pair("virtual-reality", "Virtual Reality"),
-            Pair("webtoons", "Webtoons"),
-            Pair("yuri", "Yuri"),
-            Pair("zombies", "Zombies")
-        )
-    )
-
-    private open class UriPartFilter(displayName: String, val vals: Array<Pair<String, String>>) :
-        Filter.Select<String>(displayName, vals.map { it.second }.toTypedArray()) {
-        fun toUriPart() = vals[state].first
+    private interface UrlEncoded {
+        fun encode(url: HttpUrl.Builder)
     }
 
-    private inline fun <reified T> Iterable<*>.findInstance() = find { it is T } as? T
+    // essentially a named pair
+    protected class LabeledValue(val displayname: String, val _value: String?) {
+        val value: String get() = _value ?: displayname
+        override fun toString(): String = displayname
+    }
+
+    private open class Select<T>(header: String, values: Array<T>, state: Int = 0) : Filter.Select<T>(header, values, state) {
+        val selected: T
+            get() = this.values[this.state]
+    }
+
+    private open class MultiSelect<T>(header: String, val elems: List<T>) :
+        Filter.Group<Filter.CheckBox>(header, elems.map { object : Filter.CheckBox("$it") {} }) {
+        val selected: Sequence<T>
+            get() = this.elems.asSequence().filterIndexed { i, _ -> this.state[i].state }
+    }
+
+    // filters
+    override fun getFilterList() = FilterList(
+        Filter.Header("NOTE: Ignored if using text search!"),
+        GenreFilter(),
+        StatusFilter(),
+        TypesFilter(),
+        OrderByFilter(),
+    )
+
+    private fun GenreFilter() = object : MultiSelect<LabeledValue>("Genre", getGenreList()), UrlEncoded {
+        override fun encode(url: HttpUrl.Builder) {
+            selected.forEach { url.addQueryParameter("genre[]", it.value) }
+        }
+    }
+
+    private fun StatusFilter() = object : Select<LabeledValue>("Status", getPublicationStatus()), UrlEncoded {
+        override fun encode(url: HttpUrl.Builder) {
+            url.addQueryParameter("status", selected.value)
+        }
+    }
+
+    private fun TypesFilter() = object : Select<LabeledValue>("Type", getContentType()), UrlEncoded {
+        override fun encode(url: HttpUrl.Builder) {
+            url.addQueryParameter("type", selected.value)
+        }
+    }
+
+    private fun OrderByFilter(state: Int = 0) = object : Select<LabeledValue>("Order By", getOrderBy(), state), UrlEncoded {
+        override fun encode(url: HttpUrl.Builder) {
+            url.addQueryParameter("order", selected.value)
+        }
+    }
+
+    // overridable
+    // some sources have numeric values for filters
+    private var genrelist: List<LabeledValue>? = null
+    protected open fun getGenreList(): List<LabeledValue> {
+        // Filters are fetched immediately once an extension loads
+        // We're only able to get filters after a loading the manga directory, and resetting
+        // the filters is the only thing that seems to reinflate the view
+        return genrelist ?: listOf(LabeledValue("Press reset to attempt to fetch genres", ""))
+    }
+
+    private fun getPublicationStatus() = arrayOf(
+        LabeledValue("All", ""),
+        LabeledValue("Ongoing", "ongoing"),
+        LabeledValue("Completed", "completed"),
+        LabeledValue("Hiatus", "hiatus")
+    )
+
+    private fun getContentType() = arrayOf(
+        LabeledValue("All", ""),
+        LabeledValue("Manga", "manga"),
+        LabeledValue("Manhwa", "manhwa"),
+        LabeledValue("Manhua", "manhua"),
+        LabeledValue("Comic", "comic")
+    )
+
+    private fun getOrderBy() = arrayOf(
+        LabeledValue("Default", ""),
+        LabeledValue("A-Z", "title"),
+        LabeledValue("Z-A", "titlereverse"),
+        LabeledValue("Update", "update"),
+        LabeledValue("Added", "latest"),
+        LabeledValue("Popular", "popular")
+    )
 }
diff --git a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReaderGenerator.kt b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReaderGenerator.kt
index 01f8fb48d..c4a43b26b 100644
--- a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReaderGenerator.kt
+++ b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/wpmangareader/WPMangaReaderGenerator.kt
@@ -9,7 +9,7 @@ class WPMangaReaderGenerator : ThemeSourceGenerator {
 
     override val themeClass = "WPMangaReader"
 
-    override val baseVersionCode: Int = 4
+    override val baseVersionCode: Int = 5
 
     override val sources = listOf(