From f14515c057d03469d8bae9d4bcf2c95f0eb6c970 Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Tue, 12 Sep 2023 08:14:49 -0300
Subject: [PATCH] feat(tr/tranimeizle): Add hosts selection preference (#2183)

---
 src/tr/tranimeizle/build.gradle               |  6 ++-
 .../tr/tranimeizle/TRAnimeIzle.kt             | 40 ++++++++++++++++++-
 2 files changed, 43 insertions(+), 3 deletions(-)

diff --git a/src/tr/tranimeizle/build.gradle b/src/tr/tranimeizle/build.gradle
index e872a08a4..4aebcfb18 100644
--- a/src/tr/tranimeizle/build.gradle
+++ b/src/tr/tranimeizle/build.gradle
@@ -7,7 +7,7 @@ ext {
     extName = 'TR Anime Izle'
     pkgNameSuffix = 'tr.tranimeizle'
     extClass = '.TRAnimeIzle'
-    extVersionCode = 5
+    extVersionCode = 6
     libVersion = '13'
 }
 
@@ -21,6 +21,8 @@ dependencies {
     implementation(project(":lib-sibnet-extractor"))
     implementation(project(":lib-streamlare-extractor"))
     implementation(project(":lib-voe-extractor"))
-    implementation(project(":lib-yourupload-extractor"))}
+    implementation(project(":lib-vudeo-extractor"))
+    implementation(project(":lib-yourupload-extractor"))
+}
 
 apply from: "$rootDir/common.gradle"
diff --git a/src/tr/tranimeizle/src/eu/kanade/tachiyomi/animeextension/tr/tranimeizle/TRAnimeIzle.kt b/src/tr/tranimeizle/src/eu/kanade/tachiyomi/animeextension/tr/tranimeizle/TRAnimeIzle.kt
index 1fcbe25d6..c1de9a2e7 100644
--- a/src/tr/tranimeizle/src/eu/kanade/tachiyomi/animeextension/tr/tranimeizle/TRAnimeIzle.kt
+++ b/src/tr/tranimeizle/src/eu/kanade/tachiyomi/animeextension/tr/tranimeizle/TRAnimeIzle.kt
@@ -22,6 +22,7 @@ import eu.kanade.tachiyomi.lib.sendvidextractor.SendvidExtractor
 import eu.kanade.tachiyomi.lib.sibnetextractor.SibnetExtractor
 import eu.kanade.tachiyomi.lib.streamlareextractor.StreamlareExtractor
 import eu.kanade.tachiyomi.lib.voeextractor.VoeExtractor
+import eu.kanade.tachiyomi.lib.vudeoextractor.VudeoExtractor
 import eu.kanade.tachiyomi.lib.youruploadextractor.YourUploadExtractor
 import eu.kanade.tachiyomi.network.GET
 import eu.kanade.tachiyomi.network.POST
@@ -183,6 +184,7 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
 
         val allFansubs = PREF_FANSUB_SELECTION_ENTRIES
         val chosenFansubs = preferences.getStringSet(PREF_FANSUB_SELECTION_KEY, allFansubs.toSet())!!
+        val chosenHosts = preferences.getStringSet(PREF_HOSTS_SELECTION_KEY, PREF_HOSTS_SELECTION_DEFAULT)!!
 
         return doc.select("div.fansubSelector").toList()
             // Filter-out non-chosen fansubs that were included in the fansub selection preference.
@@ -199,6 +201,8 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
                     .execute()
                     .use { it.asJsoup() }
                     .select("li.sourceBtn")
+                    .toList()
+                    .filter { it.selectFirst("p")?.ownText().orEmpty() in chosenHosts }
                     .parallelMap {
                         runCatching {
                             getVideosFromId(it.attr("data-id"))
@@ -227,6 +231,7 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
     private val sibnetExtractor by lazy { SibnetExtractor(client) }
     private val streamlareExtractor by lazy { StreamlareExtractor(client) }
     private val voeExtractor by lazy { VoeExtractor(client) }
+    private val vudeoExtractor by lazy { VudeoExtractor(client) }
     private val yourUploadExtractor by lazy { YourUploadExtractor(client) }
 
     private fun getVideosFromId(id: String): List<Video> {
@@ -251,11 +256,12 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
             "mixdrop" in url -> mixDropExtractor.videoFromUrl(url)
             "mp4upload" in url -> mp4uploadExtractor.videosFromUrl(url, headers)
             "myvi." in url -> mytvExtractor.videosFromUrl(url)
-            "ok.ru" in url -> okruExtractor.videosFromUrl(url)
+            "ok.ru" in url || "odnoklassniki.ru" in url -> okruExtractor.videosFromUrl(url)
             "sendvid.com" in url -> sendvidExtractor.videosFromUrl(url)
             "video.sibnet" in url -> sibnetExtractor.videosFromUrl(url)
             "streamlare.com" in url -> streamlareExtractor.videosFromUrl(url)
             "voe.sx" in url -> voeExtractor.videoFromUrl(url)?.let(::listOf) ?: emptyList()
+            "//vudeo." in url -> vudeoExtractor.videosFromUrl(url)
             "yourupload.com" in url -> {
                 yourUploadExtractor.videoFromUrl(url, headers)
                     // ignore error links
@@ -326,6 +332,19 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
                 }.getOrDefault(false)
             }
         }.also(screen::addPreference)
+
+        MultiSelectListPreference(screen.context).apply {
+            key = PREF_HOSTS_SELECTION_KEY
+            title = PREF_HOSTS_SELECTION_TITLE
+            entries = PREF_HOSTS_SELECTION_ENTRIES
+            entryValues = PREF_HOSTS_SELECTION_ENTRIES
+            setDefaultValue(PREF_HOSTS_SELECTION_DEFAULT)
+
+            setOnPreferenceChangeListener { _, newValue ->
+                @Suppress("UNCHECKED_CAST")
+                preferences.edit().putStringSet(key, newValue as Set<String>).commit()
+            }
+        }.also(screen::addPreference)
     }
 
     // ============================= Utilities ==============================
@@ -416,5 +435,24 @@ class TRAnimeIzle : ParsedAnimeHttpSource(), ConfigurableAnimeSource {
         private const val PREF_ADDITIONAL_FANSUBS_DIALOG_MESSAGE = "Example: AntichristHaters Fansub, 2cm erect subs"
         private const val PREF_ADDITIONAL_FANSUBS_SUMMARY = "You can add more fansubs to the previous preference from here."
         private const val PREF_ADDITIONAL_FANSUBS_TOAST = "Reopen the extension's preferences for it to take effect."
+
+        private const val PREF_HOSTS_SELECTION_KEY = "pref_hosts_selection"
+        private const val PREF_HOSTS_SELECTION_TITLE = "Enable/disable video hosts"
+        private val PREF_HOSTS_SELECTION_ENTRIES = arrayOf(
+            "Filemoon",
+            "MixDrop",
+            "Mp4upload",
+            "Myvi",
+            "Ok.RU",
+            "SendVid",
+            "Sibnet",
+            "Streamlare",
+            "Voe",
+            "Vudeo",
+            "Yourupload",
+        )
+
+        // XDDDDDDDDD
+        private val PREF_HOSTS_SELECTION_DEFAULT by lazy { PREF_HOSTS_SELECTION_ENTRIES.toSet() }
     }
 }