Update main.py

main.py

@@ -2,8 +2,9 @@ import os
 import re
 import json
 import base64
-import requests
+import httpx
 import random
+import asyncio
 from bs4 import BeautifulSoup
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import StreamingResponse
@@ -19,55 +20,65 @@ app = FastAPI()
 # --- Configurations ---
 ACCESS_KEY = os.getenv("ACCESS_KEY", "0000")
 SERVICE_ACCOUNT_JSON_STR = os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
-
-# User-Agent Generator with fallback
 ua = UserAgent(fallback='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36')
 
+# Global Async Client: Video ကျော်ကြည့်တဲ့အခါ Connection အသစ်တွေ မြန်မြန်ဆောက်နိုင်ဖို့
+client = httpx.AsyncClient(
+    timeout=httpx.Timeout(20.0, read=None), 
+    follow_redirects=True,
+    limits=httpx.Limits(max_connections=100, max_keepalive_connections=20)
+)
+
 @app.get("/")
-def index():
-    return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
+async def index():
+    return {"message": "High-Speed Async Proxy Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
 
 def get_all_service_accounts():
-    if not SERVICE_ACCOUNT_JSON_STR:
-        return []
+    if not SERVICE_ACCOUNT_JSON_STR: return []
     try:
         data = json.loads(SERVICE_ACCOUNT_JSON_STR)
         return data if isinstance(data, list) else [data]
-    except Exception as e:
-        print(f"JSON Parsing Error: {e}")
-        return []
+    except: return []
 
-def get_token_for_account(cred_dict):
+async def get_token_for_account(cred_dict):
     try:
         scopes = ['https://www.googleapis.com/auth/drive.readonly']
         creds = service_account.Credentials.from_service_account_info(cred_dict, scopes=scopes)
+        # Token refresh ကို thread ထဲမှာ run ပါတယ် (Google library က sync ဖြစ်လို့)
+        loop = asyncio.get_event_loop()
         auth_req = google.auth.transport.requests.Request()
-        creds.refresh(auth_req)
+        await loop.run_in_executor(None, creds.refresh, auth_req)
         return creds.token
-    except Exception as e:
-        print(f"Auth Error for {cred_dict.get('client_email')}: {e}")
-        return None
+    except: return None
 
 def get_google_file_id(url):
     fid = re.search(r'/(?:d|file/d|open\?id=)/([a-zA-Z0-9_-]+)', url)
     return fid.group(1) if fid else None
 
 @app.get("/download")
-def download_proxy(request: Request, url: str, key: str = None):
+async def download_proxy(request: Request, url: str, key: str = None):
     if key != ACCESS_KEY:
         raise HTTPException(status_code=403, detail="Access Denied")
 
+    range_header = request.headers.get('range')
+
+    # --- MediaFire & Others ---
     if "drive.google.com" not in url:
         target_link = None
-        current_headers = {'User-Agent': ua.random, 'Accept-Language': 'en-US,en;q=0.9'}
+        headers = {'User-Agent': ua.random, 'Accept-Language': 'en-US,en;q=0.9'}
 
         if "mediafire.com" in url:
             try:
-                current_headers['Referer'] = 'https://www.mediafire.com/'
-                r = requests.get(url, headers=current_headers, timeout=15)
-                soup = BeautifulSoup(r.text, 'html.parser')
-                btn = soup.find('a', {'id': 'downloadButton'})
-                target_link = btn.get('href') if btn else None
+                headers['Referer'] = 'https://www.mediafire.com/'
+                r = await client.get(url, headers=headers)
+                # Regex ဖြင့် Direct Link ကို အမြန်ဆုံး ရှာဖွေခြင်း
+                match = re.search(r'href="(https?://download[^"]+)"', r.text)
+                if match:
+                    target_link = match.group(1)
+                else:
+                    soup = BeautifulSoup(r.text, 'html.parser')
+                    btn = soup.find('a', {'id': 'downloadButton'})
+                    target_link = btn.get('href') if btn else None
             except Exception as e:
                 print(f"MediaFire Error: {e}")
         
@@ -81,8 +92,9 @@ def download_proxy(request: Request, url: str, key: str = None):
         if not target_link:
             raise HTTPException(status_code=400, detail="Could not resolve direct link")
         
-        return stream_file(target_link, request.headers.get('range'))
+        return await stream_file(target_link, range_header)
 
+    # --- Google Drive ---
     file_id = get_google_file_id(url)
     if not file_id:
         raise HTTPException(status_code=400, detail="Invalid Google Drive Link")
@@ -91,61 +103,50 @@ def download_proxy(request: Request, url: str, key: str = None):
     random.shuffle(accounts)
 
     for account in accounts:
-        token = get_token_for_account(account)
+        token = await get_token_for_account(account)
         if not token: continue
 
         api_link = f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media"
         headers = {"Authorization": f"Bearer {token}"}
-        if request.headers.get('range'):
-            headers['Range'] = request.headers.get('range')
+        if range_header: headers['Range'] = range_header
 
         try:
-            r = requests.get(api_link, headers=headers, stream=True, allow_redirects=True, timeout=15)
+            # Stream mode ဖြင့် ချိတ်ဆက်ခြင်း
+            req = client.build_request("GET", api_link, headers=headers)
+            r = await client.send(req, stream=True)
             if r.status_code in [200, 206]:
-                return process_response(r)
-        except Exception as e:
-            print(f"Account Request Error: {e}")
+                return await process_response(r)
+            await r.aclose()
+        except: continue
 
     # Public Fallback
-    try:
-        public_url = f"https://drive.google.com/uc?export=download&id={file_id}"
-        headers = {'User-Agent': ua.random} 
-        if request.headers.get('range'):
-            headers['Range'] = request.headers.get('range')
-        r = requests.get(public_url, headers=headers, stream=True, allow_redirects=True, timeout=15)
-        return process_response(r)
-    except:
-        raise HTTPException(status_code=500, detail="Download failed.")
-
-def stream_file(target_url, range_header):
+    public_url = f"https://drive.google.com/uc?export=download&id={file_id}"
+    return await stream_file(public_url, range_header)
+
+async def stream_file(target_url, range_header):
     headers = {'User-Agent': ua.random}
-    if range_header:
-        headers['Range'] = range_header
+    if range_header: headers['Range'] = range_header
+    
     try:
-        r = requests.get(target_url, headers=headers, stream=True, allow_redirects=True, timeout=15)
-        return process_response(r)
+        req = client.build_request("GET", target_url, headers=headers)
+        r = await client.send(req, stream=True)
+        return await process_response(r)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-def process_response(r):
+async def process_response(r):
     excluded = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
     response_headers = {n: v for n, v in r.headers.items() if n.lower() not in excluded}
+    
+    # Video Seeking အတွက် မရှိမဖြစ်လိုအပ်သော Headers များ
     response_headers['Accept-Ranges'] = 'bytes'
     if 'Content-Length' in r.headers:
         response_headers['Content-Length'] = r.headers['Content-Length']
     if 'Content-Range' in r.headers:
         response_headers['Content-Range'] = r.headers['Content-Range']
 
-    def iterfile():
-        try:
-            # Chunk size ကို 5MB တိုးမြှင့်ထားပါတယ် (Speed ပိုကောင်းစေရန်)
-            for chunk in r.iter_content(chunk_size=1024*1024):
-                if chunk: yield chunk
-        finally:
-            r.close()
-
     return StreamingResponse(
-        iterfile(),
+        r.aiter_bytes(chunk_size=131072), # 128KB chunks for perfect buffering
         status_code=r.status_code,
         headers=response_headers,
         media_type=r.headers.get('Content-Type', 'application/octet-stream')