Update main.py

main.py

@@ -1,163 +1,199 @@
-import asyncio
-import requests
 import time
+import asyncio
+import aiohttp
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
+from cachetools import TTLCache
 
-app = FastAPI()
-
+# ================= CONFIG =================
 IMGBB_API_KEY = "093dee93ec19418d26cc76f40e053725"
+ENKA_URL = "https://enka.network/u/{uid}/"
+
+# cache RAM: 100 UID, sống 60s
+INFO_CACHE = TTLCache(maxsize=100, ttl=60)
+
+# playwright globals
+playwright = None
+browser = None
+context = None
+
+app = FastAPI(title="Enka API Pro Max")
+
+
+# ================= STARTUP / SHUTDOWN =================
+@app.on_event("startup")
+async def startup():
+    global playwright, browser, context
+
+    playwright = await async_playwright().start()
+    browser = await playwright.chromium.launch(
+        headless=True,
+        args=[
+            "--no-sandbox",
+            "--disable-dev-shm-usage",
+            "--disable-gpu",
+            "--disable-background-networking",
+            "--disable-extensions",
+            "--disable-sync",
+            "--disable-default-apps",
+        ]
+    )
 
-async def get_page(uid: str, fast_mode=False):
-    p = await async_playwright().start()
-    browser = await p.chromium.launch(headless=True, args=['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'])
     context = await browser.new_context(
-        viewport={'width': 1280, 'height': 800},
-        user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+        viewport={"width": 1280, "height": 800},
+        user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64)"
     )
+
+
+@app.on_event("shutdown")
+async def shutdown():
+    await context.close()
+    await browser.close()
+    await playwright.stop()
+
+
+# ================= UTILS =================
+async def new_page(fast_mode=True):
     page = await context.new_page()
 
     if fast_mode:
-        # Chặn bớt tài nguyên nhưng vẫn để lại Images vì cần link Namecard và Stats icon
-        await page.route("**/*", lambda route: route.abort() if route.request.resource_type in ["media", "font"] else route.continue_())
+        await page.route(
+            "**/*",
+            lambda route: route.abort()
+            if route.request.resource_type in [
+                "media", "font", "stylesheet", "other"
+            ]
+            else route.continue_()
+        )
+
+    return page
 
-    url = f"https://enka.network/u/{uid}/"
-    await page.goto(url, wait_until="networkidle", timeout=60000)
-    return browser, page
 
+# ================= /INFO =================
 @app.get("/info")
 async def info(uid: str):
-    start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=True)
+    if uid in INFO_CACHE:
+        return {"cached": True, **INFO_CACHE[uid]}
+
+    start = time.time()
+    page = await new_page(fast_mode=True)
+
     try:
-        # BƯỚC MỚI: Click lấy Namecard
-        namecard_url = None
-        try:
-            # Tìm nút Danh Thiếp qua data-icon="cards"
-            card_btn = 'button[data-icon="cards"]'
-            await page.wait_for_selector(card_btn, timeout=5000)
-            await page.click(card_btn)
-            
-            # Đợi panel Loda hiện lên
-            namecard_selector = '.Loda .bg'
-            await page.wait_for_selector(namecard_selector, timeout=5000)
-            
-            # Lấy link ảnh từ background-image
-            namecard_url = await page.evaluate("""
-                () => {
-                    const bgDiv = document.querySelector('.Loda .bg');
-                    if (!bgDiv) return null;
-                    const style = bgDiv.style.backgroundImage;
-                    const match = style.match(/url\\(["']?(.*?)["']?\\)/);
-                    return match ? "https://enka.network" + match[1] : null;
-                }
-            """)
-            
-            # Đóng panel để tránh block UI (không bắt buộc nhưng nên làm)
-            await page.keyboard.press("Escape")
-        except Exception as e:
-            print(f"Lỗi lấy Namecard: {e}")
-
-        # LẤY CÁC THÔNG TIN CÒN LẠI
+        await page.goto(
+            ENKA_URL.format(uid=uid),
+            wait_until="domcontentloaded",
+            timeout=30000
+        )
+
         data = await page.evaluate("""
-            () => {
-                const name = document.querySelector('.details h1')?.innerText;
-                const signature = document.querySelector('.signature')?.innerText;
-                const avatar_img = document.querySelector('.avatar-icon img')?.getAttribute('src');
-                
-                const arContainer = document.querySelector('.ar');
-                let ar = null, wl = null;
-                if (arContainer) {
-                    const text = arContainer.innerText;
-                    const arMatch = text.match(/AR\\s*(\\d+)/i);
-                    const wlMatch = text.match(/WL\\s*(\\d+)/i);
-                    ar = arMatch ? arMatch[1] : null;
-                    wl = wlMatch ? wlMatch[1] : null;
-                }
+        () => {
+            const q = s => document.querySelector(s);
+            const bg = el => {
+                const m = el?.style?.backgroundImage?.match(/url\\(["']?(.*?)["']?\\)/);
+                return m ? "https://enka.network" + m[1] : null;
+            };
+
+            const arText = q('.ar')?.innerText || "";
 
-                const stats = Array.from(document.querySelectorAll('tr.stat')).map(tr => {
-                    const val = tr.querySelector('td:first-child')?.innerText.trim();
-                    const label = tr.querySelector('td:last-child')?.innerText.trim();
-                    const iconTd = tr.querySelector('td.icon');
-                    let iconUrl = null;
-                    if (iconTd) {
-                        const style = iconTd.getAttribute('style');
-                        const match = style.match(/url\\(["']?(.*?)["']?\\)/);
-                        if (match && match[1]) iconUrl = "https://enka.network" + match[1].replace(/\\\\/g, "");
-                    }
-                    return { value: val, label: label, icon: iconUrl };
-                });
-
-                const characters = Array.from(document.querySelectorAll('.CharacterList .avatar')).map(el => {
-                    const figure = el.querySelector('figure.chara');
-                    if (!figure) return null;
-                    const style = figure.getAttribute('style');
-                    const icon_path = style.match(/url\\((.*)\\)/)[1].replace(/["']/g, "");
-                    const level = el.querySelector('.level')?.innerText || "0";
-                    const charName = icon_path.split('_').pop().split('.')[0];
-                    return { name: charName, level, icon: "https://enka.network" + icon_path };
-                }).filter(i => i !== null);
-
-                return {
-                    player: { name, ar, wl, signature, avatar: avatar_img ? "https://enka.network" + avatar_img : null, stats },
-                    characters: characters
-                };
-            }
+            return {
+                player: {
+                    name: q('.details h1')?.innerText || null,
+                    signature: q('.signature')?.innerText || null,
+                    avatar: q('.avatar-icon img')?.src
+                        ? "https://enka.network" + q('.avatar-icon img').getAttribute("src")
+                        : null,
+                    ar: arText.match(/AR\\s*(\\d+)/)?.[1] || null,
+                    wl: arText.match(/WL\\s*(\\d+)/)?.[1] || null,
+                },
+                characters: [...document.querySelectorAll('.CharacterList .avatar')]
+                    .map(e => {
+                        const f = e.querySelector('figure');
+                        if (!f) return null;
+                        const icon = f.style.backgroundImage
+                            ?.match(/url\\((.*)\\)/)?.[1];
+                        return icon ? {
+                            icon: "https://enka.network" + icon.replace(/['"]/g, ""),
+                            level: e.querySelector('.level')?.innerText || "0"
+                        } : null;
+                    })
+                    .filter(Boolean)
+            };
+        }
         """)
-        
-        await browser.close()
-        # Gộp Namecard vào JSON kết quả
-        data['player']['namecard'] = namecard_url
-        
-        return {
-            "execution_time": f"{round(time.time() - start_time, 2)}s",
+
+        await page.close()
+
+        result = {
             "uid": uid,
+            "execution_time": f"{round(time.time() - start, 2)}s",
             **data
         }
+
+        INFO_CACHE[uid] = result
+        return result
+
     except Exception as e:
-        if 'browser' in locals(): await browser.close()
+        await page.close()
         return {"error": str(e)}
 
+
+# ================= /GEN =================
 @app.get("/gen")
-async def generate(uid: str, char: str = None):
-    start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=False)
+async def generate(uid: str):
+    start = time.time()
+    page = await new_page(fast_mode=False)
+
     try:
-        if char:
-            formatted_char = char.strip().capitalize()
-            char_selector = f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{formatted_char}')]"
-            try:
-                await page.wait_for_selector(char_selector, timeout=5000)
-                await page.click(char_selector)
-                await asyncio.sleep(1)
-            except: pass
+        await page.goto(
+            ENKA_URL.format(uid=uid),
+            wait_until="domcontentloaded",
+            timeout=30000
+        )
 
         await page.click('button[data-icon="image"]')
         await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=20000)
-        
+
         img_base64 = await page.evaluate("""
-            async () => {
-                const img = document.querySelector('.Loda img[src^="blob:"]');
-                const res = await fetch(img.src);
-                const blob = await res.blob();
-                return new Promise(r => {
-                    const reader = new FileReader();
-                    reader.onloadend = () => r(reader.result.split(',')[1]);
-                    reader.readAsDataURL(blob);
-                });
-            }
+        async () => {
+            const img = document.querySelector('.Loda img');
+            const res = await fetch(img.src);
+            const blob = await res.blob();
+
+            return await new Promise(resolve => {
+                const r = new FileReader();
+                r.onloadend = () => resolve(r.result.split(',')[1]);
+                r.readAsDataURL(blob);
+            });
+        }
         """)
-        await browser.close()
-        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_API_KEY, "image": img_base64})
+
+        await page.close()
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                "https://api.imgbb.com/1/upload",
+                data={
+                    "key": IMGBB_API_KEY,
+                    "image": img_base64
+                }
+            ) as r:
+                js = await r.json()
+
         return {
-            "execution_time": f"{round(time.time() - start_time, 2)}s",
             "uid": uid,
-            "card_url": res.json().get('data', {}).get('url')
+            "execution_time": f"{round(time.time() - start, 2)}s",
+            "card_url": js.get("data", {}).get("url")
         }
+
     except Exception as e:
-        if 'browser' in locals(): await browser.close()
+        await page.close()
         return {"error": str(e)}
 
+
+# ================= ROOT =================
 @app.get("/")
 def home():
-    return {"status": "Enka API Pro Max is Live"}
+    return {
+        "status": "Enka API Pro Max is Live",
+        "cache_size": len(INFO_CACHE)
+    }