"""Adds support for rotation policies.

Creates a default rotation policy (30 days) with the name
'default' ensures that all existing certificates use the default
policy.

Revision ID: a02a678ddc25
Revises: 8ae67285ff14
Create Date: 2017-07-12 11:45:49.257927
"""

# revision identifiers, used by Alembic.
revision = "a02a678ddc25"
down_revision = "8ae67285ff14"

from alembic import op
import sqlalchemy as sa
from sqlalchemy.sql import text


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        "rotation_policies",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("name", sa.String(), nullable=True),
        sa.Column("days", sa.Integer(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
    )
    op.add_column(
        "certificates", sa.Column("rotation_policy_id", sa.Integer(), nullable=True)
    )
    op.create_foreign_key(
        None, "certificates", "rotation_policies", ["rotation_policy_id"], ["id"]
    )

    conn = op.get_bind()
    stmt = text("insert into rotation_policies (days, name) values (:days, :name)")
    stmt = stmt.bindparams(days=30, name="default")
    conn.execute(stmt)

    stmt = text("select id from rotation_policies where name=:name")
    stmt = stmt.bindparams(name="default")
    rotation_policy_id = conn.execute(stmt).fetchone()[0]

    stmt = text("update certificates set rotation_policy_id=:rotation_policy_id")
    stmt = stmt.bindparams(rotation_policy_id=rotation_policy_id)
    conn.execute(stmt)
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint(None, "certificates", type_="foreignkey")
    op.drop_column("certificates", "rotation_policy_id")
    op.drop_index(
        "certificate_replacement_associations_ix",
        table_name="certificate_replacement_associations",
    )
    op.create_index(
        "certificate_replacement_associations_ix",
        "certificate_replacement_associations",
        ["replaced_certificate_id", "certificate_id"],
        unique=True,
    )
    op.drop_table("rotation_policies")
    # ### end Alembic commands ###