#!/usr/bin/env python3
"""
draw_card.py — Draw a card from The Chaos deck.

Usage:
    python3 draw_card.py <deck> <table> [count]

Decks: souls, cook, creatures, curiosities
Tables: see the YAML files (e.g. traits, rumours, type, drink_or_drug)

Examples:
    python3 draw_card.py souls thing
    python3 draw_card.py cook rumours 3
    python3 draw_card.py creatures type appearance
    python3 draw_card.py curiosities creepy_v<NEW_CHAR_HERE>_vibe
"""

import sys
import yaml
import random
import os

DECK_DIR = os.path.join(os.path.dirname(__file__), '..', 'themes', 'default', 'rules', 'deck')

DECKS = {
    'souls': 'souls.yaml',
    'cook': 'cook.yaml',
    'creatures': 'creatures.yaml',
    'curiosities': 'curiosities.yaml',
}

def load_deck(name):
    path = os.path.join(DECK_DIR, DECKS[name])
    with open(path) as f:
        return yaml.safe_load(f)

def draw_from_table(data, table_name):
    entry = data.get(table_name)
    if not entry:
        print(f"Table '{table_name}' not found in deck.")
        sys.exit(1)

    table = entry.get('table')
    if table:
        keys = sorted(table.keys(), key=int)
        roll = random.randint(keys[0], keys[-1])
        result = table[roll]
        if isinstance(result, dict):
            return result.get('name', str(result)), roll
        return result, roll

    one_to_six = entry.get('animal_table') or entry.get('trinket_table')
    if one_to_six and entry.get('has_table'):
        r1 = random.randint(1, 6)
        r2 = random.randint(1, 6)
        return f"{one_to_six[r1]} and has {entry['has_table'][r2]}", f"{r1},{r2}"

    faction_keys = [k for k in entry if k.startswith('table_')]
    if faction_keys:
        key = random.choice(faction_keys)
        items = entry[key]['factions']
        return random.choice(items), key

    return None, None

def main():
    if len(sys.argv) < 3:
        print(__doc__.strip())
        sys.exit(1)

    deck_name = sys.argv[1].lower()
    table_name = sys.argv[2].lower()
    count = int(sys.argv[3]) if len(sys.argv) > 3 else 1

    if deck_name not in DECKS:
        print(f"Unknown deck '{deck_name}'. Choose from: {', '.join(DECKS.keys())}")
        sys.exit(1)

    data = load_deck(deck_name)

    print(f"── [{deck_name}] drawing from '{table_name}' ──")
    for _ in range(count):
        result, roll = draw_from_table(data, table_name)
        if result:
            if roll is not None:
                print(f"  [{roll}] → {result}")
            else:
                print(f"  → {result}")
        else:
            # Try direct pick from top-level
            if table_name == 'rules':
                print("  (No card — consult mechanics.md)")
            else:
                print(f"  No result for '{table_name}'")

if __name__ == '__main__':
    main()