From bcf55beb437248180d9449e67ca9abb28023acc6 Mon Sep 17 00:00:00 2001
From: Secozzi <49240133+Secozzi@users.noreply.github.com>
Date: Sat, 14 Oct 2023 12:49:00 +0000
Subject: [PATCH] fix(en/fmovies): Fix vidsrc extractor + vrf helper (#2363)

---
 src/en/fmovies/build.gradle                   |  2 +-
 .../en/fmovies/FMoviesHelper.kt               |  2 +-
 .../en/fmovies/extractors/VidsrcExtractor.kt  | 46 +++++++++++++++----
 src/en/seez/build.gradle                      |  2 +-
 .../animeextension/en/seez/VrfHelper.kt       |  2 +-
 .../en/seez/extractors/VidsrcExtractor.kt     | 44 ++++++++++++++----
 6 files changed, 76 insertions(+), 22 deletions(-)

diff --git a/src/en/fmovies/build.gradle b/src/en/fmovies/build.gradle
index 2f01ebd7c4..3d82c1fa51 100644
--- a/src/en/fmovies/build.gradle
+++ b/src/en/fmovies/build.gradle
@@ -8,7 +8,7 @@ ext {
     extName = 'FMovies'
     pkgNameSuffix = 'en.fmovies'
     extClass = '.FMovies'
-    extVersionCode = 10
+    extVersionCode = 11
     libVersion = '13'
 }
 
diff --git a/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/FMoviesHelper.kt b/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/FMoviesHelper.kt
index 293fb0ec4a..f0e0b045ac 100644
--- a/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/FMoviesHelper.kt
+++ b/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/FMoviesHelper.kt
@@ -39,7 +39,7 @@ class FMoviesHelper(private val client: OkHttpClient, private val headers: Heade
 
     fun getVidSrc(query: String, host: String): String {
         val url = API_URL.newBuilder().apply {
-            addPathSegment(if (host == "vidstream.pro") "rawVizcloud" else "rawMcloud")
+            addPathSegment(if (host.contains("mcloud", true)) "rawMcloud" else "rawVizcloud")
             addQueryParameter("apikey", API_KEY)
         }.build().toString()
 
diff --git a/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/extractors/VidsrcExtractor.kt b/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/extractors/VidsrcExtractor.kt
index 79d3e3a71a..005d6f3758 100644
--- a/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/extractors/VidsrcExtractor.kt
+++ b/src/en/fmovies/src/eu/kanade/tachiyomi/animeextension/en/fmovies/extractors/VidsrcExtractor.kt
@@ -21,19 +21,27 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
     private val playlistUtils by lazy { PlaylistUtils(client, headers) }
 
     fun videosFromUrl(url: String, name: String): List<Video> {
-        val host = when (name) {
-            "Vidplay" -> "vidstream.pro"
-            else -> "mcloud.to"
-        }
+        val httpUrl = url.toHttpUrl()
+        val host = httpUrl.host
         val referer = "https://$host/"
 
-        val httpUrl = url.toHttpUrl()
+        val query = buildString {
+            append(httpUrl.pathSegments.last())
+            append("?")
+            append(
+                httpUrl.queryParameterNames.joinToString("&") {
+                    "$it=${httpUrl.queryParameter(it)}"
+                },
+            )
+        }
 
-        val query = "${httpUrl.pathSegments.last()}?t=${httpUrl.queryParameter("t")!!}"
-        val rawUrl = vrfHelper.getVidSrc(query, host)
+        val rawUrl = vrfHelper.getVidSrc(query, host).addAutoStart()
 
         val refererHeaders = headers.newBuilder().apply {
-            add("Referer", referer)
+            add("Accept", "application/json, text/javascript, */*; q=0.01")
+            add("Host", host)
+            add("Referer", url.addAutoStart())
+            add("X-Requested-With", "XMLHttpRequest")
         }.build()
 
         val infoJson = client.newCall(
@@ -41,18 +49,36 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
         ).execute().parseAs<VidsrcResponse>()
 
         val subtitleList = httpUrl.queryParameter("sub.info")?.let {
+            val subtitlesHeaders = headers.newBuilder().apply {
+                add("Accept", "application/json, text/javascript, */*; q=0.01")
+                add("Host", it.toHttpUrl().host)
+                add("Origin", "https://$host")
+                add("Referer", referer)
+            }.build()
+
             client.newCall(
-                GET(it, headers = refererHeaders),
+                GET(it, headers = subtitlesHeaders),
             ).execute().parseAs<List<FMoviesSubs>>().map {
                 Track(it.file, it.label)
             }
         } ?: emptyList()
 
         return infoJson.result.sources.distinctBy { it.file }.flatMap {
-            playlistUtils.extractFromHls(it.file, subtitleList = subtitleList, referer = referer, videoNameGen = { q -> "$name - $q" })
+            val url = it.file
+                .toHttpUrl()
+                .newBuilder()
+                .fragment(null)
+                .build()
+                .toString()
+
+            playlistUtils.extractFromHls(url, subtitleList = subtitleList, referer = referer, videoNameGen = { q -> "$name - $q" })
         }
     }
 
+    private fun String.addAutoStart(): String {
+        return this.toHttpUrl().newBuilder().setQueryParameter("autostart", "true").build().toString()
+    }
+
     private inline fun <reified T> Response.parseAs(transform: (String) -> String = { it }): T {
         val responseBody = use { transform(it.body.string()) }
         return json.decodeFromString(responseBody)
diff --git a/src/en/seez/build.gradle b/src/en/seez/build.gradle
index 00744d08df..3a20bf6aa8 100644
--- a/src/en/seez/build.gradle
+++ b/src/en/seez/build.gradle
@@ -8,7 +8,7 @@ ext {
     extName = 'Seez'
     pkgNameSuffix = 'en.seez'
     extClass = '.Seez'
-    extVersionCode = 3
+    extVersionCode = 4
     libVersion = '13'
 }
 
diff --git a/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/VrfHelper.kt b/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/VrfHelper.kt
index 5f6e1c1a1d..efd4d95ef0 100644
--- a/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/VrfHelper.kt
+++ b/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/VrfHelper.kt
@@ -27,7 +27,7 @@ class VrfHelper(private val client: OkHttpClient, private val headers: Headers)
 
     fun getVidSrc(query: String, host: String): String {
         val url = API_URL.newBuilder().apply {
-            addPathSegment("rawVizcloud")
+            addPathSegment(if (host.contains("mcloud", true)) "rawMcloud" else "rawVizcloud")
             addQueryParameter("apikey", API_KEY)
         }.build().toString()
 
diff --git a/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/extractors/VidsrcExtractor.kt b/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/extractors/VidsrcExtractor.kt
index 0855ad8e7d..2e4218da10 100644
--- a/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/extractors/VidsrcExtractor.kt
+++ b/src/en/seez/src/eu/kanade/tachiyomi/animeextension/en/seez/extractors/VidsrcExtractor.kt
@@ -21,17 +21,27 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
     private val playlistUtils by lazy { PlaylistUtils(client, headers) }
 
     fun videosFromUrl(url: String, name: String): List<Video> {
-        val host = "vidstream.pro"
-
+        val httpUrl = url.toHttpUrl()
+        val host = httpUrl.host
         val referer = "https://$host/"
 
-        val httpUrl = url.toHttpUrl()
+        val query = buildString {
+            append(httpUrl.pathSegments.last())
+            append("?")
+            append(
+                httpUrl.queryParameterNames.joinToString("&") {
+                    "$it=${httpUrl.queryParameter(it)}"
+                },
+            )
+        }
 
-        val query = "${httpUrl.pathSegments.last()}?t=${httpUrl.queryParameter("t")!!}"
-        val rawUrl = vrfHelper.getVidSrc(query, host)
+        val rawUrl = vrfHelper.getVidSrc(query, host).addAutoStart()
 
         val refererHeaders = headers.newBuilder().apply {
-            add("Referer", referer)
+            add("Accept", "application/json, text/javascript, */*; q=0.01")
+            add("Host", host)
+            add("Referer", url.addAutoStart())
+            add("X-Requested-With", "XMLHttpRequest")
         }.build()
 
         val infoJson = client.newCall(
@@ -39,18 +49,36 @@ class VidsrcExtractor(private val client: OkHttpClient, private val headers: Hea
         ).execute().parseAs<VidsrcResponse>()
 
         val subtitleList = httpUrl.queryParameter("sub.info")?.let {
+            val subtitlesHeaders = headers.newBuilder().apply {
+                add("Accept", "application/json, text/javascript, */*; q=0.01")
+                add("Host", it.toHttpUrl().host)
+                add("Origin", "https://$host")
+                add("Referer", referer)
+            }.build()
+
             client.newCall(
-                GET(it, headers = refererHeaders),
+                GET(it, headers = subtitlesHeaders),
             ).execute().parseAs<List<FMoviesSubs>>().map {
                 Track(it.file, it.label)
             }
         } ?: emptyList()
 
         return infoJson.result.sources.distinctBy { it.file }.flatMap {
-            playlistUtils.extractFromHls(it.file, subtitleList = subtitleList, referer = referer, videoNameGen = { q -> "$name - $q" })
+            val url = it.file
+                .toHttpUrl()
+                .newBuilder()
+                .fragment(null)
+                .build()
+                .toString()
+
+            playlistUtils.extractFromHls(url, subtitleList = subtitleList, referer = referer, videoNameGen = { q -> "$name - $q" })
         }
     }
 
+    private fun String.addAutoStart(): String {
+        return this.toHttpUrl().newBuilder().setQueryParameter("autostart", "true").build().toString()
+    }
+
     private inline fun <reified T> Response.parseAs(transform: (String) -> String = { it }): T {
         val responseBody = use { transform(it.body.string()) }
         return json.decodeFromString(responseBody)