Tachiyomi-Extensions/multisrc/overrides/mangareader/mangafire/src/MangaFire.kt

package eu.kanade.tachiyomi.extension.all.mangafire

import eu.kanade.tachiyomi.multisrc.mangareader.MangaReader
import eu.kanade.tachiyomi.network.GET
import eu.kanade.tachiyomi.source.model.Filter
import eu.kanade.tachiyomi.source.model.FilterList
import eu.kanade.tachiyomi.source.model.Page
import eu.kanade.tachiyomi.source.model.SChapter
import eu.kanade.tachiyomi.source.model.SManga
import kotlinx.serialization.Serializable
import kotlinx.serialization.decodeFromString
import kotlinx.serialization.json.Json
import kotlinx.serialization.json.JsonPrimitive
import kotlinx.serialization.json.int
import okhttp3.HttpUrl.Companion.toHttpUrl
import okhttp3.Request
import okhttp3.Response
import org.jsoup.Jsoup
import org.jsoup.nodes.Document
import org.jsoup.nodes.Element
import org.jsoup.select.Evaluator
import uy.kohesive.injekt.injectLazy

open class MangaFire(
    override val lang: String,
    private val langCode: String = lang,
) : MangaReader() {
    override val name = "MangaFire"

    override val baseUrl = "https://mangafire.to"

    private val json: Json by injectLazy()

    override val client = network.client.newBuilder()
        .addInterceptor(ImageInterceptor)
        .build()

    override fun latestUpdatesRequest(page: Int) =
        GET("$baseUrl/filter?sort=recently_updated&language[]=$langCode&page=$page", headers)

    override fun popularMangaRequest(page: Int) =
        GET("$baseUrl/filter?sort=most_viewed&language[]=$langCode&page=$page", headers)

    override fun searchMangaRequest(page: Int, query: String, filters: FilterList): Request {
        val urlBuilder = baseUrl.toHttpUrl().newBuilder()
        if (query.isNotBlank()) {
            urlBuilder.addPathSegment("filter").apply {
                addQueryParameter("keyword", query)
                addQueryParameter("page", page.toString())
            }
        } else {
            urlBuilder.addPathSegment("filter").apply {
                addQueryParameter("language[]", langCode)
                addQueryParameter("page", page.toString())
                filters.ifEmpty(::getFilterList).forEach { filter ->
                    when (filter) {
                        is Group -> {
                            filter.state.forEach {
                                if (it.state) {
                                    addQueryParameter(filter.param, it.id)
                                }
                            }
                        }
                        is Select -> {
                            addQueryParameter(filter.param, filter.selection)
                        }
                        is GenresFilter -> {
                            filter.state.forEach {
                                if (it.state != 0) {
                                    addQueryParameter(filter.param, it.selection)
                                }
                            }
                            if (filter.combineMode) {
                                addQueryParameter("genre_mode", "and")
                            }
                        }
                        else -> {}
                    }
                }
            }
        }
        return GET(urlBuilder.build(), headers)
    }

    override fun searchMangaSelector() = ".mangas.items .inner"

    override fun searchMangaNextPageSelector() = ".page-item.active + .page-item .page-link"

    override fun searchMangaFromElement(element: Element) =
        SManga.create().apply {
            element.selectFirst("a.color-light")!!.let {
                url = it.attr("href")
                title = it.attr("title")
            }
            element.selectFirst(Evaluator.Tag("img"))!!.let {
                thumbnail_url = it.attr("src")
            }
        }

    override fun mangaDetailsParse(document: Document) = SManga.create().apply {
        val root = document.selectFirst(".detail .top .wrapper")!!
        val mangaTitle = root.selectFirst(Evaluator.Class("name"))!!.ownText()
        title = mangaTitle
        description = document.run {
            val description = selectFirst(Evaluator.Class("summary"))!!.ownText()
            when (val altTitle = root.selectFirst(Evaluator.Class("al-name"))!!.ownText()) {
                "", mangaTitle -> description
                else -> "$description\n\nAlternative Title: $altTitle"
            }
        }
        thumbnail_url = root.selectFirst(Evaluator.Tag("img"))!!.attr("src")
        status = when (root.selectFirst(Evaluator.Class("status"))!!.ownText()) {
            "Completed" -> SManga.COMPLETED
            "Releasing" -> SManga.ONGOING
            "On_hiatus" -> SManga.ON_HIATUS
            "Discontinued" -> SManga.CANCELLED
            else -> SManga.UNKNOWN
        }
        with(root.selectFirst(Evaluator.Class("more-info"))!!) {
            author = selectFirst("span:contains(Author:) + span")?.text()
            val type = selectFirst("span:contains(Type:) + span")?.text()
            val genres = selectFirst("span:contains(Genres:) + span")?.text()
            genre = listOfNotNull(type, genres).joinToString()
        }
    }

    override val chapterType get() = "chapter"
    override val volumeType get() = "volume"

    override fun chapterListRequest(mangaUrl: String, type: String): Request {
        val id = mangaUrl.substringAfterLast('.')
        return GET("$baseUrl/ajax/read/$id/list?viewby=$type", headers)
    }

    override fun parseChapterElements(response: Response, isVolume: Boolean): List<Element> {
        val result = json.decodeFromString<ResponseDto<ChapterListDto>>(response.body.string()).result
        val container = result.parseHtml(if (isVolume) volumeType else chapterType)
            ?.selectFirst(".numberlist[data-lang=$langCode]")
            ?: return emptyList()
        return container.children().map { it.child(0) }
    }

    override fun pageListRequest(chapter: SChapter): Request {
        val typeAndId = chapter.url.substringAfterLast('#')
        return GET("$baseUrl/ajax/read/$typeAndId", headers)
    }

    override fun pageListParse(response: Response): List<Page> {
        val result = json.decodeFromString<ResponseDto<PageListDto>>(response.body.string()).result

        return result.pages.mapIndexed { index, image ->
            val url = image.url
            val offset = image.offset
            val imageUrl = if (offset > 0) "$url#${ImageInterceptor.SCRAMBLED}_$offset" else url

            Page(index, imageUrl = imageUrl)
        }
    }

    override fun getFilterList() =
        FilterList(
            Filter.Header("NOTE: Ignored if using text search!"),
            Filter.Separator(),
            TypeFilter(),
            GenresFilter(),
            StatusFilter(),
            YearFilter(),
            ChapterCountFilter(),
            SortFilter(),
        )

    @Serializable
    class ChapterListDto(private val html: String, private val link_format: String) {
        fun parseHtml(type: String): Document? {
            if ("LANG/$type-NUMBER" !in link_format) return null
            return Jsoup.parseBodyFragment(html)
        }
    }

    @Serializable
    class PageListDto(private val images: List<List<JsonPrimitive>>) {
        val pages get() = images.map { Image(it[0].content, it[2].int) }
    }

    class Image(val url: String, val offset: Int)

    @Serializable
    class ResponseDto<T>(val result: T)
}