from dream_customs.models import FakeTextClient
from dream_customs.pipeline import build_intake, generate_model_led_pact
from dream_customs.prompts import (
    dream_brief_prompt,
    pact_critique_prompt,
    pact_draft_prompt,
    pact_rewrite_prompt,
)
from dream_customs.schema import DreamBrief, PactCard, PactCritique


def test_dream_brief_prompt_requires_english_demo_brief():
    intake = build_intake(
        dream_text="I kept missing an elevator. The buttons melted like wax, and floor 14 froze.",
        visual_clues=["Visible text: 14", "Object: melted button"],
        mood="Uneasy",
    )

    prompt = dream_brief_prompt(intake)

    assert "English demo" in prompt
    assert "anchors" in prompt
    assert "visual_evidence" in prompt
    assert "Do not diagnose" in prompt
    assert "emotional_hypothesis" in prompt
    assert "today_bridge" in prompt
    assert "safety_flags" in prompt
    assert "language" in prompt
    assert '"en"' in prompt
    assert "Return strict JSON" in prompt


def test_pact_draft_prompt_uses_brief_not_template_fallback():
    brief = DreamBrief(
        anchors=["elevator", "melted wax", "floor 14"],
        emotional_hypothesis="The dream may be protecting a fear of getting stuck.",
        today_bridge="Choose one stalled task and name the next small movement.",
        visual_evidence=["Visible text: 14"],
    )

    prompt = pact_draft_prompt(brief, "I want to stop freezing before a task.")

    assert "Write natural English" in prompt
    assert "Use at least two anchors" in prompt
    assert "Do not use template phrases" in prompt
    assert "Use visual_evidence if present" in prompt
    assert "elevator" in prompt
    assert "visitor_name" in prompt
    assert "permit_id" in prompt
    assert "contraband" in prompt
    assert "risk_level" in prompt
    assert "alliance_reading" in prompt
    assert "practical_suggestion" in prompt
    assert "weird_task" in prompt
    assert "bedtime_release" in prompt
    assert "safety_note" in prompt


def test_pact_critique_prompt_checks_screenshot_regression():
    brief = DreamBrief(
        anchors=["elevator", "melted wax", "floor 14"],
        emotional_hypothesis="The dream may be protecting a fear of getting stuck.",
        today_bridge="Choose one stalled task and name the next small movement.",
        visual_evidence=["Visible text: 14"],
    )
    card = PactCard(
        visitor_name="The Elevator",
        permit_id="DREAM20260608-014",
        contraband=["stuck feeling"],
        risk_level="low",
        alliance_reading="The an elevator is trying to keep you from rushing.",
        practical_suggestion="Take the the next small step.",
        weird_task="Tap an imaginary floor 14 button.",
        bedtime_release="The elevator can rest outside my room tonight.",
    )

    prompt = pact_critique_prompt(brief, card)

    assert "the an" in prompt
    assert "the the" in prompt
    assert "invented details" in prompt
    assert "natural English" in prompt
    assert "passes" in prompt
    assert "issues" in prompt
    assert "rewrite_instruction" in prompt
    assert "diagnosis" in prompt
    assert "frightening certainty" in prompt
    assert "generic wellness" in prompt
    assert "missing anchors" in prompt


def test_pact_rewrite_prompt_uses_critique_instruction():
    brief = DreamBrief(
        anchors=["elevator", "melted wax", "floor 14"],
        emotional_hypothesis="The dream may be protecting a fear of getting stuck.",
        today_bridge="Choose one stalled task and name the next small movement.",
        visual_evidence=["Visible text: 14"],
    )
    card = PactCard(
        visitor_name="The Elevator",
        permit_id="DREAM20260608-014",
        contraband=["stuck feeling"],
        risk_level="low",
        alliance_reading="The an elevator is trying to keep you from rushing.",
        practical_suggestion="Take the the next small step.",
        weird_task="Tap an imaginary floor 14 button.",
        bedtime_release="The elevator can rest outside my room tonight.",
    )
    critique = PactCritique(
        passes=False,
        issues=["repeated article"],
        rewrite_instruction="Rewrite without repeated articles while preserving floor 14.",
    )

    prompt = pact_rewrite_prompt(brief, card, critique)

    assert "Rewrite without repeated articles" in prompt
    assert "Return strict JSON" in prompt
    assert "floor 14" in prompt
    assert "visitor_name" in prompt
    assert "permit_id" in prompt
    assert "contraband" in prompt
    assert "risk_level" in prompt
    assert "alliance_reading" in prompt
    assert "practical_suggestion" in prompt
    assert "weird_task" in prompt
    assert "bedtime_release" in prompt
    assert "safety_note" in prompt
    assert "safe non-diagnostic" in prompt
    assert "natural English" in prompt
    assert "Do not add objects" in prompt


def test_fake_text_client_supports_model_led_methods():
    client = FakeTextClient()

    brief = client.generate_brief("brief prompt")
    card = client.generate_pact_draft("draft prompt")
    critique = client.critique_pact("critique prompt")

    assert brief.language == "en"
    assert brief.anchors
    assert card.visitor_name
    assert critique.passes


def test_model_led_pact_rewrites_screenshot_elevator_regression():
    class CriticTextClient(FakeTextClient):
        def generate_pact_draft(self, prompt):
            return PactCard(
                visitor_name="An Elevator",
                permit_id="DREAM20260608-015",
                contraband=["melted buttons", "froze at 14"],
                risk_level="medium: handle gently, without treating it as a warning sign",
                alliance_reading="I am safe, but the wax is sticky and the floor is cold.",
                practical_suggestion="Pick one real task that feels like the an elevator.",
                weird_task="I will try to push the floor button with my hand instead of the lever.",
                bedtime_release="Tonight, the an elevator and the the button are logged.",
            )

        def critique_pact(self, prompt):
            return PactCritique(
                passes=False,
                issues=["repeated article", "invented lever"],
                rewrite_instruction="Rewrite without repeated articles or the invented lever.",
            )

        def rewrite_pact(self, prompt):
            return PactCard(
                visitor_name="The Elevator Stuck at 14",
                permit_id="DREAM20260608-015",
                contraband=["melted buttons", "floor 14", "the urge to freeze before starting"],
                risk_level="medium: handle gently, without treating it as a warning sign",
                alliance_reading=(
                    "The elevator, wax-soft buttons, and frozen 14 can be treated as a small scene "
                    "about getting stuck before the first move."
                ),
                practical_suggestion="Choose one stalled task and write only the next button-sized action.",
                weird_task="Draw floor 14 on a sticky note, tap it once, and spend five minutes on the first step.",
                bedtime_release=(
                    "Tonight, the elevator can stay on floor 14 while tomorrow's first button waits quietly."
                ),
            )

    intake = build_intake(
        dream_text="I kept missing an elevator. The buttons melted like wax, and the floor number froze at 14.",
        visual_clues=["Visible text: 14", "Object: melted elevator button"],
        mood="Uneasy",
    )

    card, html = generate_model_led_pact(intake, "", CriticTextClient())
    joined = card.to_plain_text().lower()

    assert "the an" not in joined
    assert "the the" not in joined
    assert "lever" not in joined
    assert "floor 14" in joined
    assert "melted" in joined or "wax" in joined
    assert "Today's Clearance Pass" in html or "Today's Pact" in html