"""Parser for KO Storage."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility095237Parser(BaseParser):
    """Extract storage units from KO Storage."""

    platform = "custom_facility_095237"

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

        containers = soup.select("div.col-12.col-lg-12.col-md-12.col-sm-6.unit-item")

        for container in containers:
            unit = UnitResult()
            text = container.get_text(separator=" ", strip=True)

            # Extract size
            size_el = container.select_one("span.row-unit-size-title.text-nowrap")
            size_text = size_el.get_text(strip=True) if size_el else None
            if not size_text:
                # Fallback: regex on full text
                m = re.search(r"(\d+\s*[\'\'\u2032]?\s*[xX\u00d7]\s*\d+\s*[\'\'\u2032]?)", text)
                if m:
                    size_text = m.group(1)
            if size_text:
                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}

            # Extract price
            price_el = container.select_one("span.non-web-rate.text-nowrap")
            price_text = price_el.get_text(strip=True) if price_el else None
            if not price_text:
                # Fallback: regex on full text
                pm = re.search(r"\$(\d[\d,.]*)", text)
                if pm:
                    price_text = pm.group(0)
            if price_text:
                pm = re.search(r"\$(\d[\d,.]*)", price_text)
                if pm:
                    unit.price = self.normalize_price(pm.group(1))

            unit.description = text[:200]

            if unit.size or unit.price:
                result.units.append(unit)

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

        return result