feat: add niuzi rank, and niuzi bank rank

gradle/libs.versions.toml

@@ -1,6 +1,6 @@
 [versions]
 kotlinVersion = "2.0.10"
-ronebotVersion = "1.11.0"
+ronebotVersion = "1.11.1"
 shadowVersion = "8.3.0"
 okhttpVersion = "5.0.0-alpha.14"
 exposedVersion = "0.53.0"

src/main/kotlin/cn/rtast/fancybot/Const.kt

@@ -103,5 +103,6 @@ val commands = listOf(
     BankTransferCommand(), CreateBankAccountCommand(),
     BankBalanceCommand(), DepositCommand(),
     DMSearchCommand(), LlamaCommand(),
-    QQMusicCommand(), BullShitGenerateCommand()
+    QQMusicCommand(), BullShitGenerateCommand(),
+    NiuziRankCommand(), NiuziBankRankCommand()
 )

src/main/kotlin/cn/rtast/fancybot/commands/niuzi/NiuziRankCommand.kt

@@ -0,0 +1,50 @@
+/*
+ * Copyright © 2024 RTAkland
+ * Author: RTAkland
+ * Date: 2024/9/26
+ */
+
+
+package cn.rtast.fancybot.commands.niuzi
+
+import cn.rtast.fancybot.niuziBankManager
+import cn.rtast.fancybot.niuziManager
+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.OneBotListener
+
+
+private suspend fun getUserInfo(listener: OneBotListener, groupId: Long, userId: Long): String {
+    val info = listener.getGroupMemberInfo(groupId, userId)
+    return info.card ?: info.nickname
+}
+
+class NiuziRankCommand : BaseCommand() {
+    override val commandNames = listOf("牛子排行榜")
+
+    override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
+        val allNiuzi = niuziManager.getAllNiuzi().sortedBy { it.length }.reversed()
+        val msg = MessageChain.Builder().addText("牛子长度排行榜如下").addNewLine()
+        allNiuzi.forEach {
+            val userName = getUserInfo(listener, message.groupId, it.userId)
+            msg.addText("$userName | ${it.length}cm").addNewLine()
+        }
+        message.reply(msg.build())
+    }
+}
+
+class NiuziBankRankCommand : BaseCommand() {
+    override val commandNames = listOf("牛子富豪榜")
+
+    override suspend fun executeGroup(listener: OneBotListener, message: GroupMessage, args: List<String>) {
+        val allNiuziBankAccounts = niuziBankManager.getAllAccount()
+            .sortedBy { it.balance }.reversed()
+        val msg = MessageChain.Builder().addText("牛子银行富豪榜排行如下").addNewLine()
+        allNiuziBankAccounts.forEach {
+            val userName = getUserInfo(listener, message.groupId, it.userId)
+            msg.addText("$userName | ${it.balance}cm").addNewLine()
+        }
+        message.reply(msg.build())
+    }
+}

src/main/kotlin/cn/rtast/fancybot/util/file/NiuziBankManager.kt

@@ -81,4 +81,10 @@ class NiuziBankManager {
         }
         return this.getUser(from)
     }
+
+    suspend fun getAllAccount(): List<NiuziBankAccount> =
+        suspendedTransaction {
+            NiuziBankTable.selectAll()
+                .map { NiuziBankAccount(it[userId], it[balance], it[interestRate], it[timestamp]) }
+        }
 }

src/main/kotlin/cn/rtast/fancybot/util/file/NiuziManager.kt

@@ -42,16 +42,10 @@ class NiuziManager {
         }
     }
 
-    suspend fun getUser(id: Long): Niuzi? {
+    suspend fun getUser(id: Long): Niuzi? =
-        return suspendedTransaction {
+        suspendedTransaction {
-            NiuziTable.selectAll().where { userId eq id }.map {
+            NiuziTable.selectAll().where { userId eq id }.map { Niuzi(it[userId], it[length], it[timestamp]) }
-                Niuzi(
+                .singleOrNull()
-                    it[userId],
-                    it[length],
-                    it[timestamp]
-                )
-            }.singleOrNull()
-        }
         }
 
     suspend fun isSigned(id: Long): Boolean {
@@ -106,4 +100,7 @@ class NiuziManager {
         }
         return getUser(id)
     }
+
+    suspend fun getAllNiuzi(): List<Niuzi> =
+        suspendedTransaction { NiuziTable.selectAll().map { Niuzi(it[userId], it[length], it[timestamp]) } }
 }