91 lines
3.9 KiB
Kotlin
91 lines
3.9 KiB
Kotlin
/*
|
|
* Copyright © 2024 RTAkland
|
|
* Author: RTAkland
|
|
* Date: 2024/9/4
|
|
*/
|
|
|
|
|
|
package cn.rtast.fancybot.commands.lookup
|
|
|
|
import cn.rtast.fancybot.configManager
|
|
import cn.rtast.fancybot.entity.weather.Geo
|
|
import cn.rtast.fancybot.entity.weather.Weather
|
|
import cn.rtast.fancybot.util.Http
|
|
import cn.rtast.fancybot.util.Resources
|
|
import cn.rtast.fancybot.util.drawCenteredText
|
|
import cn.rtast.fancybot.util.drawCustomImage
|
|
import cn.rtast.fancybot.util.str.encodeToBase64
|
|
import cn.rtast.fancybot.util.toBufferedImage
|
|
import cn.rtast.fancybot.util.toByteArray
|
|
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.OBMessage
|
|
import java.awt.Color
|
|
import java.awt.Font
|
|
import java.awt.image.BufferedImage
|
|
|
|
class WeatherCommand : BaseCommand() {
|
|
override val commandNames = listOf("/weather", "/天气", "天气")
|
|
|
|
private val canvasWidth = 800
|
|
private val canvasHeight = 600
|
|
private val font = Font("Serif", Font.PLAIN, 60).deriveFont(Font.BOLD or Font.ITALIC)
|
|
private val locationFont = Font("Serif", Font.PLAIN, 50).deriveFont(Font.BOLD or Font.ITALIC)
|
|
private val backgroundColor = Color(209, 238, 238)
|
|
private val textColor = Color.BLACK
|
|
private val licenseColor = Color(155, 48, 255)
|
|
|
|
private fun createWeatherCard(weather: Weather, geoResult: Geo.Location): String {
|
|
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/999.png")!!
|
|
g2d.color = backgroundColor
|
|
g2d.fillRect(0, 0, canvasWidth, canvasHeight)
|
|
g2d.drawCustomImage(weatherIcon.toBufferedImage(), canvasWidth / 2 - 230, canvasHeight / 2 - 80, 230.0, 230.0)
|
|
g2d.color = textColor
|
|
g2d.font = font
|
|
g2d.drawString("${weather.now.temp}℃", canvasWidth / 2, canvasHeight / 2 + 20)
|
|
g2d.drawString(weather.now.text, canvasWidth / 2, canvasHeight / 2 + 100)
|
|
g2d.font = locationFont
|
|
g2d.drawCenteredText("${geoResult.adm1}-${geoResult.adm2}-${geoResult.name}", canvasWidth / 2, 90)
|
|
g2d.color = licenseColor
|
|
g2d.drawCenteredText("数据来源: ${weather.refer.sources.joinToString(",")}", canvasWidth / 2, canvasHeight - 80)
|
|
g2d.dispose()
|
|
return emptyImage.toByteArray().encodeToBase64()
|
|
}
|
|
|
|
override suspend fun executeGroup(listener: OBMessage, message: GroupMessage, args: List<String>) {
|
|
if (args.isEmpty()) {
|
|
val msg = MessageChain.Builder()
|
|
.addAt(message.sender.userId)
|
|
.addText("发送/weather <地区名>即可查看实时天气哦~")
|
|
.build()
|
|
listener.sendGroupMessage(message.groupId, msg)
|
|
return
|
|
}
|
|
try {
|
|
val locationName = args.first()
|
|
val lookupLocation = Http.get<Geo>(
|
|
"https://geoapi.qweather.com/v2/city/lookup",
|
|
mapOf("location" to locationName, "key" to configManager.qweatherKey)
|
|
).location.first()
|
|
val weather = Http.get<Weather>(
|
|
"https://devapi.qweather.com/v7/weather/now",
|
|
mapOf("location" to lookupLocation.id, "key" to configManager.qweatherKey)
|
|
)
|
|
val msg = MessageChain.Builder()
|
|
.addReply(message.messageId)
|
|
.addImage(this.createWeatherCard(weather, lookupLocation), true)
|
|
.build()
|
|
listener.sendGroupMessage(message.groupId, msg)
|
|
} catch (_: NullPointerException) {
|
|
val msg = MessageChain.Builder()
|
|
.addAt(message.sender.userId)
|
|
.addText("没有查询到该城市的天气信息哦~")
|
|
.build()
|
|
listener.sendGroupMessage(message.groupId, msg)
|
|
}
|
|
}
|
|
} |