package com.lagradost.cloudstream3.movieproviders

import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.loadExtractor
import org.jsoup.Jsoup
import org.jsoup.nodes.Element
import java.util.*

class LayarKacaProvider : MainAPI() {
    override var mainUrl = "https://149.56.24.226"
    override var name = "LayarKaca"
    override val hasMainPage = true
    override val lang = "id"
    override val hasDownloadSupport = true
    override val supportedTypes = setOf(
        TvType.Movie,
        TvType.TvSeries,
        TvType.AsianDrama
    )

    override suspend fun getMainPage(): HomePageResponse {
        val document = app.get(mainUrl).document

        val homePageList = ArrayList<HomePageList>()

        document.select("section.hot-block,section#newseries").forEach { block ->
            val header = block.select("footer.load-more > a").text().trim()
            val items = block.select("div.slider-item").mapNotNull {
                it.toTopSearchResult()
            }
            if (items.isNotEmpty()) homePageList.add(HomePageList(header, items))
        }

        document.select("div#newest").forEach { block ->
            val header = block.select(".header > h2 > a").text()
            val items = block.select("div.item").mapNotNull {
                it.toMainSearchResult()
            }
            if (items.isNotEmpty()) homePageList.add(HomePageList(header, items))
        }

        document.select("section#recomendation,section#populer,section#seriespopuler")
            .forEach { block ->
                val header = block.select(".header > h2 > a").text()
                val items = block.select("div.item").mapNotNull {
                    it.toBottomSearchResult()
                }
                if (items.isNotEmpty()) homePageList.add(HomePageList(header, items))
            }

        return HomePageResponse(homePageList)
    }

    private fun Element.toTopSearchResult(): SearchResponse {
        val title = this.selectFirst("h3.caption")!!.text().trim()
        val href = this.selectFirst("a")!!.attr("href")
        val posterUrl = fixUrl(this.selectFirst("a > img")?.attr("src").toString())
        val type =
            if (this.select("div.quality-top").isNotEmpty()) TvType.Movie else TvType.TvSeries
        return if (type == TvType.Movie) {
            val quality = getQualityFromString(this.select("div.quality-top").text().trim())
            return newMovieSearchResponse(title, href, TvType.Movie) {
                this.posterUrl = posterUrl
                this.quality = quality
            }
        } else {
            val episode = this.select("div.last-episode > span").text().toIntOrNull()
            newAnimeSearchResponse(title, href, TvType.TvSeries) {
                this.posterUrl = posterUrl
                addDubStatus(dubExist = false, subExist = true, subEpisodes = episode)
            }
        }

    }

    private fun Element.toMainSearchResult(): SearchResponse {
        val title = this.selectFirst("h3.caption")!!.text().trim()
        val href = this.selectFirst("a")!!.attr("href")
        val posterUrl = fixUrl(this.select("a > img").attr("src").toString())
        val quality = getQualityFromString(this.select("div.quality-top").text().trim())
        return newMovieSearchResponse(title, href, TvType.Movie) {
            this.posterUrl = posterUrl
            this.quality = quality
        }

    }

    private fun Element.toBottomSearchResult(): SearchResponse {
        val title = this.selectFirst("h1.grid-title > a")!!.ownText().trim()
        val href = this.selectFirst("h1.grid-title > a")!!.attr("href")
        val posterUrl = fixUrl(this.selectFirst(".grid-poster > a > img")?.attr("src").toString())
        val type = if (this.select("div.quality").isNotEmpty()) TvType.Movie else TvType.TvSeries
        return if (type == TvType.Movie) {
            val quality = getQualityFromString(this.select("div.quality").text().trim())
            return newMovieSearchResponse(title, href, TvType.Movie) {
                this.posterUrl = posterUrl
                this.quality = quality
            }
        } else {
            newTvSeriesSearchResponse(title, href, TvType.TvSeries) {
                this.posterUrl = posterUrl
            }
        }

    }

    override suspend fun search(query: String): List<SearchResponse> {
        val link = "$mainUrl/?s=$query"
        val document = app.get(link).document

        return document.select("div.search-item").map {
            val title = it.selectFirst("h2 > a")!!.text().trim()
            val href = it.selectFirst("h2 > a")!!.attr("href")
            val posterUrl = fixUrl(it.selectFirst("img.img-thumbnail")?.attr("src").toString())
            newTvSeriesSearchResponse(title, href, TvType.TvSeries) {
                this.posterUrl = posterUrl
            }
        }
    }

    override suspend fun load(url: String): LoadResponse {
        val document = app.get(url).document

        val title = document.selectFirst("li.last > span[itemprop=name]")?.text()?.trim().toString()
        val poster = fixUrl(document.select("img.img-thumbnail").attr("src").toString())
        val tags = document.select("div.content > div:nth-child(5) > h3 > a").map { it.text() }

        val year = Regex("\\d, (\\d+)").find(
            document.select("div.content > div:nth-child(7) > h3").text().trim()
        )?.groupValues?.get(1).toString().toIntOrNull()
        val tvType = if (document.select("div.serial-wrapper")
                .isNotEmpty()
        ) TvType.TvSeries else TvType.Movie
        val description = document.select("div.content > blockquote").text().trim()
        val rating =
            document.selectFirst("div.content > div:nth-child(6) > h3")?.text()?.toRatingInt()
        val actors = document.select("div.col-xs-9.content > div:nth-child(3) > h3 > a").map {
            ActorData(
                Actor(
                    it.text()
                )
            )
        }

        val recommendations = document.select("div.row.item-media").map {
            val recName = it.selectFirst("h3")?.text()?.trim().toString()
            val recHref = it.selectFirst(".content-media > a")!!.attr("href")
            val recPosterUrl = fixUrl(it.selectFirst(".poster-media > a > img")?.attr("src").toString())
            newTvSeriesSearchResponse(recName, recHref, TvType.TvSeries) {
                this.posterUrl = recPosterUrl
            }
        }

        return if (tvType == TvType.TvSeries) {
            val episodes = document.select("div.episode-list > a:matches(\\d+)").map {
                val href = fixUrl(it.attr("href"))
                val episode = it.text().toIntOrNull()
                val season =
                    it.attr("href").substringAfter("season-").substringBefore("-").toIntOrNull()
                Episode(
                    href,
                    "Episode $episode",
                    season,
                    episode,
                )
            }.reversed()
            newTvSeriesLoadResponse(title, url, TvType.TvSeries, episodes) {
                this.posterUrl = poster
                this.year = year
                this.plot = description
                this.tags = tags
                this.rating = rating
                this.actors = actors
                this.recommendations = recommendations
            }
        }
        else {
            newMovieLoadResponse(title, url, TvType.Movie, url) {
                this.posterUrl = poster
                this.year = year
                this.plot = description
                this.tags = tags
                this.rating = rating
                this.actors = actors
                this.recommendations = recommendations
            }
        }
    }

    override suspend fun loadLinks(
        data: String,
        isCasting: Boolean,
        subtitleCallback: (SubtitleFile) -> Unit,
        callback: (ExtractorLink) -> Unit
    ): Boolean {

        val document = app.get(data).document

        val sources = if(data.contains("-episode-")) {
            document.select("script").mapNotNull { script ->
                if(script.data().contains("var data =")) {
                    val scriptData = script.toString().substringAfter("var data = '").substringBefore("';")
                    Jsoup.parse(scriptData).select("li").map {
                        fixUrl(it.select("a").attr("href"))
                    }
                } else {
                    null
                }
            }[0]
        } else {
            document.select("ul#loadProviders > li").map {
                fixUrl(it.select("a").attr("href"))
            }
        }

        sources.apmap {
            loadExtractor(it, data, callback)
        }

        return true
    }


}