Update main.py

main.py

@@ -25,6 +25,7 @@ from app.gs_code import GSCodeApp
 from app.gm_crate import GmailLogic
 from app.gs_daily import GSDailyApp
 from app.invite_logic import DiscordInviteLogic
+from app.tik_dldrct import TikTokDirect
 
 xs_app = XoSoApp()
 lq_app = ClLqApp()
@@ -32,6 +33,7 @@ gm_app = GmailLogic()
 gs_app = GSDailyApp()
 bypass_app = BypassUV()
 tik_app = TikTokDownloader()
+tik_direct = TikTokDirect()
 hentai_app = HentaiApp()
 ios_app = IOSAppFetcher()
 gs_code_app = GSCodeApp()
@@ -76,13 +78,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_app.start()
+        bypass_app.start(), ios_app.start(), xs_app.start(), tik_direct.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_app.stop()
+        bypass_app.stop(), ios_app.stop(), xs_app.stop(), tik_direct.stop()
     )
 
 app = FastAPI(lifespan=lifespan)
@@ -163,42 +165,54 @@ async def hentai_random(apikey: Optional[str] = Query(None), limit: str = Query(
     return await ht_random_app.get_random(key, API_KEY, limit)
 
 @app.get("/api/v1/tiktok/download")
-async def tiktok_download(url: str = Query(...), apikey: Optional[str] = Query(None)):
-    if apikey != API_KEY:
-        raise HTTPException(status_code=403, detail="Key sai rồi")
+async def tiktok_download(url: str = Query(...), hd: int = Query(1), 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. Go touch grass")
     
-    res = await tik_app.fetch_video_info(url)
+    res = await tik_app.fetch_video(url, hd)
     if not res["ok"]:
         raise HTTPException(status_code=400, detail=res["error"])
     return res
 
 @app.get("/api/v1/tiktok/dl-direct")
-async def tiktok_direct_handler(token: str = Query(...)):
-    # KHÔNG CẦN API KEY Ở ĐÂY
-    
-    real_url = await tik_app.get_real_stream_url(token)
-    
-    if real_url == "limit_exceeded":
-        raise HTTPException(status_code=403, detail="Link này đã hết lượt tải (Max 3 lần)")
-    if not real_url:
-        raise HTTPException(status_code=404, detail="Không tìm thấy file hoặc token hết hạn")
+async def tiktok_direct(
+    url: str = Query(None),
+    token: str = Query(None),
+    type: str = Query("mp4"),
+    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")
 
-    async def stream():
-        async with httpx.AsyncClient() as client:
-            async with client.stream("GET", real_url) as r:
-                async for chunk in r.aiter_bytes():
-                    yield chunk
+    if url:
+        res = await tik_direct.generate_token(url, type)
+        if not res["ok"]:
+            raise HTTPException(status_code=400, detail=res["error"])
+        return res
 
-    # Tự động lấy type từ cache dựa trên token
-    media_type = tik_app.token_cache[token]["type"]
-    filename = f"celeste_{int(time.time())}.{media_type}"
+    if token:
+        data = tik_direct.get_token_data(token)
+        if not data:
+            raise HTTPException(status_code=404, detail="Token expired or invalid")
 
-    return StreamingResponse(
-        stream(),
-        media_type="video/mp4" if media_type == "mp4" else "audio/mpeg",
-        headers={"Content-Disposition": f"attachment; filename={filename}"}
-    )
+        async def stream():
+            async with httpx.AsyncClient() as client:
+                async with client.stream("GET", data["url"]) as r:
+                    async for chunk in r.aiter_bytes():
+                        yield chunk
+
+        filename = f"tiktok_{int(time.time())}.{data['type']}"
+
+        return StreamingResponse(
+            stream(),
+            media_type="video/mp4" if data["type"] == "mp4" else "audio/mpeg",
+            headers={"Content-Disposition": f"attachment; filename={filename}"}
+        )
 
+    raise HTTPException(status_code=400, detail="Missing url or token")
 
 @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")):