"""Parser for Mountain Country Storage (mountaincountrystorage.com)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility001972Parser(BaseParser):
    """Extract storage units from Mountain Country Storage.

    Unit data is in modal divs with id="rent-unit-type-id-XXXXX-modal".
    Each modal contains the unit title (with size) and monthly price.
    """

    platform = "custom_facility_001972"

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

        # Find all unit modals: <div id="rent-unit-type-id-XXXXX-modal">
        modals = [e for e in soup.find_all(True) if e.get("id", "").startswith("rent-unit-type")]

        for modal in modals:
            title_elem = modal.select_one(".modal-title")
            if not title_elem:
                continue
            title = title_elem.get_text(strip=True)

            # Title format: "Non-climate storage (5 x 10 x 8)" or "Covered parking (11 x 30 x 12.5)"
            size_m = re.search(r"\((\d+\s*x\s*\d+)", title, re.I)
            if not size_m:
                continue
            size_str = size_m.group(1).replace(" ", "")
            size = self.normalize_size(size_str)

            # Monthly price is in <strong>$XX.XX\n/ month</strong>
            monthly_elem = modal.select_one(".text-center.small strong")
            price = None
            if monthly_elem:
                price_m = re.search(r"\$[\d,.]+", monthly_elem.get_text())
                if price_m:
                    price = self.normalize_price(price_m.group())

            # Description from title prefix (before the parenthesis)
            desc = title.split("(")[0].strip() if "(" in title else title

            result.units.append(UnitResult(size=size, price=price, description=desc))

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