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

pkg/pipeline/bansess/bansess.py

@@ -0,0 +1,76 @@
+from __future__ import annotations
+import re
+
+from .. import stage, entities, stagemgr
+from ...core import entities as core_entities
+from ...config import manager as cfg_mgr
+
+
+@stage.stage_class('BanSessionCheckStage')
+class BanSessionCheckStage(stage.PipelineStage):
+
+    banlist_mgr: cfg_mgr.ConfigManager
+
+    async def initialize(self):
+        self.banlist_mgr = await cfg_mgr.load_python_module_config(
+            "banlist.py",
+            "res/templates/banlist-template.py"
+        )
+
+    async def process(
+        self,
+        query: core_entities.Query,
+        stage_inst_name: str
+    ) -> entities.StageProcessResult:
+        
+        if not self.banlist_mgr.data['enable']:
+            return entities.StageProcessResult(
+                result_type=entities.ResultType.CONTINUE,
+                new_query=query
+            )
+        
+        result = False
+
+        if query.launcher_type == 'group':
+            if not self.banlist_mgr.data['enable_group']:  # 未启用群聊响应
+                result = True
+            # 检查是否显式声明发起人QQ要被person忽略
+            elif query.sender_id in self.banlist_mgr.data['person']:
+                result = True
+            else:
+                for group_rule in self.banlist_mgr.data['group']:
+                    if type(group_rule) == int:
+                        if group_rule == query.launcher_id:
+                            result = True
+                    elif type(group_rule) == str:
+                        if group_rule.startswith('!'):
+                            reg_str = group_rule[1:]
+                            if re.match(reg_str, str(query.launcher_id)):
+                                result = False
+                                break
+                        else:
+                            if re.match(group_rule, str(query.launcher_id)):
+                                result = True
+        elif query.launcher_type == 'person':
+            if not self.banlist_mgr.data['enable_private']:
+                result = True
+            else:
+                for person_rule in self.banlist_mgr.data['person']:
+                    if type(person_rule) == int:
+                        if person_rule == query.launcher_id:
+                            result = True
+                    elif type(person_rule) == str:
+                        if person_rule.startswith('!'):
+                            reg_str = person_rule[1:]
+                            if re.match(reg_str, str(query.launcher_id)):
+                                result = False
+                                break
+                        else:
+                            if re.match(person_rule, str(query.launcher_id)):
+                                result = True
+
+        return entities.StageProcessResult(
+            result_type=entities.ResultType.CONTINUE if not result else entities.ResultType.INTERRUPT,
+            new_query=query,
+            debug_notice=f'根据禁用列表忽略消息: {query.launcher_type}_{query.launcher_id}' if result else ''
+        )