"""Parser for First Security Self Storage."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility032773Parser(BaseParser):
    """Extract storage units from First Security Self Storage."""

    platform = "custom_facility_032773"

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

        # Sizes are grouped in column divs under a "text-center" row
        row = soup.find("div", class_=re.compile(r"row.*text-center|text-center.*row"))
        if row:
            col_divs = row.find_all("div", class_=re.compile(r"col-"))
            for col in col_divs:
                # Each column has sizes separated by <br>
                text = col.get_text(separator="\n", strip=True)
                for line in text.split("\n"):
                    line = line.strip()
                    if re.match(r"^\d+[xX]\d+$", line):
                        unit = UnitResult()
                        unit.size = line
                        w, ln, sq = self.normalize_size(line)
                        if w is not None:
                            unit.metadata = {"width": w, "length": ln, "sqft": sq}
                        result.units.append(unit)

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

        return result