zulaAPI.py

# Drake Hooks + WaterTrooper
# Casino Claim 2
# Zula API (SeleniumBase UC; 0/1/2 lobby popups; no timer scraping)
# Exposes: async def zula_uc(ctx, channel)

import os
import tempfile
import discord
from dotenv import load_dotenv
from seleniumbase import SB

load_dotenv()

# Expect "email:password" in ZULA
ZULA_CRED = os.getenv("ZULA", "")

# ───────────────────────────────────────────────────────────
# Helpers (FC/Sportzino-style)
# ───────────────────────────────────────────────────────────
async def _send_post_claim(sb: SB, channel: discord.abc.Messageable, path: str, caption: str):
    """Only used on successful claim to avoid screenshot spam."""
    try:
        sb.save_screenshot(path)
        await channel.send(caption, file=discord.File(path))
    finally:
        try:
            if os.path.exists(path):
                os.remove(path)
        except Exception:
            pass

async def _send_status_shot(sb: SB, channel: discord.abc.Messageable, caption: str, prefix: str):
    """
    One-off screenshot for 'unavailable' or 'error' states.
    Creates a temp file, attaches it, and cleans up.
    """
    fd, tmp_path = tempfile.mkstemp(prefix=f"{prefix}_", suffix=".png", dir="/tmp")
    os.close(fd)
    try:
        sb.save_screenshot(tmp_path)
        await channel.send(caption, file=discord.File(tmp_path))
    except Exception:
        # Fallback to text-only if screenshot fails
        try:
            await channel.send(caption)
        except Exception:
            pass
    finally:
        try:
            if os.path.exists(tmp_path):
                os.remove(tmp_path)
        except Exception:
            pass

def _force_click_xpath(sb: SB, xpath: str, timeout: float = 12) -> bool:
    """Robust click chain for stubborn elements."""
    try:
        sb.wait_for_element_visible(xpath, timeout=timeout)
    except Exception:
        return False
    try:
        sb.scroll_to(xpath)
    except Exception:
        pass
    for mode in ("click", "slow", "js", "directjs"):
        try:
            if mode == "click":
                sb.click_xpath(xpath, timeout=2)
            elif mode == "slow":
                sb.slow_click(xpath)
            elif mode == "js":
                sb.js_click(xpath)
            else:
                el = sb.find_element(xpath)
                sb.execute_script("arguments[0].click();", el)
            return True
        except Exception:
            continue
    return False

def _try_click_any(sb: SB, xpaths, timeout_each=10) -> bool:
    for xp in xpaths:
        if _force_click_xpath(sb, xp, timeout=timeout_each):
            return True
    return False

def _close_lobby_popups_flexible(sb: SB):
    """
    Close up to TWO popups that share the same close button XPath.
    Works if there are 0, 1, or 2 popups.
    Includes both /div[4]/... and /div[5]/... variants.
    """
    popup_close_xpaths = [
        "/html/body/div[4]/div/div[1]/div/div/button",
        "/html/body/div[5]/div/div[1]/div/div/button",  # newly added
    ]
    closed = 0
    # Try up to two times; if not present, we simply move on.
    for i in range(2):
        clicked_any = _try_click_any(sb, popup_close_xpaths, timeout_each=6)
        if clicked_any:
            closed += 1
            sb.wait(0.6)  # allow next modal to mount (if any)
        else:
            # If not clickable yet, give UI a moment and retry once per slot
            sb.wait(0.8)
            if _try_click_any(sb, popup_close_xpaths, timeout_each=3):
                closed += 1
                sb.wait(0.5)
    print(f"[Zula] Closed lobby popups: {closed} (0–2 expected)")

def _extra_popup_cleanup(sb: SB):
    """
    Extra cleanup for any additional popups that might obscure header buttons.
    """
    popup_xpaths = [
        "/html/body/div[4]/div/div[1]/div/div/button",  # primary close
        "/html/body/div[5]/div/div[1]/div/div/button",  # newly added variant
        "/html/body/div[4]/div/div/div[2]/button[4]",
        "/html/body/div[4]/div[4]/div/div/div/div[1]/div",
        "//button[contains(translate(., 'CLOSE', 'close'),'close')]",
        "//button[contains(.,'Close')]",
    ]
    for _ in range(2):
        if _try_click_any(sb, popup_xpaths, timeout_each=4):
            print("[Zula] Closed an extra popup.")
            sb.wait(0.3)
    try:
        sb.press_keys("body", "ESCAPE")
    except Exception:
        pass

# ───────────────────────────────────────────────────────────
# Main UC-based flow
# ───────────────────────────────────────────────────────────
async def zula_uc(ctx, channel: discord.abc.Messageable):
    """
    Zula via SeleniumBase (uc=True).
    - Handles 0/1/2 identical lobby popups gracefully.
    - Sends exactly one screenshot only on successful claim.
    - If no claim available (or auth fails), sends one line: 'Zula: countdown not available (or auth failed).'
    """
    await channel.send("Launching **Zula** (UC)…")

    if ":" not in ZULA_CRED:
        await channel.send("❌ Missing `ZULA` as 'email:password' in your .env.")
        return

    username, password = ZULA_CRED.split(":", 1)

    try:
        with SB(uc=True, headed=True) as sb:
            # 1) Login
            sb.uc_open_with_reconnect("https://www.zulacasino.com/login", 4)
            sb.wait_for_ready_state_complete()
            print("[Zula] Login page loaded.")

            try:
                sb.type("input[id='emailAddress']", username)
                sb.type("input[id='password']", password)
                try:
                    sb.wait(10)
                    sb.uc_gui_click_captcha()
                except Exception:
                    pass
            except Exception as e:
                print(f"[Zula][ERROR] Login fields not found: {e}")
                await _send_status_shot(sb, channel, "Zula: countdown not available (or auth failed).", "zula_unavailable")  # need new screenshot here
                return

            # Submit login
            submitted = False
            try:
                sb.press_keys("input[id='password']", "\n")
                submitted = True
            except Exception:
                pass
            if not submitted:
                submitted = _try_click_any(
                    sb,
                    ["//button[@type='submit']", "//button[contains(.,'Log in')]"],
                    timeout_each=10,
                )
            if not submitted:
                print("[Zula][ERROR] Could not submit login.")
                await _send_status_shot(sb, channel, "Zula: countdown not available (or auth failed).", "zula_unavailable")  # need new screenshot here
                return

            # 2) Post-login settle and refresh into lobby
            sb.wait(8)
            sb.refresh_page()
            sb.wait_for_ready_state_complete()
            print("[Zula] Post-login refresh complete (lobby expected).")

            # 3) Close 0/1/2 lobby popups (supports /div[4]/... and /div[5]/...)
            _close_lobby_popups_flexible(sb)

            # 4) Extra safety cleanup
            _extra_popup_cleanup(sb)

            # 5) Open Rewards / Free Coins
            opened_rewards = _try_click_any(
                sb,
                [
                    "/html/body/div[1]/div[2]/div[1]/div/nav/div[2]/button[1]",
                    "/html/body/div[1]/div[2]/div/nav/div[2]/button[1]",
                    "//button[contains(.,'Free Coins') or contains(.,'Rewards') or contains(.,'Get Coins')]",
                ],
                timeout_each=12,
            )
            if not opened_rewards:
                print("[Zula] Rewards/Free Coins button not found.")
                # need new screenshot here
                await _send_status_shot(sb, channel, "Zula: countdown not available (or auth failed).", "zula_unavailable")
                return

            sb.wait(10)  # allow the rewards modal to render fully
            print("[Zula] Rewards modal should be open (proceeding).")

            # 6) Click Collect / Claim inside Rewards modal
            collected = _try_click_any(
                sb,
                [
                    "/html/body/div[6]/div/div[1]/div/div/div[2]/div[1]/div[2]/div[1]/div/button",
                    "/html/body/div[4]/div/div[1]/div/div/div[2]/div[1]/div[2]/div[1]/div/button",
                    "//button[contains(.,'Collect') and not(@disabled)]",
                    "//button[.//span[contains(.,'Collect')] and not(@disabled)]",
                    "//button[contains(.,'Claim') and not(@disabled)]",
                ],
                timeout_each=12,
            )

            if collected:
                sb.wait(3)
                await _send_post_claim(sb, channel, "zula_claimed.png", "Zula Daily Bonus Claimed!")
                print("[Zula] Claimed successfully.")
            else:
                print("[Zula] No claim available (likely already claimed).")
                await _send_status_shot(sb, channel, "Zula: countdown not available (or auth failed).", "zula_unavailable")

    except Exception as e:
        print(f"[Zula][ERROR] Exception during automation: {e}")
        # need new screenshot here as well
        try:
            # Best effort to grab a screenshot even on exceptions outside 'with SB'
            with SB(uc=True, headed=True) as sb_fallback:
                await _send_status_shot(sb_fallback, channel, "Zula: bonus not available (or auth failed).", "zula_error")
        except Exception:
            # If even fallback SB fails, send text-only
            await channel.send("Zula: countdown not available (or auth failed).")