"""function_library — current_version_id points at the active published snapshot.

Revision ID: 052
Revises: 051
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import UUID as PGUUID

revision = "052"
down_revision = "051"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "function_library",
        sa.Column(
            "current_version_id",
            PGUUID(as_uuid=True),
            sa.ForeignKey("function_library_versions.id", ondelete="SET NULL"),
            nullable=True,
        ),
    )
    op.create_index(
        "ix_function_library_current_version_id",
        "function_library",
        ["current_version_id"],
    )


def downgrade() -> None:
    op.drop_index("ix_function_library_current_version_id", table_name="function_library")
    op.drop_column("function_library", "current_version_id")