SeaCogs/aurora/utilities/factory.py

509 lines
19 KiB
Python
Raw Normal View History

2024-02-28 10:58:57 -05:00
# pylint: disable=cyclic-import
2024-01-08 04:18:54 -05:00
from datetime import datetime, timedelta
from typing import Union
2024-01-08 04:18:54 -05:00
2024-05-06 16:47:21 -04:00
from discord import Color, Embed, Guild, Interaction, InteractionMessage, Member, Role, User
from redbot.core import commands
2024-05-06 16:47:21 -04:00
from redbot.core.utils.chat_formatting import bold, box, error, humanize_timedelta, warning
2024-01-08 04:18:54 -05:00
2024-05-06 17:23:59 -04:00
from aurora.models.moderation import Moderation
from aurora.models.partials import PartialUser
from aurora.utilities.config import config
2024-05-06 16:47:21 -04:00
from aurora.utilities.utils import get_bool_emoji, get_next_case_number, get_pagesize_str
2024-02-02 11:21:56 -05:00
async def message_factory(
color: Color,
guild: Guild,
reason: str,
moderation_type: str,
moderator: Union[Member, User] = None,
duration: timedelta = None,
response: InteractionMessage = None,
role: Role = None,
2024-02-02 11:21:56 -05:00
) -> Embed:
2024-01-08 04:18:54 -05:00
"""This function creates a message from set parameters, meant for contacting the moderated user.
Args:
color (Color): The color of the embed.
guild (Guild): The guild the moderation occurred in.
reason (str): The reason for the moderation.
moderation_type (str): The type of moderation.
moderator (Union[Member, User], optional): The moderator who performed the moderation. Defaults to None.
duration (timedelta, optional): The duration of the moderation. Defaults to None.
response (InteractionMessage, optional): The response message. Defaults to None.
role (Role, optional): The role that was added or removed. Defaults to None.
2024-01-08 04:18:54 -05:00
Returns:
embed: The message embed.
"""
2024-02-14 10:39:26 -05:00
if response is not None and moderation_type not in [
2024-02-02 11:21:56 -05:00
"kicked",
"banned",
"tempbanned",
"unbanned",
]:
2024-01-08 04:18:54 -05:00
guild_name = f"[{guild.name}]({response.jump_url})"
else:
guild_name = guild.name
title = moderation_type
2024-01-08 04:18:54 -05:00
if moderation_type in ["tempbanned", "muted"] and duration:
embed_duration = f" for {humanize_timedelta(timedelta=duration)}"
2024-01-08 04:18:54 -05:00
else:
embed_duration = ""
if moderation_type == "note":
embed_desc = "received a"
elif moderation_type == "addrole":
embed_desc = f"received the {role.name} role"
title = "Role Added"
moderation_type = ""
elif moderation_type == "removerole":
embed_desc = f"lost the {role.name} role"
title = "Role Removed"
moderation_type = ""
2024-01-08 04:18:54 -05:00
else:
embed_desc = "been"
2024-02-02 11:21:56 -05:00
embed = Embed(
title=str.title(title),
2024-02-02 11:21:56 -05:00
description=f"You have {embed_desc} {moderation_type}{embed_duration} in {guild_name}.",
color=color,
timestamp=datetime.now(),
)
2024-01-08 04:18:54 -05:00
if await config.guild(guild).show_moderator() and moderator is not None:
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Moderator", value=f"`{moderator.name} ({moderator.id})`", inline=False
)
2024-01-08 04:18:54 -05:00
2024-02-02 11:21:56 -05:00
embed.add_field(name="Reason", value=f"`{reason}`", inline=False)
2024-01-08 04:18:54 -05:00
if guild.icon.url is not None:
embed.set_author(name=guild.name, icon_url=guild.icon.url)
else:
embed.set_author(name=guild.name)
2024-02-02 11:21:56 -05:00
embed.set_footer(
text=f"Case #{await get_next_case_number(guild.id):,}",
icon_url="attachment://arrow.png",
2024-02-02 11:21:56 -05:00
)
2024-01-08 04:18:54 -05:00
return embed
2024-02-02 11:21:56 -05:00
async def log_factory(
interaction: Interaction, moderation: Moderation, resolved: bool = False
2024-02-02 11:21:56 -05:00
) -> Embed:
2024-01-08 04:18:54 -05:00
"""This function creates a log embed from set parameters, meant for moderation logging.
Args:
interaction (discord.Interaction): The interaction object.
moderation (aurora.models.Moderation): The moderation object.
2024-01-08 04:18:54 -05:00
resolved (bool, optional): Whether the case is resolved or not. Defaults to False.
"""
target = await moderation.get_target()
moderator = await moderation.get_moderator()
2024-01-08 04:18:54 -05:00
if resolved:
2024-02-02 11:21:56 -05:00
embed = Embed(
title=f"📕 Case #{moderation.id:,} Resolved",
2024-02-02 11:21:56 -05:00
color=await interaction.client.get_embed_color(interaction.channel),
)
2024-01-08 04:18:54 -05:00
resolved_by = await moderation.get_resolved_by()
embed.description = f"**Type:** {str.title(moderation.moderation_type)}\n**Target:** {target.name} ({target.id})\n**Moderator:** {moderator.name} ({moderator.id})\n**Timestamp:** <t:{moderation.unix_timestamp}> | <t:{moderation.unix_timestamp}:R>"
if moderation.duration is not None:
2024-02-02 11:21:56 -05:00
duration_embed = (
f"{humanize_timedelta(timedelta=moderation.duration)} | <t:{moderation.end_timestamp}:R>"
if not moderation.expired
else str(humanize_timedelta(timedelta=moderation.duration))
2024-02-02 11:21:56 -05:00
)
embed.description = (
embed.description
+ f"\n**Duration:** {duration_embed}\n**Expired:** {moderation.expired}"
2024-02-02 11:21:56 -05:00
)
embed.add_field(name="Reason", value=box(moderation.reason), inline=False)
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Resolve Reason",
value=f"Resolved by `{resolved_by.name}` ({resolved_by.id}) for:\n"
+ box(moderation.resolve_reason),
2024-02-02 11:21:56 -05:00
inline=False,
)
2024-01-08 04:18:54 -05:00
else:
2024-02-02 11:21:56 -05:00
embed = Embed(
title=f"📕 Case #{moderation.id:,}",
2024-02-02 11:21:56 -05:00
color=await interaction.client.get_embed_color(interaction.channel),
)
embed.description = f"**Type:** {str.title(moderation.type)}\n**Target:** {target.name} ({target.id})\n**Moderator:** {moderator.name} ({moderator.id})\n**Timestamp:** <t:{moderation.unix_timestamp}> | <t:{moderation.unix_timestamp}:R>"
if moderation.duration:
2024-02-02 11:21:56 -05:00
embed.description = (
embed.description
+ f"\n**Duration:** {humanize_timedelta(timedelta=moderation.duration)} | <t:{moderation.unix_timestamp}:R>"
2024-02-02 11:21:56 -05:00
)
embed.add_field(name="Reason", value=box(moderation.reason), inline=False)
2024-01-08 04:18:54 -05:00
return embed
2024-02-02 11:21:56 -05:00
async def case_factory(interaction: Interaction, moderation: Moderation) -> Embed:
2024-01-08 04:18:54 -05:00
"""This function creates a case embed from set parameters.
Args:
interaction (discord.Interaction): The interaction object.
moderation (aurora.models.Moderation): The moderation object.
2024-01-08 04:18:54 -05:00
"""
target = await moderation.get_target()
moderator = await moderation.get_moderator()
2024-01-08 04:18:54 -05:00
2024-02-02 11:21:56 -05:00
embed = Embed(
title=f"📕 Case #{moderation.id:,}",
2024-02-02 11:21:56 -05:00
color=await interaction.client.get_embed_color(interaction.channel),
)
embed.description = f"**Type:** {str.title(moderation.type)}\n**Target:** `{target.name}` ({target.id})\n**Moderator:** `{moderator.name}` ({moderator.id})\n**Resolved:** {moderation.resolved}\n**Timestamp:** <t:{moderation.unix_timestamp}> | <t:{moderation.unix_timestamp}:R>"
if moderation.duration:
2024-02-02 11:21:56 -05:00
duration_embed = (
f"{humanize_timedelta(timedelta=moderation.duration)} | <t:{moderation.unix_timestamp}:R>"
if moderation.expired is False
else str(humanize_timedelta(timedelta=moderation.duration))
2024-02-02 11:21:56 -05:00
)
embed.description += f"\n**Duration:** {duration_embed}\n**Expired:** {moderation.expired}"
2024-01-08 04:18:54 -05:00
2024-02-02 11:21:56 -05:00
embed.description += (
f"\n**Changes:** {len(moderation.changes) - 1}"
if moderation.changes
2024-02-02 11:21:56 -05:00
else "\n**Changes:** 0"
)
2024-01-08 04:18:54 -05:00
if moderation.role_id:
role = await moderation.get_role()
embed.description += f"\n**Role:** {role.name}"
if moderation.metadata:
if moderation.metadata["imported_from"]:
2024-02-02 11:21:56 -05:00
embed.description += (
f"\n**Imported From:** {moderation.metadata['imported_from']}"
)
if moderation.metadata["imported_timestamp"]:
embed.description += (
f"\n**Imported Timestamp:** <t:{moderation.metadata['imported_timestamp']}> | <t:{moderation.metadata['imported_timestamp']}:R>"
2024-02-02 11:21:56 -05:00
)
embed.add_field(name="Reason", value=box(moderation.reason), inline=False)
2024-02-02 11:21:56 -05:00
if moderation.resolved:
resolved_user = await moderation.get_resolved_by()
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Resolve Reason",
value=f"Resolved by `{resolved_user.name}` ({resolved_user.id}) for:\n{box(moderation.resolve_reason)}",
2024-02-02 11:21:56 -05:00
inline=False,
)
2024-01-08 04:18:54 -05:00
return embed
2024-02-02 11:21:56 -05:00
async def changes_factory(interaction: Interaction, moderation: Moderation) -> Embed:
2024-01-08 04:18:54 -05:00
"""This function creates a changes embed from set parameters.
Args:
interaction (discord.Interaction): The interaction object.
moderation (aurora.models.Moderation): The moderation object.
2024-01-08 04:18:54 -05:00
"""
2024-02-02 11:21:56 -05:00
embed = Embed(
title=f"📕 Case #{moderation.id:,} Changes",
2024-02-02 11:21:56 -05:00
color=await interaction.client.get_embed_color(interaction.channel),
)
2024-01-08 04:18:54 -05:00
memory_dict = {}
if moderation.changes:
for change in moderation.changes:
2024-05-04 21:15:39 -04:00
if change.user_id not in memory_dict:
memory_dict[str(change.user_id)] = await change.get_user()
2024-01-08 04:18:54 -05:00
user: PartialUser = memory_dict[str(change.user_id)]
2024-01-08 04:18:54 -05:00
2024-05-04 21:15:39 -04:00
timestamp = f"<t:{change.unix_timestamp}> | <t:{change.unix_timestamp}:R>"
2024-01-08 04:18:54 -05:00
2024-05-04 21:15:39 -04:00
if change.type == "ORIGINAL":
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Original",
value=f"**User:** `{user.name}` ({user.id})\n**Reason:** {change.reason}\n**Timestamp:** {timestamp}",
2024-02-02 11:21:56 -05:00
inline=False,
)
2024-05-04 21:15:39 -04:00
elif change.type == "EDIT":
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Edit",
value=f"**User:** `{user.name}` ({user.id})\n**Reason:** {change.reason}\n**Timestamp:** {timestamp}",
2024-02-02 11:21:56 -05:00
inline=False,
)
2024-05-04 21:15:39 -04:00
elif change.type == "RESOLVE":
2024-02-02 11:21:56 -05:00
embed.add_field(
name="Resolve",
value=f"**User:** `{user.name}` ({user.id})\n**Reason:** {change.reason}\n**Timestamp:** {timestamp}",
2024-02-02 11:21:56 -05:00
inline=False,
)
2024-01-08 04:18:54 -05:00
else:
embed.description = "*No changes have been made to this case.* 🙁"
return embed
2024-02-02 11:21:56 -05:00
2024-05-06 16:47:21 -04:00
async def evidenceformat_factory(moderation: Moderation) -> str:
2024-01-08 04:18:54 -05:00
"""This function creates a codeblock in evidence format from set parameters.
Args:
interaction (discord.Interaction): The interaction object.
moderation (aurora.models.Moderation): The moderation object.
2024-01-08 04:18:54 -05:00
"""
target = await moderation.get_target()
moderator = await moderation.get_moderator()
2024-01-08 04:18:54 -05:00
content = f"Case: {moderation.id:,} ({str.title(moderation.type)})\nTarget: {target.name} ({target.id})\nModerator: {moderator.name} ({moderator.id})"
2024-01-08 04:18:54 -05:00
if moderation.duration is not None:
content += f"\nDuration: {humanize_timedelta(timedelta=moderation.duration)}"
2024-01-08 04:18:54 -05:00
content += f"\nReason: {moderation.reason}"
2024-01-08 04:18:54 -05:00
2024-02-02 11:21:56 -05:00
return box(content, "prolog")
########################################################################################################################
### Configuration Embeds #
########################################################################################################################
2024-02-02 11:21:56 -05:00
2024-01-16 09:26:54 -05:00
async def _config(ctx: commands.Context) -> Embed:
"""Generates the core embed for configuration menus to use."""
2024-02-02 11:21:56 -05:00
e = Embed(title="Aurora Configuration Menu", color=await ctx.embed_color())
e.set_thumbnail(url=ctx.bot.user.display_avatar.url)
return e
2024-02-02 11:21:56 -05:00
2024-01-16 09:26:54 -05:00
async def overrides_embed(ctx: commands.Context) -> Embed:
"""Generates a configuration menu embed for a user's overrides."""
override_settings = {
"ephemeral": await config.user(ctx.author).history_ephemeral(),
"inline": await config.user(ctx.author).history_inline(),
"inline_pagesize": await config.user(ctx.author).history_inline_pagesize(),
"pagesize": await config.user(ctx.author).history_pagesize(),
2024-02-02 11:21:56 -05:00
"auto_evidenceformat": await config.user(ctx.author).auto_evidenceformat(),
}
override_str = [
2024-02-02 11:21:56 -05:00
"- "
+ bold("Auto Evidence Format: ")
+ get_bool_emoji(override_settings["auto_evidenceformat"]),
"- " + bold("Ephemeral: ") + get_bool_emoji(override_settings["ephemeral"]),
"- " + bold("History Inline: ") + get_bool_emoji(override_settings["inline"]),
"- "
+ bold("History Inline Pagesize: ")
+ get_pagesize_str(override_settings["inline_pagesize"]),
"- "
+ bold("History Pagesize: ")
+ get_pagesize_str(override_settings["pagesize"]),
]
2024-02-02 11:21:56 -05:00
override_str = "\n".join(override_str)
2024-01-16 09:26:54 -05:00
e = await _config(ctx)
e.title += ": User Overrides"
2024-02-02 11:21:56 -05:00
e.description = (
"""
Use the buttons below to manage your user overrides.
These settings will override the relevant guild settings.\n
2024-02-02 11:21:56 -05:00
"""
+ override_str
)
return e
2024-02-02 11:21:56 -05:00
2024-01-16 09:26:54 -05:00
async def guild_embed(ctx: commands.Context) -> Embed:
"""Generates a configuration menu field value for a guild's settings."""
guild_settings = {
"show_moderator": await config.guild(ctx.guild).show_moderator(),
2024-02-02 11:21:56 -05:00
"use_discord_permissions": await config.guild(
ctx.guild
).use_discord_permissions(),
"ignore_modlog": await config.guild(ctx.guild).ignore_modlog(),
"ignore_other_bots": await config.guild(ctx.guild).ignore_other_bots(),
"dm_users": await config.guild(ctx.guild).dm_users(),
"log_channel": await config.guild(ctx.guild).log_channel(),
"history_ephemeral": await config.guild(ctx.guild).history_ephemeral(),
"history_inline": await config.guild(ctx.guild).history_inline(),
"history_pagesize": await config.guild(ctx.guild).history_pagesize(),
2024-02-02 11:21:56 -05:00
"history_inline_pagesize": await config.guild(
ctx.guild
).history_inline_pagesize(),
"auto_evidenceformat": await config.guild(ctx.guild).auto_evidenceformat(),
"respect_hierarchy": await config.guild(ctx.guild).respect_hierarchy(),
}
2024-02-02 11:21:56 -05:00
channel = ctx.guild.get_channel(guild_settings["log_channel"])
if channel is None:
channel = warning("Not Set")
else:
channel = channel.mention
guild_str = [
2024-02-02 11:21:56 -05:00
"- "
+ bold("Show Moderator: ")
+ get_bool_emoji(guild_settings["show_moderator"]),
"- "
+ bold("Use Discord Permissions: ")
+ get_bool_emoji(guild_settings["use_discord_permissions"]),
"- "
+ bold("Respect Hierarchy: ")
+ get_bool_emoji(guild_settings["respect_hierarchy"]),
"- "
2024-02-02 11:21:56 -05:00
+ bold("Ignore Modlog: ")
+ get_bool_emoji(guild_settings["ignore_modlog"]),
"- "
+ bold("Ignore Other Bots: ")
+ get_bool_emoji(guild_settings["ignore_other_bots"]),
"- " + bold("DM Users: ") + get_bool_emoji(guild_settings["dm_users"]),
"- "
+ bold("Auto Evidence Format: ")
+ get_bool_emoji(guild_settings["auto_evidenceformat"]),
"- "
+ bold("Ephemeral: ")
+ get_bool_emoji(guild_settings["history_ephemeral"]),
"- "
+ bold("History Inline: ")
+ get_bool_emoji(guild_settings["history_inline"]),
"- "
+ bold("History Pagesize: ")
+ get_pagesize_str(guild_settings["history_pagesize"]),
"- "
+ bold("History Inline Pagesize: ")
+ get_pagesize_str(guild_settings["history_inline_pagesize"]),
"- " + bold("Log Channel: ") + channel,
]
2024-02-02 11:21:56 -05:00
guild_str = "\n".join(guild_str)
2024-01-16 09:26:54 -05:00
e = await _config(ctx)
e.title += ": Server Configuration"
2024-02-02 11:21:56 -05:00
e.description = (
"""
Use the buttons below to manage Aurora's server configuration.\n
2024-02-02 11:21:56 -05:00
"""
+ guild_str
)
return e
2024-02-02 11:21:56 -05:00
2024-01-16 09:26:54 -05:00
async def addrole_embed(ctx: commands.Context) -> Embed:
"""Generates a configuration menu field value for a guild's addrole whitelist."""
roles = []
async with config.guild(ctx.guild).addrole_whitelist() as whitelist:
for role in whitelist:
evalulated_role = ctx.guild.get_role(role) or error(f"`{role}` (Not Found)")
if isinstance(evalulated_role, Role):
roles.append({
"id": evalulated_role.id,
"mention": evalulated_role.mention,
"position": evalulated_role.position
})
else:
roles.append({
"id": role,
"mention": error(f"`{role}` (Not Found)"),
"position": 0
})
if roles:
roles = sorted(roles, key=lambda x: x["position"], reverse=True)
roles = [role["mention"] for role in roles]
whitelist_str = "\n".join(roles)
else:
whitelist_str = warning("No roles are on the addrole whitelist!")
2024-01-16 09:26:54 -05:00
e = await _config(ctx)
e.title += ": Addrole Whitelist"
2024-02-02 11:21:56 -05:00
e.description = (
"Use the select menu below to manage this guild's addrole whitelist."
)
if len(whitelist_str) > 4000 and len(whitelist_str) < 5000:
lines = whitelist_str.split("\n")
chunks = []
chunk = ""
for line in lines:
if len(chunk) + len(line) > 1024:
chunks.append(chunk)
chunk = line
else:
2024-02-02 11:21:56 -05:00
chunk += "\n" + line if chunk else line
chunks.append(chunk)
for chunk in chunks:
e.add_field(name="", value=chunk)
else:
e.description += "\n\n" + whitelist_str
return e
2024-02-02 11:21:56 -05:00
2024-01-16 09:26:54 -05:00
async def immune_embed(ctx: commands.Context) -> Embed:
"""Generates a configuration menu embed for a guild's immune roles."""
roles = []
async with config.guild(ctx.guild).immune_roles() as immune_roles:
for role in immune_roles:
evalulated_role = ctx.guild.get_role(role) or error(f"`{role}` (Not Found)")
if isinstance(evalulated_role, Role):
roles.append({
"id": evalulated_role.id,
"mention": evalulated_role.mention,
"position": evalulated_role.position
})
else:
roles.append({
"id": role,
"mention": error(f"`{role}` (Not Found)"),
"position": 0
})
if roles:
roles = sorted(roles, key=lambda x: x["position"], reverse=True)
roles = [role["mention"] for role in roles]
immune_str = "\n".join(roles)
else:
immune_str = warning("No roles are set as immune roles!")
2024-01-16 09:26:54 -05:00
e = await _config(ctx)
e.title += ": Immune Roles"
e.description = "Use the select menu below to manage this guild's immune roles."
if len(immune_str) > 4000 and len(immune_str) < 5000:
2024-02-02 11:21:56 -05:00
lines = immune_str.split("\n")
chunks = []
chunk = ""
for line in lines:
if len(chunk) + len(line) > 1024:
chunks.append(chunk)
chunk = line
else:
2024-02-02 11:21:56 -05:00
chunk += "\n" + line if chunk else line
chunks.append(chunk)
for chunk in chunks:
e.add_field(name="", value=chunk)
else:
2024-02-02 11:21:56 -05:00
e.description += "\n\n" + immune_str
return e