"""增加应用宝和平精英充值任务"""

from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = "20260812_0020"
down_revision: Union[str, None] = "20260808_0019"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    bind = op.get_bind()
    if sa.inspect(bind).has_table("yyb_recharge_tasks"):
        return
    op.create_table(
        "yyb_recharge_tasks",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("task_id", sa.String(64), nullable=False),
        sa.Column("created_by", sa.Integer(), sa.ForeignKey("users.id"), nullable=False),
        sa.Column("worker_job_id", sa.String(64), nullable=False),
        sa.Column("provider", sa.String(16), nullable=False, server_default=""),
        sa.Column("platform", sa.String(16), nullable=False, server_default="android"),
        sa.Column("points", sa.Integer(), nullable=True),
        sa.Column("product_id", sa.String(128), nullable=True, server_default=""),
        sa.Column("zone_id", sa.String(64), nullable=True, server_default=""),
        sa.Column("zone_name", sa.String(128), nullable=True, server_default=""),
        sa.Column("role_id", sa.String(64), nullable=True, server_default=""),
        sa.Column("role_name", sa.String(128), nullable=True, server_default=""),
        sa.Column("status", sa.String(32), nullable=False, server_default="created"),
        sa.Column("phase", sa.String(32), nullable=False, server_default="login"),
        sa.Column("message", sa.String(512), nullable=True, server_default=""),
        sa.Column("result", sa.JSON(), nullable=True),
        sa.Column("login_qr_data", sa.Text(), nullable=True),
        sa.Column("payment_qr_data", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(), nullable=True),
        sa.Column("finished_at", sa.DateTime(), nullable=True),
    )
    op.create_index("uq_yyb_recharge_tasks_task_id", "yyb_recharge_tasks", ["task_id"], unique=True)
    op.create_index("ix_yyb_recharge_tasks_created_by", "yyb_recharge_tasks", ["created_by"])
    op.create_index("ix_yyb_recharge_tasks_worker_job_id", "yyb_recharge_tasks", ["worker_job_id"])
    op.create_index("ix_yyb_recharge_tasks_status", "yyb_recharge_tasks", ["status"])


def downgrade() -> None:
    bind = op.get_bind()
    if sa.inspect(bind).has_table("yyb_recharge_tasks"):
        op.drop_table("yyb_recharge_tasks")