Add support for assigning roles for meeting

Author: patrick91

.env.sample

@@ -4,3 +4,5 @@ DISCORD_TOKEN=123
 CALENDAR_URL=abc
 NOTES_LINK=abc
 SCHEDULED_EVENT_CHANNEL_ID=123
+REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID=123
+MEETING_WATCHERS_ROLE_ID=123

src/bot.py

@@ -1,6 +1,7 @@
 import asyncio
 
 import nextcord
+from discord import RawReactionActionEvent
 
 from src.database import (
     add_notification_for_date,
@@ -9,7 +10,12 @@
 )
 from src.date_utils import add_localized_times_to_embed
 
-from .config import GENERAL_CHANNEL_ID, NOTES_LINK
+from .config import (
+    GENERAL_CHANNEL_ID,
+    MEETING_WATCHERS_ROLE_ID,
+    NOTES_LINK,
+    REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID,
+)
 from .meeting import find_next_event_and_notify_core_team
 
 
@@ -19,44 +25,89 @@ def __init__(self, *args, **kwargs):
 
         self.bg_task = self.loop.create_task(self.check_events_for_next_week())
 
-    async def on_raw_reaction_add(self, reaction):
+    async def on_remove_checkmark_reaction(
+        self, reaction: RawReactionActionEvent
+    ) -> None:
+        message_id = reaction.message_id
 
-        assert self.user
+        if message_id == REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID:
+            # reaction.member happens to be always `None`, so we use the
+            # `user_id` instead
 
-        if reaction.user_id == self.user.id:
-            return
+            if reaction.guild_id is None:
+                print("Guild is None")
+                return
+
+            guild = await self.fetch_guild(reaction.guild_id)
 
+            assert guild
+
+            member = await guild.fetch_member(reaction.user_id)
+
+            if member:
+                await member.remove_roles(
+                    nextcord.Object(id=MEETING_WATCHERS_ROLE_ID),
+                    reason="Opted out of meeting notifications via reaction",
+                )
+
+    async def on_add_checkmark_reaction(self, reaction: RawReactionActionEvent) -> None:
         message_id = reaction.message_id
-        emoji = reaction.emoji
 
-        if emoji.name == "✅" and (
-            notification := get_notification_for_discord_message_id(
-                message_id, "core_devs"
+        if message_id == REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID:
+            if reaction.member is None:
+                print("Member is None")
+                return
+
+            await reaction.member.add_roles(
+                nextcord.Object(id=MEETING_WATCHERS_ROLE_ID),
+                reason="Opted in for meeting notifications via reaction",
             )
+
+            return
+
+        if meeting_notification := get_notification_for_discord_message_id(
+            message_id, "core_devs"
         ):
-            event_date = notification.date.isoformat()
-            public_notification = get_notification_for_date(event_date, "public")
+            event_date = meeting_notification.date.isoformat()
 
-            if public_notification:
+            if get_notification_for_date(event_date, "public") is not None:
                 return
 
             channel = client.get_channel(GENERAL_CHANNEL_ID)
 
             assert isinstance(channel, nextcord.channel.TextChannel)
 
             embed = nextcord.Embed(color=5814783)
-            add_localized_times_to_embed(embed, notification.date)
+            add_localized_times_to_embed(embed, meeting_notification.date)
 
             message = await channel.send(
                 "Hey @everyone 👋 the next monthly meeting will happen "
-                f"{notification.date.humanize()} 📅\n"
+                f"{meeting_notification.date.humanize()} 📅\n"
                 f"Realtime notes will be posted here: {NOTES_LINK}.\n\n"
                 "Feel free to add any topics you'd like to discuss in the meeting! 🍓",
                 embed=embed,
             )
 
             add_notification_for_date(event_date, message.id, "public")
 
+    async def on_raw_reaction_add(self, reaction: RawReactionActionEvent) -> None:
+        assert self.user
+
+        if reaction.user_id == self.user.id:
+            return
+
+        if reaction.emoji.name == "✅":
+            await self.on_add_checkmark_reaction(reaction)
+
+    async def on_raw_reaction_remove(self, reaction: RawReactionActionEvent) -> None:
+        assert self.user
+
+        if reaction.user_id == self.user.id:
+            return
+
+        if reaction.emoji.name == "✅":
+            await self.on_remove_checkmark_reaction(reaction)
+
     async def check_events_for_next_week(self):
         await self.wait_until_ready()
         print("Bot is ready!")

src/config.py

@@ -10,3 +10,7 @@
 DISCORD_TOKEN = os.environ["DISCORD_TOKEN"]
 CALENDAR_URL = os.environ["CALENDAR_URL"]
 NOTES_LINK = os.environ["NOTES_LINK"]
+REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID = int(
+    os.environ["REACT_FOR_MEETINGS_NOTIFICATION_MESSAGE_ID"]
+)
+MEETING_WATCHERS_ROLE_ID = int(os.environ["MEETING_WATCHERS_ROLE_ID"])

src/database.py

@@ -67,7 +67,7 @@ def add_notification_for_date(
 
 
 def get_notification_for_discord_message_id(
-    discord_message_id: str, notification_type: str
+    discord_message_id: int, notification_type: str
 ) -> Notification | None:
     cursor.execute(
         "SELECT date, discord_message_id FROM meeting_notifications "