Restore commit e77a637 - main.py

main.py

@@ -9,8 +9,6 @@ from fastapi import FastAPI, Query, Request
 from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse, Response
 from urllib.parse import quote
 import requests
-import urllib3
-urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 from bs4 import BeautifulSoup
 
 app = FastAPI()
@@ -133,81 +131,24 @@ def _parse_match_from_li(li, status_type="live"):
     return {"home":home_el.get_text(strip=True),"away":away_el.get_text(strip=True),"score":score or"VS","minute":minute,"league":league,"time":time_el.get_text(strip=True) if time_el else "","event_id":event_id,"home_logo":home_logo.get("src","") if home_logo else "","away_logo":away_logo.get("src","") if away_logo else "","status":status_type}
 
 # ===== VIDEO PROXY =====
-def _get_proxy_headers(url):
-    """Get appropriate headers for proxying a URL."""
-    headers = dict(HEADERS)
-    if "vtvdigital" in url:
-        headers["Referer"] = "https://vtv.vn/"
-        headers["Origin"] = "https://vtv.vn"
-    elif "fptplay" in url:
-        headers["Referer"] = "https://fptplay.vn/"
-        headers["Origin"] = "https://fptplay.vn"
-    elif "canthotv" in url:
-        headers["Referer"] = "https://canthotv.vn/"
-    elif "xemtv" in url:
-        headers["Referer"] = "https://xemtv.us/"
-    return headers
-
 @app.get("/api/proxy/m3u8")
 def proxy_m3u8(url: str = Query(...)):
     try:
-        headers = _get_proxy_headers(url)
-        verify_ssl = "vtvdigital" not in url
-        r = requests.get(url, headers=headers, timeout=15, verify=verify_ssl)
+        r = requests.get(url, headers=HEADERS, timeout=15)
         if r.status_code != 200: return Response(status_code=502, content="upstream error")
         lines = r.text.strip().split('\n'); rewritten = []
-        base_url = url.rsplit('/', 1)[0] + '/'
         for line in lines:
-            line_s = line.strip()
-            if not line_s or line_s.startswith('#'):
-                if 'URI=' in line_s:
-                    line_s = re.sub(
-                        r'URI="([^"]+)"',
-                        lambda m: 'URI="' + ("/api/proxy/seg?url=" + quote(m.group(1), safe="")
-                                   if not m.group(1).startswith('http')
-                                   else "/api/proxy/seg?url=" + quote(m.group(1), safe="")) + '"',
-                        line_s
-                    )
-                rewritten.append(line_s)
-            else:
-                seg_url = line_s
-                if not seg_url.startswith('http'):
-                    seg_url = base_url + seg_url
-                rewritten.append("/api/proxy/seg?url=" + quote(seg_url, safe=""))
-        return Response(content='\n'.join(rewritten).encode('utf-8'), media_type="application/vnd.apple.mpegurl", headers={"Access-Control-Allow-Origin":"*","Cache-Control":"no-cache"})
+            if line.startswith('#') or not line.strip(): rewritten.append(line)
+            else: rewritten.append("/api/proxy/seg?url=" + quote(line.strip(), safe=""))
+        return Response(content='\n'.join(rewritten).encode('utf-8'), media_type="application/vnd.apple.mpegurl", headers={"Access-Control-Allow-Origin":"*","Cache-Control":"public, max-age=300"})
     except: return Response(status_code=502, content="proxy error")
 
 @app.get("/api/proxy/seg")
 def proxy_segment(url: str = Query(...)):
     try:
-        headers = _get_proxy_headers(url)
-        verify_ssl = "vtvdigital" not in url
-        r = requests.get(url, headers=headers, timeout=30, verify=verify_ssl)
+        r = requests.get(url, headers=HEADERS, timeout=30)
         if r.status_code != 200: return Response(status_code=502, content="upstream error")
         data = r.content
-        # Handle sub-playlist
-        if b"#EXTM3U" in data[:500] or ".m3u8" in url:
-            lines = data.decode("utf-8", errors="replace").split('\n')
-            rewritten = []
-            base_url = url.rsplit('/', 1)[0] + '/'
-            for line in lines:
-                line_s = line.strip()
-                if not line_s or line_s.startswith('#'):
-                    if 'URI=' in line_s:
-                        line_s = re.sub(
-                            r'URI="([^"]+)"',
-                            lambda m: 'URI="' + ("/api/proxy/seg?url=" + quote(m.group(1), safe="")
-                                       if not m.group(1).startswith('http')
-                                       else "/api/proxy/seg?url=" + quote(m.group(1), safe="")) + '"',
-                            line_s
-                        )
-                    rewritten.append(line_s)
-                else:
-                    seg_url = line_s
-                    if not seg_url.startswith('http'):
-                        seg_url = base_url + seg_url
-                    rewritten.append("/api/proxy/seg?url=" + quote(seg_url, safe=""))
-            return Response(content='\n'.join(rewritten).encode("utf-8"), media_type="application/vnd.apple.mpegurl", headers={"Access-Control-Allow-Origin": "*", "Cache-Control": "no-cache"})
         if len(data) > 188 and data[0:4] == b'\x89PNG' and data[188] == 0x47: data = data[188:]
         return Response(content=data, media_type="video/mp2t", headers={"Access-Control-Allow-Origin":"*","Cache-Control":"public, max-age=3600"})
     except: return Response(status_code=502, content="proxy error")
@@ -215,7 +156,7 @@ def proxy_segment(url: str = Query(...)):
 @app.get("/api/proxy/video")
 def proxy_video(url: str = Query(...), request: Request = None):
     try:
-        req_headers = _get_proxy_headers(url)
+        req_headers = dict(HEADERS)
         if request and request.headers.get("range"): req_headers["Range"] = request.headers["range"]
         r = requests.get(url, headers=req_headers, timeout=30, stream=True)
         resp_headers = {"Access-Control-Allow-Origin":"*","Accept-Ranges":"bytes","Content-Type":r.headers.get("Content-Type","video/mp4")}