"""Parser for Kenwood Mini Storage (Austin, MN)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility020571Parser(BaseParser):
    """Extract storage units from Kenwood Mini Storage."""

    platform = "custom_facility_020571"

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

        # Units are in section.box elements with h3 containing "N units\nWxH" format
        containers = soup.select("section.box")

        for container in containers:
            h3 = container.select_one("h3")
            if not h3:
                continue

            h3_text = h3.get_text(separator="\n", strip=True)

            # Extract size from h3 text (format: "20 units\n5' x 10'" or similar)
            size_match = re.search(r"(\d+'\s*x\s*\d+'|\d+\s*x\s*\d+)", h3_text, re.IGNORECASE)
            if not size_match:
                continue

            size_text = size_match.group(1).strip()

            # Extract unit count from h3 text
            count_match = re.search(r"(\d+)\s*units?", h3_text, re.IGNORECASE)
            count = int(count_match.group(1)) if count_match else None

            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}
                if count:
                    unit.metadata["count"] = count

            result.units.append(unit)

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

        return result