✨ after_rules

Author: RF-Tar-Railt

src/nonebot_plugin_alconna/matcher.py

@@ -943,6 +943,7 @@ def ensure_context(self, bot: Bot, event: Event):
 def on_alconna(
     command: Alconna | str,
     rule: None | Rule | T_RuleChecker = None,
+    after_rule: None | Rule | T_RuleChecker = None,
     skip_for_unmatch: bool = True,
     auto_send_output: bool | None = None,
     aliases: set[str] | tuple[str, ...] | None = None,
@@ -968,6 +969,7 @@ def on_alconna(
     参数:
         command: Alconna 命令
         rule: 事件响应规则
+        after_rule: 作用于 Alconna 解析后的事件响应规则
         skip_for_unmatch: 是否在解析失败时跳过
         auto_send_output: 是否自动发送输出信息并跳过
         aliases: 命令别名
@@ -1018,6 +1020,7 @@ def on_alconna(
         response_self,
         aliases,
         rule,
+        after_rule,
     )
     executor = _rule.executor
     params = (

src/nonebot_plugin_alconna/rule.py

@@ -65,6 +65,7 @@ class AlconnaRule:
         "_path",
         "_tasks",
         "_waiter",
+        "after_rules",
         "auto_send",
         "before_rules",
         "command",
@@ -89,6 +90,7 @@ def __init__(
         response_self: Optional[bool] = None,
         _aliases: Optional[Union[set[str], tuple[str, ...]]] = None,
         before_rule: Optional[Union[Rule, T_RuleChecker]] = None,
+        after_rule: Optional[Union[Rule, T_RuleChecker]] = None,
     ):
         if isinstance(comp_config, bool):
             self.comp_config = {} if comp_config else None
@@ -156,6 +158,7 @@ def _update(cmd_id: int):
         self._namespace = command.namespace
         self._tasks: dict[str, asyncio.Task] = {}
         self.before_rules = Rule() & before_rule
+        self.after_rules = Rule() & after_rule
 
         self._comp_help = ""
         if self.comp_config is not None:
@@ -357,6 +360,8 @@ async def __call__(
         state[ALCONNA_RESULT] = CommandResult(result=arp, output=may_help_text)
         state[ALCONNA_EXEC_RESULT] = cmd.exec_result
         state[ALCONNA_EXTENSION] = selected
+        if not await self.after_rules(bot, event, state, stack, dependency_cache):
+            return False
         return True
 
     async def send(self, text: str, bot: Bot, event: Event, arp: Arparma) -> Any:

src/nonebot_plugin_alconna/shortcut.py

@@ -43,6 +43,7 @@ def funcommand(
     use_cmd_sep: bool | None = None,
     response_self: bool | None = None,
     rule: Rule | T_RuleChecker | None = None,
+    after_rule: Rule | T_RuleChecker | None = None,
     permission: Permission | T_PermissionChecker | None = None,
     *,
     handlers: list[T_Handler | Dependent] | None = None,
@@ -65,6 +66,7 @@ def wrapper(func: Callable[..., MReturn]) -> type[AlconnaMatcher]:
         matcher = on_alconna(
             FuncMounter(func, _config),
             rule,
+            after_rule,
             skip_for_unmatch,
             auto_send_output,
             extensions=extensions,
@@ -105,6 +107,7 @@ def args_gen(pattern: str, types: dict):
     def build(
         self,
         rule: Rule | T_RuleChecker | None = None,
+        after_rule: Rule | T_RuleChecker | None = None,
         skip_for_unmatch: bool = True,
         auto_send_output: bool | None = None,
         aliases: set[str] | tuple[str, ...] | None = None,

src/nonebot_plugin_alconna/uniseg/message.py

@@ -6,7 +6,7 @@
 from json import dumps, loads
 from pathlib import Path
 from types import FunctionType
-from typing import TYPE_CHECKING, Any, Callable, Literal, NoReturn, TypeVar, Union
+from typing import TYPE_CHECKING, Any, Callable, Literal, NoReturn, TypeVar, Union, Protocol
 from typing_extensions import Self, SupportsIndex, deprecated
 
 from nonebot.exception import FinishedException
@@ -46,6 +46,7 @@
 from .template import UniMessageTemplate
 
 TS = TypeVar("TS", bound=Segment)
+_TM = TypeVar("_TM", bound="str | Message | UniMessage")
 
 
 class _method:
@@ -58,7 +59,11 @@ def __get__(self, instance, owner):
         return self.__func__.__get__(instance, owner)
 
 
-current_send_wrapper = ContextModel("nonebot_plugin_alconna.uniseg.send_wrapper")
+class SendWrapper(Protocol):
+    async def __call__(self, bot: Bot, event: Event, send: _TM) -> _TM: ...
+
+
+current_send_wrapper: ContextModel[SendWrapper] = ContextModel("nonebot_plugin_alconna.uniseg.send_wrapper")
 MessageContainer = Union[str, Segment, Sequence["MessageContainer"], "UniMessage"]