import asyncio from typing import Callable from aiocqhttp.message import * from . import NoneBot from .command import handle_command, SwitchException from .log import logger from .natural_language import handle_natural_language from .typing import Context_T _message_preprocessors = set() def message_preprocessor(func: Callable) -> Callable: _message_preprocessors.add(func) return func async def handle_message(bot: NoneBot, ctx: Context_T) -> None: _log_message(ctx) if not ctx['message']: ctx['message'].append(MessageSegment.text('')) coros = [] for processor in _message_preprocessors: coros.append(processor(bot, ctx)) if coros: await asyncio.wait(coros) if 'to_me' not in ctx: if ctx['message_type'] == 'private': ctx['to_me'] = True else: # group or discuss ctx['to_me'] = False at_me_seg = MessageSegment.at(ctx['self_id']) # check the first segment first_msg_seg = ctx['message'][0] if first_msg_seg == at_me_seg: ctx['to_me'] = True del ctx['message'][0] if not ctx['to_me']: # check the last segment i = -1 last_msg_seg = ctx['message'][i] if last_msg_seg.type == 'text' and \ not last_msg_seg.data['text'].strip() and \ len(ctx['message']) >= 2: i -= 1 last_msg_seg = ctx['message'][i] if last_msg_seg == at_me_seg: ctx['to_me'] = True del ctx['message'][i:] if not ctx['message']: ctx['message'].append(MessageSegment.text('')) while True: try: handled = await handle_command(bot, ctx) break except SwitchException as e: # we are sure that there is no session existing now ctx['message'] = e.new_ctx_message ctx['to_me'] = True if handled: logger.info(f'Message {ctx["message_id"]} is handled as a command') return handled = await handle_natural_language(bot, ctx) if handled: logger.info(f'Message {ctx["message_id"]} is handled ' f'as natural language') return def _log_message(ctx: Context_T) -> None: msg_from = str(ctx['user_id']) if ctx['message_type'] == 'group': msg_from += f'@[群:{ctx["group_id"]}]' elif ctx['message_type'] == 'discuss': msg_from += f'@[讨论组:{ctx["discuss_id"]}]' logger.info(f'Self: {ctx["self_id"]}, ' f'Message {ctx["message_id"]} from {msg_from}: ' f'{ctx["message"]}')