RTAkland/FancyBot
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: improve code and add image type option

Browse Source
This commit is contained in:
RTAkland
2024-09-14 20:24:54 +08:00
parent 86581e7998
commit eb65a89425
8 files changed
+27 -15
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/main/kotlin/cn/rtast/fancybot/commands/lookup/WeatherCommand.kt
+1 -1
View File
@@ -37,7 +37,7 @@ class WeatherCommand : BaseCommand() {
val emptyImage = BufferedImage(canvasWidth, canvasHeight, BufferedImage.TYPE_INT_RGB)
val g2d = emptyImage.createGraphics()
val weatherIcon = Resources.loadFromResourcesAsBytes("qweather/${weather.now.icon}.png")
?: Resources.loadFromResourcesAsBytes("qweather/100.png")!!
?: Resources.loadFromResourcesAsBytes("qweather/999.png")!!
g2d.color = Color(209, 238, 238)
g2d.fillRect(0, 0, canvasWidth, canvasHeight)
g2d.drawCustomImage(weatherIcon.toBufferedImage(), canvasWidth / 2 - 230, canvasHeight / 2 - 80, 230.0, 230.0)
src/main/kotlin/cn/rtast/fancybot/commands/parse/BVParseCommand.kt
+4 -7
View File
@@ -15,6 +15,7 @@ import cn.rtast.fancybot.util.drawCustomImage
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.fancybot.util.str.formatNumber
import cn.rtast.fancybot.util.str.setTruncat
import cn.rtast.fancybot.util.toByteArray
import cn.rtast.rob.entity.GroupMessage
import cn.rtast.rob.util.ob.MessageChain
import cn.rtast.rob.util.ob.OBMessage
@@ -24,7 +25,6 @@ import java.awt.Color
import java.awt.Font
import java.awt.RenderingHints
import java.awt.image.BufferedImage
import java.io.ByteArrayOutputStream
import java.net.URI
import javax.imageio.ImageIO
@@ -69,8 +69,8 @@ object BVParseCommand {
): String {
val coverImage = ImageIO.read(URI(picUrl).toURL())
val faceImage = ImageIO.read(URI(authorFace).toURL())
val backgroundImage = BufferedImage(CANVAS_WIDTH, CANVAS_HEIGHT, BufferedImage.TYPE_INT_RGB)
val g2d = backgroundImage.createGraphics()
val canvas = BufferedImage(CANVAS_WIDTH, CANVAS_HEIGHT, BufferedImage.TYPE_INT_RGB)
val g2d = canvas.createGraphics()
g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON)
g2d.color = Color(43, 43, 43)
g2d.fillRect(0, 0, CANVAS_WIDTH, CANVAS_HEIGHT)
@@ -109,10 +109,7 @@ object BVParseCommand {
// draw cover image
g2d.drawCustomImage(coverImage, 430, 140, 600.0, 300.0, true)
g2d.dispose()
val byteArrayOutputStream = ByteArrayOutputStream()
ImageIO.write(backgroundImage, "png", byteArrayOutputStream)
val imageBytes = byteArrayOutputStream.toByteArray()
return imageBytes.encodeToBase64()
return canvas.toByteArray().encodeToBase64()
}
suspend fun parse(listener: OBMessage, bvid: String, message: GroupMessage) {
src/main/kotlin/cn/rtast/fancybot/commands/parse/GitHubParseCommand.kt
+2 -5
View File
@@ -14,6 +14,7 @@ import cn.rtast.fancybot.util.Resources
import cn.rtast.fancybot.util.drawCustomImage
import cn.rtast.fancybot.util.str.encodeToBase64
import cn.rtast.fancybot.util.str.setTruncat
import cn.rtast.fancybot.util.toByteArray
import cn.rtast.rob.entity.GroupMessage
import cn.rtast.rob.util.ob.MessageChain
import cn.rtast.rob.util.ob.OBMessage
@@ -21,7 +22,6 @@ import java.awt.Color
import java.awt.Font
import java.awt.RenderingHints
import java.awt.image.BufferedImage
import java.io.ByteArrayOutputStream
import java.net.URI
import javax.imageio.ImageIO
@@ -117,10 +117,7 @@ object GitHubParseCommand {
g2d.drawString(truncatedDescription, 80, 450)
val avatarImage = ImageIO.read(URI(repoStat.owner.avatarUrl).toURL())
g2d.drawCustomImage(avatarImage, 1200, 160, 300.0, 300.0, true)
val byteArrayOutputStream = ByteArrayOutputStream()
ImageIO.write(canvas, "png", byteArrayOutputStream)
val imageBytes = byteArrayOutputStream.toByteArray()
return imageBytes.encodeToBase64()
return canvas.toByteArray().encodeToBase64()
}
suspend fun parse(listener: OBMessage, message: GroupMessage) {