package com.lagradost.cloudstream3.animeproviders

import com.fasterxml.jackson.annotation.JsonProperty
import com.fasterxml.jackson.module.kotlin.readValue
import com.lagradost.cloudstream3.*
import org.jsoup.Jsoup
import java.lang.Exception
import java.net.URLEncoder

class ShiroProvider : MainAPI() {
    companion object {
        var token: String? = null
    }

    fun autoLoadToken(): Boolean {
        if (token != null) return true
        return loadToken()
    }

    fun loadToken(): Boolean {
        return try {
            val response = khttp.get(mainUrl, headers = baseHeader)

            val jsMatch = Regex("""src="(/static/js/main.*?)"""").find(response.text)
            val (destructed) = jsMatch!!.destructured
            val jsLocation = "$mainUrl$destructed"
            val js = khttp.get(jsLocation, headers = baseHeader)
            val tokenMatch = Regex("""token:"(.*?)"""").find(js.text)
            token = (tokenMatch!!.destructured).component1()

            token != null
        } catch (e: Exception) {
            false
        }
    }

    override val mainUrl: String
        get() = "https://shiro.is"

    data class ShiroSearchResponseShow(
        @JsonProperty("image") val image: String,
        @JsonProperty("_id") val _id: String,
        @JsonProperty("slug") val slug: String,
        @JsonProperty("name") val name: String,
    )
    data class ShiroSearchResponse(
        @JsonProperty("data") val data: List<ShiroSearchResponseShow>,
        @JsonProperty("status") val status: String
    )

    data class ShiroFullSearchResponseCurrentPage(
        @JsonProperty("items") val items: List<ShiroSearchResponseShow>
    )
    data class ShiroFullSearchResponseNavItems(
        @JsonProperty("currentPage") val currentPage: ShiroFullSearchResponseCurrentPage
    )

    data class ShiroFullSearchResponseNav(
        @JsonProperty("nav") val nav: ShiroFullSearchResponseNavItems
    )
    data class ShiroFullSearchResponse(
        @JsonProperty("data") val data: ShiroFullSearchResponseNav,
        @JsonProperty("status") val status: String
    )

    override fun search(query: String): ArrayList<Any>? {
        if(!autoLoadToken()) return null
        try {
            val returnValue: ArrayList<Any> = ArrayList()
            val response = khttp.get("https://tapi.shiro.is/advanced?search=${
                URLEncoder.encode(
                    query,
                    "UTF-8"
                )
            }&token=$token")
            val mapped = response.let { mapper.readValue<ShiroSearchResponse>(it.text) }



            return returnValue
        }
        catch (e : Exception) {
            return null
        }
    }
}