feat :

配置项: pre 为空时则不发送前置消息传话目标为 -1 时则转发全部好友传话目标为 -2 时则转发全部群聊传话目标为 -3 时则转发全部好友/群聊
gdpl2112 · Jun 4, 2023 · 9d3c478 · 9d3c478
1 parent 7e741bc
commit 9d3c478
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 17 deletions.
diff --git a/README.md b/README.md
@@ -32,3 +32,11 @@
 赋予该用户123456权限
 
     /perm permit u123456 io.github.gdpl2112.transmitMsg.TransmitMessage:op
+
+<hr>
+
+配置项: pre 为 空时则不发送前置消息
+
+传话目标 为 -1 时 则转发全部好友
+传话目标 为 -2 时 则转发全部群聊
+传话目标 为 -3 时 则转发全部好友/群聊
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -7,7 +7,7 @@ plugins {
 }
 
 group = "io.github.kloping"
-version = "1.0"
+version = "1.1"
 
 repositories {
     maven("https://maven.aliyun.com/repository/public")

diff --git a/src/main/kotlin/M1ListenerHost.kt b/src/main/kotlin/M1ListenerHost.kt
@@ -3,6 +3,7 @@ package io.github.gdpl2112.transmitMsg
 import net.mamoe.mirai.Bot
 import net.mamoe.mirai.console.permission.AbstractPermitteeId
 import net.mamoe.mirai.console.permission.PermissionService.Companion.hasPermission
+import net.mamoe.mirai.contact.Contact
 import net.mamoe.mirai.event.EventHandler
 import net.mamoe.mirai.event.SimpleListenerHost
 import net.mamoe.mirai.event.events.FriendMessageEvent
@@ -51,22 +52,6 @@ object M1ListenerHost : SimpleListenerHost() {
         }
     }
 
-    private suspend fun send(st: SyncTransmit, bot: Bot) {
-        var f = bot.getFriend(st.target)
-        if (f != null) {
-            f.sendMessage(TransmitMessage.config.pre.replace("$1", st.qid.toString()))
-            st.data?.let { f.sendMessage(it) }
-        } else {
-            for (group in bot.groups) {
-                var m = group.get(st.target);
-                if (m != null) {
-                    m.sendMessage(TransmitMessage.config.pre.replace("$1", st.qid.toString()))
-                    st.data?.let { m.sendMessage(it) }
-                }
-            }
-        }
-    }
-
     @EventHandler
     suspend fun onFriendMessageEvent(event: FriendMessageEvent) {
         if (!AbstractPermitteeId.ExactFriend(event.sender.id).hasPermission(TransmitMessage.p0)) {
@@ -109,4 +94,54 @@ object M1ListenerHost : SimpleListenerHost() {
         }
         return text.trim()
     }
+
+    private suspend fun send(st: SyncTransmit, bot: Bot) {
+        when (st.target) {
+            -1L -> {
+                sendAllFriends(bot, st)
+            }
+
+            -2L -> {
+                sendAllGroups(bot, st)
+            }
+
+            -3L -> {
+                sendAllFriends(bot, st)
+                sendAllGroups(bot, st)
+            }
+        }
+        var f = bot.getFriend(st.target)
+        if (f != null) {
+            sendNow(f, st)
+        } else {
+            for (group in bot.groups) {
+                var m = group.get(st.target);
+                if (m != null) {
+                    sendNow(m, st)
+                }
+            }
+        }
+    }
+
+    private suspend fun sendAllGroups(bot: Bot, st: SyncTransmit) {
+        for (group in bot.groups) {
+            sendNow(group, st)
+        }
+    }
+
+    private suspend fun sendAllFriends(bot: Bot, st: SyncTransmit) {
+        for (f in bot.friends) {
+            sendNow(f, st)
+        }
+    }
+
+    private suspend fun sendNow(contact: Contact, st: SyncTransmit) {
+        if (!TransmitMessage.config.pre.trim().isEmpty()) contact.sendMessage(
+            TransmitMessage.config.pre.replace(
+                "$1",
+                st.qid.toString()
+            )
+        )
+        st.data?.let { contact.sendMessage(it) }
+    }
 }