ayaka.deal

  1'''处理收到的消息，将其分割为cmd和args，设置上下文相关变量的值，并将消息传递给对应的群组'''
  2import asyncio
  3import datetime
  4from typing import List
  5
  6from .driver import Message, MessageSegment, Bot, MessageEvent, GroupMessageEvent
  7from .on import AyakaTrigger
  8from .config import ayaka_root_config
  9from .constant import _bot, _event, _group, _arg, _args, _message, _cmd, private_listener_dict
 10from .group import get_group
 11
 12
 13async def deal_event(bot: Bot, event: MessageEvent):
 14    if ayaka_root_config.exclude_old_msg:
 15        time_i = int(datetime.datetime.now().timestamp())
 16        if event.time < time_i - 60:
 17            return
 18
 19    _bot.set(bot)
 20    _event.set(event)
 21
 22    bot_id = int(bot.self_id)
 23
 24    if isinstance(event, GroupMessageEvent):
 25        group_id = event.group_id
 26        await deal_group(bot_id, group_id)
 27
 28    else:
 29        id = event.user_id
 30        group_ids = private_listener_dict.get(id, [])
 31        ts = [asyncio.create_task(deal_group(bot_id, group_id))
 32              for group_id in group_ids]
 33        await asyncio.gather(*ts)
 34
 35
 36async def deal_group(bot_id: int, group_id: int):
 37    # 群组
 38    group = get_group(bot_id, group_id)
 39    _group.set(group)
 40
 41    # 消息
 42    message = _event.get().message
 43    _message.set(message)
 44
 45    # 命令、参数
 46    cmd = get_cmd(message)
 47    _cmd.set(cmd)
 48
 49    # 让super先来
 50    triggers = []
 51    for app in group.apps:
 52        for t in app.super_triggers:
 53            triggers.append(t)
 54
 55    # 若执行了super命令，则退出
 56    if await deal_triggers(triggers):
 57        return
 58
 59    # 普通命令继续
 60    triggers = []
 61
 62    # 当前正在运行
 63    app = group.running_app
 64    if app:
 65        for t in app.state_triggers:
 66            if t.state == "*":
 67                triggers.append(t)
 68            else:
 69                # 子状态
 70                # s1 = 添加.
 71                # s2 = 添加.姓名.
 72                s1 = t.state + "."
 73                s2 = app.state + "."
 74                if s2.startswith(s1):
 75                    triggers.append(t)
 76
 77    # 其他桌面应用
 78    else:
 79        for app in group.apps:
 80            for t in app.no_state_triggers:
 81                triggers.append(t)
 82
 83    await deal_triggers(triggers)
 84
 85
 86async def deal_triggers(triggers: List[AyakaTrigger]):
 87    triggers.sort(key=lambda t: str(t.state), reverse=True)
 88
 89    cmd = _cmd.get()
 90    msg = _message.get()
 91
 92    # 命令
 93    if cmd:
 94        for t in triggers:
 95            if t.cmd == cmd:
 96                # 设置去掉命令后的消息
 97                arg = remove_cmd(cmd, msg)
 98                _arg.set(arg)
 99                _args.set(divide_message(arg))
100                await t.run()
101                return True
102
103    _arg.set(msg)
104    _args.set(divide_message(msg))
105
106    # 命令退化成消息
107    for t in triggers:
108        if not t.cmd:
109            await t.run()
110
111
112def get_cmd(message: Message):
113    '''返回命令'''
114    prefix = ayaka_root_config.prefix
115    sep = ayaka_root_config.separate
116    first = ""
117    for m in message:
118        if m.type == "text":
119            first += str(m)
120        else:
121            break
122        
123    if not first or not first.startswith(prefix):
124        return ""
125
126    if not sep:
127        return first
128    
129    first += sep
130    return first.split(sep, 1)[0][len(prefix):]
131
132
133def divide_message(message: Message) -> List[MessageSegment]:
134    args = []
135    sep = ayaka_root_config.separate
136
137    for m in message:
138        if m.is_text():
139            ss = str(m).split(sep)
140            args.extend(MessageSegment.text(s) for s in ss if s)
141        else:
142            args.append(m)
143
144    return args
145
146
147def remove_cmd(cmd: str, message: Message) -> Message:
148    prefix = ayaka_root_config.prefix
149    sep = ayaka_root_config.separate
150    m = message[0]
151    if m.is_text():
152        m_str = str(m)
153        m_str = m_str[len(prefix+cmd):].lstrip(sep)
154        if not m_str:
155            return Message(message[1:])
156        return Message([MessageSegment.text(m_str), *message[1:]])
157    return message