Update app/gs_daily.py

app/gs_daily.py

@@ -47,19 +47,38 @@ class GSDailyApp:
         if self.playwright:
             await self.playwright.stop()
 
-    async def run_daily_and_capture(self, cookie_string: str):
+    async def run_daily_and_capture(self, cookie_string: str, discord_id: str):
         try:
-            iphone = self.playwright.devices["iPhone 14 Pro"]
-            context = await self.browser.new_context(**iphone)
+            if discord_id not in self.sessions:
+                iphone = self.playwright.devices["iPhone 14 Pro"]
 
+                context = await self.browser.new_context(**iphone)
+                page = await context.new_page()
+
+                self.sessions[discord_id] = {
+                    "context": context,
+                    "page": page
+                }
+
+            session = self.sessions[discord_id]
+            page = session["page"]
+            context = session["context"]
+
+            await context.clear_cookies()
             cookies = parse_cookie_string(cookie_string)
             await context.add_cookies(cookies)
 
-            page = await context.new_page()
-
-            await page.goto(CHECKIN_URL, timeout=60000, wait_until="domcontentloaded")
+            await page.goto(CHECKIN_URL, wait_until="domcontentloaded")
             await page.wait_for_load_state("networkidle")
-            await page.wait_for_timeout(3000)
+            await page.wait_for_timeout(2000)
+
+            login_button = page.locator("span:has-text('Đăng Nhập')")
+
+            if await login_button.count() > 0:
+                return {
+                    "ok": False,
+                    "error": "Cookie không hợp lệ hoặc đã hết hạn"
+                }
 
             active_item = page.locator("div[class*='actived']").first
 
@@ -67,7 +86,7 @@ class GSDailyApp:
                 return {
                     "ok": True,
                     "claimed": False,
-                    "message": "Hôm nay đã điểm danh rồi hoặc không tìm thấy ngày hợp lệ"
+                    "message": "Hôm nay đã điểm danh rồi"
                 }
 
             day_text = await active_item.locator("div[class*='item-day']").inner_text()
@@ -77,21 +96,19 @@ class GSDailyApp:
             day_number = int(match.group()) if match else None
 
             count_text = await active_item.locator("div[class*='item-cnt']").inner_text()
-
             icon_url = await active_item.locator("img").first.get_attribute("src")
 
             await active_item.click()
             await page.wait_for_timeout(2000)
 
-            await context.close()
-
             return {
                 "ok": True,
                 "claimed": True,
                 "day": day_number,
                 "reward_count": count_text,
-                    "icon": icon_url,
-            "message": f"Đã nhận ngày thứ {day_number}"
+                "icon": icon_url,
+                "discord_id": discord_id,
+                "message": f"Đã nhận ngày thứ {day_number}"
             }
 
         except Exception as e: