"""add recovery idempotency and import history columns

Revision ID: 20260910_recovery
Revises: 48094a7d1b92
"""
from alembic import op
import sqlalchemy as sa

revision = "20260910_recovery"
down_revision = "48094a7d1b92"
branch_labels = None
depends_on = None

def upgrade() -> None:
    op.add_column("submission_attempt", sa.Column("idempotency_key", sa.String(128), nullable=True))
    op.add_column("submission_attempt", sa.Column("catch_report_id", sa.Uuid(), nullable=True))
    op.create_foreign_key("fk_submission_attempt_report", "submission_attempt", "catch_report", ["catch_report_id"], ["id"])
    op.create_index("ix_submission_attempt_idempotency_key", "submission_attempt", ["idempotency_key"], unique=True)
    op.add_column("import_record_event", sa.Column("changes", sa.JSON(), nullable=True))
    op.add_column("import_record_event", sa.Column("provenance", sa.JSON(), nullable=True))

def downgrade() -> None:
    op.drop_column("import_record_event", "provenance")
    op.drop_column("import_record_event", "changes")
    op.drop_index("ix_submission_attempt_idempotency_key", table_name="submission_attempt")
    op.drop_constraint("fk_submission_attempt_report", "submission_attempt", type_="foreignkey")
    op.drop_column("submission_attempt", "catch_report_id")
    op.drop_column("submission_attempt", "idempotency_key")