Update app/gs_daily.py

app/gs_daily.py

@@ -1,5 +1,4 @@
 # app/gs_daily.py
-
 import asyncio
 from datetime import datetime
 from pathlib import Path
@@ -51,10 +50,8 @@ class GSDailyApp:
     async def run_daily_and_capture(self, cookie_string: str):
         try:
             iphone = self.playwright.devices["iPhone 14 Pro"]
-
             context = await self.browser.new_context(**iphone)
 
-            # 🔥 Add cookie từ header string
             cookies = parse_cookie_string(cookie_string)
             await context.add_cookies(cookies)
 
@@ -64,19 +61,37 @@ class GSDailyApp:
             await page.wait_for_load_state("networkidle")
             await page.wait_for_timeout(3000)
 
-            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-            file_path = SCREENSHOT_DIR / f"gs_daily_{timestamp}.png"
+            active_item = page.locator("div[class*='actived']").first
+
+            if await active_item.count() == 0:
+                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ệ"
+                }
+
+            day_text = await active_item.locator("div[class*='item-day']").inner_text()
+
+            import re
+            match = re.search(r"\d+", day_text)
+            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 page.screenshot(
-                path=str(file_path),
-                full_page=True
-            )
+            await active_item.click()
+            await page.wait_for_timeout(2000)
 
             await context.close()
 
             return {
                 "ok": True,
-                "file": str(file_path)
+                "claimed": True,
+                "day": day_number,
+                "reward_count": count_text,
+                    "icon": icon_url,
+            "message": f"Đã nhận ngày thứ {day_number}"
             }
 
         except Exception as e: