From 76c8129672716bf4448cad1d48b996bc3fa03ec7 Mon Sep 17 00:00:00 2001
From: Dan <usernamekepo@gmail.com>
Date: Wed, 8 Nov 2023 07:10:00 +0700
Subject: [PATCH] feat(id/samehadaku): Add search filters (#2478)

---
 src/id/samehadaku/build.gradle                |   2 +-
 .../id/samehadaku/Samehadaku.kt               | 131 +++++++------
 .../id/samehadaku/SamehadakuFilters.kt        | 183 ++++++++++++++++++
 3 files changed, 253 insertions(+), 63 deletions(-)
 create mode 100644 src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/SamehadakuFilters.kt

diff --git a/src/id/samehadaku/build.gradle b/src/id/samehadaku/build.gradle
index 364c700ae..1daec5f02 100644
--- a/src/id/samehadaku/build.gradle
+++ b/src/id/samehadaku/build.gradle
@@ -5,7 +5,7 @@ ext {
     extName = 'Samehadaku'
     pkgNameSuffix = 'id.samehadaku'
     extClass = '.Samehadaku'
-    extVersionCode = 2
+    extVersionCode = 3
     libVersion = '13'
 }
 
diff --git a/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/Samehadaku.kt b/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/Samehadaku.kt
index 5fe622b49..c8e9c93dd 100644
--- a/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/Samehadaku.kt
+++ b/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/Samehadaku.kt
@@ -6,10 +6,11 @@ import androidx.preference.ListPreference
 import androidx.preference.PreferenceScreen
 import eu.kanade.tachiyomi.animesource.ConfigurableAnimeSource
 import eu.kanade.tachiyomi.animesource.model.AnimeFilterList
+import eu.kanade.tachiyomi.animesource.model.AnimesPage
 import eu.kanade.tachiyomi.animesource.model.SAnime
 import eu.kanade.tachiyomi.animesource.model.SEpisode
 import eu.kanade.tachiyomi.animesource.model.Video
-import eu.kanade.tachiyomi.animesource.online.ParsedAnimeHttpSource
+import eu.kanade.tachiyomi.animesource.online.AnimeHttpSource
 import eu.kanade.tachiyomi.network.GET
 import eu.kanade.tachiyomi.network.POST
 import eu.kanade.tachiyomi.util.asJsoup
@@ -27,9 +28,9 @@ import uy.kohesive.injekt.api.get
 import java.text.SimpleDateFormat
 import java.util.Locale
 
-class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
+class Samehadaku : ConfigurableAnimeSource, AnimeHttpSource() {
     override val name: String = "Samehadaku"
-    override val baseUrl: String = "https://samehadaku.rent"
+    override val baseUrl: String = "https://samehadaku.mom"
     override val lang: String = "id"
     override val supportsLatest: Boolean = true
 
@@ -42,75 +43,76 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
     override fun popularAnimeRequest(page: Int): Request =
         GET("$baseUrl/daftar-anime-2/page/$page/?order=popular")
 
-    override fun popularAnimeSelector(): String = "div.relat > article"
-
-    override fun popularAnimeFromElement(element: Element): SAnime =
-        getAnimeFromAnimeElement(element)
-
-    override fun popularAnimeNextPageSelector(): String = "div.pagination > a.arrow_pag"
+    override fun popularAnimeParse(response: Response): AnimesPage =
+        getAnimeParse(response.asJsoup(), "div.relat > article")
 
     // =============================== Latest ===============================
 
     override fun latestUpdatesRequest(page: Int): Request =
         GET("$baseUrl/daftar-anime-2/page/$page/?order=latest")
 
-    override fun latestUpdatesSelector(): String = "div.relat > article"
-
-    override fun latestUpdatesFromElement(element: Element): SAnime =
-        getAnimeFromAnimeElement(element)
-
-    override fun latestUpdatesNextPageSelector(): String = "div.pagination > a.arrow_pag"
+    override fun latestUpdatesParse(response: Response): AnimesPage =
+        getAnimeParse(response.asJsoup(), "div.relat > article")
 
     // =============================== Search ===============================
 
-    override fun searchAnimeRequest(page: Int, query: String, filters: AnimeFilterList): Request =
-        GET("$baseUrl/page/$page/?s=$query")
+    override fun searchAnimeRequest(page: Int, query: String, filters: AnimeFilterList): Request {
+        val params = SamehadakuFilters.getSearchParameters(filters)
+        return GET("$baseUrl/daftar-anime-2/page/$page/?s=$query${params.filter}", headers)
+    }
 
-    override fun searchAnimeSelector(): String = "main.site-main.relat > article"
+    override fun searchAnimeParse(response: Response): AnimesPage {
+        val doc = response.asJsoup()
+        val searchSelector = "main.site-main.relat > article"
+        return if (doc.selectFirst(searchSelector) != null) {
+            getAnimeParse(doc, searchSelector)
+        } else {
+            getAnimeParse(doc, "div.relat > article")
+        }
+    }
 
-    override fun searchAnimeFromElement(element: Element): SAnime =
-        getAnimeFromAnimeElement(element)
+    // ============================== Filters ===============================
 
-    override fun searchAnimeNextPageSelector(): String = "div.pagination > a.arrow_pag"
+    override fun getFilterList(): AnimeFilterList = SamehadakuFilters.FILTER_LIST
 
     // =========================== Anime Details ============================
 
-    override fun animeDetailsParse(document: Document): SAnime {
-        val detail = document.selectFirst("div.infox > div.spe")!!
+    override fun animeDetailsParse(response: Response): SAnime {
+        val doc = response.asJsoup()
+        val detail = doc.selectFirst("div.infox > div.spe")!!
         return SAnime.create().apply {
             author = detail.getInfo("Studio")
             status = parseStatus(detail.getInfo("Status"))
-            title = document.selectFirst("h3.anim-detail")!!.text().split("Detail Anime ")[1]
+            title = doc.selectFirst("h3.anim-detail")!!.text().split("Detail Anime ")[1]
             thumbnail_url =
-                document.selectFirst("div.infoanime.widget_senction > div.thumb > img")!!
+                doc.selectFirst("div.infoanime.widget_senction > div.thumb > img")!!
                     .attr("src")
             description =
-                document.selectFirst("div.entry-content.entry-content-single > p")!!.text()
+                doc.selectFirst("div.entry-content.entry-content-single > p")!!.text()
         }
     }
 
     // ============================== Episodes ==============================
 
-    override fun episodeListSelector(): String =
-        "div.whites.lsteps.widget_senction > div.lstepsiode.listeps > ul > li"
-
-    override fun episodeFromElement(element: Element): SEpisode {
-        val episode = element.selectFirst("span.eps > a")!!.text()
-        return SEpisode.create().apply {
-            setUrlWithoutDomain(element.selectFirst("a")!!.attr("href"))
-            episode_number = episode.toFloatOrNull() ?: 1F
-            name = element.selectFirst("span.lchx > a")!!.text()
-            date_upload = element.selectFirst("span.date")!!.text().toDate()
-        }
+    override fun episodeListParse(response: Response): List<SEpisode> {
+        val doc = response.asJsoup()
+        return doc.select("div.lstepsiode > ul > li")
+            .map {
+                val episode = it.selectFirst("span.eps > a")!!
+                SEpisode.create().apply {
+                    setUrlWithoutDomain(episode.attr("href"))
+                    episode_number = episode.text().trim().toFloatOrNull() ?: 1F
+                    name = it.selectFirst("span.lchx > a")!!.text()
+                    date_upload = it.selectFirst("span.date")!!.text().toDate()
+                }
+            }
     }
 
     // ============================ Video Links =============================
 
-    override fun videoListSelector(): String = "#server > ul > li > div"
-
     override fun videoListParse(response: Response): List<Video> {
         val doc = response.asJsoup()
-        return doc.select(videoListSelector())
+        return doc.select("#server > ul > li > div")
             .parallelMapNotNull {
                 runCatching { getEmbedLinks(it) }.getOrNull()
             }
@@ -119,17 +121,11 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
             }.flatten()
     }
 
-    override fun videoFromElement(element: Element) = throw Exception("not used")
-
-    override fun videoUrlParse(document: Document) = throw Exception("not used")
-
     // ============================= Utilities ==============================
 
     override fun List<Video>.sort(): List<Video> {
         val quality = preferences.getString(PREF_QUALITY_KEY, PREF_QUALITY_DEFAULT)!!
-        return sortedWith(
-            compareByDescending { it.quality.contains(quality) },
-        )
+        return sortedWith(compareByDescending { it.quality.contains(quality) })
     }
 
     private fun String?.toDate(): Long {
@@ -137,7 +133,7 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
             .getOrNull() ?: 0L
     }
 
-    private fun Element.getInfo(info: String, cut: Boolean = true): String? {
+    private fun Element.getInfo(info: String, cut: Boolean = true): String {
         return selectFirst("span:has(b:contains($info))")!!.text()
             .let {
                 when {
@@ -153,13 +149,23 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
         }
     }
 
-    private fun getAnimeFromAnimeElement(element: Element): SAnime {
-        return SAnime.create().apply {
-            setUrlWithoutDomain(element.selectFirst("div > a")!!.attr("href"))
-            title = element.selectFirst("div.title > h2")!!.text()
-            thumbnail_url =
-                element.selectFirst("div.content-thumb > img")!!.attr("src")
+    private fun getAnimeParse(document: Document, query: String): AnimesPage {
+        val animes = document.select(query).map {
+            SAnime.create().apply {
+                setUrlWithoutDomain(it.selectFirst("div > a")!!.attr("href"))
+                title = it.selectFirst("div.title > h2")!!.text()
+                thumbnail_url = it.selectFirst("div.content-thumb > img")!!.attr("src")
+            }
         }
+        val hasNextPage = try {
+            val pagination = document.selectFirst("div.pagination")!!
+            val totalPage = pagination.selectFirst("span:nth-child(1)")!!.text().split(" ").last()
+            val currentPage = pagination.selectFirst("span.page-numbers.current")!!.text()
+            currentPage.toInt() < totalPage.toInt()
+        } catch (_: Exception) {
+            false
+        }
+        return AnimesPage(animes, hasNextPage)
     }
 
     private fun parseStatus(status: String?): Int {
@@ -177,7 +183,6 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
             add("nume", element.attr("data-nume"))
             add("type", element.attr("data-type"))
         }.build()
-
         return client.newCall(POST("$baseUrl/wp-admin/admin-ajax.php", body = form))
             .execute()
             .use {
@@ -191,18 +196,20 @@ class Samehadaku : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
         return when {
             "wibufile" in link -> {
                 client.newCall(GET(link)).execute().use {
+                    if (!it.isSuccessful) return emptyList()
                     val videoUrl = it.body.string().substringAfter("cast(\"").substringBefore("\"")
                     listOf(Video(videoUrl, server, videoUrl, headers))
                 }
             }
 
-//            "krakenfiles" in link -> {
-//                client.newCall(GET(link)).execute().use {
-//                    val doc = it.asJsoup()
-//                    val videoUrl = doc.selectFirst("#my-video > source")!!.attr("src")
-//                    listOf(Video(videoUrl, server, videoUrl, headers))
-//                }
-//            }
+            "krakenfiles" in link -> {
+                client.newCall(GET(link)).execute().use {
+                    val doc = it.asJsoup()
+                    val getUrl = doc.selectFirst("source")!!.attr("src")
+                    val videoUrl = "https:${getUrl.replace("&amp;", "&")}"
+                    listOf(Video(videoUrl, server, videoUrl, headers))
+                }
+            }
 
             "blogger" in link -> {
                 client.newCall(GET(link)).execute().use {
diff --git a/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/SamehadakuFilters.kt b/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/SamehadakuFilters.kt
new file mode 100644
index 000000000..e1d0786df
--- /dev/null
+++ b/src/id/samehadaku/src/eu/kanade/tachiyomi/animeextension/id/samehadaku/SamehadakuFilters.kt
@@ -0,0 +1,183 @@
+package eu.kanade.tachiyomi.animeextension.id.samehadaku
+
+import eu.kanade.tachiyomi.animesource.model.AnimeFilter
+import eu.kanade.tachiyomi.animesource.model.AnimeFilterList
+
+object SamehadakuFilters {
+    open class QueryPartFilter(
+        displayName: String,
+        val vals: Array<Pair<String, String>>,
+    ) : AnimeFilter.Select<String>(
+        displayName,
+        vals.map { it.first }.toTypedArray(),
+    ) {
+        fun toQueryPart(name: String) = "&$name=${vals[state].second}"
+    }
+
+    open class CheckBoxFilterList(name: String, values: List<CheckBox>) :
+        AnimeFilter.Group<AnimeFilter.CheckBox>(name, values)
+
+    private class CheckBoxVal(name: String, state: Boolean = false) :
+        AnimeFilter.CheckBox(name, state)
+
+    private inline fun <reified R> AnimeFilterList.asQueryPart(name: String): String {
+        return (this.getFirst<R>() as QueryPartFilter).toQueryPart(name)
+    }
+
+    private inline fun <reified R> AnimeFilterList.getFirst(): R {
+        return this.filterIsInstance<R>().first()
+    }
+
+    private inline fun <reified R> AnimeFilterList.parseCheckbox(
+        options: Array<Pair<String, String>>,
+        name: String,
+    ): String {
+        return (this.getFirst<R>() as CheckBoxFilterList).state
+            .mapNotNull { checkbox ->
+                if (checkbox.state) {
+                    options.find { it.first == checkbox.name }!!.second
+                } else {
+                    null
+                }
+            }.joinToString("&$name[]=").let {
+                if (it.isBlank()) {
+                    ""
+                } else {
+                    "&$name[]=$it"
+                }
+            }
+    }
+
+    class GenreFilter : CheckBoxFilterList(
+        "Genre",
+        FiltersData.GENRE.map { CheckBoxVal(it.first, false) },
+    )
+
+    class TypeFilter : QueryPartFilter("Type", FiltersData.TYPE)
+
+    class StatusFilter : QueryPartFilter("Status", FiltersData.STATUS)
+
+    class OrderFilter : QueryPartFilter("Sort By", FiltersData.ORDER)
+
+    val FILTER_LIST
+        get() = AnimeFilterList(
+            AnimeFilter.Header("Ignored With Text Search!!"),
+            GenreFilter(),
+            TypeFilter(),
+            StatusFilter(),
+            OrderFilter(),
+        )
+
+    data class FilterSearchParams(
+        val filter: String = "",
+    )
+
+    internal fun getSearchParameters(filters: AnimeFilterList): FilterSearchParams {
+        if (filters.isEmpty()) return FilterSearchParams()
+        return FilterSearchParams(
+            filters.parseCheckbox<GenreFilter>(FiltersData.GENRE, "genre") +
+                filters.asQueryPart<TypeFilter>("type") +
+                filters.asQueryPart<StatusFilter>("status") +
+                filters.asQueryPart<OrderFilter>("order"),
+        )
+    }
+
+    private object FiltersData {
+        val ORDER = arrayOf(
+            Pair("A-Z", "title"),
+            Pair("Z-A", "titlereverse"),
+            Pair("Latest Update", "update"),
+            Pair("Latest Added", "latest"),
+            Pair("Popular", "popular"),
+        )
+
+        val STATUS = arrayOf(
+            Pair("All", ""),
+            Pair("Currently Airing", "Currently Airing"),
+            Pair("Finished Airing", "Finished Airing"),
+        )
+
+        val TYPE = arrayOf(
+            Pair("All", ""),
+            Pair("TV", "TV"),
+            Pair("OVA", "OVA"),
+            Pair("ONA", "ONA"),
+            Pair("Special", "Special"),
+            Pair("Movie", "Movie"),
+        )
+
+        val GENRE = arrayOf(
+            Pair("Action", "action"),
+            Pair("Adult Cast", "adult-cast"),
+            Pair("Adventure", "adventure"),
+            Pair("Boys Love", "boys-love"),
+            Pair("Childcare", "childcare"),
+            Pair("Comedy", "comedy"),
+            Pair("Delinquents", "delinquents"),
+            Pair("Dementia", "dementia"),
+            Pair("Demons", "demons"),
+            Pair("Detective", "detective"),
+            Pair("Drama", "drama"),
+            Pair("ecch", "ecch"),
+            Pair("Ecchi", "ecchi"),
+            Pair("Fantasy", "fantasy"),
+            Pair("Gag Humor", "gag-humor"),
+            Pair("Game", "game"),
+            Pair("Gore", "gore"),
+            Pair("Gourmet", "gourmet"),
+            Pair("Harem", "harem"),
+            Pair("High Stakes Game", "high-stakes-game"),
+            Pair("Historical", "historical"),
+            Pair("Horror", "horror"),
+            Pair("Isekai", "isekai"),
+            Pair("Iyashikei", "iyashikei"),
+            Pair("Kids", "kids"),
+            Pair("Magic", "magic"),
+            Pair("Martial Arts", "martial-arts"),
+            Pair("Mecha", "mecha"),
+            Pair("Medical", "medical"),
+            Pair("Military", "military"),
+            Pair("Music", "music"),
+            Pair("Mystery", "mystery"),
+            Pair("Mythology", "mythology"),
+            Pair("Organized Crime", "organized-crime"),
+            Pair("Otaku Culture", "otaku-culture"),
+            Pair("Parody", "parody"),
+            Pair("Performing Arts", "performing-arts"),
+            Pair("pilihan", "pilihan"),
+            Pair("Police", "police"),
+            Pair("poling", "poling"),
+            Pair("Psychological", "psychological"),
+            Pair("Recomendation", "recomendation"),
+            Pair("Reincarnation", "reincarnation"),
+            Pair("Romance", "romance"),
+            Pair("Romantic Subtext", "romantic-subtext"),
+            Pair("Samurai", "samurai"),
+            Pair("School", "school"),
+            Pair("Sci-Fi", "sci-fi"),
+            Pair("Seinen", "seinen"),
+            Pair("Shoujo", "shoujo"),
+            Pair("Shoujo Ai", "shoujo-ai"),
+            Pair("Shounen", "shounen"),
+            Pair("Showbiz", "showbiz"),
+            Pair("Slice of Life", "slice-of-life"),
+            Pair("Space", "space"),
+            Pair("Sports", "sports"),
+            Pair("Spring 2023", "spring-2023"),
+            Pair("Strategy Game", "strategy-game"),
+            Pair("Super Power", "super-power"),
+            Pair("Supernatural", "supernatural"),
+            Pair("Survival", "survival"),
+            Pair("Suspense", "suspense"),
+            Pair("Team Sports", "team-sports"),
+            Pair("Thriller", "thriller"),
+            Pair("Time Travel", "time-travel"),
+            Pair("Vampire", "vampire"),
+            Pair("Video Game", "video-game"),
+            Pair("vidiogame", "vidiogame"),
+            Pair("Visual Arts", "visual-arts"),
+            Pair("Work Life", "work-life"),
+            Pair("Workplace", "workplace"),
+        )
+    }
+}