Spaces:

Starchik1
/

proxy

Running

App Files Files Community

Starchik1 commited on 11 days ago

Commit

9bb2d26

verified ·

1 Parent(s): f269387

Update app.py

Browse files

Files changed (1) hide show

app.py +301 -48

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from datetime import datetime, timedelta
 from pathlib import Path
 from contextlib import asynccontextmanager
-from fastapi import FastAPI, UploadFile, File, HTTPException, Request, Form
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse
 import aiofiles
 from apscheduler.schedulers.background import BackgroundScheduler
@@ -112,7 +112,62 @@ async def lifespan(app: FastAPI):
 app = FastAPI(title="Free File Sharing", lifespan=lifespan)
-# ------------------- Web Interface -------------------
 @app.get("/", response_class=HTMLResponse)
 async def index():
     return HTMLResponse("""
@@ -488,7 +543,249 @@ function showSuccess(msg) { showMessage(msg, false); }
 </html>
 """)
-# ------------------- API Endpoints -------------------
 @app.post("/upload/init")
 async def init_upload(data: dict):
     file_count = data.get("file_count", 1)
@@ -586,48 +883,4 @@ async def delete_by_token(delete_token: str):
     if not upload:
         raise HTTPException(status_code=404, detail="Invalid token or files already deleted")
     await delete_upload(upload['id'])
-    return {"status": "deleted"}
-@app.get("/{short_code}")
-async def download(short_code: str, request: Request):
-    conn = get_conn()
-    upload = conn.execute(
-        "SELECT id, is_complete, is_deleted FROM uploads WHERE short_code=?",
-        (short_code,)
-    ).fetchone()
-    if not upload or upload['is_deleted']:
-        conn.close()
-        raise HTTPException(status_code=404, detail="Link not found or files deleted")
-    if not upload['is_complete']:
-        conn.close()
-        return HTMLResponse("<h3>Files are still uploading, please try later</h3>", status_code=202)
-    files = conn.execute("SELECT original_name, stored_name FROM files WHERE upload_id=?", (upload['id'],)).fetchall()
-    conn.close()
-    upload_dir = UPLOAD_DIR / upload['id']
-    if not upload_dir.exists():
-        raise HTTPException(status_code=404, detail="Files not found on disk")
-    if len(files) == 1:
-        file_path = upload_dir / files[0]['original_name']
-        if not file_path.exists():
-            raise HTTPException(status_code=404, detail="File missing")
-        return FileResponse(file_path, filename=files[0]['original_name'])
-    else:
-        zip_filename = f"files_{short_code}.zip"
-        async def zip_stream():
-            zip_buffer = io.BytesIO()
-            with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zf:
-                for f in files:
-                    file_path = upload_dir / f['original_name']
-                    if file_path.exists():
-                        zf.write(file_path, f['original_name'])
-            zip_buffer.seek(0)
-            while True:
-                data = zip_buffer.read(8192)
-                if not data:
-                    break
-                yield data
-        return StreamingResponse(zip_stream(), media_type="application/zip", headers={
-            "Content-Disposition": f'attachment; filename="{zip_filename}"'
-        })

 from pathlib import Path
 from contextlib import asynccontextmanager
+from fastapi import FastAPI, UploadFile, File, HTTPException, Request, Form, Query
 from fastapi.responses import HTMLResponse, FileResponse, StreamingResponse
 import aiofiles
 from apscheduler.schedulers.background import BackgroundScheduler
 app = FastAPI(title="Free File Sharing", lifespan=lifespan)
+# ------------------- Beautiful error page -------------------
+def error_html(message: str, status_code: int = 404) -> str:
+    return f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Error {status_code}</title>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <style>
+            body {{
+                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                min-height: 100vh;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                margin: 0;
+                padding: 20px;
+            }}
+            .error-card {{
+                background: white;
+                border-radius: 20px;
+                padding: 40px;
+                text-align: center;
+                box-shadow: 0 20px 60px rgba(0,0,0,0.3);
+                max-width: 400px;
+            }}
+            h1 {{
+                font-size: 5em;
+                color: #ff5252;
+                margin: 0;
+            }}
+            p {{
+                color: #555;
+                font-size: 1.1em;
+                margin: 20px 0;
+            }}
+            a {{
+                color: #667eea;
+                text-decoration: none;
+                font-weight: bold;
+            }}
+        </style>
+    </head>
+    <body>
+        <div class="error-card">
+            <h1>{status_code}</h1>
+            <p>{message}</p>
+            <a href="/">← Back to main page</a>
+        </div>
+    </body>
+    </html>
+    """
+# ------------------- Main upload page -------------------
 @app.get("/", response_class=HTMLResponse)
 async def index():
     return HTMLResponse("""
 </html>
 """)
+# ------------------- 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("<h3>Files are still uploading, please try later</h3>", 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"""
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>Download - {display_name}</title>
+        <script src="https://cdn.jsdelivr.net/npm/qrcodejs@1.0.0/qrcode.min.js"></script>
+        <style>
+            body {{
+                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                min-height: 100vh;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                padding: 20px;
+                margin: 0;
+            }}
+            .download-card {{
+                background: white;
+                border-radius: 20px;
+                padding: 30px;
+                max-width: 500px;
+                width: 100%;
+                box-shadow: 0 20px 60px rgba(0,0,0,0.3);
+                text-align: center;
+            }}
+            .file-icon {{
+                font-size: 4em;
+                margin-bottom: 15px;
+            }}
+            h2 {{
+                color: #333;
+                word-break: break-all;
+                margin-bottom: 15px;
+            }}
+            .info {{
+                color: #555;
+                font-size: 0.95em;
+                margin: 10px 0;
+            }}
+            .btn-download {{
+                display: inline-block;
+                padding: 15px 40px;
+                background: linear-gradient(135deg, #667eea, #764ba2);
+                color: white;
+                border: none;
+                border-radius: 30px;
+                font-weight: bold;
+                font-size: 1.1em;
+                cursor: pointer;
+                text-decoration: none;
+                transition: transform 0.2s, box-shadow 0.2s;
+                box-shadow: 0 4px 10px rgba(102,126,234,0.4);
+                margin: 15px 0;
+            }}
+            .btn-download:hover {{
+                transform: translateY(-2px);
+                box-shadow: 0 6px 15px rgba(102,126,234,0.6);
+            }}
+            .qr-section {{
+                margin-top: 20px;
+                position: relative;
+            }}
+            .qr-container {{
+                display: inline-block;
+                cursor: pointer;
+                position: relative;
+            }}
+            .qr-hint {{
+                position: absolute;
+                bottom: 100%;
+                left: 50%;
+                transform: translateX(-50%);
+                background: #333;
+                color: white;
+                padding: 5px 12px;
+                border-radius: 20px;
+                font-size: 0.8em;
+                white-space: nowrap;
+                opacity: 0;
+                transition: opacity 0.3s;
+                pointer-events: none;
+            }}
+            .qr-container:hover .qr-hint {{
+                opacity: 1;
+            }}
+            .direct-link {{
+                margin-top: 10px;
+                font-size: 0.9em;
+                color: #666;
+            }}
+            .direct-link a {{
+                color: #667eea;
+                text-decoration: none;
+            }}
+            .back-link {{
+                margin-top: 20px;
+                display: block;
+                color: #667eea;
+                text-decoration: none;
+            }}
+        </style>
+    </head>
+    <body>
+        <div class="download-card">
+            <div class="file-icon">📁</div>
+            <h2>{display_name}</h2>
+            <div class="info">Size: {format_size(total_size)}</div>
+            <div class="info">Uploaded: {created}</div>
+            <div class="info">Expires: {expires}</div>
+            <a href="/dl/{short_code}" class="btn-download" id="downloadBtn">⬇ Download ({format_size(total_size)})</a>
+            <div class="qr-section">
+                <div class="qr-container" id="qrContainer">
+                    <div id="qrcode"></div>
+                    <div class="qr-hint">Click to switch to direct link</div>
+                </div>
+            </div>
+            <div class="direct-link">
+                Direct: <a href="/dl/{short_code}" id="directLink">/dl/{short_code}</a>
+            </div>
+            <a href="/" class="back-link">← Upload more files</a>
+        </div>
+        <script>
+            const pageUrl = window.location.href.split('?')[0];
+            const directUrl = pageUrl.replace(/\/[^\/]+$/, '/dl/{short_code}');
+            let showingDirect = false;
+            const qrDiv = document.getElementById('qrcode');
+            const qrContainer = document.getElementById('qrContainer');
+            const directLink = document.getElementById('directLink');
+            function makeQR(url) {{
+                qrDiv.innerHTML = '';
+                new QRCode(qrDiv, {{
+                    text: url,
+                    width: 128,
+                    height: 128,
+                }});
+            }}
+            makeQR(pageUrl);
+            qrContainer.addEventListener('click', () => {{
+                showingDirect = !showingDirect;
+                if (showingDirect) {{
+                    makeQR(directUrl);
+                    directLink.textContent = directUrl;
+                }} else {{
+                    makeQR(pageUrl);
+                    directLink.textContent = '/dl/{short_code}';
+                }}
+            }});
+        </script>
+    </body>
+    </html>
+    """
+    return HTMLResponse(html)
+@app.get("/dl/{short_code}")
+async def direct_download(short_code: str, request: Request):
+    conn = get_conn()
+    upload = conn.execute(
+        "SELECT id, is_complete, is_deleted FROM uploads WHERE short_code=?",
+        (short_code,)
+    ).fetchone()
+    if not upload or upload['is_deleted'] or not upload['is_complete']:
+        conn.close()
+        return HTMLResponse(status_code=404, content=error_html("File not found."))
+    files = conn.execute("SELECT original_name, size FROM files WHERE upload_id=?", (upload['id'],)).fetchall()
+    conn.close()
+    return await serve_direct_download(upload['id'], short_code, files)
+async def serve_direct_download(upload_id: str, short_code: str, files):
+    upload_dir = UPLOAD_DIR / upload_id
+    if not upload_dir.exists():
+        return HTMLResponse(status_code=404, content=error_html("Files missing."))
+    if len(files) == 1:
+        file_path = upload_dir / files[0]['original_name']
+        if not file_path.exists():
+            return HTMLResponse(status_code=404, content=error_html("File missing."))
+        return FileResponse(file_path, filename=files[0]['original_name'])
+    else:
+        zip_filename = f"files_{short_code}.zip"
+        async def zip_stream():
+            zip_buffer = io.BytesIO()
+            with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zf:
+                for f in files:
+                    file_path = upload_dir / f['original_name']
+                    if file_path.exists():
+                        zf.write(file_path, f['original_name'])
+            zip_buffer.seek(0)
+            while True:
+                data = zip_buffer.read(8192)
+                if not data:
+                    break
+                yield data
+        return StreamingResponse(zip_stream(), media_type="application/zip", headers={
+            "Content-Disposition": f'attachment; filename="{zip_filename}"'
+        })
+# ------------------- Upload endpoints (unchanged) -------------------
 @app.post("/upload/init")
 async def init_upload(data: dict):
     file_count = data.get("file_count", 1)
     if not upload:
         raise HTTPException(status_code=404, detail="Invalid token or files already deleted")
     await delete_upload(upload['id'])
+    return {"status": "deleted"}