Syntrel/cogs/moderation/ban.py

import discord
from discord import app_commands
from discord.ext import commands
from discord.ext.commands import Context


def ban_command():
    @commands.hybrid_command(
        name="ban",
        description="Bans a user from the server.",
    )
    @app_commands.describe(
        user="The user that should be banned.",
        reason="The reason why the user should be banned.",
        delete_messages="Delete messages from the user (choose time period).",
    )
    @app_commands.choices(
        delete_messages=[
            app_commands.Choice(name="Don't delete any messages", value="none"),
            app_commands.Choice(name="Last 1 hour", value="1h"),
            app_commands.Choice(name="Last 6 hours", value="6h"),
            app_commands.Choice(name="Last 12 hours", value="12h"),
            app_commands.Choice(name="Last 24 hours", value="1d"),
            app_commands.Choice(name="Last 3 days", value="3d"),
            app_commands.Choice(name="Last 7 days", value="7d"),
        ]
    )
    async def ban(
        self,
        context,
        user: discord.User,
        *,
        reason: str = "Not specified",
        delete_messages: str = "none",
    ):
        try:
            member = context.guild.get_member(user.id)
            if not member:
                try:
                    member = await context.guild.fetch_member(user.id)
                except discord.NotFound:
                    try:
                        await context.guild.ban(user, reason=reason)
                        embed = discord.Embed(
                            title="Ban",
                            description=f"**{user}** was banned by **{context.author}**!",
                            color=0x7289DA,
                        ).set_author(
                            name="Moderation",
                            icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                        )
                        embed.add_field(name="Reason:", value=reason)
                        await context.send(embed=embed)
                        return
                    except discord.Forbidden:
                        embed = discord.Embed(
                            title="Error!",
                            description="I don't have permission to ban this user.",
                            color=0xE02B2B,
                        ).set_author(
                            name="Moderation",
                            icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                        )
                        await context.send(embed=embed, ephemeral=True)
                        return
                    except Exception:
                        embed = discord.Embed(
                            title="Error!",
                            description="An error occurred while trying to ban the user.",
                            color=0xE02B2B,
                        ).set_author(
                            name="Moderation",
                            icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                        )
                        await context.send(embed=embed, ephemeral=True)
                        return

            if (
                not context.author.guild_permissions.ban_members
                and context.author != context.guild.owner
            ):
                embed = discord.Embed(
                    title="Missing Permissions!",
                    description="You don't have the `Ban Members` permission to use this command.",
                    color=0xE02B2B,
                ).set_author(
                    name="Moderation",
                    icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                )
                await context.send(embed=embed, ephemeral=True)
                return

            if member and member.top_role >= context.guild.me.top_role:
                embed = discord.Embed(
                    title="Cannot Ban User",
                    description="This user has a higher or equal role to me. Make sure my role is above theirs.",
                    color=0xE02B2B,
                ).set_author(
                    name="Moderation",
                    icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                )
                await context.send(embed=embed, ephemeral=True)
                return

            if member and context.author != context.guild.owner:
                if member.top_role >= context.author.top_role:
                    embed = discord.Embed(
                        title="Cannot Ban User",
                        description="You cannot ban this user as they have a higher or equal role to you.",
                        color=0xE02B2B,
                    ).set_author(
                        name="Moderation",
                        icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                    )
                    await context.send(embed=embed, ephemeral=True)
                    return

            delete_message_days = 0
            delete_all_messages = False

            if delete_messages != "none":
                if delete_messages == "all":
                    delete_all_messages = True
                    delete_message_days = 7
                elif delete_messages.endswith("h"):
                    hours = int(delete_messages[:-1])
                    delete_message_days = min(hours / 24, 7)
                elif delete_messages.endswith("d"):
                    days = int(delete_messages[:-1])
                    delete_message_days = min(days, 7)

            try:
                if member:
                    try:
                        dm_embed = discord.Embed(
                            title="Ban",
                            description=f"You were banned by **{context.author}** from **{context.guild.name}**!\nReason: {reason}",
                            color=0xE02B2B,
                        ).set_author(
                            name="Moderation",
                            icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                        )
                        await member.send(embed=dm_embed)
                    except (discord.Forbidden, discord.HTTPException):
                        pass

                if member:
                    await member.ban(
                        reason=reason, delete_message_days=delete_message_days
                    )
                else:
                    await context.guild.ban(
                        user, reason=reason, delete_message_days=delete_message_days
                    )

                if delete_all_messages:
                    await self.delete_all_user_messages(context.guild, user.id)

                embed = discord.Embed(
                    title="Ban",
                    description=f"**{user}** was banned by **{context.author}**!",
                    color=0x7289DA,
                ).set_author(
                    name="Moderation",
                    icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                )
                embed.add_field(name="Reason:", value=reason)

                if delete_messages != "none":
                    if delete_all_messages:
                        embed.add_field(
                            name="Messages Deleted:", value="All messages", inline=False
                        )
                    else:
                        delete_time_text = self.format_delete_time(delete_messages)
                        embed.add_field(
                            name="Messages Deleted:",
                            value=delete_time_text,
                            inline=False,
                        )

                await context.send(embed=embed)

            except discord.Forbidden:
                embed = discord.Embed(
                    title="Error!",
                    description="I don't have permission to ban this user. Make sure my role is above theirs.",
                    color=0xE02B2B,
                ).set_author(
                    name="Moderation",
                    icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                )
                await context.send(embed=embed, ephemeral=True)
            except discord.HTTPException as e:
                if "Cannot ban the owner of a guild" in str(e):
                    embed = discord.Embed(
                        title="Cannot Ban User",
                        description="You cannot ban the server owner.",
                        color=0xE02B2B,
                    ).set_author(
                        name="Moderation",
                        icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                    )
                else:
                    embed = discord.Embed(
                        title="Error!",
                        description=f"Discord API error: {str(e)}",
                        color=0xE02B2B,
                    ).set_author(
                        name="Moderation",
                        icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                    )
                await context.send(embed=embed, ephemeral=True)
            except Exception as e:
                embed = discord.Embed(
                    title="Debug Error!",
                    description=f"Error type: {type(e).__name__}\nError message: {str(e)}",
                    color=0xE02B2B,
                ).set_author(
                    name="Moderation",
                    icon_url="https://yes.nighty.works/raw/CPKHQd.png",
                )
                await context.send(embed=embed, ephemeral=True)

        except Exception as e:
            embed = discord.Embed(
                title="Error!",
                description="An unexpected error occurred.",
                color=0xE02B2B,
            ).set_author(
                name="Moderation", icon_url="https://yes.nighty.works/raw/CPKHQd.png"
            )
            await context.send(embed=embed, ephemeral=True)

    async def delete_all_user_messages(
        self, guild: discord.Guild, user_id: int
    ) -> None:
        for channel in guild.text_channels:
            try:
                permissions = channel.permissions_for(guild.me)
                if not (
                    permissions.read_message_history and permissions.manage_messages
                ):
                    continue

                deleted = True
                while deleted:
                    deleted = False
                    try:
                        async for message in channel.history(limit=100):
                            if message.author.id == user_id:
                                try:
                                    await message.delete()
                                    deleted = True
                                except (
                                    discord.NotFound,
                                    discord.Forbidden,
                                    discord.HTTPException,
                                ):
                                    continue
                    except (discord.Forbidden, discord.HTTPException):
                        break

            except (discord.Forbidden, discord.HTTPException):
                continue

    def format_delete_time(self, delete_option: str) -> str:
        time_formats = {
            "1h": "Last 1 hour",
            "6h": "Last 6 hours",
            "12h": "Last 12 hours",
            "1d": "Last 24 hours",
            "3d": "Last 3 days",
            "7d": "Last 7 days",
        }
        return time_formats.get(delete_option, "Unknown time period")

    return ban