""")
# ------------------- Download page (like Mega) -------------------
def format_size(size_bytes):
if size_bytes < 1024:
return f"{size_bytes} B"
elif size_bytes < 1024*1024:
return f"{size_bytes/1024:.1f} KB"
elif size_bytes < 1024*1024*1024:
return f"{size_bytes/(1024*1024):.1f} MB"
else:
return f"{size_bytes/(1024*1024*1024):.2f} GB"
@app.get("/{short_code}")
async def download_page(short_code: str, request: Request, direct: bool = Query(False)):
conn = get_conn()
upload = conn.execute(
"SELECT id, is_complete, is_deleted, created_at, expires_at FROM uploads WHERE short_code=?",
(short_code,)
).fetchone()
if not upload or upload['is_deleted']:
conn.close()
return HTMLResponse(status_code=404, content=error_html("Link not found or files have been deleted."))
if not upload['is_complete']:
conn.close()
return HTMLResponse("
Files are still uploading, please try later
", status_code=202)
files = conn.execute("SELECT original_name, size FROM files WHERE upload_id=?", (upload['id'],)).fetchall()
conn.close()
# Если запрошена прямая загрузка (параметр ?direct или путь /dl/...)
if direct:
return await serve_direct_download(upload['id'], short_code, files)
# Иначе показываем красивую страницу
total_size = sum(f['size'] for f in files)
file_names = [f['original_name'] for f in files]
display_name = file_names[0] if len(files) == 1 else f"{len(files)} files"
created = upload['created_at']
expires = upload['expires_at']
html = f"""
Download - {display_name}