Main.kt

import dev.inmo.tgbotapi.extensions.api.answers.answerCallbackQuery
import dev.inmo.tgbotapi.extensions.api.edit.edit
import dev.inmo.tgbotapi.extensions.api.send.reply
import dev.inmo.tgbotapi.extensions.api.send.sendMessage
import dev.inmo.tgbotapi.extensions.api.telegramBot
import dev.inmo.tgbotapi.extensions.behaviour_builder.buildBehaviourWithLongPolling
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onCommand
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onContentMessage
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onDataCallbackQuery
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onMessageDataCallbackQuery
import dev.inmo.tgbotapi.extensions.utils.botCommandTextSourceOrNull
import dev.inmo.tgbotapi.extensions.utils.extensions.raw.text
import dev.inmo.tgbotapi.extensions.utils.types.buttons.flatInlineKeyboard
import dev.inmo.tgbotapi.extensions.utils.types.buttons.inlineKeyboard
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.CallbackDataInlineKeyboardButton
import dev.inmo.tgbotapi.types.buttons.InlineKeyboardButtons.URLInlineKeyboardButton
import dev.inmo.tgbotapi.types.message.HTMLParseMode
import dev.inmo.tgbotapi.types.message.abstracts.CommonMessage
import dev.inmo.tgbotapi.types.message.abstracts.ContentMessage
import dev.inmo.tgbotapi.types.message.content.MessageContent
import dev.inmo.tgbotapi.types.message.content.TextContent
import dev.inmo.tgbotapi.utils.row

private val firstMenu = "<b>Menu 1</b>\n\nA beautiful menu with a shiny inline button.";
private val secondMenu = "<b>Menu 2</b>\n\nA better menu with even more shiny inline buttons.";
private val next = CallbackDataInlineKeyboardButton("Next", "next")
private val back = CallbackDataInlineKeyboardButton("Back", "back")
private val url = URLInlineKeyboardButton("Tutorial", "https://madhead.me/posts/from-botfather-to-hello-world")
private val firstMenuMarkup = flatInlineKeyboard { next }
private val secondMenuMarkup = inlineKeyboard {
    row { back }
    row { url }
}

suspend fun main(args: Array<String>) {
    val token = args.first()
    val bot = telegramBot(token)
    var screaming = false

    bot.buildBehaviourWithLongPolling() {
        onCommand("scream") {
            screaming = true
        }

        onCommand("whisper") {
            screaming = false
        }

        onCommand("menu") {
            bot.sendMessage(
                chat = it.chat,
                text = firstMenu,
                parseMode = HTMLParseMode,
                replyMarkup = firstMenuMarkup
            )
        }

        onContentMessage(
            initialFilter = CommonMessage<MessageContent>::isNotCommand
        ) {
            val text = it.text

            if (text != null) {
                bot.reply(it, if (screaming) text.uppercase() else text)
            } else {
                bot.reply(it, it)
            }
        }

        onMessageDataCallbackQuery("next") {
            bot.edit(
                chatId = it.message.chat.id,
                messageId = it.message.messageId,
                text = secondMenu,
                parseMode = HTMLParseMode,
                replyMarkup = secondMenuMarkup,
            )
        }

        onMessageDataCallbackQuery("back") {
            bot.edit(
                message = it.message as ContentMessage<TextContent>,
                text = firstMenu,
                parseMode = HTMLParseMode,
                replyMarkup = firstMenuMarkup,
            )
        }

        onDataCallbackQuery {
            bot.answerCallbackQuery(it)
        }
    }.join()
}

fun CommonMessage<*>.isNotCommand(): Boolean =
    (this.content as? TextContent)?.textSources?.none { it.botCommandTextSourceOrNull() != null } ?: true