import uuid

from sqlalchemy import (
    ARRAY,
    DECIMAL,
    VARCHAR,
    BigInteger,
    Boolean,
    Column,
    DateTime,
    ForeignKey,
    Index,
    Integer,
    MetaData,
    String,
    UniqueConstraint,
    ForeignKeyConstraint,
)
from sqlalchemy.dialects.postgresql import JSONB, UUID
from sqlalchemy.ext.compiler import compiles
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy.sql import and_, expression

"""
Naming conventions doc
https://docs.sqlalchemy.org/en/13/core/constraints.html#configuring-constraint-naming-conventions
"""
convention = {
    "ix": "ix_%(column_0_label)s",
    "uq": "uq_%(table_name)s_%(column_0_name)s",
    "ck": "ck_%(table_name)s_%(constraint_name)s",
    "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
    "pk": "pk_%(table_name)s",
}
metadata = MetaData(naming_convention=convention)
Base = declarative_base(metadata=metadata)

"""
Creating a utcnow function which runs on the Posgres database server when created_at and updated_at
fields are populated.
Following:
1. https://docs.sqlalchemy.org/en/13/core/compiler.html#utc-timestamp-function
2. https://www.postgresql.org/docs/current/functions-datetime.html#FUNCTIONS-DATETIME-CURRENT
3. https://stackoverflow.com/a/33532154/13659585
"""


class utcnow(expression.FunctionElement):
    type = DateTime


@compiles(utcnow, "postgresql")
def pg_utcnow(element, compiler, **kwargs):
    return "TIMEZONE('utc', statement_timestamp())"


class DropperContract(Base):  # type: ignore
    __tablename__ = "dropper_contracts"
    __table_args__ = (UniqueConstraint("blockchain", "address"),)

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    blockchain = Column(VARCHAR(128), nullable=False)
    address = Column(VARCHAR(256), index=True)
    title = Column(VARCHAR(128), nullable=True)
    description = Column(String, nullable=True)
    image_uri = Column(String, nullable=True)

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )


class DropperClaim(Base):  # type: ignore
    __tablename__ = "dropper_claims"

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    dropper_contract_id = Column(
        UUID(as_uuid=True),
        ForeignKey("dropper_contracts.id", ondelete="CASCADE"),
        nullable=False,
    )
    claim_id = Column(BigInteger, nullable=True)
    title = Column(VARCHAR(128), nullable=True)
    description = Column(String, nullable=True)
    terminus_address = Column(VARCHAR(256), nullable=True, index=True)
    terminus_pool_id = Column(BigInteger, nullable=True, index=True)
    claim_block_deadline = Column(BigInteger, nullable=True)
    active = Column(Boolean, default=False, nullable=False)

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )

    __table_args__ = (
        Index(
            "uq_dropper_claims_dropper_contract_id_claim_id",
            "dropper_contract_id",
            "claim_id",
            unique=True,
            postgresql_where=and_(claim_id.isnot(None), active.is_(True)),
        ),
    )


class DropperClaimant(Base):  # type: ignore
    __tablename__ = "dropper_claimants"
    __table_args__ = (UniqueConstraint("dropper_claim_id", "address"),)

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    dropper_claim_id = Column(
        UUID(as_uuid=True),
        ForeignKey("dropper_claims.id", ondelete="CASCADE"),
        nullable=False,
    )
    address = Column(VARCHAR(256), nullable=False, index=True)
    amount = Column(BigInteger, nullable=False)
    raw_amount = Column(String, nullable=True)
    added_by = Column(VARCHAR(256), nullable=False, index=True)
    signature = Column(String, nullable=True, index=True)

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )


class CallRequestType(Base):  # type: ignore
    """
    CallRequestType contains versions of call requests like:
    raw or dropper-v0.2.0.
    """

    __tablename__ = "call_request_types"

    name = Column(
        VARCHAR(128),
        primary_key=True,
        unique=True,
    )
    description = Column(String, nullable=True)


class MetatxRequester(Base):  # type: ignore
    """
    MetatxRequester represents id of user from bugout authorization.
    """

    __tablename__ = "metatx_requesters"

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        unique=True,
    )

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )

    registered_contracts = relationship(
        "RegisteredContract",
        back_populates="metatx_requester",
        cascade="all, delete, delete-orphan",
    )
    call_requests = relationship(
        "CallRequest",
        back_populates="metatx_requester",
        cascade="all, delete, delete-orphan",
    )


class Blockchain(Base):  # type: ignore
    __tablename__ = "blockchains"

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
    )
    name = Column(VARCHAR(128), nullable=False, index=True, unique=True)
    chain_id = Column(Integer, nullable=False, index=True, unique=False)
    testnet = Column(Boolean, default=False, nullable=False)

    registered_contracts = relationship(
        "RegisteredContract",
        back_populates="blockchain",
        cascade="all, delete, delete-orphan",
    )


class RegisteredContract(Base):  # type: ignore
    __tablename__ = "registered_contracts"
    __table_args__ = (
        UniqueConstraint(
            "blockchain_id",
            "metatx_requester_id",
            "address",
        ),
    )

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
    )
    metatx_requester_id = Column(
        UUID(as_uuid=True),
        ForeignKey("metatx_requesters.id", ondelete="CASCADE"),
        nullable=False,
    )
    blockchain_id = Column(
        UUID(as_uuid=True),
        ForeignKey("blockchains.id", ondelete="CASCADE"),
        nullable=False,
    )

    address = Column(VARCHAR(256), nullable=False, index=True)
    title = Column(VARCHAR(128), nullable=False)
    description = Column(String, nullable=True)
    image_uri = Column(String, nullable=True)

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )

    call_requests = relationship(
        "CallRequest",
        back_populates="registered_contract",
        cascade="all, delete, delete-orphan",
    )

    metatx_requester = relationship(
        "MetatxRequester", back_populates="registered_contracts"
    )
    blockchain = relationship("Blockchain", back_populates="registered_contracts")


class CallRequest(Base):
    __tablename__ = "call_requests"
    __table_args__ = (
        UniqueConstraint(
            "registered_contract_id",
            "request_id",
        ),
    )

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    registered_contract_id = Column(
        UUID(as_uuid=True),
        ForeignKey("registered_contracts.id", ondelete="CASCADE"),
        nullable=False,
    )
    call_request_type_name = Column(
        VARCHAR(128),
        ForeignKey("call_request_types.name", ondelete="CASCADE"),
        nullable=False,
    )
    metatx_requester_id = Column(
        UUID(as_uuid=True),
        ForeignKey("metatx_requesters.id", ondelete="CASCADE"),
        nullable=False,
    )

    caller = Column(VARCHAR(256), nullable=False, index=True)
    method = Column(String, nullable=False, index=True)
    request_id = Column(DECIMAL, nullable=False, index=True)
    parameters = Column(JSONB, nullable=False)

    expires_at = Column(DateTime(timezone=True), nullable=True, index=True)

    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )

    registered_contract = relationship(
        "RegisteredContract", back_populates="call_requests"
    )
    metatx_requester = relationship("MetatxRequester", back_populates="call_requests")


class Leaderboard(Base):  # type: ignore
    __tablename__ = "leaderboards"

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    title = Column(VARCHAR(128), nullable=False)
    description = Column(String, nullable=True)
    resource_id = Column(UUID(as_uuid=True), nullable=True, index=True)
    blockchain_ids = Column(ARRAY(Integer), nullable=False, default=[])

    wallet_connect = Column(Boolean, default=False, nullable=False)
    columns_names = Column(JSONB, nullable=False, default={})
    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )


class LeaderboardVersion(Base):  # type: ignore
    __tablename__ = "leaderboard_versions"
    __table_args__ = (UniqueConstraint("leaderboard_id", "version_number"),)

    leaderboard_id = Column(
        UUID(as_uuid=True),
        ForeignKey("leaderboards.id", ondelete="CASCADE"),
        primary_key=True,
        nullable=False,
    )
    version_number = Column(DECIMAL, primary_key=True, nullable=False)
    published = Column(Boolean, default=False, nullable=False)
    created_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        nullable=False,
        index=True,
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )


class LeaderboardScores(Base):  # type: ignore
    __tablename__ = "leaderboard_scores"
    __table_args__ = (
        UniqueConstraint("leaderboard_id", "address", "leaderboard_version_number"),
        ForeignKeyConstraint(
            ["leaderboard_id", "leaderboard_version_number"],
            [
                "leaderboard_versions.leaderboard_id",
                "leaderboard_versions.version_number",
            ],
            ondelete="CASCADE",
        ),
    )

    id = Column(
        UUID(as_uuid=True),
        primary_key=True,
        default=uuid.uuid4,
        unique=True,
        nullable=False,
    )
    leaderboard_id = Column(
        UUID(as_uuid=True),
        nullable=False,
    )
    leaderboard_version_number = Column(
        DECIMAL,
        nullable=False,
    )
    address = Column(VARCHAR(256), nullable=False, index=True)
    score = Column(BigInteger, nullable=False)
    points_data = Column(JSONB, nullable=True)
    created_at = Column(
        DateTime(timezone=True), server_default=utcnow(), nullable=False
    )
    updated_at = Column(
        DateTime(timezone=True),
        server_default=utcnow(),
        onupdate=utcnow(),
        nullable=False,
    )