Update main.py

main.py

@@ -60,7 +60,6 @@ async def download_proxy(request: Request, url: str, key: str = None):
     if key != ACCESS_KEY:
         raise HTTPException(status_code=403, detail="Access Denied")
 
-    # URL ကို အကြိမ်ကြိမ် Decode လုပ်ပြီး သန့်စင်ခြင်း
     clean_url = urllib.parse.unquote(url)
     while "%" in clean_url:
         prev_url = clean_url
@@ -72,19 +71,26 @@ async def download_proxy(request: Request, url: str, key: str = None):
     # --- MediaFire ---
     if "mediafire.com" in clean_url:
         try:
-            # MediaFire Page ကို ခေါ်ယူခြင်း
-            r = await client.get(clean_url, headers={'User-Agent': ua.random})
+            # Player ကလာတဲ့ Request ကို Browser ကလာသလို အသွင်ဖျက်ရန်
+            browser_headers = {
+                'User-Agent': ua.random,
+                'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
+                'Accept-Language': 'en-US,en;q=0.5',
+                'Connection': 'keep-alive',
+                'Upgrade-Insecure-Requests': '1',
+            }
+            
+            r = await client.get(clean_url, headers=browser_headers)
             if r.status_code != 200:
-                raise HTTPException(status_code=r.status_code, detail="Failed to access MediaFire page")
+                raise HTTPException(status_code=r.status_code, detail="MediaFire page access denied")
 
-            # Direct Link ရှာဖွေခြင်း
             target_link = None
-            # Method 1: Regex
+            # Method 1: Regex for Direct Download Link
             match = re.search(r"https?://download[0-9]+\.mediafire\.com/[a-zA-Z0-9_-]+/[a-zA-Z0-9_-]+/[^\s'\"]+", r.text)
             if match:
                 target_link = match.group(0).replace('"', '').replace("'", "")
             
-            # Method 2: BeautifulSoup (Regex မတွေ့ရင်)
+            # Method 2: BS4 Fallback
             if not target_link:
                 soup = BeautifulSoup(r.text, 'html.parser')
                 btn = soup.find('a', {'id': 'downloadButton'})
@@ -92,23 +98,14 @@ async def download_proxy(request: Request, url: str, key: str = None):
                     target_link = btn.get('href')
 
             if not target_link:
-                raise HTTPException(status_code=404, detail="Direct link not found in MediaFire page")
+                raise HTTPException(status_code=404, detail="Direct link extraction failed")
+            
+            # Direct Link ကို Stream လုပ်တဲ့အခါ Referer ထည့်ပေးခြင်းဖြင့် 500 Error ကို ကျော်လွှားနိုင်သည်
+            return await stream_file(target_link, range_header, referer=clean_url)
             
-            return await stream_file(target_link, range_header)
         except Exception as e:
             print(f"MediaFire Error: {str(e)}")
-            raise HTTPException(status_code=500, detail=str(e))
-
-    # --- Dropbox ---
-    elif "dropbox.com" in clean_url:
-        target_link = clean_url.replace("?dl=0", "").split("?")[0] + "?dl=1"
-        return await stream_file(target_link, range_header)
-            
-    # --- OneDrive ---
-    elif "1drv.ms" in clean_url or "onedrive.live.com" in clean_url:
-        encoded_url = base64.b64encode(bytes(clean_url, 'utf-8')).decode('utf-8').replace('=', '').replace('/', '_').replace('+', '-')
-        target_link = f"https://api.onedrive.com/v1.0/shares/u!{encoded_url}/root/content"
-        return await stream_file(target_link, range_header)
+            raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")
 
     # --- Google Drive ---
     elif "drive.google.com" in clean_url:
@@ -122,11 +119,9 @@ async def download_proxy(request: Request, url: str, key: str = None):
         for account in accounts:
             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 range_header: headers['Range'] = range_header
-
             try:
                 req = client.build_request("GET", api_link, headers=headers)
                 r = await client.send(req, stream=True)
@@ -141,28 +136,28 @@ async def download_proxy(request: Request, url: str, key: str = None):
     else:
         return await stream_file(clean_url, range_header)
 
-async def stream_file(target_url, range_header):
+async def stream_file(target_url, range_header, referer=None):
     headers = {'User-Agent': ua.random}
     if range_header: headers['Range'] = range_header
+    if referer: headers['Referer'] = referer # MediaFire အတွက် အရေးကြီးသည်
     
     try:
-        # Redirect တွေကို ကိုင်တွယ်နိုင်ရန် client object ကို သု���းသည်
         req = client.build_request("GET", target_url, headers=headers)
         r = await client.send(req, stream=True)
+        # အကယ်၍ မူရင်း Server က 403/500 ပြန်လာလျှင် Proxy ကနေ Error ပြန်ရန်
+        if r.status_code >= 400:
+            await r.aclose()
+            raise HTTPException(status_code=r.status_code, detail="Remote server rejected the request")
         return await process_response(r)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 async def process_response(r):
-    # Streaming အတွက် လိုအပ်သော Header များကို သန့်စင်ယူခြင်း
     safe_headers = [
         'content-type', 'content-length', 'content-range', 
         'accept-ranges', 'content-disposition', 'cache-control'
     ]
-    
     response_headers = {n: v for n, v in r.headers.items() if n.lower() in safe_headers}
-    
-    # Header ထဲမှာ မပါလာရင်တောင် Bytes range လက်ခံကြောင်း ပြောထားမှ Player တွေ အဆင်ပြေမည်
     if 'accept-ranges' not in [k.lower() for k in response_headers.keys()]:
         response_headers['Accept-Ranges'] = 'bytes'