from sqlalchemy import Column, String, Numeric, DateTime, ForeignKey, Integer, Enum, Text, BigInteger
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import declarative_base, relationship
from datetime import datetime
from enum import Enum as PyEnum
import uuid

Base = declarative_base()

class SubscriptionStatus(PyEnum):
    ACTIVE = "active"
    EXPIRED = "expired"
    CANCELLED = "cancelled"

class TransactionStatus(PyEnum):
    PENDING = "pending"
    SUCCESS = "success"
    FAILED = "failed"

class TransactionType(PyEnum):
    DEPOSIT = "deposit"
    WITHDRAWAL = "withdrawal"
    PAYMENT = "payment"

# Пользователи
class User(Base):
    __tablename__ = 'users'
    
    telegram_id = Column(BigInteger, primary_key=True)
    username = Column(String(255))
    balance = Column(Numeric(10, 2), default=0.0)
    ref_code = Column(String(7), unique=True)  # Реферальный код пользователя
    invited_by = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    
    # Relationships
    subscriptions = relationship("Subscription", back_populates="user")
    transactions = relationship("Transaction", back_populates="user")
    sent_referrals = relationship("Referral", 
                                foreign_keys="Referral.inviter_id",
                                back_populates="inviter")
    received_referrals = relationship("Referral", 
                                    foreign_keys="Referral.invited_id", 
                                    back_populates="invited")

# Реферальные связи
class Referral(Base):
    __tablename__ = 'referrals'
    
    id = Column(Integer, primary_key=True)
    inviter_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
    invited_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
    created_at = Column(DateTime, default=datetime.utcnow)
    
    inviter = relationship("User", foreign_keys=[inviter_id], back_populates="sent_referrals")
    invited = relationship("User", foreign_keys=[invited_id], back_populates="received_referrals")

# Тарифные планы
class Plan(Base):
    __tablename__ = 'plans'
    
    id = Column(Integer, primary_key=True)
    name = Column(String(100), nullable=False)
    price = Column(Numeric(10, 2), nullable=False)
    duration_days = Column(Integer, nullable=False)
    description = Column(Text)
    
    subscriptions = relationship("Subscription", back_populates="plan")

# Подписки
class Subscription(Base):
    __tablename__ = 'subscriptions'
    
    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    user_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
    plan_id = Column(Integer, ForeignKey('plans.id'), nullable=False)
    vpn_server_id = Column(String)  # ID сервера в Marzban
    status = Column(Enum(SubscriptionStatus), default=SubscriptionStatus.ACTIVE)
    start_date = Column(DateTime, nullable=False)
    end_date = Column(DateTime, nullable=False)
    created_at = Column(DateTime, default=datetime.utcnow)
    
    user = relationship("User", back_populates="subscriptions")
    plan = relationship("Plan", back_populates="subscriptions")

# Транзакции
class Transaction(Base):
    __tablename__ = 'transactions'
    
    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    user_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
    amount = Column(Numeric(10, 2), nullable=False)
    status = Column(Enum(TransactionStatus), default=TransactionStatus.PENDING)
    type = Column(Enum(TransactionType), nullable=False)
    payment_provider = Column(String(100))
    payment_id = Column(String, unique=True)  # ID платежа в внешней системе
    created_at = Column(DateTime, default=datetime.utcnow)
    
    user = relationship("User", back_populates="transactions")