"""add durable execution event revision

Revision ID: 0003_execution_events
Revises: 0002_sessions
"""

import sqlalchemy as sa
from alembic import op

revision = "0003_execution_events"
down_revision = "0002_sessions"
branch_labels = None
depends_on = None


def upgrade() -> None:
    with op.batch_alter_table("executions") as batch:
        batch.add_column(sa.Column("revision", sa.Integer(), nullable=False, server_default="0"))
    op.create_table(
        "execution_events",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column(
            "execution_id",
            sa.String(36),
            sa.ForeignKey("executions.id", ondelete="CASCADE"),
            nullable=False,
        ),
        sa.Column("revision", sa.Integer(), nullable=False),
        sa.Column("payload", sa.JSON(), nullable=False),
        sa.UniqueConstraint("execution_id", "revision", name="uq_execution_event_revision"),
    )


def downgrade() -> None:
    op.drop_table("execution_events")
    with op.batch_alter_table("executions") as batch:
        batch.drop_column("revision")