"""Parser for Green Valley Mini Storage LLC."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility059225Parser(BaseParser):
    """Extract storage units from Green Valley Mini Storage LLC (unittrac platform)."""

    platform = "custom_facility_059225"

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

        # Each unit is in an outer table with width="100%" containing an image,
        # a nested table with the size/sqft, and a badge div for availability.
        containers = soup.find_all("table", attrs={"width": "100%"})

        for container in containers:
            # Must contain a size span (bold, 15px font)
            size_el = container.select_one('span[style*="font-size: 15px"]')
            if not size_el:
                continue

            size_text = size_el.get_text(strip=True)
            if not re.search(r"\d", size_text):
                continue

            unit = UnitResult()
            unit.size = size_text
            w, ln, sq = self.normalize_size(size_text)
            if w is not None:
                unit.metadata = {"width": w, "length": ln, "sqft": sq}

            # Availability badge
            avail_el = container.find("div", class_=re.compile(r"badge"))
            if avail_el:
                avail_text = avail_el.get_text(strip=True)
                avail_match = re.search(r"(\d+)\s+Available", avail_text, re.I)
                if avail_match:
                    avail_count = int(avail_match.group(1))
                    if unit.metadata is None:
                        unit.metadata = {}
                    unit.metadata["available"] = avail_count
                    unit.scarcity = f"{avail_count} available"

            result.units.append(unit)

        if not result.units:
            result.warnings.append("No units found on page")

        return result