From 600b10a1769a092a0781d5232b9bbb36c5085aa5 Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Mon, 22 May 2023 22:03:56 +0000
Subject: [PATCH] fix(en/KickAssAnime): Fix video extractor (#1634)

* fix: Update Dynamic AES Extractor and fix video extractor on sapphire videos

* fix: Ignore exceptions on video extractor

* fix: Prevent empty video URLs

* chore: Bump version
---
 src/en/kickassanime/build.gradle              |  2 +-
 .../en/kickassanime/KickAssAnime.kt           |  4 +-
 .../en/kickassanime/dto/KickAssAnimeDto.kt    |  9 ++-
 .../extractors/AESKeyExtractor.kt             | 57 +++++++++++++------
 .../extractors/KickAssAnimeExtractor.kt       | 15 +++--
 5 files changed, 63 insertions(+), 24 deletions(-)

diff --git a/src/en/kickassanime/build.gradle b/src/en/kickassanime/build.gradle
index 4ce8f25d0..8e611e274 100644
--- a/src/en/kickassanime/build.gradle
+++ b/src/en/kickassanime/build.gradle
@@ -9,7 +9,7 @@ ext {
     pkgNameSuffix = 'en.kickassanime'
     extClass = '.KickAssAnime'
     libVersion = '13'
-    extVersionCode = 26
+    extVersionCode = 27
 }
 
 dependencies {
diff --git a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/KickAssAnime.kt b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/KickAssAnime.kt
index 2c115a4c8..d329e383c 100644
--- a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/KickAssAnime.kt
+++ b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/KickAssAnime.kt
@@ -129,7 +129,9 @@ class KickAssAnime : ConfigurableAnimeSource, AnimeHttpSource() {
     override fun videoListParse(response: Response): List<Video> {
         val videos = response.parseAs<ServersDto>()
         val extractor = KickAssAnimeExtractor(client, json, headers)
-        return videos.servers.flatMap(extractor::videosFromUrl)
+        return videos.servers.mapNotNull {
+            runCatching { extractor.videosFromUrl(it) }.getOrNull()
+        }.flatten()
     }
 
     // =========================== Anime Details ============================
diff --git a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/dto/KickAssAnimeDto.kt b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/dto/KickAssAnimeDto.kt
index bea693b61..1125c3417 100644
--- a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/dto/KickAssAnimeDto.kt
+++ b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/dto/KickAssAnimeDto.kt
@@ -70,7 +70,14 @@ data class VideoDto(
     val dash: String = "",
     val subtitles: List<SubtitlesDto> = emptyList(),
 ) {
-    val playlistUrl by lazy { if (hls.isBlank()) "https:$dash" else hls }
+    val playlistUrl by lazy {
+        hls.ifEmpty { dash }.let { uri ->
+            when {
+                uri.startsWith("//") -> "https:$uri"
+                else -> uri
+            }
+        }
+    }
 
     @Serializable
     data class SubtitlesDto(val name: String, val language: String, val src: String)
diff --git a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/AESKeyExtractor.kt b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/AESKeyExtractor.kt
index f3e7fc06d..7254a9401 100644
--- a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/AESKeyExtractor.kt
+++ b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/AESKeyExtractor.kt
@@ -6,10 +6,10 @@ import android.os.Looper
 import android.webkit.JavascriptInterface
 import android.webkit.WebSettings
 import android.webkit.WebView
+import eu.kanade.tachiyomi.lib.cryptoaes.CryptoAES.decodeHex
 import eu.kanade.tachiyomi.network.GET
 import okhttp3.OkHttpClient
-import uy.kohesive.injekt.Injekt
-import uy.kohesive.injekt.api.get
+import uy.kohesive.injekt.injectLazy
 import java.util.concurrent.CountDownLatch
 import java.util.concurrent.TimeUnit
 
@@ -18,21 +18,27 @@ class AESKeyExtractor(private val client: OkHttpClient) {
         val KEY_MAP = mutableMapOf(
             // Default key. if it changes, the extractor will update it.
             "PinkBird" to "7191d608bd4deb4dc36f656c4bbca1b7".toByteArray(),
-            "SapphireDuck" to "f04274d54a9e01ed4a728c5c1889886e".toByteArray(), // i hate sapphire
+            "SapphireDuck" to "2940ba141ba490377b3f0a28ce56641a".toByteArray(), // i hate sapphire
         )
 
-        private const val ERROR_MSG_GENERIC = "the AES key was not found."
-        private const val ERROR_MSG_VAR = "the AES key variable was not found"
+        // ..... dont try reading them.
         private val KEY_VAR_REGEX by lazy { Regex("\\.AES\\[.*?\\]\\((\\w+)\\),") }
+        private val KEY_FUNC_REGEX by lazy {
+            Regex(",\\w+\\.SHA1\\)\\(\\[.*?function.*?\\(\\w+=(\\w+).*?function")
+        }
     }
 
-    private val context = Injekt.get<Application>()
+    private val context: Application by injectLazy()
     private val handler by lazy { Handler(Looper.getMainLooper()) }
 
     class ExtractorJSI(private val latch: CountDownLatch, private val prefix: String) {
         @JavascriptInterface
         fun setKey(key: String) {
-            AESKeyExtractor.KEY_MAP.set(prefix, key.toByteArray())
+            val keyBytes = when {
+                key.length == 32 -> key.toByteArray()
+                else -> key.decodeHex()
+            }
+            AESKeyExtractor.KEY_MAP.set(prefix, keyBytes)
             latch.countDown()
         }
     }
@@ -78,7 +84,7 @@ class AESKeyExtractor(private val client: OkHttpClient) {
             webView = null
         }
 
-        return AESKeyExtractor.KEY_MAP.get(prefix) ?: throw Exception(ERROR_MSG_GENERIC)
+        return AESKeyExtractor.KEY_MAP.get(prefix) ?: throw Exception()
     }
 
     private fun patchScriptFromUrl(url: String): String {
@@ -90,18 +96,37 @@ class AESKeyExtractor(private val client: OkHttpClient) {
     }
 
     private fun patchScriptFromHtml(baseUrl: String, body: String): String {
-        val scriptPath = body.substringAfter("script src=\"").substringBefore('"')
-        val scriptUrl = "$baseUrl/$scriptPath"
+        val scriptPath = body.substringAfter("src=\"assets/").substringBefore('"')
+        val scriptUrl = "$baseUrl/assets/$scriptPath"
         val scriptBody = client.newCall(GET(scriptUrl)).execute().body.string()
 
-        val varWithKeyName = KEY_VAR_REGEX.find(scriptBody)
-            ?.groupValues
-            ?.last()
-            ?: Exception(ERROR_MSG_VAR)
+        return when {
+            KEY_FUNC_REGEX.containsMatchIn(scriptBody) -> patchScriptWithFunction(scriptBody) // Sapphire
+            KEY_VAR_REGEX.containsMatchIn(scriptBody) -> patchScriptWithVar(scriptBody) // PinkBird
+            else -> throw Exception() // ????
+        }
+    }
 
-        val varWithKeyBody = scriptBody.substringAfter("var $varWithKeyName=")
+    private fun patchScriptWithVar(script: String): String {
+        val varWithKeyName = KEY_VAR_REGEX.find(script)
+            ?.groupValues
+            ?.lastOrNull()
+            ?: throw Exception()
+
+        val varWithKeyBody = script.substringAfter("var $varWithKeyName=")
             .substringBefore(";")
 
-        return scriptBody.replace(varWithKeyBody, "AESKeyExtractor.setKey($varWithKeyBody)")
+        return script.replace(varWithKeyBody, "AESKeyExtractor.setKey($varWithKeyBody)")
+    }
+
+    private fun patchScriptWithFunction(script: String): String {
+        val (match, functionName) = KEY_FUNC_REGEX.find(script)
+            ?.groupValues
+            ?: throw Exception()
+        val patchedMatch = match.replace(
+            ";function",
+            ";AESKeyExtractor.setKey($functionName().toString());function",
+        )
+        return script.replace(match, patchedMatch)
     }
 }
diff --git a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/KickAssAnimeExtractor.kt b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/KickAssAnimeExtractor.kt
index e875bd890..95cfdfe6e 100644
--- a/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/KickAssAnimeExtractor.kt
+++ b/src/en/kickassanime/src/eu/kanade/tachiyomi/animeextension/en/kickassanime/extractors/KickAssAnimeExtractor.kt
@@ -9,6 +9,7 @@ import eu.kanade.tachiyomi.lib.cryptoaes.CryptoAES.decodeHex
 import eu.kanade.tachiyomi.network.GET
 import kotlinx.serialization.json.Json
 import okhttp3.Headers
+import okhttp3.HttpUrl.Companion.toHttpUrl
 import okhttp3.OkHttpClient
 import okhttp3.Request
 import org.jsoup.Jsoup
@@ -83,7 +84,7 @@ class KickAssAnimeExtractor(
         return when {
             videoObject.hls.isBlank() ->
                 extractVideosFromDash(masterPlaylist, prefix, subtitles)
-            else -> extractVideosFromHLS(masterPlaylist, prefix, subtitles)
+            else -> extractVideosFromHLS(masterPlaylist, prefix, subtitles, videoObject.playlistUrl)
         }
     }
 
@@ -111,15 +112,20 @@ class KickAssAnimeExtractor(
         }
     }
 
-    private fun extractVideosFromHLS(playlist: String, prefix: String, subs: List<Track>): List<Video> {
+    private fun extractVideosFromHLS(playlist: String, prefix: String, subs: List<Track>, playlistUrl: String): List<Video> {
         val separator = "#EXT-X-STREAM-INF"
-        return playlist.substringAfter(separator).split(separator).map {
+        return playlist.substringAfter(separator).split(separator).mapNotNull {
             val resolution = it.substringAfter("RESOLUTION=")
                 .substringBefore("\n")
                 .substringAfter("x")
                 .substringBefore(",") + "p"
 
-            val videoUrl = it.substringAfter("\n").substringBefore("\n")
+            val videoUrl = it.substringAfter("\n").substringBefore("\n").let { url ->
+                when {
+                    url.startsWith("/") -> "https://" + playlistUrl.toHttpUrl().host + url
+                    else -> url
+                }
+            }.ifEmpty { return@mapNotNull null }
 
             Video(videoUrl, "$prefix - $resolution", videoUrl, subtitleTracks = subs)
         }
@@ -143,7 +149,6 @@ class KickAssAnimeExtractor(
     }
 
     // ============================= Utilities ==============================
-
     @SuppressLint("DefaultLocale")
     private fun Element.formatBits(attribute: String = "bandwidth"): String? {
         var bits = attr(attribute).toLongOrNull() ?: 0L