From b236fc2a8448c249f891f04a9ea880283b0bd691 Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Tue, 2 May 2023 09:20:52 -0300
Subject: [PATCH] feat(lib): Convert GdrivePlayerExtractor to shared lib
 (#1573)

---
 lib/gdriveplayer-extractor/build.gradle.kts   | 21 ++++++
 .../GdrivePlayerExtractor.kt                  | 68 +++++++++++++++++++
 2 files changed, 89 insertions(+)
 create mode 100644 lib/gdriveplayer-extractor/build.gradle.kts
 create mode 100644 lib/gdriveplayer-extractor/src/main/java/eu/kanade/tachiyomi/lib/gdriveplayerextractor/GdrivePlayerExtractor.kt

diff --git a/lib/gdriveplayer-extractor/build.gradle.kts b/lib/gdriveplayer-extractor/build.gradle.kts
new file mode 100644
index 000000000..a7df5d0ba
--- /dev/null
+++ b/lib/gdriveplayer-extractor/build.gradle.kts
@@ -0,0 +1,21 @@
+plugins {
+    id("com.android.library")
+    kotlin("android")
+    id("kotlinx-serialization")
+}
+
+android {
+    compileSdk = AndroidConfig.compileSdk
+    namespace = "eu.kanade.tachiyomi.lib.gdriveplayerextractor"
+
+    defaultConfig {
+        minSdk = AndroidConfig.minSdk
+        targetSdk = AndroidConfig.targetSdk
+    }
+}
+
+dependencies {
+    compileOnly(libs.bundles.common)
+    implementation(project(":lib-cryptoaes"))
+    implementation(project(":lib-unpacker"))
+}
diff --git a/lib/gdriveplayer-extractor/src/main/java/eu/kanade/tachiyomi/lib/gdriveplayerextractor/GdrivePlayerExtractor.kt b/lib/gdriveplayer-extractor/src/main/java/eu/kanade/tachiyomi/lib/gdriveplayerextractor/GdrivePlayerExtractor.kt
new file mode 100644
index 000000000..a5012a733
--- /dev/null
+++ b/lib/gdriveplayer-extractor/src/main/java/eu/kanade/tachiyomi/lib/gdriveplayerextractor/GdrivePlayerExtractor.kt
@@ -0,0 +1,68 @@
+package eu.kanade.tachiyomi.lib.gdriveplayerextractor
+
+import eu.kanade.tachiyomi.animesource.model.Track
+import eu.kanade.tachiyomi.animesource.model.Video
+import eu.kanade.tachiyomi.lib.cryptoaes.CryptoAES.decryptWithSalt
+import eu.kanade.tachiyomi.lib.unpacker.Unpacker
+import eu.kanade.tachiyomi.network.GET
+import kotlinx.serialization.decodeFromString
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.JsonObject
+import kotlinx.serialization.json.jsonPrimitive
+import okhttp3.Headers
+import okhttp3.OkHttpClient
+import org.jsoup.Jsoup
+
+class GdrivePlayerExtractor(private val client: OkHttpClient) {
+
+    fun videosFromUrl(url: String, name: String, headers: Headers): List<Video> {
+        val newUrl = url.replace(".us", ".to").replace(".me", ".to")
+        val body = client.newCall(GET(newUrl, headers = headers)).execute()
+            .body.string()
+
+        val subtitleList = Jsoup.parse(body).selectFirst("div:contains(\\.srt)")
+            ?.let { element ->
+                val subUrl = "https://gdriveplayer.to/?subtitle=" + element.text()
+                listOf(Track(subUrl, "Subtitles"))
+            } ?: emptyList()
+
+        val eval = Unpacker.unpack(body).replace("\\", "")
+        val json = Json.decodeFromString<JsonObject>(REGEX_DATAJSON.getFirst(eval))
+        val sojson = REGEX_SOJSON.getFirst(eval)
+            .split(Regex("\\D+"))
+            .joinToString("") {
+                Char(it.toInt()).toString()
+            }
+        val password = REGEX_PASSWORD.getFirst(sojson)
+        val decrypted = decryptAES(password, json) ?: return emptyList()
+
+        val secondEval = Unpacker.unpack(decrypted).replace("\\", "")
+        return REGEX_VIDEOURL.findAll(secondEval)
+            .distinctBy { it.groupValues[2] } // remove duplicates by quality
+            .map {
+                val qualityStr = it.groupValues[2]
+                val quality = "$PLAYER_NAME ${qualityStr}p - $name"
+                val videoUrl = "https:" + it.groupValues[1] + "&res=$qualityStr"
+                Video(videoUrl, quality, videoUrl, subtitleTracks = subtitleList)
+            }.toList()
+    }
+
+    private fun decryptAES(password: String, json: JsonObject): String? {
+        val salt = json["s"]!!.jsonPrimitive.content
+        val ciphertext = json["ct"]!!.jsonPrimitive.content
+        return decryptWithSalt(ciphertext, salt, password)
+    }
+
+    private fun Regex.getFirst(item: String): String {
+        return find(item)?.groups?.elementAt(1)?.value!!
+    }
+
+    companion object {
+        private const val PLAYER_NAME = "GDRIVE"
+
+        private val REGEX_DATAJSON = Regex("data=\"(\\S+?)\";")
+        private val REGEX_PASSWORD = Regex("var pass = \"(\\S+?)\"")
+        private val REGEX_SOJSON = Regex("null,['|\"](\\w+)['|\"]")
+        private val REGEX_VIDEOURL = Regex("file\":\"(\\S+?)\".*?res=(\\d+)")
+    }
+}