Reaction roles working. Management of message and reactions is messy and needs reinforcement.

cogs/react_roles.py

@@ -0,0 +1,132 @@
+import os
+
+import discord
+from discord.ext import commands
+from data import react_roles
+
+from util.console import console, panel, track_iterable as track
+
+GUILD_IDS = [int(os.getenv("DEV_GUILD_ID"))]
+REACTION_ROLE_CHANNEL_ID = int(os.getenv("REACT_ROLE_CHANNEL_ID"))
+
+
+def update_react_role_embed(category_name):
+    # Generate role list
+    roles = react_roles.get_react_roles_by_category(category_name)
+    role_list = []
+    for role in roles:
+        emoji = role[2]
+        description = role[3]
+        role_list.append(f"{emoji} - {description}")
+
+    # Create an embed for the react-role category
+    embed = discord.Embed(title=f"**{category_name.capitalize()} Roles**",
+                          description="React to this message to gain access to the relevant text and voice channels.\n\n" + "\n".join(role_list),
+                          color=discord.Color.dark_orange())
+    return embed
+
+class ReactRoles(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    # Slash command to add a new reaction role Category / Message
+    @commands.slash_command(name="new_react_role_category",
+                             description="Create a new category of react roles",
+                             guild_ids=GUILD_IDS)
+    @commands.has_permissions(manage_roles=True)
+    async def new_react_role_category(self, ctx: discord.ApplicationContext, category_name: str):
+        # Verify the category is unique
+        existing_categories = react_roles.get_react_roles_categories()
+        if category_name.lower() in [cat[1] for cat in existing_categories]:
+            await ctx.respond(f"Category '{category_name}' already exists.", ephemeral=True)
+            return
+
+        # Send a new Embed message
+        embed = discord.Embed(title=f"**{category_name.capitalize()} Roles**",
+                              description="React to this message to gain access to the relevant text and voice channels.",
+                              color=discord.Color.dark_orange())
+        message = await ctx.guild.get_channel(REACTION_ROLE_CHANNEL_ID).send(embed=embed)
+
+        # Create the category in the database
+        react_roles.add_react_role_category_to_db(message.id, category_name)
+
+        # Send a confirmation message
+        await ctx.respond(f"React role category '{category_name}' created successfully!", ephemeral=True)
+
+    #Slash command to add a new reaction role to an existing category
+    @commands.slash_command(name="new_react_role",
+                             description="Create a new react role",
+                             guild_ids=GUILD_IDS)
+    @commands.has_permissions(manage_roles=True)
+    async def new_react_role(self, ctx: discord.ApplicationContext, category_name: str, role: discord.Role, emoji: str, description: str):
+        # Verify the category exists
+        existing_categories = react_roles.get_react_roles_categories()
+        if category_name.lower() not in [cat[1] for cat in existing_categories]:
+            await ctx.respond(f"Category '{category_name}' does not exist.", ephemeral=True)
+            return
+
+        # Get the message ID of the category
+        message_id = next(cat[0] for cat in existing_categories if cat[1] == category_name.lower())
+
+        # Add the react-role to the database
+        react_roles.add_react_role_to_db(message_id, role.id, emoji, description)
+        message = await self.bot.get_channel(REACTION_ROLE_CHANNEL_ID).fetch_message(message_id)
+        await message.edit(embed=update_react_role_embed(category_name))
+        await message.add_reaction(emoji)
+
+        # Send a confirmation message
+        await ctx.respond(f"React role '{emoji}' - '{role.name}' added to category '{category_name}'.", ephemeral=True)
+
+
+    # Event listener for reaction add
+    @commands.Cog.listener()
+    async def on_raw_reaction_add(self, payload: discord.RawReactionActionEvent):
+        # Check if the reaction is in the correct channel and not from the bot itself
+        if payload.channel_id != REACTION_ROLE_CHANNEL_ID or payload.user_id == self.bot.user.id:
+            return
+
+        # Ge all react roles for the message_id that was reacted to
+        message_id = payload.message_id
+        emoji = str(payload.emoji)
+        react_roles_list = react_roles.get_react_roles_by_message_id(message_id)
+
+        if react_roles_list:
+            for role in react_roles_list:
+                if role[2] == emoji:
+                    guild = self.bot.get_guild(payload.guild_id)
+                    member = payload.member
+                    role_obj = guild.get_role(role[1])
+
+                    # Add the role to the member
+                    if role_obj not in member.roles:
+                        await member.add_roles(role_obj)
+                        console.log (f"[green]✔ Added role:[/] {emoji} {role_obj.name} to {member.name}")
+
+    # Event Listener for reaction remove
+    @commands.Cog.listener()
+    async def on_raw_reaction_remove(self, payload: discord.RawReactionActionEvent):
+        # Check if the reaction is in the correct channel and not from the bot itself
+        if payload.channel_id != REACTION_ROLE_CHANNEL_ID or payload.user_id == self.bot.user.id:
+            return
+
+        # Ge all react roles for the message_id that was reacted to
+        message_id = payload.message_id
+        emoji = str(payload.emoji)
+        react_roles_list = react_roles.get_react_roles_by_message_id(message_id)
+
+        if react_roles_list:
+            for role in react_roles_list:
+                if role[2] == emoji:
+                    guild = self.bot.get_guild(payload.guild_id)
+
+                    # Member object is not available in the payload, so we need to fetch it...
+                    member = await guild.fetch_member(payload.user_id)
+                    role_obj = guild.get_role(role[1])
+
+                    # Add the role to the member
+                    if role_obj in member.roles:
+                        await member.remove_roles(role_obj)
+                        console.log (f"[red]✖ Removed role:[/] {emoji} {role_obj.name} from {member.name}")
+
+def setup(bot):
+    bot.add_cog(ReactRoles(bot))