1 files changed, 96 insertions, 0 deletions
diff --git a/code/utils/command_tree.py b/code/utils/command_tree.py
new file mode 100644
index 0000000..504d3a9
--- /dev/null
+++ b/code/utils/command_tree.py
@@ -0,0 +1,96 @@
+import discord
+from discord import app_commands
+from discord.ext.commands.errors import *
+import datetime
+
+from config import BOT_COLOR
+from custom_sources import LoadError
+
+
+# Create a custom AppCommandError for the create_player function
+class CheckPlayerError(app_commands.AppCommandError):
+    def __init__(self, info):
+        self.info = info
+
+
+class Tree(app_commands.CommandTree):
+    async def on_error(
+        self, interaction: discord.Interaction, error: app_commands.AppCommandError
+    ):
+        music_commands = [
+            "play",
+            "clear",
+            "np",
+            "pause",
+            "queue",
+            "remove",
+            "resume",
+            "shuffle",
+            "skip",
+            "stop",
+        ]
+
+        if isinstance(error, CommandNotFound):
+            return
+
+        # Custom Error class for the `create_player` function
+        # Issues that arise may be user not in vc, user not in correct vc, missing perms, etc.
+        elif isinstance(error, CheckPlayerError):
+            embed = discord.Embed(
+                title=error.info["title"],
+                description=error.info["description"],
+                color=BOT_COLOR,
+            )
+            embed.set_footer(
+                text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                    "%Y-%m-%d %H:%M:%S"
+                )
+                + " UTC"
+            )
+            try:
+                await interaction.response.send_message(embed=embed, ephemeral=True)
+            except discord.errors.InteractionResponded:
+                await interaction.followup.send(embed=embed, ephemeral=True)
+
+        # If `create_player` fails to create a player and fails
+        # to raise a `CheckPlayerError`, this will catch it
+        elif (
+            isinstance(error, app_commands.CheckFailure)
+            and interaction.command.name in music_commands
+        ):
+            embed = discord.Embed(
+                title="Player Creation Error",
+                description="An error occured when trying to create a player. Please submit a bug report with </bug:1224840889906499626> if this issue persists.",
+                color=BOT_COLOR,
+            )
+            embed.set_footer(
+                text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                    "%Y-%m-%d %H:%M:%S"
+                )
+                + " UTC"
+            )
+            try:
+                await interaction.response.send_message(embed=embed, ephemeral=True)
+            except discord.errors.InteractionResponded:
+                await interaction.followup.send(embed=embed, ephemeral=True)
+
+        # If a Spotify song is linked but cannot be found on a provider (e.g. YouTube)
+        elif isinstance(error, LoadError):
+            embed = discord.Embed(
+                title="Nothing Found",
+                description="Spotify does not allow direct play, meaning songs have to be found on a supported provider. In this case, the song couldn't be found. Please try again with a different song, or try searching for just the name and artist manually rather than sending a link.",
+                color=BOT_COLOR,
+            )
+            embed.set_footer(
+                text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                    "%Y-%m-%d %H:%M:%S"
+                )
+                + " UTC"
+            )
+            try:
+                await interaction.response.send_message(embed=embed, ephemeral=True)
+            except discord.errors.InteractionResponded:
+                await interaction.followup.send(embed=embed, ephemeral=True)
+
+        else:
+            raise error