Shamrock: support /set_guild_member_role

qqinterface/src/main/java/com/tencent/qqnt/kernel/nativeinterface/IKernelGuildService.java

@@ -68,5 +68,6 @@ public interface IKernelGuildService {
 
     void deleteRole(long guild, long role, IGProResultCallback cb);
 
+    void setMemberRoles(long guild, long u1, long u2, long tinyId, ArrayList<Long> addRoles, ArrayList<Long> removeRoles, IGProResultCallback cb);
 
 }

xposed/src/main/java/moe/fuqiuluo/qqinterface/servlet/GProSvc.kt

@@ -305,4 +305,16 @@ internal object GProSvc: BaseSvc() {
             }
         }
     }
+
+    fun setMemberRole(guildId: ULong, tinyId: ULong, roleId: ULong, isSet: Boolean) {
+        val kernelGProService = NTServiceFetcher.kernelService.wrapperSession.guildService
+        val addList = arrayListOf<Long>()
+        val rmList = arrayListOf<Long>()
+        (if (isSet) addList else rmList).add(roleId.toLong())
+        kernelGProService.setMemberRoles(guildId.toLong(), 0, 0, tinyId.toLong(), addList, rmList) { code, msg, result ->
+            if (code != 0) {
+                LogCenter.log("setMemberRole failed: $code($msg) => $result", Level.WARN)
+            }
+        }
+    }
 }

xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/SetGuildMemberRole.kt

@@ -0,0 +1,42 @@
+package moe.fuqiuluo.shamrock.remote.action.handlers
+
+import kotlinx.serialization.json.JsonElement
+import moe.fuqiuluo.qqinterface.servlet.GProSvc
+import moe.fuqiuluo.shamrock.remote.action.ActionSession
+import moe.fuqiuluo.shamrock.remote.action.IActionHandler
+import moe.fuqiuluo.shamrock.tools.EmptyJsonString
+import moe.fuqiuluo.shamrock.tools.asString
+import moe.fuqiuluo.symbols.OneBotHandler
+
+@OneBotHandler("set_guild_member_role")
+internal object SetGuildMemberRole: IActionHandler() {
+    override suspend fun internalHandle(session: ActionSession): String {
+        val guildId = session.getString("guild_id").toULong()
+        val role = session.getString("role_id").toULong()
+        val set = session.getBooleanOrDefault("set", false)
+        return if (session.has("user_id")) {
+            val userId = session.getString("user_id").toULong()
+            invoke(guildId, userId, role, set, echo = session.echo)
+        } else if (session.isArray("users")) {
+            invoke(guildId, session.getArray("users").map {
+                it.asString.toULong()
+            }, role, set, echo = session.echo)
+        } else {
+            logic("missing user_id or users", echo = session.echo)
+        }
+    }
+
+    operator fun invoke(guildId: ULong, users: List<ULong>, roleId: ULong, set: Boolean, echo: JsonElement = EmptyJsonString): String {
+        users.forEach {
+            GProSvc.setMemberRole(guildId, it, roleId, set)
+        }
+        return ok("success", echo = echo)
+    }
+
+    operator fun invoke(guildId: ULong, user: ULong, roleId: ULong, set: Boolean, echo: JsonElement = EmptyJsonString): String {
+        GProSvc.setMemberRole(guildId, user, roleId, set)
+        return ok("success", echo = echo)
+    }
+
+    override val requiredParams: Array<String> = arrayOf("guild_id", "role_id")
+}

xposed/src/main/java/moe/fuqiuluo/shamrock/remote/api/GuildAction.kt

@@ -19,12 +19,14 @@ import moe.fuqiuluo.shamrock.remote.action.handlers.GetGuildRoles
 import moe.fuqiuluo.shamrock.remote.action.handlers.GetGuildServiceProfile
 import moe.fuqiuluo.shamrock.remote.action.handlers.SendGuildMessage
 import moe.fuqiuluo.shamrock.remote.action.handlers.SendMessage
+import moe.fuqiuluo.shamrock.remote.action.handlers.SetGuildMemberRole
 import moe.fuqiuluo.shamrock.tools.fetchGetOrNull
 import moe.fuqiuluo.shamrock.tools.fetchGetOrThrow
 import moe.fuqiuluo.shamrock.tools.fetchOrNull
 import moe.fuqiuluo.shamrock.tools.fetchOrThrow
 import moe.fuqiuluo.shamrock.tools.fetchPostJsonArray
 import moe.fuqiuluo.shamrock.tools.fetchPostJsonObject
+import moe.fuqiuluo.shamrock.tools.fetchPostJsonString
 import moe.fuqiuluo.shamrock.tools.fetchPostOrNull
 import moe.fuqiuluo.shamrock.tools.fetchPostOrThrow
 import moe.fuqiuluo.shamrock.tools.getOrPost
@@ -130,4 +132,22 @@ fun Routing.guildAction() {
         val roleId = fetchOrThrow("role_id").toULong()
         call.respondText(DeleteGuildRole(guildId, roleId), ContentType.Application.Json)
     }
+
+    getOrPost("/set_guild_member_role") {
+        val guildId = fetchOrThrow("guild_id").toULong()
+        val roleId = fetchOrThrow("role_id").toULong()
+        val set = fetchOrNull("set")?.toBoolean() ?: false
+        val userId = fetchOrNull("user_id")?.toULong()
+        val users = fetchOrNull("users")?.split(",")?.map { it.toULong() }
+        call.respondText(
+            if (userId != null) {
+                SetGuildMemberRole(guildId, userId, roleId, set)
+            } else if (users != null) {
+                SetGuildMemberRole(guildId, users, roleId, set)
+            } else {
+                throw IllegalArgumentException("missing user_id or users")
+            },
+            ContentType.Application.Json
+        )
+    }
 }