Create app/panel_login.py

app/panel_login.py

@@ -0,0 +1,78 @@
+import asyncio
+import os
+import json
+from playwright.async_api import async_playwright
+from playwright_stealth import stealth_async
+
+class PanelLogin:
+    def __init__(self, username, password, server_id):
+        self.username = username
+        self.password = password
+        self.server_id = server_id
+        self.login_url = "https://cp.pikamc.vn/auth/login"
+        self.api_ws_url = f"https://cp.pikamc.vn/api/client/servers/{server_id}/websocket"
+        self.debug_path = "debug_screenshot.png"
+
+    async def fetch_ws_credentials(self):
+        async with async_playwright() as p:
+            # Khởi chạy trình duyệt (phù hợp với Docker Jammy)
+            browser = await p.chromium.launch(headless=True, args=["--no-sandbox"])
+            context = await browser.new_context(
+                viewport={'width': 1280, 'height': 720},
+                user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36"
+            )
+            
+            page = await context.new_page()
+            await stealth_async(page)
+
+            try:
+                # Bước 1: Đăng nhập
+                print(f"[*] Đang truy cập: {self.login_url}")
+                await page.goto(self.login_url, wait_until="networkidle")
+                
+                await page.fill('input[name="username"]', self.username)
+                await page.fill('input[name="password"]', self.password)
+                
+                # Click Đăng nhập
+                await page.click('button[type="submit"]')
+                
+                # Đợi chuyển hướng (Thường là về trang chủ / hoặc /server/...)
+                await page.wait_for_load_state("networkidle")
+                
+                # Chụp ảnh debug sau khi login
+                await page.screenshot(path=self.debug_path)
+                print(f"[+] Đã chụp ảnh debug tại: {self.debug_path}")
+
+                # Bước 2: Truy cập thẳng vào API lấy Websocket Token
+                # Playwright sẽ sử dụng Cookie đã lưu sau khi login thành công
+                print(f"[*] Đang lấy Token từ: {self.api_ws_url}")
+                await page.goto(self.api_ws_url)
+                
+                # Lấy nội dung JSON trả về từ trang API
+                content = await page.inner_text('pre') if await page.query_selector('pre') else await page.content()
+                
+                # Nếu nội dung nằm trong body (đối với một số trình duyệt)
+                if not content.startswith('{'):
+                    content = await page.locator('body').inner_text()
+
+                data = json.loads(content)
+                
+                await browser.close()
+                
+                if "data" in data:
+                    return {
+                        "ok": True,
+                        "token": data["data"]["token"],
+                        "socket": data["data"]["socket"]
+                    }
+                else:
+                    return {"ok": False, "error": "Không tìm thấy data trong JSON", "raw": content}
+
+            except Exception as e:
+                await page.screenshot(path="error_debug.png")
+                await browser.close()
+                return {"ok": False, "error": str(e)}
+
+# Cách sử dụng nhanh để test:
+# login_svc = PanelLogin("user", "pass", "45feca4d-3f79-4bcc-ae11-114335a1c77e")
+# result = asyncio.run(login_svc.fetch_ws_credentials())