BooruGrabberTelegramBot/src/main/kotlin/App.kt

import dev.inmo.krontab.utils.asFlow
import dev.inmo.micro_utils.coroutines.*
import dev.inmo.micro_utils.pagination.utils.doForAllWithNextPaging
import dev.inmo.micro_utils.repos.*
import dev.inmo.micro_utils.repos.cache.cache.FullKVCache
import dev.inmo.micro_utils.repos.cache.cache.KVCache
import dev.inmo.micro_utils.repos.cache.cached
import dev.inmo.micro_utils.repos.exposed.keyvalue.ExposedKeyValueRepo
import dev.inmo.micro_utils.repos.exposed.onetomany.ExposedKeyValuesRepo
import dev.inmo.micro_utils.repos.mappers.withMapper
import dev.inmo.tgbotapi.bot.ktor.telegramBot
import dev.inmo.tgbotapi.extensions.api.bot.getMe
import dev.inmo.tgbotapi.extensions.api.bot.setMyCommands
import dev.inmo.tgbotapi.extensions.api.delete
import dev.inmo.tgbotapi.extensions.api.send.media.*
import dev.inmo.tgbotapi.extensions.api.send.reply
import dev.inmo.tgbotapi.extensions.behaviour_builder.buildBehaviourWithLongPolling
import dev.inmo.tgbotapi.extensions.behaviour_builder.triggers_handling.onCommand
import dev.inmo.tgbotapi.requests.abstracts.FileUrl
import dev.inmo.tgbotapi.types.*
import dev.inmo.tgbotapi.types.chat.ChannelChat
import dev.inmo.tgbotapi.types.chat.PrivateChat
import dev.inmo.tgbotapi.types.media.TelegramMediaPhoto
import java.io.File
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import kotlinx.serialization.json.Json
import models.Config
import net.kodehawa.lib.imageboards.ImageBoard
import net.kodehawa.lib.imageboards.entities.BoardImage

/**
 * This method by default expects one argument in [args] field: telegram bot configuration
 */
suspend fun main(args: Array<String>) {
    // create json to decode config
    val json = Json { ignoreUnknownKeys = true }
    // decode config
    val config: Config = json.decodeFromString(Config.serializer(), File(args.first()).readText())
    // that is your bot
    val bot = telegramBot(config.token)

    ImageBoard.setUserAgent("WhoAmI?")

    // that is kotlin coroutine scope which will be used in requests and parallel works under the hood
    val scope = CoroutineScope(Dispatchers.Default + ContextSafelyExceptionHandler { it.printStackTrace() })

    val repo = ExposedKeyValueRepo(
        config.database.database,
        { long("chat_id") },
        { text("config") },
        "configs"
    ).withMapper(
        { chatId },
        { json.encodeToString(ChatSettings.serializer(), this) },
        { ChatId(this) },
        { json.decodeFromString(ChatSettings.serializer(), this) },
    ).cached(FullKVCache(), scope = scope)

    val chatsUrlsSeen = ExposedKeyValuesRepo(
        config.database.database,
        { long("chat_id") },
        { text("url") },
        "chatsUrlsSeen"
    ).withMapper(
        { chatId },
        { this },
        { ChatId(this) },
        { this },
    ).cached(KVCache(128), scope = scope)

    val chatsChangingMutex = Mutex()
    val chatsSendingJobs = mutableMapOf<ChatId, Job>()

    // here should be main logic of your bot
    bot.buildBehaviourWithLongPolling(scope) {
        // in this lambda you will be able to call methods without "bot." prefix
        val me = getMe()

        suspend fun triggerSendForChat(chatId: ChatId, settings: ChatSettings) {
            val result = let {
                val result = mutableListOf<BoardImage>()
                var i = 0
                while (result.size < settings.count) {
                    val images = settings.makeRequest(i).takeIf { it.isNotEmpty() } ?: break
                    result.addAll(
                        images.filterNot {
                            chatsUrlsSeen.contains(chatId, it.url)
                        }
                    )
                    i++
                }
                val toDrop = (result.size - settings.count).takeIf { it > 0 } ?: return@let result
                result.dropLast(toDrop)
            }.takeIf { it.isNotEmpty() } ?: return
            runCatchingSafely {
                val urls = result.map { it.url }
                chatsUrlsSeen.add(chatId, urls)
                when {
                    urls.isEmpty() -> return@runCatchingSafely
                    urls.size == 1 -> sendPhoto(
                        chatId,
                        FileUrl(urls.first()),
                        if (settings.attachUrls) urls.first() else null
                    )
                    settings.gallery -> urls.chunked(mediaCountInMediaGroup.last + 1).forEach {
                        sendVisualMediaGroup(
                            chatId,
                            it.map {
                                TelegramMediaPhoto(FileUrl(it), if (settings.attachUrls) it else null)
                            }
                        )
                    }
                    else -> urls.forEach {
                        sendPhoto(
                            chatId,
                            FileUrl(it),
                            if (settings.attachUrls) it else null
                        )
                    }
                }
            }.onFailure {
                triggerSendForChat(chatId, settings)
            }
        }

        suspend fun refreshChatJob(chatId: ChatId, settings: ChatSettings?) {
            val settings = settings ?: repo.get(chatId)
            chatsChangingMutex.withLock {
                chatsSendingJobs[chatId] ?.cancel()
                settings ?.scheduler ?.let {
                    chatsSendingJobs[chatId] = it.asFlow().subscribeSafelyWithoutExceptions(scope) {
                        triggerSendForChat(chatId, settings)
                    }
                }
            }
        }

        doForAllWithNextPaging {
            repo.keys(it).also {
                it.results.forEach {
                    runCatchingSafely {
                        refreshChatJob(it, null)
                    }
                }
            }
        }

        repo.onNewValue.subscribeSafelyWithoutExceptions(this) {
            refreshChatJob(it.first, it.second)
        }
        repo.onValueRemoved.subscribeSafelyWithoutExceptions(this) {
            refreshChatJob(it, null)
        }

        onCommand(Regex("(help|start)"), requireOnlyCommandInMessage = true) {
            reply(it, EnableArgsParser(onlyQueryIsRequired = false).getFormattedHelp().takeIf { it.isNotBlank() } ?: return@onCommand)
        }
        onCommand("enable", requireOnlyCommandInMessage = false) {
            val args = it.content.textSources.drop(1).joinToString("") { it.source }.split(" ")
            val parser = EnableArgsParser(onlyQueryIsRequired = false)
            runCatchingSafely {
                parser.parse(args)
                repo.set(ChatId(it.chat.id.chatId), parser.resultSettings ?: return@runCatchingSafely)
            }.onFailure { e ->
                e.printStackTrace()
                if (it.chat is PrivateChat) {
                    reply(it, parser.getFormattedHelp())
                }
            }
            runCatchingSafely {
                if (it.chat is ChannelChat) {
                    delete(it)
                }
            }
        }
        onCommand("request", requireOnlyCommandInMessage = false) {
            val args = it.content.textSources.drop(1).joinToString("") { it.source }.trim().takeIf { it.isNotBlank() } ?.split(" ")

            val chatSettings = if (args.isNullOrEmpty()) {
                repo.get(ChatId(it.chat.id.chatId)) ?: run {
                    if (it.chat is PrivateChat) {
                        reply(it, "Unable to find default config")
                    }
                    return@onCommand
                }
            } else {
                val parser = EnableArgsParser(onlyQueryIsRequired = true, repo.get(ChatId(it.chat.id.chatId)) ?: ChatSettings.DEFAULT)
                runCatchingSafely {
                    parser.parse(args)
                    parser.resultSettings
                }.onFailure { e ->
                    e.printStackTrace()
                    if (it.chat is PrivateChat) {
                        reply(it, parser.getFormattedHelp())
                    }
                }.getOrNull()
            }

            triggerSendForChat(ChatId(it.chat.id.chatId), chatSettings ?: return@onCommand)
        }
        onCommand("disable", requireOnlyCommandInMessage = true) {
            runCatchingSafely {
                repo.unset(ChatId(it.chat.id.chatId))
            }
            runCatchingSafely {
                delete(it)
            }
        }

        setMyCommands(
            listOf(
                BotCommand("start", "Will return the help for the enable command"),
                BotCommand("help", "Will return the help for the enable command"),
                BotCommand("request", "Will trigger image immediately with custom settings from arguments or default settings of chat if any"),
                BotCommand("enable", "Will enable images grabbing for current chat or update exists settings"),
                BotCommand("disable", "Will disable bot for current chat"),
            )
        )

        println(me)
    }.join()
}