"""Parser for Plum Storage Center (StoragePug platform)."""

from __future__ import annotations

import re

from bs4 import BeautifulSoup

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


class Facility000943Parser(BaseParser):
    """Extract storage units from Plum Storage Center.

    Site uses StoragePug (Vue.js SPA). Units are rendered as
    ``div.unit-card`` elements inside a ``div.cobra-wrapper`` container.
    """

    platform = "custom_facility_000943"

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

        cards = soup.find_all("div", class_="unit-card")
        if not cards:
            result.warnings.append("No unit-card elements found")
            return result

        for card in cards:
            unit = UnitResult()

            # Size — e.g. "5' x 10'"
            size_el = card.find("span", class_="dimensions")
            if size_el:
                raw_size = size_el.get_text(strip=True)
                unit.size = raw_size
                w, ln, sq = self.normalize_size(raw_size)
                if w is not None:
                    unit.metadata = {"width": w, "length": ln, "sqft": sq}

            # Price — prefer the discounted (current) price in .price-normal,
            # fall back to any price text in .unit-price
            price_val = None
            price_normal = card.find("div", class_="price-normal")
            if price_normal:
                price_val = self.normalize_price(price_normal.get_text(strip=True))

            if price_val is None:
                price_span = card.find("span", class_="unit-price")
                if price_span:
                    # May contain both discount and normal; grab first $ amount
                    m = re.search(r"\$[\d,.]+", price_span.get_text(strip=True))
                    if m:
                        price_val = self.normalize_price(m.group())

            unit.price = price_val

            # Original / strikethrough price
            price_discount = card.find("div", class_="price-discount")
            if price_discount:
                original = self.normalize_price(price_discount.get_text(strip=True))
                if original and original != price_val:
                    unit.metadata = unit.metadata or {}
                    unit.metadata["original_price"] = original

            # Unit type — e.g. "Drive-up", "Climate Control, Indoor Access"
            type_el = card.find("div", class_="unit-type-name")
            if type_el:
                unit_type = type_el.get_text(strip=True)
                unit.metadata = unit.metadata or {}
                unit.metadata["type"] = unit_type

            # Availability badge — e.g. "Last Unit", "2 Units Left"
            badge = card.find("span", class_=lambda c: c and "card-urgency-badge" in c and "badge-location" not in c)
            if badge:
                avail_text = badge.get_text(strip=True)
                if avail_text and avail_text not in ("", "Pittsburgh, PA"):
                    unit.metadata = unit.metadata or {}
                    unit.metadata["availability"] = avail_text

            # Build a human-readable description
            parts = []
            if unit.size:
                parts.append(unit.size)
            if unit.metadata and unit.metadata.get("type"):
                parts.append(unit.metadata["type"])
            unit.description = " | ".join(parts) if parts else None

            # Only keep units with at least a size or price
            if unit.size or unit.price:
                result.units.append(unit)

        if not result.units:
            result.warnings.append("unit-card elements found but no units extracted")

        return result