"""Parser for A-K Storage (New Richmond, WI)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility006457Parser(BaseParser):
    """Extract storage units from A-K Storage (Squarespace site)."""

    platform = "custom_facility_006457"

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

        # Units are listed as h2 elements with class 'list-item-content__title'
        containers = soup.select("div.list-item-content__text-wrapper")

        for container in containers:
            size_el = container.select_one("h2.list-item-content__title")
            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}

            # Extract description if available
            desc_el = container.select_one("div.list-item-content__description")
            if desc_el:
                unit.description = desc_el.get_text(strip=True)

            result.units.append(unit)

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

        return result