AquaStream/app/src/main/java/com/lagradost/cloudstream3/movieproviders/HDMovie5.kt

package com.lagradost.cloudstream3.movieproviders

import com.fasterxml.jackson.annotation.JsonProperty
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.httpsify
import com.lagradost.cloudstream3.utils.loadExtractor
import org.jsoup.Jsoup

class HDMovie5 : MainAPI() {
    override var mainUrl = "https://hdmovie5.tv"
    override var name = "HDMovie"
    override val lang = "hi"

    override val hasQuickSearch = true
    override val hasMainPage = true
    override val supportedTypes = setOf(
        TvType.Movie,
        TvType.TvSeries,
    )

    override suspend fun getMainPage(): HomePageResponse {
        val doc = app.get(mainUrl).document.select("div.content")
        val list = mapOf(
            "Featured Movies" to "featured",
            "Updated Movies" to "normal"
        )
        return HomePageResponse(list.map { item ->
            HomePageList(item.key,
                doc.select("div.${item.value}>.item").map {
                    val data = it.select(".data")
                    val a = data.select("a")
                    MovieSearchResponse(
                        a.text(),
                        a.attr("href"),
                        this.name,
                        TvType.Movie,
                        it.select("img").attr("src"),
                        data.select("span").text().toIntOrNull()
                    )
                }
            )
        })
    }

    private data class QuickSearchResponse(
        val title: String,
        val url: String,
        val img: String,
        val extra: Extra
    ) {
        data class Extra(
            val date: String
        )
    }

    override suspend fun quickSearch(query: String): List<SearchResponse> {
        return app.get("$mainUrl/wp-json/dooplay/search/?keyword=$query&nonce=ddbde04d9c")
            .parsed<Map<String, QuickSearchResponse>>().map {
                val res = it.value
                MovieSearchResponse(
                    res.title,
                    res.url,
                    this.name,
                    TvType.Movie,
                    res.img,
                    res.extra.date.toIntOrNull()
                )
            }
    }

    override suspend fun search(query: String): List<SearchResponse> {
        return app.get("$mainUrl/?s=$query").document.select(".search-page>div.result-item").map {
            val image = it.select(".image")
            MovieSearchResponse(
                image.select("img").attr("alt"),
                image.select("a").attr("href"),
                this.name,
                TvType.Movie,
                image.select("img").attr("src"),
                it.select(".year").text().toIntOrNull()
            )
        }
    }

    override suspend fun load(url: String): LoadResponse {
        val doc = app.get(url).document
        val info = doc.select(".sheader")
        val links = doc.select("#playeroptionsul>li")
        val data = links.joinToString(",") { it.attr("data-post") }
        return MovieLoadResponse(
            info.select(".data>h1").text(),
            url,
            this.name,
            TvType.Movie,
            data,
            info.select(".poster>img").attr("src"),
            info.select(".date").text().substringAfter(", ").toIntOrNull(),
            doc.select(".wp-content>p").let { it.getOrNull(it.size - 1)?.text() },
            (doc.select("#repimdb>strong").text().toFloatOrNull()?.times(1000))?.toInt(),
            info.select(".sgeneros>a").map { it.text() },
            info.select(".runtime").text().substringBefore(" Min.").toIntOrNull(),
            null,
            doc.select("#single_relacionados>article>a").map {
                val img = it.select("img")
                MovieSearchResponse(
                    img.attr("alt"),
                    it.attr("href"),
                    this.name,
                    TvType.Movie,
                    img.attr("src")
                )
            },
            doc.select("#cast>.persons>.person").mapNotNull {
                if (it.attr("itemprop") != "director") {
                    ActorData(
                        Actor(
                            it.select("meta").attr("content"),
                            it.select("img").attr("src")
                        )
                    )
                } else null
            },
        )
    }

    private data class PlayerAjaxResponse(
        @JsonProperty("embed_url")
        val embedURL: String? = null
    )

    override suspend fun loadLinks(
        data: String,
        isCasting: Boolean,
        subtitleCallback: (SubtitleFile) -> Unit,
        callback: (ExtractorLink) -> Unit
    ): Boolean {
        return data.split(",").apmapIndexed { index, it ->
            //println("loadLinks:::: $index $it")
            val p = app.post(
                "$mainUrl/wp-admin/admin-ajax.php",
                data = mapOf(
                    "action" to "doo_player_ajax",
                    "post" to it,
                    "nume" to "${index + 1}",
                    "type" to "movie"
                )
            )
           // println("TEXT::::: ${p.text}")
            val html = p.parsedSafe<PlayerAjaxResponse>()?.embedURL ?: return@apmapIndexed false
            val doc = Jsoup.parse(html)
            val link = doc.select("iframe").attr("src")
            loadExtractor(httpsify(link), "$mainUrl/", callback)
        }.contains(true)
    }
}