# pylint: skip-file

from sqlalchemy import (
    MetaData, Table, Column, ForeignKey, PrimaryKeyConstraint,
    LargeBinary, Text, SmallInteger, Integer, BigInteger, Boolean,
    text
)
from .constants import TXO_TYPES, CLAIM_TYPE_CODES


SCHEMA_VERSION = '1.4'


metadata = MetaData()


Version = Table(
    'version', metadata,
    Column('version', Text, primary_key=True),
)


PubkeyAddress = Table(
    'pubkey_address', metadata,
    Column('address', Text, primary_key=True),
    Column('used_times', Integer, server_default='0'),
)


AccountAddress = Table(
    'account_address', metadata,
    Column('account', Text, primary_key=True),
    Column('address', Text, ForeignKey(PubkeyAddress.columns.address), primary_key=True),
    Column('chain', SmallInteger),
    Column('pubkey', LargeBinary),
    Column('chain_code', LargeBinary),
    Column('n', Integer),
    Column('depth', SmallInteger),
)


Block = Table(
    'block', metadata,
    Column('block_hash', LargeBinary, primary_key=True),
    Column('previous_hash', LargeBinary),
    Column('file_number', SmallInteger),
    Column('height', Integer),
    Column('timestamp', Integer),
    Column('block_filter', LargeBinary, nullable=True)
)


TX = Table(
    'tx', metadata,
    Column('block_hash', LargeBinary, nullable=True),
    Column('tx_hash', LargeBinary, primary_key=True),
    Column('raw', LargeBinary),
    Column('height', Integer),
    Column('position', SmallInteger),
    Column('timestamp', Integer, nullable=True),
    Column('day', Integer, nullable=True),
    Column('is_verified', Boolean, server_default='FALSE'),
    Column('purchased_claim_hash', LargeBinary, nullable=True),
    Column('tx_filter', LargeBinary, nullable=True)
)


TXO = Table(
    'txo', metadata,
    Column('tx_hash', LargeBinary, ForeignKey(TX.columns.tx_hash)),
    Column('txo_hash', LargeBinary, primary_key=True),
    Column('address', Text),
    Column('position', SmallInteger),
    Column('amount', BigInteger),
    Column('height', Integer),
    Column('spent_height', Integer, server_default='0'),
    Column('script_offset', Integer),
    Column('script_length', Integer),
    Column('is_reserved', Boolean, server_default='0'),

    # claims
    Column('txo_type', SmallInteger, server_default='0'),
    Column('claim_id', Text, nullable=True),
    Column('claim_hash', LargeBinary, nullable=True),
    Column('claim_name', Text, nullable=True),
    Column('channel_hash', LargeBinary, nullable=True),  # claims in channel
    Column('signature', LargeBinary, nullable=True),
    Column('signature_digest', LargeBinary, nullable=True),

    # channels
    Column('public_key', LargeBinary, nullable=True),
    Column('public_key_hash', LargeBinary, nullable=True),
)

txo_join_account = TXO.join(AccountAddress, TXO.columns.address == AccountAddress.columns.address)


def pg_add_txo_constraints_and_indexes(execute):
    execute(text("ALTER TABLE txo ADD PRIMARY KEY (txo_hash);"))
    # find appropriate channel public key for signing a content claim
    execute(text(f"""
        CREATE INDEX txo_channel_hash_w_height_desc_and_pub_key
        ON txo (claim_hash, height desc) INCLUDE (public_key)
        WHERE txo_type={TXO_TYPES['channel']};
    """))
    # update supports for a claim
    execute(text(f"""
        CREATE INDEX txo_unspent_supports
        ON txo (claim_hash) INCLUDE (amount)
        WHERE spent_height = 0 AND txo_type={TXO_TYPES['support']};
    """))
    # claim changes by height
    execute(text(f"""
        CREATE INDEX txo_claim_changes
        ON txo (height DESC) INCLUDE (txo_hash)
        WHERE spent_height = 0 AND txo_type IN {tuple(CLAIM_TYPE_CODES)};
    """))
    # supports added
    execute(text(f"""
        CREATE INDEX txo_added_supports_by_height ON txo (height DESC)
        INCLUDE (claim_hash) WHERE txo_type={TXO_TYPES['support']};
    """))
    # supports spent
    execute(text(f"""
        CREATE INDEX txo_spent_supports_by_height ON txo (spent_height DESC)
        INCLUDE (claim_hash) WHERE txo_type={TXO_TYPES['support']};
    """))


TXI = Table(
    'txi', metadata,
    Column('tx_hash', LargeBinary, ForeignKey(TX.columns.tx_hash)),
    Column('txo_hash', LargeBinary, ForeignKey(TXO.columns.txo_hash), primary_key=True),
    Column('address', Text, nullable=True),
    Column('position', SmallInteger),
    Column('height', Integer),
)

txi_join_account = TXI.join(AccountAddress, TXI.columns.address == AccountAddress.columns.address)


def pg_add_txi_constraints_and_indexes(execute):
    execute(text("ALTER TABLE txi ADD PRIMARY KEY (txo_hash);"))


Claim = Table(
    'claim', metadata,
    Column('claim_hash', LargeBinary, primary_key=True),
    Column('claim_id', Text),
    Column('claim_name', Text),
    Column('normalized', Text),
    Column('address', Text),
    Column('txo_hash', LargeBinary, ForeignKey(TXO.columns.txo_hash)),
    Column('amount', BigInteger),
    Column('staked_amount', BigInteger),
    Column('timestamp', Integer),  # last updated timestamp
    Column('creation_timestamp', Integer),
    Column('release_time', Integer, nullable=True),
    Column('height', Integer),  # last updated height
    Column('creation_height', Integer),
    Column('activation_height', Integer),
    Column('expiration_height', Integer),
    Column('takeover_height', Integer, nullable=True),
    Column('sync_height', Integer),  # claim dynamic values up-to-date as of this height (eg. staked_amount)
    Column('is_controlling', Boolean),

    # short_url: normalized#shortest-unique-claim_id
    Column('short_url', Text),
    # canonical_url: channel's-short_url/normalized#shortest-unique-claim_id-within-channel
    # canonical_url is computed dynamically

    Column('title', Text, nullable=True),
    Column('author', Text, nullable=True),
    Column('description', Text, nullable=True),

    Column('claim_type', SmallInteger),
    Column('claim_reposted_count', Integer, server_default='0'),
    Column('staked_support_count', Integer, server_default='0'),
    Column('staked_support_amount', BigInteger, server_default='0'),

    # streams
    Column('stream_type', SmallInteger, nullable=True),
    Column('media_type', Text, nullable=True),
    Column('fee_amount', BigInteger, server_default='0'),
    Column('fee_currency', Text, nullable=True),
    Column('duration', Integer, nullable=True),

    # reposts
    Column('reposted_claim_hash', LargeBinary, nullable=True),

    # claims which are channels
    Column('signed_claim_count', Integer, server_default='0'),
    Column('signed_support_count', Integer, server_default='0'),

    # claims which are inside channels
    Column('channel_hash', LargeBinary, nullable=True),
    Column('is_signature_valid', Boolean, nullable=True),

    Column('trending_group', BigInteger, server_default='0'),
    Column('trending_mixed', BigInteger, server_default='0'),
    Column('trending_local', BigInteger, server_default='0'),
    Column('trending_global', BigInteger, server_default='0'),
)


def pg_add_claim_constraints_and_indexes(execute):
    execute(text("ALTER TABLE claim ADD PRIMARY KEY (claim_hash);"))
    # finding claims that aren't updated with latest TXO
    execute(text("CREATE INDEX claim_txo_hash ON claim (txo_hash);"))
    # used to calculate content in a channel
    execute(text("""
        CREATE INDEX signed_content ON claim (channel_hash)
        INCLUDE (amount) WHERE is_signature_valid;
    """))


Tag = Table(
    'tag', metadata,
    Column('claim_hash', LargeBinary),
    Column('tag', Text),
)


Support = Table(
    'support', metadata,

    Column('txo_hash', LargeBinary, ForeignKey(TXO.columns.txo_hash), primary_key=True),
    Column('claim_hash', LargeBinary),
    Column('address', Text),
    Column('amount', BigInteger),
    Column('height', Integer),
    Column('timestamp', Integer),

    # support metadata
    Column('emoji', Text),

    # signed supports
    Column('channel_hash', LargeBinary, nullable=True),
    Column('signature', LargeBinary, nullable=True),
    Column('signature_digest', LargeBinary, nullable=True),
    Column('is_signature_valid', Boolean, nullable=True),
)


def pg_add_support_constraints_and_indexes(execute):
    execute(text("ALTER TABLE support ADD PRIMARY KEY (txo_hash);"))
    execute(text("""
        CREATE INDEX signed_support ON support (channel_hash)
        INCLUDE (amount) WHERE is_signature_valid;
    """))