From d651b419bc76f385bc3d4988014b1e521c5ce49c Mon Sep 17 00:00:00 2001
From: Secozzi <49240133+Secozzi@users.noreply.github.com>
Date: Sun, 5 Feb 2023 01:20:13 +0100
Subject: [PATCH] Update video extraction, add extractor for Cdope (#1234)

---
 src/fr/animevostfr/build.gradle               |  2 +-
 .../fr/animevostfr/AnimeVostFr.kt             | 79 ++++++++++++++-----
 .../animevostfr/extractors/CdopeExtractor.kt  | 62 +++++++++++++++
 3 files changed, 121 insertions(+), 22 deletions(-)
 create mode 100644 src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/extractors/CdopeExtractor.kt

diff --git a/src/fr/animevostfr/build.gradle b/src/fr/animevostfr/build.gradle
index 0642f1947..ba4550577 100644
--- a/src/fr/animevostfr/build.gradle
+++ b/src/fr/animevostfr/build.gradle
@@ -6,7 +6,7 @@ ext {
     extName = 'AnimeVostFr'
     pkgNameSuffix = 'fr.animevostfr'
     extClass = '.AnimeVostFr'
-    extVersionCode = 1
+    extVersionCode = 2
     libVersion = '13'
 }
 
diff --git a/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/AnimeVostFr.kt b/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/AnimeVostFr.kt
index b6feb9977..ef1cc10f6 100644
--- a/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/AnimeVostFr.kt
+++ b/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/AnimeVostFr.kt
@@ -5,6 +5,7 @@ import android.content.SharedPreferences
 import android.util.Log
 import androidx.preference.ListPreference
 import androidx.preference.PreferenceScreen
+import eu.kanade.tachiyomi.animeextension.fr.animevostfr.extractors.CdopeExtractor
 import eu.kanade.tachiyomi.animesource.ConfigurableAnimeSource
 import eu.kanade.tachiyomi.animesource.model.AnimeFilter
 import eu.kanade.tachiyomi.animesource.model.AnimeFilterList
@@ -132,7 +133,7 @@ class AnimeVostFr : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
         return if (type == "MOVIE") {
             return listOf(
                 SEpisode.create().apply {
-                    url = response.request.url.toString() + "?server=download"
+                    url = response.request.url.toString()
                     name = "Movie"
                 }
             )
@@ -148,42 +149,78 @@ class AnimeVostFr : ConfigurableAnimeSource, ParsedAnimeHttpSource() {
             .substringAfterLast("-episode-")
             .substringBefore("-")
         return SEpisode.create().apply {
-            setUrlWithoutDomain(element.attr("href") + "?server=download")
+            setUrlWithoutDomain(element.attr("href"))
             name = "Épisode $number"
             episode_number = number.toFloat()
         }
     }
 
     override fun fetchVideoList(episode: SEpisode): Observable<List<Video>> {
-        val response = client.newCall(GET(episode.url)).execute()
+        val videoList = mutableListOf<Video>()
+
+        val url = if (episode.url.startsWith("https:")) {
+            episode.url
+        } else {
+            baseUrl + episode.url
+        }
+
+        val response = client.newCall(GET(url)).execute()
         val parsedResponse = response.asJsoup()
         if (parsedResponse.select("title").text().contains("Warning"))
             throw Exception(parsedResponse.select("body").text())
         val epId = parsedResponse.select("link[rel=shortlink]").attr("href")
             .substringAfter("?p=")
+
+        parsedResponse.select("div.list-server > select > option").forEach { server ->
+            videoList.addAll(
+                extractVideos(
+                    server.attr("value"),
+                    server.text(),
+                    epId
+                )
+            )
+        }
+
+        return Observable.just(videoList)
+    }
+
+    private fun extractVideos(serverValue: String, serverName: String, epId: String): List<Video> {
         Log.i("bruh", "ID: $epId \nLink: $")
         val xhr = Headers.headersOf("x-requested-with", "XMLHttpRequest")
-        val epLink = client.newCall(GET("$baseUrl/ajax-get-link-stream/?server=download&filmId=$epId", xhr))
+        val epLink = client.newCall(GET("$baseUrl/ajax-get-link-stream/?server=$serverValue&filmId=$epId", xhr))
             .execute().body!!.string()
-        val playlistInterceptor = CloudFlareInterceptor()
-        val cfClient = client.newBuilder().addInterceptor(playlistInterceptor).build()
-        val headers = Headers.headersOf(
-            "referer", "$baseUrl/",
-            "user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36"
-        )
-        val playlistResponse = cfClient.newCall(GET(epLink, headers)).execute().body!!.string()
-        val headersVideo = Headers.headersOf(
-            "referer", epLink,
-            "user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36"
-        )
+
         val playlist = mutableListOf<Video>()
-        playlistResponse.substringAfter("#EXT-X-STREAM-INF:")
-            .split("#EXT-X-STREAM-INF:").map {
-                val quality = it.substringAfter("RESOLUTION=").split(",")[0].split("\n")[0].substringAfter("x") + "p"
-                val videoUrl = it.substringAfter("\n").substringBefore("\n")
-                playlist.add(Video(videoUrl, quality, videoUrl, headers = headersVideo))
+        when {
+            epLink.contains("comedyshow.to") -> {
+                val playlistInterceptor = CloudFlareInterceptor()
+                val cfClient = client.newBuilder().addInterceptor(playlistInterceptor).build()
+                val headers = Headers.headersOf(
+                    "referer", "$baseUrl/",
+                    "user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36"
+                )
+                val playlistResponse = cfClient.newCall(GET(epLink, headers)).execute().body!!.string()
+                val headersVideo = Headers.headersOf(
+                    "referer", epLink,
+                    "user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36"
+                )
+
+                playlistResponse.substringAfter("#EXT-X-STREAM-INF:")
+                    .split("#EXT-X-STREAM-INF:").map {
+                        val quality = it.substringAfter("RESOLUTION=").split(",")[0].split("\n")[0].substringAfter("x") + "p ($serverName)"
+                        val videoUrl = it.substringAfter("\n").substringBefore("\n")
+                        playlist.add(Video(videoUrl, quality, videoUrl, headers = headersVideo))
+                    }
             }
-        return Observable.just(playlist.sort())
+            epLink.contains("cdopetimes.xyz") -> {
+                val extractor = CdopeExtractor(client)
+                playlist.addAll(
+                    extractor.videosFromUrl(epLink)
+                )
+            }
+        }
+
+        return playlist.sort()
     }
 
     override fun videoListSelector() = throw Exception("not used")
diff --git a/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/extractors/CdopeExtractor.kt b/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/extractors/CdopeExtractor.kt
new file mode 100644
index 000000000..f5b3f0042
--- /dev/null
+++ b/src/fr/animevostfr/src/eu/kanade/tachiyomi/animeextension/fr/animevostfr/extractors/CdopeExtractor.kt
@@ -0,0 +1,62 @@
+package eu.kanade.tachiyomi.animeextension.fr.animevostfr.extractors
+
+import eu.kanade.tachiyomi.animesource.model.Video
+import eu.kanade.tachiyomi.network.POST
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.decodeFromString
+import kotlinx.serialization.json.Json
+import okhttp3.Headers
+import okhttp3.MediaType.Companion.toMediaType
+import okhttp3.OkHttpClient
+import okhttp3.RequestBody.Companion.toRequestBody
+
+@Serializable
+data class CdopeResponse(
+    val data: List<FileObject>
+) {
+    @Serializable
+    data class FileObject(
+        val file: String,
+        val label: String,
+        val type: String
+    )
+}
+
+class CdopeExtractor(private val client: OkHttpClient) {
+    fun videosFromUrl(url: String): List<Video> {
+        val videoList = mutableListOf<Video>()
+        val id = url.substringAfter("/v/")
+        val body = "r=&d=cdopetimes.xyz".toRequestBody("application/x-www-form-urlencoded".toMediaType())
+        val headers = Headers.headersOf(
+            "Accept", "*/*",
+            "Content-Type", "application/x-www-form-urlencoded; charset=UTF-8",
+            "Host", "cdopetimes.xyz",
+            "Origin", "https://cdopetimes.xyz",
+            "Referer", url,
+            "User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/109.0",
+            "X-Requested-With", "XMLHttpRequest"
+        )
+        val response = client.newCall(
+            POST("https://cdopetimes.xyz/api/source/$id", body = body, headers = headers)
+        ).execute()
+
+        Json { ignoreUnknownKeys = true }.decodeFromString<CdopeResponse>(response.body!!.string()).data.forEach { file ->
+            val videoHeaders = Headers.headersOf(
+                "Accept", "video/webm,video/ogg,video/*;q=0.9,application/ogg;q=0.7,audio/*;q=0.6,*/*;q=0.5",
+                "Referer", "https://cdopetimes.xyz/",
+                "User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/109.0",
+            )
+
+            videoList.add(
+                Video(
+                    file.file,
+                    "${file.label} (Cdope - ${file.type})",
+                    file.file,
+                    headers = videoHeaders
+                )
+            )
+        }
+
+        return videoList
+    }
+}