Merge branch 'main' of https://github.com/LiteyukiStudio/LiteyukiBot

liteyuki/message/matcher.py

@@ -34,16 +34,17 @@ class Matcher:
     def __str__(self):
         return f"Matcher(rule={self.rule}, priority={self.priority}, block={self.block})"
 
-    def handle(self, handler: EventHandler) -> EventHandler:
+    def handle(self) -> Callable[[EventHandler], EventHandler]:
         """
         添加处理函数，装饰器
-        Args:
-            handler:
         Returns:
-            EventHandler
+            装饰器 handler
         """
-        self.handlers.append(handler)
-        return handler
+        def decorator(handler: EventHandler) -> EventHandler:
+            self.handlers.append(handler)
+            return handler
+
+        return decorator
 
     async def run(self, event: MessageEvent) -> None:
         """

liteyuki/message/on.py

@@ -15,7 +15,7 @@ from liteyuki.comm.storage import shared_memory
 from liteyuki.log import logger
 from liteyuki.message.event import MessageEvent
 from liteyuki.message.matcher import Matcher
-from liteyuki.message.rule import Rule
+from liteyuki.message.rule import Rule, empty_rule
 
 _matcher_list: list[Matcher] = []
 _queue: Queue = Queue()
@@ -36,7 +36,7 @@ async def _(event: MessageEvent):
                 break
 
 
-def on_message(rule: Rule = Rule(), priority: int = 0, block: bool = True) -> Matcher:
+def on_message(rule: Rule = empty_rule, priority: int = 0, block: bool = False) -> Matcher:
     matcher = Matcher(rule, priority, block)
     # 按照优先级插入
     for i, m in enumerate(_matcher_list):
@@ -46,3 +46,10 @@ def on_message(rule: Rule = Rule(), priority: int = 0, block: bool = True) -> Ma
     else:
         _matcher_list.append(matcher)
     return matcher
+
+
+def on_keywords(keywords: list[str], rule=empty_rule, priority: int = 0, block: bool = False) -> Matcher:
+    @Rule
+    async def on_keywords_rule(event: MessageEvent):
+        return any(keyword in event.raw_message for keyword in keywords)
+    return on_message(on_keywords_rule & rule, priority, block)

liteyuki/message/rule.py

@@ -8,26 +8,37 @@ Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved
 @File    : rule.py
 @Software: PyCharm
 """
-
+import inspect
 from typing import Optional, TypeAlias, Callable, Coroutine
 
 from liteyuki.message.event import MessageEvent
 
-RuleHandler: TypeAlias = Callable[[MessageEvent], Coroutine[None, None, bool]]
+RuleHandlerFunc: TypeAlias = Callable[[MessageEvent], Coroutine[None, None, bool]]
 """规则函数签名"""
 
 
 class Rule:
-    def __init__(self, handler: Optional[RuleHandler] = None):
+    def __init__(self, handler: RuleHandlerFunc):
         self.handler = handler
 
     def __or__(self, other: "Rule") -> "Rule":
-        return Rule(lambda event: self.handler(event) or other.handler(event))
+        async def combined_handler(event: MessageEvent) -> bool:
+            return await self.handler(event) or await other.handler(event)
+
+        return Rule(combined_handler)
 
     def __and__(self, other: "Rule") -> "Rule":
-        return Rule(lambda event: self.handler(event) and other.handler(event))
+        async def combined_handler(event: MessageEvent) -> bool:
+            return await self.handler(event) and await other.handler(event)
+
+        return Rule(combined_handler)
 
     async def __call__(self, event: MessageEvent) -> bool:
         if self.handler is None:
             return True
         return await self.handler(event)
+
+
+@Rule
+async def empty_rule(event: MessageEvent) -> bool:
+    return True