From 5c90af8709e1fe0eb7681680a1147cf4b6912cf8 Mon Sep 17 00:00:00 2001
From: Claudemirovsky <63046606+Claudemirovsky@users.noreply.github.com>
Date: Sun, 3 Sep 2023 07:46:10 -0300
Subject: [PATCH] fix(pt/pobreflix): Fix corner case in extractor (#2134)

---
 .../src/extractors/PainelfxExtractor.kt       | 61 ++++++++++++++-----
 .../multisrc/dooplay/DooPlayGenerator.kt      |  2 +-
 2 files changed, 46 insertions(+), 17 deletions(-)

diff --git a/multisrc/overrides/dooplay/pobreflix/src/extractors/PainelfxExtractor.kt b/multisrc/overrides/dooplay/pobreflix/src/extractors/PainelfxExtractor.kt
index 4edaad879..d597dcc8b 100644
--- a/multisrc/overrides/dooplay/pobreflix/src/extractors/PainelfxExtractor.kt
+++ b/multisrc/overrides/dooplay/pobreflix/src/extractors/PainelfxExtractor.kt
@@ -11,28 +11,57 @@ import okhttp3.OkHttpClient
 
 class PainelfxExtractor(private val client: OkHttpClient) {
     fun videosFromUrl(url: String, headers: Headers): List<Video> {
-        val doc = client.newCall(GET(url)).execute().asJsoup()
+        val docHeaders = headers.newBuilder().set("Referer", "https://gastronomiabrasileira.net/").build()
+        val doc = client.newCall(GET(url, docHeaders)).execute().use { it.asJsoup() }
         val lang = when (url.substringAfterLast("/")) {
             "leg" -> "Legendado"
             else -> "Dublado"
         }
-        return doc.select("div.panel-body > button").flatMap { elem ->
-            val form = FormBody.Builder()
-                .add("idS", elem.attr("idS"))
-                .build()
-            val host = url.toHttpUrl().host
-            val newHeaders = headers.newBuilder().set("Referer", "https://$host/").build()
-            val req = client.newCall(POST("https://$host/CallEpi", body = form)).execute()
-            val decoded = req.body.string().decodeHex().let(::String).replace("\\", "")
-            if (decoded.contains("video\"")) {
-                decoded.substringAfter("video").split("{").drop(1).map {
-                    val videoUrl = it.substringAfter("file\":\"").substringBefore('"')
-                    val quality = it.substringAfter("label\":\"").substringBefore('"')
-                    Video(videoUrl, "$lang - $quality", videoUrl, newHeaders)
+
+        val host = url.toHttpUrl().host
+        val videoHeaders = headers.newBuilder().set("Referer", "https://$host/").build()
+
+        val buttons = doc.select("div.panel-body > button")
+
+        val encodedHexList = when {
+            buttons.isNotEmpty() -> {
+                buttons.map { elem ->
+                    val form = FormBody.Builder()
+                        .add("idS", elem.attr("idS"))
+                        .build()
+                    client.newCall(POST("https://$host/CallEpi", body = form)).execute()
+                        .use { it.body.string() }
                 }
-            } else {
-                emptyList()
             }
+            else -> {
+                val script = doc.selectFirst("script:containsData(idS:)")?.data() ?: return emptyList()
+                val idList = script.split("idS:").drop(1).map { it.substringAfter('"').substringBefore('"') }
+
+                idList.map { idS ->
+                    val form = FormBody.Builder()
+                        .add("id", idS)
+                        .build()
+                    client.newCall(POST("https://$host/CallPlayer", body = form)).execute()
+                        .use { it.body.string() }
+                }
+            }
+        }
+
+        return encodedHexList.flatMap {
+            videosFromHex(it, lang, videoHeaders)
+        }
+    }
+
+    private fun videosFromHex(hex: String, lang: String, videoHeaders: Headers): List<Video> {
+        val decoded = hex.decodeHex().let(::String).replace("\\", "")
+        return if (decoded.contains("video\"")) {
+            decoded.substringAfter("video").split("{").drop(1).map {
+                val videoUrl = it.substringAfter("file\":\"").substringBefore('"')
+                val quality = it.substringAfter("label\":\"").substringBefore('"')
+                Video(videoUrl, "$lang - $quality", videoUrl, videoHeaders)
+            }
+        } else {
+            emptyList()
         }
     }
 
diff --git a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/dooplay/DooPlayGenerator.kt b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/dooplay/DooPlayGenerator.kt
index bcec470d8..fb3e56c7d 100644
--- a/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/dooplay/DooPlayGenerator.kt
+++ b/multisrc/src/main/java/eu/kanade/tachiyomi/multisrc/dooplay/DooPlayGenerator.kt
@@ -27,7 +27,7 @@ class DooPlayGenerator : ThemeSourceGenerator {
         SingleLang("Multimovies", "https://multimovies.shop", "en", isNsfw = false, overrideVersionCode = 9),
         SingleLang("pactedanime", "https://pactedanime.com", "en", isNsfw = false, className = "PactedAnime", overrideVersionCode = 4),
         SingleLang("Pi Fansubs", "https://pifansubs.org", "pt-BR", isNsfw = true, overrideVersionCode = 17),
-        SingleLang("Pobreflix", "https://pobreflix.biz", "pt-BR", isNsfw = true),
+        SingleLang("Pobreflix", "https://pobreflix.biz", "pt-BR", isNsfw = true, overrideVersionCode = 1),
         SingleLang("UniqueStream", "https://uniquestream.net", "en", isNsfw = false, overrideVersionCode = 2),
     )