com.github.insanusmokrassar.AutoPostTelegramBot.plugins.scheduler.commands.GetSchedulesCommand.kt Maven / Gradle / Ivy

Go to download
package com.github.insanusmokrassar.AutoPostTelegramBot.plugins.scheduler.commands

import com.github.insanusmokrassar.AutoPostTelegramBot.base.database.tables.PostsMessagesTable
import com.github.insanusmokrassar.AutoPostTelegramBot.base.plugins.commonLogger
import com.github.insanusmokrassar.AutoPostTelegramBot.plugins.scheduler.PostsSchedulesTable
import com.github.insanusmokrassar.AutoPostTelegramBot.utils.commands.Command
import com.github.insanusmokrassar.AutoPostTelegramBot.utils.extensions.executeSync
import com.pengrad.telegrambot.TelegramBot
import com.pengrad.telegrambot.model.Message
import com.pengrad.telegrambot.request.ForwardMessage
import com.pengrad.telegrambot.request.SendMessage
import kotlinx.coroutines.experimental.delay
import kotlinx.coroutines.experimental.launch
import java.lang.ref.WeakReference

class GetSchedulesCommand(
    private val postsSchedulesTable: PostsSchedulesTable,
    private val botWR: WeakReference,
    private val sourceChatId: Long
) : Command() {
    override val commandRegex: Regex = Regex("^/getPublishSchedule( \\d+)?$")

    override fun onCommand(updateId: Int, message: Message) {
        val bot = botWR.get() ?: return

        val chatId = message.from() ?.id() ?: message.chat().id()

        val count = message.text().split(" ").let {
            if (it.size == 1) {
                null
            } else {
                it[1].toInt()
            }
        }

        try {
            bot.executeSync(
                SendMessage(
                    chatId,
                    "Let me prepare data"
                )
            )
        } catch (e: Exception) {
            commonLogger.throwing(
                this::class.java.simpleName,
                "sending message \"Prepare data\" to user",
                e
            )
            return
        }

        launch {
            postsSchedulesTable.registeredPostsTimes().sortedBy {
                it.second
            }.let {
                if (count == null || it.size <= count) {
                    it
                } else {
                    it.subList(0, count)
                }
            }.forEach {
                bot.executeSync(
                    ForwardMessage(
                        chatId,
                        sourceChatId,
                        PostsMessagesTable.getMessagesOfPost(it.first).firstOrNull() ?.messageId ?: return@forEach
                    )
                )
                bot.executeSync(
                    SendMessage(
                        chatId,
                        "Post time: ${it.second}"
                    )
                )
                delay(1000)
            }
        }
    }
}