mudae series tracking

alembic/versions/5bee6f5c272e_add_mudae_series_map.py

@@ -0,0 +1,36 @@
+"""Add Mudae series Map
+
+Revision ID: 5bee6f5c272e
+Revises: ca032db10a65
+Create Date: 2024-06-17 19:43:08.163862
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '5bee6f5c272e'
+down_revision: Union[str, None] = 'ca032db10a65'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        'mudaeSeriesRequests',
+        sa.Column('series', sa.String(), nullable=False),
+        sa.Column('requestedBy', sa.BIGINT(), nullable=False),
+        sa.PrimaryKeyConstraint('series', 'requestedBy'),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('mudaeSeriesRequests')
+    # ### end Alembic commands ###

alexBot/cogs/mudae.py

@@ -1,31 +1,58 @@
+import asyncio
 import logging
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, List
 
 import discord
+from discord import app_commands
 from datetime import datetime, timedelta
 
+import regex
+import sqlalchemy
+
 from ..tools import Cog
 
 from typing import Optional
 
+from alexBot import database as db
+
 if TYPE_CHECKING:
     from bot import Bot
 
 
 log = logging.getLogger(__name__)
 
+GAMESERVER = 1069353044808056902
+PRIMARY_COMMAND_CHANNEL = 1077272164887183450
+MENTION_ROLE = 1251159714532691979
+MUDAE_BOT = 432610292342587392
+
+SERIES_REGEX = regex.compile(r'\*\*\d{1,3} - (.+)\*\*')
+
 
 class Mudae(Cog):
     def __init__(self, bot: "Bot"):
         super().__init__(bot)
         self.lastPinged: Optional[datetime] = None
         self.lastMessage: Optional[datetime] = None
+        self.seriesExtractMenu = app_commands.ContextMenu(
+            name='Extract Liked Series',
+            callback=self.extract_series,
+        )
+
+    async def cog_load(self) -> None:
+        self.bot.tree.add_command(
+            self.seriesExtractMenu,
+            guilds=[
+                discord.Object(GAMESERVER),
+            ],
+        )
 
     @Cog.listener()
     async def on_message(self, message: discord.Message):
-        if message.channel.id != 1077272164887183450:
+        if message.channel.id != PRIMARY_COMMAND_CHANNEL:
             return
         if message.content.lower() in ["$m", "$mg", "$ma"]:
+            self.bot.loop.create_task(self.message_series_detector(message))
             if self.lastMessage is None:
                 self.lastMessage = datetime.now()
             elif datetime.now() - self.lastMessage < timedelta(seconds=60):
@@ -37,7 +64,100 @@ async def on_message(self, message: discord.Message):
             elif datetime.now() - self.lastPinged < timedelta(minutes=5):
                 return
             self.lastPinged = datetime.now()
-            await message.channel.send("<@&1251159714532691979>", allowed_mentions=discord.AllowedMentions(roles=True))
+            await message.channel.send(f"<@&{MENTION_ROLE}>", allowed_mentions=discord.AllowedMentions(roles=True))
+
+    async def message_series_detector(self, message: discord.Message):
+        msg = await self.bot.wait_for(
+            "message",
+            check=lambda m: m.author.id == MUDAE_BOT and m.channel.id == PRIMARY_COMMAND_CHANNEL and len(m.embeds) == 1,
+        )
+        # get the series name from the embed
+        description = msg.embeds[0].description
+        assert description
+        is_collectable = "React with any emoji to claim!" in description
+        series_name = description.split("\n")[0]
+        if is_collectable:
+            async with db.async_session() as session:
+                matches = await session.scalars(
+                    sqlalchemy.select(db.MudaeSeriesRequest).where(db.MudaeSeriesRequest.series == series_name)
+                )
+                if matches:
+                    mentions = [f"<@{match.requestedBy}>" for match in matches]
+                    await msg.reply(
+                        f"Series Liked by {', '.join(mentions)}", allowed_mentions=discord.AllowedMentions(users=True)
+                    )
+
+    @staticmethod
+    def get_series_name_from_description(series: str) -> List[str]:
+        return [match.groups()[0] for match in SERIES_REGEX.findall(series)]
+
+    async def extract_series(self, interaction: discord.Interaction, message: discord.Message):
+        # user will be from the interaction
+        # series can be extracted from the message clicked on
+        # need to loop on message updates, check footer for competion 'Page n / n'
+        # there may be no footer, can validated liked series page by checking author field
+
+        if message.author.id != MUDAE_BOT:
+            await interaction.response.send_message(
+                "you need to run $ml and right click *that* message", ephemeral=True
+            )
+        # it's a mudae message! check if it's a liked series message
+        if not message.embeds or not message.embeds[0].author or not message.embeds[0].author.name:
+            await interaction.response.send_message("This message does not contain an embed.", ephemeral=True)
+        if 'Liked Series' not in message.embeds[0].author.name:  # type: ignore
+            await interaction.response.send_message(
+                "This is not a $ml response. please right click on the message from $ml.", ephemeral=True
+            )
+
+        serieses: List[str] = []
+        # it's a message we care about! do we have a paginator?
+        if message.embeds[0].footer:
+            assert message.embeds[0].footer.text is str
+            if 'Page' not in message.embeds[0].footer.text:
+                await interaction.response.send_message("This shound not happen...", ephemeral=True)
+            # extract page details:
+            regex_result = regex.match(r'Page (\d+) / (\d+)', message.embeds[0].footer.text)
+            if not regex_result:
+                await interaction.response.send_message("This should not happen...", ephemeral=True)
+                return
+            current_page, total_pages = regex_result.groups()
+            current_page, total_pages = int(current_page), int(total_pages)
+            if current_page != 1:
+                await interaction.response.send_message("Please start from the first page.", ephemeral=True)
+                return
+            serieses.extend(self.get_series_name_from_description(message.embeds[0].description))  # type: ignore
+            # ^ captures first page
+            while current_page < total_pages:
+
+                await interaction.response.send_message(
+                    "tab through your liked series, i'll save them!", ephemeral=True
+                )
+                # get the next page
+                before, after = await self.bot.wait_for(
+                    "message_update",
+                    check=lambda before, after: before.id == message.id
+                    and after.embeds[0].footer.text == f"Page {current_page + 1} / {total_pages}",
+                    timeout=60,
+                )
+                serieses.extend(self.get_series_name_from_description(after.embeds[0].description))
+                # captures pages 2 thru n
+                current_page += 1
+                if current_page == total_pages:
+                    break
+        else:
+            serieses.extend(self.get_series_name_from_description(message.embeds[0].description))  # type: ignore
+            # captures single page
+
+        async with db.async_session() as session:
+            await session.delete(
+                sqlalchemy.select(db.MudaeSeriesRequest).where(db.MudaeSeriesRequest.requestedBy == interaction.user.id)
+            )
+            for series in serieses:
+                await session.merge(db.MudaeSeriesRequest(series=series, requestedBy=interaction.user.id))
+        if interaction.response.is_done():
+            await interaction.followup.send("added your liked series!", ephemeral=True)
+        else:
+            await interaction.response.send_message("added your liked series!", ephemeral=True)
 
 
 async def setup(bot: "Bot"):

alexBot/database.py

@@ -184,6 +184,12 @@ class SugeryUser(Base):
     )
 
 
+class MudaeSeriesRequest(Base):
+    __tablename__ = "mudaeSeriesRequests"
+    series: Mapped[str] = mapped_column(String(), primary_key=True)
+    requestedBy: Mapped[int] = mapped_column(BIGINT(), nullable=False, primary_key=True)
+
+
 user = config.db_user
 pw = config.db_pw
 db = config.db_name

docker-compose.yml

@@ -7,8 +7,6 @@ services:
     restart: unless-stopped
     depends_on:
       - db
-    environment:
-
     env_file:
       - .env
   db: