data/dialogs.py

Part of Dungeon Explorer.

"""NPC dialog trees: Python dicts with callable conditions and actions.

Each NPC has a dialog tree: dict of node_id -> {text, portrait, choices}.
Choices can have `condition` (callable taking game_state) and `action` (string key).
"""

SHOPKEEPER = {
    "_npc_name": "Shopkeeper",
    "greeting": {
        "text": "Welcome, adventurer! Looking to buy or sell?",
        "portrait": "shopkeeper",
        "choices": [
            {"text": "Show me your wares.", "next": "shop_open", "action": "open_shop"},
            {"text": "I'd like to sell something.", "next": "sell_open", "action": "open_sell"},
            {"text": "Just browsing. Goodbye.", "next": None},
        ],
    },
    "shop_open": {
        "text": "Take your time. I've got the finest gear this side of the dungeon.",
        "portrait": "shopkeeper",
        "choices": [
            {"text": "Thanks.", "next": None},
        ],
    },
    "sell_open": {
        "text": "Let's see what you've got. I'll give you a fair price.",
        "portrait": "shopkeeper",
        "choices": [
            {"text": "Thanks.", "next": None},
        ],
    },
}

INNKEEPER = {
    "_npc_name": "Innkeeper",
    "greeting": {
        "text": "Weary traveller! A rest costs 50 gold and will fully restore your health.",
        "portrait": "innkeeper",
        "choices": [
            {
                "text": "Rest (50 gold, full heal)",
                "next": "rested",
                "action": "inn_rest",
                "condition": lambda gs: gs.get("gold", 0) >= 50,
            },
            {
                "text": "I can't afford that. (Need 50 gold)",
                "next": None,
                "condition": lambda gs: gs.get("gold", 0) < 50,
            },
            {"text": "No thanks.", "next": None},
        ],
    },
    "rested": {
        "text": "There you go, good as new! The dungeon awaits.",
        "portrait": "innkeeper",
        "choices": [{"text": "Thanks!", "next": None}],
    },
}

BLACKSMITH = {
    "_npc_name": "Blacksmith",
    "greeting": {
        "text": "I can forge or repair your gear. Forging combines 3 same-rarity items. Repairs cost 50 gold.",
        "portrait": "blacksmith",
        "choices": [
            {"text": "Forge (combine 3 same-rarity items)", "next": "forge_done", "action": "open_blacksmith"},
            {
                "text": "Repair all equipment (50 gold)",
                "next": "repair_done",
                "action": "repair_equipment",
                "condition": lambda gs: gs.get("gold", 0) >= 50,
            },
            {"text": "I can't afford repairs right now.", "next": None, "condition": lambda gs: gs.get("gold", 0) < 50},
            {"text": "Not right now.", "next": None},
        ],
    },
    "forge_done": {
        "text": "There you go! Come back when you have more materials.",
        "portrait": "blacksmith",
        "choices": [{"text": "Thanks!", "next": None}],
    },
    "repair_done": {
        "text": "Good as new! Take care of your gear out there.",
        "portrait": "blacksmith",
        "choices": [{"text": "Thanks!", "next": None}],
    },
}


def _well_cost(gs):
    """Compute well cost dynamically from game state."""
    from nodes.well import Well

    return Well.get_cost(gs.get("max_dungeon_reached", 0))


WELL_KEEPER = {
    "_npc_name": "Well Keeper",
    "greeting": {
        "text": "The Well of Fate... Toss a coin and the dungeon reshapes itself. "
        "But beware, your checkpoints will be lost.",
        "portrait": "well_keeper",
        "choices": [
            {
                "text": "Toss a coin. (Re-roll dungeon, lose checkpoints)",
                "next": "rerolled",
                "action": "well_reroll",
                "condition": lambda gs: gs.get("gold", 0) >= _well_cost(gs),
            },
            {
                "text": "I can't afford that right now.",
                "next": None,
                "condition": lambda gs: gs.get("gold", 0) < _well_cost(gs),
            },
            {"text": "That's too risky. Nevermind.", "next": None},
        ],
    },
    "rerolled": {
        "text": "The waters stir... The dungeon has been reborn. Your checkpoints are gone. "
        "May fate be kinder this time.",
        "portrait": "well_keeper",
        "choices": [{"text": "Into the depths once more.", "next": None}],
    },
}

# Quest Board NPC
QUEST_BOARD = {
    "_npc_name": "Quest Board",
    "greeting": {
        "text": "The Quest Board stands before you. Adventurers post requests here.",
        "portrait": "quest_board",
        "choices": [
            {"text": "View available quests.", "next": "board_open", "action": "open_quest_board"},
            {"text": "Not right now.", "next": None},
        ],
    },
    "board_open": {
        "text": "Good luck, adventurer. Return when you've completed a task!",
        "portrait": "quest_board",
        "choices": [{"text": "Thanks.", "next": None}],
    },
}

# Enchanter NPC
ENCHANTER = {
    "_npc_name": "Enchanter",
    "greeting": {
        "text": "I can imbue your equipment with magical properties... for a price.",
        "portrait": "enchanter",
        "choices": [
            {"text": "Enchant my gear.", "next": "enchant_open", "action": "open_enchanter"},
            {"text": "Maybe later.", "next": None},
        ],
    },
    "enchant_open": {
        "text": "Choose wisely. The enchantment is random, but always powerful.",
        "portrait": "enchanter",
        "choices": [{"text": "I see.", "next": None}],
    },
}

# Registry: NPC id -> dialog tree
DIALOGS = {
    "shopkeeper": SHOPKEEPER,
    "innkeeper": INNKEEPER,
    "blacksmith": BLACKSMITH,
    "well_keeper": WELL_KEEPER,
    "quest_board": QUEST_BOARD,
    "enchanter": ENCHANTER,
}