# gscode_logic.py
import aiohttp
from bs4 import BeautifulSoup
from typing import List, Dict

GENSHIN_API = (
    "https://genshin-impact.fandom.com/api.php"
    "?action=parse"
    "&page=Promotional_Code"
    "&prop=text"
    "&format=json"
)

class GenshinCodeLogic:
    def __init__(self):
        self.session: aiohttp.ClientSession | None = None

    async def start(self):
        if not self.session:
            self.session = aiohttp.ClientSession(
                headers={"User-Agent": "Mozilla/5.0 (GenshinCodeBot)"}
            )

    async def stop(self):
        if self.session:
            await self.session.close()
            self.session = None

    async def fetch_codes(self) -> Dict:
        async with self.session.get(
            GENSHIN_API,
            timeout=aiohttp.ClientTimeout(total=20)
        ) as resp:
            if resp.status != 200:
                return {"count": 0, "cards": []}

            data = await resp.json()

        html = data.get("parse", {}).get("text", {}).get("*")
        if not html:
            return {"count": 0, "cards": []}

        soup = BeautifulSoup(html, "html.parser")

        cards: List[Dict] = []

        for row in soup.select("table.wikitable tbody tr"):
            tds = row.find_all("td")
            if len(tds) < 3:
                continue

            code_els = tds[0].find_all("code")
            if not code_els:
                continue

            codes = [c.get_text(strip=True) for c in code_els]

            server = tds[1].get_text(" ", strip=True)

            rewards = []
            for item in tds[2].select(".item-text"):
                text = item.get_text(" ", strip=True)
                if text:
                    rewards.append(text)

            validity = (
                tds[3].get_text(" ", strip=True)
                if len(tds) >= 4 else None
            )

            cards.append({
                "codes": codes,
                "server": server,
                "rewards": rewards,
                "validity": validity
            })

        return {
            "count": sum(len(card["codes"]) for card in cards),
            "cards": cards
        }