from sqlalchemy.orm import Session
from app import models

INITIAL_CARDS = [
    {"name": "Goblin Grunt", "rarity": "COMMON"},
    {"name": "Elven Archer", "rarity": "COMMON"},
    {"name": "Orc Warrior", "rarity": "COMMON"},
    {"name": "Dwarven Shield", "rarity": "COMMON"},
    {"name": "Healing Potion", "rarity": "COMMON"},
    {"name": "Fireball Scroll", "rarity": "RARE"},
    {"name": "Silver Sword", "rarity": "RARE"},
    {"name": "Mystic Amulet", "rarity": "RARE"},
    {"name": "Dragon Scale", "rarity": "LEGENDARY"},
    {"name": "King's Crown", "rarity": "LEGENDARY"},
]

INITIAL_CHESTS = [
    {"name": "Wooden Chest", "cost_gold": 100, "cards_per_open": 3}
]

def seed_data(db: Session):
    # Seed Cards
    for card_data in INITIAL_CARDS:
        exists = db.query(models.Card).filter_by(name=card_data["name"]).first()
        if not exists:
            card = models.Card(**card_data)
            db.add(card)
    
    # Seed Chests
    for chest_data in INITIAL_CHESTS:
        exists = db.query(models.Chest).filter_by(name=chest_data["name"]).first()
        if not exists:
            chest = models.Chest(**chest_data)
            db.add(chest)
    
    db.commit()