104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
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
|
|
from core.dirty_check import rickroll
|
|
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:
|
|
message = await rickroll()
|
|
return await self.sendMessage(message)
|
|
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())
|