GitHub/xiaomusic
1
0
Fork 0
mirror of https://github.com/hanxi/xiaomusic.git synced 2026-03-23 09:29:44 +08:00
Code Packages Releases Activity

refactor: 修复告警

Browse Source
This commit is contained in:
涵曦
2024-10-09 07:02:58 +08:00
parent 9d9939be9f
commit c9384aac08
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
xiaomusic/httpserver.py
View File

@@ -499,6 +499,15 @@ def access_key_verification(file_path, key, code):
range_pattern = re.compile(r"bytes=(\d+)-(\d*)")
def safe_redirect(config, url):
url = try_add_access_control_param(config, url)
url = url.replace("\\", "")
if not urllib.parse.urlparse(url).netloc and not urllib.parse.urlparse(url).scheme:
RedirectResponse(url=url)
return True
return False
@app.get("/music/{file_path:path}")
async def music_file(request: Request, file_path: str, key: str = "", code: str = ""):
if not access_key_verification(f"/music/{file_path}", key, code):
@@ -517,8 +526,8 @@ async def music_file(request: Request, file_path: str, key: str = "", code: str
temp_mp3_file = remove_id3_tags(absolute_file_path, config)
if temp_mp3_file:
log.info(f"ID3 tag removed {absolute_file_path} to {temp_mp3_file}")
url = try_add_access_control_param(config, f"/music/{temp_mp3_file}")
return RedirectResponse(url=url)
if safe_redirect(f"/music/{temp_mp3_file}"):
return
else:
log.info(f"No ID3 tag remove needed: {absolute_file_path}")
@@ -526,8 +535,8 @@ async def music_file(request: Request, file_path: str, key: str = "", code: str
temp_mp3_file = convert_file_to_mp3(absolute_file_path, config)
if temp_mp3_file:
log.info(f"Converted file: {absolute_file_path} to {temp_mp3_file}")
url = try_add_access_control_param(config, f"/music/{temp_mp3_file}")
return RedirectResponse(url=url)
if safe_redirect(f"/music/{temp_mp3_file}"):
return
else:
log.warning(f"Failed to convert file to MP3 format: {absolute_file_path}")