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-07-01 11:01:24 +08:00
|
|
|
from .natural_language import handle_natural_language
|
2018-06-27 22:50:01 +08:00
|
|
|
from .log import logger
|
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-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
|
2018-06-30 17:26:36 +08:00
|
|
|
first_message_seg = ctx['message'][0]
|
|
|
|
if first_message_seg == MessageSegment.at(ctx['self_id']):
|
|
|
|
ctx['to_me'] = True
|
|
|
|
del ctx['message'][0]
|
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:
|
2018-07-01 11:01:24 +08:00
|
|
|
logger.debug('Message is handled as command')
|
2018-06-27 16:36:40 +08:00
|
|
|
return
|
2018-06-30 10:58:56 +08:00
|
|
|
|
2018-07-01 11:01:24 +08:00
|
|
|
handled = await handle_natural_language(bot, ctx)
|
|
|
|
if handled:
|
|
|
|
logger.debug('Message is handled as natural language')
|
|
|
|
return
|