From 360af2a277a435ec9c99a21e2c00aef40a0004f2 Mon Sep 17 00:00:00 2001
From: hexated <hexated@gmail.com>
Date: Sun, 6 Nov 2022 12:45:43 +0700
Subject: [PATCH] fixed missing channel in TimefourTv

---
 TimefourTv/build.gradle.kts                   |  2 +-
 .../src/main/kotlin/com/hexated/TimefourTv.kt | 66 +++++++++++++++++--
 .../kotlin/com/hexated/TimefourTvExtractor.kt |  2 +-
 3 files changed, 63 insertions(+), 7 deletions(-)

diff --git a/TimefourTv/build.gradle.kts b/TimefourTv/build.gradle.kts
index badb97c4..5c62e04d 100644
--- a/TimefourTv/build.gradle.kts
+++ b/TimefourTv/build.gradle.kts
@@ -1,5 +1,5 @@
 // use an integer for version numbers
-version = 4
+version = 5
 
 
 cloudstream {
diff --git a/TimefourTv/src/main/kotlin/com/hexated/TimefourTv.kt b/TimefourTv/src/main/kotlin/com/hexated/TimefourTv.kt
index 7cc15433..1a1f4ef1 100644
--- a/TimefourTv/src/main/kotlin/com/hexated/TimefourTv.kt
+++ b/TimefourTv/src/main/kotlin/com/hexated/TimefourTv.kt
@@ -2,6 +2,8 @@ package com.hexated
 
 import com.hexated.TimefourTvExtractor.getLink
 import com.lagradost.cloudstream3.*
+import com.lagradost.cloudstream3.utils.AppUtils
+import com.lagradost.cloudstream3.utils.AppUtils.toJson
 import com.lagradost.cloudstream3.utils.ExtractorLink
 import com.lagradost.cloudstream3.utils.Qualities
 import org.jsoup.nodes.Element
@@ -11,6 +13,7 @@ open class TimefourTv : MainAPI() {
     override var name = "Time4tv"
     override val hasDownloadSupport = false
     override val hasMainPage = true
+    private val time4tvPoster = "$mainUrl/images/logo.png"
     override val supportedTypes = setOf(
         TvType.Live
     )
@@ -22,6 +25,7 @@ open class TimefourTv : MainAPI() {
         "$mainUrl/sports-channels" to "Sport Channels",
         "$mainUrl/live-sports-streams" to "Live Sport Channels",
         "$mainUrl/news-channels" to "News Channels",
+        "$mainUrl/schedule.php" to "Schedule",
     )
 
     override suspend fun getMainPage(
@@ -49,9 +53,27 @@ open class TimefourTv : MainAPI() {
             items.add(HomePageList(request.name, home, true))
         }
 
+        if (nonPaged && request.name == "Schedule") {
+            val res = app.get(request.data).document
+            val schedule = res.select("div.search_p h2").mapNotNull {
+                it.toSearchSchedule()
+            }
+            items.add(HomePageList(request.name, schedule, true))
+        }
+
         return newHomePageResponse(items)
     }
 
+    private fun Element.toSearchSchedule(): LiveSearchResponse? {
+        return LiveSearchResponse(
+            this.text() ?: return null,
+            Schedule(this.text()).toJson(),
+            this@TimefourTv.name,
+            TvType.Live,
+            posterUrl = time4tvPoster
+        )
+    }
+
     private fun Element.toSearchResult(): LiveSearchResponse? {
         return LiveSearchResponse(
             this.selectFirst("div.channelName")?.text() ?: return null,
@@ -63,8 +85,34 @@ open class TimefourTv : MainAPI() {
 
     }
 
+    private suspend fun loadSchedule(url: String): LoadResponse {
+        val doc = app.get("$mainUrl/schedule.php").document
+
+        val episode =
+            doc.selectFirst("div.search_p h2:contains($url)")?.nextElementSibling()?.select("span")
+                ?.mapIndexedNotNull { index, ele ->
+                    val title = ele.select("a").text()
+                    val href = ele.select("a").attr("href")
+                    val desc = ele.parent()?.textNodes()?.get(index).toString()
+                    Episode(
+                        href,
+                        title,
+                        posterUrl = time4tvPoster,
+                        description = desc,
+                    )
+                } ?: throw ErrorLoadingException("Referest Page")
+
+        return newTvSeriesLoadResponse(url, url, TvType.TvSeries, episode) {
+        }
+    }
+
     override suspend fun load(url: String): LoadResponse? {
-        val document = app.get(url).document
+        val data = AppUtils.parseJson<Schedule>(url)
+
+        if (data.name?.startsWith("http") == false) return loadSchedule(data.name)
+
+        val res = app.get(url)
+        val document = res.document
         val title = document.selectFirst("div.channelHeading h1")?.text() ?: return null
         val poster =
             fixUrlNull(document.selectFirst("meta[property=\"og:image\"]")?.attr("content"))
@@ -104,10 +152,14 @@ open class TimefourTv : MainAPI() {
         callback: (ExtractorLink) -> Unit
     ): Boolean {
 
-        val link = if (data.startsWith(mainUrl)) {
-            app.get(data, allowRedirects = false).document.selectFirst("iframe")?.attr("src")
-        } else {
-            data
+        val link = when {
+            data.contains("/channel") -> app.get(data).document.selectFirst("div.tv_palyer iframe")?.attr("src")
+            data.startsWith(mainUrl) -> {
+                app.get(data, allowRedirects = false).document.selectFirst("iframe")?.attr("src")
+            }
+            else -> {
+                data
+            }
         } ?: throw ErrorLoadingException()
         getLink(fixUrl(link))?.let { m3uLink ->
             callback.invoke(
@@ -124,4 +176,8 @@ open class TimefourTv : MainAPI() {
         return true
     }
 
+    data class Schedule(
+        val name: String? = null
+    )
+
 }
\ No newline at end of file
diff --git a/TimefourTv/src/main/kotlin/com/hexated/TimefourTvExtractor.kt b/TimefourTv/src/main/kotlin/com/hexated/TimefourTvExtractor.kt
index 85ace2bd..18973922 100644
--- a/TimefourTv/src/main/kotlin/com/hexated/TimefourTvExtractor.kt
+++ b/TimefourTv/src/main/kotlin/com/hexated/TimefourTvExtractor.kt
@@ -71,7 +71,7 @@ object TimefourTvExtractor : TimefourTv() {
             return getSportLink(url)
         }
 
-        val (channel, iframe) = if (url.contains("width=")) {
+        val (channel, iframe) = if (url.contains("width=") || url.contains("/link")) {
             val doc = app.get(url, referer = "$mainUrl/").document
             val tempIframe = doc.selectFirst("iframe")?.attr("src") ?: return null