From 8f70751e123484d49ac026d1ef4f0dfe071ca767 Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Sun, 21 Jan 2024 12:41:50 -0300
Subject: [PATCH] fix(lib/sendvid-extractor): Don't try parsing a .mp4 as a
 .m3u8 file (#2803)

---
 lib/sendvid-extractor/build.gradle.kts        |  1 +
 .../lib/sendvidextractor/SendvidExtractor.kt  | 44 ++++++-------------
 2 files changed, 15 insertions(+), 30 deletions(-)

diff --git a/lib/sendvid-extractor/build.gradle.kts b/lib/sendvid-extractor/build.gradle.kts
index 64448bbeb..ae77b397d 100644
--- a/lib/sendvid-extractor/build.gradle.kts
+++ b/lib/sendvid-extractor/build.gradle.kts
@@ -14,4 +14,5 @@ android {
 
 dependencies {
     compileOnly(libs.bundles.common)
+    implementation(project(":lib-playlist-utils"))
 }
diff --git a/lib/sendvid-extractor/src/main/java/eu/kanade/tachiyomi/lib/sendvidextractor/SendvidExtractor.kt b/lib/sendvid-extractor/src/main/java/eu/kanade/tachiyomi/lib/sendvidextractor/SendvidExtractor.kt
index e8d29c391..53a323ba4 100644
--- a/lib/sendvid-extractor/src/main/java/eu/kanade/tachiyomi/lib/sendvidextractor/SendvidExtractor.kt
+++ b/lib/sendvid-extractor/src/main/java/eu/kanade/tachiyomi/lib/sendvidextractor/SendvidExtractor.kt
@@ -1,6 +1,7 @@
 package eu.kanade.tachiyomi.lib.sendvidextractor
 
 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
@@ -8,38 +9,21 @@ import okhttp3.HttpUrl.Companion.toHttpUrl
 import okhttp3.OkHttpClient
 
 class SendvidExtractor(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 videoList = mutableListOf<Video>()
-        val document = client.newCall(GET(url)).execute().asJsoup()
+        val document = client.newCall(GET(url, headers)).execute().use { it.asJsoup() }
         val masterUrl = document.selectFirst("source#video_source")?.attr("src") ?: return emptyList()
 
-        val masterHeaders = headers.newBuilder()
-            .add("Accept", "*/*")
-            .add("Host", masterUrl.toHttpUrl().host)
-            .add("Origin", "https://${url.toHttpUrl().host}")
-            .add("Referer", "https://${url.toHttpUrl().host}/")
-            .build()
-        val masterPlaylist = client.newCall(
-            GET(masterUrl, headers = masterHeaders),
-        ).execute().body.string()
-
-        val masterBase = "https://${masterUrl.toHttpUrl().host}${masterUrl.toHttpUrl().encodedPath}"
-            .substringBeforeLast("/") + "/"
-
-        masterPlaylist.substringAfter("#EXT-X-STREAM-INF:").split("#EXT-X-STREAM-INF:")
-            .forEach {
-                val quality = "Sendvid:" + it.substringAfter("RESOLUTION=").substringAfter("x").substringBefore(",") + "p "
-                val videoUrl = masterBase + it.substringAfter("\n").substringBefore("\n")
-
-                val videoHeaders = headers.newBuilder()
-                    .add("Accept", "*/*")
-                    .add("Host", videoUrl.toHttpUrl().host)
-                    .add("Origin", "https://${url.toHttpUrl().host}")
-                    .add("Referer", "https://${url.toHttpUrl().host}/")
-                    .build()
-
-                videoList.add(Video(videoUrl, prefix + quality, videoUrl, headers = videoHeaders))
-            }
-        return videoList
+        return if (masterUrl.contains(".m3u8")) {
+            playlistUtils.extractFromHls(masterUrl, url, videoNameGen = { prefix + "Sendvid:$it" })
+        } else {
+            val httpUrl = "https://${url.toHttpUrl()}"
+            val newHeaders = headers.newBuilder()
+                .set("Origin", httpUrl)
+                .set("Referer", "$httpUrl/")
+                .build()
+            listOf(Video(masterUrl, prefix + "Sendvid:default", masterUrl, newHeaders))
+        }
     }
 }