"""Parser for Prime Storage."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility109027Parser(BaseParser):
    """Extract storage locations from Prime Storage (multi-location directory).

    The homepage lists all Prime Storage locations with starting prices.
    Individual unit data requires visiting each location page.
    """

    platform = "custom_facility_109027"

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

        # Location cards have starting prices
        cards = soup.select("div.psc\\:card, div[class*='psc:card']")

        for card in cards:
            text = card.get_text(separator=" | ", strip=True)

            # Look for "Starting At $XX/MO" pattern
            price_match = re.search(r"Starting At \$(\d+)/MO", text, re.I)
            if not price_match:
                continue

            unit = UnitResult()
            unit.price = self.normalize_price(price_match.group(1))

            # Extract location name
            # Cards have format "Starting At $XX/MO | Location Name | Address"
            parts = text.split("|")
            if len(parts) >= 2:
                unit.description = parts[1].strip()

            if unit.price is not None:
                result.units.append(unit)

        if not result.units:
            result.warnings.append(
                "No unit data found - this is a multi-location directory page."
            )

        return result