mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
905 lines
38 KiB
Python
905 lines
38 KiB
Python
# Ensō~Chan - A Multi Purpose Discord Bot That Has Everything Your Server Needs!
|
|
# Copyright (C) 2020 Hamothy
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import datetime
|
|
from datetime import timedelta
|
|
from typing import Optional
|
|
|
|
import discord
|
|
from discord import Member, Embed, DMChannel, NotFound
|
|
from discord.ext.commands import command, guild_only, has_guild_permissions, bot_has_guild_permissions, Greedy, \
|
|
cooldown, BucketType, Cog
|
|
|
|
|
|
# TODO: CREATE A BITARRAY SO THAT THE MODLOG EVENTS ARE TOGGLEABLE
|
|
# TODO: MAKE SURE THAT THE BITARRAY IS ONLY IMPLEMENTED AFTER ALL EVENTS ARE CODED
|
|
|
|
async def send_to_modlogs(self, ctx, target, reason, action):
|
|
"""
|
|
Function to send the moderation actions to modlogs channel
|
|
|
|
On top of the normal logging, this function sends another embed
|
|
if the user has used the inbuilt mute command which allows for a
|
|
reason to be given
|
|
|
|
"""
|
|
|
|
# Get the channel of the modlog within the guild
|
|
if modlog := self.bot.get_modlog_for_guild(ctx.guild.id):
|
|
channel = ctx.guild.get_channel(modlog)
|
|
|
|
desc = f"**Member -->{target.mention} | {target}\nID -->** {target.id}" \
|
|
f"\n\n**Auctioned By --> {ctx.author.mention} | {ctx.author}\nID -->** {ctx.author.id}"
|
|
embed = Embed(title=f"Member {action}",
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_thumbnail(url=target.avatar_url)
|
|
embed.add_field(name="Reason", value=reason, inline=False)
|
|
embed.set_footer(text=f"Member {action}")
|
|
|
|
await channel.send(embed=embed)
|
|
|
|
|
|
async def check(ctx, members):
|
|
"""
|
|
Check Function
|
|
|
|
- Checks if all arguments are given
|
|
- Checks if user mentions themselves
|
|
|
|
Error will be thrown in these two cases
|
|
"""
|
|
|
|
if not len(members):
|
|
desc = f"Not Correct Syntax!\nUse **{ctx.prefix}help** to find how to use **{ctx.command}**"
|
|
await ctx.bot.generate_embed(ctx, desc=desc)
|
|
return True
|
|
|
|
elif ctx.author in members:
|
|
await ctx.bot.generate_embed(ctx, desc=f"**{ctx.bot.cross} Forbidden Action {ctx.bot.cross}**")
|
|
return True
|
|
|
|
|
|
async def ummute_members(self, ctx, targets, reason):
|
|
"""
|
|
|
|
Method to allow members to be unmuted
|
|
|
|
2 embeds will be sent, one to the channel that the user is in
|
|
And if the user has the modlogs channel setup, an embed will be logged there
|
|
|
|
"""
|
|
|
|
for target in targets:
|
|
if (ctx.guild.me.top_role.position > target.top_role.position
|
|
and not target.guild_permissions.administrator):
|
|
|
|
# Get the roles of the user
|
|
result = await self.bot.check_cache(target.id, ctx.guild.id)
|
|
|
|
# Get muted roles of the user from cache/database and give them back
|
|
role_ids = result["muted_roles"]
|
|
roles = [ctx.guild.get_role(int(id_)) for id_ in role_ids.split(", ") if len(id_)]
|
|
|
|
# Clear all the roles of the user
|
|
await self.bot.clear_roles(member=target)
|
|
|
|
await target.edit(roles=roles)
|
|
|
|
# Send confirmation to the channel that the user is in
|
|
await self.bot.generate_embed(ctx, desc=f"{ctx.bot.tick} **{target}** Was Unmuted! {ctx.bot.tick}")
|
|
|
|
await send_to_modlogs(self, ctx, target, reason, action="Unmuted")
|
|
|
|
# Send error message if the User could not be muted
|
|
else:
|
|
desc = f"**{target.mention} Could Not Be Unmuted!**"
|
|
await self.bot.generate_embed(ctx, desc=desc)
|
|
|
|
|
|
async def mute_members(self, ctx, targets, reason, muted):
|
|
"""
|
|
|
|
Method to allow members to be muted
|
|
|
|
2 embeds will be sent, one to the channel that the user is in
|
|
And if the user has the modlogs channel setup, an embed will be logged there
|
|
|
|
"""
|
|
|
|
for target in targets:
|
|
|
|
# When user is already muted, send error message
|
|
if muted in target.roles:
|
|
await self.bot.generate_embed(ctx, desc=f"**{ctx.bot.cross} User Is Already Muted! {ctx.bot.cross}**")
|
|
|
|
else:
|
|
|
|
# Store managed roles into the list of roles to be added to the user (including muted)
|
|
roles = [role for role in target.roles if role.managed]
|
|
roles.append(muted)
|
|
|
|
if (ctx.guild.me.top_role.position > target.top_role.position
|
|
and not target.guild_permissions.administrator):
|
|
|
|
# Store the current roles of the user within database
|
|
await self.bot.store_roles(target=target, ctx=ctx, member=target)
|
|
# Give the user the muted role (and any integration roles if need be)
|
|
await target.edit(roles=roles, reason=reason)
|
|
|
|
# Send confirmation to the channel that the user is in
|
|
embed = Embed(description=f"{ctx.bot.tick} **{target}** Was Muted! {ctx.bot.tick}",
|
|
colour=self.bot.admin_colour)
|
|
|
|
if self.bot.get_roles_persist(ctx.guild.id) == 0:
|
|
embed.add_field(name="**WARNING: ROLE PERSIST NOT ENABLED**",
|
|
value="The bot **will not give** the roles back to the user if they leave the server."
|
|
"\nAllowing the user to bypass the Mute by leaving and rejoining."
|
|
f"\nPlease enable Role Persist by doing **{ctx.prefix}rolepersist enable**",
|
|
inline=True)
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
await send_to_modlogs(self, ctx, target, reason, action="Muted")
|
|
|
|
# Send error message if the User could not be muted
|
|
else:
|
|
await self.bot.generate_embed(ctx, desc=f"**{target.mention} Could Not Be Muted!**")
|
|
|
|
|
|
async def ban_members(self, ctx, targets, reason):
|
|
"""
|
|
|
|
Method to allow members to be banned
|
|
|
|
2 embeds will be sent, one to the channel that the user is in
|
|
And if the user has the modlogs channel setup, an embed will be logged there
|
|
|
|
"""
|
|
|
|
for target in targets:
|
|
if (ctx.guild.me.top_role.position > target.top_role.position
|
|
and not target.guild_permissions.administrator):
|
|
|
|
await target.ban(reason=reason)
|
|
|
|
await self.bot.generate_embed(ctx, desc=f"{ctx.bot.tick} **{target}** Was Banned! {ctx.bot.tick}")
|
|
|
|
await send_to_modlogs(self, ctx, target, reason, action="Banned")
|
|
|
|
# Send error message if the User could not be banned
|
|
else:
|
|
await self.bot.generate_embed(ctx, desc=f"**{target.mention} Could Not Be Banned!**")
|
|
|
|
|
|
async def unban_members(self, ctx, targets, reason):
|
|
"""
|
|
|
|
Method to allow members to be unbanned
|
|
|
|
2 embeds will be sent, one to the channel that the user is in
|
|
And if the user has the modlogs channel setup, an embed will be logged there
|
|
|
|
"""
|
|
|
|
# Get the list of banned users from the server
|
|
bans = await ctx.guild.bans()
|
|
ban_ids = list(map(lambda m: m.user.id, bans))
|
|
|
|
for target in targets:
|
|
if target not in ban_ids:
|
|
await self.bot.generate_embed(ctx, desc=f"{ctx.bot.cross} **Member Is Not In Ban's List!** {ctx.bot.cross}")
|
|
|
|
else:
|
|
# Get the member and unban them
|
|
user = await self.bot.fetch_user(target)
|
|
await ctx.guild.unban(user, reason=reason)
|
|
|
|
# Send confirmation to the channel that the user is in
|
|
await self.bot.generate_embed(ctx, desc=f"{ctx.bot.tick} **{user}** Was Unbanned! {ctx.bot.tick}")
|
|
|
|
await send_to_modlogs(self, ctx, user, reason, action="Unbanned")
|
|
|
|
|
|
async def kick_members(self, ctx, targets, reason):
|
|
"""
|
|
|
|
Method to allow the kick member log to be sent to the modlog channel
|
|
|
|
2 embeds will be sent, one to the channel that the user is in
|
|
And if the user has the modlogs channel setup, an embed will be logged there
|
|
|
|
"""
|
|
|
|
for target in targets:
|
|
if (ctx.guild.me.top_role.position > target.top_role.position
|
|
and not target.guild_permissions.administrator):
|
|
|
|
await target.kick(reason=reason)
|
|
|
|
await self.bot.generate_embed(ctx, desc=f"{ctx.bot.tick} **{target}** Was Kicked! {ctx.bot.tick}")
|
|
|
|
await send_to_modlogs(self, ctx, target, reason, action="Kicked")
|
|
|
|
# Send error message if the User could not be kicked
|
|
else:
|
|
await self.bot.generate_embed(ctx, desc=f"**{target.mention} Could Not Be Kicked!**")
|
|
|
|
|
|
class Moderation(Cog):
|
|
"""Moderation Commands! (Kick/Ban/Mute etc)"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@Cog.listener()
|
|
async def on_ready(self):
|
|
print(f"{self.__class__.__name__} Cog has been loaded\n-----")
|
|
|
|
@command(name="kick", usage="`<member>...` `[reason]`")
|
|
@guild_only()
|
|
@has_guild_permissions(kick_members=True)
|
|
@bot_has_guild_permissions(kick_members=True)
|
|
@cooldown(1, 1, BucketType.user)
|
|
async def kick_member(self, ctx, members: Greedy[Member], *, reason: Optional[str] = "No Reason Given"):
|
|
"""
|
|
Kick Member(s) from Server
|
|
Multiple Members can be Kicked at Once
|
|
"""
|
|
|
|
if not await check(ctx, members):
|
|
with ctx.typing():
|
|
# Send embed of the kicked member
|
|
await kick_members(self, ctx, members, reason)
|
|
|
|
@command(name="mute", usage="`<member>...` `[reason]`")
|
|
@has_guild_permissions(manage_roles=True)
|
|
@bot_has_guild_permissions(manage_roles=True)
|
|
async def mute(self, ctx, members: Greedy[Member], *, reason: Optional[str] = "No Reason Given"):
|
|
"""
|
|
Mute Member(s) from Server
|
|
Multiple Members can be Muted At Once
|
|
"""
|
|
|
|
if not await check(ctx, members):
|
|
with ctx.typing():
|
|
|
|
# Get muted role from the server
|
|
role = discord.utils.get(ctx.guild.roles, name="Muted")
|
|
|
|
# Create muted role when no muted role exists and mute member(s)
|
|
if role is None:
|
|
muted = await ctx.guild.create_role(name="Muted")
|
|
for channel in ctx.guild.channels:
|
|
await channel.set_permissions(muted, send_messages=False)
|
|
|
|
await mute_members(self, ctx, members, reason, muted)
|
|
|
|
else:
|
|
await mute_members(self, ctx, members, reason, role)
|
|
|
|
@command(name="unmute", usage="`<member>...` `[reason]`")
|
|
@has_guild_permissions(manage_roles=True)
|
|
@bot_has_guild_permissions(manage_roles=True)
|
|
async def unmute(self, ctx, members: Greedy[Member], *, reason: Optional[str] = "No Reason Given"):
|
|
"""
|
|
Unmute Member(s) from Server
|
|
Multiple Members can be unmuted at once
|
|
"""
|
|
unmute = False
|
|
|
|
if not await check(ctx, members):
|
|
with ctx.typing():
|
|
role = discord.utils.get(ctx.guild.roles, name="Muted")
|
|
|
|
if role is None:
|
|
embed = Embed(description="**❌ No Muted Role Was Found! ❌**",
|
|
colour=self.bot.admin_colour)
|
|
await ctx.send(embed=embed)
|
|
|
|
else:
|
|
|
|
for member in members:
|
|
if role in member.roles:
|
|
await ummute_members(self, ctx, members, reason)
|
|
unmute = True
|
|
|
|
if role not in member.roles and unmute is False:
|
|
embed = Embed(description=f"**❌ {member.mention} Is Not Muted! ❌**",
|
|
colour=self.bot.admin_colour)
|
|
await ctx.send(embed=embed)
|
|
|
|
@command(name="ban", usage="`<member>...` `[reason]`")
|
|
@guild_only()
|
|
@has_guild_permissions(ban_members=True)
|
|
@bot_has_guild_permissions(ban_members=True)
|
|
@cooldown(1, 1, BucketType.user)
|
|
async def ban(self, ctx, members: Greedy[Member], *, reason: Optional[str] = "No Reason Given"):
|
|
"""
|
|
Ban Member(s) from Server
|
|
Multiple Members can be banned at once
|
|
"""
|
|
|
|
if not await check(ctx, members):
|
|
with ctx.typing():
|
|
await ban_members(self, ctx, members, reason)
|
|
|
|
@command(name="forceban", aliases=["powerban", "ultraban"], usage="`<member>...` `[reason]`")
|
|
@guild_only()
|
|
@has_guild_permissions(ban_members=True)
|
|
@bot_has_guild_permissions(ban_members=True)
|
|
@cooldown(1, 1, BucketType.user)
|
|
async def force_ban(self, ctx, users: Greedy[int], *, reason: Optional[str] = "No Reason Given"):
|
|
"""Ban User(s) from Server (MUST PROVIDE ID)"""
|
|
|
|
if not await check(ctx, users):
|
|
# Get the list of banned users from the server
|
|
bans = await ctx.guild.bans()
|
|
ban_ids = list(map(lambda m: m.user.id, bans))
|
|
|
|
# Power ban users from guilds without them being in there
|
|
for user in users:
|
|
if user in ban_ids:
|
|
await self.bot.generate_embed(ctx, desc="❌ **Member Is Already Banned!** ❌")
|
|
else:
|
|
await ctx.guild.ban(discord.Object(id=user))
|
|
target = await self.bot.fetch_user(user)
|
|
# Send confirmation to the channel that the user is in
|
|
await self.bot.generate_embed(ctx, desc=f"✅ **{target}** Was Power Banned! ✅")
|
|
|
|
await send_to_modlogs(self, ctx, target, reason, action="Power Banned")
|
|
|
|
@command(name="unban", usage="`<member>...` `[reason]`")
|
|
@guild_only()
|
|
@has_guild_permissions(ban_members=True)
|
|
@bot_has_guild_permissions(ban_members=True)
|
|
@cooldown(1, 1, BucketType.user)
|
|
async def unban(self, ctx, members: Greedy[int], *, reason: Optional[str] = "No Reason Given"):
|
|
"""
|
|
Unban Member(s) from Server
|
|
Multiple Members can be Unbanned At Once
|
|
"""
|
|
if not await check(ctx, members):
|
|
with ctx.typing():
|
|
await unban_members(self, ctx, members, reason)
|
|
|
|
@command(name="purge")
|
|
@guild_only()
|
|
@has_guild_permissions(manage_messages=True)
|
|
@bot_has_guild_permissions(manage_messages=True, read_message_history=True)
|
|
@cooldown(1, 1, BucketType.user)
|
|
async def purge(self, ctx, amount: int = None):
|
|
"""
|
|
Purge Messages from Channel
|
|
(No Amount Will Default to 50 Messages Deleted)
|
|
"""
|
|
|
|
# When an amount is specified and is between 0 and 100
|
|
if amount:
|
|
if 0 < amount <= 100:
|
|
|
|
# Delete the message sent and then the amount specified
|
|
# (Only messages sent within the last 14 days)
|
|
|
|
await ctx.message.delete()
|
|
deleted = await ctx.channel.purge(limit=amount,
|
|
after=datetime.datetime.utcnow() - timedelta(days=14))
|
|
|
|
await ctx.send(f"Deleted **{len(deleted):,}** messages.", delete_after=5)
|
|
|
|
# Send error if amount is not between 0 and 100
|
|
else:
|
|
await self.bot.generate_embed(ctx, desc="The amount provided is not between **0** and **100**")
|
|
|
|
# Delete the last 50 messages if no amount is given
|
|
else:
|
|
|
|
# Delete the message sent and then the amount specified
|
|
# (Only messages sent within the last 14 days)
|
|
|
|
await ctx.message.delete()
|
|
deleted = await ctx.channel.purge(limit=50,
|
|
after=datetime.datetime.utcnow() - timedelta(days=14))
|
|
|
|
await ctx.send(f"Deleted **{len(deleted):,}** messages.", delete_after=5)
|
|
|
|
@ban.error
|
|
@unban.error
|
|
@force_ban.error
|
|
async def ban_command_error(self, ctx, exc):
|
|
"""Catching error if channel is not recognised"""
|
|
|
|
error = getattr(exc, "original", exc)
|
|
|
|
if isinstance(error, NotFound):
|
|
text = "**❌ User Not Detected... Aborting Process** ❌"
|
|
await self.bot.generate_embed(ctx, desc=text)
|
|
|
|
@Cog.listener()
|
|
async def on_raw_bulk_message_delete(self, payload):
|
|
"""Logging Bulk Message Deletion from Server"""
|
|
|
|
if modlogs := self.bot.get_modlog_for_guild(payload.guild_id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
deleted_msgs_channel = self.bot.get_channel(payload.channel_id)
|
|
desc = f"**Bulk Delete in {deleted_msgs_channel.mention} | {len(payload.message_ids)} messages deleted**"
|
|
|
|
# Set up embed showing the messages deleted and what channel they were deleted in
|
|
embed = Embed(
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=deleted_msgs_channel.guild.name, icon_url=deleted_msgs_channel.guild.icon_url)
|
|
embed.set_footer(text="Bulk Message Deletion")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_member_remove(self, member):
|
|
"""Log Member Leaves from Server"""
|
|
|
|
if member == self.bot.user: return
|
|
|
|
removed_at = datetime.datetime.utcnow()
|
|
if modlogs := self.bot.get_modlog_for_guild(member.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
embed = Embed(title="Member Removed",
|
|
description=f"**Member --> {member.mention} | {member}"
|
|
f"\nID -->** {member.id}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=removed_at)
|
|
embed.add_field(name="Account Creation Date",
|
|
value=member.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"), inline=True)
|
|
embed.add_field(name="Member Left Date", value=removed_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_thumbnail(url=member.avatar_url)
|
|
embed.set_footer(text=f"Member Left")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_member_join(self, member):
|
|
"""Log Member Joins to Server"""
|
|
|
|
if member == self.bot.user: return
|
|
|
|
joined_at = datetime.datetime.utcnow()
|
|
if modlogs := self.bot.get_modlog_for_guild(member.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
embed = Embed(title="Member Joined",
|
|
description=f"**Member --> {member.mention} | {member}"
|
|
f"\nID -->** {member.id}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=joined_at)
|
|
embed.add_field(name="Account Creation Date",
|
|
value=member.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"), inline=True)
|
|
embed.add_field(name="Member Joined Date", value=joined_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_thumbnail(url=member.avatar_url)
|
|
embed.set_footer(text=f"Member Joined")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_member_ban(self, guild, user):
|
|
"""Logs Member Bans to Server"""
|
|
|
|
if user == self.bot.user: return
|
|
|
|
banned_at = datetime.datetime.utcnow()
|
|
if modlogs := self.bot.get_modlog_for_guild(guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
embed = Embed(title="Member Banned",
|
|
description=f"**Member --> {user.mention} | {user}"
|
|
f"\nID -->** {user.id}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=banned_at)
|
|
embed.add_field(name="Account Creation Date",
|
|
value=user.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"), inline=True)
|
|
embed.add_field(name="Member Banned Date", value=banned_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_footer(text="Member Banned")
|
|
embed.set_thumbnail(url=user.avatar_url)
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_member_unban(self, guild, user):
|
|
"""Logs Member Unbans to Server"""
|
|
|
|
unbanned_at = datetime.datetime.utcnow()
|
|
if modlogs := self.bot.get_modlog_for_guild(guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
embed = Embed(title="Member Unbanned",
|
|
description=f"**Member --> {user.mention} | {user}"
|
|
f"\nID -->** {user.id}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=unbanned_at)
|
|
embed.add_field(name="Account Creation Date",
|
|
value=user.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"), inline=True)
|
|
embed.add_field(name="Member Unbanned Date", value=unbanned_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_footer(text="Member Unbanned")
|
|
embed.set_thumbnail(url=user.avatar_url)
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_member_update(self, before, after):
|
|
"""Logging Member Profile Updates"""
|
|
|
|
if before == self.bot.user: return
|
|
|
|
if modlogs := self.bot.get_modlog_for_guild(after.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
# Logging Nickname Updates
|
|
if before.nick != after.nick:
|
|
embed = Embed(description=f"**Nickname Updated"
|
|
f"\nID -->** {after.id}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after, icon_url=after.avatar_url)
|
|
embed.add_field(name="Before",
|
|
value=before.nick, inline=False)
|
|
embed.add_field(name="After",
|
|
value=after.nick, inline=False)
|
|
embed.set_footer(text="Nickname Updated")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
# Logging Role additions/removals from Members
|
|
if after.roles != before.roles:
|
|
# Retrieve the roles that were added/removed to/from the Member
|
|
new_roles = [roles for roles in after.roles if roles not in before.roles]
|
|
old_roles = [roles for roles in before.roles if roles not in after.roles]
|
|
|
|
# As long as roles were added to the Member, log the role(s) that were given
|
|
if len(new_roles) >= 1:
|
|
new_roles_string = " **|** ".join(r.mention for r in new_roles)
|
|
|
|
# Change the description of the embed depending on how many roles were added
|
|
if len(new_roles) == 1:
|
|
desc = f"**Role Added --> {new_roles_string}**"
|
|
footer = "Role Added"
|
|
else:
|
|
desc = f"**Roles Added --> {new_roles_string}**"
|
|
footer = "Roles Added"
|
|
|
|
embed = Embed(title=footer,
|
|
description=f"**Member --> {after.mention} | {after}"
|
|
f"\nID -->** {after.id}"
|
|
f"\n\n{desc}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after, icon_url=after.avatar_url)
|
|
embed.set_footer(text=footer)
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
# As long as roles were removed from the member, log the role(s) that were removed
|
|
if len(old_roles) >= 1:
|
|
old_roles_string = " **|** ".join(r.mention for r in old_roles)
|
|
|
|
# Change the description of the embed depending on how many roles were removed
|
|
if len(old_roles) == 1:
|
|
desc = f"**Role Removed --> {old_roles_string}**"
|
|
footer = "Role Removed"
|
|
else:
|
|
desc = f"**Roles Removed --> {old_roles_string}**"
|
|
footer = "Roles Removed"
|
|
|
|
embed = Embed(title=footer,
|
|
description=f"**Member --> {after.mention} | {after}"
|
|
f"\nID -->** {after.id}"
|
|
f"\n\n{desc}",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after, icon_url=after.avatar_url)
|
|
embed.set_footer(text=footer)
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_message_edit(self, before, after):
|
|
"""Logging Message Edits (Within Cache)"""
|
|
|
|
msg_channel = self.bot.get_channel(after.channel.id)
|
|
|
|
# Get the channel within the cache
|
|
if not isinstance(msg_channel, DMChannel):
|
|
# Get the channel within the cache
|
|
channel = self.bot.get_modlog_for_guild(after.guild.id)
|
|
else:
|
|
return
|
|
|
|
# When no modlogs channel is returned, do nothing
|
|
if channel:
|
|
modlogs_channel = self.bot.get_channel(channel)
|
|
|
|
# Logging Message Content Edits
|
|
# Not logging any message edits from bots
|
|
if before.content != after.content and not after.author.bot:
|
|
desc = f"**Channel -->** <#{after.channel.id}>" \
|
|
f"\n**Message ID -->** {after.id}" \
|
|
f"\n**Edited Message -->** [Jump To Message]({after.jump_url})"
|
|
|
|
# When the message context exceeds 500 characters, only display the first 500 characters in the logs
|
|
before_value = f"{before.content[:500]} ..." if len(before.content) >= 500 else before.content
|
|
after_value = f"{after.content[:500]} ..." if len(after.content) >= 500 else after.content
|
|
|
|
embed = Embed(title="Message Edited",
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after.author, icon_url=after.author.avatar_url)
|
|
embed.add_field(name="Before",
|
|
value=before_value, inline=False)
|
|
embed.add_field(name="After",
|
|
value=after_value, inline=False)
|
|
embed.set_footer(text="Message Edited")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_raw_message_edit(self, payload):
|
|
"""Logging Message Edits Not Stored Within Internal Cache"""
|
|
|
|
msg_channel = self.bot.get_channel(int(payload.data["channel_id"]))
|
|
|
|
# Get the channel within the cache
|
|
if not isinstance(msg_channel, DMChannel):
|
|
channel = self.bot.get_modlog_for_guild(int(payload.data["guild_id"]))
|
|
else:
|
|
return
|
|
|
|
# Only log this message edit when the message does not exist within the internal cache
|
|
# and modlogs channel is set up
|
|
if channel and not payload.cached_message:
|
|
modlogs_channel = self.bot.get_channel(channel)
|
|
|
|
desc = f"**Channel -->** {msg_channel.mention}" \
|
|
f"\n**Message ID -->** {payload.message_id}" \
|
|
f"\n**Message Content -->** N/A"
|
|
embed = Embed(title="Raw Message Edited",
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=modlogs_channel.guild.name, icon_url=modlogs_channel.guild.icon_url)
|
|
embed.set_footer(text="Raw Message Edited")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_message_delete(self, message):
|
|
"""Logging Message Deletions (Within Cache)"""
|
|
|
|
# Get the channel within the cache
|
|
channel = self.bot.get_modlog_for_guild(message.guild.id)
|
|
|
|
# When no modlogs channel is returned, do nothing
|
|
if channel and not message.author.bot:
|
|
modlogs_channel = self.bot.get_channel(channel)
|
|
|
|
if not message.attachments:
|
|
desc = f"**Channel --> {message.channel.mention}**" \
|
|
f"\n**Author ID -->** {message.author.id}" \
|
|
f"\n**Message ID -->** {message.id}"
|
|
fields = [("Message Content", message.content, False)]
|
|
else:
|
|
attach_string = "".join(f"[Here]({attach.proxy_url})" for attach in message.attachments)
|
|
desc = f"**Channel --> {message.channel.mention}**" \
|
|
f"\n**Author ID -->** {message.author.id}" \
|
|
f"\n**Message ID -->** {message.id}"
|
|
fields = [("Message Content", message.content, False),
|
|
("Attachment Link(s)", attach_string, False)]
|
|
|
|
embed = Embed(title="Message Deleted",
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=message.author, icon_url=message.author.avatar_url)
|
|
embed.set_footer(text="Message Deleted")
|
|
|
|
# Add fields to the embed
|
|
for name, value, inline in fields:
|
|
embed.add_field(name=name, value=value, inline=inline)
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_raw_message_delete(self, payload):
|
|
"""Logging Message Deletions Not Stored Within Internal Cache"""
|
|
|
|
msg_channel = self.bot.get_channel(payload.channel_id)
|
|
|
|
# Get the channel within the cache
|
|
if not isinstance(msg_channel, DMChannel):
|
|
channel = self.bot.get_modlog_for_guild(payload.guild_id)
|
|
else:
|
|
return
|
|
|
|
# Only log this message deletion when the message does not exist within the internal cache
|
|
# and modlogs channel is set up
|
|
if channel and not payload.cached_message:
|
|
modlogs_channel = self.bot.get_channel(channel)
|
|
|
|
desc = f"**Channel -->** {msg_channel.mention}" \
|
|
f"\n**Message ID -->** {payload.message_id}" \
|
|
f"\n**Message Content -->** N/A"
|
|
embed = Embed(title="Raw Message Deleted",
|
|
description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=modlogs_channel.guild.name, icon_url=modlogs_channel.guild.icon_url)
|
|
embed.set_footer(text="Raw Message Deleted")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_guild_channel_delete(self, channel):
|
|
"""Logging channel deletions within the guild"""
|
|
|
|
deleted_at = datetime.datetime.utcnow()
|
|
|
|
if modlogs := self.bot.get_modlog_for_guild(channel.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
category = channel.category if channel.category else self.bot.cross
|
|
|
|
desc = f"**Channel Deleted |** #{channel.name}\n" \
|
|
f"**Category |** {category}\n" \
|
|
f"**Position |** #{channel.position} / {len(channel.guild.channels)}\n"
|
|
embed = Embed(description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.add_field(name="Created Date",
|
|
value=channel.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.add_field(name="Deleted Date",
|
|
value=deleted_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_author(name=modlogs_channel.guild.name, icon_url=modlogs_channel.guild.icon_url)
|
|
embed.set_footer(text="Channel Deleted")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_guild_channel_create(self, channel):
|
|
"""Logging channel creations within the guild"""
|
|
|
|
if modlogs := self.bot.get_modlog_for_guild(channel.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
category = channel.category if channel.category else self.bot.cross
|
|
|
|
desc = f"**Channel Created |** {channel.mention}\n" \
|
|
f"**Category |** {category}\n" \
|
|
f"**Position |** #{channel.position} / {len(channel.guild.channels)}\n"
|
|
embed = Embed(description=desc,
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.add_field(name="Created Date",
|
|
value=channel.created_at.strftime("%a, %b %d, %Y\n%I:%M:%S %p"),
|
|
inline=True)
|
|
embed.set_author(name=modlogs_channel.guild.name, icon_url=modlogs_channel.guild.icon_url)
|
|
embed.set_footer(text="Channel Created")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
@Cog.listener()
|
|
async def on_guild_channel_update(self, before, after):
|
|
"""Logging channel updates"""
|
|
|
|
if modlogs := self.bot.get_modlog_for_guild(after.guild.id):
|
|
modlogs_channel = self.bot.get_channel(modlogs)
|
|
|
|
# Logging Channel Name Updates
|
|
if before.name != after.name:
|
|
embed = Embed(description=f"**Channel Name Changed**",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after.guild, icon_url=after.guild.icon_url)
|
|
embed.add_field(name="Before",
|
|
value=f"#{before.name}", inline=False)
|
|
embed.add_field(name="After",
|
|
value=after.mention, inline=False)
|
|
embed.set_footer(text=f"ID: {after.id}")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
# Logging Channel Position Updates
|
|
if before.position != after.position:
|
|
embed = Embed(description=f"**{after.mention} Position Changed**",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after.guild, icon_url=after.guild.icon_url)
|
|
embed.add_field(name="Before",
|
|
value=f"#{before.position} / {len(before.guild.channels)}", inline=False)
|
|
embed.add_field(name="After",
|
|
value=f"#{after.position} / {len(after.guild.channels)}", inline=False)
|
|
embed.set_footer(text=f"ID: {after.id}")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
# Logging any roles added/removed from channel permissions
|
|
if before.changed_roles != after.changed_roles:
|
|
new_roles = [roles for roles in after.changed_roles]
|
|
old_roles = [roles for roles in before.changed_roles]
|
|
|
|
def role_string(roles):
|
|
# Check if the amount of roles is above 20
|
|
if len(roles) > 20:
|
|
# Retrieve the length of the remaining roles
|
|
length = len(roles) - 20
|
|
|
|
# Store the first 20 roles in a string (highest to lowest)
|
|
string = f"{' **|** '.join(map(str, (role.mention for role in list(reversed(roles))[:20])))} and **{length}** more"
|
|
return string
|
|
|
|
else:
|
|
# Display all roles as it is lower than 20
|
|
string = f"{' **|** '.join(map(str, (role.mention for role in list(reversed(roles[1:])))))}"
|
|
return string
|
|
|
|
embed = Embed(description=f"**{after.mention} Role Overrides Updated**",
|
|
colour=self.bot.admin_colour,
|
|
timestamp=datetime.datetime.utcnow())
|
|
embed.set_author(name=after.guild, icon_url=after.guild.icon_url)
|
|
embed.add_field(name="Before",
|
|
value=role_string(old_roles) or after.guild.default_role.mention, inline=False)
|
|
embed.add_field(name="After",
|
|
value=role_string(new_roles) or after.guild.default_role.mention, inline=False)
|
|
embed.set_footer(text=f"ID: {after.id}")
|
|
|
|
await modlogs_channel.send(embed=embed)
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Moderation(bot))
|
|
|
|
|
|
"""
|
|
print(before.overwrites)
|
|
print(after.overwrites)
|
|
|
|
before_diffkeys = [k for k in before.overwrites if
|
|
before.overwrites[k].pair() != after.overwrites[k].pair()]
|
|
after_diffkeys = [k for k in after.overwrites if after.overwrites[k].pair() != before.overwrites[k].pair()]
|
|
|
|
before_array = []
|
|
after_array = []
|
|
for k in before_diffkeys:
|
|
for pair in k.permissions:
|
|
before_array += [pair]
|
|
|
|
for k in after_diffkeys:
|
|
for pair in k.permissions:
|
|
after_array += [pair]
|
|
|
|
old_roles = [p for p in before_array if p in after_array]
|
|
after_roles = [p for p in after_array if p in before_array]
|
|
|
|
print(old_roles)
|
|
print(after_roles)
|
|
"""
|