Project initHEADmain

1 files changed, 112 insertions, 0 deletions

diff --git a/cogs/Automod.py b/cogs/Automod.py
new file mode 100644
index 0000000..6a98fb8
--- /dev/null
+++ b/cogs/Automod.py
@@ -0,0 +1,112 @@
+from discord.ext import commands
+import discord
+import importlib
+import utils
+importlib.reload(utils)
+
+
+class Automod(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.group(invoke_without_command=True)
+    async def automod(self, ctx, *arg):
+        """Configure some server moderation for the server."""
+        prefix = await utils.prefix(self.bot, ctx.message)
+        emb = await utils.embed(ctx, f"Commands for `{prefix}automod`",
+                                "Automod helps keep unwanted words out of the chat.")
+        emb = await utils.field(emb, f"{prefix}automod add `[words / phrase]`",
+                                f"Adds the given word or phrase to the word filter. These words will be deleted.")
+        emb = await utils.field(emb, f"{prefix}automod list",
+                                f"Shows the words Automod will remove from the chat.")
+        emb = await utils.field(emb, f"{prefix}automod remove `[words / phrase]`",
+                                f"Removes the given word or phrase from the filter.")
+        await ctx.send(embed=emb)
+
+    @automod.command()
+    @commands.bot_has_permissions(manage_messages=True)
+    @commands.has_permissions(manage_messages=True)
+    async def add(self, ctx, *, words):
+        await utils.sql_set("INSERT INTO `automod_words` (guild_id, word) VALUES (%s, '%s')", (ctx.guild.id, words))
+        await ctx.send(content=f"Added {words} into the word filter.")
+
+
+    @automod.command()
+    async def list(self, ctx):
+        dict_words = await utils.sql_get("SELECT `word` from `automod_words` WHERE `guild_id` = %s", (ctx.guild.id))
+        lst_words = [x['word'] for x in dict_words]
+
+        emb = await utils.embed(ctx, "Blacklisted words:", "\n".join(lst_words))
+        await ctx.send(embed=emb)
+
+    @automod.command()
+    @commands.bot_has_permissions(manage_messages=True)
+    @commands.has_permissions(manage_messages=True)
+    async def remove(self, ctx, *, words):
+        dict_words = await utils.sql_get("SELECT `word` from `automod_words` WHERE `guild_id` = %s", (ctx.guild.id))
+        lst_words = [x['word'] for x in dict_words]
+
+        for word in lst_words:
+            if word == words:
+                await utils.sql_set("DELETE FROM `automod_words` WHERE `guild_id`=%s AND `word`='%s'", (ctx.guild.id, words))
+                emb = await utils.embed(ctx, "Automod - Remove complete:", f"Removed `{word}` from the filter.")
+                await ctx.send(embed=emb)
+                break
+
+    @commands.Cog.listener()
+    async def on_message(self, message):
+        if message.author == self.bot.user:
+            return
+
+        bl_words = await utils.sql_get("SELECT `word` FROM `automod_words` WHERE `guild_id` = %s", (message.guild.id,))
+
+        lst_words = []
+        for word in bl_words:
+            lst_words.append(word["word"])
+        lst_words = [x.lower() for x in lst_words]
+        msg_words = [x.lower() for x in message.content.lower().split(' ')]
+        matches = []
+        for db_word in lst_words:
+            if db_word.count(" ") == 0:
+                for msg_word in msg_words:
+                    if db_word.lower() == msg_word.lower():
+                        if str(db_word).casefold() == str(msg_word).casefold():
+                            print("single word match")
+                            matches.append(db_word)
+
+            elif db_word.count(" ") > 0:
+                if db_word.lower() in message.content.lower():
+                    print("phrase match")
+                    matches.append(db_word)
+
+        if not matches == []:   
+            await message.delete()
+            emb = await utils.embed(message, "Message Removed",
+                                    f"[AUTOMOD] Removed a message from {message.author} as it contained a blacklisted word: `{', '.join(matches)}`",
+                                    colour="orange")
+            emb = await utils.footer(emb, f"{message.author.name}\n" +
+                                     f"{message.content}",
+                                     url=message.author.avatar_url)
+            await utils.log(message.guild.id, self.bot, emb=emb)
+
+        # result = set(lst_words).intersection(msg_words)
+        #
+        # #  await message.channel.send(content=result)
+        # print(f"{len(result)}: {result}")
+        # if len(result) > 0:
+        #     await message.delete()
+        #     emb = await utils.embed(message, "Message Removed", f"[AUTOMOD] Removed a message from {message.author} as it contained a blacklisted word: `{', '.join(result)}`")
+        #     emb = await utils.footer(emb, f"{message.author.name}\n" +
+        #                              f"{message.content}",
+        #                              url=message.author.avatar_url)
+        #     await utils.log(message, self.bot, emb=emb)
+
+
+def setup(bot):
+    print("INFO: Loading [Automod]... ", end="")
+    bot.add_cog(Automod(bot))
+    print("Done!")
+
+
+def teardown(bot):
+    print("INFO: Unloading [Automod]")