# -----------------------------------------------------------------------------
# Lou-Ka — Agrégateur de logements à louer (province de Québec)
# Auteur : Simon-Pierre Boucher — contact@spboucher.ai
# connectors/solis_levis.py : Solis Lévis (solislevis.com) — RPA 55+ autonomes
#   Immeuble unique de 7 étages sur la rue Weyman à Lévis (Studio, 3½, 4½,
#   dès ~1 231 $/mois, plusieurs inclusions : eau chaude, électricité,
#   chauffage, climatisation). WordPress rendu serveur : la page
#   /appartements-retraites-autonomes/ embarque le sélecteur d'unités complet
#   dans un <script> « units = [[…]] » — un tableau par étage, chaque unité
#   avec numéro, état (Available/Reserved/Sold), typologie (0=Studio, 1=3½,
#   2=4½), type de plan, prix mensuel, superficies (logement + balcon) et
#   image du plan. Une annonce Lou-Ka par unité DISPONIBLE, prix exact.
# -----------------------------------------------------------------------------
from __future__ import annotations

import json
import re

from bs4 import BeautifulSoup

from ..schema import Listing
from .base import BaseConnector

PAGE_URL = "https://solislevis.com/appartements-retraites-autonomes/"
ADDRESS = "Rue Weyman, Lévis"

# typologie du sélecteur (data-typology des filtres de la page)
_TYPOLOGY = {0: "Studio", 1: "3½", 2: "4½"}

_UNITS_RE = re.compile(r"units\s*=\s*(\[\[[\s\S]*?\]\])\s*;")
_MAP_SOLIS_RE = re.compile(
    r'\{"nom":"Solis","adresse":"[^"]*","latitude":"([\d.\-]+)",'
    r'"longitude":"([\d.\-]+)"')
_IMG_RE = re.compile(
    r"https://solislevis\.com/app/uploads/[^\"\s\\']+\.(?:jpe?g|webp|png)", re.I)


class SolisLevisConnector(BaseConnector):
    source_id = "solis_levis"
    request_delay = 0.6
    max_images = 12

    # -- inclusions de l'immeuble (listes sous « Tous nos appartements
    #    offrent » et « Le prix mensuel inclut ») --------------------------------
    @staticmethod
    def _inclusions(soup: BeautifulSoup) -> list[str]:
        feats: list[str] = []
        # « Tous nos appartements offrent : » — liste <li> qui suit le titre
        el = soup.find(string=re.compile("appartements offrent", re.I))
        if el is not None:
            ul = el.parent.find_next("ul")
            if ul is not None:
                for li in ul.find_all("li")[:8]:
                    t = re.sub(r"\s+", " ", li.get_text(" ", strip=True)).strip()
                    if 3 <= len(t) <= 80 and t not in feats:
                        feats.append(t)
        # « Le prix mensuel inclut : » — grille d'icônes (p.icon-descrpition,
        # coquille du thème incluse)
        for p in soup.select("p.icon-descrpition")[:8]:
            t = re.sub(r"\s+", " ", p.get_text(" ", strip=True)).strip()
            if 2 <= len(t) <= 80 and t not in feats:
                feats.append(t)
        return feats

    def fetch(self) -> list[Listing]:
        listings: list[Listing] = []
        try:
            html = self.get(PAGE_URL).text
        except Exception:
            return listings

        m = _UNITS_RE.search(html)
        if not m:
            return listings
        try:
            floors = json.loads(m.group(1))
        except ValueError:
            return listings

        soup = BeautifulSoup(html, "html.parser")

        # coordonnées : point « Solis » de la carte du quartier (repli : centre
        # du secteur Weyman/Lévis relevé sur la page)
        lat, lng = 46.79696, -71.15874
        gm = _MAP_SOLIS_RE.search(html)
        if gm:
            try:
                lat, lng = float(gm.group(1)), float(gm.group(2))
            except ValueError:
                pass

        # description : intro de la page + promotion en cours le cas échéant
        paras = []
        for p in soup.find_all("p"):
            t = re.sub(r"\s+", " ", p.get_text(" ", strip=True)).strip()
            if len(t) >= 100:
                paras.append(t)
                break
        promo = ""
        pm = re.search(r"Promotion en cours\s*:\s*([^<]{5,120})", html)
        if pm:
            promo = "Promotion en cours : " + pm.group(1).strip()
        desc = "\n\n".join(x for x in (paras[0] if paras else "", promo) if x)

        inclusions = self._inclusions(soup)

        # photos de l'immeuble (héro/galerie de la page ; les plans d'unités
        # viennent en plus, annonce par annonce)
        building_imgs = [u for u in dict.fromkeys(_IMG_RE.findall(html))
                         if not re.search(r"logo|icon|favicon|plan|spinner|"
                                          r"\d{3}(?:@\d{3})?\.png", u, re.I)]

        for floor in floors:
            for unit in floor:
                if (unit.get("state") or "") != "Available":
                    continue
                number = unit.get("number")
                price = unit.get("price")
                if number is None or not price:
                    continue
                typology = _TYPOLOGY.get(unit.get("typology"), "")
                if not typology:
                    continue
                level = int(number) // 100
                floor_label = "RDC" if level == 1 else f"{level}e étage"
                details = {
                    "Résidence pour aînés": "oui",
                    "Étage": floor_label,
                    "Type de plan": unit.get("type") or "",
                }
                area = unit.get("indoor_area")
                balcony = unit.get("balcony_area")
                if balcony:
                    details["Balcon"] = f"{balcony} pi²"
                images: list[str] = []
                if unit.get("plan_img_src"):
                    images.append(unit["plan_img_src"])
                images += building_imgs
                listings.append(Listing(
                    source=self.source_id,
                    external_id=f"solis-{number}",
                    url=PAGE_URL,
                    title=f"{typology} — Solis Lévis, unité {number}",
                    address=ADDRESS,
                    city="Lévis",
                    unit_type=typology,
                    price=float(price),
                    availability="Disponible",
                    area_sqft=float(area) if area else None,
                    description=desc,
                    amenities=list(inclusions),
                    details=details,
                    images=images[: self.max_images],
                    lat=lat,
                    lng=lng,
                ))
        return listings