feat: add dm search

This commit is contained in:
2024-09-24 19:11:04 +08:00
parent 7688f538fb
commit 98f4f5d10f
3 files changed
+90

No files matched your search

@@ -200,6 +200,7 @@ val commands = listOf(
NiuziBankCommand(), WithdrawCommand(), NiuziBankCommand(), WithdrawCommand(),
BankTransferCommand(), CreateBankAccountCommand(), BankTransferCommand(), CreateBankAccountCommand(),
BankBalanceCommand(), DepositCommand(), BankBalanceCommand(), DepositCommand(),
DMSearchCommand()
) )
val START_UP_TIME = Instant.now().epochSecond val START_UP_TIME = Instant.now().epochSecond
@@ -0,0 +1,63 @@
/*
* Copyright © 2024 RTAkland
* Author: RTAkland
* Date: 2024/9/24
*/
package cn.rtast.fancybot.commands.lookup
import cn.rtast.fancybot.entity.bgm.BGMSearch
import cn.rtast.fancybot.util.Http
import cn.rtast.fancybot.util.str.encodeToBase64
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.NodeMessageChain
import cn.rtast.rob.util.ob.OneBotListener
import okhttp3.OkHttpClient
import okhttp3.Request
class DMSearchCommand : BaseCommand() {
override val commandNames = listOf("/dm", "动漫", "dm")
private val client = OkHttpClient()
companion object {
private const val API_URL = "https://api.bgm.tv/search/subject/#{}?type=2&responseGroup=small"
}
private fun followRedirect(url: String): ByteArray {
val request = Request.Builder().url(url).build()
client.newCall(request).execute().use { response ->
return response.body.bytes()
}
}
override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
if (args.isEmpty()) {
message.reply("发送`/dm <关键词>`即可搜索番剧啦~")
return
}
try {
val keyword = args.joinToString(" ")
val response = Http.get<BGMSearch>(API_URL.replace("#{}", keyword)).list
val nodeMsg = NodeMessageChain.Builder()
response.asSequence().take(5).forEach {
val imageBase64 = this.followRedirect(it.images.large).encodeToBase64()
val tempMsg = MessageChain.Builder()
.addImage(imageBase64, true)
.addText("${it.name} | ${it.nameCN}")
.addNewLine()
.addText(it.url)
.build()
nodeMsg.addMessageChain(tempMsg, message.sender.userId)
}
val footerNode = MessageChain.Builder().addText("数据来源: Bangumi").build()
nodeMsg.addMessageChain(footerNode, message.sender.userId)
listener.sendGroupForwardMsg(message.groupId, nodeMsg.build())
} catch (_: Exception) {
message.reply("没有搜索到结果呢~")
}
}
}
@@ -0,0 +1,26 @@
/*
* Copyright © 2024 RTAkland
* Author: RTAkland
* Date: 2024/9/24
*/
package cn.rtast.fancybot.entity.bgm
import com.google.gson.annotations.SerializedName
data class BGMSearch(
val list: List<Result>
) {
data class Result(
val url: String,
val images: Images,
val name: String,
@SerializedName("name_cn")
val nameCN: String,
)
data class Images(
val large: String
)
}