Support FLAC audio playback and file lookup

votify/interface/audio.py

@@ -51,7 +51,7 @@ class SpotifyAudioInterface(SpotifyBaseInterface):
             audio_track=StreamInfo(
                 stream_url=stream_url,
                 widevine_pssh=pssh,
-                file_format="mp4",
+                file_format="flac" if audio_quality == AudioQuality.FLAC else "mp4",
             ),
         )
 
@@ -67,12 +67,15 @@ class SpotifyAudioInterface(SpotifyBaseInterface):
         playback_info: dict,
         format_id: str,
     ) -> str | None:
+        manifest_key = (
+            "file_ids_mp4flac"
+            if self.audio_quality == AudioQuality.FLAC
+            else "file_ids_mp4"
+        )
         file_id = next(
             (
                 file_info["file_id"]
-                for file_info in playback_info.get("manifest", {}).get(
-                    "file_ids_mp4", []
-                )
+                for file_info in playback_info.get("manifest", {}).get(manifest_key, [])
                 if file_info["format"] == format_id
             ),
             None,
@@ -95,7 +98,7 @@ class SpotifyAudioInterface(SpotifyBaseInterface):
         file_id: str,
     ) -> str:
         seek_table_response = await self.api.get_seek_table(file_id)
-        pssh = seek_table_response["pssh"]
+        pssh = seek_table_response["pssh_widevine"]
 
         logger.debug(f"Received PSSH: {pssh}")
 

votify/interface/base.py

@@ -105,10 +105,19 @@ class SpotifyBaseInterface:
     def is_video(self, playback_info: dict) -> bool:
         return bool(playback_info["manifest"].get("manifest_ids_video"))
 
-    async def get_playback_info(self, media_id: str, media_type: str) -> dict | None:
+    async def get_playback_info(
+        self,
+        media_id: str,
+        media_type: str,
+        flac: bool = False,
+    ) -> dict | None:
         playback_info_response = await self.api.get_playback_info(
             media_id=media_id,
             media_type=media_type,
+            file_formats=[
+                "manifest_ids_video",
+                "file_ids_mp4flac" if flac else "file_ids_mp4",
+            ],
         )
 
         playback_info_key = next(iter(playback_info_response.get("media", {})), None)

votify/interface/interface.py

@@ -5,7 +5,7 @@ from InquirerPy import inquirer
 from InquirerPy.base.control import Choice
 
 from .audio import SpotifyAudioInterface
-from .enums import ArtistMediaOption
+from .enums import ArtistMediaOption, AudioQuality
 from .episode import SpotifyEpisodeInterface
 from .episode_video import SpotifyEpisodeVideoInterface
 from .exceptions import (
@@ -61,6 +61,7 @@ class SpotifyInterface:
         playback_info = await self.base.get_playback_info(
             media_id=track_id,
             media_type="track",
+            flac=self.song.audio_quality == AudioQuality.FLAC,
         )
         assert playback_info, "Playback info should be available for playable track"