"""Wardrobe slot extraction and validation."""

from __future__ import annotations

from .schema import OutfitGraph, WardrobeSlot

SLOT_BLUEPRINTS: list[tuple[str, str, str, str, str]] = [
    ("hair_headwear", "sleek hair or sculptural headwear", "hair / metal", "obsidian with neon edge", "identity lock"),
    ("upper_body", "structured bodice or fitted top", "chantilly_lace", "black lace and mesh", "lace / mesh adapter"),
    ("outerwear", "long technical coat or jacket", "patent_leather", "jet black", "FLUX.2 material texture"),
    ("hands", "gloves with interface details", "patent_leather", "black with crimson nodes", "detail LoRA"),
    ("lower_body", "tailored lower silhouette", "layered_garments", "dark graphite", "garment consistency"),
    ("footwear", "strong platform boots", "polished_leather", "matte black sole", "boot detail"),
    ("jewelry", "choker, buckles, and hard points", "crimson_hardware", "crimson metal", "hardware glow"),
    ("props", "holographic sigils and data tools", "holographic_glass", "cyan and magenta", "NEXUS glyphs"),
    ("background_context", "rain, code, and location atmosphere", "neon_rain", "obsidian/cyan/crimson", "environmental continuity"),
]


def _material_token(value: object) -> str:
    return str(value).strip().lower().replace("-", "_").replace(" ", "_")


def build_outfit_graph(prompt: str, adult_mode: bool = False, controls: dict | None = None) -> OutfitGraph:
    """
    Constructs an outfit graph with wardrobe slots configured by prompt analysis and optional controls.
    
    Parameters:
    	prompt (str): Text to analyze for slot configuration and locking keywords.
    	adult_mode (bool): If True, marks upper_body and lower_body slots as adult-only. Defaults to False.
    	controls (dict | None): Optional configuration dict with keys: locked_slots, locate_focus, outerwear, upper_body, footwear, hardware, palette.
    
    Returns:
    	OutfitGraph: Graph containing configured wardrobe slots and a computed score.
    """
    lowered = prompt.lower()
    controls = controls or {}
    locked_slots = set(controls.get("locked_slots") or [])
    locate_focus = set(controls.get("locate_focus") or [])
    slots: list[WardrobeSlot] = []
    for index, (name, description, material, palette, lora_hint) in enumerate(SLOT_BLUEPRINTS):
        if name == "outerwear" and controls.get("outerwear"):
            description = str(controls["outerwear"])
        if name == "upper_body" and controls.get("upper_body"):
            material = str(controls["upper_body"])
        if name == "footwear" and controls.get("footwear"):
            description = str(controls["footwear"])
        if name == "jewelry" and controls.get("hardware"):
            material = _material_token(controls["hardware"])
        if controls.get("palette"):
            palette = str(controls["palette"])
        locked = any(token in lowered for token in name.split("_")) or material.replace("_", " ") in lowered
        if name == "outerwear" and "coat" in lowered:
            locked = True
        if name == "footwear" and ("boot" in lowered or "platform" in lowered):
            locked = True
        if name == "jewelry" and ("crimson" in lowered or "hardware" in lowered):
            locked = True
        if name in locked_slots:
            locked = True
        slots.append(
            WardrobeSlot(
                name=name,
                description=description,
                material=material,
                palette=palette,
                lora_hint=lora_hint,
                locked=locked,
                adult_only=False if not adult_mode else name in {"upper_body", "lower_body"},
                locate_region="manual-focus" if name in locate_focus else "auto-map",
                edit_priority=max(1, 5 - index // 2),
            )
        )
    locked_count = sum(1 for slot in slots if slot.locked)
    score = round(0.68 + min(0.24, locked_count * 0.035), 2)
    return OutfitGraph(slots=slots, score=score)