added discord bot

Author: kenzoengineer

.gitignore

@@ -22,3 +22,9 @@ dist-ssr
 *.njsproj
 *.sln
 *.sw?
+
+# Python
+
+__pycache__
+.venv
+.env

bot.sh

@@ -0,0 +1,5 @@
+cd discord_bot
+python -m venv .venv
+source ./venv/Scripts/activate
+pip install -r requirements.txt
+python index.py

discord_bot/cogs/commands.py

@@ -0,0 +1,119 @@
+import discord
+from discord.ext import commands
+from discord import app_commands
+from util.state import StateBot
+from util.embeds import Embeds
+from util.views import JudgeView
+from random import shuffle
+from datetime import date, timedelta
+
+
+class Commands(commands.Cog):
+    def __init__(self, bot):
+        self.bot: StateBot = bot
+        self.TOTAL_SUBMISSIONS = 2
+
+    def get_judge(self) -> list[str, str]:
+        """Returns the judge for the week
+
+        Returns:
+            list[str, str]: the name of the judge, then their id
+        """
+        NAMES = ["Souren", "Kevin", "Ken", "Artom", "Andrew"]
+        IDS = [
+            "290550631591182336",
+            "183383346569543681",
+            "290630197458501633",
+            "161921661346643968",
+            "196069761082327041",
+        ]
+        start_monday: date = date(2025, 6, 9)
+        nearest_monday = date.today() + timedelta((7 - date.today().weekday()) % 7)
+        day_delta = nearest_monday - start_monday
+        week_delta = day_delta.days // 7
+        return [NAMES[week_delta % len(NAMES)], IDS[week_delta % len(IDS)]]
+
+    @app_commands.command(name="submit", description="submit a song to music monday")
+    @app_commands.describe(song="The spotify link for the song")
+    async def submit_song(self, interaction: discord.Interaction, song: str):
+        if song.find("spotify") == -1:
+            return await interaction.response.send_message(
+                embed=Embeds.error(
+                    "Link did not contain the word 'spotify', are you sure you submitted a spotify link?"
+                ),
+                ephemeral=True,
+            )
+
+        self.bot.state.add_submission(interaction.user.id, song)
+
+        if self.TOTAL_SUBMISSIONS == len(self.bot.state.get_all_submissions()):
+            await interaction.channel.send(
+                content=f"<@{self.get_judge()[1]}>, all songs have been submitted, you can now judge with `/submissions`"
+            )
+
+        return await interaction.response.send_message(
+            embed=Embeds.emphasis(title="Submission received", body="Thank you!")
+        )
+
+    @app_commands.command(
+        name="submissions",
+        description="lists submissions blind if 4 have been given, otherwise errors",
+    )
+    async def list_submissions(self, interaction: discord.Interaction):
+        subs = self.bot.state.get_all_submissions()
+        if len(subs) < self.TOTAL_SUBMISSIONS:
+            return await interaction.response.send_message(
+                embed=Embeds.error(
+                    f"We aren't at {self.TOTAL_SUBMISSIONS} submissions yet! So far we have {len(subs)}"
+                ),
+                ephemeral=True,
+            )
+
+        word_map = ["one", "two", "three", "four"]
+
+        # randomize the order of the songs
+        userIds = list(subs.keys())
+        shuffle(userIds)
+
+        for [idx, userId] in enumerate(userIds):
+            await interaction.channel.send(content=f":{word_map[idx]}: {subs[userId]}")
+        return await interaction.response.send_message(
+            content="Judge, please vote for your favourite this song this week!",
+            view=JudgeView(subs, userIds),
+        )
+
+    @app_commands.command(name="reveal", description="Reveals who submitted what song")
+    async def reveal_submissions(self, interaction: discord.Interaction):
+        subs = self.bot.state.get_all_submissions()
+        for userId in subs:
+            await interaction.channel.send(f"<@{userId}> {subs[userId]}")
+        await interaction.channel.send(
+            embed=Embeds.info(
+                title="Here is who sent each song",
+                body="Congrats on another successful music monday",
+            )
+        )
+
+    ## debug commands
+
+    @app_commands.command(
+        name="clearall", description="clears ALL submissions (debug command)"
+    )
+    async def clearall_debug(self, interaction: discord.Interaction):
+        return await interaction.response.send_message(
+            embed=Embeds.info(title="Removed all submissions", body=":wave:")
+        )
+
+    @app_commands.command(
+        name="whojudge", description="Who is the judge this week (debugcommand)"
+    )
+    async def whojudge_debug(self, interaction: discord.Interaction):
+        return await interaction.response.send_message(
+            embed=Embeds.info(
+                title="The next judge is", body=f"# {self.get_judge()[0]}!"
+            )
+        )
+
+
+async def setup(bot):
+    await bot.add_cog(Commands(bot))

discord_bot/index.py

@@ -0,0 +1,40 @@
+import discord
+import os
+from dotenv import load_dotenv
+from discord.ext import commands
+from util.state import State
+
+load_dotenv()
+
+# placeholder command prefix; We only use slash commands
+client = commands.Bot(command_prefix="<>", intents=discord.Intents.all())
+
+# load state
+
+client.state = State()
+
+
+# load all cogs
+async def load():
+    for f in os.listdir("./cogs"):
+        if f.endswith(".py"):
+            await client.load_extension(f"cogs.{f[:-3]}")
+
+
+@client.event
+async def on_ready():
+    print("Bot is ready")
+    try:
+        synched_commands = await client.tree.sync()
+        print(f"Synched {len(synched_commands)} commands")
+    except Exception as e:
+        print("Error synching")
+
+
+async def setup_bot():
+    await load()
+
+
+if __name__ == "__main__":
+    client.setup_hook = setup_bot
+    client.run(os.getenv("TOKEN"))

discord_bot/requirements.txt

@@ -0,0 +1,13 @@
+aiohappyeyeballs==2.6.1
+aiohttp==3.12.13
+aiosignal==1.4.0
+attrs==25.3.0
+audioop-lts==0.2.1
+discord.py==2.5.2
+frozenlist==1.7.0
+idna==3.10
+multidict==6.6.3
+propcache==0.3.2
+python-dotenv==1.1.1
+tzdata==2025.2
+yarl==1.20.1

discord_bot/util/embeds.py

@@ -0,0 +1,66 @@
+import discord
+
+
+class Embeds:
+
+    ### Generic embeds
+
+    @staticmethod
+    def error(body: str):
+        return discord.Embed(
+            colour=discord.Colour.red(), title="Error!", description=body
+        )
+
+    @staticmethod
+    def info(title: str, body: str, footer: str = ""):
+        embed = discord.Embed(
+            colour=discord.Colour.blue(), title=title, description=body
+        )
+        if len(footer):
+            embed.set_footer(text=footer)
+        return embed
+
+    @staticmethod
+    def megainfo(
+        title: str,
+        body: str,
+        fields: list[tuple[str, str, bool]] = [],
+        footer: str = "",
+        img: str = "",
+        thumbnail: str = "",
+        author: str = "",
+    ):
+        embed = discord.Embed(
+            colour=discord.Colour.orange(),
+            title=title,
+            description=body,
+        )
+        for field in fields:
+            embed.add_field(name=field[0], value=field[1], inline=field[2])
+
+        if len(footer):
+            embed.set_footer(text=footer)
+        if len(img):
+            embed.set_image(url=img)
+        if len(thumbnail):
+            embed.set_thumbnail(url=thumbnail)
+        if len(author):
+            embed.set_author(name=author)
+        return embed
+
+    @staticmethod
+    def emphasis(title: str, body: str, footer: str = ""):
+        embed = discord.Embed(
+            color=discord.Colour.pink(), title=title, description=body
+        )
+        if len(footer):
+            embed.set_footer(text=footer)
+        return embed
+
+    ### Specific embeds
+
+    @staticmethod
+    def word_not_found(word: str):
+        return Embeds.error(
+            f"You aren't tracking __'{word}'__. Please choose a different word."
+        )

discord_bot/util/state.py

@@ -0,0 +1,45 @@
+from datetime import datetime
+from zoneinfo import ZoneInfo
+from discord.ext import commands
+
+type submissions_map = dict[str, str]
+
+
+class State:
+    """Main state holder for the bot
+
+    No safeguards, assumes the following things:
+        - All submissions take place over the course of one day
+        - Music monday takes place from 00:00 - 23:59 in EST
+    """
+
+    def __init__(self):
+        self.submission_date = None
+        self.submissions: submissions_map = dict()
+
+    def get_date_today(self):
+        return datetime.now(ZoneInfo("America/New_York")).date()
+
+    def set_date_today(self):
+        self.submission_date = self.get_date_today()
+
+    def add_submission(self, user_id: str, song_link: str):
+        if self.submission_date == None:
+            self.set_date_today()
+
+        # reset submissions upon a new day
+        if self.submission_date != self.get_date_today():
+            self.set_date_today()
+            self.submissions = dict()
+
+        self.submissions[user_id] = song_link
+
+    def get_all_submissions(self):
+        return self.submissions
+
+    def clear_all_submissions(self):
+        self.submissions = dict()
+
+
+class StateBot(commands.Bot):
+    state: State

discord_bot/util/views.py

@@ -0,0 +1,28 @@
+import discord
+from discord.ext import commands
+
+from util.embeds import Embeds
+
+
+class JudgeView(discord.ui.View):
+    def __init__(self, songs: dict[str, str], userIds: list[str]):
+        super().__init__()
+        self.songs = songs
+        for idx, userId in enumerate(userIds):
+            self.add_item(SongButton(str(idx + 1), userId, songs))
+
+
+class SongButton(discord.ui.Button):
+    def __init__(self, label: str, key: str, songs: dict[str, str]):
+        super().__init__(label=label, style=discord.ButtonStyle.primary)
+        self.index = key
+        self.songs = songs
+
+    async def callback(self, interaction: discord.Interaction):
+        await interaction.response.send_message(
+            embed=Embeds.emphasis(
+                title="Congratulations to the winner!", body=f"<@{self.index}>"
+            )
+        )
+        await interaction.channel.send(content=f"{self.songs[self.index]}")
+        self.view.stop()