58 lines
2.8 KiB
Python
58 lines
2.8 KiB
Python
from urllib.parse import urlparse, parse_qs,unquote,quote
|
|
from fastapi import APIRouter, HTTPException, Response, Request
|
|
from curl_cffi.requests import AsyncSession
|
|
|
|
router = APIRouter()
|
|
User_Agent= "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36 OPR/111.0.0.0"
|
|
async def fetch_m3u8(url):
|
|
"""Fetch the M3U8 file from the URL and store it in memory."""
|
|
async with AsyncSession() as client:
|
|
response = await client.get(url,headers = {"User-Agent": User_Agent, "user-agent": User_Agent})
|
|
response.raise_for_status() # Raise an error for bad responses
|
|
return response.text
|
|
|
|
@router.get("/clone/manifest.m3u8")
|
|
async def clone_m3u8(d: str = None):
|
|
if d:
|
|
try:
|
|
d = unquote(d)
|
|
m3u8_content = await fetch_m3u8(d)
|
|
return Response(content=m3u8_content, media_type='application/vnd.apple.mpegurl')
|
|
except Exception as e:
|
|
print(f"Failed to fetch M3U8 file: {e}")
|
|
raise HTTPException(status_code=404, detail="M3U8 content not found")
|
|
#This is just for testing, ignore that endpoint and the func
|
|
|
|
|
|
@router.api_route("/vixcloud/manifest.m3u8", methods=["GET", "HEAD"])
|
|
async def clone2_m3u8(d:str,token:str,expires:str,h:str = None, b:str = None, request: Request = None):
|
|
try:
|
|
m3u8 = f'{d}?token={token}&expires={expires}'
|
|
if h:
|
|
m3u8 = m3u8 + f'&h={h}'
|
|
if b:
|
|
m3u8 = m3u8 + f'&b={b}'
|
|
forwarded_proto = request.headers.get("x-forwarded-proto")
|
|
scheme = forwarded_proto if forwarded_proto else request.url.scheme
|
|
instance_url = f"{scheme}://{request.url.netloc}"
|
|
m3u8_content = await fetch_m3u8(m3u8)
|
|
modified_playlist = m3u8_content.replace("https://vixcloud.co/playlist/", f"{instance_url}/clony/")
|
|
|
|
return Response(content=modified_playlist, media_type='application/vnd.apple.mpegurl')
|
|
except Exception as e:
|
|
print(f"Failed to fetch M3U8 file: {e}")
|
|
raise HTTPException(status_code=404, detail="M3U8 content not found")
|
|
|
|
@router.api_route("/clony/{segment:path}", methods=["GET", "HEAD"])
|
|
async def clony_m3u8(segment: str, request: Request):
|
|
base_url = "https://vixcloud.co/playlist/"
|
|
full_url = f"{base_url}{segment}?{request.query_params}"
|
|
m3u8_content = await fetch_m3u8(full_url)
|
|
return Response(content=m3u8_content, media_type='application/vnd.apple.mpegurl')
|
|
|
|
@router.api_route("/storage/enc.key", methods=["GET", "HEAD"])
|
|
async def get_key():
|
|
async with AsyncSession() as client:
|
|
response = await client.get("https://vixcloud.co/storage/enc.key",headers = {"User-Agent": User_Agent, "user-agent": User_Agent})
|
|
response.raise_for_status() # Raise an error for bad responses
|
|
return response.text |