nonebot2/none/message.py

36 lines
1.1 KiB
Python
Raw Normal View History

2018-06-15 06:58:24 +08:00
from typing import Dict, Any
from aiocqhttp import CQHttp
from aiocqhttp.message import MessageSegment
2018-06-27 22:05:12 +08:00
from .command import handle_command
2018-06-27 22:50:01 +08:00
from .log import logger
2018-06-30 10:58:56 +08:00
from .helpers import send
2018-06-15 06:58:24 +08:00
2018-06-27 16:36:40 +08:00
async def handle_message(bot: CQHttp, ctx: Dict[str, Any]) -> None:
2018-06-29 22:57:53 +08:00
# TODO: 支持让插件自己选择是否响应没有 at 的消息
2018-06-15 06:58:24 +08:00
if ctx['message_type'] != 'private':
# group or discuss
2018-06-30 10:58:56 +08:00
ctx['to_me'] = False
indexes_to_remove = []
for i, seg in enumerate(ctx['message']):
if seg == MessageSegment.at(ctx['self_id']):
ctx['to_me'] = True
indexes_to_remove.append(i)
for i in reversed(indexes_to_remove):
del ctx['message'][i]
2018-06-15 06:58:24 +08:00
if not ctx['message']:
ctx['message'].append(MessageSegment.text(''))
2018-06-30 10:58:56 +08:00
else:
ctx['to_me'] = True
2018-06-15 06:58:24 +08:00
2018-06-27 16:36:40 +08:00
handled = await handle_command(bot, ctx)
2018-06-15 06:58:24 +08:00
if handled:
logger.debug('Message is handled as a command')
2018-06-27 16:36:40 +08:00
return
2018-06-30 10:58:56 +08:00
elif ctx['to_me']:
await send(bot, ctx, '你在说什么我看不懂诶')
# TODO: NLP