✨ config cache_message

Author: RF-Tar-Railt

src/nonebot_plugin_alconna/builtins/extensions/discord.py

@@ -35,6 +35,7 @@
     ApplicationCommandInteractionDataOption,
 )
 
+from nonebot_plugin_alconna.extension import cache_msg
 from nonebot_plugin_alconna.matcher import _M, AlconnaMatcher
 from nonebot_plugin_alconna import At, Image, Extension, UniMessage, log
 
@@ -342,7 +343,7 @@ def validate(self, bot, event: Event) -> bool:
     async def message_provider(self, event: Event, state: T_State, bot, use_origin: bool = False):
         if not isinstance(event, ApplicationCommandInteractionEvent):
             return None
-        if event.id in self.cache:
+        if cache_msg and event.id in self.cache:
             return self.cache[event.id]
         data = event.data
         cmd = f"/{data.name}"

src/nonebot_plugin_alconna/builtins/extensions/onebot11.py

@@ -6,6 +6,7 @@
 from nonebot.internal.adapter import Bot, Event
 from nonebot.adapters.onebot.v11 import Event as OneBot11Event
 
+from nonebot_plugin_alconna.extension import cache_msg
 from nonebot_plugin_alconna import Extension, UniMessage, get_message_id
 
 
@@ -42,9 +43,9 @@ async def message_provider(
     ) -> UniMessage | None:
         if event.get_type() == "message_sent" and hasattr(event, "message"):
             msg_id = get_message_id(event, bot)
-            if use_origin and (uni_msg := self.cache.get(msg_id)) is not None:
+            if use_origin and cache_msg and (uni_msg := self.cache.get(msg_id)) is not None:
                 return uni_msg
-            if (uni_msg := self.cache.get(msg_id)) is not None:
+            if cache_msg and (uni_msg := self.cache.get(msg_id)) is not None:
                 return uni_msg
             msg = Message._validate(event.message)  # type: ignore
             uni_msg = UniMessage.generate_without_reply(message=msg, bot=bot)

src/nonebot_plugin_alconna/builtins/extensions/reply.py

@@ -4,6 +4,7 @@
 from arclet.alconna import Alconna
 from nonebot.internal.adapter import Bot, Event
 
+from nonebot_plugin_alconna.extension import cache_msg
 from nonebot_plugin_alconna import Reply, Extension, UniMessage
 from nonebot_plugin_alconna.uniseg import reply_fetch, get_message_id
 
@@ -94,7 +95,7 @@ async def message_provider(self, event, state, bot, use_origin: bool = False):
         except (NotImplementedError, ValueError):
             return None
         msg_id = get_message_id(event, bot)
-        if msg_id in self.cache:
+        if cache_msg and msg_id in self.cache:
             return self.cache[msg_id]
         uni_msg = UniMessage.generate_sync(message=msg, bot=bot)
         self.cache[msg_id] = uni_msg

src/nonebot_plugin_alconna/config.py

@@ -46,3 +46,6 @@ class Config(BaseModel):
 
     alconna_response_self: bool = False
     """是否响应自身发送的消息"""
+
+    alconna_cache_message: bool = True
+    """是否缓存已解析的消息"""

src/nonebot_plugin_alconna/extension.py

@@ -11,10 +11,12 @@
 
 from tarina import LRU, lang
 from nonebot.typing import T_State
+from nonebot import get_plugin_config
 from arclet.alconna import Alconna, Arparma
 from nonebot.compat import PydanticUndefined
 from nonebot.adapters import Bot, Event, Message
 
+from .config import Config
 from .uniseg import UniMessage, get_message_id
 
 OutputType = Literal["help", "shortcut", "completion", "error"]
@@ -34,6 +36,12 @@ class Interface(Generic[TE]):
     default: Any
 
 
+try:
+    cache_msg = get_plugin_config(Config).alconna_cache_message
+except ValueError:
+    cache_msg = True
+
+
 class Extension(metaclass=ABCMeta):
     _overrides: dict[str, bool]
 
@@ -148,9 +156,9 @@ async def message_provider(
             raise exc
         if event.get_type() == "message":
             msg_id = get_message_id(event, bot)
-            if use_origin and (uni_msg := unimsg_origin_cache.get(msg_id)) is not None:
+            if use_origin and cache_msg and (uni_msg := unimsg_origin_cache.get(msg_id)) is not None:
                 return uni_msg
-            if (uni_msg := unimsg_cache.get(msg_id)) is not None:
+            if cache_msg and (uni_msg := unimsg_cache.get(msg_id)) is not None:
                 return uni_msg
             msg = event.get_message()
             uni_msg = UniMessage.generate_without_reply(message=msg, bot=bot)