cloudstream-extensions-arabic/EgyDeadProvider/src/main/kotlin/com/egydead/EgyDeadProvider.kt

package com.egydead

import com.fasterxml.jackson.annotation.JsonProperty
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.LoadResponse.Companion.addTrailer
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.Qualities
import com.lagradost.cloudstream3.utils.loadExtractor
import org.jsoup.nodes.Element
import java.net.URI

class EgyDead : MainAPI() {
    override var lang = "ar"
    override var mainUrl = "https://tv.egydead.live"
    override var name = "EgyDead"
    override val usesWebView = false
    override val hasMainPage = true
    override val supportedTypes = setOf(TvType.TvSeries, TvType.Movie, TvType.Anime)

    private fun String.getIntFromText(): Int? {
        return Regex("""\d+""").find(this)?.groupValues?.firstOrNull()?.toIntOrNull()
    }
    private fun String.cleanTitle(): String {
        return this.replace("جميع مواسم مسلسل|مترجم كامل|مشاهدة فيلم|مترجم|انمي|الموسم.*|مترجمة كاملة|مسلسل|كاملة".toRegex(), "")
    }
    private fun Element.toSearchResponse(): SearchResponse {
        val title = select("h1.BottomTitle").text().cleanTitle()
        val posterUrl = select("img").attr("src")
        val tvType = if (select("span.cat_name").text().contains("افلام")) TvType.Movie else TvType.TvSeries
        return MovieSearchResponse(
            title,
            select("a").attr("href"),
            this@EgyDead.name,
            tvType,
            posterUrl,
            )
    }

    override val mainPage = mainPageOf(
        "$mainUrl/category/افلام-اجنبي/?page=" to "English Movies",
        "$mainUrl/category/افلام-اسيوية/?page=" to "Asian Movies",
        "$mainUrl/season/?page=" to "Series",
    )

    override suspend fun getMainPage(
        page: Int,
        request: MainPageRequest
    ): HomePageResponse {
        val document = app.get(request.data + page).document
        val home = document.select("li.movieItem").mapNotNull {
            it.toSearchResponse()
        }
        return newHomePageResponse(request.name, home)
    }

    override suspend fun search(query: String): List<SearchResponse> {
        val doc = app.get("$mainUrl/?s=$query").document
        return doc.select("li.movieItem").mapNotNull {
            if(it.select("a").attr("href").contains("/episode/")) return@mapNotNull null
            it.toSearchResponse()
        }
    }

    override suspend fun load(url: String): LoadResponse {
        val doc = app.get(url).document
        val title = doc.select("div.singleTitle em").text().cleanTitle()
        val isMovie = !url.contains("/serie/|/season/".toRegex())

        val posterUrl = doc.select("div.single-thumbnail > img").attr("src")
        val rating = doc.select("a.IMDBRating em").text().getIntFromText()
        val synopsis = doc.select("div.extra-content:contains(القصه) p").text()
        val year = doc.select("ul > li:contains(السنه) > a").text().getIntFromText()
        val tags = doc.select("ul > li:contains(النوع) > a").map { it.text() }
        val recommendations = doc.select("div.related-posts > ul > li").mapNotNull { element ->
            element.toSearchResponse()
        }
        val youtubeTrailer = doc.select("div.popupContent > iframe").attr("src")
        return if (isMovie) {
            newMovieLoadResponse(
                title,
                url,
                TvType.Movie,
                url
            ) {
                this.posterUrl = posterUrl
                this.recommendations = recommendations
                this.plot = synopsis
                this.tags = tags
                this.rating = rating
                this.year = year
                addTrailer(youtubeTrailer)
            }
        } else {
            val seasonList = doc.select("div.seasons-list ul > li > a").reversed()
            val episodes = arrayListOf<Episode>()
            if(seasonList.isNotEmpty()) {
                seasonList.apmapIndexed { index, season ->
                    app.get(
                        season.attr("href"),
                    ).document.select("div.EpsList > li > a").map {
                        episodes.add(Episode(
                            it.attr("href"),
                            it.attr("title"),
                            index+1,
                            it.text().getIntFromText()
                        ))
                    }
                }
            } else {
                doc.select("div.EpsList > li > a").map {
                    episodes.add(Episode(
                        it.attr("href"),
                        it.attr("title"),
                        0,
                        it.text().getIntFromText()
                    ))
                }
            }
            newTvSeriesLoadResponse(title, url, TvType.TvSeries, episodes.distinct().sortedBy { it.episode }) {
                this.posterUrl = posterUrl
                this.tags = tags
                this.plot = synopsis
                this.recommendations = recommendations
                this.rating = rating
                this.year = year
                addTrailer(youtubeTrailer)
            }
        }
    }
    data class LinkBox (
        @JsonProperty("data"   ) var data   : Data?   = Data(),
    )
    data class Data (
        @JsonProperty("rList"     ) var rList     : ArrayList<RList> = arrayListOf(),
    )
    data class RList (
        @JsonProperty("resolution" ) var resolution : String? = null,
        @JsonProperty("size"       ) var size       : Double?    = null,
//        @JsonProperty("sub_type"   ) var subType    : String? = null,
        @JsonProperty("url"        ) var url        : String? = null,

    )
    private fun bytesToHumanReadableSize(bytes: Double) = when {
        bytes >= 1 shl 30 -> "%.1f GB".format(bytes / (1 shl 30))
        bytes >= 1 shl 20 -> "%.1f MB".format(bytes / (1 shl 20))
        bytes >= 1 shl 10 -> "%.0f kB".format(bytes / (1 shl 10))
        else -> "$bytes bytes"
    }
    override suspend fun loadLinks(
        data: String,
        isCasting: Boolean,
        subtitleCallback: (SubtitleFile) -> Unit,
        callback: (ExtractorLink) -> Unit
    ): Boolean {
        val doc = app.post(data, data = mapOf("View" to "1")).document
        val watchList = doc.select("ul.serversList > li")
        watchList.apmap { li ->
            val iframeUrl = li.attr("data-link")
            val quality = li.select("small").text().getIntFromText() ?: Qualities.Unknown.value
            if(iframeUrl.contains("www.linkbox.to")) {
                val apiUrl = "https://" + URI(iframeUrl).host + "/api/open/get_url?itemId=" + iframeUrl.substringAfter("id=").substringBefore("&")
                val json = app.get(apiUrl).parsed<LinkBox>()
                json.data?.rList?.forEach {
                    callback.invoke(
                        ExtractorLink(
                            this.name,
                            "LinkBox " + bytesToHumanReadableSize(it.size ?: 0.0),
                            it.url ?: return@forEach,
                            mainUrl,
                            quality,
                            false
                        )
                    )
                }
            }
            else loadExtractor(iframeUrl, data, subtitleCallback, callback)
        }
        return true
    }
}