Update main.py

main.py

@@ -9,11 +9,8 @@ import uvicorn
 
 app = FastAPI()
 
-# --- Configurations ---
+# --- Configurations (Hugging Face Secrets မှ ဆွဲယူမည်) ---
 ACCESS_KEY = os.getenv("ACCESS_KEY", "786969")
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
-
-# OAuth Credentials (Hugging Face Secrets ထဲမှာ ထည့်ထားပေးပါ)
 CLIENT_ID = os.getenv("GOOGLE_CLIENT_ID")
 CLIENT_SECRET = os.getenv("GOOGLE_CLIENT_SECRET")
 REFRESH_TOKEN = os.getenv("GOOGLE_REFRESH_TOKEN")
@@ -23,57 +20,51 @@ def index():
     return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
 
 def get_google_access_token():
-    """Refresh Token ကိုသုံးပြီး Access Token အသစ်တောင်းယူခြင်း"""
+    """OAuth Token ကို Refresh လုပ်ခြင်း"""
     if not (CLIENT_ID and CLIENT_SECRET and REFRESH_TOKEN):
         return None
     try:
         url = "https://oauth2.googleapis.com/token"
-        data = {
+        payload = {
             "client_id": CLIENT_ID,
             "client_secret": CLIENT_SECRET,
             "refresh_token": REFRESH_TOKEN,
             "grant_type": "refresh_token",
         }
-        r = requests.post(url, data=data, timeout=10)
+        r = requests.post(url, data=payload, timeout=10)
         return r.json().get("access_token")
     except:
         return None
 
 def get_direct_url(url):
-    """Link အမျိုးအစားအလိုက် Direct Link နှင့် လိုအပ်ပါက Bearer Token ပြန်ပေးခြင်း"""
-    token = None
-    
+    """Link အမျိုးအစားအလိုက် Direct URL နှင့် Token ခွဲခြားထုတ်ယူခြင်း"""
     if "drive.google.com" in url:
         fid = re.search(r'/(?:d|file/d|open\?id=)/([a-zA-Z0-9_-]+)', url)
         if fid:
             file_id = fid.group(1)
-            # OAuth သုံးနိုင်လျှင် (Private Files အတွက် အကောင်းဆုံး)
-            access_token = get_google_access_token()
-            if access_token:
-                return f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media", access_token
-            
-            # API Key ရှိလျှင် (Public Files အတွက်)
-            if GOOGLE_API_KEY and GOOGLE_API_KEY != "YOUR_GOOGLE_API_KEY_HERE":
-                return f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media&key={GOOGLE_API_KEY}", None
-            
-            # API Key မရှိလျှင် ရိုးရိုး Direct Link
-            return f"https://drive.google.com/uc?export=download&id={file_id}", None
-            
+            token = get_google_access_token()
+            if token:
+                # OAuth သုံး၍ Google Drive API v3 မှ ဆွဲထုတ်ခြင်း
+                return f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media", token
+            else:
+                # Token မရှိပါက ရိုးရိုး URL သုံးမည်
+                return f"https://drive.google.com/uc?export=download&id={file_id}", None
+
     elif "mediafire.com" in url:
         try:
             r = requests.get(url, headers={'User-Agent': 'Mozilla/5.0'}, timeout=10)
             soup = BeautifulSoup(r.text, 'html.parser')
             btn = soup.find('a', {'id': 'downloadButton'})
-            return btn.get('href') if btn else None, None
-        except: return None, None
-        
+            if btn: return btn.get('href'), None
+        except: pass
+
     elif "dropbox.com" in url:
         return url.replace("?dl=0", "").split("?")[0] + "?dl=1", None
-        
+
     elif "1drv.ms" in url or "onedrive.live.com" in url:
         encoded_url = base64.b64encode(bytes(url, 'utf-8')).decode('utf-8').replace('=', '').replace('/', '_').replace('+', '-')
         return f"https://api.onedrive.com/v1.0/shares/u!{encoded_url}/root/content", None
-        
+
     return None, None
 
 @app.get("/download")
@@ -82,42 +73,30 @@ async def download_proxy(request: Request, url: str, key: str = None):
     if key != ACCESS_KEY:
         raise HTTPException(status_code=403, detail="Access Denied")
 
-    # Direct Link နှင့် Token ရယူခြင်း
     target_link, token = get_direct_url(url)
     if not target_link:
         raise HTTPException(status_code=400, detail="Invalid Link")
 
-    # Header များ ပြင်ဆင်ခြင်း
+    # Header ပြင်ဆင်ခြင်း
     headers = {}
     range_header = request.headers.get('range')
     if range_header:
         headers['Range'] = range_header
     
-    # OAuth Token ရှိလျှင် Authorization Header ထည့်မည်
     if token:
         headers['Authorization'] = f"Bearer {token}"
 
     session = requests.Session()
+    # Google/Target Server ထံသို့ Request ပို့ခြင်း
     r = session.get(target_link, headers=headers, stream=True, allow_redirects=True)
     
-    # ---------------------------------------------------------
-    # GDrive Virus Scan Bypass (OAuth မသုံးထားသော Public Links များအတွက်)
-    # ---------------------------------------------------------
-    if "drive.google.com" in url and not token:
-        content_type = r.headers.get('Content-Type', '').lower()
-        if 'text/html' in content_type:
-            # HTML ဖြစ်မှသာ r.text ကိုဖတ်မည် (RAM မပြည့်စေရန်)
-            if "Google Drive - Virus scan warning" in r.text or "confirm=" not in target_link:
-                confirm_token = None
-                for c_key, c_val in session.cookies.items():
-                    if c_key.startswith("download_warning"):
-                        confirm_token = c_val
-                        break
-                if confirm_token:
-                    new_link = f"{target_link}&confirm={confirm_token}"
-                    r = session.get(new_link, headers=headers, stream=True, allow_redirects=True)
-
-    # Header များကို Proxy လုပ်၍ ပြန်ပို့ခြင်း
+    # Error Handling (404 ဖြစ်ပါက Public link ဖြင့် Bypass လုပ်ကြည့်မည်)
+    if r.status_code == 404 and "drive.google.com" in url:
+        file_id = re.search(r'/files/([a-zA-Z0-9_-]+)', target_link).group(1)
+        fallback_url = f"https://drive.google.com/uc?export=download&id={file_id}"
+        r = session.get(fallback_url, headers=headers, stream=True, allow_redirects=True)
+
+    # Header များကို ပြန်လည်ပေးပို့ရန် ပြင်ဆင်ခြင်း
     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}
     response_headers['Accept-Ranges'] = 'bytes'