2021-12-25 14:01:33 +00:00
|
|
|
import datetime
|
2021-08-07 07:56:48 +00:00
|
|
|
import re
|
2021-08-02 15:35:45 +00:00
|
|
|
import traceback
|
2021-12-31 14:44:34 +00:00
|
|
|
from typing import List, Union
|
2021-08-02 15:35:45 +00:00
|
|
|
|
|
|
|
import discord
|
2023-05-02 12:07:19 +00:00
|
|
|
import filetype
|
2021-08-07 07:56:48 +00:00
|
|
|
|
2023-08-29 18:13:34 +00:00
|
|
|
from bots.discord.client import client
|
|
|
|
from bots.discord.info import client_name
|
2022-07-14 13:23:15 +00:00
|
|
|
from config import Config
|
2023-12-14 03:11:00 +00:00
|
|
|
from core.builtins import Bot, Plain, Image, MessageSession as MessageSessionT, MessageTaskManager
|
2023-02-05 14:33:33 +00:00
|
|
|
from core.builtins.message.chain import MessageChain
|
2023-06-29 14:53:46 +00:00
|
|
|
from core.builtins.message.internal import Embed, ErrorMessage, Voice
|
2022-01-18 12:32:43 +00:00
|
|
|
from core.logger import Logger
|
2023-09-03 17:21:01 +00:00
|
|
|
from core.types import FetchTarget as FetchTargetT, FinishedSession as FinS
|
2023-05-02 12:07:19 +00:00
|
|
|
from core.utils.http import download_to_cache
|
2021-08-30 18:53:39 +00:00
|
|
|
from database import BotDBUtil
|
2021-08-02 15:35:45 +00:00
|
|
|
|
2022-07-14 13:23:15 +00:00
|
|
|
enable_analytics = Config('enable_analytics')
|
|
|
|
|
|
|
|
|
2022-01-13 12:08:14 +00:00
|
|
|
async def convert_embed(embed: Embed):
|
2021-12-25 14:01:33 +00:00
|
|
|
if isinstance(embed, Embed):
|
2022-01-13 12:08:14 +00:00
|
|
|
files = []
|
2023-12-19 08:50:45 +00:00
|
|
|
embeds = discord.Embed(title=embed.title if embed.title else discord.Embed.Empty,
|
|
|
|
description=embed.description if embed.description else discord.Embed.Empty,
|
|
|
|
color=embed.color if embed.color else discord.Embed.Empty,
|
|
|
|
url=embed.url if embed.url else discord.Embed.Empty,
|
2021-12-31 14:44:34 +00:00
|
|
|
timestamp=datetime.datetime.fromtimestamp(
|
2023-12-19 08:50:45 +00:00
|
|
|
embed.timestamp) if embed.timestamp else discord.Embed.Empty, )
|
|
|
|
if embed.image:
|
2022-01-13 12:08:14 +00:00
|
|
|
upload = discord.File(await embed.image.get(), filename="image.png")
|
|
|
|
files.append(upload)
|
|
|
|
embeds.set_image(url="attachment://image.png")
|
2023-12-19 08:50:45 +00:00
|
|
|
if embed.thumbnail:
|
2022-01-13 12:08:14 +00:00
|
|
|
upload = discord.File(await embed.thumbnail.get(), filename="thumbnail.png")
|
|
|
|
files.append(upload)
|
|
|
|
embeds.set_thumbnail(url="attachment://thumbnail.png")
|
2023-12-19 08:50:45 +00:00
|
|
|
if embed.author:
|
2021-12-25 14:01:33 +00:00
|
|
|
embeds.set_author(name=embed.author)
|
2023-12-19 08:50:45 +00:00
|
|
|
if embed.footer:
|
2021-12-25 14:01:33 +00:00
|
|
|
embeds.set_footer(text=embed.footer)
|
2023-12-19 08:50:45 +00:00
|
|
|
if embed.fields:
|
2021-12-25 14:01:33 +00:00
|
|
|
for field in embed.fields:
|
|
|
|
embeds.add_field(name=field.name, value=field.value, inline=field.inline)
|
2022-01-13 12:08:14 +00:00
|
|
|
return embeds, files
|
2021-12-25 14:01:33 +00:00
|
|
|
|
|
|
|
|
2022-01-15 14:28:35 +00:00
|
|
|
class FinishedSession(FinS):
|
|
|
|
async def delete(self):
|
|
|
|
"""
|
|
|
|
用于删除这条消息。
|
|
|
|
"""
|
|
|
|
try:
|
|
|
|
for x in self.result:
|
|
|
|
await x.delete()
|
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2022-01-15 14:28:35 +00:00
|
|
|
|
|
|
|
|
2023-09-03 17:21:01 +00:00
|
|
|
class MessageSession(MessageSessionT):
|
2021-08-02 15:35:45 +00:00
|
|
|
class Feature:
|
|
|
|
image = True
|
2023-06-29 14:53:46 +00:00
|
|
|
voice = True
|
2022-01-16 13:24:15 +00:00
|
|
|
embed = True
|
2021-10-11 14:45:28 +00:00
|
|
|
forward = False
|
2021-11-16 14:19:48 +00:00
|
|
|
delete = True
|
2022-01-16 13:24:15 +00:00
|
|
|
quote = True
|
|
|
|
wait = True
|
2021-08-02 15:35:45 +00:00
|
|
|
|
2023-12-14 03:11:00 +00:00
|
|
|
async def send_message(self, message_chain, quote=True, disable_secret_check=False, allow_split_image=True,
|
|
|
|
callback=None
|
2023-09-01 14:38:32 +00:00
|
|
|
) -> FinishedSession:
|
|
|
|
message_chain = MessageChain(message_chain)
|
|
|
|
if not message_chain.is_safe and not disable_secret_check:
|
|
|
|
return await self.send_message(Plain(ErrorMessage(self.locale.t("error.message.chain.unsafe"))))
|
|
|
|
self.sent.append(message_chain)
|
2021-12-31 14:44:34 +00:00
|
|
|
count = 0
|
|
|
|
send = []
|
2023-12-09 13:42:46 +00:00
|
|
|
for x in message_chain.as_sendable(self):
|
2021-12-31 14:44:34 +00:00
|
|
|
if isinstance(x, Plain):
|
|
|
|
send_ = await self.session.target.send(x.text,
|
|
|
|
reference=self.session.message if quote and count == 0
|
2023-04-30 03:30:59 +00:00
|
|
|
and self.session.message else None)
|
2023-09-01 14:38:32 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.target_id}]: {x.text}')
|
2021-12-31 14:44:34 +00:00
|
|
|
elif isinstance(x, Image):
|
|
|
|
send_ = await self.session.target.send(file=discord.File(await x.get()),
|
|
|
|
reference=self.session.message if quote and count == 0
|
2023-04-30 03:30:59 +00:00
|
|
|
and self.session.message else None)
|
2023-09-01 14:38:32 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.target_id}]: Image: {str(x.__dict__)}')
|
2023-06-29 14:53:46 +00:00
|
|
|
elif isinstance(x, Voice):
|
|
|
|
send_ = await self.session.target.send(file=discord.File(x.path),
|
|
|
|
reference=self.session.message if quote and count == 0
|
|
|
|
and self.session.message else None)
|
2023-09-01 14:38:32 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.target_id}]: Voice: {str(x.__dict__)}')
|
2023-06-29 14:53:46 +00:00
|
|
|
|
2021-12-31 14:44:34 +00:00
|
|
|
elif isinstance(x, Embed):
|
2022-01-13 12:08:14 +00:00
|
|
|
embeds, files = await convert_embed(x)
|
|
|
|
send_ = await self.session.target.send(embed=embeds,
|
2021-12-31 14:44:34 +00:00
|
|
|
reference=self.session.message if quote and count == 0
|
2023-04-30 03:30:59 +00:00
|
|
|
and self.session.message else None,
|
2022-01-13 12:08:14 +00:00
|
|
|
files=files)
|
2023-09-01 14:38:32 +00:00
|
|
|
Logger.info(f'[Bot] -> [{self.target.target_id}]: Embed: {str(x.__dict__)}')
|
2021-12-31 14:44:34 +00:00
|
|
|
else:
|
2023-12-14 03:11:00 +00:00
|
|
|
send_ = None
|
2021-12-31 14:44:34 +00:00
|
|
|
if send_:
|
|
|
|
send.append(send_)
|
|
|
|
count += 1
|
2023-09-03 11:49:14 +00:00
|
|
|
msg_ids = []
|
2022-06-28 06:11:03 +00:00
|
|
|
for x in send:
|
2023-09-03 11:49:14 +00:00
|
|
|
msg_ids.append(x.id)
|
2023-12-14 03:11:00 +00:00
|
|
|
if callback:
|
|
|
|
MessageTaskManager.add_callback(x.id, callback)
|
2022-06-28 06:11:03 +00:00
|
|
|
|
2023-09-03 11:49:14 +00:00
|
|
|
return FinishedSession(self, msg_ids, send)
|
2021-08-02 15:35:45 +00:00
|
|
|
|
2023-09-01 14:38:32 +00:00
|
|
|
async def check_native_permission(self):
|
2023-08-28 16:24:38 +00:00
|
|
|
if not self.session.message:
|
|
|
|
channel = await client.fetch_channel(self.session.target)
|
|
|
|
author = await channel.guild.fetch_member(self.session.sender)
|
|
|
|
else:
|
|
|
|
channel = self.session.message.channel
|
|
|
|
author = self.session.message.author
|
|
|
|
try:
|
|
|
|
if channel.permissions_for(author).administrator \
|
|
|
|
or isinstance(channel, discord.DMChannel):
|
|
|
|
return True
|
|
|
|
except Exception:
|
|
|
|
Logger.error(traceback.format_exc())
|
2022-01-08 08:37:03 +00:00
|
|
|
return False
|
|
|
|
|
2023-09-01 14:38:32 +00:00
|
|
|
async def to_message_chain(self):
|
2023-05-02 12:07:19 +00:00
|
|
|
lst = []
|
|
|
|
lst.append(Plain(self.session.message.content))
|
|
|
|
for x in self.session.message.attachments:
|
|
|
|
d = await download_to_cache(x.url)
|
|
|
|
if filetype.is_image(d):
|
|
|
|
lst.append(Image(d))
|
|
|
|
return MessageChain(lst)
|
|
|
|
|
2023-09-01 14:38:32 +00:00
|
|
|
def as_display(self, text_only=False):
|
2023-01-31 13:09:29 +00:00
|
|
|
msg = self.session.message.content
|
2023-01-31 13:10:51 +00:00
|
|
|
msg = re.sub(r'<@(.*?)>', r'Discord|Client|\1', msg)
|
2023-01-31 13:09:29 +00:00
|
|
|
return msg
|
2021-08-02 15:35:45 +00:00
|
|
|
|
|
|
|
async def delete(self):
|
|
|
|
try:
|
2022-08-04 15:52:53 +00:00
|
|
|
await self.session.message.delete()
|
2021-08-23 12:44:31 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2021-08-02 15:35:45 +00:00
|
|
|
|
2023-09-04 07:04:28 +00:00
|
|
|
sendMessage = send_message
|
|
|
|
asDisplay = as_display
|
|
|
|
toMessageChain = to_message_chain
|
|
|
|
checkNativePermission = check_native_permission
|
|
|
|
|
2021-08-02 15:35:45 +00:00
|
|
|
class Typing:
|
2023-09-03 17:21:01 +00:00
|
|
|
def __init__(self, msg: MessageSessionT):
|
2021-08-02 15:35:45 +00:00
|
|
|
self.msg = msg
|
|
|
|
|
|
|
|
async def __aenter__(self):
|
|
|
|
async with self.msg.session.target.typing() as typing:
|
|
|
|
return typing
|
|
|
|
|
|
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
2021-08-07 07:56:48 +00:00
|
|
|
pass
|
|
|
|
|
|
|
|
|
2023-08-27 14:51:16 +00:00
|
|
|
class FetchedSession(Bot.FetchedSession):
|
2022-01-15 13:46:23 +00:00
|
|
|
|
2023-09-01 14:38:32 +00:00
|
|
|
async def send_direct_message(self, message_chain, disable_secret_check=False, allow_split_image=True):
|
2022-01-15 13:46:23 +00:00
|
|
|
try:
|
2023-09-03 11:49:14 +00:00
|
|
|
get_channel = await client.fetch_channel(self.session.target)
|
2022-01-15 13:46:23 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2022-01-15 13:46:23 +00:00
|
|
|
return False
|
2023-09-03 11:49:14 +00:00
|
|
|
self.session.target = self.session.sender = self.parent.session.target = self.parent.session.sender = get_channel
|
2023-09-01 14:38:32 +00:00
|
|
|
return await self.parent.send_direct_message(message_chain, disable_secret_check=disable_secret_check)
|
2022-01-15 13:46:23 +00:00
|
|
|
|
|
|
|
|
2023-08-27 14:51:16 +00:00
|
|
|
Bot.FetchedSession = FetchedSession
|
|
|
|
|
|
|
|
|
2023-09-03 17:21:01 +00:00
|
|
|
class FetchTarget(FetchTargetT):
|
2023-08-27 14:51:16 +00:00
|
|
|
name = client_name
|
2023-08-30 05:15:33 +00:00
|
|
|
|
|
|
|
@staticmethod
|
2023-09-01 14:38:32 +00:00
|
|
|
async def fetch_target(target_id, sender_id=None) -> Union[Bot.FetchedSession]:
|
2023-09-03 11:49:14 +00:00
|
|
|
match_channel = re.match(r'^(Discord\|(?:DM\||)Channel)\|(.*)', target_id)
|
|
|
|
if match_channel:
|
|
|
|
target_from = sender_from = match_channel.group(1)
|
|
|
|
target_id = match_channel.group(2)
|
2023-09-01 14:38:32 +00:00
|
|
|
if sender_id:
|
2023-09-03 11:49:14 +00:00
|
|
|
match_sender = re.match(r'^(Discord\|Client)\|(.*)', sender_id)
|
|
|
|
if match_sender:
|
|
|
|
sender_from = match_sender.group(1)
|
|
|
|
sender_id = match_sender.group(2)
|
2023-08-30 05:15:33 +00:00
|
|
|
else:
|
2023-09-01 14:38:32 +00:00
|
|
|
sender_id = target_id
|
2023-08-30 05:15:33 +00:00
|
|
|
|
2023-09-03 11:49:14 +00:00
|
|
|
return Bot.FetchedSession(target_from, target_id, sender_from, sender_id)
|
2021-08-30 18:53:39 +00:00
|
|
|
|
2021-10-08 11:54:27 +00:00
|
|
|
@staticmethod
|
2023-09-01 14:38:32 +00:00
|
|
|
async def fetch_target_list(target_list: list) -> List[Bot.FetchedSession]:
|
2021-10-08 11:54:27 +00:00
|
|
|
lst = []
|
2023-09-01 14:38:32 +00:00
|
|
|
for x in target_list:
|
2021-10-08 11:54:27 +00:00
|
|
|
fet = await FetchTarget.fetch_target(x)
|
|
|
|
if fet:
|
|
|
|
lst.append(fet)
|
|
|
|
return lst
|
|
|
|
|
2021-08-30 18:53:39 +00:00
|
|
|
@staticmethod
|
2023-08-27 14:51:16 +00:00
|
|
|
async def post_message(module_name, message, user_list: List[Bot.FetchedSession] = None, i18n=False, **kwargs):
|
2023-12-19 08:50:45 +00:00
|
|
|
if user_list:
|
2021-08-30 18:53:39 +00:00
|
|
|
for x in user_list:
|
|
|
|
try:
|
2023-12-13 08:59:56 +00:00
|
|
|
msgchain = message
|
|
|
|
if isinstance(message, str):
|
|
|
|
if i18n:
|
|
|
|
msgchain = MessageChain([Plain(x.parent.locale.t(message, **kwargs))])
|
|
|
|
else:
|
|
|
|
msgchain = MessageChain([Plain(message)])
|
|
|
|
await x.send_direct_message(msgchain)
|
2022-07-14 13:23:15 +00:00
|
|
|
if enable_analytics:
|
|
|
|
BotDBUtil.Analytics(x).add('', module_name, 'schedule')
|
2021-08-30 18:53:39 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2021-08-30 18:53:39 +00:00
|
|
|
else:
|
2023-08-30 09:24:24 +00:00
|
|
|
get_target_id = BotDBUtil.TargetInfo.get_enabled_this(module_name, "Discord")
|
2021-08-30 18:53:39 +00:00
|
|
|
for x in get_target_id:
|
2023-09-05 15:10:48 +00:00
|
|
|
fetch = await FetchTarget.fetch_target(x.targetId)
|
2021-08-30 18:53:39 +00:00
|
|
|
if fetch:
|
|
|
|
try:
|
2023-12-13 08:59:56 +00:00
|
|
|
msgchain = message
|
|
|
|
if isinstance(message, str):
|
|
|
|
if i18n:
|
|
|
|
msgchain = MessageChain([Plain(fetch.parent.locale.t(message, **kwargs))])
|
|
|
|
else:
|
|
|
|
msgchain = MessageChain([Plain(message)])
|
|
|
|
await fetch.send_direct_message(msgchain)
|
2022-07-14 13:23:15 +00:00
|
|
|
if enable_analytics:
|
|
|
|
BotDBUtil.Analytics(fetch).add('', module_name, 'schedule')
|
2021-08-30 18:53:39 +00:00
|
|
|
except Exception:
|
2022-01-20 13:31:50 +00:00
|
|
|
Logger.error(traceback.format_exc())
|
2023-02-04 14:42:21 +00:00
|
|
|
|
|
|
|
|
|
|
|
Bot.MessageSession = MessageSession
|
|
|
|
Bot.FetchTarget = FetchTarget
|