49 lines
1.8 KiB
Kotlin
49 lines
1.8 KiB
Kotlin
import dev.inmo.micro_utils.fsm.common.State
|
|
import dev.inmo.tgbotapi.extensions.api.send.sendMessage
|
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.*
|
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.expectations.*
|
|
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.*
|
|
import dev.inmo.tgbotapi.extensions.utils.extensions.parseCommandsWithParams
|
|
import dev.inmo.tgbotapi.extensions.utils.formatting.*
|
|
import dev.inmo.tgbotapi.types.ChatId
|
|
import dev.inmo.tgbotapi.types.message.abstracts.CommonMessage
|
|
import dev.inmo.tgbotapi.types.message.content.TextContent
|
|
import kotlinx.coroutines.*
|
|
|
|
sealed interface BotState : State
|
|
data class ExpectContentOrStopState(override val context: ChatId, val sourceMessage: CommonMessage<TextContent>) : BotState
|
|
data class StopState(override val context: ChatId) : BotState
|
|
|
|
suspend fun main(args: Array<String>) {
|
|
val botToken = args.first()
|
|
|
|
telegramBotWithBehaviourAndFSMAndStartLongPolling(botToken, CoroutineScope(Dispatchers.IO)) {
|
|
strictlyOn<ExpectContentOrStopState> {
|
|
sendMessage(
|
|
it.context,
|
|
buildEntities {
|
|
+"Send me some content or " + botCommand("stop") + " if you want to stop sending"
|
|
}
|
|
)
|
|
|
|
val content = waitContentMessage().first()
|
|
when {
|
|
content is TextContent && content.parseCommandsWithParams().keys.contains("stop") -> StopState(it.context)
|
|
else -> {
|
|
execute(content.createResend(it.context))
|
|
it
|
|
}
|
|
}
|
|
}
|
|
strictlyOn<StopState> {
|
|
sendMessage(it.context, "You have stopped sending of content")
|
|
|
|
null
|
|
}
|
|
|
|
command("start") {
|
|
startChain(ExpectContentOrStopState(it.chat.id, it))
|
|
}
|
|
}.second.join()
|
|
}
|