"""Parser for Self Storage Douglasville (GA)."""
from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility48Parser(BaseParser):
    """Extract storage units from selfstoragedouglasville.com.

    The rates table has columns: Unit size, Conventional, Climate Controlled,
    Choose Your Unit. Size format includes ceiling height: ``10' x 20' x 15'``.
    Prices may be ``n/a`` for unavailable types.
    """

    platform = "custom_facility_48"

    _SIZE_RE = re.compile(
        r"(\d+)['\u2019\u2032]?\s*x\s*(\d+)['\u2019\u2032]?",
        re.IGNORECASE,
    )

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

        # Find the rates table
        table = soup.select_one(".affordableArea table.table")
        if not table:
            # Fallback: find table with "Unit size" header
            for t in soup.find_all("table"):
                if t.find("th", string=re.compile(r"Unit\s*size", re.IGNORECASE)):
                    table = t
                    break

        if not table:
            result.warnings.append("No rates table found")
            return result

        rows = table.find_all("tr")
        for row in rows[1:]:  # Skip header row
            cells = row.find_all("td")
            if len(cells) < 3:
                continue

            size_text = cells[0].get_text(strip=True)
            conventional_text = cells[1].get_text(strip=True)
            climate_text = cells[2].get_text(strip=True)

            size_match = self._SIZE_RE.search(size_text)
            if not size_match:
                continue

            width = float(size_match.group(1))
            length = float(size_match.group(2))

            conventional_price = self.normalize_price(conventional_text)
            climate_price = self.normalize_price(climate_text)

            base_meta = {"width": width, "length": length, "sqft": width * length}

            # Create unit for conventional if available
            if conventional_price is not None:
                meta = {**base_meta, "driveUpAccess": True}
                unit = UnitResult(
                    size=f"{int(width)}' x {int(length)}'",
                    sale_price=conventional_price,
                    description=f"{size_text} Conventional ${conventional_price}",
                    metadata=meta,
                )
                result.units.append(unit)

            # Create unit for climate controlled if available
            if climate_price is not None:
                meta = {**base_meta, "climateControlled": True}
                unit = UnitResult(
                    size=f"{int(width)}' x {int(length)}'",
                    sale_price=climate_price,
                    description=f"{size_text} Climate Controlled ${climate_price}",
                    metadata=meta,
                )
                result.units.append(unit)

        if not result.units:
            result.warnings.append("Rates table found but no units extracted")

        return result