diff --git a/ratings/source/src/jvmMain/kotlin/Plugin.kt b/ratings/source/src/jvmMain/kotlin/Plugin.kt index a9a2631..9cb3228 100644 --- a/ratings/source/src/jvmMain/kotlin/Plugin.kt +++ b/ratings/source/src/jvmMain/kotlin/Plugin.kt @@ -22,6 +22,7 @@ import dev.inmo.plaguposter.ratings.models.Rating import dev.inmo.plaguposter.ratings.repo.RatingsRepo import dev.inmo.plaguposter.ratings.source.models.* import dev.inmo.plaguposter.ratings.source.repos.* +import dev.inmo.plaguposter.ratings.utils.postsByRatings import dev.inmo.tgbotapi.extensions.api.answers.answer import dev.inmo.tgbotapi.extensions.api.delete import dev.inmo.tgbotapi.extensions.api.edit.edit @@ -34,8 +35,9 @@ import dev.inmo.tgbotapi.extensions.utils.extensions.sameMessage import dev.inmo.tgbotapi.extensions.utils.types.buttons.dataButton import dev.inmo.tgbotapi.extensions.utils.types.buttons.flatInlineKeyboard import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.CallbackDataInlineKeyboardButton +import dev.inmo.tgbotapi.types.message.textsources.bold import dev.inmo.tgbotapi.types.message.textsources.regular -import kotlinx.coroutines.flow.filter +import dev.inmo.tgbotapi.utils.buildEntities import kotlinx.coroutines.flow.first import kotlinx.serialization.Serializable import kotlinx.serialization.json.* @@ -79,6 +81,7 @@ object Plugin : Plugin { val postsRepo = koin.get() val config = koin.get() val panelApi = koin.getOrNull() + val chatConfig = koin.get() onPollUpdates (markerFactory = { it.id }) { poll -> val postId = pollsToPostsIdsRepo.get(poll.id) ?: return@onPollUpdates @@ -208,6 +211,22 @@ object Plugin : Plugin { } } } + onCommand("ratings", requireOnlyCommandInMessage = true) { + if (it.chat.id == chatConfig.sourceChatId) { + val ratings = ratingsRepo.postsByRatings().toList().sortedByDescending { it.first } + val textSources = buildEntities { + + "Ratings amount: " + bold("${ratings.sumOf { it.second.size }}") + "\n\n" + ratings.forEach { + + "• " + bold("% 3.1f".format(it.first.double)) + ": " + bold(it.second.size.toString()) + "\n" + } + } + runCatchingSafely { + edit(it, textSources) + }.onFailure { _ -> + reply(it, textSources) + } + } + } koin.getOrNull() ?.apply { addTemplate( diff --git a/ratings/src/commonMain/kotlin/utils/PostsByRatings.kt b/ratings/src/commonMain/kotlin/utils/PostsByRatings.kt index 2c939f0..ab7176a 100644 --- a/ratings/src/commonMain/kotlin/utils/PostsByRatings.kt +++ b/ratings/src/commonMain/kotlin/utils/PostsByRatings.kt @@ -1,8 +1,15 @@ package dev.inmo.plaguposter.ratings.utils import dev.inmo.micro_utils.pagination.utils.getAll +import dev.inmo.micro_utils.repos.pagination.getAll +import dev.inmo.plaguposter.posts.models.PostId +import dev.inmo.plaguposter.ratings.models.Rating import dev.inmo.plaguposter.ratings.repo.RatingsRepo -suspend fun RatingsRepo.postsByRatings() { - getAll() +suspend fun RatingsRepo.postsByRatings(): Map> { + return getAll { keys(it) }.groupBy { + it.second + }.map { + it.key to it.value.map { it.first } + }.toMap() }