From 4b7639197f3700a80feedce81bf0902ce1c40adc Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Mon, 9 Oct 2023 08:25:22 -0300
Subject: [PATCH] feat(all/chineseanime): Add VatchusExtractor (#2331)

---
 .../chineseanime/additional.gradle            |  3 +-
 .../chineseanime/src/ChineseAnime.kt          | 14 +++--
 .../src/extractors/VatchusExtractor.kt        | 56 +++++++++++++++++++
 .../animestream/AnimeStreamGenerator.kt       |  2 +-
 4 files changed, 67 insertions(+), 8 deletions(-)
 create mode 100644 multisrc/overrides/animestream/chineseanime/src/extractors/VatchusExtractor.kt

diff --git a/multisrc/overrides/animestream/chineseanime/additional.gradle b/multisrc/overrides/animestream/chineseanime/additional.gradle
index 4f8c322f9..f5e445431 100644
--- a/multisrc/overrides/animestream/chineseanime/additional.gradle
+++ b/multisrc/overrides/animestream/chineseanime/additional.gradle
@@ -1,4 +1,5 @@
 dependencies {
     implementation(project(":lib-dailymotion-extractor"))
     implementation(project(":lib-streamwish-extractor"))
-}
\ No newline at end of file
+    implementation(project(":lib-playlist-utils"))
+}
diff --git a/multisrc/overrides/animestream/chineseanime/src/ChineseAnime.kt b/multisrc/overrides/animestream/chineseanime/src/ChineseAnime.kt
index e2ef937a8..5be6f8efd 100644
--- a/multisrc/overrides/animestream/chineseanime/src/ChineseAnime.kt
+++ b/multisrc/overrides/animestream/chineseanime/src/ChineseAnime.kt
@@ -2,6 +2,7 @@ package eu.kanade.tachiyomi.animeextension.all.chineseanime
 
 import androidx.preference.ListPreference
 import androidx.preference.PreferenceScreen
+import eu.kanade.tachiyomi.animeextension.all.chineseanime.extractors.VatchusExtractor
 import eu.kanade.tachiyomi.animesource.model.Video
 import eu.kanade.tachiyomi.lib.dailymotionextractor.DailymotionExtractor
 import eu.kanade.tachiyomi.lib.streamwishextractor.StreamWishExtractor
@@ -23,15 +24,16 @@ class ChineseAnime : AnimeStream(
     override val filtersSelector = "div.filter > ul"
 
     // ============================ Video Links =============================
+    private val dailymotionExtractor by lazy { DailymotionExtractor(client, headers) }
+    private val streamwishExtractor by lazy { StreamWishExtractor(client, headers) }
+    private val vatchusExtractor by lazy { VatchusExtractor(client, headers) }
+
     override fun getVideoList(url: String, name: String): List<Video> {
         val prefix = "$name - "
         return when {
-            url.contains("dailymotion") -> {
-                DailymotionExtractor(client, headers).videosFromUrl(url, prefix)
-            }
-            url.contains("embedwish") -> {
-                StreamWishExtractor(client, headers).videosFromUrl(url, prefix)
-            }
+            url.contains("dailymotion") -> dailymotionExtractor.videosFromUrl(url, prefix)
+            url.contains("embedwish") -> streamwishExtractor.videosFromUrl(url, prefix)
+            url.contains("vatchus") -> vatchusExtractor.videosFromUrl(url, prefix)
             else -> emptyList()
         }
     }
diff --git a/multisrc/overrides/animestream/chineseanime/src/extractors/VatchusExtractor.kt b/multisrc/overrides/animestream/chineseanime/src/extractors/VatchusExtractor.kt
new file mode 100644
index 000000000..051e57db3
--- /dev/null
+++ b/multisrc/overrides/animestream/chineseanime/src/extractors/VatchusExtractor.kt
@@ -0,0 +1,56 @@
+package eu.kanade.tachiyomi.animeextension.all.chineseanime.extractors
+
+import android.util.Base64
+import eu.kanade.tachiyomi.animesource.model.Track
+import eu.kanade.tachiyomi.animesource.model.Video
+import eu.kanade.tachiyomi.lib.playlistutils.PlaylistUtils
+import eu.kanade.tachiyomi.network.GET
+import eu.kanade.tachiyomi.util.asJsoup
+import okhttp3.Headers
+import okhttp3.OkHttpClient
+
+class VatchusExtractor(private val client: OkHttpClient, private val headers: Headers) {
+    private val playlistUtils by lazy { PlaylistUtils(client, headers) }
+
+    fun videosFromUrl(url: String, prefix: String): List<Video> {
+        val doc = client.newCall(GET(url, headers)).execute()
+            .use { it.asJsoup() }
+
+        val script = doc.selectFirst("script:containsData(document.write)")
+            ?.data()
+            ?: return emptyList()
+
+        val numberList = script.substringAfter(" = [").substringBefore("];")
+            .replace("\"", "")
+            .split(",")
+            .map(String::trim)
+            .filter(String::isNotBlank)
+            .map { String(Base64.decode(it, Base64.DEFAULT)) }
+            .mapNotNull { it.filter(Char::isDigit).toIntOrNull() }
+
+        val offset = numberList.first() - 60
+        val decodedData = numberList.joinToString("") {
+            Char(it - offset).toString()
+        }.trim()
+
+        val playlistUrl = decodedData.substringAfter("file:'").substringBefore("'")
+        val subs = decodedData.substringAfter("tracks:[").substringBefore("]")
+            .split("{")
+            .drop(1)
+            .filter { it.contains(""""kind":"captions"""") }
+            .mapNotNull {
+                val trackUrl = it.substringAfter("file\":\"").substringBefore('"')
+                    .takeIf { link -> link.startsWith("http") }
+                    ?: return@mapNotNull null
+                val language = it.substringAfter("label\":\"").substringBefore('"')
+                Track(trackUrl, language)
+            }
+
+        return playlistUtils.extractFromHls(
+            playlistUrl,
+            url,
+            subtitleList = subs,
+            videoNameGen = { prefix + it },
+        )
+    }
+}
diff --git a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/animestream/AnimeStreamGenerator.kt b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/animestream/AnimeStreamGenerator.kt
index 7f32feddd..f3d3fc4d9 100644
--- a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/animestream/AnimeStreamGenerator.kt
+++ b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/animestream/AnimeStreamGenerator.kt
@@ -17,7 +17,7 @@ class AnimeStreamGenerator : ThemeSourceGenerator {
         SingleLang("AnimeTitans", "https://animetitans.com", "ar", isNsfw = false, overrideVersionCode = 13),
         SingleLang("AnimeXin", "https://animexin.vip", "all", isNsfw = false, overrideVersionCode = 6),
         SingleLang("AsyaAnimeleri", "https://asyaanimeleri.com", "tr", isNsfw = false, overrideVersionCode = 1),
-        SingleLang("ChineseAnime", "https://chineseanime.top", "all", isNsfw = false),
+        SingleLang("ChineseAnime", "https://chineseanime.top", "all", isNsfw = false, overrideVersionCode = 1),
         SingleLang("desu-online", "https://desu-online.pl", "pl", className = "DesuOnline", isNsfw = false, overrideVersionCode = 3),
         SingleLang("DonghuaStream", "https://donghuastream.co.in", "en", isNsfw = false),
         SingleLang("Hstream", "https://hstream.moe", "en", isNsfw = true, overrideVersionCode = 3),