Update app/aternos_api.py

app/aternos_api.py

@@ -1,83 +1,79 @@
 import asyncio
 from playwright.async_api import async_playwright
-from playwright_stealth import stealth_async 
+from playwright_stealth import stealth_async
 
 class AternosAPI:
     def __init__(self):
         self.playwright = None
+        # Chỉ cần 1 con xoay này là đủ, không cần list lằng nhằng
+        self.proxy_config = {
+            "server": "socks5://p.webshare.io:80",
+            "username": "szusjfnw-rotate",
+            "password": "mftzj9sljp5t"
+        }
 
     async def start(self):
-        self.playwright = await async_playwright().start()
+        if not self.playwright:
+            self.playwright = await async_playwright().start()
 
     async def stop(self):
         if self.playwright:
             await self.playwright.stop()
             self.playwright = None
 
-    async def check_with_proxy(self, proxy: str):
+    async def check_access(self):
+        """Hàm chính để FastAPI gọi vào"""
         if not self.playwright:
             await self.start()
 
-        browser = None
-        try:
-            # Dùng các args để giảm thiểu việc bị phát hiện
-            browser = await self.playwright.chromium.launch(
-                headless=True,
-                proxy={"server": proxy},
-                args=[
-                    "--disable-blink-features=AutomationControlled",
-                    "--no-sandbox",
-                    "--disable-dev-shm-usage",
-                ]
-            )
+        browser = await self.playwright.chromium.launch(
+            headless=True,
+            proxy=self.proxy_config,
+            args=[
+                "--no-sandbox",
+                "--disable-blink-features=AutomationControlled",
+                "--disable-dev-shm-usage"
+            ]
+        )
 
-            context = await browser.new_context(
-                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()
-            
-            # Sử dụng stealth để ẩn danh
-            # Nếu vẫn báo lỗi ImportError, hãy thay dòng dưới bằng: 
-            # from playwright_stealth import stealth_async -> import trực tiếp trong hàm
-            await stealth_async(page)
+        # Giả lập vân tay trình duyệt thật
+        context = await browser.new_context(
+            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36",
+            viewport={'width': 1920, 'height': 1080}
+        )
+        
+        page = await context.new_page()
+        # Xóa dấu vết bot
+        await stealth_async(page)
 
-            # Thử truy cập
-            resp = await page.goto(
-                "https://aternos.org/server/",
-                wait_until="networkidle", # Chờ cho network ổn định
-                timeout=30000
-            )
+        try:
+            # Bước 1: Truy cập trang chủ để lấy "vía" (cookies nền)
+            await page.goto("https://aternos.org/go/", wait_until="networkidle", timeout=60000)
+            await asyncio.sleep(3) 
 
-            # Đợi thêm một chút để Cloudflare challenge (nếu có) tự giải quyết
+            # Bước 2: Vào thẳng trang server
+            response = await page.goto("https://aternos.org/server/", wait_until="domcontentloaded", timeout=60000)
+            
+            # Đợi thêm chút nếu gặp màn hình chờ của Cloudflare
             await asyncio.sleep(5)
 
             title = await page.title()
             content = await page.content()
-            
-            is_blocked = "just a moment" in title.lower() or "cloudflare" in content.lower()
+            status = response.status if response else 0
+
+            # Logic kiểm tra chuẩn xác
+            is_cf_challenge = "just a moment" in title.lower() or "cloudflare" in content.lower()
+            is_success = status == 200 and not is_cf_challenge
 
             return {
-                "ok": not is_blocked,
-                "status": resp.status if resp else None,
+                "ok": is_success,
+                "status": status,
                 "title": title,
-                "blocked": is_blocked
+                "proxy_working": status != 407,
+                "cloudflare_blocked": is_cf_challenge
             }
 
         except Exception as e:
-            return {"ok": False, "error": str(e), "blocked": True}
+            return {"ok": False, "error": str(e)}
         finally:
-            if browser:
-                await browser.close()
-
-    async def check_access(self):
-        # Hàm này để gọi từ FastAPI
-        PROXIES = [
-            "http://szusjfnw:mftzj9sljp5t@31.59.20.176:6754",
-            # ... bỏ danh sách proxy của bạn vào đây
-        ]
-        for p in PROXIES:
-            res = await self.check_with_proxy(p)
-            if res["ok"]:
-                return res
-        return {"ok": False, "message": "All proxies failed"}
+            await browser.close()