"""Parser for All Valley Storage."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility038550Parser(BaseParser):
    """Extract storage units from All Valley Storage."""

    platform = "custom_facility_038550"

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

        # Sizes are grouped in a list: Small (5x5, 5x10), Medium (10x10, 10x15), Large (10x20, 10x30)
        unit_list = soup.find("ul", class_=re.compile(r"defaultList"))
        if unit_list:
            for li in unit_list.find_all("li"):
                text = li.get_text(strip=True)
                # Extract size category
                category_match = re.match(r"^(\w+\s*units?)", text, re.I)
                category = category_match.group(1) if category_match else None
                # Extract all dimensions from the list item
                dims = re.findall(r"\b(\d+x\d+)\b", text, re.I)
                for dim in dims:
                    unit = UnitResult()
                    unit.size = dim
                    if category:
                        unit.description = category
                    w, ln, sq = self.normalize_size(dim)
                    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 on page")

        return result