fix(self-protect): checker

BakaBotTeam · Sep 30, 2024 · a3cd168 · a3cd168
1 parent df7342c
commit a3cd168
Show file tree

Hide file tree

Showing 4 changed files with 896 additions and 2 deletions.
diff --git a/src/main/kotlin/ltd/guimc/lgzbot/PluginMain.kt b/src/main/kotlin/ltd/guimc/lgzbot/PluginMain.kt
@@ -73,6 +73,7 @@ object PluginMain : KotlinPlugin(
     lateinit var adRegex: Array<Regex>
     lateinit var adPinyinRegex: Array<Regex>
     lateinit var seriousRegex: Array<Regex>
+    lateinit var spRegex: Array<Regex>
     lateinit var fbValue: Array<String>
     lateinit var webHookService: WebHookService
     lateinit var configReloadThread: Thread
@@ -91,6 +92,7 @@ object PluginMain : KotlinPlugin(
         adRegex = getDefaultRegex()
         adPinyinRegex = getDefaultPinyinRegex()
         seriousRegex = getSeriousRegex()
+        spRegex = getSeriousRegex()
         fbValue = getFbValue()
         webHookService = WebHookService(GithubWebHookReciver())
 

diff --git a/src/main/kotlin/ltd/guimc/lgzbot/listener/message/SelfMessageListener.kt b/src/main/kotlin/ltd/guimc/lgzbot/listener/message/SelfMessageListener.kt
@@ -16,9 +16,9 @@ object SelfMessageListener : ListenerHost {
     suspend fun MessagePreSendEvent.onEvent() {
         val newMessageChain = MessageChainBuilder()
         this.message.toMessageChain().forEach { it ->
-            if (it is PlainText && RegexUtils.matchRegex(PluginMain.seriousRegex, it.content)) {
+            if (it is PlainText && RegexUtils.matchRegex(PluginMain.spRegex, it.content)) {
                 this.cancel()
-                newMessageChain.add(it.content.replaceRegex(PluginMain.seriousRegex))
+                newMessageChain.add(it.content.replaceRegex(PluginMain.spRegex))
             } else {
                 newMessageChain.add(it)
             }

diff --git a/src/main/kotlin/ltd/guimc/lgzbot/utils/RegexUtils.kt b/src/main/kotlin/ltd/guimc/lgzbot/utils/RegexUtils.kt
@@ -58,6 +58,20 @@ object RegexUtils {
         return regexList.toTypedArray()
     }
 
+    fun getSpRegex(): Array<Regex> {
+        // Read from resources regex.txt
+        val regexFile = RegexUtils::class.java.getResourceAsStream("/self-protect.txt")
+        val regexList = mutableListOf<Regex>()
+        regexFile?.bufferedReader()?.use { reader ->
+            reader.lines().forEach {
+                // regexList.add(Regex(PinyinUtils.convertToPinyin(it)))
+                regexList.add(Regex(it))
+            }
+        }
+
+        return regexList.toTypedArray()
+    }
+
     // 匹配正则表达式列表 返回是否匹配
     fun matchRegex(regexList: Array<Regex>, message: String): Boolean {
         try {