DecoraterBot-cogs/moderation.py at discord-py-cog-rewrite · DecoraterBot-devs/DecoraterBot-cogs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
# coding=utf-8
"""
moderation plugin for DecoraterBot.
"""
import discord
from discord import app_commands
from discord.ext import commands


class Moderation(commands.Cog):
    """
    Moderation Commands Extension to the
    default DecoraterBot Moderation commands.
    """

    @app_commands.command(
        name=app_commands.locale_str('prune', str_id=34),
        description=app_commands.locale_str('Mass delete messages sent by any user.', str_id=35))
    @app_commands.describe(num=app_commands.locale_str('The amount of messages to prune.', str_id=36))
    @app_commands.guild_only()
    @app_commands.checks.has_permissions(send_messages=True, manage_messages=True)
    @app_commands.checks.bot_has_permissions(send_messages=True, manage_messages=True)
    async def prune_command(self, interaction: discord.Interaction, num: app_commands.Range[int, 1, 100] = 1):
        """
        Bot Commands.
        :param interaction: Interaction.
        :param num: Some number.
        :return: Nothing.
        """
        await interaction.response.defer(thinking=True)
        reply_data = await self.prune_command_iterator_helper(interaction, num)
        if reply_data is not None:
            await interaction.followup.send(content=reply_data)

    @app_commands.command(
        name=app_commands.locale_str('clear', str_id=37),
        description=app_commands.locale_str('Clears the messages that was sent by the bot.', str_id=38))
    @app_commands.guild_only()
    @app_commands.checks.has_permissions(send_messages=True)
    @app_commands.checks.bot_has_permissions(send_messages=True)
    async def clear_command(self, interaction: discord.Interaction):
        """
        Bot Commands.
        :param interaction: Messages.
        :return: Nothing.
        """
        await interaction.response.defer(thinking=True)
        reply_data = await self.clear_command_iterator_helper(interaction)
        if reply_data is not None:
            await interaction.followup.send(content=reply_data)

    # Helpers.
    @staticmethod
    async def prune_command_iterator_helper(interaction: discord.Interaction, num: int) -> str:
        """
        Prunes Messages.
        :param interaction: Interaction Context.
        :param num: Some number.
        :return: message string with the number of messages deleted.
        """
        reason = await interaction.translate(app_commands.locale_str('', str_id=26))
        deleted = await interaction.channel.purge(
            limit=num,
            before=interaction.created_at,
            reason=reason)
        result: str = await interaction.translate(app_commands.locale_str('', str_id=27))
        return result.format(len(deleted))

    @staticmethod
    async def clear_command_iterator_helper(interaction: discord.Interaction) -> str:
        """
        Clears the bot's messages.
        :param interaction: Message Context.
        :return: message string with the number of messages deleted.
        """
        reason: str = await interaction.translate(app_commands.locale_str('', str_id=28))
        deleted = await interaction.channel.purge(
            limit=100,
            before=interaction.created_at,
            check=lambda e: e.author == interaction.client.user,
            reason=reason.format(interaction.client.user.name))
        result: str = await interaction.translate(app_commands.locale_str('', str_id=29))
        return result.format(len(deleted))

    @staticmethod
    async def automod_helper(execution: discord.AutoModAction, reason: str):
        if execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.mention_spam \
                or execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.harmful_link:
            if execution.action.type == discord.AutoModRuleActionType.send_alert_message:
                # ban the member.
                await execution.member.ban(
                    delete_message_seconds=86400 * 7,
                    reason=reason)
                await execution.guild.get_channel(execution.action.channel_id).send(
                    content=f'Banned {execution.member.name} for \'{reason}\'.')
        else:
            # timeout the member for 120 seconds.
            # await execution.member.timeout(until=datetime.datetime.utcnow().minute + , reason=reason)
            await execution.guild.get_channel(execution.action.channel_id).send(
                content=f'Timed out {execution.member.name} for \'{reason}\'.')

    # Events.
    @prune_command.error
    @clear_command.error
    async def on_cog_app_command_error(self, interaction: discord.Interaction, error: app_commands.AppCommandError):
        if isinstance(error, app_commands.BotMissingPermissions):
            if interaction.command.name == app_commands.locale_str('prune', str_id=34):
                await interaction.response.send_message(
                    content=await interaction.translate(app_commands.locale_str('', str_id=30)))
            elif interaction.command.name == app_commands.locale_str('clear', str_id=37):
                await interaction.response.send_message(
                    content=await interaction.translate(app_commands.locale_str('', str_id=32)))
        elif isinstance(error, app_commands.MissingPermissions):
            if interaction.command.name == app_commands.locale_str('prune', str_id=34):
                await interaction.response.send_message(
                    content=await interaction.translate(app_commands.locale_str('', str_id=31)))
            elif interaction.command.name == app_commands.locale_str('clear', str_id=37):
                await interaction.response.send_message(
                    content=await interaction.translate(app_commands.locale_str('', str_id=33)))

    @commands.Cog.listener()
    async def on_automod_action(self, execution: discord.AutoModAction):
        if execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.mention_spam:
            await self.automod_helper(execution, '[AutoMod] Mention Spam')
        elif execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.harmful_link:
            await self.automod_helper(execution, '[AutoMod] Harmful Link')
        elif execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.keyword:
            await self.automod_helper(execution, '[AutoMod] Keyword')
        elif execution.rule_trigger_type == discord.enums.AutoModRuleTriggerType.keyword_preset:
            await self.automod_helper(execution, '[AutoMod] Keyword Preset')


async def setup(bot):
    """
    DecoraterBot's Moderation Plugin.
    """
    await bot.add_cog(Moderation())