package com.lagradost

import com.fasterxml.jackson.annotation.JsonProperty
import com.lagradost.cloudstream3.*
import com.lagradost.cloudstream3.LoadResponse.Companion.addActors
import com.lagradost.cloudstream3.utils.AppUtils.parseJson
import com.lagradost.cloudstream3.utils.ExtractorLink
import com.lagradost.cloudstream3.utils.getQualityFromName

class TheFlixToProvider : MainAPI() {
    companion object {
        var latestCookies: Map<String, String> = emptyMap()
    }

    override var name = "TheFlix.to"
    override var mainUrl = "https://theflix.to"
    override val instantLinkLoading = false
    override val hasMainPage = true
    override val supportedTypes = setOf(
        TvType.Movie,
        TvType.TvSeries,
    )


    data class HomeJson(
        @JsonProperty("props") val props: HomeProps = HomeProps(),
    )

    data class HomeProps(
        @JsonProperty("pageProps") val pageProps: PageProps = PageProps(),
    )

    data class PageProps(
        @JsonProperty("moviesListTrending") val moviesListTrending: MoviesListTrending = MoviesListTrending(),
        @JsonProperty("moviesListNewArrivals") val moviesListNewArrivals: MoviesListNewArrivals = MoviesListNewArrivals(),
        @JsonProperty("tvsListTrending") val tvsListTrending: TvsListTrending = TvsListTrending(),
        @JsonProperty("tvsListNewEpisodes") val tvsListNewEpisodes: TvsListNewEpisodes = TvsListNewEpisodes(),
    )


    data class MoviesListTrending(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )

    data class MoviesListNewArrivals(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )

    data class TvsListTrending(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )

    data class TvsListNewEpisodes(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )

    data class Docs(
        @JsonProperty("name") val name: String = String(),
        @JsonProperty("originalLanguage") val originalLanguage: String? = null,
        @JsonProperty("popularity") val popularity: Double? = null,
        @JsonProperty("runtime") val runtime: Int? = null,
        @JsonProperty("status") val status: String? = null,
        @JsonProperty("voteAverage") val voteAverage: Double? = null,
        @JsonProperty("voteCount") val voteCount: Int? = null,
        @JsonProperty("cast") val cast: String? = null,
        @JsonProperty("director") val director: String? = null,
        @JsonProperty("overview") val overview: String? = null,
        @JsonProperty("posterUrl") val posterUrl: String? = null,
        @JsonProperty("releaseDate") val releaseDate: String? = null,
        @JsonProperty("createdAt") val createdAt: String? = null,
        @JsonProperty("updatedAt") val updatedAt: String? = null,
        @JsonProperty("conversionDate") val conversionDate: String? = null,
        @JsonProperty("id") val id: Int? = null,
        @JsonProperty("available") val available: Boolean? = null,
        @JsonProperty("videos") val videos: ArrayList<String>? = arrayListOf(),
    )


    private suspend fun getCookies(): Map<String, String> {
        //  val cookieResponse = app.post(
        //      "https://theflix.to:5679/authorization/session/continue?contentUsageType=Viewing",
        //    headers = mapOf(
        //          "Host" to "theflix.to:5679",
        //          "User-Agent" to "Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0",
        //          "Accept" to "application/json, text/plain,"
        //          "Accept-Language" to "en-US,en;q=0.5",
        //          "Content-Type" to "application/json;charset=utf-8",
        //          "Content-Length" to "35",
        //          "Origin" to "https://theflix.to",
        //          "DNT" to "1",
        //          "Connection" to "keep-alive",
        //          "Referer" to "https://theflix.to/",
        //          "Sec-Fetch-Dest" to "empty",
        //          "Sec-Fetch-Mode" to "cors",
        //          "Sec-Fetch-Site" to "same-site",)).okhttpResponse.headers.values("Set-Cookie")

        val cookies = app.post(
            "$mainUrl:5679/authorization/session/continue?contentUsageType=Viewing",
            headers = mapOf(
                "Host" to "theflix.to:5679",
                "User-Agent" to USER_AGENT,
                "Accept" to "application/json, text/plain, */*",
                "Accept-Language" to "en-US,en;q=0.5",
                "Content-Type" to "application/json;charset=utf-8",
                "Content-Length" to "35",
                "Origin" to mainUrl,
                "DNT" to "1",
                "Connection" to "keep-alive",
                "Referer" to mainUrl,
                "Sec-Fetch-Dest" to "empty",
                "Sec-Fetch-Mode" to "cors",
                "Sec-Fetch-Site" to "same-site",
            )
        ).cookies
        /* val cookieRegex = Regex("(theflix\\..*?id\\=[a-zA-Z0-9]{0,8}[a-zA-Z0-9_-]+)")
       val findcookie = cookieRegex.findAll(cookieResponse.toString()).map { it.value }.toList()
       val cookiesstring = findcookie.toString().replace(", ","; ").replace("[","").replace("]","")
       val cookiesmap = mapOf("Cookie" to cookiesstring) */
        latestCookies = cookies
        return latestCookies
    }


    override suspend fun getMainPage(page: Int, request: MainPageRequest): HomePageResponse {
        val items = ArrayList<HomePageList>()
        val doc = app.get(mainUrl).document
        val scriptText = doc.selectFirst("script[type=application/json]")!!.data()
        if (scriptText.contains("moviesListTrending")) {
            val json = parseJson<HomeJson>(scriptText)
            val homePageProps = json.props.pageProps
            listOf(
                Triple(
                    homePageProps.moviesListNewArrivals.docs,
                    homePageProps.moviesListNewArrivals.type,
                    "New Movie arrivals"
                ),
                Triple(
                    homePageProps.moviesListTrending.docs,
                    homePageProps.moviesListTrending.type,
                    "Trending Movies"
                ),
                Triple(
                    homePageProps.tvsListTrending.docs,
                    homePageProps.tvsListTrending.type,
                    "Trending TV Series"
                ),
                Triple(
                    homePageProps.tvsListNewEpisodes.docs,
                    homePageProps.tvsListNewEpisodes.type,
                    "New Episodes"
                )
            ).map { (docs, type, homename) ->
                val home = docs.map { info ->
                    val title = info.name
                    val poster = info.posterUrl
                    val typeinfo =
                        if (type?.contains("TV") == true) TvType.TvSeries else TvType.Movie
                    val link =
                        if (typeinfo == TvType.Movie) "$mainUrl/movie/${info.id}-${cleanTitle(title)}"
                        else "$mainUrl/tv-show/${info.id}-${
                            cleanTitle(title).replace(
                                "?",
                                ""
                            )
                        }/season-1/episode-1"
                    TvSeriesSearchResponse(
                        title,
                        link,
                        this.name,
                        typeinfo,
                        poster,
                        null,
                        null,
                    )
                }
                items.add(HomePageList(homename, home))
            }

        }

        if (items.size <= 0) throw ErrorLoadingException()
        return HomePageResponse(items)
    }

    data class SearchJson(
        @JsonProperty("props") val props: SearchProps = SearchProps(),
    )

    data class SearchProps(
        @JsonProperty("pageProps") val pageProps: SearchPageProps = SearchPageProps(),
    )

    data class SearchPageProps(
        @JsonProperty("mainList") val mainList: SearchMainList = SearchMainList(),
    )

    data class SearchMainList(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )


    override suspend fun search(query: String): List<SearchResponse> {
        val search = ArrayList<SearchResponse>()
        val urls = listOf(
            "$mainUrl/movies/trending?search=$query",
            "$mainUrl/tv-shows/trending?search=$query"
        )
        urls.apmap { url ->
            val doc = app.get(url).document
            val scriptText = doc.selectFirst("script[type=application/json]")!!.data()
            if (scriptText.contains("pageProps")) {
                val json = parseJson<SearchJson>(scriptText)
                val searchPageProps = json.props.pageProps.mainList
                val pair = listOf(Pair(searchPageProps.docs, searchPageProps.type))
                pair.map { (docs, type) ->
                    docs.map { info ->
                        val title = info.name
                        val poster = info.posterUrl
                        val typeinfo =
                            if (type?.contains("TV") == true) TvType.TvSeries else TvType.Movie
                        val link = if (typeinfo == TvType.Movie) "$mainUrl/movie/${info.id}-${
                            cleanTitle(title)
                        }"
                        else "$mainUrl/tv-show/${info.id}-${cleanTitle(title)}/season-1/episode-1"
                        if (typeinfo == TvType.Movie) {
                            search.add(
                                MovieSearchResponse(
                                    title,
                                    link,
                                    this.name,
                                    TvType.Movie,
                                    poster,
                                    null
                                )
                            )
                        } else {
                            search.add(
                                TvSeriesSearchResponse(
                                    title,
                                    link,
                                    this.name,
                                    TvType.TvSeries,
                                    poster,
                                    null,
                                    null
                                )
                            )
                        }
                    }
                }
            }
        }
        return search
    }

    data class LoadMain(
        @JsonProperty("props") val props: LoadProps? = LoadProps(),
        @JsonProperty("page") val page: String? = null,
        @JsonProperty("buildId") val buildId: String? = null,
        @JsonProperty("runtimeConfig") val runtimeConfig: RuntimeConfig? = RuntimeConfig(),
        @JsonProperty("isFallback") val isFallback: Boolean? = null,
        @JsonProperty("gssp") val gssp: Boolean? = null,
        @JsonProperty("customServer") val customServer: Boolean? = null,
        @JsonProperty("appGip") val appGip: Boolean? = null
    )

    data class LoadProps(
        @JsonProperty("pageProps") val pageProps: LoadPageProps? = LoadPageProps(),
        @JsonProperty("__N_SSP") val _NSSP: Boolean? = null
    )

    data class LoadPageProps(
        @JsonProperty("selectedTv") val selectedTv: TheFlixMetadata? = TheFlixMetadata(),
        @JsonProperty("movie") val movie: TheFlixMetadata? = TheFlixMetadata(),
        @JsonProperty("recommendationsList") val recommendationsList: RecommendationsList? = RecommendationsList(),
        @JsonProperty("basePageSegments") val basePageSegments: ArrayList<String>? = arrayListOf()
    )

    data class TheFlixMetadata(
        @JsonProperty("episodeRuntime") val episodeRuntime: Int? = null,
        @JsonProperty("name") val name: String? = null,
        @JsonProperty("numberOfSeasons") val numberOfSeasons: Int? = null,
        @JsonProperty("numberOfEpisodes") val numberOfEpisodes: Int? = null,
        @JsonProperty("originalLanguage") val originalLanguage: String? = null,
        @JsonProperty("popularity") val popularity: Double? = null,
        @JsonProperty("status") val status: String? = null,
        @JsonProperty("voteAverage") val voteAverage: Double? = null,
        @JsonProperty("voteCount") val voteCount: Int? = null,
        @JsonProperty("cast") val cast: String? = null,
        @JsonProperty("director") val director: String? = null,
        @JsonProperty("overview") val overview: String? = null,
        @JsonProperty("posterUrl") val posterUrl: String? = null,
        @JsonProperty("releaseDate") val releaseDate: String? = null,
        @JsonProperty("createdAt") val createdAt: String? = null,
        @JsonProperty("updatedAt") val updatedAt: String? = null,
        @JsonProperty("id") val id: Int? = null,
        @JsonProperty("available") val available: Boolean? = null,
        @JsonProperty("genres") val genres: ArrayList<Genres>? = arrayListOf(),
        @JsonProperty("seasons") val seasons: ArrayList<Seasons>? = arrayListOf(),
        @JsonProperty("videos") val videos: ArrayList<String>? = arrayListOf(),
        @JsonProperty("runtime") val runtime: Int? = null,
    )

    data class Seasons(
        @JsonProperty("name") val name: String? = null,
        @JsonProperty("numberOfEpisodes") val numberOfEpisodes: Int? = null,
        @JsonProperty("seasonNumber") val seasonNumber: Int? = null,
        @JsonProperty("overview") val overview: String? = null,
        @JsonProperty("posterUrl") val posterUrl: String? = null,
        @JsonProperty("releaseDate") val releaseDate: String? = null,
        @JsonProperty("createdAt") val createdAt: String? = null,
        @JsonProperty("updatedAt") val updatedAt: String? = null,
        @JsonProperty("id") val id: Int? = null,
        @JsonProperty("episodes") val episodes: ArrayList<Episodes>? = arrayListOf()
    )

    data class Episodes(
        @JsonProperty("episodeNumber") val episodeNumber: Int? = null,
        @JsonProperty("name") val name: String? = null,
        @JsonProperty("seasonNumber") val seasonNumber: Int? = null,
        @JsonProperty("voteAverage") val voteAverage: Double? = null,
        @JsonProperty("voteCount") val voteCount: Int? = null,
        @JsonProperty("overview") val overview: String? = null,
        @JsonProperty("releaseDate") val releaseDate: String? = null,
        @JsonProperty("createdAt") val createdAt: String? = null,
        @JsonProperty("updatedAt") val updatedAt: String? = null,
        @JsonProperty("id") val id: Int? = null,
        @JsonProperty("videos") val videos: ArrayList<String>? = arrayListOf()
    )


    data class Genres(
        @JsonProperty("name") val name: String? = null,
        @JsonProperty("id") val id: Int? = null
    )

    data class RuntimeConfig(
        @JsonProperty("AddThisService") val AddThisService: RuntimeConfigData? = RuntimeConfigData(),
        @JsonProperty("Application") val Application: RuntimeConfigData? = RuntimeConfigData(),
        @JsonProperty("GtmService") val GtmService: RuntimeConfigData? = RuntimeConfigData(),
        @JsonProperty("Services") val Services: RuntimeConfigData? = RuntimeConfigData(),
    )

    data class RuntimeConfigData(
        @JsonProperty("PublicId") val PublicId: String? = null,
        @JsonProperty("ContentUsageType") val ContentUsageType: String? = null,
        @JsonProperty("IsDevelopmentMode") val IsDevelopmentMode: Boolean? = null,
        @JsonProperty("IsDevelopmentOrProductionMode") val IsDevelopmentOrProductionMode: Boolean? = null,
        @JsonProperty("IsProductionMode") val IsProductionMode: Boolean? = null,
        @JsonProperty("IsStagingMode") val IsStagingMode: Boolean? = null,
        @JsonProperty("IsTestMode") val IsTestMode: Boolean? = null,
        @JsonProperty("Mode") val Mode: String? = null,
        @JsonProperty("Name") val Name: String? = null,
        @JsonProperty("Url") val Url: String? = null,
        @JsonProperty("UseFilterInfoInUrl") val UseFilterInfoInUrl: Boolean? = null,
        @JsonProperty("TrackingId") val TrackingId: String? = null,
        @JsonProperty("Server") val Server: Server? = Server(),
        @JsonProperty("TmdbServer") val TmdbServer: TmdbServer? = TmdbServer(),
    )

    data class TmdbServer(
        @JsonProperty("Url") val Url: String? = null
    )


    data class Server(
        @JsonProperty("Url") val Url: String? = null
    )

    data class RecommendationsList(
        @JsonProperty("docs") val docs: ArrayList<Docs> = arrayListOf(),
        @JsonProperty("total") val total: Int? = null,
        @JsonProperty("page") val page: Int? = null,
        @JsonProperty("limit") val limit: Int? = null,
        @JsonProperty("pages") val pages: Int? = null,
        @JsonProperty("type") val type: String? = null,
    )

    private fun cleanTitle(title: String): String {
        val dotTitle = title.substringBefore("/season")
        if (dotTitle.contains(Regex("\\..\\."))) { //For titles containing more than two dots (S.W.A.T.)
            return (dotTitle.removeSuffix(".")
                .replace(" - ", "-")
                .replace(".", "-").replace(" ", "-")
                .replace("-&", "")
                .replace(Regex("(:|-&)"), "")
                .replace("'", "-")).lowercase()
        }
        return (title
            .replace(" - ", "-")
            .replace(" ", "-")
            .replace("-&", "")
            .replace("/", "-")
            .replace(Regex("(:|-&|\\.)"), "")
            .replace("'", "-")).lowercase()
    }

    private suspend fun getLoadMan(url: String): LoadMain {
        getCookies()
        val og = app.get(url, headers = latestCookies)
        val soup = og.document
        val script = soup.selectFirst("script[type=application/json]")!!.data()
        return parseJson(script)
    }

    override suspend fun load(url: String): LoadResponse? {
        val tvtype = if (url.contains("movie")) TvType.Movie else TvType.TvSeries
        val json = getLoadMan(url)
        val episodes = ArrayList<Episode>()
        val isMovie = tvtype == TvType.Movie
        val pageMain = json.props?.pageProps

        val metadata: TheFlixMetadata? = if (isMovie) pageMain?.movie else pageMain?.selectedTv

        val available = metadata?.available

        val comingsoon = !available!!

        val movieId = metadata.id

        val movietitle = metadata.name

        val poster = metadata.posterUrl

        val description = metadata.overview

        if (!isMovie) {
            metadata.seasons?.map { seasons ->
                val seasonPoster = seasons.posterUrl ?: metadata.posterUrl
                seasons.episodes?.forEach { epi ->
                    val episodenu = epi.episodeNumber
                    val seasonum = epi.seasonNumber
                    val title = epi.name
                    val epDesc = epi.overview
                    val test = epi.videos
                    val ratinginfo = (epi.voteAverage)?.times(10)?.toInt()
                    val rating = if (ratinginfo?.equals(0) == true) null else ratinginfo
                    val eps = Episode(
                        "$mainUrl/tv-show/$movieId-${cleanTitle(movietitle!!)}/season-$seasonum/episode-$episodenu",
                        title,
                        seasonum,
                        episodenu,
                        description = epDesc!!,
                        posterUrl = seasonPoster,
                        rating = rating,
                    )
                    if (test!!.isNotEmpty()) {
                        episodes.add(eps)
                    } else {
                        //Nothing, will prevent seasons/episodes with no videos to be added
                    }
                }
            }
        }
        val rating = metadata.voteAverage?.toFloat()?.times(1000)?.toInt()

        val tags = metadata.genres?.mapNotNull { it.name }

        val recommendationsitem = pageMain?.recommendationsList?.docs?.map { loadDocs ->
            val title = loadDocs.name
            val posterrec = loadDocs.posterUrl
            val link = if (isMovie) "$mainUrl/movie/${loadDocs.id}-${cleanTitle(title)}"
            else "$mainUrl/tv-show/${loadDocs.id}-${cleanTitle(title)}/season-1/episode-1"
            MovieSearchResponse(
                title,
                link,
                this.name,
                tvtype,
                posterrec,
                year = null
            )
        }

        val year = metadata.releaseDate?.substringBefore("-")

        val runtime = metadata.runtime?.div(60) ?: metadata.episodeRuntime?.div(60)
        val cast = metadata.cast?.split(",")

        return when (tvtype) {
            TvType.TvSeries -> {
                return newTvSeriesLoadResponse(movietitle!!, url, TvType.TvSeries, episodes) {
                    this.posterUrl = poster
                    this.year = year?.toIntOrNull()
                    this.plot = description
                    this.duration = runtime
                    addActors(cast)
                    this.tags = tags
                    this.recommendations = recommendationsitem
                    this.comingSoon = comingsoon
                    this.rating = rating
                }
            }
            TvType.Movie -> {
                newMovieLoadResponse(movietitle!!, url, TvType.Movie, url) {
                    this.year = year?.toIntOrNull()
                    this.posterUrl = poster
                    this.plot = description
                    this.duration = runtime
                    addActors(cast)
                    this.tags = tags
                    this.recommendations = recommendationsitem
                    this.comingSoon = comingsoon
                    this.rating = rating
                }
            }
            else -> null
        }
    }


    data class VideoData(
        @JsonProperty("url") val url: String? = null,
        @JsonProperty("id") val id: String? = null,
        @JsonProperty("type") val type: String? = null,
    )


    override suspend fun loadLinks(
        data: String,
        isCasting: Boolean,
        subtitleCallback: (SubtitleFile) -> Unit,
        callback: (ExtractorLink) -> Unit
    ): Boolean {
        val json = getLoadMan(data)
        val authhost = json.runtimeConfig?.Services?.Server?.Url
        val isMovie = data.contains("/movie/")
        val qualityReg = Regex("(\\d+p)")
        if (isMovie) {
            json.props?.pageProps?.movie?.videos?.apmap { id ->
                val videoData = app.get(
                    "$authhost/movies/videos/$id/request-access?contentUsageType=Viewing",
                    cookies = latestCookies
                ).parsedSafe<VideoData>() ?: return@apmap false
                val extractedLink = videoData.url

                if (!extractedLink.isNullOrEmpty()) {
                    val quality = qualityReg.find(extractedLink)?.value ?: ""
                    callback(
                        ExtractorLink(
                            name,
                            name,
                            extractedLink,
                            this.mainUrl + "/",
                            getQualityFromName(quality),
                            false
                        )
                    )
                } else null
            }
        } else {
            val dataRegex = Regex("(season-(\\d+)\\/episode-(\\d+))")
            val cleandatainfo =
                dataRegex.find(data)?.value?.replace(Regex("(season-|episode-)"), "")
                    ?.replace("/", "x")
            val tesatt = cleandatainfo.let { str ->
                str?.split("x")?.mapNotNull { subStr -> subStr.toIntOrNull() }
            }
            val epID = tesatt?.getOrNull(1)
            val seasonid = tesatt?.getOrNull(0)
            json.props?.pageProps?.selectedTv?.seasons?.map {
                it.episodes?.map {
                    val epsInfo = Triple(it.seasonNumber, it.episodeNumber, it.videos)
                    if (epsInfo.first == seasonid && epsInfo.second == epID) {
                        epsInfo.third?.apmap { id ->
                            val jsonserie = app.get(
                                "$authhost/tv/videos/$id/request-access?contentUsageType=Viewing",
                                cookies = latestCookies
                            ).parsedSafe<VideoData>() ?: return@apmap false
                            val extractedlink = jsonserie.url
                            if (!extractedlink.isNullOrEmpty()) {
                                val quality = qualityReg.find(extractedlink)?.value ?: ""
                                callback(
                                    ExtractorLink(
                                        name,
                                        name,
                                        extractedlink,
                                        this.mainUrl + "/",
                                        getQualityFromName(quality),
                                        false
                                    )
                                )
                            } else null
                        }
                    }
                }
            }
        }
        return true
    }
}