Update main.py

main.py

@@ -1,5 +1,6 @@
 import os
 import re
+import json
 import base64
 import requests
 from bs4 import BeautifulSoup
@@ -7,74 +8,113 @@ from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import StreamingResponse
 import uvicorn
 
+# Google Auth Libraries များကို Import လုပ်ခြင်း
+from google.oauth2 import service_account
+import google.auth.transport.requests
+
 app = FastAPI()
 
 # --- Configurations ---
 ACCESS_KEY = os.getenv("ACCESS_KEY", "786969")
-CLIENT_ID = os.getenv("GOOGLE_CLIENT_ID")
-CLIENT_SECRET = os.getenv("GOOGLE_CLIENT_SECRET")
-REFRESH_TOKEN = os.getenv("GOOGLE_REFRESH_TOKEN")
+# Service Account ရဲ့ JSON ဖိုင်ထဲက စာသားအကုန်လုံးကို Copy ကူးပြီး Secret ထဲမှာ ထည့်ရပါမည်
+SERVICE_ACCOUNT_JSON_STR = os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
+
+@app.get("/")
+def index():
+    return {"message": "Proxy is Online!", "usage": "/download?url=[LINK]&key=[YOUR_KEY]"}
 
 def get_google_access_token():
-    """OAuth Access Token ကို ရယူခြင်း"""
-    if not (CLIENT_ID and CLIENT_SECRET and REFRESH_TOKEN):
+    """Service Account JSON ကို အသုံးပြု၍ Access Token ရယူခြင်း"""
+    if not SERVICE_ACCOUNT_JSON_STR:
+        print("Missing GOOGLE_SERVICE_ACCOUNT_JSON credential!")
         return None
+    
     try:
-        url = "https://oauth2.googleapis.com/token"
-        payload = {
-            "client_id": CLIENT_ID,
-            "client_secret": CLIENT_SECRET,
-            "refresh_token": REFRESH_TOKEN,
-            "grant_type": "refresh_token",
-        }
-        r = requests.post(url, data=payload, timeout=10)
-        return r.json().get("access_token")
-    except:
+        # Secret ထဲမှ JSON string ကို Dictionary အဖြစ် ပြောင်းလဲခြင်း
+        creds_dict = json.loads(SERVICE_ACCOUNT_JSON_STR)
+        
+        # Scopes သတ်မှတ်ခြင်း (Drive ကို ဖတ်ရန်)
+        scopes = ['https://www.googleapis.com/auth/drive.readonly']
+        
+        # Service Account Credentials တည်ဆောက်ခြင်း
+        creds = service_account.Credentials.from_service_account_info(
+            creds_dict,
+            scopes=scopes
+        )
+        
+        # Token အသစ်တောင်းခံခြင်း
+        auth_req = google.auth.transport.requests.Request()
+        creds.refresh(auth_req)
+        
+        return creds.token
+    except Exception as e:
+        print(f"Service Account Auth Error: {e}")
         return None
 
 def get_direct_url(url):
     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)
-            token = get_google_access_token()
-            if token:
-                # Sign-in page မတက်အောင် ရိုးရိုး link အစား API v3 link ကိုပဲ သုံးမည်
-                return f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media", token
-            else:
-                # Token မရှိပါက fallback အနေဖြင့်သာ ရိုးရိုး link ကို သုံးမည်
-                return f"https://drive.google.com/uc?export=download&id={file_id}", None
-    
-    # ... (MediaFire & အခြား logic များ အတူတူပင်ဖြစ်သည်) ...
+        if not fid:
+            return None, None
+        file_id = fid.group(1)
+        
+        access_token = get_google_access_token()
+        if access_token:
+            # Service Account မှ Token ရပါက API v3 Link ဖြင့် ပြန်ပေးမည်
+            return f"https://www.googleapis.com/drive/v3/files/{file_id}?alt=media", access_token
+        
+        return None, 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
+        
+    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")
 async def download_proxy(request: Request, url: str, key: str = None):
+    # Proxy Password စစ်ဆေးခြင်း
     if key != ACCESS_KEY:
         raise HTTPException(status_code=403, detail="Access Denied")
 
     target_link, token = get_direct_url(url)
     if not target_link:
-        raise HTTPException(status_code=400, detail="Invalid Link")
+        raise HTTPException(status_code=400, detail="Invalid Link or Auth Failed")
 
+    # Header ပြင်ဆင်ခြင်း
     headers = {}
     range_header = request.headers.get('range')
     if range_header:
         headers['Range'] = range_header
     
-    # Token ရှိလျှင် Authorization Header ကို သေချာပေါက် ထည့်ရမည်
+    # Token ရှိပါက Bearer Token ထည့်ပေးခြင်း
     if token:
         headers['Authorization'] = f"Bearer {token}"
 
     session = requests.Session()
-    # Google API သို့ Request ပို့ခြင်း
-    r = session.get(target_link, headers=headers, stream=True, allow_redirects=True)
-    
-    # အကယ်၍ Sign-in Page တက်နေသေးပါက (Response က HTML ဖြစ်နေလျှင်) Error ပြရန်
+    try:
+        r = session.get(target_link, headers=headers, stream=True, allow_redirects=True)
+        r.raise_for_status() # 403 / 404 ဖြစ်ပါက တန်းပြီး Error ဖမ်းမည်
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=400, detail=f"Failed to fetch file: {str(e)}")
+        
+    # Sign-in page သို့မဟုတ် HTML Error Page တက်လာပါက
     if "text/html" in r.headers.get("Content-Type", "").lower() and "google.com" in target_link:
-        # ဤနေရာတွင် Sign-in Page ကို stream လုပ်မည့်အစား error ပြလိုက်ခြင်းက ပိုကောင်းပါသည်
-        raise HTTPException(status_code=401, detail="Google Authentication Failed. Please check your OAuth Credentials.")
+        raise HTTPException(status_code=401, detail="Google Authentication Failed. Token may be invalid.")
 
+    # 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'