vndangkhoa/spotify-clone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix NAS playback: Dynamic MIME type and v2 stream cache

Browse source
This commit is contained in:
Khoa Vo 2026-01-01 16:16:06 +07:00
parent f073846d4b
commit 7bc230a8fd
1 changed files with 31 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
backend/api/routes.py
View file

@ -608,18 +608,25 @@ async def stream_audio(id: str):
""" """
try: try:
# Check Cache for stream URL # Check Cache for stream URL
cache_key = f"stream:{id}" cache_key = f"v2:stream:{id}" # v2 cache key for new format
cached_url = cache.get(cache_key) cached_data = cache.get(cache_key)
stream_url = None stream_url = None
if cached_url: mime_type = "audio/mp4"
print(f"DEBUG: Using cached stream URL for '{id}'")
stream_url = cached_url if cached_data:
else: print(f"DEBUG: Using cached stream data for '{id}'")
if isinstance(cached_data, dict):
stream_url = cached_data.get('url')
mime_type = cached_data.get('mime', 'audio/mp4')
else:
stream_url = cached_data # Legacy fallback
if not stream_url:
print(f"DEBUG: Fetching new stream URL for '{id}'") print(f"DEBUG: Fetching new stream URL for '{id}'")
url = f"https://www.youtube.com/watch?v={id}" url = f"https://www.youtube.com/watch?v={id}"
ydl_opts = { ydl_opts = {
'format': 'bestaudio[ext=m4a]/bestaudio/best', # Prefer m4a for direct streaming 'format': 'bestaudio[ext=m4a]/bestaudio/best',
'quiet': True, 'quiet': True,
'noplaylist': True, 'noplaylist': True,
'nocheckcertificate': True, 'nocheckcertificate': True,
@ -627,27 +634,37 @@ async def stream_audio(id: str):
'socket_timeout': 30, 'socket_timeout': 30,
'retries': 3, 'retries': 3,
'force_ipv4': True, 'force_ipv4': True,
'extractor_args': {'youtube': {'player_client': ['ios', 'android']}}, # Remove 'web' to avoid blocking 'extractor_args': {'youtube': {'player_client': ['ios', 'android']}},
} }
# Extract direct URL
try: try:
with yt_dlp.YoutubeDL(ydl_opts) as ydl: with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info = ydl.extract_info(url, download=False) info = ydl.extract_info(url, download=False)
stream_url = info.get('url') stream_url = info.get('url')
print(f"DEBUG: Got stream URL format: {info.get('format')}, ext: {info.get('ext')}") ext = info.get('ext')
# Determine MIME type
if ext == 'm4a':
mime_type = "audio/mp4"
elif ext == 'webm':
mime_type = "audio/webm"
else:
mime_type = "audio/mpeg" # Fallback
print(f"DEBUG: Got stream URL format: {info.get('format')}, ext: {ext}, mime: {mime_type}")
except Exception as ydl_error: except Exception as ydl_error:
print(f"DEBUG: yt-dlp extraction error: {type(ydl_error).__name__}: {str(ydl_error)}") print(f"DEBUG: yt-dlp extraction error: {type(ydl_error).__name__}: {str(ydl_error)}")
raise ydl_error raise ydl_error
if stream_url: if stream_url:
# Cache for 1 hour (3600 seconds) - URLs expire cache_data = {"url": stream_url, "mime": mime_type}
cache.set(cache_key, stream_url, ttl_seconds=3600) cache.set(cache_key, cache_data, ttl_seconds=3600)
if not stream_url: if not stream_url:
raise HTTPException(status_code=404, detail="Audio stream not found") raise HTTPException(status_code=404, detail="Audio stream not found")
# Stream the content print(f"Streaming {id} with Content-Type: {mime_type}")
def iterfile(): def iterfile():
try: try:
with requests.get(stream_url, stream=True, timeout=30) as r: with requests.get(stream_url, stream=True, timeout=30) as r:
@ -656,7 +673,6 @@ async def stream_audio(id: str):
yield chunk yield chunk
except requests.exceptions.HTTPError as http_err: except requests.exceptions.HTTPError as http_err:
print(f"DEBUG: Stream HTTP Error: {http_err}") print(f"DEBUG: Stream HTTP Error: {http_err}")
# Invalidate cache on 403
if http_err.response.status_code == 403: if http_err.response.status_code == 403:
cache.delete(cache_key) cache.delete(cache_key)
raise raise
@ -664,7 +680,7 @@ async def stream_audio(id: str):
print(f"DEBUG: Stream Iterator Error: {e}") print(f"DEBUG: Stream Iterator Error: {e}")
raise raise
return StreamingResponse(iterfile(), media_type="audio/mp4") return StreamingResponse(iterfile(), media_type=mime_type)
except HTTPException: except HTTPException:
raise raise