Update app/aternos_api.py

app/aternos_api.py

@@ -1,6 +1,6 @@
 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):
@@ -12,67 +12,72 @@ class AternosAPI:
     async def stop(self):
         if self.playwright:
             await self.playwright.stop()
+            self.playwright = None
 
     async def check_with_proxy(self, proxy: str):
-        # Cấu hình trình duyệt cực kỳ cẩn thận
-        browser = await self.playwright.chromium.launch(
-            headless=True, # Nếu chạy Docker/Linux thì để True, nhưng tốt nhất là False nếu có thể
-            proxy={"server": proxy},
-            args=[
-                "--disable-blink-features=AutomationControlled",
-                "--no-sandbox",
-                "--disable-dev-shm-usage",
-                "--disable-infobars",
-                "--window-position=0,0",
-                "--ignore-certificate-errors",
-            ]
-        )
-
-        # Tạo context với Fingerprint giả lập người dùng 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},
-            device_scale_factor=1,
-        )
-
-        page = await context.new_page()
-
-        # Áp dụng Stealth để xóa sạch dấu vết automation
-        await stealth_async(page)
+        if not self.playwright:
+            await self.start()
 
+        browser = None
         try:
-            # Truy cập trang chủ trước để lấy cookie nền (mẹo nhỏ)
-            await page.goto("https://aternos.org/", wait_until="networkidle", timeout=30000)
-            await asyncio.sleep(2) # Chờ CF xử lý nhẹ
+            # 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",
+                ]
+            )
+
+            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)
 
-            # Mới bắt đầu vào trang server
+            # Thử truy cập
             resp = await page.goto(
                 "https://aternos.org/server/",
-                wait_until="networkidle",
-                timeout=40000
+                wait_until="networkidle", # Chờ cho network ổn định
+                timeout=30000
             )
 
-            # Chờ thêm 5 giây nếu thấy vòng xoay của Cloudflare
-            if "just a moment" in (await page.title()).lower():
-                print("Detected Cloudflare Challenge... waiting 10s")
-                await asyncio.sleep(10)
+            # Đợi thêm một chút để Cloudflare challenge (nếu có) tự giải quyết
+            await asyncio.sleep(5)
 
             title = await page.title()
             content = await page.content()
             
-            # Kiểm tra xem có bị lỗi Internal từ phía Proxy không
-            status = resp.status if resp else 500
-            
-            blocked = "just a moment" in title.lower() or "cloudflare" in content.lower()
+            is_blocked = "just a moment" in title.lower() or "cloudflare" in content.lower()
 
             return {
-                "ok": not blocked and status == 200,
-                "status": status,
+                "ok": not is_blocked,
+                "status": resp.status if resp else None,
                 "title": title,
-                "blocked": blocked
+                "blocked": is_blocked
             }
 
         except Exception as e:
             return {"ok": False, "error": str(e), "blocked": True}
         finally:
-            await browser.close()
+            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"}