"""Create tables and columns for the acme issuer.

Revision ID: 3adfdd6598df
Revises: 556ceb3e3c3e
Create Date: 2018-04-10 13:25:47.007556

"""

# revision identifiers, used by Alembic.
revision = "3adfdd6598df"
down_revision = "556ceb3e3c3e"

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.postgresql import JSON
from sqlalchemy_utils import ArrowType

from lemur.utils import Vault


def upgrade():
    # create provider table
    print("Creating dns_providers table")
    op.create_table(
        "dns_providers",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("name", sa.String(length=256), nullable=True),
        sa.Column("description", sa.String(length=1024), nullable=True),
        sa.Column("provider_type", sa.String(length=256), nullable=True),
        sa.Column("credentials", Vault(), nullable=True),
        sa.Column("api_endpoint", sa.String(length=256), nullable=True),
        sa.Column(
            "date_created", ArrowType(), server_default=sa.text("now()"), nullable=False
        ),
        sa.Column("status", sa.String(length=128), nullable=True),
        sa.Column("options", JSON),
        sa.Column("domains", sa.JSON(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("name"),
    )
    print("Adding dns_provider_id column to certificates")
    op.add_column(
        "certificates", sa.Column("dns_provider_id", sa.Integer(), nullable=True)
    )
    print("Adding dns_provider_id column to pending_certs")
    op.add_column(
        "pending_certs", sa.Column("dns_provider_id", sa.Integer(), nullable=True)
    )
    print("Adding options column to pending_certs")
    op.add_column("pending_certs", sa.Column("options", JSON))

    print("Creating pending_dns_authorizations table")
    op.create_table(
        "pending_dns_authorizations",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("account_number", sa.String(length=128), nullable=True),
        sa.Column("domains", JSON, nullable=True),
        sa.Column("dns_provider_type", sa.String(length=128), nullable=True),
        sa.Column("options", JSON, nullable=True),
    )

    print("Creating certificates_dns_providers_fk foreign key")
    op.create_foreign_key(
        "certificates_dns_providers_fk",
        "certificates",
        "dns_providers",
        ["dns_provider_id"],
        ["id"],
        ondelete="cascade",
    )

    print("Altering column types in the api_keys table")
    op.alter_column("api_keys", "issued_at", existing_type=sa.BIGINT(), nullable=True)
    op.alter_column("api_keys", "revoked", existing_type=sa.BOOLEAN(), nullable=True)
    op.alter_column("api_keys", "ttl", existing_type=sa.BIGINT(), nullable=True)
    op.alter_column("api_keys", "user_id", existing_type=sa.INTEGER(), nullable=True)

    print("Creating dns_providers_id foreign key on pending_certs table")
    op.create_foreign_key(
        None,
        "pending_certs",
        "dns_providers",
        ["dns_provider_id"],
        ["id"],
        ondelete="CASCADE",
    )


def downgrade():
    print("Removing dns_providers_id foreign key on pending_certs table")
    op.drop_constraint(None, "pending_certs", type_="foreignkey")
    print("Reverting column types in the api_keys table")
    op.alter_column("api_keys", "user_id", existing_type=sa.INTEGER(), nullable=False)
    op.alter_column("api_keys", "ttl", existing_type=sa.BIGINT(), nullable=False)
    op.alter_column("api_keys", "revoked", existing_type=sa.BOOLEAN(), nullable=False)
    op.alter_column("api_keys", "issued_at", existing_type=sa.BIGINT(), nullable=False)
    print("Reverting certificates_dns_providers_fk foreign key")
    op.drop_constraint(
        "certificates_dns_providers_fk", "certificates", type_="foreignkey"
    )

    print("Dropping pending_dns_authorizations table")
    op.drop_table("pending_dns_authorizations")
    print("Undoing modifications to pending_certs table")
    op.drop_column("pending_certs", "options")
    op.drop_column("pending_certs", "dns_provider_id")
    print("Undoing modifications to certificates table")
    op.drop_column("certificates", "dns_provider_id")

    print("Deleting dns_providers table")
    op.drop_table("dns_providers")