PlaguPoster/ratings/source/src/commonMain/kotlin/buttons/RootButtons.kt

package dev.inmo.plaguposter.ratings.source.buttons

import com.soywiz.klock.DateFormat
import dev.inmo.micro_utils.coroutines.runCatchingSafely
import dev.inmo.micro_utils.pagination.FirstPagePagination
import dev.inmo.micro_utils.pagination.Pagination
import dev.inmo.micro_utils.pagination.SimplePagination
import dev.inmo.micro_utils.pagination.utils.paginate
import dev.inmo.plaguposter.posts.repo.ReadPostsRepo
import dev.inmo.plaguposter.ratings.models.Rating
import dev.inmo.plaguposter.ratings.repo.RatingsRepo
import dev.inmo.plaguposter.ratings.utils.postsByRatings
import dev.inmo.tgbotapi.extensions.api.answers.answer
import dev.inmo.tgbotapi.extensions.api.edit.edit
import dev.inmo.tgbotapi.extensions.behaviour_builder.BehaviourContext
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onMessageDataCallbackQuery
import dev.inmo.tgbotapi.extensions.utils.formatting.makeLinkToMessage
import dev.inmo.tgbotapi.extensions.utils.types.buttons.dataButton
import dev.inmo.tgbotapi.extensions.utils.types.buttons.inlineKeyboard
import dev.inmo.tgbotapi.extensions.utils.types.buttons.urlButton
import dev.inmo.tgbotapi.types.ChatIdentifier
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardMarkup
import dev.inmo.tgbotapi.utils.row

const val RootButtonsShowRatingData = "ratings_buttons_show"
const val RootButtonsShowRatingPageData = "ratings_buttons_show_page"
const val RootButtonsToPageData = "ratings_buttons_to_page"

suspend fun RatingsRepo.buildRootButtons(
    pagination: Pagination = FirstPagePagination(16),
    rowSize: Int = 4
): InlineKeyboardMarkup {
    val postsByRatings = postsByRatings().toList().paginate(pagination)
    return inlineKeyboard {
        if (postsByRatings.pagesNumber > 1) {
            row {
                if (postsByRatings.page > 0) {
                    dataButton("<", "$RootButtonsToPageData ${postsByRatings.page - 1} ${postsByRatings.size}")
                }
                dataButton("${postsByRatings.page}: \uD83D\uDD04", "$RootButtonsToPageData ${postsByRatings.page} ${postsByRatings.size}")
                if (postsByRatings.pagesNumber - postsByRatings.page > 1) {
                    dataButton(">", "$RootButtonsToPageData ${postsByRatings.page + 1} ${postsByRatings.size}")
                }
            }
        }
        postsByRatings.results.chunked(rowSize).map {
            row {
                it.forEach { (rating, posts) ->
                    dataButton("${rating.double}: ${posts.size}", "$RootButtonsShowRatingData ${rating.double}")
                }
            }
        }
    }
}

val defaultPostCreationTimeFormat: DateFormat = DateFormat("dd.MM.yy HH:mm")

suspend fun RatingsRepo.buildRatingButtons(
    postsRepo: ReadPostsRepo,
    rating: Rating,
    pagination: Pagination = FirstPagePagination(8),
    rowSize: Int = 2,
    postCreationTimeFormat: DateFormat = defaultPostCreationTimeFormat
): InlineKeyboardMarkup {
    val postsByRatings = getPosts(rating .. rating, true).keys.paginate(pagination)
    return inlineKeyboard {
        if (postsByRatings.pagesNumber > 1) {
            row {
                if (postsByRatings.page > 0) {
                    dataButton("<", "$RootButtonsShowRatingPageData ${postsByRatings.page - 1} ${postsByRatings.size} ${rating.double}")
                }
                dataButton("${postsByRatings.page}: \uD83D\uDD04", "$RootButtonsShowRatingPageData ${postsByRatings.page} ${postsByRatings.size} ${rating.double}")
                if (postsByRatings.pagesNumber - postsByRatings.page > 1) {
                    dataButton(">", "$RootButtonsShowRatingPageData ${postsByRatings.page + 1} ${postsByRatings.size} ${rating.double}")
                }
            }
        }
        postsByRatings.results.chunked(rowSize).map {
            row {
                it.forEach { postId ->
                    val firstMessageInfo = postsRepo.getFirstMessageInfo(postId) ?: return@forEach
                    val postCreationTime = postsRepo.getPostCreationTime(postId) ?: return@forEach
                    urlButton(
                        postCreationTime.format(postCreationTimeFormat),
                        makeLinkToMessage(
                            firstMessageInfo.chatId,
                            firstMessageInfo.messageId
                        )
                    )
                }
            }
        }
        row {
            dataButton("↩️", "$RootButtonsToPageData 0 16")
        }
    }
}

suspend fun BehaviourContext.includeRootNavigationButtonsHandler(
    allowedChats: Set<ChatIdentifier>,
    ratingsRepo: RatingsRepo,
    postsRepo: ReadPostsRepo
) {
    suspend fun registerPageQueryListener(
        dataPrefix: String,
        onPageUpdate: suspend (pagination: Pagination, additionalParams: Array<String>) -> InlineKeyboardMarkup?
    ) {
        onMessageDataCallbackQuery(
            initialFilter = { it.message.chat.id in allowedChats }
        ) {
            val args = it.data.split(" ").takeIf { it.size >= 3 } ?: return@onMessageDataCallbackQuery
            val (prefix, pageRaw, sizeRaw) = args

            if (prefix == dataPrefix) {
                runCatchingSafely {
                    val page = pageRaw.toIntOrNull() ?: return@runCatchingSafely
                    val size = sizeRaw.toIntOrNull() ?: return@runCatchingSafely

                    edit(
                        it.message,
                        onPageUpdate(SimplePagination(page, size), args.drop(3).toTypedArray()) ?: return@runCatchingSafely
                    )
                }

                answer(it)
            }
        }
    }
    suspend fun registerPageQueryListener(
        dataPrefix: String,
        onPageUpdate: suspend (pagination: Pagination) -> InlineKeyboardMarkup?
    ) = registerPageQueryListener(dataPrefix) { pagination, _ ->
        onPageUpdate(pagination)
    }
    registerPageQueryListener(
        RootButtonsToPageData,
        ratingsRepo::buildRootButtons
    )
    registerPageQueryListener(
        RootButtonsShowRatingPageData
    ) { pagination, params ->
        params.firstOrNull() ?.toDoubleOrNull() ?.let { rating ->
            ratingsRepo.buildRatingButtons(postsRepo, Rating(rating), pagination)
        }
    }
    onMessageDataCallbackQuery(
        initialFilter = { it.message.chat.id in allowedChats }
    ) {
        val (prefix, ratingRaw) = it.data.split(" ").takeIf { it.size == 2 } ?: return@onMessageDataCallbackQuery

        if (prefix == RootButtonsShowRatingData) {
            runCatchingSafely {
                val rating = ratingRaw.toDoubleOrNull() ?: return@runCatchingSafely
                edit(it.message, ratingsRepo.buildRatingButtons(postsRepo, Rating(rating)))
            }

            answer(it)
        }
    }
}