From ee6aed3db3c78e2f592d13bc417ee28637ca86ee Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Sat, 17 Jun 2023 18:48:12 +0000
Subject: [PATCH] feat(lib): New lib: Mp4uploadExtractor (#1753)

---
 lib/mp4upload-extractor/build.gradle.kts      | 20 +++++++++++
 .../mp4uploadextractor/Mp4uploadExtractor.kt  | 36 +++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 lib/mp4upload-extractor/build.gradle.kts
 create mode 100644 lib/mp4upload-extractor/src/main/java/eu/kanade/tachiyomi/lib/mp4uploadextractor/Mp4uploadExtractor.kt

diff --git a/lib/mp4upload-extractor/build.gradle.kts b/lib/mp4upload-extractor/build.gradle.kts
new file mode 100644
index 000000000..e90bb0294
--- /dev/null
+++ b/lib/mp4upload-extractor/build.gradle.kts
@@ -0,0 +1,20 @@
+plugins {
+    id("com.android.library")
+    kotlin("android")
+}
+
+android {
+    compileSdk = AndroidConfig.compileSdk
+    namespace = "eu.kanade.tachiyomi.lib.mp4uploadextractor"
+
+    defaultConfig {
+        minSdk = AndroidConfig.minSdk
+    }
+}
+
+dependencies {
+    implementation("dev.datlag.jsunpacker:jsunpacker:1.0.1") {
+        exclude(group = "org.jetbrains.kotlin", module = "kotlin-stdlib-jdk8")
+    }
+    compileOnly(libs.bundles.common)
+}
diff --git a/lib/mp4upload-extractor/src/main/java/eu/kanade/tachiyomi/lib/mp4uploadextractor/Mp4uploadExtractor.kt b/lib/mp4upload-extractor/src/main/java/eu/kanade/tachiyomi/lib/mp4uploadextractor/Mp4uploadExtractor.kt
new file mode 100644
index 000000000..aebd186f9
--- /dev/null
+++ b/lib/mp4upload-extractor/src/main/java/eu/kanade/tachiyomi/lib/mp4uploadextractor/Mp4uploadExtractor.kt
@@ -0,0 +1,36 @@
+package eu.kanade.tachiyomi.lib.mp4uploadextractor
+
+import eu.kanade.tachiyomi.animesource.model.Video
+import eu.kanade.tachiyomi.network.GET
+import eu.kanade.tachiyomi.util.asJsoup
+import okhttp3.Headers
+import okhttp3.OkHttpClient
+import dev.datlag.jsunpacker.JsUnpacker
+
+class Mp4uploadExtractor(private val client: OkHttpClient) {
+    fun videosFromUrl(url: String, headers: Headers, prefix: String = "", suffix: String = ""): List<Video> {
+        val newHeaders = headers.newBuilder()
+            .add("referer", REFERER)
+            .build()
+
+        val doc = client.newCall(GET(url, newHeaders)).execute().use { it.asJsoup() }
+
+        val script = doc.selectFirst("script:containsData(eval):containsData(p,a,c,k,e,d)")?.data()
+            ?.let(JsUnpacker::unpackAndCombine)
+            ?: doc.selectFirst("script:containsData(player.src)")?.data()
+            ?: return emptyList()
+
+        val videoUrl = script.substringAfter(".src(").substringBefore(")")
+            .substringAfter("src:").substringAfter('"').substringBefore('"')
+
+        val resolution = QUALITY_REGEX.find(script)?.groupValues?.let { "${it[1]}p" } ?: "Unknown resolution"
+        val quality = "${prefix}Mp4Upload - $resolution$suffix"
+
+        return listOf(Video(videoUrl, quality, videoUrl, newHeaders))
+    }
+
+    companion object {
+        private val QUALITY_REGEX by lazy { """\WHEIGHT=(\d+)""".toRegex() }
+        private const val REFERER = "https://mp4upload.com/"
+    }
+}