"""Parser for Storage Rentals of America (Cortland, OH)."""

from __future__ import annotations

import re
from typing import Optional

from bs4 import BeautifulSoup

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


class Facility112038Parser(BaseParser):
    """Extract storage units from SROA using data-testid unit containers."""

    platform = "custom_facility_112038"

    @staticmethod
    def _extract_price(text: str) -> Optional[float]:
        """Extract numeric price from text like '$71/mo*' or 'was $85/mo'."""
        m = re.search(r"\$(\d+(?:\.\d+)?)", text)
        if m:
            return float(m.group(1))
        return None

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

        # SROA uses data-testid="storageUnitsItem-container-unitPackageModal"
        containers = soup.find_all(
            attrs={"data-testid": "storageUnitsItem-container-unitPackageModal"}
        )

        for container in containers:
            unit = UnitResult()

            # Size element: "10 x 10| Medium" or "10 x 20| Large | Parking Space"
            size_el = container.find(
                attrs={"data-testid": "storageUnitsItem-text-unitSize"}
            )
            if size_el:
                size_text = size_el.get_text(strip=True)
                # Extract just the dimension part before the pipe
                size_part = size_text.split("|")[0].strip()
                unit.size = size_part
                w, ln, sq = self.normalize_size(size_part)
                if w is not None:
                    unit.metadata = {"width": w, "length": ln, "sqft": sq}

            # Try online price first (data-testid="storageUnitsItem-text-priceOnline")
            online_price_el = container.find(
                attrs={"data-testid": "storageUnitsItem-text-priceOnline"}
            )
            if online_price_el:
                unit.price = self._extract_price(online_price_el.get_text(strip=True))

            # Fall back to standard price if online not found
            if unit.price is None:
                standard_price_el = container.find(
                    attrs={"data-testid": "storageUnitsItem-text-priceStandard"}
                )
                if standard_price_el:
                    unit.price = self._extract_price(standard_price_el.get_text(strip=True))

            if unit.size or unit.price:
                result.units.append(unit)

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

        return result