"""Parser for KJ Storage (storewithkjstorage.com)."""

from __future__ import annotations

import json

from bs4 import BeautifulSoup

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


class Facility003661Parser(BaseParser):
    """Extract storage units from KJ Storage via embedded JSON in <pre> tags."""

    platform = "custom_facility_003661"

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

        # Unit data is embedded as JSON inside <pre> tags within tables.
        # Each <pre> contains a JSON object with dcWidth, dcLength, dcPreferredRate, sTypeName.
        # The first <pre> is a summary record (fields are dicts with min/max); skip those.
        for pre in soup.find_all("pre"):
            text = pre.get_text(strip=True)
            if not text.startswith("{"):
                continue
            try:
                data = json.loads(text)
            except (json.JSONDecodeError, ValueError):
                continue

            width = data.get("dcWidth")
            length = data.get("dcLength")
            rate = data.get("dcPreferredRate")
            type_name = data.get("sTypeName", "")

            # Skip summary records where fields are dicts (not scalar values)
            if isinstance(width, dict) or isinstance(length, dict):
                continue
            if not (width and length):
                continue

            size_str = f"{int(width)}x{int(length)}"
            size = self.normalize_size(size_str)
            price = self.normalize_price(str(rate)) if rate is not None else None
            climate = data.get("bClimate", False)
            type_name_str = type_name if isinstance(type_name, str) else ""
            desc_parts = [type_name_str] if type_name_str else []
            if climate:
                desc_parts.append("Climate Controlled")
            if "parking" in type_name_str.lower() or "vehicle" in type_name_str.lower():
                desc_parts.append("Parking")

            vacant = data.get("iTotalVacant", 1)
            scarcity = None if (vacant and int(vacant) > 0) else "sold out"

            result.units.append(
                UnitResult(
                    size=size,
                    price=price,
                    description=" | ".join(desc_parts) if desc_parts else None,
                    scarcity=scarcity,
                )
            )

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