Update main.py

main.py

@@ -1,6 +1,7 @@
 import os
 import re
 import json
+import time
 import httpx
 import random
 import asyncio
@@ -23,6 +24,10 @@ ACCESS_KEY = os.getenv("ACCESS_KEY", "0000")
 SERVICE_ACCOUNT_JSON_STR = os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
 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')
 
+# MediaFire အတွက် Cache Memory
+MEDIAFIRE_CACHE = {}
+CACHE_TTL = 1800 # Cache သက်တမ်း စက္ကန့် 1800 (မိနစ် 30)
+
 client = httpx.AsyncClient(
     timeout=httpx.Timeout(60.0, read=None), 
     follow_redirects=True,
@@ -55,7 +60,6 @@ def get_google_file_id(url):
     return fid.group(1) if fid else None
 
 def get_clean_filename(url):
-    """URL ထဲကနေ ဖိုင်နာမည်ကို သန့်စင်စွာထုတ်ယူခြင်း"""
     decoded_url = urllib.parse.unquote(url)
     name = decoded_url.split('/')[-1].split('?')[0]
     if not name or '.' not in name:
@@ -71,13 +75,30 @@ async def download_proxy(request: Request, url: str, key: str = None):
     filename = get_clean_filename(clean_url)
     range_header = request.headers.get('range')
     
-    # 1. သတ်မှတ်ထားသော User-Agent ကို Request တစ်ခုလုံးအတွက် အသုံးပြုရန်
     req_ua = ua.random 
+    current_time = time.time()
 
     # --- MediaFire Section ---
     if "mediafire.com" in clean_url:
+        
+        # 1. Cache ထဲမှာရှိမရှိ အရင်စစ်ဆေးခြင်း (Player များ Seek လုပ်ရာတွင် လွယ်ကူစေရန်)
+        cached_data = MEDIAFIRE_CACHE.get(clean_url)
+        if cached_data and (current_time - cached_data['time']) < CACHE_TTL:
+            try:
+                return await stream_file(
+                    cached_data['link'], range_header, filename, 
+                    referer=clean_url, req_ua=cached_data['ua'], cookies=cached_data['cookies']
+                )
+            except HTTPException as e:
+                # အကယ်၍ Cached Link သည် သက်တမ်းကုန်သွား၍ 415 ပြန်လာပါက Cache ကိုဖျက်ပြီး အသစ်ပြန်ရှာမည်
+                if e.status_code == 415:
+                    print(f"Cache expired for {clean_url}, refetching...")
+                    del MEDIAFIRE_CACHE[clean_url]
+                else:
+                    raise
+                    
+        # 2. Cache မရှိပါက သို့မဟုတ် Expire ဖြစ်သွားပါက အသစ်ပြန်ရှာခြင်း
         try:
-            # Browser အသွင်ဖြင့် စာမျက်နှာကိုအရင်ဖတ်ခြင်း (Cookies များကို သိမ်းဆည်းရန် temp_client ကိုသုံးမည်)
             async with httpx.AsyncClient(headers={'User-Agent': req_ua}, follow_redirects=True) as temp_client:
                 page_res = await temp_client.get(clean_url)
                 if page_res.status_code != 200:
@@ -85,8 +106,6 @@ async def download_proxy(request: Request, url: str, key: str = None):
                 
                 html_content = page_res.text
                 target_link = None
-                
-                # 2. MediaFire မှပေးသော Session Cookies ကို ယူထားခြင်း
                 cookies = temp_client.cookies 
 
                 match = re.search(r"https?://download[0-9]+\.mediafire\.com/[a-zA-Z0-9_-]+/[a-zA-Z0-9_-]+/[^\s'\"]+", html_content)
@@ -103,14 +122,21 @@ async def download_proxy(request: Request, url: str, key: str = None):
                     if target_link.startswith("//"): target_link = f"https:{target_link}"
                     elif target_link.startswith("/"): target_link = f"https://www.mediafire.com{target_link}"
                     
-                    # Target Link, UA နှင့် Cookies ကို Stream ထံ ပို့ပေးခြင်း
+                    # နောက်ထပ် Range Request များအတွက် Memory ထဲတွင် မှတ်သားထားခြင်း
+                    MEDIAFIRE_CACHE[clean_url] = {
+                        'link': target_link,
+                        'ua': req_ua,
+                        'cookies': dict(cookies),
+                        'time': current_time
+                    }
+                    
                     return await stream_file(target_link, range_header, filename, referer=clean_url, req_ua=req_ua, cookies=cookies)
                 else:
                     raise HTTPException(status_code=404, detail="Could not extract direct download link")
 
         except Exception as e:
             print(f"MediaFire Logic Error: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"415 Error Avoided: {str(e)}")
+            raise HTTPException(status_code=500, detail=str(e))
 
     # --- Google Drive Section ---
     elif "drive.google.com" in clean_url:
@@ -141,17 +167,14 @@ async def download_proxy(request: Request, url: str, key: str = None):
         return await stream_file(clean_url, range_header, filename, req_ua=req_ua)
 
 async def stream_file(target_url, range_header, filename, referer=None, req_ua=None, cookies=None):
-    # တူညီသော User-Agent ကို ပြန်သုံးခြင်း
     headers = {'User-Agent': req_ua if req_ua else ua.random}
     if range_header: headers['Range'] = range_header
     if referer: headers['Referer'] = referer
     
     try:
-        # Cookies ပါထည့်၍ Request လုပ်ခြင်း
         req = client.build_request("GET", target_url, headers=headers, cookies=cookies)
         r = await client.send(req, stream=True)
         
-        # HTML ပြန်လာရင် ဗီဒီယိုမဟုတ်ကြောင်း သေချာစေရန်
         ctype = r.headers.get('content-type', '').lower()
         if 'text/html' in ctype:
             await r.aclose()