"""Parser for Gypsum Mini Storage (StorEdge / Storable platform)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup, Tag

from src.parsers.base import BaseParser, ParseResult, UnitResult


class Facility004216Parser(BaseParser):
    """Extract storage units from Gypsum Mini Storage.

    Site uses the Storable/StorEdge rental center with Material UI card
    components.  Each unit is rendered inside a ``MuiCard-root`` div with:

    * An ``<h5>`` (or similar heading) containing the size, e.g. "5' x 5' Unit"
    * A ``MuiTypography-h6`` element with the current/discounted monthly price
    * An optional ``MuiTypography-caption`` span with the original
      (struck-through) price
    * A ``MuiTypography-body1`` paragraph with the unit category
    """

    platform = "custom_facility_004216"

    _SIZE_RE = re.compile(
        r"(\d+)[′'']?\s*[xX×]\s*(\d+)[′'']?"
    )

    def parse(self, html: str, url: str = "") -> ParseResult:
        soup = BeautifulSoup(html, "lxml")
        result = ParseResult(platform=self.platform, parser_name=self.__class__.__name__)

        cards = soup.find_all("div", class_="MuiCard-root")
        seen: set[str] = set()

        for card in cards:
            if not isinstance(card, Tag):
                continue

            # --- size from heading ---
            heading = card.find(["h3", "h4", "h5", "h6"])
            if heading is None:
                continue
            heading_text = heading.get_text(strip=True)
            m = self._SIZE_RE.search(heading_text)
            if not m:
                continue

            size_raw = f"{m.group(1)}' x {m.group(2)}'"
            if size_raw in seen:
                continue
            seen.add(size_raw)

            # --- prices ---
            # Discounted / current price lives in an h6.MuiTypography-h6
            price_el = card.find(
                "h6",
                class_=lambda c: c and "MuiTypography-h6" in c,
            )
            # Fallback: subtitle1 (contains "$X /mo")
            if price_el is None:
                price_el = card.find(
                    "h6",
                    class_=lambda c: c and "MuiTypography-subtitle1" in c,
                )
            price = self.normalize_price(price_el.get_text(strip=True)) if price_el else None

            # Original / strike-through price (caption)
            strike_el = card.find(
                "span",
                class_=lambda c: c and "MuiTypography-caption" in c,
            )
            original_price = (
                self.normalize_price(strike_el.get_text(strip=True))
                if strike_el
                else None
            )

            # --- category ---
            cat_el = card.find(
                "p",
                class_=lambda c: c and "MuiTypography-body1" in c,
            )
            category = cat_el.get_text(strip=True) if cat_el else None

            # --- build unit ---
            unit = UnitResult()
            unit.size = size_raw
            w, ln, sq = self.normalize_size(size_raw)
            metadata: dict[str, object] = {}
            if w is not None:
                metadata.update(width=w, length=ln, sqft=sq)
            if original_price is not None:
                metadata["original_price"] = original_price
            if category:
                metadata["category"] = category
            if metadata:
                unit.metadata = metadata

            unit.price = price
            unit.description = heading_text
            result.units.append(unit)

        if not result.units:
            result.warnings.append("No MuiCard-root unit cards found")

        return result