"""Arbitrum Sepolia blockchain support

Revision ID: b38b75b11247
Revises: 2391c3cc5050
Create Date: 2024-02-20 15:16:40.393985

"""

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = "b38b75b11247"
down_revision = "2391c3cc5050"
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        "arbitrum_sepolia_blocks",
        sa.Column("block_number", sa.BigInteger(), nullable=False),
        sa.Column("difficulty", sa.BigInteger(), nullable=True),
        sa.Column("extra_data", sa.VARCHAR(length=128), nullable=True),
        sa.Column("gas_limit", sa.BigInteger(), nullable=True),
        sa.Column("gas_used", sa.BigInteger(), nullable=True),
        sa.Column("base_fee_per_gas", sa.Numeric(precision=78, scale=0), nullable=True),
        sa.Column("hash", sa.VARCHAR(length=256), nullable=True),
        sa.Column("logs_bloom", sa.VARCHAR(length=1024), nullable=True),
        sa.Column("miner", sa.VARCHAR(length=256), nullable=True),
        sa.Column("nonce", sa.VARCHAR(length=256), nullable=True),
        sa.Column("parent_hash", sa.VARCHAR(length=256), nullable=True),
        sa.Column("receipt_root", sa.VARCHAR(length=256), nullable=True),
        sa.Column("uncles", sa.VARCHAR(length=256), nullable=True),
        sa.Column("size", sa.Integer(), nullable=True),
        sa.Column("state_root", sa.VARCHAR(length=256), nullable=True),
        sa.Column("timestamp", sa.BigInteger(), nullable=True),
        sa.Column("total_difficulty", sa.VARCHAR(length=256), nullable=True),
        sa.Column("transactions_root", sa.VARCHAR(length=256), nullable=True),
        sa.Column(
            "indexed_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("TIMEZONE('utc', statement_timestamp())"),
            nullable=False,
        ),
        sa.Column("sha3_uncles", sa.VARCHAR(length=256), nullable=True),
        sa.Column("l1_block_number", sa.BigInteger(), nullable=True),
        sa.Column("send_count", sa.BigInteger(), nullable=True),
        sa.Column("send_root", sa.VARCHAR(length=256), nullable=True),
        sa.Column("mix_hash", sa.VARCHAR(length=256), nullable=True),
        sa.PrimaryKeyConstraint(
            "block_number", name=op.f("pk_arbitrum_sepolia_blocks")
        ),
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_blocks_block_number"),
        "arbitrum_sepolia_blocks",
        ["block_number"],
        unique=True,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_blocks_hash"),
        "arbitrum_sepolia_blocks",
        ["hash"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_blocks_timestamp"),
        "arbitrum_sepolia_blocks",
        ["timestamp"],
        unique=False,
    )
    op.create_table(
        "arbitrum_sepolia_labels",
        sa.Column("id", sa.UUID(), nullable=False),
        sa.Column("label", sa.VARCHAR(length=256), nullable=False),
        sa.Column("block_number", sa.BigInteger(), nullable=True),
        sa.Column("address", sa.VARCHAR(length=256), nullable=True),
        sa.Column("transaction_hash", sa.VARCHAR(length=256), nullable=True),
        sa.Column("label_data", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
        sa.Column("block_timestamp", sa.BigInteger(), nullable=True),
        sa.Column("log_index", sa.Integer(), nullable=True),
        sa.Column(
            "created_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("TIMEZONE('utc', statement_timestamp())"),
            nullable=False,
        ),
        sa.PrimaryKeyConstraint("id", name=op.f("pk_arbitrum_sepolia_labels")),
        sa.UniqueConstraint("id", name=op.f("uq_arbitrum_sepolia_labels_id")),
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_labels_address"),
        "arbitrum_sepolia_labels",
        ["address"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_labels_block_number"),
        "arbitrum_sepolia_labels",
        ["block_number"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_labels_block_timestamp"),
        "arbitrum_sepolia_labels",
        ["block_timestamp"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_labels_label"),
        "arbitrum_sepolia_labels",
        ["label"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_labels_transaction_hash"),
        "arbitrum_sepolia_labels",
        ["transaction_hash"],
        unique=False,
    )
    op.create_table(
        "arbitrum_sepolia_transactions",
        sa.Column("hash", sa.VARCHAR(length=256), nullable=False),
        sa.Column("block_number", sa.BigInteger(), nullable=False),
        sa.Column("from_address", sa.VARCHAR(length=256), nullable=True),
        sa.Column("to_address", sa.VARCHAR(length=256), nullable=True),
        sa.Column("gas", sa.Numeric(precision=78, scale=0), nullable=True),
        sa.Column("gas_price", sa.Numeric(precision=78, scale=0), nullable=True),
        sa.Column("max_fee_per_gas", sa.Numeric(precision=78, scale=0), nullable=True),
        sa.Column(
            "max_priority_fee_per_gas", sa.Numeric(precision=78, scale=0), nullable=True
        ),
        sa.Column("input", sa.Text(), nullable=True),
        sa.Column("nonce", sa.VARCHAR(length=256), nullable=True),
        sa.Column("transaction_index", sa.BigInteger(), nullable=True),
        sa.Column("transaction_type", sa.Integer(), nullable=True),
        sa.Column("value", sa.Numeric(precision=78, scale=0), nullable=True),
        sa.Column(
            "indexed_at",
            sa.DateTime(timezone=True),
            server_default=sa.text("TIMEZONE('utc', statement_timestamp())"),
            nullable=False,
        ),
        sa.Column("y_parity", sa.BigInteger(), nullable=True),
        sa.ForeignKeyConstraint(
            ["block_number"],
            ["arbitrum_sepolia_blocks.block_number"],
            name=op.f(
                "fk_arbitrum_sepolia_transactions_block_number_arbitrum_sepolia_blocks"
            ),
            ondelete="CASCADE",
        ),
        sa.PrimaryKeyConstraint("hash", name=op.f("pk_arbitrum_sepolia_transactions")),
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_block_number"),
        "arbitrum_sepolia_transactions",
        ["block_number"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_from_address"),
        "arbitrum_sepolia_transactions",
        ["from_address"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_gas"),
        "arbitrum_sepolia_transactions",
        ["gas"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_gas_price"),
        "arbitrum_sepolia_transactions",
        ["gas_price"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_hash"),
        "arbitrum_sepolia_transactions",
        ["hash"],
        unique=True,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_to_address"),
        "arbitrum_sepolia_transactions",
        ["to_address"],
        unique=False,
    )
    op.create_index(
        op.f("ix_arbitrum_sepolia_transactions_value"),
        "arbitrum_sepolia_transactions",
        ["value"],
        unique=False,
    )
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###)
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_value"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_to_address"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_hash"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_gas_price"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_gas"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_from_address"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_transactions_block_number"),
        table_name="arbitrum_sepolia_transactions",
    )
    op.drop_table("arbitrum_sepolia_transactions")
    op.drop_index(
        op.f("ix_arbitrum_sepolia_labels_transaction_hash"),
        table_name="arbitrum_sepolia_labels",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_labels_label"), table_name="arbitrum_sepolia_labels"
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_labels_block_timestamp"),
        table_name="arbitrum_sepolia_labels",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_labels_block_number"),
        table_name="arbitrum_sepolia_labels",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_labels_address"), table_name="arbitrum_sepolia_labels"
    )
    op.drop_table("arbitrum_sepolia_labels")
    op.drop_index(
        op.f("ix_arbitrum_sepolia_blocks_timestamp"),
        table_name="arbitrum_sepolia_blocks",
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_blocks_hash"), table_name="arbitrum_sepolia_blocks"
    )
    op.drop_index(
        op.f("ix_arbitrum_sepolia_blocks_block_number"),
        table_name="arbitrum_sepolia_blocks",
    )
    op.drop_table("arbitrum_sepolia_blocks")
    # ### end Alembic commands ###