ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-11-27 18:45:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

💥 🐛 add support for non-plaintext start message

Browse Source
This commit is contained in:
Mix 2021-02-07 02:21:31 +08:00
parent bf7b2a8cbe
commit abcdbc4de9
2 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
nonebot/plugin.py
View File

@ -13,7 +13,7 @@ from types import ModuleType
from dataclasses import dataclass
from importlib._bootstrap import _load
from contextvars import Context, ContextVar, copy_context
from typing import Any, Set, List, Dict, Type, Tuple, Union, Optional, TYPE_CHECKING
from typing import Any, Set, List, Dict, Type, Tuple, Union, Optional, TYPE_CHECKING, Iterable
from nonebot.log import logger
from nonebot.matcher import Matcher
@ -22,7 +22,7 @@ from nonebot.typing import T_State, T_StateFactory, T_Handler, T_RuleChecker
from nonebot.rule import Rule, startswith, endswith, keyword, command, shell_command, ArgumentParser, regex
if TYPE_CHECKING:
from nonebot.adapters import Bot, Event
from nonebot.adapters import Bot, Event, MessageSegment
plugins: Dict[str, "Plugin"] = {}
"""
@ -421,13 +421,16 @@ def on_command(cmd: Union[str, Tuple[str, ...]],
"""
async def _strip_cmd(bot: "Bot", event: "Event", state: T_State):
message = event.get_message()
segment = message.pop(0)
new_message = message.__class__(
str(segment)
[len(state["_prefix"]["raw_command"]):].lstrip()) # type: ignore
for new_segment in reversed(new_message):
message.insert(0, new_segment)
message: Iterable[MessageSegment] = event.get_message()
text_processed = False
for index, segment in enumerate(message):
segment: MessageSegment = message.pop(index)
if segment.is_text() and not text_processed:
segment, *_ = message.__class__(
str(segment)[len(state["_prefix"]["raw_command"]):].lstrip(
)) # type: ignore
text_processed = True
message.insert(index, segment)
handlers = kwargs.pop("handlers", [])
handlers.insert(0, _strip_cmd)

7
nonebot/rule.py
View File

@ -25,7 +25,7 @@ from nonebot.exception import ParserExit
from nonebot.typing import T_State, T_RuleChecker
if TYPE_CHECKING:
from nonebot.adapters import Bot, Event
from nonebot.adapters import Bot, Event, MessageSegment
class Rule:
@ -137,8 +137,9 @@ class TrieRule:
prefix = None
suffix = None
message = event.get_message()
message_seg = message[0]
if message_seg.is_text():
message_seg: Optional[MessageSegment] = next(
filter(lambda x: x.is_text(), message), None)
if message_seg is not None:
prefix = cls.prefix.longest_prefix(str(message_seg).lstrip())
message_seg_r = message[-1]
if message_seg_r.is_text():