from sqlalchemy import (
    Boolean,
    DateTime,
    Float,
    ForeignKey,
    Index,
    Integer,
    String,
    Text,
    UniqueConstraint,
)
from sqlalchemy.orm import DeclarativeBase, mapped_column, relationship
from sqlalchemy.sql import func, text


class Base(DeclarativeBase):
    pass


class Token(Base):
    __tablename__ = "tokens"

    id = mapped_column(Integer, primary_key=True, autoincrement=True)
    provider = mapped_column(String, nullable=False)
    token = mapped_column(String, nullable=False)
    token_type = mapped_column(String, default="user", nullable=False)
    priority = mapped_column(Integer, default=0, server_default=text("0"), nullable=False)
    is_enabled = mapped_column(Boolean, default=True, server_default=text("true"), nullable=False)
    created_at = mapped_column(DateTime, default=func.now(), server_default=func.now(), nullable=False)
    last_chat_cleanup = mapped_column(DateTime, nullable=True)

    __table_args__ = (
        UniqueConstraint("provider", "token", name="uq_provider_token"),
    )

    stats = relationship("TokenStats", back_populates="token_rel", uselist=False, cascade="all, delete-orphan")


class TokenStats(Base):
    __tablename__ = "token_stats"

    id = mapped_column(Integer, primary_key=True, autoincrement=True)
    token_id = mapped_column(
        Integer, ForeignKey("tokens.id", ondelete="CASCADE"), nullable=False, unique=True
    )
    total_requests = mapped_column(Integer, default=0, server_default=text("0"), nullable=False)
    successful_requests = mapped_column(Integer, default=0, server_default=text("0"), nullable=False)
    failed_requests = mapped_column(Integer, default=0, server_default=text("0"), nullable=False)
    last_success_time = mapped_column(DateTime, nullable=True)
    last_failure_time = mapped_column(DateTime, nullable=True)

    token_rel = relationship("Token", back_populates="stats")


class RequestLog(Base):
    __tablename__ = "request_logs"
    __table_args__ = (
        Index("ix_request_logs_timestamp", "timestamp"),
        Index("ix_request_logs_provider_timestamp", "provider", "timestamp"),
    )

    id = mapped_column(Integer, primary_key=True, autoincrement=True)
    provider = mapped_column(String, nullable=False)
    endpoint = mapped_column(String, nullable=False, default="", server_default=text("''"))
    source = mapped_column(String, nullable=False, default="unknown", server_default=text("'unknown'"))
    protocol = mapped_column(String, nullable=False, default="unknown", server_default=text("'unknown'"))
    client_name = mapped_column(String, nullable=False, default="Unknown", server_default=text("'Unknown'"))
    auth_token = mapped_column(Text, nullable=True)
    upstream_auth_token = mapped_column(Text, nullable=True)
    model = mapped_column(String, nullable=False)
    status_code = mapped_column(Integer, nullable=False, default=200, server_default=text("200"))
    success = mapped_column(Boolean, nullable=False)
    duration = mapped_column(Float, nullable=False, default=0.0, server_default=text("0.0"))
    first_token_time = mapped_column(Float, nullable=False, default=0.0, server_default=text("0.0"))
    input_tokens = mapped_column(Integer, nullable=False, default=0, server_default=text("0"))
    output_tokens = mapped_column(Integer, nullable=False, default=0, server_default=text("0"))
    cache_creation_tokens = mapped_column(Integer, nullable=False, default=0, server_default=text("0"))
    cache_read_tokens = mapped_column(Integer, nullable=False, default=0, server_default=text("0"))
    total_tokens = mapped_column(Integer, nullable=False, default=0, server_default=text("0"))
    error_message = mapped_column(Text, nullable=True)
    timestamp = mapped_column(DateTime, default=func.now(), server_default=func.now(), nullable=False)


class ConfigItem(Base):
    __tablename__ = "config_items"

    key = mapped_column(String, primary_key=True)
    value = mapped_column(Text, nullable=False)