musicdlp/backend/main.py

61 lines
1.5 KiB
Python

import asyncio
import json
import websockets
import config
import response
import ydl_pool
type SocketT = websockets.WebSocketServerProtocol
async def handler(socket: SocketT) -> None:
ydls = ydl_pool.Downloader(None, None) # type: ignore # TODO
async for message in socket:
try:
data = json.loads(message)
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']),
))
case 'download': # download by URL
ydls.choose_ydl(data['site'])
ret = await ydls.download(
data['url'],
data.get('items'),
)
await socket.send(response.dl_end(ret))
# 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()
async def main() -> None:
cfg = config.get()
async with websockets.serve(handler, cfg.host, cfg.port):
await asyncio.Future()
if __name__ == '__main__':
asyncio.run(main(), debug=True)