import dev.inmo.micro_utils.coroutines.subscribeSafelyWithoutExceptions
import dev.inmo.tgbotapi.extensions.api.files.downloadFile
import dev.inmo.tgbotapi.extensions.api.files.downloadFileToTemp
import dev.inmo.tgbotapi.extensions.api.get.getFileAdditionalInfo
import dev.inmo.tgbotapi.extensions.api.send.reply
import dev.inmo.tgbotapi.extensions.api.send.replyWithAnimation
import dev.inmo.tgbotapi.extensions.api.send.replyWithAudio
import dev.inmo.tgbotapi.extensions.api.send.replyWithDocument
import dev.inmo.tgbotapi.extensions.api.send.replyWithMediaGroup
import dev.inmo.tgbotapi.extensions.api.send.replyWithPhoto
import dev.inmo.tgbotapi.extensions.api.send.replyWithSticker
import dev.inmo.tgbotapi.extensions.api.send.replyWithVideo
import dev.inmo.tgbotapi.extensions.api.send.replyWithVideoNote
import dev.inmo.tgbotapi.extensions.api.send.replyWithVoice
import dev.inmo.tgbotapi.extensions.api.send.withAction
import dev.inmo.tgbotapi.extensions.behaviour_builder.telegramBotWithBehaviourAndLongPolling
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.onMedia
import dev.inmo.tgbotapi.requests.abstracts.asMultipartFile
import dev.inmo.tgbotapi.requests.send.SendAction
import dev.inmo.tgbotapi.types.actions.BotAction
import dev.inmo.tgbotapi.types.actions.TypingAction
import dev.inmo.tgbotapi.types.media.TelegramMediaAudio
import dev.inmo.tgbotapi.types.media.TelegramMediaDocument
import dev.inmo.tgbotapi.types.media.TelegramMediaPhoto
import dev.inmo.tgbotapi.types.media.TelegramMediaVideo
import dev.inmo.tgbotapi.types.message.content.AnimationContent
import dev.inmo.tgbotapi.types.message.content.AudioContent
import dev.inmo.tgbotapi.types.message.content.DocumentContent
import dev.inmo.tgbotapi.types.message.content.MediaGroupContent
import dev.inmo.tgbotapi.types.message.content.PhotoContent
import dev.inmo.tgbotapi.types.message.content.StickerContent
import dev.inmo.tgbotapi.types.message.content.VideoContent
import dev.inmo.tgbotapi.types.message.content.VideoNoteContent
import dev.inmo.tgbotapi.types.message.content.VoiceContent
import dev.inmo.tgbotapi.utils.filenameFromUrl
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import java.io.File

/**
 * This bot will download incoming files
 */
suspend fun main(args: Array<String>) {
    val botToken = args.first()
    val directoryOrFile = args.getOrNull(1) ?.let { File(it) } ?: File("/tmp/")
    directoryOrFile.mkdirs()

    telegramBotWithBehaviourAndLongPolling(botToken, CoroutineScope(Dispatchers.IO)) {
        onCommand("start") {
            reply(it, "Send me any media (like photo or video) to download it")
        }
        onMedia(initialFilter = null) {
            val content = it.content
            val pathedFile = bot.getFileAdditionalInfo(content.media)
            val outFile = File(directoryOrFile, pathedFile.filePath.filenameFromUrl)
            runCatching {
                bot.downloadFile(content.media, outFile)
            }.onFailure {
                it.printStackTrace()
            }.onSuccess { _ ->
                reply(it, "Saved to ${outFile.absolutePath}")
                withAction(it.chat.id, TypingAction) {
                    when (content) {
                        is PhotoContent -> replyWithPhoto(
                            it,
                            outFile.asMultipartFile()
                        )
                        is AnimationContent -> replyWithAnimation(
                            it,
                            outFile.asMultipartFile()
                        )
                        is VideoContent -> replyWithVideo(
                            it,
                            outFile.asMultipartFile()
                        )
                        is StickerContent -> replyWithSticker(
                            it,
                            outFile.asMultipartFile()
                        )
                        is MediaGroupContent<*> -> replyWithMediaGroup(
                            it,
                            content.group.map {
                                when (val innerContent = it.content) {
                                    is AudioContent -> TelegramMediaAudio(
                                        downloadFileToTemp(innerContent.media).asMultipartFile()
                                    )
                                    is DocumentContent -> TelegramMediaDocument(
                                        downloadFileToTemp(innerContent.media).asMultipartFile()
                                    )
                                    is PhotoContent -> TelegramMediaPhoto(
                                        downloadFileToTemp(innerContent.media).asMultipartFile()
                                    )
                                    is VideoContent -> TelegramMediaVideo(
                                        downloadFileToTemp(innerContent.media).asMultipartFile()
                                    )
                                }
                            }
                        )
                        is AudioContent -> replyWithAudio(
                            it,
                            outFile.asMultipartFile()
                        )
                        is DocumentContent -> replyWithDocument(
                            it,
                            outFile.asMultipartFile()
                        )
                        is VoiceContent -> replyWithVoice(
                            it,
                            outFile.asMultipartFile()
                        )
                        is VideoNoteContent -> replyWithVideoNote(
                            it,
                            outFile.asMultipartFile()
                        )
                    }
                }
            }
        }
        allUpdatesFlow.subscribeSafelyWithoutExceptions(this) { println(it) }
    }.second.join()
}