backend/main.py

@@ -6,14 +6,16 @@ import websockets
 import config
 import response
 import ydl_pool
+import ydl_wrap
 
 
 type SocketT = websockets.WebSocketServerProtocol
 
 
+
 async def handler(socket: SocketT) -> None:
 
-    ydls = ydl_pool.Downloader(None, None)  # type: ignore  # TODO
+    ydls = ydl_pool.Downloaders()
 
     async for message in socket:
 
@@ -23,18 +25,20 @@ async def handler(socket: SocketT) -> None:
             match data['action']:
 
                 case 'list':  # list tracks in album
-                    ydls.choose_ydl(data['site'])
-                    await socket.send(response.playlist(
-                        await ydls.get_playlist_items(data['url']),
+                    await socket.send(response.ok_playlist(
+                        await ydl_wrap.get_playlist_items(
+                            ydls.get_ydl(data['site']),
+                            data['url'],
+                        )
                     ))
 
                 case 'download':  # download by URL
-                    ydls.choose_ydl(data['site'])
-                    ret = await ydls.download(
+                    ret = await ydl_wrap.download(
+                        ydls.get_ydl(data['site']),
                         data['url'],
                         data.get('items'),
                     )
-                    await socket.send(response.dl_end(ret))
+                    await socket.send(response.ok_downloaded(ret))
 
                 # TODO: cancellation
 

backend/response.py

@@ -1,29 +1,25 @@
 import json
 import traceback
 
-def playlist(items: list[str]) -> str:
+OK = '{"ok":true}'
+
+def ok_playlist(items: list[str]) -> str:
     return json.dumps({
-        "type": "items",
+        "ok": True,
         "data": items,
     })
 
-def dl_progress(msg: str) -> str:
+def ok_downloaded(ret: int) -> str:
     return json.dumps({
-        "type": "dl_progress",
-        "data": msg,
-    })
-
-def dl_end(ret: int) -> str:
-    return json.dumps({
-        "type": "dl_end",
+        "type": "downloaded",
         "data": ret,
     })
 
 def error(ex: Exception) -> str:
     traceback.print_tb(ex.__traceback__)
     return json.dumps({
-        "type": "error",
-        "data": {
+        "ok": False,
+        "error": {
             "type": ex.__class__.__qualname__,
             "message": str(ex),
         }

backend/ydl_pool.py

@@ -1,6 +1,3 @@
-import asyncio
-from typing import Callable, Awaitable, Iterable
-
 from yt_dlp import YoutubeDL
 from yt_dlp.postprocessor import FFmpegExtractAudioPP
 
@@ -37,12 +34,9 @@ create_ydl_fn = {
 ydl_fn_keys = create_ydl_fn.keys()
 
 
-class Downloader:
+class Downloaders:
 
-    def __init__(
-            self,
-            progress_cb: Callable[[str], Awaitable],
-            lyrics_cb: Callable[[list[str]], Awaitable]) -> None:
+    def __init__(self) -> None:
 
         self.ydls: dict[str, YoutubeDL | None] = {
             'youtube': None,
@@ -50,15 +44,10 @@ class Downloader:
             'yandex': None,
         }
 
-        self.cur_ydl: YoutubeDL | None = None
+    def get_ydl(self, site: str) -> YoutubeDL:
 
-        self.progress_cb = progress_cb
-        self.lyrics_cb = lyrics_cb
-
-    def choose_ydl(self, site: str) -> None:
-
-        ydl = self.ydls[site]
         cfg = config.get()
+        ydl = self.ydls[site]
 
         if ydl is None:
             ydl = create_ydl_fn[site]()
@@ -71,62 +60,8 @@ class Downloader:
             if cookies.exists():
                 ydl.params['cookiefile'] = str(cookies)
 
-        self.cur_ydl = ydl
-
-    def get_cur_ydl(self) -> YoutubeDL:
-
-        ydl = self.cur_ydl
-        if ydl is None:
-            raise RuntimeError('ydl object not initialized')
         return ydl
 
-    async def get_playlist_items(self, url: str) -> list[str]:
-
-        return await asyncio.get_event_loop().run_in_executor(
-            None,
-            Downloader._target_get_playlist_items,
-            self.get_cur_ydl(),
-            url,
-        )
-
-    @staticmethod
-    def _target_get_playlist_items(ydl: YoutubeDL, url: str) -> list[str]:
-
-        info = ydl.extract_info(url, download=False, process=True)
-        if info is None:
-            raise RuntimeError('ydl.extract_info returned None')
-        return [
-            entry['track'] if 'track' in entry else entry['title']
-            for entry in info['entries']
-        ]
-
-    async def download(
-            self,
-            url: str,
-            playlist_items: Iterable[int] | None = None) -> int:
-
-        return await asyncio.get_event_loop().run_in_executor(
-            None,
-            Downloader._target_download,
-            self.get_cur_ydl(),
-            url,
-            playlist_items,
-        )
-
-    @staticmethod
-    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
-
     def cleanup(self) -> None:
 
         for ydl in self.ydls.values():

backend/ydl_wrap.py

@@ -0,0 +1,53 @@
+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=True)
+    if info is None:
+        raise RuntimeError('ydl.extract_info returned None')
+    return [
+        entry['track'] if 'track' in entry else 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