"""Persist compact activity analysis and route previews."""

import sqlalchemy as sa
from alembic import op

revision = "f08db2902a74"
down_revision = "e57a0f824c61"
branch_labels = None
depends_on = None


def upgrade():
    with op.batch_alter_table("activity") as batch_op:
        batch_op.add_column(
            sa.Column("analysis_version", sa.Integer(), nullable=False, server_default="0")
        )
        batch_op.add_column(
            sa.Column("analysis_summary", sa.JSON(), nullable=False, server_default="{}")
        )
        batch_op.add_column(
            sa.Column("display_data", sa.JSON(), nullable=False, server_default="{}")
        )
        batch_op.add_column(
            sa.Column("route_geometry", sa.JSON(), nullable=False, server_default="[]")
        )


def downgrade():
    with op.batch_alter_table("activity") as batch_op:
        for name in ("route_geometry", "display_data", "analysis_summary", "analysis_version"):
            batch_op.drop_column(name)