package org.koitharu.kotatsu.parsers

import androidx.annotation.VisibleForTesting
import org.koitharu.kotatsu.parsers.model.*
import org.koitharu.kotatsu.parsers.util.Paginator

@InternalParsersApi
abstract class PagedMangaParser(
	context: MangaLoaderContext,
	source: MangaSource,
	@VisibleForTesting(otherwise = VisibleForTesting.PROTECTED) @JvmField internal val pageSize: Int,
	searchPageSize: Int = pageSize,
) : MangaParser(context, source) {

	@JvmField
	protected val paginator = Paginator(pageSize)

	@JvmField
	protected val searchPaginator = Paginator(searchPageSize)

	final override suspend fun getList(offset: Int, filter: MangaListFilter?): List<Manga> {
		return getList(
			paginator = if (filter is MangaListFilter.Search) {
				searchPaginator
			} else {
				paginator
			},
			offset = offset,
			filter = filter,
		)
	}

	@InternalParsersApi
	@Deprecated("You should use getListPage for PagedMangaParser", level = DeprecationLevel.HIDDEN)
	final override suspend fun getList(
		offset: Int,
		query: String?,
		tags: Set<MangaTag>?,
		tagsExclude: Set<MangaTag>?,
		sortOrder: SortOrder,
	): List<Manga> = throw UnsupportedOperationException("You should use getListPage for PagedMangaParser")

	@Deprecated("")
	open suspend fun getListPage(
		page: Int,
		query: String?,
		tags: Set<MangaTag>?,
		tagsExclude: Set<MangaTag>?,
		sortOrder: SortOrder,
	): List<Manga> = throw NotImplementedError("Please implement getListPage(page, filter) instead")

	open suspend fun getListPage(page: Int, filter: MangaListFilter?): List<Manga> {
		return when (filter) {
			is MangaListFilter.Advanced -> getListPage(
				page = page,
				query = null,
				tags = filter.tags,
				tagsExclude = filter.tagsExclude,
				sortOrder = filter.sortOrder,
			)

			is MangaListFilter.Search -> getListPage(
				page = page,
				query = filter.query,
				tags = null,
				tagsExclude = null,
				sortOrder = defaultSortOrder,
			)

			null -> getListPage(
				page = page,
				query = null,
				tags = null,
				tagsExclude = null,
				sortOrder = defaultSortOrder,
			)
		}
	}

	private suspend fun getList(
		paginator: Paginator,
		offset: Int,
		filter: MangaListFilter?,
	): List<Manga> {
		val page = paginator.getPage(offset)
		val list = getListPage(page, filter)
		paginator.onListReceived(offset, page, list.size)
		return list
	}
}