"""Parser for U-STOR Self Storage Pinellas Park (WordPress/Avada with DRTS fields)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility026153Parser(BaseParser):
    """Extract storage units from U-STOR Self Storage Pinellas Park.

    This site uses WordPress with Avada theme and DRTS (Directory) plugin.
    Unit sizes are in structured div elements with class patterns like
    'field_storage_size'. Prices are not listed per-unit (call for pricing).
    """

    platform = "custom_facility_026153"

    _SIZE_CLASS_RE = re.compile(r"field_storage_size")

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

        # Extract sizes from DRTS display elements
        size_divs = soup.find_all(class_=self._SIZE_CLASS_RE)
        seen_sizes: set[str] = set()

        for div in size_divs:
            size_text = div.get_text(strip=True)
            if not size_text:
                continue

            w, ln, sqft = self.normalize_size(size_text)
            if w is None or w < 3 or ln < 3:
                continue

            size_key = f"{int(w)}x{int(ln)}"
            if size_key in seen_sizes:
                continue
            seen_sizes.add(size_key)

            # Try to find description from sibling elements
            description = ""
            parent = div.parent
            if parent:
                desc_el = parent.find(class_=re.compile(r"description"))
                if desc_el:
                    description = desc_el.get_text(strip=True)[:200]

            unit = UnitResult()
            unit.size = size_text.strip()
            unit.metadata = {"width": w, "length": ln, "sqft": sqft}
            if description:
                unit.description = description
            result.units.append(unit)

        if not result.units:
            result.warnings.append("No units found via DRTS size fields")

        return result