Update app/aternos_api.py

app/aternos_api.py

@@ -20,14 +20,16 @@ class AternosAPI:
             await self.playwright.stop()
             self.playwright = None
 
-        async def check_access(self, action_type="login"):
-        if not self.playwright: await self.start()
+    async def run_action(self, user, pwd, action_type="login"):
+        if not self.playwright:
+            await self.start()
 
         browser = await self.playwright.chromium.launch(
             headless=False, 
             proxy=self.proxy_config,
             args=["--no-sandbox", "--disable-blink-features=AutomationControlled"]
         )
+        
         context = await browser.new_context(
             viewport={'width': 1366, 'height': 768},
             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"
@@ -35,39 +37,43 @@ class AternosAPI:
         page = await context.new_page()
 
         try:
-            # 1. Vượt Cloudflare (Giữ nguyên vì bạn đã vượt qua được bước này rồi)
-            await page.goto("https://aternos.org/go/", wait_until="domcontentloaded", timeout=60000)
-            for i in range(10):
-                if "just a moment" not in (await page.title()).lower(): break
+            # 1. Vượt Cloudflare
+            print("[INFO] Truy cập Aternos...")
+            await page.goto("https://aternos.org/go", wait_until="domcontentloaded", timeout=60000)
+            
+            for i in range(15):
+                title = await page.title()
+                if "just a moment" not in title.lower():
+                    break
                 await page.mouse.move(random.randint(100, 600), random.randint(100, 600))
                 await asyncio.sleep(2)
 
-            # 2. ĐIỀN FORM - Cần chắc chắn nó hiển thị
-            print("[INFO] Chờ form đăng nhập...")
+            # 2. Nhập liệu
+            print("[INFO] Đang điền form...")
             user_input = await page.wait_for_selector("input.username", state="visible", timeout=20000)
-            
-            # Xóa sạch nếu có chữ cũ và điền mới
-            await user_input.click(click_count=3) 
+            await user_input.click(click_count=3)
             await page.keyboard.press("Backspace")
-            await page.type("input.username", "Cleskry", delay=100)
+            await page.type("input.username", user, delay=100)
             
-            await page.type("input.password", "Saik_yuaik1", delay=100)
+            await page.type("input.password", pwd, delay=100)
             
-            # 3. BẤM NÚT - Chỗ này quan trọng
-            print("[INFO] Đang thực hiện click nút Đăng nhập...")
-            login_btn = await page.wait_for_selector("button.login-button", state="visible")
+            # 3. Click Đăng nhập & Đợi chuyển hướng
+            print("[INFO] Click Login...")
+            login_btn = await page.wait_for_selector("button.login-button")
             
-            # Chiêu cuối: Force Click và đợi URL đổi
-            await asyncio.gather(
-                page.wait_for_url("**/server**", timeout=20000), # Đợi URL đổi sang /server hoặc /servers
-                login_btn.click(force=True) # Ép click kể cả khi có gì đó che
-            )
+            # Kết hợp vừa bấm vừa đợi URL thay đổi
+            try:
+                await asyncio.gather(
+                    page.wait_for_url("**/server*", timeout=20000),
+                    login_btn.click(force=True)
+                )
+            except:
+                print("[WARN] Timeout hoặc kẹt chuyển hướng")
 
-            # 4. XỬ LÝ SAU LOGIN
-            await asyncio.sleep(2) # Đợi trang mới ổn định
+            await asyncio.sleep(3)
             final_url = page.url
             
-            # Nếu lọt vào danh sách nhiều server
+            # Xử lý nếu có danh sách server
             if "/servers/" in final_url:
                 await page.click(".server-body")
                 await page.wait_for_url("**/server/**", timeout=10000)
@@ -75,29 +81,31 @@ class AternosAPI:
 
             is_success = "/server" in final_url and "go" not in final_url
 
-            # 5. NẾU TYPE LÀ START - THÌ BẤM TIẾP
-            msg = "Đã vào được trang quản trị!"
-            if is_success and action_type == "start":
-                start_btn = await page.query_selector("#start")
-                if start_btn:
-                    await start_btn.click()
-                    # Chờ xác nhận EULA (nếu có)
-                    try:
-                        confirm = await page.wait_for_selector(".btn-success", timeout=5000)
-                        await confirm.click()
-                        msg = "Login & Server Started!"
-                    except:
-                        msg = "Login & Start Clicked!"
+            # 4. Thực hiện Action Start/Stop
+            msg = "Login thành công" if is_success else "Thất bại"
+            if is_success and action_type in ["start", "stop"]:
+                target_btn = f"#{action_type}"
+                btn = await page.query_selector(target_btn)
+                if btn and await btn.is_visible():
+                    await btn.click()
+                    if action_type == "start":
+                        try:
+                            confirm = await page.wait_for_selector(".btn-success", timeout=5000)
+                            await confirm.click()
+                            msg = "Server đang khởi động (EULA accepted)"
+                        except:
+                            msg = "Server đang khởi động"
+                    else:
+                        msg = "Server đã bấm dừng"
 
             return {
                 "ok": is_success,
+                "msg": msg,
                 "url": final_url,
-                "title": await page.title(),
-                "msg": msg if is_success else "Vẫn kẹt ở trang login (Sai pass hoặc lỗi click)"
+                "title": await page.title()
             }
 
         except Exception as e:
-            return {"ok": False, "error": str(e), "url": page.url}
+            return {"ok": False, "error": str(e)}
         finally:
-            await browser.close()
-
+            await browser.close()