1 files changed, 80 insertions, 0 deletions
diff --git a/code/utils/source_helpers/parse.py b/code/utils/source_helpers/parse.py
new file mode 100644
index 0000000..8489515
--- /dev/null
+++ b/code/utils/source_helpers/parse.py
@@ -0,0 +1,80 @@
+import discord
+
+from utils.source_helpers.apple import (
+    album as apple_album,
+    playlist as apple_playlist,
+    song as apple_song,
+)
+from utils.source_helpers.spotify import (
+    album as spotify_album,
+    playlist as spotify_playlist,
+    song as spotify_song,
+)
+from utils.custom_sources import AppleSource, SpotifySource
+
+
+async def parse_custom_source(
+    self, provider: str, query: str, user: discord.User
+):
+    """
+    Parse the query and run the appropriate functions to get the results/info
+
+    Return the results and an embed or None, None
+    """
+    load_funcs = {
+        "apple": {
+            "album": apple_album.load,
+            "playlist": apple_playlist.load,
+            "song": apple_song.load,
+        },
+        "spotify": {
+            "album": spotify_album.load,
+            "playlist": spotify_playlist.load,
+            "song": spotify_song.load,
+        },
+    }
+
+    headers = {
+        "apple": self.bot.apple_headers,
+        "spotify": self.bot.spotify_headers,
+    }
+
+    sources = {
+        "apple": AppleSource,
+        "spotify": SpotifySource,
+    }
+
+    # Catch all songs
+    if "?i=" in query or "/track/" in query:
+        song, embed = await load_funcs[provider]["song"](
+            headers[provider], query, user
+        )
+
+        if song:
+            results = await sources[provider].load_item(self, user, song)
+        else:
+            return None, embed
+    # Catch all playlists
+    elif "/playlist/" in query:
+        playlist, embed = await load_funcs[provider]["playlist"](
+            headers[provider], query, user
+        )
+
+        if playlist:
+            results = await sources[provider].load_playlist(
+                self, user, playlist
+            )
+        else:
+            return None, embed
+    # Catch all albums
+    elif "/album/" in query:
+        album, embed = await load_funcs[provider]["album"](
+            headers[provider], query, user
+        )
+
+        if album:
+            results = await sources[provider].load_album(self, user, album)
+        else:
+            return None, embed
+
+    return results, embed