chore: improve use experience
This commit is contained in:
2 files changed
+75
-60
No files matched your search
@@ -65,6 +65,8 @@ class FancyBot : OneBotListener {
|
||||
val messageId = message.messageId
|
||||
println("$sender($senderId: $groupId >>> $messageId): $msg")
|
||||
|
||||
AsciiArtCommand.callback(message)
|
||||
|
||||
if (message.rawMessage.startsWith("https://github.com/") || message.rawMessage.startsWith("git@github.com:")) {
|
||||
GitHubParseCommand.parse(this, message)
|
||||
}
|
||||
|
||||
@@ -22,10 +22,12 @@ import javax.imageio.ImageIO
|
||||
import kotlin.math.roundToInt
|
||||
|
||||
class AsciiArtCommand : BaseCommand() {
|
||||
override val commandNames = listOf("/ascii")
|
||||
override val commandNames = listOf("/ascii", "/asc")
|
||||
|
||||
private val fontSize = 12
|
||||
private val font = Font("Monospaced", Font.PLAIN, fontSize)
|
||||
companion object {
|
||||
private const val FONT_SIZE = 12
|
||||
private val font = Font("Monospaced", Font.PLAIN, FONT_SIZE)
|
||||
val waitingList = mutableListOf<Long>()
|
||||
|
||||
private fun BufferedImage.convertToAscii(): String {
|
||||
val width = this.width
|
||||
@@ -60,9 +62,10 @@ class AsciiArtCommand : BaseCommand() {
|
||||
return asciiArt.toString()
|
||||
}
|
||||
|
||||
|
||||
private fun String.saveAsciiArtToImage(width: Int, height: Int): String {
|
||||
val imgWidth = width * fontSize / 2
|
||||
val imgHeight = height * fontSize / 2
|
||||
val imgWidth = width * FONT_SIZE / 2
|
||||
val imgHeight = height * FONT_SIZE / 2
|
||||
val img = BufferedImage(imgWidth, imgHeight, BufferedImage.TYPE_INT_RGB)
|
||||
val g2d = img.createGraphics()
|
||||
g2d.color = Color.BLACK
|
||||
@@ -71,30 +74,40 @@ class AsciiArtCommand : BaseCommand() {
|
||||
g2d.font = font
|
||||
val lines = this.split("\n")
|
||||
for ((index, line) in lines.withIndex()) {
|
||||
g2d.drawString(line, 0, (index + 1) * fontSize)
|
||||
g2d.drawString(line, 0, (index + 1) * FONT_SIZE)
|
||||
}
|
||||
g2d.dispose()
|
||||
return img.toByteArray().encodeToBase64()
|
||||
}
|
||||
|
||||
override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
|
||||
if (args.isEmpty()) {
|
||||
val msg = MessageChain.Builder().addText("发送`/ascii <图片>`可以将图片转成ascii字符").build()
|
||||
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!!
|
||||
val bufferedImage = ImageIO.read(URI(url).toURL())
|
||||
val imageBase64 =
|
||||
bufferedImage.convertToAscii().saveAsciiArtToImage(bufferedImage.width, bufferedImage.height)
|
||||
val msg = MessageChain.Builder().addImage(imageBase64, true).build()
|
||||
message.reply(msg)
|
||||
return
|
||||
} else {
|
||||
message.reply("回复错误已取消本次操作")
|
||||
}
|
||||
val image = message.message.find { it.type == ArrayMessageType.image }
|
||||
if (image == null) {
|
||||
val msg = MessageChain.Builder().addText("输入一个图片来继续操作").build()
|
||||
waitingList.removeIf { it == message.sender.userId }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
|
||||
if (message.sender.userId !in waitingList) {
|
||||
val msg = MessageChain.Builder()
|
||||
.addText("请继续发送一张图片,如果输入错误则取消本次操作")
|
||||
.build()
|
||||
message.reply(msg)
|
||||
waitingList.add(message.sender.userId)
|
||||
return
|
||||
} else {
|
||||
val url = image.data.file!!
|
||||
val bufferedImage = ImageIO.read(URI(url).toURL())
|
||||
val ascii = bufferedImage.convertToAscii()
|
||||
val image = ascii.saveAsciiArtToImage(bufferedImage.width, bufferedImage.height)
|
||||
val msg = MessageChain.Builder().addImage(image, true).build()
|
||||
message.reply(msg)
|
||||
message.reply("先发送`/ascii`再继续操作吧~")
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user