"""Drop XP / tier related columns (+data) if they exist

Revision ID: 20250521_remove_xp_system
Revises: 05d6342e2105
Create Date: 2025-05-21
"""
from alembic import op
import sqlalchemy as sa

revision      = "20250521_remove_xp_system"
down_revision = "05d6342e2105"
branch_labels = None
depends_on    = None

# added "data" here ↓↓↓
XP_COLUMNS = (
    "videos_downloaded",
    "mb_usage",
    "level",
    "xp",
    "tier",
    "admin",
    "vip_badge",
    "bonus_active_until",
    "score",
    "data",               # ← drop the NOT-NULL JSON/profile blob
)

def upgrade() -> None:
    conn      = op.get_bind()
    existing  = {c["name"] for c in sa.inspect(conn).get_columns("users")}
    with op.batch_alter_table("users") as batch:
        for col in XP_COLUMNS:
            if col in existing:
                batch.drop_column(col)

def downgrade() -> None:
    conn      = op.get_bind()
    existing  = {c["name"] for c in sa.inspect(conn).get_columns("users")}
    with op.batch_alter_table("users") as batch:
        # Only re-add if missing; "data" comes back as JSON/Text and NULL-able
        add = lambda name, *args, **kw: (
            batch.add_column(sa.Column(name, *args, **kw))
            if name not in existing else None
        )
        add("videos_downloaded", sa.Integer(),  server_default="0", nullable=False)
        add("mb_usage",         sa.Float(),    server_default="0", nullable=False)
        add("level",            sa.Integer(),  server_default="1", nullable=False)
        add("xp",               sa.Integer(),  server_default="0", nullable=False)
        add("tier",             sa.Text(),     server_default="Online")
        add("admin",            sa.Boolean(),  server_default="false", nullable=False)
        add("vip_badge",        sa.Text())
        add("bonus_active_until", sa.DateTime(timezone=True))
        add("score",            sa.Float(),    server_default="0")
        add("data",             sa.JSON(),     server_default="{}")      # <<<