forked from cswimr/SeaCogs
225 lines
13 KiB
Python
225 lines
13 KiB
Python
# pylint: disable=cyclic-import
|
|
|
|
from typing import Union
|
|
from datetime import datetime, timedelta
|
|
|
|
import humanize
|
|
from discord import Color, Embed, Guild, Interaction, InteractionMessage, User, Member
|
|
from redbot.core.utils.chat_formatting import box
|
|
|
|
from .config import config
|
|
from .utils import fetch_channel_dict, fetch_user_dict, get_next_case_number
|
|
|
|
|
|
async def message_factory(color: Color, guild: Guild, reason: str, moderation_type: str, moderator: Union[Member, User] = None, duration: timedelta = None, response: InteractionMessage = None) -> Embed:
|
|
"""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.
|
|
|
|
Returns:
|
|
embed: The message embed.
|
|
"""
|
|
if response is not None and not moderation_type in ["kicked", "banned", "tempbanned", "unbanned"]:
|
|
guild_name = f"[{guild.name}]({response.jump_url})"
|
|
else:
|
|
guild_name = guild.name
|
|
|
|
if moderation_type in ["tempbanned", "muted"] and duration:
|
|
embed_duration = f" for {humanize.precisedelta(duration)}"
|
|
else:
|
|
embed_duration = ""
|
|
|
|
if moderation_type == "note":
|
|
embed_desc = "received a"
|
|
else:
|
|
embed_desc = "been"
|
|
|
|
embed = Embed(title=str.title(moderation_type), description=f"You have {embed_desc} {moderation_type}{embed_duration} in {guild_name}.", color=color, timestamp=datetime.now())
|
|
|
|
if await config.guild(guild).show_moderator() and moderator is not None:
|
|
embed.add_field(name='Moderator', value=f"`{moderator.name} ({moderator.id})`", inline=False)
|
|
|
|
embed.add_field(name='Reason', value=f"`{reason}`", inline=False)
|
|
|
|
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)
|
|
|
|
embed.set_footer(text=f"Case #{await get_next_case_number(guild.id):,}", icon_url="https://cdn.discordapp.com/attachments/1070822161389994054/1159469476773904414/arrow-right-circle-icon-512x512-2p1e2aaw.png?ex=65312319&is=651eae19&hm=3cebdd28e805c13a79ec48ef87c32ca532ffa6b9ede2e48d0cf8e5e81f3a6818&")
|
|
|
|
return embed
|
|
|
|
async def log_factory(interaction: Interaction, case_dict: dict, resolved: bool = False) -> Embed:
|
|
"""This function creates a log embed from set parameters, meant for moderation logging.
|
|
|
|
Args:
|
|
interaction (Interaction): The interaction object.
|
|
case_dict (dict): The case dictionary.
|
|
resolved (bool, optional): Whether the case is resolved or not. Defaults to False.
|
|
"""
|
|
if resolved:
|
|
if case_dict['target_type'] == 'USER':
|
|
target_user = await fetch_user_dict(interaction, case_dict['target_id'])
|
|
target_name = f"`{target_user['name']}`" if target_user['discriminator'] == "0" else f"`{target_user['name']}#{target_user['discriminator']}`"
|
|
elif case_dict['target_type'] == 'CHANNEL':
|
|
target_user = await fetch_channel_dict(interaction, case_dict['target_id'])
|
|
if target_user['mention']:
|
|
target_name = f"{target_user['mention']}"
|
|
else:
|
|
target_name = f"`{target_user['name']}`"
|
|
|
|
moderator_user = await fetch_user_dict(interaction, case_dict['moderator_id'])
|
|
moderator_name = f"`{moderator_user['name']}`" if moderator_user['discriminator'] == "0" else f"`{moderator_user['name']}#{moderator_user['discriminator']}`"
|
|
|
|
embed = Embed(title=f"📕 Case #{case_dict['moderation_id']:,} Resolved", color=await interaction.client.get_embed_color(interaction.channel))
|
|
|
|
embed.description = f"**Type:** {str.title(case_dict['moderation_type'])}\n**Target:** {target_name} ({target_user['id']})\n**Moderator:** {moderator_name} ({moderator_user['id']})\n**Timestamp:** <t:{case_dict['timestamp']}> | <t:{case_dict['timestamp']}:R>"
|
|
|
|
if case_dict['duration'] != 'NULL':
|
|
td = timedelta(**{unit: int(val) for unit, val in zip(["hours", "minutes", "seconds"], case_dict["duration"].split(":"))})
|
|
duration_embed = f"{humanize.precisedelta(td)} | <t:{case_dict['end_timestamp']}:R>" if case_dict["expired"] == '0' else str(humanize.precisedelta(td))
|
|
embed.description = embed.description + f"\n**Duration:** {duration_embed}\n**Expired:** {bool(case_dict['expired'])}"
|
|
|
|
embed.add_field(name='Reason', value=box(case_dict['reason']), inline=False)
|
|
|
|
resolved_user = await fetch_user_dict(interaction, case_dict['resolved_by'])
|
|
resolved_name = resolved_user['name'] if resolved_user['discriminator'] == "0" else f"{resolved_user['name']}#{resolved_user['discriminator']}"
|
|
embed.add_field(name='Resolve Reason', value=f"Resolved by `{resolved_name}` ({resolved_user['id']}) for:\n" + box(case_dict['resolve_reason']), inline=False)
|
|
else:
|
|
if case_dict['target_type'] == 'USER':
|
|
target_user = await fetch_user_dict(interaction, case_dict['target_id'])
|
|
target_name = f"`{target_user['name']}`" if target_user['discriminator'] == "0" else f"`{target_user['name']}#{target_user['discriminator']}`"
|
|
elif case_dict['target_type'] == 'CHANNEL':
|
|
target_user = await fetch_channel_dict(interaction, case_dict['target_id'])
|
|
if target_user['mention']:
|
|
target_name = target_user['mention']
|
|
else:
|
|
target_name = f"`{target_user['name']}`"
|
|
|
|
moderator_user = await fetch_user_dict(interaction, case_dict['moderator_id'])
|
|
moderator_name = f"`{moderator_user['name']}`" if moderator_user['discriminator'] == "0" else f"`{moderator_user['name']}#{moderator_user['discriminator']}`"
|
|
|
|
embed = Embed(title=f"📕 Case #{case_dict['moderation_id']:,}", color=await interaction.client.get_embed_color(interaction.channel))
|
|
embed.description = f"**Type:** {str.title(case_dict['moderation_type'])}\n**Target:** {target_name} ({target_user['id']})\n**Moderator:** {moderator_name} ({moderator_user['id']})\n**Timestamp:** <t:{case_dict['timestamp']}> | <t:{case_dict['timestamp']}:R>"
|
|
|
|
if case_dict['duration'] != 'NULL':
|
|
td = timedelta(**{unit: int(val) for unit, val in zip(["hours", "minutes", "seconds"], case_dict["duration"].split(":"))})
|
|
embed.description = embed.description + f"\n**Duration:** {humanize.precisedelta(td)} | <t:{case_dict['end_timestamp']}:R>"
|
|
|
|
embed.add_field(name='Reason', value=box(case_dict['reason']), inline=False)
|
|
return embed
|
|
|
|
async def case_factory(interaction: Interaction, case_dict: dict) -> Embed:
|
|
"""This function creates a case embed from set parameters.
|
|
|
|
Args:
|
|
interaction (Interaction): The interaction object.
|
|
case_dict (dict): The case dictionary.
|
|
"""
|
|
if case_dict['target_type'] == 'USER':
|
|
target_user = await fetch_user_dict(interaction, case_dict['target_id'])
|
|
target_name = f"`{target_user['name']}`" if target_user['discriminator'] == "0" else f"`{target_user['name']}#{target_user['discriminator']}`"
|
|
elif case_dict['target_type'] == 'CHANNEL':
|
|
target_user = await fetch_channel_dict(interaction, case_dict['target_id'])
|
|
if target_user['mention']:
|
|
target_name = f"{target_user['mention']}"
|
|
else:
|
|
target_name = f"`{target_user['name']}`"
|
|
|
|
moderator_user = await fetch_user_dict(interaction, case_dict['moderator_id'])
|
|
moderator_name = f"`{moderator_user['name']}`" if moderator_user['discriminator'] == "0" else f"`{moderator_user['name']}#{moderator_user['discriminator']}`"
|
|
|
|
embed = Embed(title=f"📕 Case #{case_dict['moderation_id']:,}", color=await interaction.client.get_embed_color(interaction.channel))
|
|
embed.description = f"**Type:** {str.title(case_dict['moderation_type'])}\n**Target:** {target_name} ({target_user['id']})\n**Moderator:** {moderator_name} ({moderator_user['id']})\n**Resolved:** {bool(case_dict['resolved'])}\n**Timestamp:** <t:{case_dict['timestamp']}> | <t:{case_dict['timestamp']}:R>"
|
|
|
|
if case_dict['duration'] != 'NULL':
|
|
td = timedelta(**{unit: int(val) for unit, val in zip(["hours", "minutes", "seconds"], case_dict["duration"].split(":"))})
|
|
duration_embed = f"{humanize.precisedelta(td)} | <t:{case_dict['end_timestamp']}:R>" if bool(case_dict['expired']) is False else str(humanize.precisedelta(td))
|
|
embed.description += f"\n**Duration:** {duration_embed}\n**Expired:** {bool(case_dict['expired'])}"
|
|
|
|
embed.description += f"\n**Changes:** {len(case_dict['changes']) - 1}" if case_dict['changes'] else "\n**Changes:** 0"
|
|
|
|
if case_dict['metadata']:
|
|
if case_dict['metadata']['imported_from']:
|
|
embed.description += f"\n**Imported From:** {case_dict['metadata']['imported_from']}"
|
|
|
|
embed.add_field(name='Reason', value=box(case_dict['reason']), inline=False)
|
|
|
|
if case_dict['resolved'] == 1:
|
|
resolved_user = await fetch_user_dict(interaction, case_dict['resolved_by'])
|
|
resolved_name = f"`{resolved_user['name']}`" if resolved_user['discriminator'] == "0" else f"`{resolved_user['name']}#{resolved_user['discriminator']}`"
|
|
embed.add_field(name='Resolve Reason', value=f"Resolved by {resolved_name} ({resolved_user['id']}) for:\n{box(case_dict['resolve_reason'])}", inline=False)
|
|
|
|
return embed
|
|
|
|
async def changes_factory(interaction: Interaction, case_dict: dict) -> Embed:
|
|
"""This function creates a changes embed from set parameters.
|
|
|
|
Args:
|
|
interaction (Interaction): The interaction object.
|
|
case_dict (dict): The case dictionary.
|
|
"""
|
|
embed = Embed(title=f"📕 Case #{case_dict['moderation_id']:,} Changes", color=await interaction.client.get_embed_color(interaction.channel))
|
|
|
|
memory_dict = {}
|
|
|
|
if case_dict['changes']:
|
|
for change in case_dict['changes']:
|
|
if change['user_id'] not in memory_dict:
|
|
memory_dict[str(change['user_id'])] = await fetch_user_dict(interaction, change['user_id'])
|
|
|
|
user = memory_dict[str(change['user_id'])]
|
|
name = user['name'] if user['discriminator'] == "0" else f"{user['name']}#{user['discriminator']}"
|
|
|
|
timestamp = f"<t:{change['timestamp']}> | <t:{change['timestamp']}:R>"
|
|
|
|
if change['type'] == 'ORIGINAL':
|
|
embed.add_field(name='Original', value=f"**User:** `{name}` ({user['id']})\n**Reason:** {change['reason']}\n**Timestamp:** {timestamp}", inline=False)
|
|
|
|
elif change['type'] == 'EDIT':
|
|
embed.add_field(name='Edit', value=f"**User:** `{name}` ({user['id']})\n**Reason:** {change['reason']}\n**Timestamp:** {timestamp}", inline=False)
|
|
|
|
elif change['type'] == 'RESOLVE':
|
|
embed.add_field(name='Resolve', value=f"**User:** `{name}` ({user['id']})\n**Reason:** {change['reason']}\n**Timestamp:** {timestamp}", inline=False)
|
|
|
|
else:
|
|
embed.description = "*No changes have been made to this case.* 🙁"
|
|
|
|
return embed
|
|
|
|
async def evidenceformat_factory(interaction: Interaction, case_dict: dict) -> str:
|
|
"""This function creates a codeblock in evidence format from set parameters.
|
|
|
|
Args:
|
|
interaction (Interaction): The interaction object.
|
|
case_dict (dict): The case dictionary.
|
|
"""
|
|
if case_dict['target_type'] == 'USER':
|
|
target_user = await fetch_user_dict(interaction, case_dict['target_id'])
|
|
target_name = target_user['name'] if target_user['discriminator'] == "0" else f"{target_user['name']}#{target_user['discriminator']}"
|
|
|
|
elif case_dict['target_type'] == 'CHANNEL':
|
|
target_user = await fetch_channel_dict(interaction, case_dict['target_id'])
|
|
target_name = target_user['name']
|
|
|
|
moderator_user = await fetch_user_dict(interaction, case_dict['moderator_id'])
|
|
moderator_name = moderator_user['name'] if moderator_user['discriminator'] == "0" else f"{moderator_user['name']}#{moderator_user['discriminator']}"
|
|
|
|
content = f"Case: {case_dict['moderation_id']:,} ({str.title(case_dict['moderation_type'])})\nTarget: {target_name} ({target_user['id']})\nModerator: {moderator_name} ({moderator_user['id']})"
|
|
|
|
if case_dict['duration'] != 'NULL':
|
|
hours, minutes, seconds = map(int, case_dict['duration'].split(':'))
|
|
td = timedelta(hours=hours, minutes=minutes, seconds=seconds)
|
|
content += f"\nDuration: {humanize.precisedelta(td)}"
|
|
|
|
content += f"\nReason: {case_dict['reason']}"
|
|
|
|
return box(content, 'prolog')
|