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

pkg/pipeline/resprule/resprule.py

@@ -0,0 +1,62 @@
+from __future__ import annotations
+
+import mirai
+
+from ...core import app
+from . import entities as rule_entities, rule
+from .rules import atbot, prefix, regexp, random
+
+from .. import stage, entities, stagemgr
+from ...core import entities as core_entities
+from ...config import manager as cfg_mgr
+
+
+@stage.stage_class("GroupRespondRuleCheckStage")
+class GroupRespondRuleCheckStage(stage.PipelineStage):
+    """群组响应规则检查器
+    """
+
+    rule_matchers: list[rule.GroupRespondRule]
+
+    async def initialize(self):
+        """初始化检查器
+        """
+        self.rule_matchers = [
+            atbot.AtBotRule(self.ap),
+            prefix.PrefixRule(self.ap),
+            regexp.RegExpRule(self.ap),
+            random.RandomRespRule(self.ap),
+        ]
+
+        for rule_matcher in self.rule_matchers:
+            await rule_matcher.initialize()
+
+    async def process(self, query: core_entities.Query, stage_inst_name: str) -> entities.StageProcessResult:
+        
+        if query.launcher_type != 'group':
+            return entities.StageProcessResult(
+                result_type=entities.ResultType.CONTINUE,
+                new_query=query
+            )
+
+        rules = self.ap.cfg_mgr.data['response_rules']
+
+        use_rule = rules['default']
+
+        if str(query.launcher_id) in use_rule:
+            use_rule = use_rule[str(query.launcher_id)]
+
+        for rule_matcher in self.rule_matchers:  # 任意一个匹配就放行
+            res = await rule_matcher.match(str(query.message_chain), query.message_chain, use_rule)
+            if res.matching:
+                query.message_chain = res.replacement
+
+                return entities.StageProcessResult(
+                    result_type=entities.ResultType.CONTINUE,
+                    new_query=query,
+                )
+        
+        return entities.StageProcessResult(
+            result_type=entities.ResultType.INTERRUPT,
+            new_query=query
+        )