From 917d4f0d2681280cf4bfc867a426b0d2b5a311f0 Mon Sep 17 00:00:00 2001 From: RTAkland Date: Thu, 26 Sep 2024 12:33:09 +0800 Subject: [PATCH] fix: fix mface and image ascii art generator --- .../cn/rtast/fancybot/commands/parse/AsciiArtCommand.kt | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/main/kotlin/cn/rtast/fancybot/commands/parse/AsciiArtCommand.kt b/src/main/kotlin/cn/rtast/fancybot/commands/parse/AsciiArtCommand.kt index 7651c43..a59679e 100644 --- a/src/main/kotlin/cn/rtast/fancybot/commands/parse/AsciiArtCommand.kt +++ b/src/main/kotlin/cn/rtast/fancybot/commands/parse/AsciiArtCommand.kt @@ -86,8 +86,11 @@ class AsciiArtCommand : BaseCommand() { suspend fun callback(message: GroupMessage) { if (message.sender.userId in waitingList) { - if (message.message.any { it.type == ArrayMessageType.image }) { - val url = message.message.find { it.type == ArrayMessageType.image }!!.data.file!! + if (message.message.any { it.type == ArrayMessageType.image } + || message.message.any { it.type == ArrayMessageType.mface }) { + val url = if (message.message.any { it.type == ArrayMessageType.image }) + message.message.find { it.type == ArrayMessageType.image }!!.data.file!! + else message.message.find { it.type == ArrayMessageType.mface }!!.data.url!! val bufferedImage = withContext(Dispatchers.IO) { ImageIO.read(URI(url).toURL()) } val imageBase64 = bufferedImage.convertToAscii().saveAsciiArtToImage(bufferedImage.width, bufferedImage.height)