Update main.py

main.py

@@ -22,6 +22,8 @@ from app.ht_noti import HentaiApp
 from app.ios_app import IOSAppFetcher
 from app.tik_dl import TiktokDl
 from app.tik_cdn import TiktokCdn
+from app.yt_dl import YouTubeDl
+from app.yt_cdn import YouTubeCdn
 from app.byps_uv import BypassUV
 from app.gs_code import GSCodeApp
 from app.gm_crate import GmailLogic
@@ -36,6 +38,8 @@ bypass_app = BypassUV()
 tik_app = TiktokDl()
 tik_direct = TiktokCdn()
 hentai_app = HentaiApp()
+yt_app = YouTubeDl()
+yt_cdn = YouTubeCdn()
 ios_app = IOSAppFetcher()
 gs_code_app = GSCodeApp()
 invite_app = DiscordInviteLogic()
@@ -79,13 +83,13 @@ async def lifespan(app: FastAPI):
     await asyncio.gather(
         invite_app.start(), lq_app.start(), gs_app.start(),
         gs_code_app.start(), hentai_app.start(), ht_random_app.start(),
-        bypass_app.start(), ios_app.start(), xs_app.start(), tik_direct.start()
+        bypass_app.start(), ios_app.start(), xs_app.start(), tik_direct.start(), yt_cdn.start())
     )
     yield
     await asyncio.gather(
         invite_app.stop(), lq_app.stop(), gs_app.stop(),
         gs_code_app.stop(), hentai_app.stop(), ht_random_app.stop(),
-        bypass_app.stop(), ios_app.stop(), xs_app.stop(), tik_direct.stop()
+        bypass_app.stop(), ios_app.stop(), xs_app.stop(), tik_direct.stop(), yt_cdn.stop())
     )
 
 app = FastAPI(lifespan=lifespan)
@@ -262,7 +266,47 @@ async def tiktok_cdn(
         headers={"Content-Disposition": f"attachment; filename=Celeste_{int(time.time())}.{media_type_req}"}
     )
 
-    raise HTTPException(status_code=400, detail="Provide either 'url' or 'token'")
+    raise HTTPException(status_code=400, detail="Provide either 'url' or '
+
+@app.get("/api/v1/youtube/dl")
+async def youtube_dl(
+    url: str = Query(...),
+    format: str = Query("1080"),
+    apikey: Optional[str] = Query(None),
+    x_api_key: Optional[str] = Header(None, alias="X-API-Key")
+):
+    key = x_api_key or apikey
+    if not API_KEY or key != API_KEY:
+        raise HTTPException(status_code=403, detail="API key invalid")
+    
+    return await yt_app.fetch_info(url, yt_cdn, format)
+
+@app.get("/api/v1/youtube/cdn")
+async def youtube_cdn_proxy(
+    token: str = Query(...),
+):
+    data = yt_cdn.consume_token(token)
+    if not data:
+        raise HTTPException(status_code=404, detail="Token invalid or expired")
+
+    async def stream_file():
+        async with httpx.AsyncClient(follow_redirects=True, timeout=120.0) as client:
+            async with client.stream("GET", data["url"]) as r:
+                async for chunk in r.aiter_bytes(chunk_size=1024*1024):
+                    yield chunk
+
+    media_types = {
+        "mp4": "video/mp4",
+        "mp3": "audio/mpeg",
+        "wav": "audio/wav",
+        "jpg": "image/jpeg"
+    }
+    
+    return StreamingResponse(
+        stream_file(),
+        media_type=media_types.get(data["type"], "application/octet-stream"),
+        headers={"Content-Disposition": f"attachment; filename=Celeskry_YT_{int(time.time())}.{data['type']}"}
+    )
 
 @app.get("/api/v1/bypass")
 async def bypass_link(url: str = Query(...), apikey: Optional[str] = Query(None), x_api_key: Optional[str] = Header(None, alias="X-API-Key")):