FancyBot/src/main/kotlin/cn/rtast/fancybot/commands/misc/HTTPCatCommand.kt

/*
 * Copyright © 2024 RTAkland
 * Author: RTAkland
 * Date: 2024/10/23
 */


package cn.rtast.fancybot.commands.misc

import cn.rtast.fancybot.annotations.CommandDescription
import cn.rtast.fancybot.enums.HttpStatusCode
import cn.rtast.fancybot.util.misc.Resources
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.rob.entity.GroupMessage
import cn.rtast.rob.util.BaseCommand
import cn.rtast.rob.util.ob.MessageChain
import cn.rtast.rob.util.ob.OneBotListener

@CommandDescription("随机HTTP猫猫")
class HTTPCatCommand : BaseCommand() {
    override val commandNames = listOf("http猫猫", "HTTP猫猫")

    private val notFoundCatBase64 = Resources.loadFromResourcesAsBytes("httpcat/0.jpg")!!.encodeToBase64()

    private fun loadHTTPCat(code: String): String {
        return Resources.loadFromResourcesAsBytes("httpcat/${code}.jpg")?.encodeToBase64() ?: notFoundCatBase64
    }

    override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
        val statusCode = if (args.isEmpty()) HttpStatusCode.entries.random().code.toString() else args.first()
        val image = loadHTTPCat(statusCode)
        val msg = MessageChain.Builder().addImage(image, true).build()
        message.reply(msg)
    }
}