TachiyomiSY-Plus/app/src/main/java/exh/md/handlers/PageHandler.kt

package exh.md.handlers

import eu.kanade.tachiyomi.data.preference.PreferencesHelper
import eu.kanade.tachiyomi.data.track.mdlist.MdList
import eu.kanade.tachiyomi.network.GET
import eu.kanade.tachiyomi.network.asObservableSuccess
import eu.kanade.tachiyomi.source.model.Page
import eu.kanade.tachiyomi.source.model.SChapter
import exh.md.utils.MdUtil
import okhttp3.CacheControl
import okhttp3.Headers
import okhttp3.OkHttpClient
import okhttp3.Request
import rx.Observable

class PageHandler(
    private val client: OkHttpClient,
    private val headers: Headers,
    private val apiChapterParser: ApiChapterParser,
    private val mangaPlusHandler: MangaPlusHandler,
    private val preferences: PreferencesHelper,
    private val mdList: MdList,
) {

    fun fetchPageList(chapter: SChapter, isLogged: Boolean, usePort443Only: Boolean, dataSaver: Boolean): Observable<List<Page>> {
        if (chapter.scanlator.equals("MangaPlus")) {
            return client.newCall(pageListRequest(chapter))
                .asObservableSuccess()
                .map { response ->
                    val chapterId = apiChapterParser.externalParse(response)
                    mangaPlusHandler.fetchPageList(chapterId)
                }
        }

        val atHomeRequestUrl = if (usePort443Only) {
            "${MdUtil.atHomeUrl}/${MdUtil.getChapterId(chapter.url)}?forcePort443=true"
        } else {
            "${MdUtil.atHomeUrl}/${MdUtil.getChapterId(chapter.url)}"
        }

        val (client, headers) = if (isLogged) {
            client to MdUtil.getAuthHeaders(headers, preferences, mdList)
        } else {
            client to headers
        }

        return client.newCall(pageListRequest(chapter))
            .asObservableSuccess()
            .map { response ->
                val host = MdUtil.atHomeUrlHostUrl(atHomeRequestUrl, client, headers, CacheControl.FORCE_NETWORK)
                apiChapterParser.pageListParse(response, host, dataSaver)
            }
    }

    private fun pageListRequest(chapter: SChapter): Request {
        return GET("${MdUtil.chapterUrl}${MdUtil.getChapterId(chapter.url)}", headers, CacheControl.FORCE_NETWORK)
    }
}