feat: Add EventRoleManager for managing event notification roles (#112)

Author: phoenixpereira

src/main.py

@@ -21,6 +21,7 @@
 from commands import admin_commands, gemini, help_menu, skullboard
 from constants.colours import LIGHT_YELLOW
 from utils import spam_detection, time
+from utils.event_roles import EventRoleManager
 
 # Load environment variables from .env file
 load_dotenv()
@@ -44,12 +45,25 @@
 
 
 class DuckBot(commands.Bot):
+    async def on_scheduled_event_user_add(self, event, user):
+        """Delegate to EventRoleManager."""
+        await self.event_role_manager.on_scheduled_event_user_add(event, user)
+
+    async def on_scheduled_event_user_remove(self, event, user):
+        """Delegate to EventRoleManager."""
+        await self.event_role_manager.on_scheduled_event_user_remove(event, user)
+
+    async def on_scheduled_event_update(self, before, after):
+        """Delegate to EventRoleManager."""
+        await self.event_role_manager.on_scheduled_event_update(before, after)
+
     def __init__(self):
         super().__init__(command_prefix="", intents=intents)
         self.synced = False  # Make sure that the command tree will be synced only once
         self.skullboard_manager = skullboard.SkullboardManager(
             self
         )  # Initialise SkullboardManager
+        self.event_role_manager = EventRoleManager(self)  # Initialise EventRoleManager
         self.prev_day = None
         self.expiry_loop = None
 

src/utils/event_roles.py

@@ -0,0 +1,89 @@
+import logging
+
+from discord.enums import EventStatus
+
+
+def get_event_role_name(event_name: str) -> str:
+    """Return the role name for a given event."""
+    return f"Event: {event_name}"
+
+
+class EventRoleManager:
+    """Manages auto-assignment and deletion of event notification roles."""
+
+    def __init__(self, bot):
+        self.bot = bot
+
+    async def on_scheduled_event_user_add(self, event, user):
+        """Auto-assigns a notification role (per event) when a user is interested in an event."""
+        guild = self.bot.get_guild(event.guild_id)
+        if not guild:
+            return
+        member = guild.get_member(user.id)
+        if not member:
+            member = await guild.fetch_member(user.id)
+        role_name = get_event_role_name(event.name)
+        role = next((r for r in guild.roles if r.name == role_name), None)
+        if not role:
+            try:
+                role = await guild.create_role(
+                    name=role_name, reason=f"Auto-created for event: {event.name}"
+                )
+                logging.info(f"Created new event notification role: {role_name}")
+            except Exception as e:
+                logging.error(f"Failed to create event notification role: {e}")
+                return
+        if role and member:
+            try:
+                await member.add_roles(
+                    role, reason=f"Interested in event: {event.name}"
+                )
+                logging.info(
+                    f"Added event notification role '{role_name}' to {member.display_name} for event {event.name}"
+                )
+            except Exception as e:
+                logging.error(f"Failed to add event notification role: {e}")
+
+    async def on_scheduled_event_user_remove(self, event, user):
+        """Removes the notification role (per event) if user is no longer interested in the event."""
+        guild = self.bot.get_guild(event.guild_id)
+        if not guild:
+            return
+        member = guild.get_member(user.id)
+        if not member:
+            member = await guild.fetch_member(user.id)
+        role_name = get_event_role_name(event.name)
+        role = next((r for r in guild.roles if r.name == role_name), None)
+        if role and member:
+            try:
+                await member.remove_roles(
+                    role, reason=f"No longer interested in event: {event.name}"
+                )
+                logging.info(
+                    f"Removed event notification role '{role_name}' from {member.display_name} for event {event.name}"
+                )
+            except Exception as e:
+                logging.error(f"Failed to remove event notification role: {e}")
+
+    async def on_scheduled_event_update(self, before, after):
+        """Deletes the notification role (per event) after the event ends or is cancelled."""
+        if before.status != after.status and after.status in (
+            EventStatus.completed,
+            EventStatus.cancelled,
+        ):
+            guild = self.bot.get_guild(after.guild_id)
+            if not guild:
+                return
+            role_name = get_event_role_name(after.name)
+            role = next((r for r in guild.roles if r.name == role_name), None)
+            if not role:
+                return
+            try:
+                await role.delete(
+                    reason=f"Event {after.name} ended or was cancelled; auto-deleted notification role."
+                )
+                logging.info(
+                    f"Deleted event notification role '{role_name}' after event {after.name} ended or was cancelled."
+                )
+            except Exception as e:
+                logging.error(f"Failed to delete event notification role: {e}")