from sqlalchemy import Column, BigInteger, Float, Text, Boolean, ForeignKey
from sqlalchemy.orm import declarative_base, relationship

Base = declarative_base()

class Connection(Base):
    """
    Модель для таблицы connections
    """
    __tablename__ = 'connections'

    tg_id = Column(BigInteger, primary_key=True, nullable=False)
    balance = Column(Float, nullable=False, default=0.0)
    trial = Column(BigInteger, nullable=False, default=0)

    # Связь с ключами и рефералами
    keys = relationship('Key', back_populates='connection')
    referrals_received = relationship('Referral', foreign_keys='Referral.referred_tg_id', back_populates='referred')
    referrals_sent = relationship('Referral', foreign_keys='Referral.referrer_tg_id', back_populates='referrer')


class Key(Base):
    """
    Модель для таблицы keys
    """
    __tablename__ = 'keys'

    tg_id = Column(BigInteger, ForeignKey('connections.tg_id'), primary_key=True, nullable=False)
    client_id = Column(Text, primary_key=True, nullable=False)
    email = Column(Text, nullable=False)
    created_at = Column(BigInteger, nullable=False)
    expiry_time = Column(BigInteger, nullable=False)
    key = Column(Text, nullable=False)
    server_id = Column(Text, nullable=False, default='server1')
    notified = Column(Boolean, nullable=False, default=False)
    notified_24h = Column(Boolean, nullable=False, default=False)

    # Связь с подключением
    connection = relationship('Connection', back_populates='keys')


class Referral(Base):
    """
    Модель для таблицы referrals
    """
    __tablename__ = 'referrals'

    referred_tg_id = Column(BigInteger, ForeignKey('connections.tg_id'), primary_key=True, nullable=False)
    referrer_tg_id = Column(BigInteger, ForeignKey('connections.tg_id'), nullable=False)
    reward_issued = Column(Boolean, default=False)

    # Связи с подключениями
    referred = relationship('Connection', foreign_keys=[referred_tg_id], back_populates='referrals_received')
    referrer = relationship('Connection', foreign_keys=[referrer_tg_id], back_populates='referrals_sent')