"""Parser for Premium Spaces storage facility 050371."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility050371Parser(BaseParser):
    """Extract storage units from Premium Spaces (se-vapor platform).

    Units are rendered in div.unit-item elements. Each contains:
    - div.unit-item__dimension: size text (e.g. "4.5' x 4.5'")
    - div.unit-item__category: category label
    - div.unit-item-price__web-rate: web rate price string (e.g. "$36.00")
    - div.unit-item__promotions span.unit-item-promotion: promo text
    """

    platform = "custom_facility_050371"

    _CLEAN_RE = re.compile(r"[^\d.]")

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

        seen: set[tuple[str, str]] = set()

        for card in soup.find_all(class_="unit-item"):
            # Skip nested sub-elements — only top-level unit-item containers
            if not card.get("data-unit-id"):
                continue

            # --- size ---
            dim_el = card.find(class_="unit-item__dimension")
            if not dim_el:
                continue
            size_text = dim_el.get_text(strip=True)
            # Remove span/state wrappers; keep just the text
            if not size_text:
                continue

            # --- price ---
            price_el = card.find(class_="unit-item-price__web-rate")
            if not price_el:
                continue
            # Remove the label span before extracting price text
            for label in price_el.find_all(class_="unit-item-price__label"):
                label.decompose()
            price_text = price_el.get_text(strip=True)
            price = self.normalize_price(price_text)
            if price is None:
                continue

            key = (size_text, price_text)
            if key in seen:
                continue
            seen.add(key)

            # --- promotion ---
            promo_el = card.find(class_="unit-item-promotion")
            promo = promo_el.get_text(strip=True) if promo_el else None

            # --- description (category + features) ---
            cat_el = card.find(class_="unit-item__category")
            desc = cat_el.get_text(strip=True) if cat_el else None

            unit = UnitResult(
                size=size_text,
                price=price,
                promotion=promo,
                description=desc,
            )
            w, ln, sq = self.normalize_size(size_text)
            if w is not None:
                unit.metadata = {"width": w, "length": ln, "sqft": sq}
            result.units.append(unit)

        if not result.units:
            result.warnings.append("No units found in Premium Spaces unit-item cards")

        return result