Update main.py

main.py

@@ -1,134 +1,25 @@
-import asyncio
-import requests
-import time
 from fastapi import FastAPI
-from playwright.async_api import async_playwright
+from contextlib import asynccontextmanager
+from app.enka_logic import EnkaApp
 
-app = FastAPI()
+enka = EnkaApp()
 
-IMGBB_API_KEY = "093dee93ec19418d26cc76f40e053725"
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    await enka.start()
+    yield
+    await enka.stop()
 
-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',
-        '--disable-gpu'
-    ])
-    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"
-    )
-    page = await context.new_page()
-
-    # CHẶN RÁC ĐỂ TĂNG TỐC
-    async def block_waste(route):
-        bad_types = ["media", "font", "other"]
-        if fast_mode: bad_types.extend(["image", "stylesheet"])
-        if any(x in route.request.url for x in ["google", "analytics", "doubleclick"]):
-            await route.abort()
-        else:
-            await route.continue_()
-
-    await page.route("**/*", block_waste)
-    await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=60000)
-    return browser, page
+app = FastAPI(lifespan=lifespan)
 
 @app.get("/info")
-async def info(uid: str):
-    start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=True)
-    try:
-        # CLICK NAMECARD (Dùng try/except thay cho .catch)
-        try:
-            await page.click('button[data-icon="cards"]', timeout=3000)
-            await page.wait_for_selector('.Loda .bg', timeout=3000)
-        except:
-            pass # Bỏ qua nếu không có namecard hoặc quá chậm
-        
-        data = await page.evaluate("""
-            () => {
-                const arText = document.querySelector('.ar')?.innerText || "";
-                const stats = Array.from(document.querySelectorAll('tr.stat')).map(tr => {
-                    const style = tr.querySelector('td.icon')?.getAttribute('style') || "";
-                    const iconMatch = style.match(/url\\(["']?(.*?)["']?\\)/);
-                    return {
-                        value: tr.querySelector('td:first-child')?.innerText.trim(),
-                        label: tr.querySelector('td:last-child')?.innerText.trim(),
-                        icon: iconMatch ? "https://enka.network" + iconMatch[1].replace(/\\\\/g, "") : null
-                    };
-                });
-
-                const ncBg = document.querySelector('.Loda .bg')?.style.backgroundImage;
-                const ncMatch = ncBg ? ncBg.match(/url\\(["']?(.*?)["']?\\)/) : null;
-                
-                const characters = Array.from(document.querySelectorAll('.CharacterList .avatar')).map(el => {
-                    const style = el.querySelector('figure.chara')?.getAttribute('style') || "";
-                    const path = style.match(/url\\((.*)\\)/)?.[1].replace(/["']/g, "");
-                    return {
-                        name: path?.split('_').pop().split('.')[0],
-                        level: el.querySelector('.level')?.innerText,
-                        icon: "https://enka.network" + path
-                    };
-                }).filter(i => i.name);
-
-                return {
-                    player: {
-                        name: document.querySelector('.details h1')?.innerText,
-                        signature: document.querySelector('.signature')?.innerText,
-                        ar: arText.match(/AR\\s*(\\d+)/i)?.[1],
-                        wl: arText.match(/WL\\s*(\\d+)/i)?.[1],
-                        avatar: "https://enka.network" + document.querySelector('.avatar-icon img')?.getAttribute('src'),
-                        namecard: ncMatch ? "https://enka.network" + ncMatch[1] : null,
-                        stats: stats
-                    },
-                    characters: characters
-                };
-            }
-        """)
-        
-        await browser.close()
-        return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, **data}
-    except Exception as e:
-        if 'browser' in locals(): await browser.close()
-        return {"error": str(e)}
+async def get_info(uid: str):
+    return await enka.fetch_info(uid)
 
 @app.get("/gen")
-async def generate(uid: str, char: str = None):
-    start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=False)
-    try:
-        if char:
-            target = char.strip().capitalize()
-            try:
-                await page.click(f"//figure[contains(@style, '{target}')]", timeout=5000)
-                await asyncio.sleep(0.8)
-            except: pass
-
-        await page.click('button[data-icon="image"]')
-        await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=15000)
-        
-        img_base64 = await page.evaluate("""
-            async () => {
-                const img = document.querySelector('.Loda img');
-                const res = await fetch(img.src);
-                const blob = await res.blob();
-                return new Promise(r => {
-                    const rd = new FileReader();
-                    rd.onloadend = () => r(rd.result.split(',')[1]);
-                    rd.readAsDataURL(blob);
-                });
-            }
-        """)
-        
-        await browser.close()
-        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_API_KEY, "image": img_base64}, timeout=20)
-        return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, "card_url": res.json().get('data', {}).get('url')}
-    except Exception as e:
-        if 'browser' in locals(): await browser.close()
-        return {"error": str(e)}
+async def get_gen(uid: str, char: str = None):
+    return await enka.fetch_gen(uid, char)
 
 @app.get("/")
 def home():
-    return {"status": "Enka API Fixed & Running"}
+    return {"status": "Modular API is Live", "logic": "Loaded from ./app/enka_logic.py"}