115 lines
3 KiB
Python
115 lines
3 KiB
Python
import asyncio
|
|
|
|
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
|
from models import MusicPlayer, PlayerState, WSConnectionType
|
|
from services import ConnectionManager, DownloadService
|
|
|
|
# Setup
|
|
tags_metadata = [
|
|
{"name": "player", "description": "Interact with the Music Player"},
|
|
{"name": "experimental"},
|
|
{"name": "queue"},
|
|
]
|
|
api_app = FastAPI(openapi_tags=tags_metadata, title="Dashdio API")
|
|
player = MusicPlayer()
|
|
dl_service = DownloadService()
|
|
ws_manager = ConnectionManager()
|
|
|
|
app = FastAPI(title="Dashdio App")
|
|
|
|
# Interface
|
|
app.mount("/api", api_app)
|
|
app.mount("/", StaticFiles(directory="ui", html=True), name="ui")
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def start_event_loop():
|
|
asyncio.create_task(state_broadcast_loop())
|
|
asyncio.create_task(queue_broadcast_loop())
|
|
|
|
|
|
async def state_broadcast_loop():
|
|
while True:
|
|
await player._state_event.wait()
|
|
await ws_manager.broadcast(WSConnectionType.state, player.get_state())
|
|
player._state_event.clear()
|
|
|
|
|
|
async def queue_broadcast_loop():
|
|
while True:
|
|
await player._queue_event.wait()
|
|
await ws_manager.broadcast(WSConnectionType.queue, player.get_queue())
|
|
player._queue_event.clear()
|
|
|
|
|
|
# Status updates
|
|
@api_app.websocket("/player")
|
|
async def websocket_player(websocket: WebSocket):
|
|
await ws_manager.connect(websocket, WSConnectionType.state)
|
|
try:
|
|
while True:
|
|
await websocket.receive_text()
|
|
await ws_manager.send(websocket, player.get_state())
|
|
except WebSocketDisconnect:
|
|
ws_manager.disconnect(websocket)
|
|
|
|
|
|
# Queue updates
|
|
@api_app.websocket("/queue")
|
|
async def websocket_queue(websocket: WebSocket):
|
|
await ws_manager.connect(websocket, WSConnectionType.queue)
|
|
try:
|
|
while True:
|
|
await websocket.receive_text()
|
|
await ws_manager.send(websocket, player.get_queue())
|
|
except WebSocketDisconnect:
|
|
ws_manager.disconnect(websocket)
|
|
|
|
|
|
@api_app.get("/queue", tags=["queue"])
|
|
def get_queue():
|
|
return player.get_queue()
|
|
|
|
|
|
@api_app.post("/queue", tags=["queue"])
|
|
async def post_to_queue(url: str):
|
|
track = await dl_service.download(url)
|
|
await player.add_to_queue(track)
|
|
|
|
|
|
@api_app.delete("/queue/{track_id}", tags=["queue"])
|
|
def delete_from_queue(track_id: str):
|
|
player.remove_from_queue_by_id(track_id)
|
|
|
|
|
|
@api_app.post("/player/play", tags=["player"])
|
|
async def player_play():
|
|
await player.play()
|
|
|
|
|
|
@api_app.post("/player/stop", tags=["player"])
|
|
async def player_stop():
|
|
await player.stop()
|
|
|
|
|
|
@api_app.post("/player/skip", tags=["player"])
|
|
async def player_skip():
|
|
await player.next()
|
|
|
|
|
|
# Player
|
|
@api_app.put("/player/volume", tags=["player"])
|
|
async def set_volume(volume: float):
|
|
await player.set_volume(volume)
|
|
|
|
|
|
@api_app.get("/player/volume", tags=["player"])
|
|
def get_volume():
|
|
return {"volume": player.get_volume()}
|
|
|
|
|
|
@api_app.get("/player", tags=["player"])
|
|
def get_player_state() -> PlayerState:
|
|
return player.get_state()
|