from sqlalchemy import Column, String, Numeric, DateTime, Boolean, ForeignKey, Integer, Enum, Text
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
from sqlalchemy.orm import declarative_base, relationship, sessionmaker
from datetime import datetime
from enum import Enum as PyEnum
import uuid

Base = declarative_base()

def generate_uuid():
    return str(uuid.uuid4())


class TicketStatus(PyEnum):
    OPEN = "open"
    PENDING = "pending"
    CLOSED = "closed"


"""Пользователи"""
class User(Base):
    __tablename__ = 'users'

    id = Column(UUID(as_uuid=True), primary_key=True, default=generate_uuid)
    telegram_id = Column(String, unique=True, nullable=False)  # telegram_id как уникальный идентификатор
    username = Column(String)
    balance = Column(Numeric(10, 2), default=0.0)
    referrer_id = Column(String, ForeignKey('users.telegram_id'), nullable=True)  # Ссылка на telegram_id
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    referrals = relationship("User", backref="referrer", remote_side=[telegram_id])  # Ссылка на telegram_id
    subscriptions = relationship("Subscription", back_populates="user")
    transactions = relationship("Transaction", back_populates="user")
    admins = relationship("Administrators", back_populates="user")


"""Подписки"""
class Subscription(Base):
    __tablename__ = 'subscriptions'

    id = Column(String, primary_key=True, default=generate_uuid)
    user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
    vpn_server_id = Column(String)
    plan = Column(String)
    expiry_date = Column(DateTime)
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    user = relationship("User", back_populates="subscriptions")

"""Транзакции"""
class Transaction(Base):
    __tablename__ = 'transactions'

    id = Column(String, primary_key=True, default=generate_uuid)
    user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
    amount = Column(Numeric(10, 2))
    transaction_type = Column(String)
    created_at = Column(DateTime, default=datetime.utcnow)

    user = relationship("User", back_populates="transactions")

"""Тикет"""
class SupportTicket(Base):
    __tablename__ = "support_tickets"

    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
    subject = Column(String, nullable=False)
    message = Column(String, nullable=False)
    status = Column(Enum(TicketStatus), default=TicketStatus.OPEN, nullable=False)
    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

"""Сообщения из тикетов"""
class TicketMessage(Base):
    __tablename__ = "ticket_messages"

    id = Column(Integer, primary_key=True, index=True)
    ticket_id = Column(Integer, ForeignKey("support_tickets.id"), nullable=False)  # ID тикета
    sender = Column(String, nullable=False)  # "user" или "support"
    message = Column(Text, nullable=False)  # Текст сообщения
    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)  #

"""Администраторы"""
class Administrators(Base):
    __tablename__ = 'admins'

    id = Column(String, primary_key=True, default=generate_uuid)
    user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
    
    user = relationship("User", back_populates="admins")