61 lines
1.5 KiB
Python
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)
|