"""
Collection models for user card collections
"""

from sqlalchemy import Column, Integer, String, DateTime, Boolean, ForeignKey, Text, Float, JSON
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
from app.database import Base


class Collection(Base):
    __tablename__ = "collections"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, nullable=False)
    description = Column(Text)
    is_public = Column(Boolean, default=False)
    owner_id = Column(Integer, ForeignKey("users.id"), nullable=False)
    
    # Metadata
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())

    # Relationships
    owner = relationship("User", back_populates="collections")
    collection_cards = relationship("CollectionCard", back_populates="collection", cascade="all, delete-orphan")

    @property
    def total_cards(self):
        """Calculate total number of cards in collection"""
        return sum(cc.quantity for cc in self.collection_cards)

    @property
    def total_value(self):
        """Calculate total estimated value of collection"""
        total = 0
        for cc in self.collection_cards:
            if cc.card.current_price:
                total += cc.card.current_price * cc.quantity
        return total

    def __repr__(self):
        return f"<Collection(id={self.id}, name='{self.name}', owner_id={self.owner_id})>"


class CollectionCard(Base):
    __tablename__ = "collection_cards"

    id = Column(Integer, primary_key=True, index=True)
    collection_id = Column(Integer, ForeignKey("collections.id"), nullable=False)
    card_id = Column(Integer, ForeignKey("cards.id"), nullable=False)
    
    # Ownership details
    quantity = Column(Integer, default=1)
    condition = Column(String, default="NM")  # NM, LP, MP, HP, DMG
    foil = Column(Boolean, default=False)
    language = Column(String, default="English")
    
    # Storage and notes
    location = Column(String)  # Binder 1, Page 5, etc.
    notes = Column(Text)
    purchase_price = Column(Float)  # What user paid for it
    purchase_date = Column(DateTime)
    
    # User's card images
    user_images = Column(JSON)  # Array of image URLs/paths for this specific card copy
    condition_notes = Column(Text)  # Detailed condition description
    
    # Metadata
    added_at = Column(DateTime(timezone=True), server_default=func.now())
    updated_at = Column(DateTime(timezone=True), onupdate=func.now())

    # Relationships
    collection = relationship("Collection", back_populates="collection_cards")
    card = relationship("Card", back_populates="collection_cards")

    def __repr__(self):
        return f"<CollectionCard(collection_id={self.collection_id}, card_id={self.card_id}, quantity={self.quantity})>"