refactor: 请求处理控制流基础架构

pkg/pipeline/resprule/rules/atbot.py

@@ -0,0 +1,28 @@
+from __future__ import annotations
+
+import mirai
+
+from .. import rule as rule_model
+from .. import entities
+
+
+class AtBotRule(rule_model.GroupRespondRule):
+
+    async def match(
+        self,
+        message_text: str,
+        message_chain: mirai.MessageChain,
+        rule_dict: dict
+    ) -> entities.RuleJudgeResult:
+        
+        if message_chain.has(mirai.At(self.ap.im_mgr.bot_account_id)) and rule_dict['at']:
+            message_chain.remove(mirai.At(self.ap.im_mgr.bot_account_id))
+            return entities.RuleJudgeResult(
+                matching=True,
+                replacement=message_chain,
+            )
+
+        return entities.RuleJudgeResult(
+            matching=False,
+            replacement = message_chain
+        )

pkg/pipeline/resprule/rules/prefix.py

@@ -0,0 +1,29 @@
+import mirai
+
+from .. import rule as rule_model
+from .. import entities
+
+
+class PrefixRule(rule_model.GroupRespondRule):
+
+    async def match(
+        self,
+        message_text: str,
+        message_chain: mirai.MessageChain,
+        rule_dict: dict
+    ) -> entities.RuleJudgeResult:
+        prefixes = rule_dict['prefix']
+
+        for prefix in prefixes:
+            if message_text.startswith(prefix):
+                return entities.RuleJudgeResult(
+                    matching=True,
+                    replacement=mirai.MessageChain([
+                        mirai.Plain(message_text[len(prefix):])
+                    ]),
+                )
+
+        return entities.RuleJudgeResult(
+            matching=False,
+            replacement=message_chain
+        )

pkg/pipeline/resprule/rules/random.py

@@ -0,0 +1,22 @@
+import random
+
+import mirai
+
+from .. import rule as rule_model
+from .. import entities
+
+
+class RandomRespRule(rule_model.GroupRespondRule):
+
+    async def match(
+        self,
+        message_text: str,
+        message_chain: mirai.MessageChain,
+        rule_dict: dict
+    ) -> entities.RuleJudgeResult:
+        random_rate = rule_dict['random_rate']
+        
+        return entities.RuleJudgeResult(
+            matching=random.random() < random_rate,
+            replacement=message_chain
+        )

pkg/pipeline/resprule/rules/regexp.py

@@ -0,0 +1,31 @@
+import re
+
+import mirai
+
+from .. import rule as rule_model
+from .. import entities
+
+
+class RegExpRule(rule_model.GroupRespondRule):
+
+    async def match(
+        self,
+        message_text: str,
+        message_chain: mirai.MessageChain,
+        rule_dict: dict
+    ) -> entities.RuleJudgeResult:
+        regexps = rule_dict['regexp']
+
+        for regexp in regexps:
+            match = re.match(regexp, message_text)
+
+            if match:
+                return entities.RuleJudgeResult(
+                    matching=True,
+                    replacement=message_chain,
+                )
+        
+        return entities.RuleJudgeResult(
+            matching=False,
+            replacement=message_chain
+        )