from __future__ import annotations

import re
from typing import Any

try:
    from .krea_action_context import axis_values_text
    from .krea_action_positions import action_position_phrase, mentions_rear_entry
    from .krea_detail import detail_clauses, join_detail_clauses, limit_detail_for_density
except ImportError:  # Allows local smoke tests with `python -c`.
    from krea_action_context import axis_values_text
    from krea_action_positions import action_position_phrase, mentions_rear_entry
    from krea_detail import detail_clauses, join_detail_clauses, limit_detail_for_density


def _clean(value: Any) -> str:
    text = "" if value is None else str(value)
    text = text.replace("\n", " ")
    text = re.sub(r"\s+", " ", text).strip()
    text = re.sub(r"\s+([,.;:])", r"\1", text)
    return text


def normalize_climax_view_clause(clause: str, role_graph: str) -> str:
    lower = clause.lower()
    if "view" not in lower and "frame" not in lower:
        return clause
    angle_match = re.search(
        r"\b(front-facing|close-up|wide full-body|wide|overhead|mirror-reflected|low-angle|side-profile|bed-level)\b",
        lower,
    )
    if not angle_match:
        return clause
    angle = angle_match.group(1)
    if angle == "wide":
        angle = "wide full-body"
    position = action_position_phrase(role_graph)
    if position:
        return f"{angle} aftermath view with the {position} readable"
    return f"{angle} aftermath view"


def climax_clause_duplicates_role(clause: str, role_graph: str) -> bool:
    clause_lower = clause.lower()
    role_lower = role_graph.lower()
    role_has_ejaculation = any(token in role_lower for token in ("ejaculates semen", "visible semen", "semen lands"))
    if role_has_ejaculation and re.search(
        r"\b(?:cum clearly visible|explicit semen aftermath visible|hardcore ejaculation detail visible|"
        r"post-ejaculation fluids anatomically clear|sexual fluids and body contact visible|"
        r"visible external ejaculation|hardcore ejaculation scene|visible orgasm aftermath)\b",
        clause_lower,
    ):
        return True
    duplicate_pairs = (
        (("lower back", "ass"), ("lower back", "ass")),
        (("ass",), ("ass",)),
        (("pussy", "thigh"), ("pussy", "thigh")),
        (("face", "lips"), ("face", "lips")),
        (("tongue", "chin"), ("face", "lips", "mouth", "tongue")),
        (("breast",), ("breast", "chest")),
        (("belly",), ("belly", "torso")),
        (("body",), ("body",)),
    )
    if any(token in clause_lower for token in ("cum", "semen", "fluid")):
        for clause_tokens, role_tokens in duplicate_pairs:
            if any(token in clause_lower for token in clause_tokens) and any(token in role_lower for token in role_tokens):
                return True
    return False


def climax_role_graph(role_graph: str, hard_item: str, axis_values: Any = None) -> str:
    role_graph = _clean(role_graph).rstrip(".")
    text = " ".join(part.lower() for part in (role_graph, _clean(hard_item), axis_values_text(axis_values)) if part)
    if "the woman" not in text or "the man" not in text:
        return role_graph
    if "lying between two partners" in text or "lies between" in text:
        return "the woman lies between two partners, the man under her hips and another partner over her torso as visible semen lands on her body"
    if "held between front-and-back partners" in text:
        return "the woman is held between the man behind her and another partner in front of her as visible semen lands across her body"
    if "kneeling between standing partners" in text:
        return "the woman kneels between standing partners gathered around her face and torso for visible ejaculation"
    if "side-lying with thighs parted" in text:
        return "the woman lies on her side with thighs parted while the man kneels beside her hips and ejaculates semen across her thighs and pussy"
    if "sitting on the edge of the bed" in text:
        return "the woman sits on the edge of the bed with knees spread while the man stands close between her legs and ejaculates semen across her body"
    if "lying at the bed edge with thighs open" in text:
        return "the woman lies at the bed edge with thighs open while the man kneels between her legs and ejaculates semen across her pussy and thighs"
    if "reclining with thighs open" in text or "lying on the back with legs spread" in text:
        return (
            "the woman lies on her back with thighs open after ejaculation; thick semen and clear fluid cover her exposed pussy "
            "and inner thighs as the exact-center aftermath detail, her body stays still, and her face and torso remain visible behind the open thighs"
        )
    if "on all fours with hips raised" in text:
        return "the woman is on all fours with hips raised while the man is positioned behind her and ejaculates semen across her ass, thighs, and lower back"
    if "face-down ass-up" in text or "lies face-down" in text or "face down" in text:
        return "the woman lies face-down with ass raised while the man is positioned behind her and ejaculates semen across her lower back and ass"
    if "bent over with ass raised" in text or "bent over" in text:
        return "the woman bends forward with hips raised while the man stands behind her with visible semen across her lower back, ass, and thighs"
    if "kneeling with mouth open" in text:
        return "the woman kneels in front of the man at hip height as he ejaculates semen onto her face, lips, and chest"
    if "kneeling in front of a standing partner" in text:
        return "the woman kneels in front of the man at hip height while he stands over her and ejaculates semen across her body"
    if "standing with cum on the body" in text:
        return "the woman stands braced in front of the man while he stands close at hip level and ejaculates semen across her body"
    if "squatting on top of a partner" in text:
        return "the woman squats over the man's hips while the man lies on his back under her and ejaculates semen onto her body"
    if "reverse cowgirl over a partner's hips" in text:
        return "the woman straddles the man's hips facing away while the man lies on his back under her and ejaculates semen onto her body"
    if "straddles" in text or "straddling a partner" in text or "straddling a partner's hips" in text or "shared climax after penetration" in text:
        return "the woman straddles the man's hips while the man lies on his back under her and ejaculates semen onto her body"
    if "seated in a partner's lap facing them" in text:
        return "the woman sits in the man's lap facing him, legs wrapped around his hips as he ejaculates semen across her body"
    if "lower back" in text or "cum dripping from ass" in text or "cum on lower back" in text or mentions_rear_entry(text):
        return "the woman bends forward with hips raised while the man stands behind her with visible semen across her lower back, ass, and thighs"
    if "cum on face" in text or "cum on tongue" in text or "cum on lips" in text or "cum on tongue and chin" in text:
        return "the woman kneels in front of the man at hip height as he ejaculates semen onto her face, lips, and chest"
    if (
        "cum dripping from pussy" in text
        or "arousal dripping from pussy" in text
        or "open thighs" in text
    ):
        return (
            "the woman lies on her back with thighs open after ejaculation; thick semen and clear fluid cover her exposed pussy "
            "and inner thighs as the exact-center aftermath detail, her body stays still, and her face and torso remain visible behind the open thighs"
        )
    if role_graph:
        return role_graph
    return "the woman lies on her back with thighs open while the man kneels between her legs and ejaculates semen across her body"


def dedupe_climax_detail(detail: str, role_graph: str, density: str = "balanced") -> str:
    detail = _clean(detail)
    lower = role_graph.lower()
    patterns: list[str] = []
    if "solo visible ejaculation" in lower or "one hand on his penis" in lower:
        detail = re.sub(r"\bcum on lower back and ass\b", "visible semen on skin", detail, flags=re.IGNORECASE)
        detail = re.sub(r"\bcum (?:on|dripping from) ass\b", "visible semen on skin", detail, flags=re.IGNORECASE)
    if "lies on her back" in lower:
        patterns.extend((r"lying on the back with legs spread and hips lifted", r"reclining with thighs open", r"lying on the back with legs spread"))
        detail = re.sub(r"\bcum on lower back and ass\b", "cum across thighs and pussy", detail, flags=re.IGNORECASE)
        detail = re.sub(r"\bcum (?:on|dripping from) ass\b", "cum across thighs and pussy", detail, flags=re.IGNORECASE)
    if "straddles" in lower:
        patterns.extend(
            (
                r"straddling a partner's hips in cowgirl position",
                r"reverse cowgirl over a partner's hips",
                r"straddling a partner",
                r"squatting on top of a partner",
            )
        )
    if "squats over" in lower:
        patterns.append(r"squatting on top of a partner")
    if "sits in the man's lap" in lower:
        patterns.append(r"seated in a partner's lap facing them")
    if "bends forward" in lower:
        patterns.append(r"bent over with ass raised")
    if "on all fours" in lower:
        patterns.append(r"on all fours with hips raised")
    if "face-down" in lower:
        patterns.append(r"face-down ass-up on the mattress")
    if "lies on her side" in lower:
        patterns.append(r"side-lying with thighs parted")
        detail = re.sub(r"\bcum on lower back and ass\b", "cum across thighs and pussy", detail, flags=re.IGNORECASE)
        detail = re.sub(r"\bcum (?:on|dripping from) ass\b", "cum across thighs and pussy", detail, flags=re.IGNORECASE)
    if "sits on the edge" in lower:
        patterns.append(r"sitting on the edge of the bed")
    if "bed edge" in lower:
        patterns.append(r"lying at the bed edge with thighs open")
    if "kneels in front" in lower:
        patterns.extend((r"kneeling with mouth open", r"kneeling in front of a standing partner"))
    if "stands braced" in lower:
        patterns.append(r"standing with cum on the body")
    for pattern in patterns:
        detail = re.sub(rf"\b{pattern}\b,?\s*", "", detail, flags=re.IGNORECASE)
    if not any(token in lower for token in ("face", "mouth", "lips", "tongue")):
        detail = re.sub(r"\bsaliva and cum mixed on the mouth\b", "visible semen on skin", detail, flags=re.IGNORECASE)
        detail = re.sub(r"\bcum on tongue and chin\b", "visible semen on skin", detail, flags=re.IGNORECASE)
        detail = re.sub(r"\bcum on face and lips\b", "visible semen on skin", detail, flags=re.IGNORECASE)
    detail = re.sub(r",\s*,", ",", detail)
    detail = re.sub(r"\bwith\s*,\s*", "with ", detail, flags=re.IGNORECASE)
    detail = re.sub(r"^with\s+", "", detail, flags=re.IGNORECASE)
    detail = re.sub(r"^and\s+", "", detail, flags=re.IGNORECASE)
    clauses: list[str] = []
    for clause in detail_clauses(detail):
        normalized = normalize_climax_view_clause(clause, role_graph)
        if climax_clause_duplicates_role(normalized, role_graph):
            continue
        if density != "dense" and normalized.lower() in ("orgasm during penetration", "post-orgasm visible release"):
            continue
        clauses.append(normalized)
    return limit_detail_for_density(join_detail_clauses(clauses), density, True)
