Closes #20, implement mass mention prevention. TODO: Implement warning for mass mentions

jarvis/__init__.py

@@ -80,6 +80,16 @@ async def on_message(message: Message):
         if autoreact:
             for reaction in autoreact["reactions"]:
                 await message.add_reaction(reaction)
+        massmention = db.jarvis.settings.find_one(
+            {"guild": message.guild.id, "setting": "massmention"}
+        )
+        if (
+            massmention["value"] > 0
+            and len(message.mentions) >= massmention["value"]
+        ):
+            await message.delete()
+            # TODO: Implement warnings on massmention
+            await message.channel.send("Please do not mass mention.")
     await jarvis.process_commands(message)
 
 
@@ -95,7 +105,13 @@ async def on_guild_join(guild):
         )
         await asyncio.sleep(1)
         await general.send("Importing all preferences from home interface...")
-        await asyncio.sleep(5)
+
+        # Set some default settings
+        db = DBManager(get_config().mongo).mongo
+        db.jarvis.settings.insert_one(
+            {"guild": guild.id, "setting": "massmention", "value": 5}
+        )
+
         await general.send("Systems are now fully operational")
 
 

jarvis/cogs/settings.py

@@ -20,7 +20,7 @@ class SettingsCog(commands.Cog):
     def update_settings(self, setting, value, guild):
         settings = {"setting": setting, "value": value, "guild": guild}
         updated = self.db.jarvis.settings.update_one(
-            {"settings": setting, "guild": guild},
+            {"setting": setting, "guild": guild},
             {"$set": settings},
             upsert=True,
         )
@@ -31,6 +31,7 @@ class SettingsCog(commands.Cog):
         base="settings",
         name="mute",
         description="Set mute role",
+        guild_ids=[418094694325813248, 578757004059738142],
         options=[
             create_option(
                 name="role",
@@ -50,6 +51,7 @@ class SettingsCog(commands.Cog):
         base="settings",
         name="modlog",
         description="Set modlog channel",
+        guild_ids=[418094694325813248, 578757004059738142],
         options=[
             create_option(
                 name="channel",
@@ -71,6 +73,7 @@ class SettingsCog(commands.Cog):
         base="settings",
         name="userlog",
         description="Set userlog channel",
+        guild_ids=[418094694325813248, 578757004059738142],
         options=[
             create_option(
                 name="channel",
@@ -88,6 +91,26 @@ class SettingsCog(commands.Cog):
             f"Settings applied. New userlog channel is {channel.mention}"
         )
 
+    @cog_ext.cog_subcommand(
+        base="settings",
+        name="massmention",
+        description="Set massmention amount",
+        guild_ids=[418094694325813248, 578757004059738142],
+        options=[
+            create_option(
+                name="amount",
+                description="Amount of mentions (0 to disable)",
+                option_type=4,
+                required=True,
+            )
+        ],
+    )
+    @commands.has_permissions(administrator=True)
+    async def _massmention(self, ctx, amount: int):
+        await ctx.defer()
+        self.update_settings("massmention", amount, ctx.guild.id)
+        await ctx.send(f"Settings applied. New massmention limit is {amount}")
+
 
 def setup(bot):
     bot.add_cog(SettingsCog(bot))