FancyBot/src/main/kotlin/cn/rtast/fancybot/FancyBot.kt

/*
 * Copyright © 2024 RTAkland
 * Author: RTAkland
 * Date: 2024/8/26
 */


package cn.rtast.fancybot

import cn.rtast.fancybot.commands.misc.JueCommand
import cn.rtast.fancybot.commands.misc.PastebinCommand
import cn.rtast.fancybot.commands.misc.ReactionCommand
import cn.rtast.fancybot.commands.misc.ScanQRCodeCommand
import cn.rtast.fancybot.commands.misc.ShortLinkCommand.Companion.makeShortLink
import cn.rtast.fancybot.commands.parse.*
import cn.rtast.fancybot.commands.reply.ImageBedCommand
import cn.rtast.fancybot.commands.reply.SpeedUpGIFCommand
import cn.rtast.fancybot.entity.nailong.NaiLongDetectPayload
import cn.rtast.fancybot.entity.nailong.NaiLongDetectResponse
import cn.rtast.fancybot.enums.WSType
import cn.rtast.fancybot.util.*
import cn.rtast.fancybot.util.misc.ImageBed
import cn.rtast.fancybot.util.misc.convertToDate
import cn.rtast.fancybot.util.misc.initBackgroundTasks
import cn.rtast.fancybot.util.misc.initCommand
import cn.rtast.fancybot.util.misc.initFilesDir
import cn.rtast.fancybot.util.misc.initItems
import cn.rtast.fancybot.util.misc.initSetuIndex
import cn.rtast.fancybot.util.misc.makeGif
import cn.rtast.fancybot.util.misc.toBufferedImage
import cn.rtast.fancybot.util.misc.toByteArray
import cn.rtast.fancybot.util.misc.toGrayscale
import cn.rtast.fancybot.util.misc.toURL
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.fancybot.util.str.toJson
import cn.rtast.rob.ROneBotFactory
import cn.rtast.rob.entity.*
import cn.rtast.rob.entity.custom.BeKickEvent
import cn.rtast.rob.entity.custom.ErrorEvent
import cn.rtast.rob.entity.custom.MemberKickEvent
import cn.rtast.rob.entity.custom.MemberLeaveEvent
import cn.rtast.rob.entity.lagrange.FileEvent
import cn.rtast.rob.entity.lagrange.PokeEvent
import cn.rtast.rob.enums.ArrayMessageType
import cn.rtast.rob.enums.QQFace
import cn.rtast.rob.util.ob.MessageChain
import cn.rtast.rob.util.ob.OneBotAction
import cn.rtast.rob.util.ob.OneBotListener
import com.madgag.gif.fmsware.GifDecoder
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.delay
import kotlinx.coroutines.launch
import kotlinx.coroutines.withContext
import java.io.File
import java.net.URI

class FancyBot : OneBotListener {

    private val logger = Logger.getLogger<FancyBot>()
    private val coroutineScope = CoroutineScope(Dispatchers.IO)

    override suspend fun onWebsocketOpenEvent(action: OneBotAction) {
        action.sendPrivateMessage(configManager.noticeUser, "FancyBot启动完成~")
    }

    override suspend fun onGroupMessage(message: GroupMessage, json: String) {
        val sender = message.sender.nickname
        val senderId = message.sender.userId
        val msg = message.rawMessage
        val groupId = message.groupId
        val messageId = message.messageId
        logger.info("$sender($senderId: $groupId >>> $messageId): $msg")
        logger.trace("$sender($senderId: $groupId >>> $messageId: $json")

        if (message.message.any { it.type == ArrayMessageType.face && it.data.id.toString() == "419" }) {
            message.reply("你发牛魔的火车呢, 我直接就是打断")
        }

        if (message.text.contains("原神")) {
            (0..10).forEach { _ -> message.reaction(QQFace.entries.random()) }
            message.reply("你原神牛魔呢")
            delay(1500L)
            message.reply("但是话又说回来.....启动！！")
        }

        GitHubParseCommand.parse(message)
        ReverseGIFCommand.callback(message)
        AsciiArtCommand.callback(message)
        ScanQRCodeCommand.callback(message)
        BiliVideoParseCommand.parse(message)
        BiliUserParseCommand.parse(message)
        DouyinVideoParseCommand.parse(message)
        YoutubeVideoParseCommand.parse(message)

        if (message.text.toList().any { it in arrayListOf('*', '-', '/', '+', '=') }) {
            val calculateResult = CalculateCommand.parse(message.rawMessage)
            calculateResult?.let { message.reply(calculateResult) }
        }

        if (message.message.any { it.type == ArrayMessageType.at }) {
            if (message.text.contains("/取头像")) {
                val target = message.message.find { it.type == ArrayMessageType.at }?.data?.qq!!
                val avatar = JueCommand.AVATAR_REPLACE_URL.replace("#{}", target)
                val msg = MessageChain.Builder()
                if (message.text.contains("黑白")) {
                    val greyscale = avatar.toURL()
                        .readBytes().toBufferedImage()
                        .toGrayscale().toByteArray()
                    val greyscaleShortLink = ImageBed.upload(greyscale).makeShortLink()
                    msg.addImage(greyscale.encodeToBase64(), true)
                        .addText(greyscaleShortLink)
                } else {
                    val shortLink = avatar.makeShortLink()
                    msg.addImage(avatar).addText(shortLink)
                }
                message.reply(msg.build())
            }
        }

        if (message.message.any { it.type == ArrayMessageType.reply }) {
            val command = message.message.reversed().find { it.type == ArrayMessageType.text }!!.data.text!!
            val replyId = message.message.find { it.type == ArrayMessageType.reply }!!.data.id!!
            val getMsg = message.action.getMessage(replyId.toString().toLong())
            val plainTextContent = getMsg.message
                .filter { it.type == ArrayMessageType.text }
                .joinToString { it.data.text!! }
            if (command.contains("/倒放") || command.contains("/df")) ReverseGIFCommand.reverse(message, getMsg)
            if (command.contains("/图来") || command.contains("/图链")) ImageURLCommand.callback(message, getMsg)
            if (command.contains("/图床")) ImageBedCommand.execute(getMsg, message)
            if (command.contains("/reaction")) ReactionCommand.reaction(
                message.action,
                message.groupId,
                getMsg.messageId
            )
            if (command.contains("/sl") || command.contains("/short")) message.reply(
                plainTextContent.trim().makeShortLink()
            )
            if (command.contains("/ascii") || command.contains("/asc")) {
                val url = AsciiArtCommand.getImageUrl(getMsg)
                val image = AsciiArtCommand.generateAsciiArt(url)
                message.reply(image)
            }
            if (command.contains("/pb") || command.contains("/pastebin")) {
                val pastebinUrl = PastebinCommand.createPastebin(plainTextContent)
                message.reply(pastebinUrl)
            }
            if (command.contains("/加速")) {
                val multiply = command.split("加速").last().toFloat()
                SpeedUpGIFCommand.speedUp(message, getMsg, multiply)
            }
            if (command.contains("/黑白")) {
                val msg = MessageChain.Builder()
                getMsg.message.filter { it.type == ArrayMessageType.image || it.type == ArrayMessageType.mface }
                    .forEach {
                        val imgUrl = it.data.file!!
                        val decoder = GifDecoder()
                        val imageStream = withContext(Dispatchers.IO) { imgUrl.toURL().openStream() }
                        decoder.read(imageStream)
                        if (decoder.frameCount == 0) {
                            msg.addImage(imageStream.readBytes().toBufferedImage().toByteArray().encodeToBase64(), true)
                        } else {
                            val frames = (0 until decoder.frameCount).map { decoder.getFrame(it).toGrayscale() }
                            val base64 = decoder.makeGif(frames).encodeToBase64()
                            msg.addImage(base64, true)
                        }
                    }
                message.reply(msg.build())
            }
        }

        coroutineScope.launch {
            if (message.images.isNotEmpty()) {
                message.images.forEach {
                    val payload = NaiLongDetectPayload(it.file).toJson()
                    val result = Http.post<NaiLongDetectResponse>(configManager.naiLongApiUrl, payload)
                    logger.info("奶龙识别结果: ${result.result}")
                    if (result.result) {
                        message.reply("本群禁止发奶龙!")
                    }
                }
            }
            message.message.forEach {
                if (it.type == ArrayMessageType.image) {
                    val filename = it.data.file!!.split("=").last() + ".png"
                    URI(it.data.file!!).toURL().openConnection().inputStream.use { input ->
                        File("$ROOT_PATH/caches/images/$filename").outputStream().use { output ->
                            output.write(input.readBytes())
                        }
                    }
                }
            }
        }
    }

    override suspend fun onGroupMessageRevoke(message: GroupRevokeMessage) {
        if (!configManager.enableAntiRevoke) return
        val msg = MessageChain.Builder()
            .addText("用户: ${message.userId} 被: ${message.operatorId} 撤回了一条消息")
            .addNewLine()
            .addText("使用/revoke ${message.messageId} 来获取被撤回的消息")
            .build()
        message.action.sendGroupMessage(message.groupId, msg)
    }

    override suspend fun onWebsocketErrorEvent(event: ErrorEvent) {
        event.exception.printStackTrace()
    }

    override suspend fun onAddFriendRequest(event: AddFriendRequestEvent) {
        event.approve()
    }

    override suspend fun onGroupFileUpload(event: FileEvent) {
        event.saveTo("$ROOT_PATH/caches/")
    }

    override suspend fun onLeaveEvent(event: MemberLeaveEvent) {
        val nickname = event.action.getGroupMemberInfo(event.groupId, event.userId).nickname
        val msg = MessageChain.Builder()
            .addText("有人坐飞船离开了本星系~(主动退出)")
            .addNewLine()
            .addText("QQ: $nickname(${event.userId})")
            .addNewLine()
            .addText("下次再见吧~~~")
            .build()
        event.action.sendGroupMessage(event.groupId, msg)
    }

    override suspend fun onMemberKick(event: MemberKickEvent) {
        val nickname = event.action.getGroupMemberInfo(event.groupId, event.userId).nickname
        val msg = MessageChain.Builder()
            .addText("有人坐飞船离开了本星系~(管理员踢出)")
            .addNewLine()
            .addText("QQ: $nickname(${event.userId})")
            .addNewLine()
            .addText("下次再见吧~~~")
            .build()
        event.action.sendGroupMessage(event.groupId, msg)
    }

    override suspend fun onGroupPoke(event: PokeEvent) {
        if (event.targetId == configManager.selfId) {
            val msg = MessageChain.Builder()
                .addAt(event.userId)
                .addText("你${event.pokeAction.first()}牛魔呢")
                .build()
            event.action.sendGroupMessage(event.groupId!!, msg)
        }
    }

    override suspend fun onBeKicked(event: BeKickEvent) {
        blackListManager.insertGroup(event.groupId, event.operator, event.time)
        event.action.sendPrivateMessage(
            configManager.noticeUser,
            "被: ${event.groupId}踢出! 操作人: ${event.operator} 时间: ${event.time.convertToDate()} 已将其拉入黑名单!"
        )
    }
}

suspend fun main() {
    val fancyBot = FancyBot()
    val workType = configManager.wsType
    val accessToken = configManager.accessToken
    val instance = when (workType) {
        WSType.Client -> ROneBotFactory.createClient(configManager.wsAddress, accessToken, fancyBot)
        WSType.Server -> ROneBotFactory.createServer(configManager.wsPort, accessToken, fancyBot)
    }
    instance.addListeningGroups(*configManager.listeningGroups.toLongArray())

    initDatabase()
    initFilesDir()
    initCommand()
    initSetuIndex()
    initItems()
    initBackgroundTasks(instance)
}