akari-bot/bots/discord/slash_message.py

import traceback

import discord

from bots.discord.message import convert_embed
from config import Config
from core.builtins import Plain, Image, MessageSession as MS
from core.builtins.message.chain import MessageChain
from core.builtins.message.internal import Embed, ErrorMessage
from core.logger import Logger
from core.types import FinishedSession as FinS

enable_analytics = Config('enable_analytics')


class FinishedSession(FinS):
    async def delete(self):
        """
        用于删除这条消息。
        """
        try:
            for x in self.result:
                await x.delete()
        except Exception:
            Logger.error(traceback.format_exc())


class MessageSession(MS):
    command = ''

    class Feature:
        image = True
        voice = False
        embed = True
        forward = False
        delete = True
        quote = False
        wait = True

    async def sendMessage(self, msgchain, quote=True, disable_secret_check=False, allow_split_image=True
                          ) -> FinishedSession:
        msgchain = MessageChain(msgchain)
        if not msgchain.is_safe and not disable_secret_check:
            return await self.sendMessage(Plain(ErrorMessage(self.locale.t("error.message.chain.unsafe"))))
        self.sent.append(msgchain)
        count = 0
        send = []
        first_send = True
        for x in msgchain.asSendable():
            if isinstance(x, Plain):
                if first_send:
                    send_ = await self.session.message.respond(x.text)
                else:
                    send_ = await self.session.message.send(x.text)
                Logger.info(f'[Bot] -> [{self.target.targetId}]: {x.text}')
            elif isinstance(x, Image):
                if first_send:
                    send_ = await self.session.message.respond(file=discord.File(await x.get()))
                else:
                    send_ = await self.session.message.send(file=discord.File(await x.get()))
                Logger.info(f'[Bot] -> [{self.target.targetId}]: Image: {str(x.__dict__)}')
            elif isinstance(x, Embed):
                embeds, files = await convert_embed(x)
                if first_send:
                    send_ = await self.session.message.respond(embed=embeds)
                else:
                    send_ = await self.session.message.send(embed=embeds)
                Logger.info(f'[Bot] -> [{self.target.targetId}]: Embed: {str(x.__dict__)}')
            else:
                send_ = False
            if send_:
                send.append(send_)
            count += 1
            first_send = False
        msgIds = []
        for x in send:
            msgIds.append(x.id)

        return FinishedSession(self, msgIds, send)

    async def checkPermission(self):
        if self.session.message.channel.permissions_for(self.session.message.author).administrator \
                or isinstance(self.session.message.channel, discord.DMChannel) \
                or self.target.senderInfo.query.isSuperUser \
                or self.target.senderInfo.check_TargetAdmin(self.target.targetId):
            return True
        return False

    async def checkNativePermission(self):
        if self.session.message.channel.permissions_for(self.session.message.author).administrator \
                or isinstance(self.session.message.channel, discord.DMChannel):
            return True
        return False

    def asDisplay(self):
        return self.command

    async def delete(self):
        try:
            await self.session.message.delete()
        except Exception:
            Logger.error(traceback.format_exc())