backend/id3pp.py

@@ -20,8 +20,6 @@ class InfoYouTubePP(PostProcessor):
             information['track'] = information['title']
         if not 'artist' in information:
             information['artist'] = information['channel'].removesuffix(' - Topic')
-        if not 'artists' in information:
-            information['artists'] = [information['artist']]
 
         if not 'album' in information:
             try:
@@ -49,10 +47,10 @@ class ID3TagsPP(PostProcessor):
         file = mp3.MP3(information['filepath'])
 
         title = information['track']
-        artists: list[str] = information['artists']
+        artist = information['artist']
 
         file['TIT2'] = id3.TIT2(encoding=ENC_UTF8, text=title)
-        file['TPE1'] = id3.TPE1(encoding=ENC_UTF8, text=artists)  # multiple values are null-separated
+        file['TPE1'] = id3.TPE1(encoding=ENC_UTF8, text=artist)
         if 'album' in information:
             file['TALB'] = id3.TALB(encoding=ENC_UTF8, text=information['album'])
         if 'release_year' in information:
@@ -63,7 +61,7 @@ class ID3TagsPP(PostProcessor):
             file['TCON'] = id3.TCON(encoding=ENC_UTF8, text=information['genre'])
 
         try:
-            lyr_url = genius.search(title, artists[0])
+            lyr_url = genius.search(title, artist)
             file['USLT'] = id3.USLT(encoding=ENC_UTF8, text=genius.parse(lyr_url))
         except:
             pass

backend/main.py

@@ -1,5 +1,6 @@
 import asyncio
 import json
+from typing import Any
 
 import secrets
 import time
@@ -7,62 +8,48 @@ import time
 import websockets
 
 import response
-import ydl_pool
-import ydl_wrap
 
 
 # TODO: config
 HOST = '127.0.0.1'
 PORT = 5678
 
+# TODO: make dict with functions creating YoutubeDL for each site instead of this set
+SITES = {'youtube', 'yandex'}
+
+sessions = {}
+
 
 def generate_key() -> str:
-
     return hex(time.time_ns()) + secrets.token_hex(2)
 
 
 async def handler(socket: websockets.WebSocketServerProtocol) -> None:
-
-    ydls = ydl_pool.Downloaders()
-
-    async for message in socket:
-
-        try:
-            data = json.loads(message)
-
-            match data['action']:
-
-                case 'list':  # list tracks in album
-                    await socket.send(response.ok_playlist(
-                        await ydl_wrap.get_playlist_items(
-                            ydls.get_ydl(data['site']),
-                            data['url'],
-                        )
-                    ))
-
-                case 'download':  # download by URL
-                    # TODO: send all yt-dlp's output to client
-                    await socket.send(response.ok_download(
-                        await ydl_wrap.download(
-                            ydls.get_ydl(data['site']),
-                            data['url'],
-                            data.get('items'),
-                        )
-                    ))
-
-                # TODO: cancellation
-
-                case _:
-                    raise ValueError('invalid "action" field value')
-
-        except websockets.ConnectionClosed:
-            break
-
-        except Exception as ex:
-            if not socket.closed:
-                await socket.send(response.error(ex))
-
-    ydls.cleanup()
+    data = json.loads(await socket.recv())
+    if 'action' not in data:
+        await socket.send(response.error_field('action'))
+        return
+    match data['action']:
+        case 'init':  # create session
+            if data.get('site') not in SITES:
+                await socket.send(response.error_field('site'))
+                return
+            key = generate_key()
+            sessions[key] = ""  # TODO
+            await socket.send(response.ok_init(key))
+        case 'list':  # list tracks in album
+            if 'url' not in data:
+                await socket.send(response.error_field('url'))
+                return
+            await socket.send(response.ok_playlist(['title 1', 'title 2']))  # TODO
+        case 'download':  # download by URL
+            if 'url' not in data:
+                await socket.send(response.error_field('url'))
+                return
+            # TODO: pass command to the thread started in `init`
+            await socket.send(response.OK)
+        case _:
+            await socket.send(response.error_field('action'))
 
 
 async def main() -> None:

backend/response.py

@@ -1,26 +1,22 @@
 import json
-import traceback
 
 OK = '{"ok":true}'
 
+def ok_init(key: str) -> str:
+    return json.dumps({
+        "ok": True,
+        "data": key,
+    })
+
 def ok_playlist(items: list[str]) -> str:
     return json.dumps({
         "ok": True,
         "data": items,
     })
 
-def ok_download(ret: int) -> str:
-    return json.dumps({
-        "ok": True,
-        "data": ret,
-    })
-
-def error(ex: Exception) -> str:
-    traceback.print_tb(ex.__traceback__)
+def error_field(field: str) -> str:
     return json.dumps({
         "ok": False,
-        "error": {
-            "type": ex.__class__.__qualname__,
-            "message": str(ex),
-        }
+        "error": "field",
+        "data": field,
     })

backend/test_pp.py

@@ -25,7 +25,6 @@ INFO_AFTER = {
     **INFO_BEFORE,
     "track": "A Line in the Sand",
     "artist": "Linkin Park",
-    "artists": ["Linkin Park"],
     "album": "The Hunting Party",
     "track_number": 12,
 }
@@ -40,7 +39,7 @@ class TestPostProcessorsOnFakeData(TestCase):
     def test_infoytpp(self) -> None:
 
         _, info = id3pp.InfoYouTubePP().run(INFO_BEFORE)
-        self.assertDictEqual(info, INFO_AFTER)
+        self.assertEqual(info, INFO_AFTER)
 
     def test_id3(self) -> None:
 

backend/ydl_pool.py

@@ -1,61 +0,0 @@
-from yt_dlp import YoutubeDL
-from yt_dlp.postprocessor import FFmpegExtractAudioPP
-
-import id3pp
-
-
-class _CreateYDL:
-
-    @staticmethod
-    def youtube() -> YoutubeDL:
-        ydl = YoutubeDL({'format': 'ba'})
-        ydl.add_post_processor(id3pp.InfoYouTubePP(), when='before_dl')
-        ydl.add_post_processor(FFmpegExtractAudioPP(preferredcodec='mp3'), when='post_process')
-        return ydl
-
-    @staticmethod
-    def yt_proxied() -> YoutubeDL:
-        ydl = _CreateYDL.youtube()
-        ydl.params['proxy'] = 'http://127.0.0.1:1080'  # TODO
-        return ydl
-
-    @staticmethod
-    def yandex() -> YoutubeDL:
-        return YoutubeDL()  # TODO: cookies
-
-
-create_ydl_fn = {
-    'youtube': _CreateYDL.youtube,
-    'yt_proxied': _CreateYDL.yt_proxied,
-    'yandex': _CreateYDL.yandex,
-}
-
-ydl_fn_keys = create_ydl_fn.keys()
-
-
-class Downloaders:
-
-    def __init__(self) -> None:
-
-        self.ydls: dict[str, YoutubeDL | None] = {
-            'youtube': None,
-            'yt_proxied': None,
-            'yandex': None,
-        }
-
-    def get_ydl(self, site: str) -> YoutubeDL:
-
-        ydl = self.ydls[site]
-        if ydl is None:
-            ydl = create_ydl_fn[site]()
-            ydl.params['trim_file_name'] = 40  # TODO: config
-            # artists.0 instead of artist, because it can contain "feat. ..."
-            ydl.params['outtmpl']['default'] = 'music/%(artists.0)s/%(album)s/%(track)s.%(ext)s'
-            ydl.add_post_processor(id3pp.ID3TagsPP(), when='post_process')
-        return ydl
-
-    def cleanup(self) -> None:
-
-        for ydl in self.ydls.values():
-            if ydl is not None:
-                ydl.close()

backend/ydl_wrap.py

@@ -1,33 +0,0 @@
-import asyncio
-from typing import Iterable
-
-from yt_dlp import YoutubeDL
-
-
-async def get_playlist_items(ydl: YoutubeDL, url: str) -> list[str]:
-
-    return await asyncio.get_event_loop().run_in_executor(None, _target_get_playlist_items, ydl, url)
-
-
-def _target_get_playlist_items(ydl: YoutubeDL, url: str) -> list[str]:
-
-    info = ydl.extract_info(url, download=False, process=False)
-    if info is None:
-        raise RuntimeError('ydl.extract_info returned None')
-    return [entry['title'] for entry in info['entries']]
-
-
-async def download(ydl: YoutubeDL, url: str, playlist_items: Iterable[int] | None = None) -> int:
-
-    return await asyncio.get_event_loop().run_in_executor(None, _target_download, ydl, url, playlist_items)
-
-
-def _target_download(ydl: YoutubeDL, url: str, playlist_items: Iterable[int] | None = None) -> int:
-
-    if playlist_items:
-        ydl.params['playlist_items'] = ','.join(str(i) for i in playlist_items)
-
-    ret = ydl.download(url)
-    del ydl.params['playlist_items']
-
-    return ret