Add soundcloud support

2023-11-13 23:08:58 +02:00
parent e99ba9daa3
commit bc2663c17c
22 changed files with 391 additions and 7 deletions
--- a/bot/modules/soundcloud/downloader.py
+++ b/bot/modules/soundcloud/downloader.py
@@ -0,0 +1,118 @@
+from attrs import define
+from typing import Callable
+
+from .driver import SoundCloudDriver
+from .song import SongItem
+
+import m3u8
+
+
+@define
+class SoundCloudBytestream:
+    file: bytes
+    filename: str
+    duration: int
+    song: SongItem
+
+    @classmethod
+    def from_bytes(
+            cls,
+            bytes_: bytes,
+            filename: str,
+            duration: int,
+            song: SongItem
+    ):
+        return cls(
+            file=bytes_,
+            filename=filename,
+            duration=int(duration / 1000),
+            song=song
+        )
+
+
+@define
+class Downloader:
+    driver: SoundCloudDriver
+    download_url: str
+    duration: int
+    filename: str
+    method: Callable
+    song: SongItem
+
+    @classmethod
+    async def build(
+            cls,
+            song_id: str,
+            driver: SoundCloudDriver
+    ):
+        track = await driver.get_track(song_id)
+        song = SongItem.from_soundcloud(track)
+
+        if url := cls._try_get_progressive(track['media']['transcodings']):
+            method = cls._progressive
+        else:
+            url = track['media']['transcodings'][0]['url']
+            method = cls._hls if \
+                (track['media']['transcodings'][0]['format']['protocol']
+                 == 'hls') else cls._progressive
+
+        return cls(
+            driver=driver,
+            duration=track['duration'],
+            method=method,
+            download_url=url,
+            filename=f'{track["title"]}.mp3',
+            song=song
+        )
+
+    @staticmethod
+    def _try_get_progressive(urls: list) -> str | None:
+        for transcode in urls:
+            if transcode['format']['protocol'] == 'progressive':
+                return transcode['url']
+
+    async def _progressive(self, url: str) -> bytes:
+        return await self.driver.engine.read_data(
+            url=(await self.driver.engine.get(
+                url
+            ))['url']
+        )
+
+    async def _hls(self, url: str) -> bytes:
+        m3u8_obj = m3u8.loads(
+            (await self.driver.engine.read_data(
+                (await self.driver.engine.get(
+                    url=url
+                ))['url']
+            )).decode()
+        )
+
+        content = bytearray()
+        for segment in m3u8_obj.files:
+            content.extend(
+                await self.driver.engine.read_data(
+                    url=segment,
+                    append_client_id=False
+                )
+            )
+
+        return content
+
+    async def to_bytestream(self) -> SoundCloudBytestream:
+        return SoundCloudBytestream.from_bytes(
+            bytes_=await self.method(self, self.download_url),
+            filename=self.filename,
+            duration=self.duration,
+            song=self.song
+        )
+
+
+@define
+class DownloaderBuilder:
+    driver: SoundCloudDriver
+
+    async def from_id(self, song_id: str):
+        return await Downloader.build(
+            song_id=song_id,
+            driver=self.driver
+        )