from sqlalchemy import Column, Integer, String, DateTime, Boolean, ForeignKey, Text, JSON
from sqlalchemy.orm import relationship
from datetime import datetime, timezone
from database import Base


class User(Base):
    """Модель пользователя"""
    __tablename__ = "users"

    id = Column(Integer, primary_key=True)
    telegram_id = Column(Integer, unique=True, nullable=False, index=True)
    username = Column(String(255))
    first_name = Column(String(255))
    last_name = Column(String(255))
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
    is_admin = Column(Boolean, default=False)
    # Клиентский счет в формате: XX-XX-XX-XX-XX-XX-XX (7 пар цифр через дефис)
    account_number = Column(String(20), unique=True, nullable=True, index=True)

    # Связи
    participations = relationship("Participation", back_populates="user")

    def __repr__(self):
        return f"<User(telegram_id={self.telegram_id}, username={self.username})>"


class Lottery(Base):
    """Модель розыгрыша"""
    __tablename__ = "lotteries"

    id = Column(Integer, primary_key=True)
    title = Column(String(500), nullable=False)
    description = Column(Text)
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
    start_date = Column(DateTime(timezone=True))
    end_date = Column(DateTime(timezone=True))
    is_active = Column(Boolean, default=True)
    is_completed = Column(Boolean, default=False)
    prizes = Column(JSON)  # Список призов в формате JSON
    creator_id = Column(Integer, ForeignKey("users.id"), nullable=False)
    
    # Настройки для ручного управления победителями
    manual_winners = Column(JSON, default=lambda: {})  # {место: telegram_id}
    draw_results = Column(JSON)  # Результаты розыгрыша
    
    # Тип отображения победителей: "username", "chat_id", "account_number"
    winner_display_type = Column(String(20), default="username")

    # Связи
    creator = relationship("User")
    participations = relationship("Participation", back_populates="lottery")

    def __repr__(self):
        return f"<Lottery(id={self.id}, title={self.title})>"


class Participation(Base):
    """Модель участия в розыгрыше"""
    __tablename__ = "participations"

    id = Column(Integer, primary_key=True)
    user_id = Column(Integer, ForeignKey("users.id"), nullable=True)  # Опционально
    lottery_id = Column(Integer, ForeignKey("lotteries.id"), nullable=False)
    account_number = Column(String(20), nullable=True, index=True)  # Счет участника (XX-XX-XX-XX-XX-XX-XX)
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
    
    # Связи
    user = relationship("User", back_populates="participations")
    lottery = relationship("Lottery", back_populates="participations")

    def __repr__(self):
        if self.account_number:
            return f"<Participation(account={self.account_number}, lottery_id={self.lottery_id})>"
        return f"<Participation(user_id={self.user_id}, lottery_id={self.lottery_id})>"


class Winner(Base):
    """Модель победителя розыгрыша"""
    __tablename__ = "winners"

    id = Column(Integer, primary_key=True)
    lottery_id = Column(Integer, ForeignKey("lotteries.id"), nullable=False)
    user_id = Column(Integer, ForeignKey("users.id"), nullable=True)  # Опционально
    account_number = Column(String(20), nullable=True, index=True)  # Счет победителя
    place = Column(Integer, nullable=False)  # Место (1, 2, 3...)
    prize = Column(String(500))  # Описание приза
    is_manual = Column(Boolean, default=False)  # Был ли установлен вручную
    created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))

    # Связи
    user = relationship("User")
    lottery = relationship("Lottery")

    def __repr__(self):
        if self.account_number:
            return f"<Winner(lottery_id={self.lottery_id}, account={self.account_number}, place={self.place})>"
        return f"<Winner(lottery_id={self.lottery_id}, user_id={self.user_id}, place={self.place})>"