feat: add playlists

mafic/node.py

@@ -11,6 +11,7 @@
 
 from .__libraries import ExponentialBackoff, dumps, loads
 from .errors import TrackLoadException
+from .playlist import Playlist
 from .track import Track
 
 if TYPE_CHECKING:
@@ -368,7 +369,7 @@ async def __request(
 
     async def fetch_tracks(
         self, query: str, *, search_type: str
-    ) -> list[Track] | None:  # TODO: | Playlist
+    ) -> list[Track] | Playlist | None:
         if not URL_REGEX.match(query):
             query = f"{search_type}:{query}"
 
@@ -385,7 +386,7 @@ async def fetch_tracks(
             return [Track(**data["tracks"][0])]
         elif data["loadType"] == "PLAYLIST_LOADED":
             # TODO: handle playlists
-            ...
+            return Playlist(info=data["playlistInfo"], tracks=data["tracks"])
         elif data["loadType"] == "SEARCH_RESULT":
             return [Track(**track) for track in data["tracks"]]
         elif data["loadType"] == "LOAD_FAILED":

mafic/playlist.py

@@ -0,0 +1,19 @@
+# SPDX-License-Identifier: MIT
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from .track import Track
+
+if TYPE_CHECKING:
+    from .typings import PlaylistInfo, TrackWithInfo
+
+__all__ = ("Playlist",)
+
+
+class Playlist:
+    def __init__(self, *, info: PlaylistInfo, tracks: list[TrackWithInfo]):
+        self.name: str = info["name"]
+        self.selected_track: int = info["selectedTrack"]
+        self.tracks: list[Track] = [Track(**track) for track in tracks]