"""Persist immutable installed composition contract snapshots."""

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


revision = "186"
down_revision = "185"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "installed_composition_snapshots",
        sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column("project_id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("recipe_id", sa.String(), nullable=True),
        sa.Column("recipe_version", sa.Integer(), nullable=True),
        sa.Column("starter_id", sa.String(), nullable=True),
        sa.Column("starter_version", sa.String(), nullable=True),
        sa.Column("feature_packs", postgresql.JSONB(), nullable=False, server_default="[]"),
        sa.Column("app_blocks", postgresql.JSONB(), nullable=False, server_default="[]"),
        sa.Column("capability_contract_versions", postgresql.JSONB(), nullable=False, server_default="{}"),
        sa.Column("auth_contract", postgresql.JSONB(), nullable=True),
        sa.Column("auth_contract_version", sa.String(), nullable=True),
        sa.Column("runtime_contract_version", sa.String(), nullable=True),
        sa.Column("reaction_contract_version", sa.String(), nullable=True),
        sa.Column("generation_plan_hash", sa.String(), nullable=True),
        sa.Column("generated_artifact_version", sa.String(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_index("ix_installed_composition_snapshots_project_id", "installed_composition_snapshots", ["project_id"])


def downgrade() -> None:
    op.drop_index("ix_installed_composition_snapshots_project_id", table_name="installed_composition_snapshots")
    op.drop_table("installed_composition_snapshots")