from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime, func
from sqlalchemy.orm import relationship
from .database import Base


class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, autoincrement=True)
    username = Column(String(50), unique=True, nullable=False, index=True)
    password_hash = Column(String(255), nullable=False)
    role = Column(String(20), nullable=False, default="user")  # "admin" or "user"
    created_at = Column(DateTime(timezone=True), server_default=func.now())


class Set(Base):
    __tablename__ = "sets"

    id = Column(Integer, primary_key=True, autoincrement=True)
    code = Column(String(10), unique=True, nullable=False, index=True)  # e.g. "CRI"
    name = Column(String(200), nullable=False)                          # e.g. "Chaos Rising"
    release_date = Column(String(20), nullable=True)
    description = Column(Text, nullable=True)
    created_at = Column(DateTime(timezone=True), server_default=func.now())

    cards = relationship("Card", back_populates="set", cascade="all, delete-orphan",
                         order_by="Card.number")


class Card(Base):
    __tablename__ = "cards"

    id = Column(Integer, primary_key=True, autoincrement=True)
    set_id = Column(Integer, ForeignKey("sets.id", ondelete="CASCADE"), nullable=False, index=True)
    number = Column(String(10), nullable=False)        # e.g. "068"
    name = Column(String(200), nullable=False)          # e.g. "Garchomp EX"
    type = Column(String(50), nullable=True)            # e.g. "Colorless", "Fire"
    rarity = Column(String(50), nullable=True)          # e.g. "Rare", "Uncommon"
    hp = Column(Integer, nullable=True)
    stage = Column(String(50), nullable=True)           # e.g. "Basic", "Stage 1", "Stage 2"
    description = Column(Text, nullable=True)
    image_url = Column(String(500), nullable=True)

    set = relationship("Set", back_populates="cards")

    @property
    def full_code(self):
        return f"{self.set.code}-{self.number}"