From 4f4d9f63de4ca96e8ef11ec871398a615e342b00 Mon Sep 17 00:00:00 2001
From: sora <sora@gmail.com>
Date: Wed, 26 Jul 2023 21:16:46 +0700
Subject: [PATCH] Nodrakorid: fix missing episode

---
 Gomov/build.gradle.kts                        |   2 +-
 .../src/main/kotlin/com/hexated/Nodrakorid.kt | 122 +++++++++++++++++-
 2 files changed, 122 insertions(+), 2 deletions(-)

diff --git a/Gomov/build.gradle.kts b/Gomov/build.gradle.kts
index 326a8138..a14f2a82 100644
--- a/Gomov/build.gradle.kts
+++ b/Gomov/build.gradle.kts
@@ -1,5 +1,5 @@
 // use an integer for version numbers
-version = 3
+version = 4
 
 
 cloudstream {
diff --git a/Gomov/src/main/kotlin/com/hexated/Nodrakorid.kt b/Gomov/src/main/kotlin/com/hexated/Nodrakorid.kt
index 1e383927..c6c119d1 100644
--- a/Gomov/src/main/kotlin/com/hexated/Nodrakorid.kt
+++ b/Gomov/src/main/kotlin/com/hexated/Nodrakorid.kt
@@ -1,6 +1,12 @@
 package com.hexated
 
-import com.lagradost.cloudstream3.mainPageOf
+import com.fasterxml.jackson.annotation.JsonProperty
+import com.lagradost.cloudstream3.*
+import com.lagradost.cloudstream3.utils.*
+import com.lagradost.cloudstream3.utils.AppUtils.toJson
+import com.lagradost.cloudstream3.utils.AppUtils.tryParseJson
+import org.jsoup.nodes.Element
+import java.net.URI
 
 class Nodrakorid : DutaMovie() {
     override var mainUrl = "https://no-drak-or.xyz"
@@ -13,4 +19,118 @@ class Nodrakorid : DutaMovie() {
         "genre/c-drama/c-drama-c-drama/page/%d/" to "Drama China",
         "genre/thai-drama/page/%d/" to "Drama Thailand",
     )
+
+    override suspend fun load(url: String): LoadResponse {
+        return super.load(url).apply {
+            when (this) {
+                is TvSeriesLoadResponse -> {
+                    val doc = app.get(url).document
+                    this.comingSoon = false
+                    this.episodes = doc.select("div.entry-content p:contains(Episode)").distinctBy {
+                        it.text()
+                    }.map { eps ->
+                        val num = eps.text()
+                        val endSibling = eps.nextElementSiblings().select("p:contains(Episode)").firstOrNull() ?: eps.nextElementSiblings().select("div.content-moviedata").firstOrNull()
+                        val siblings = eps.nextElementSiblingsUntil(endSibling).map { ele ->
+                            ele.ownText().filter { it.isDigit() }.toIntOrNull() to ele.select("a")
+                                .map { it.attr("href") to it.text() }
+                        }.filter { it.first != null }
+                        Episode(siblings.toJson(), episode = num.filter { it.isDigit() }.toIntOrNull())
+                    }
+                }
+            }
+        }
+    }
+
+    override suspend fun loadLinks(
+        data: String,
+        isCasting: Boolean,
+        subtitleCallback: (SubtitleFile) -> Unit,
+        callback: (ExtractorLink) -> Unit
+    ): Boolean {
+        return if (data.startsWith("[")) {
+            tryParseJson<ArrayList<LinkData>>(data)?.filter { it.first != 360 }?.map {
+                it.second.apmap { link ->
+                    loadFixedExtractor(
+                        fixEmbed(link.first, link.second),
+                        it.first,
+                        "$mainUrl/",
+                        subtitleCallback,
+                        callback
+                    )
+                }
+            }
+            true
+        } else {
+            super.loadLinks(data, isCasting, subtitleCallback, callback)
+        }
+    }
+
+    private fun fixEmbed(url: String, server: String): String {
+        return when {
+            server.contains("streamsb", true) -> {
+                val host = getBaseUrl(url)
+                url.replace("$host/", "$host/e/")
+            }
+
+            server.contains("hxfile", true) -> {
+                val host = getBaseUrl(url)
+                val id = url.substringAfterLast("/")
+                "$host/embed-$id.html"
+            }
+
+            else -> url
+        }
+    }
+
+    private fun getBaseUrl(url: String): String {
+        return URI(url).let {
+            "${it.scheme}://${it.host}"
+        }
+    }
+
+    private suspend fun loadFixedExtractor(
+        url: String,
+        quality: Int?,
+        referer: String? = null,
+        subtitleCallback: (SubtitleFile) -> Unit,
+        callback: (ExtractorLink) -> Unit
+    ) {
+        loadExtractor(url, referer, subtitleCallback) { link ->
+            callback.invoke(
+                ExtractorLink(
+                    link.name,
+                    link.name,
+                    link.url,
+                    link.referer,
+                    if(link.isM3u8) link.quality else quality ?: Qualities.Unknown.value,
+                    link.isM3u8,
+                    link.headers,
+                    link.extractorData
+                )
+            )
+        }
+    }
+
+    private fun Element.nextElementSiblingsUntil(untilElement: Element?): List<Element> {
+        val siblings = mutableListOf<Element>()
+        var currentElement = this.nextElementSibling()
+
+        while (currentElement != null && currentElement != untilElement) {
+            siblings.add(currentElement)
+            currentElement = currentElement.nextElementSibling()
+        }
+
+        return siblings
+    }
+
+    data class LinkData(
+        @JsonProperty("first") var first: Int? = null,
+        @JsonProperty("second") var second: ArrayList<Second> = arrayListOf()
+    )
+
+    data class Second(
+        @JsonProperty("first") var first: String,
+        @JsonProperty("second") var second: String
+    )
 }
\ No newline at end of file