"""Parser for AAA Storage Manhattan, KS."""
from __future__ import annotations

from bs4 import BeautifulSoup

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


class Facility32Parser(BaseParser):
    """Extract storage units from aaastoragemanhattan.com.

    Units are in an HTML table with ``td`` headers (not ``th``) in the format:
    Unit Size | Monthly Rent | 6 Months Prepaid | 6 Months Savings |
    12 Months Prepaid | 12 Month Savings.
    """

    platform = "custom_facility_32"

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

        # Find the innermost pricing table containing "Unit Size"
        target_table = None
        for td in soup.find_all("td", string=lambda s: s and "Unit Size" in s):
            target_table = td.find_parent("table")
            break

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

        rows = target_table.find_all("tr")
        # Skip the title row and header row (first two rows)
        data_rows = rows[2:] if len(rows) > 2 else []

        for row in data_rows:
            cells = row.find_all("td")
            if len(cells) < 2:
                continue

            size_text = cells[0].get_text(strip=True)
            monthly_text = cells[1].get_text(strip=True)

            if not size_text or not monthly_text:
                continue

            unit = UnitResult()
            unit.description = " | ".join(c.get_text(strip=True) for c in cells)
            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(monthly_text)

            result.units.append(unit)

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

        return result