async def acloseall(self, ctx, *, reason: str = None):
for channel in ctx.guild.text_channels:
if checks.is_modmail_channel2(self.bot, channel):
msg = copy.copy(ctx.message)
msg.channel = channel
new_ctx = await self.bot.get_context(msg, cls=type(ctx))
await self.close_channel(new_ctx, reason, True)
try:
await ctx.send(embed=discord.Embed(
description="All channels are successfully closed anonymously.",
colour=self.bot.primary_colour,
))
except discord.HTTPException:
pass
async def on_message(self, message):
if message.author.bot or not message.guild or not message.channel.category_id:
return
if not checks.is_modmail_channel2(self.bot, message.channel):
return
if (message.channel.permissions_for(message.guild.me).send_messages is
False or message.channel.permissions_for(
message.guild.me).embed_links is False):
return
prefix = self.bot.tools.get_guild_prefix(self.bot, message)
if message.content.startswith(prefix):
return
if message.author.id in self.bot.banned_users:
return await message.channel.send(embed=discord.Embed(
description="You are banned from this bot.",
colour=self.bot.error_colour))
await self.send_mail_mod(message, prefix)
async def closeall(self, ctx, *, reason: str = None):
category = (await self.bot.get_data(ctx.guild.id))[2]
category = ctx.guild.get_channel(category)
if category:
for channel in category.text_channels:
if checks.is_modmail_channel2(self.bot, channel):
msg = copy.copy(ctx.message)
msg.channel = channel
new_ctx = await self.bot.get_context(msg, cls=type(ctx))
await self.close_channel(new_ctx, reason)
try:
await ctx.send(embed=discord.Embed(
description="All channels are successfully closed.",
colour=self.bot.primary_colour,
))
except discord.HTTPException:
pass
async def send_mail(self, message, guild, to_send):
self.bot.total_messages += 1
guild = self.bot.get_guild(guild)
if guild is None:
return await message.channel.send(
embed=discord.Embed(description="The server was not found.", colour=self.bot.error_colour)
)
if guild.get_member(message.author.id) is False:
return await message.channel.send(
embed=discord.Embed(
description="You are not in that server, and the message is not sent.",
colour=self.bot.error_colour,
)
)
data = self.bot.get_data(guild.id)
category = guild.get_channel(data[2])
if category is None:
return await message.channel.send(
embed=discord.Embed(
description="A ModMail category is not found. The bot is not set up properly in the server.",
colour=self.bot.error_colour,
)
)
if data[9] and str(message.author.id) in data[9].split(","):
return await message.channel.send(
embed=discord.Embed(
description="That server has blacklisted you from sending a message there.",
colour=self.bot.error_colour,
)
)
channels = [
channel for channel in guild.text_channels if checks.is_modmail_channel2(self.bot, channel, message.author.id)
]
channel = None
new_ticket = False
if len(channels) > 0:
channel = channels[0]
if not channel:
self.bot.total_tickets += 1
try:
name = "".join(
l for l in message.author.name.lower() if l not in string.punctuation and l.isprintable()
)
if name:
name = name + f"-{message.author.discriminator}"
else:
name = message.author.id
channel = await category.create_text_channel(
name, topic=f"ModMail Channel {message.author.id} (Please do not change this)"
)
new_ticket = True
log_channel = guild.get_channel(data[4])
if log_channel:
try:
embed = discord.Embed(
title="New Ticket", colour=self.bot.user_colour, timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(
text=f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
await log_channel.send(embed=embed)
except discord.Forbidden:
pass
except discord.Forbidden:
return await message.channel.send(
embed=discord.Embed(
description="The bot is missing permissions to create a channel. Please contact an admin on "
"the server.",
colour=self.bot.error_colour,
)
)
except discord.HTTPException:
return await message.channel.send(
embed=discord.Embed(
description="A HTTPException error occurred. This is most likely because the server has "
"reached the maximum number of channels (500). Please join the support server "
"if you cannot figure out what went wrong.",
colour=self.bot.error_colour,
)
)
try:
if new_ticket is True:
self.guild = guild
prefix = self.bot.tools.get_guild_prefix(self.bot, self)
embed = discord.Embed(
title="New Ticket",
description="Type a message in this channel to reply. Messages starting with the server prefix "
f"`{prefix}` are ignored, and can be used for staff discussion. Use the command "
f"`{prefix}close [reason]` to close this ticket.",
colour=self.bot.primary_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(
text=f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
await channel.send(
content=f"<@&{data[8]}>" if data[8] and data[8] not in ["@here", "@everyone"] else data[8],
embed=embed,
)
if data[5]:
embed = discord.Embed(
title="Custom Greeting Message",
description=self.bot.tools.tag_format(data[5], message.author),
colour=self.bot.mod_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(text=f"{guild.name} | {guild.id}", icon_url=guild.icon_url)
await message.channel.send(embed=embed)
embed = discord.Embed(
title="Message Sent",
description=to_send,
colour=self.bot.user_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(text=f"{guild.name} | {guild.id}", icon_url=guild.icon_url)
files = []
for file in message.attachments:
saved_file = io.BytesIO()
await file.save(saved_file)
files.append(discord.File(saved_file, file.filename))
message2 = await message.channel.send(embed=embed, files=files)
embed.title = "Message Received"
embed.set_footer(
text=f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
for count, attachment in enumerate([attachment.url for attachment in message2.attachments], start=1):
embed.add_field(
name=f"Attachment {count}",
value=attachment,
inline=False
)
for file in files:
file.reset()
await channel.send(embed=embed, files=files)
except discord.Forbidden:
return await message.channel.send(
embed=discord.Embed(
description="No permission to send message in the channel. Please contact an admin on the server.",
colour=self.bot.error_colour,
)
)
async def select_guild(self, message, prefix, msg=None):
guilds = [guild for guild in self.bot.guilds if guild.get_member(message.author.id)]
guild_list = {}
for guild in guilds:
channels = [
channel for channel in guild.text_channels
if checks.is_modmail_channel2(self.bot, channel, message.author.id)
]
channel = None
if len(channels) > 0:
channel = channels[0]
if not channel:
guild_list[str(guild.id)] = (guild.name, False)
else:
guild_list[str(guild.id)] = (guild.name, True)
embeds = []
current_embed = None
for guild, value in guild_list.items():
if not current_embed:
current_embed = discord.Embed(
title="Select Server",
description="Select the server you want this message to be sent to.\n Tip: You can "
f"also use `{prefix}send <server ID> <message>`.",
colour=self.bot.primary_colour,
)
current_embed.set_footer(text="Use the reactions to flip pages.")
current_embed.add_field(
name=f"{len(current_embed.fields) + 1}: {value[0]}",
value=f"{'Create a new ticket.' if value[1] is False else 'Existing ticket.'}\nServer ID: {guild}",
)
if len(current_embed.fields) == 10:
embeds.append(current_embed)
current_embed = None
if current_embed:
embeds.append(current_embed)
if msg:
await msg.edit(embed=embeds[0])
else:
msg = await message.channel.send(embed=embeds[0])
reactions = [
"1⃣",
"2⃣",
"3⃣",
"4⃣",
"5⃣",
"6⃣",
"7⃣",
"8⃣",
"9⃣",
"🔟",
"◀",
"▶",
]
async def add_reactions(length):
await msg.add_reaction("◀")
await msg.add_reaction("▶")
for index in range(0, length):
await msg.add_reaction(reactions[index])
def reaction_check(reaction2, user2):
return str(reaction2) in reactions and user2.id == message.author.id and reaction2.message.id == msg.id
await add_reactions(len(embeds[0].fields))
page_index = 0
chosen = -1
try:
while chosen < 0:
reaction, _ = await self.bot.wait_for("reaction_add", check=reaction_check, timeout=60)
if str(reaction) == "◀":
if page_index != 0:
page_index = page_index - 1
await msg.edit(embed=embeds[page_index])
await add_reactions(len(embeds[page_index].fields))
elif str(reaction) == "▶":
if page_index + 1 < len(embeds):
page_index = page_index + 1
await msg.edit(embed=embeds[page_index])
if len(embeds[page_index].fields) != 10:
to_remove = reactions[len(embeds[page_index].fields) : -2]
msg = await msg.channel.fetch_message(msg.id)
for this_reaction in msg.reactions:
if str(this_reaction) in to_remove:
await this_reaction.remove(self.bot.user)
elif reactions.index(str(reaction)) >= 0:
chosen = reactions.index(str(reaction))
except asyncio.TimeoutError:
await self.remove_reactions(msg)
return await msg.edit(
embed=discord.Embed(description="Time out. You did not choose anything.", colour=self.bot.error_colour)
)
await msg.delete()
guild = embeds[page_index].fields[chosen].value.split()[-1]
await self.send_mail(message, int(guild), message.content)
async def select_guild(self, message, prefix, msg=None):
data = await self.bot.cogs["Communication"].handler(
"get_user_guilds", self.bot.cluster_count,
{"user_id": message.author.id})
guilds = []
for chunk in data:
guilds.extend(chunk)
guild_list = {}
for guild in guilds:
channels = [
channel for channel in guild["text_channels"]
if checks.is_modmail_channel2(self.bot, channel,
message.author.id, True)
]
channel = None
if len(channels) > 0:
channel = channels[0]
if not channel:
guild_list[str(guild["id"])] = (guild["name"], False)
else:
guild_list[str(guild["id"])] = (guild["name"], True)
embeds = []
current_embed = None
for guild, value in guild_list.items():
if not current_embed:
current_embed = discord.Embed(
title="Select Server",
description=
"Please select the server you want to send this message to. You can do so by reacting "
f"with the corresponding emote.",
colour=self.bot.primary_colour,
)
current_embed.set_footer(
text="Use the reactions to flip pages.")
current_embed.add_field(
name=f"{len(current_embed.fields) + 1}: {value[0]}",
value=
f"{'Create a new ticket.' if value[1] is False else 'Existing ticket.'}\nServer ID: {guild}",
)
if len(current_embed.fields) == 10:
embeds.append(current_embed)
current_embed = None
if current_embed:
embeds.append(current_embed)
if len(embeds) == 0:
await message.channel.send(
embed=discord.Embed(description="Oops... No server found.",
colour=self.bot.primary_colour))
return
if msg:
await msg.edit(embed=embeds[0])
else:
msg = await message.channel.send(embed=embeds[0])
reactions = [
"1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣", "🔟", "◀", "▶"
]
async def add_reactions(length):
await msg.add_reaction("◀")
await msg.add_reaction("▶")
for index in range(0, length):
await msg.add_reaction(reactions[index])
def reaction_check(reaction2, user2):
return str(
reaction2
) in reactions and user2.id == message.author.id and reaction2.message.id == msg.id
await add_reactions(len(embeds[0].fields))
page_index = 0
chosen = -1
try:
while chosen < 0:
reaction, _ = await self.bot.wait_for("reaction_add",
check=reaction_check,
timeout=60)
if str(reaction) == "◀":
if page_index != 0:
page_index = page_index - 1
await msg.edit(embed=embeds[page_index])
await add_reactions(len(embeds[page_index].fields))
elif str(reaction) == "▶":
if page_index + 1 < len(embeds):
page_index = page_index + 1
await msg.edit(embed=embeds[page_index])
if len(embeds[page_index].fields) != 10:
to_remove = reactions[len(embeds[page_index].fields
):-2]
msg = await msg.channel.fetch_message(msg.id)
for this_reaction in msg.reactions:
if str(this_reaction) in to_remove:
await this_reaction.remove(self.bot.user)
elif reactions.index(str(reaction)) >= 0:
chosen = reactions.index(str(reaction))
except asyncio.TimeoutError:
await self.remove_reactions(msg)
await msg.edit(embed=discord.Embed(
description="Time out. You did not choose anything.",
colour=self.bot.error_colour))
return
await msg.delete()
guild = embeds[page_index].fields[chosen].value.split()[-1]
await self.send_mail(message, int(guild), message.content)
async def send_mail(self, message, guild, to_send):
self.bot.stats_messages += 1
self.bot.prom.tickets_message_counter.inc()
guild = await self.bot.cogs["Communication"].handler(
"get_guild", 1, {"guild_id": guild})
if not guild:
await message.channel.send(
embed=discord.Embed(description="The server was not found.",
colour=self.bot.error_colour))
return
guild = guild[0]
member = await self.bot.cogs["Communication"].handler(
"get_guild_member",
1,
{
"guild_id": guild["id"],
"member_id": message.author.id
},
)
if not member:
await message.channel.send(embed=discord.Embed(
description=
"You are not in that server, and the message is not sent.",
colour=self.bot.error_colour,
))
return
data = await self.bot.get_data(guild["id"])
category = await self.bot.cogs["Communication"].handler(
"get_guild_channel",
1,
{
"guild_id": guild["id"],
"channel_id": data[2]
},
)
if not category:
await message.channel.send(embed=discord.Embed(
description=
"A ModMail category is not found. The bot is not set up properly in the server.",
colour=self.bot.error_colour,
))
return
category = category[0]
if message.author.id in data[9]:
await message.channel.send(embed=discord.Embed(
description=
"That server has blacklisted you from sending a message there.",
colour=self.bot.error_colour,
))
return
channels = [
channel for channel in guild["text_channels"]
if checks.is_modmail_channel2(self.bot, channel, message.author.id,
True)
]
channel = None
new_ticket = False
if len(channels) > 0:
channel = channels[0]
if not channel:
self.bot.stats_tickets += 1
self.bot.prom.tickets_counter.inc()
try:
name = "".join(
l for l in message.author.name.lower()
if l not in string.punctuation and l.isprintable())
if name:
name = name + f"-{message.author.discriminator}"
else:
name = message.author.id
channel = await self.bot.http.create_channel(
guild["id"],
0,
name=name,
parent_id=category["id"],
topic=
f"ModMail Channel {message.author.id} (Please do not change this)",
)
new_ticket = True
log_channel = await self.bot.cogs["Communication"].handler(
"get_guild_channel",
1,
{
"guild_id": guild["id"],
"channel_id": data[4]
},
)
if log_channel:
log_channel = log_channel[0]
try:
embed = discord.Embed(
title="New Ticket",
colour=self.bot.user_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(
text=
f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
await self.bot.http.send_message(log_channel["id"],
None,
embed=embed.to_dict())
except discord.Forbidden:
pass
except discord.Forbidden:
await message.channel.send(embed=discord.Embed(
description=
"The bot is missing permissions to create a channel. Please contact an admin on "
"the server.",
colour=self.bot.error_colour,
))
return
except discord.HTTPException:
await message.channel.send(embed=discord.Embed(
description=
"A HTTPException error occurred. This is most likely because the server has "
"reached the maximum number of channels (500). Please join the support server if you cannot "
"figure out what went wrong.",
colour=self.bot.error_colour,
))
return
try:
if new_ticket is True:
prefix = self.bot.tools.get_guild_prefix(self.bot, guild, True)
embed = discord.Embed(
title="New Ticket",
description=
"Type a message in this channel to reply. Messages starting with the server prefix "
f"`{prefix}` are ignored, and can be used for staff discussion. Use the command "
f"`{prefix}close [reason]` to close this ticket.",
colour=self.bot.primary_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(
text=
f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
roles = []
for role in data[8]:
if role == guild["default_role"]["id"]:
roles.append("@everyone")
elif role == -1:
roles.append("@here")
else:
roles.append(f"<@&{role}>")
await self.bot.http.send_message(channel["id"],
" ".join(roles),
embed=embed.to_dict())
if data[5]:
embed = discord.Embed(
title="Custom Greeting Message",
description=self.bot.tools.tag_format(
data[5], message.author),
colour=self.bot.mod_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(text=f"{guild['name']} | {guild['id']}",
icon_url=guild["icon_url"])
await message.channel.send(embed=embed)
embed = discord.Embed(
title="Message Sent",
description=to_send,
colour=self.bot.user_colour,
timestamp=datetime.datetime.utcnow(),
)
embed.set_footer(text=f"{guild['name']} | {guild['id']}",
icon_url=guild["icon_url"])
files = []
for file in message.attachments:
saved_file = io.BytesIO()
await file.save(saved_file)
files.append(discord.File(saved_file, file.filename))
message2 = await message.channel.send(embed=embed, files=files)
embed.title = "Message Received"
embed.set_footer(
text=
f"{message.author.name}#{message.author.discriminator} | {message.author.id}",
icon_url=message.author.avatar_url,
)
for count, attachment in enumerate(
[attachment.url for attachment in message2.attachments],
start=1):
embed.add_field(name=f"Attachment {count}",
value=attachment,
inline=False)
for file in files:
file.reset()
if files:
await self.bot.http.send_files(channel["id"],
embed=embed.to_dict(),
files=files)
else:
await self.bot.http.send_message(channel["id"],
None,
embed=embed.to_dict())
except discord.Forbidden:
try:
await message2.delete()
except NameError:
pass
await message.channel.send(embed=discord.Embed(
description=
"No permission to send message in the channel. Please contact an admin on the server.",
colour=self.bot.error_colour,
))
