cloudstream-extensions-hexated/Minioppai/src/main/kotlin/com/hexated/Minioppai.kt

package com.hexated

import com.fasterxml.jackson.annotation.JsonProperty
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.network.CloudflareKiller
import com.lagradost.cloudstream3.utils.*
import okhttp3.Interceptor
import okhttp3.Response
import org.jsoup.Jsoup
import org.jsoup.nodes.Element
import java.net.URLDecoder

class Minioppai : MainAPI() {
    override var mainUrl = "https://minioppai.org"
    override var name = "Minioppai"
    override val hasMainPage = true
    override var lang = "id"
    override val hasDownloadSupport = true
    override val hasQuickSearch = true
    private val cloudflareKiller by lazy { CloudflareKiller() }
    private val interceptor by lazy { CloudflareInterceptor(cloudflareKiller) }
    override val supportedTypes = setOf(
        TvType.NSFW,
    )

    class CloudflareInterceptor(private val cloudflareKiller: CloudflareKiller): Interceptor {
        override fun intercept(chain: Interceptor.Chain): Response {
            val request = chain.request()
            val response = chain.proceed(request)
            val doc = Jsoup.parse(response.peekBody(1024 * 1024).string())
            if (doc.select("title").text() == "Just a moment...") {
                return cloudflareKiller.intercept(chain)
            }
            return response
        }
    }

    companion object {
        fun getStatus(t: String?): ShowStatus {
            return when (t) {
                "Completed" -> ShowStatus.Completed
                "Ongoing" -> ShowStatus.Ongoing
                else -> ShowStatus.Completed
            }
        }
    }

    override val mainPage = mainPageOf(
        "$mainUrl/watch" to "New Episode",
        "$mainUrl/popular" to "Popular Hentai",
    )

    override suspend fun getMainPage(
        page: Int,
        request: MainPageRequest
    ): HomePageResponse {
        val document = app.get("${request.data}/page/$page", interceptor = interceptor).document
        val home = document.select("div.latest a").mapNotNull {
            it.toSearchResult()
        }
        return newHomePageResponse(
            list = HomePageList(
                name = request.name,
                list = home,
                isHorizontalImages = request.name == "New Episode"
            ),
            hasNext = true
        )
    }

    private fun getProperAnimeLink(uri: String): String {
        return if (uri.contains("-episode-")) {
            uri.substringBefore("-episode-")
        } else {
            uri
        }
    }

    private fun Element.toSearchResult(): AnimeSearchResponse? {
        val title = this.selectFirst("h2.entry-title")?.text()?.trim() ?: return null
        val href = getProperAnimeLink(this.attr("href"))
        val posterUrl = fixUrlNull(this.selectFirst("img")?.attr("src"))
        val epNum = this.selectFirst("i.dot")?.text()?.filter { it.isDigit() }?.toIntOrNull()
        return newAnimeSearchResponse(title, href, TvType.NSFW) {
            this.posterUrl = posterUrl
            addSub(epNum)
            posterHeaders = cloudflareKiller.getCookieHeaders(mainUrl).toMap()
        }

    }

    override suspend fun quickSearch(query: String): List<SearchResponse>? = search(query)

    override suspend fun search(query: String): List<SearchResponse>? {
        return app.post(
            "$mainUrl/wp-admin/admin-ajax.php", data = mapOf(
                "action" to "ts_ac_do_search",
                "ts_ac_query" to query,
            ), headers = mapOf("X-Requested-With" to "XMLHttpRequest"),
            interceptor = interceptor
        ).parsedSafe<SearchResponses>()?.post?.firstOrNull()?.all?.mapNotNull { item ->
            newAnimeSearchResponse(
                item.postTitle ?: "",
                item.postLink ?: return@mapNotNull null,
                TvType.NSFW
            ) {
                this.posterUrl = item.postImage
                posterHeaders = cloudflareKiller.getCookieHeaders(mainUrl).toMap()
            }
        }
    }

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

        val title = document.selectFirst("h1.entry-title")?.text()?.trim() ?: return null
        val poster = fixUrlNull(document.selectFirst("div.limage img")?.attr("src"))
        val table = document.select("ul.data")
        val tags = table.select("ul.data li:nth-child(1) a").map { it.text() }
        val year =
            document.selectFirst("ul.data time[itemprop=dateCreated]")?.text()?.substringBefore("-")
                ?.toIntOrNull()
        val status = getStatus(document.selectFirst("ul.data li:nth-child(2) span")?.text()?.trim())
        val description = document.select("div[itemprop=description] > p").text()

        val episodes = document.select("div.epsdlist ul li").mapNotNull {
            val name = it.selectFirst("div.epl-num")?.text()
            val link = fixUrlNull(it.selectFirst("a")?.attr("href")) ?: return@mapNotNull null
            Episode(link, name = name)
        }.reversed()

        return newAnimeLoadResponse(title, url, TvType.NSFW) {
            engName = title
            posterUrl = poster
            this.year = year
            addEpisodes(DubStatus.Subbed, episodes)
            showStatus = status
            plot = description
            this.tags = tags
            posterHeaders = cloudflareKiller.getCookieHeaders(mainUrl).toMap()
        }
    }

    override suspend fun loadLinks(
        data: String,
        isCasting: Boolean,
        subtitleCallback: (SubtitleFile) -> Unit,
        callback: (ExtractorLink) -> Unit
    ): Boolean {
        val document = app.get(data, interceptor = interceptor).document
        document.select("div.server ul.mirror li a").mapNotNull {
            Jsoup.parse(base64Decode(it.attr("data-em"))).select("iframe").attr("src")
        }.apmap { link ->
            loadExtractor(
                fixUrl(decode(link.substringAfter("data="))),
                mainUrl,
                subtitleCallback,
                callback
            )
        }

        return true
    }

    private fun decode(input: String): String = URLDecoder.decode(input, "utf-8")

    data class SearchResponses(
        @JsonProperty("post") var post: ArrayList<Post> = arrayListOf()
    )

    data class All(
        @JsonProperty("ID") var ID: Int? = null,
        @JsonProperty("post_image") var postImage: String? = null,
        @JsonProperty("post_title") var postTitle: String? = null,
        @JsonProperty("post_link") var postLink: String? = null,
    )

    data class Post(
        @JsonProperty("all") var all: ArrayList<All> = arrayListOf(),
    )

}