akari-bot/MessageGen.py

import asyncio
import random
import re
from os.path import abspath

import graia.application.interrupt as inter
from graia.application.message.chain import MessageChain
from graia.application.message.elements.internal import Plain, Image, Source

from CommandGen import command
from modules.findimage import findimage


async def gen(bcc, app, message, target1, target2='0', msgtype='None'):
    im = inter.InterruptControl(bcc)
    if msgtype == 'Group':
        run = await command(message.asDisplay(), target1.id)
    else:
        run = await command(message.asDisplay())
    #    print(run)
    if run != None:
        print(run)
        msgchain = await makemsgchain(run, msgtype)
        send = await sendmessage(app, msgchain, target1, target2, msgtype,
                                 message[Source][0] if msgtype == 'Group' else 0)
        '''
        if msgtype == 'group':
            voice = re.findall(r'https?://.*?/File:.*?\.(?:ogg|m4a|mp3|flac|wav)', run, re.I)
            for x in voice:
                y = re.match(r'(https?://.*?/)File:(.*?\.(?:ogg|m4a|mp3|flac|wav))', x, re.I)
                z = await dfile(y.group(1), y.group(2))
                c = await camr(z)
                voicemsgchain = MessageChain.create(Voice.fromExternal(abspath(c)))
                await app.sendGroupMessage(group, voicemsgchain)
        '''
        if run.find('[一分钟后撤回本消息]') != -1:
            await asyncio.sleep(60)
            await app.revokeMessage(send)
        if run.find('[30秒后撤回本消息]') != -1:
            await asyncio.sleep(30)
            await app.revokeMessage(send)
        if run.find('[wait]') != -1:
            ranint = random.randint(1, 3)
            if ranint == 2:
                waitmsg = await makemsgchain('提示：你可以发送“是”字来将所有无效结果再次查询。（考虑到实现复杂性，恕不提供选择性查询）', msgtype)
                await sendmessage(app, waitmsg, target1, target2, msgtype)
            MessageEventImport = __import__('graia.application', fromlist=[f'{msgtype}Message'])
            MessageEvent = getattr(MessageEventImport, f'{msgtype}Message')
            InterruptImport = __import__('graia.application.interrupt.interrupts',
                                         fromlist=[f'{msgtype}MessageInterrupt'])
            Interrupt = getattr(InterruptImport, f'{msgtype}MessageInterrupt')
            if msgtype == 'Friend':
                event: MessageEvent = await im.wait(Interrupt(target1.id))
            else:
                event: MessageEvent = await im.wait(Interrupt(target1, target2))
            print(event)
            if event.messageChain.asDisplay() == '是':
                msg2 = await command(run)
                msgchain = await makemsgchain(msg2, msgtype)
                await sendmessage(app, msgchain, target1, target2, msgtype)

            else:
                pass


async def makemsgchain(msg, msgtype):
    msg = re.sub('\[wait\]', '', msg)
    exec('from graia.application.message.elements.internal import UploadMethods')
    mth = eval(f'UploadMethods.{msgtype}')
    if msg.find('[[usn:') != -1:
        user = re.sub(r'.*\[\[usn:|\]\]', '', msg)
        msg = re.sub(r'\[\[.*\]\]', '', msg)
        msgchain = MessageChain.create(
            [Plain(msg)])
        msgchain = msgchain.plusWith(
            [Image.fromLocalFile(filepath=abspath(f"./assests/usercard/{user}.png"), method=mth)])
    else:
        msgchain = MessageChain.create(
            [Plain(msg)])
    r = re.findall(r'(https?://.*?/File:.*?\.(?:png|gif|jpg|jpeg|webp|bmp|ico))', msg, re.I)
    for d in r:
        d1 = await findimage(d)
        print(d1)
        msgchain = msgchain.plusWith([Image.fromNetworkAddress(url=d1, method=mth)])
    return msgchain


async def sendmessage(app, msgchain, target1, target2, msgtype, quoteid=0):
    if msgtype == 'Friend':
        friend = target1
        send = await app.sendFriendMessage(friend, msgchain.asSendable())
    if msgtype == 'Group':
        group = target1
        send = await app.sendGroupMessage(group, msgchain.asSendable(), quote=quoteid if quoteid != 0 else None)
    if msgtype == 'Temp':
        group = target1
        member = target2
        send = await app.sendTempMessage(group=group, target=member, message=msgchain.asSendable())
    return send