fix: fix forward msg bug

This commit is contained in:
2024-09-23 18:22:27 +08:00
parent 07a94fda32
commit a29f9ba00a
3 files changed
+47 -20

No files matched your search

@@ -9,26 +9,41 @@ package cn.rtast.fancybot.commands.misc
import cn.rtast.fancybot.entity.Setu
import cn.rtast.fancybot.util.Http
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.rob.entity.GroupMessage
import cn.rtast.rob.segment.Image
import cn.rtast.rob.segment.Node
import cn.rtast.rob.util.BaseCommand
import cn.rtast.rob.util.ob.MessageChain
import cn.rtast.rob.util.ob.NodeMessageChain
import cn.rtast.rob.util.ob.OneBotListener
import java.net.URI
class TenSetuCommand : BaseCommand() {
override val commandNames = listOf("十张色图")
override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
val nodeImages = mutableListOf<Image>()
val nodeMsg = NodeMessageChain.Builder()
repeat(10) {
val response = Http.get<Setu>("https://api.rtast.cn/api/setu")
if (!response.r18) nodeImages.add(Image(Image.Data(response.urls.large)))
if (!response.r18) {
try {
val imageBase64 = URI(response.urls.large).toURL().readBytes().encodeToBase64()
val tempMsg = MessageChain.Builder().addImage(imageBase64, true).build()
val tempNode = Node(
Node.Data("", message.userId.toString(), tempMsg.finalArrayMsgList)
)
nodeMsg.addNode(tempNode)
} catch (_: Exception) {
}
}
}
val node = Node(Node.Data("ROBOT", "1845464277", nodeImages))
val nodeMsg = NodeMessageChain.Builder()
.addNode(node)
val footerMsg = MessageChain.Builder()
.addText("图片数量可能不足10张~")
.addNewLine()
.addText("因为过滤了R18的图片~")
.build()
listener.sendGroupForwardMsg(message.groupId, nodeMsg)
val footerNode = Node(Node.Data("", message.userId.toString(), footerMsg.finalArrayMsgList))
nodeMsg.addNode(footerNode)
listener.sendGroupForwardMsg(message.groupId, nodeMsg.build())
}
}