Add verify cog, closes #14

jarvis/__init__.py

@@ -64,10 +64,16 @@ async def on_member_join(user: Member):
         mute_role = db.jarvis.settings.find_one(
             {"guild": guild.id, "setting": "mute"}
         )
-        role = guild.get_role(mute_role["role"])
+        role = guild.get_role(mute_role["value"])
         await user.add_roles(
             role, reason="User is muted still muted from prior mute"
         )
+    unverified = db.jarvis.settings.find_one(
+        {"guild": guild.id, "setting": "unverified"}
+    )
+    if unverified:
+        role = guild.get_role(unverified["value"])
+        await user.add_roles(role, reason="User just joined and is unverified")
 
 
 @jarvis.event

jarvis/cogs/settings.py

@@ -111,6 +111,48 @@ class SettingsCog(commands.Cog):
         self.update_settings("massmention", amount, ctx.guild.id)
         await ctx.send(f"Settings applied. New massmention limit is {amount}")
 
+    @cog_ext.cog_subcommand(
+        base="settings",
+        name="verified",
+        description="Set verified role",
+        guild_ids=[418094694325813248, 578757004059738142],
+        options=[
+            create_option(
+                name="role",
+                description="verified role",
+                option_type=8,
+                required=True,
+            )
+        ],
+    )
+    @admin_or_permissions(kick_members=True, ban_members=True)
+    async def _verified(self, ctx, role: Role):
+        await ctx.defer()
+        self.update_settings("verified", role.id, ctx.guild.id)
+        await ctx.send(f"Settings applied. New verified role is `{role.name}`")
+
+    @cog_ext.cog_subcommand(
+        base="settings",
+        name="unverified",
+        description="Set unverified role",
+        guild_ids=[418094694325813248, 578757004059738142],
+        options=[
+            create_option(
+                name="role",
+                description="Unverified role role",
+                option_type=8,
+                required=True,
+            )
+        ],
+    )
+    @admin_or_permissions(kick_members=True, ban_members=True)
+    async def _unverified(self, ctx, role: Role):
+        await ctx.defer()
+        self.update_settings("unverified", role.id, ctx.guild.id)
+        await ctx.send(
+            f"Settings applied. New unverified role is `{role.name}`"
+        )
+
 
 def setup(bot):
     bot.add_cog(SettingsCog(bot))

jarvis/cogs/verify.py

@@ -0,0 +1,98 @@
+from random import randint
+
+from discord import Role
+from discord.ext import commands
+from discord_slash import ComponentContext, SlashContext, cog_ext
+from discord_slash.model import ButtonStyle
+from discord_slash.utils import manage_commands, manage_components
+from discord_slash.utils.manage_commands import create_option
+
+import jarvis
+from jarvis.config import get_config
+from jarvis.utils.db import DBManager
+
+
+def create_layout():
+    buttons = []
+    yes = randint(0, 2)
+    for i in range(3):
+        label = "YES" if i == yes else "NO"
+        id = f"no_{i}" if not i == yes else "yes"
+        buttons.append(
+            manage_components.create_button(
+                style=ButtonStyle.grey,
+                label=label,
+                custom_id=f"verify_button||{id}",
+            )
+        )
+    action_row = manage_components.spread_to_rows(*buttons, max_in_row=3)
+    return action_row
+
+
+class VerifyCog(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        self.db = DBManager(get_config().mongo).mongo
+
+    @cog_ext.cog_slash(
+        name="verify",
+        description="Verify that you've read the rules",
+        guild_ids=[418094694325813248, 578757004059738142],
+    )
+    async def _verify(self, ctx: SlashContext):
+        await ctx.defer()
+        role = self.db.jarvis.settings.find_one(
+            {"setting": "verified", "guild": ctx.guild.id}
+        )
+        if ctx.guild.get_role(role["value"]) in ctx.author.roles:
+            await ctx.send("You are already verified.", delete_after=5)
+            return
+        components = create_layout()
+        message = await ctx.send(
+            content=f"{ctx.author.mention}, "
+            + "please press the button that says `YES`",
+            components=components,
+        )
+        await message.delete(delay=15)
+
+    @cog_ext.cog_component(components=create_layout())
+    async def _process(self, ctx: ComponentContext):
+        await ctx.defer(edit_origin=True)
+        try:
+            if ctx.author.id != ctx.origin_message.mentions[0].id:
+                return
+        except Exception:
+            return
+        correct = ctx.custom_id.split("||")[-1] == "yes"
+        if correct:
+            components = ctx.origin_message.components
+            for c in components:
+                for c2 in c["components"]:
+                    c2["disabled"] = True
+            setting = self.db.jarvis.settings.find_one(
+                {"setting": "verified", "guild": ctx.guild.id}
+            )
+            role = ctx.guild.get_role(setting["value"])
+            await ctx.author.add_roles(role, reason="Verified")
+            setting = self.db.jarvis.settings.find_one(
+                {"setting": "unverified", "guild": ctx.guild.id}
+            )
+            if setting:
+                role = ctx.guild.get_role(setting["value"])
+                await ctx.author.remove_roles(role, reason="Verified")
+            await ctx.edit_origin(
+                content=f"{ctx.author.mention} has been verified",
+                components=manage_components.spread_to_rows(
+                    *components, max_in_row=5
+                ),
+            )
+            await ctx.origin_message.delete(delay=5)
+        else:
+            await ctx.edit_origin(
+                content=f"{ctx.author.mention}, incorrect. "
+                + "Please press the button that says `YES`",
+            )
+
+
+def setup(bot):
+    bot.add_cog(VerifyCog(bot))