"""Parser for 54th South Storage."""

from __future__ import annotations

from bs4 import BeautifulSoup

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


class Facility5Parser(BaseParser):
    """Extract storage units from 54th South Storage price table.

    The site uses an old-school HTML table with td-based headers
    (Monthly Rent, Unit Size, Size Equivalent, What the Unit Will Hold).
    """

    platform = "custom_facility_5"

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

        # Find the pricing table by its border attribute and centered alignment
        price_table = soup.find("table", attrs={"border": "1"})
        if not price_table:
            result.warnings.append("No pricing table found")
            return result

        rows = price_table.find_all("tr")
        if len(rows) < 2:
            result.warnings.append("Pricing table has no data rows")
            return result

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

            texts = [c.get_text(strip=True) for c in cells]
            price_text = texts[0]
            size_text = texts[1]

            unit = UnitResult()
            unit.description = " | ".join(texts)
            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}
            unit.price = self.normalize_price(price_text)

            if (unit.metadata and "width" in unit.metadata) or unit.price:
                result.units.append(unit)

        if not result.units:
            result.warnings.append("No units extracted from pricing table")

        return result