akari-bot/MessageGen.py

import asyncio
import os
import random
import re
import traceback
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 modules.findimage import findimage

try:
    cachepath = abspath('./assets/cache/')
    cachefile = os.listdir(cachepath)
    for file in cachefile:
        os.remove(f'{cachepath}/{file}')
except Exception:
    pass


async def gen(bcc, app, message, target1, target2='0', msgtype='None', runfun='command'):
    im = inter.InterruptControl(bcc)
    command = __import__('CommandGen', fromlist=[runfun])
    command = getattr(command, runfun)
    if msgtype == 'Group':
        run = await command(message.asDisplay(), target1.id)
    else:
        run = await command(message.asDisplay())
    # print(run)
    if run != None:
        await msgproc(run, app, im, command, message, target1, target2, msgtype)


async def msgproc(resultmsgraw, app, im, command, message, target1, target2='0', msgtype='None'):
    print(resultmsgraw)
    msgchain = await makemsgchain(resultmsgraw)
    send = await sendmessage(app, msgchain, target1, target2, msgtype,
                             message[Source][0] if msgtype == 'Group' else 0)
    uimgcs = re.findall(r'\[\[uimgc:.*\]\]', resultmsgraw, re.I | re.M)
    for uimgc in uimgcs:
        uimgc = re.match(r'\[\[uimgc:(.*)\]\]', uimgc)
        if uimgc:
            await uimgsend(app, message, target1, target2, msgtype, uimgc.group(1))
    r = re.findall(r'(https?://.*?/File:.*?\.(?:png|gif|jpg|jpeg|webp|bmp|ico))', resultmsgraw, re.I)
    for d in r:
        d1 = await findimage(d)
        if d1 is not None:
            await linkimgsend(app, d1, target1, target2, msgtype)
    await afterproc(resultmsgraw, app, im, command, send, message, target1, target2, msgtype)


async def makemsgchain(msg):
    msg = re.sub('\[wait\]', '', msg)
    msgbase = re.sub(r'\[\[uimgc:.*\]\]', '', msg)
    msgchain = MessageChain.create([Plain(msgbase)])
    return msgchain


async def afterproc(resultmsgraw, app, im, command, send, message, target1, target2='0', msgtype='None'):
    if resultmsgraw.find('[一分钟后撤回本消息]') != -1:
        await asyncio.sleep(60)
        await app.revokeMessage(send)
    if resultmsgraw.find('[30秒后撤回本消息]') != -1:
        await asyncio.sleep(30)
        await app.revokeMessage(send)
    if resultmsgraw.find('[wait]') != -1:
        ranint = random.randint(1, 3)
        if ranint == 2:
            waitmsg = await makemsgchain('提示：你可以发送“是”字来将所有无效结果再次查询。（考虑到实现复杂性，恕不提供选择性查询）')
            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(resultmsgraw)
            await msgproc(msg2, app, im, command, message, target1, target2, msgtype)
        else:
            pass


async def uimgsend(app, message, target1, target2, msgtype, link):
    exec('from graia.application.message.elements.internal import UploadMethods')
    mth = eval(f'UploadMethods.{msgtype}')
    try:
        msgchain = MessageChain.create(
            [Image.fromLocalFile(filepath=abspath(link), method=mth)])
        print('Sending Image...')
        await sendmessage(app, msgchain, target1, target2, msgtype,
                          message[Source][0] if msgtype == 'Group' else 0)
    except Exception:
        traceback.print_exc()
        msgchain = MessageChain.create(
            [Plain('上传过程中遇到了问题，图片发送失败。')])
        await sendmessage(app, msgchain, target1, target2, msgtype,
                          message[Source][0] if msgtype == 'Group' else 0)


async def linkimgsend(app, sendlink, target1, target2, msgtype):
    exec('from graia.application.message.elements.internal import UploadMethods')
    mth = eval(f'UploadMethods.{msgtype}')
    try:
        msgchain = MessageChain.create([Image.fromNetworkAddress(url=sendlink, method=mth)])
        print('Sending Image...')
        await sendmessage(app, msgchain, target1, target2, msgtype)
    except Exception:
        traceback.print_exc()
        msgchain = MessageChain.create(
            [Plain('上传过程中遇到了问题，图片发送失败。')])
        await sendmessage(app, msgchain, target1, target2, msgtype)


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


"""
if msgtype == 'Group':
    voice = re.findall(r'https?://.*?/File:.*?\.(?:ogg|m4a|mp3|flac|wav)', run, re.I)
    for voicelink in voice:
        try:
            findvoicename = re.match(r'(https?://.*?/)File:(.*?\.(?:ogg|m4a|mp3|flac|wav))', voicelink, re.I)
            downloadfile = await dfile(findvoicename.group(1), findvoicename.group(2))
            print(downloadfile)
            conventamr = await camr(downloadfile)
            print(conventamr)
            readfile = open(conventamr, 'rb+')
            uploadvoice = await app.uploadVoice(readfile.read())
            voicemsgchain = MessageChain.create([uploadvoice])
            await app.sendGroupMessage(target1, voicemsgchain)
            readfile.close()
            os.remove(downloadfile)
            os.remove(conventamr)
        except Exception:
            traceback.print_exc()
"""