"""rename strategy_set columns and add nick

Revision ID: 20260403_0002
Revises: 20260403_0001
Create Date: 2026-04-03 00:30:00
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa


revision = "20260403_0002"
down_revision = "20260403_0001"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("strategy_sets", sa.Column("name", sa.Text(), nullable=True))
    op.add_column("strategy_sets", sa.Column("config", sa.Text(), nullable=True, server_default=""))
    op.add_column("strategy_sets", sa.Column("nick", sa.Text(), nullable=True))

    op.execute(
        sa.text(
            """
            UPDATE strategy_sets
            SET
                name = strategy_name,
                config = strategy_config,
                nick = strategy_name || ':' || substr(md5(strategy_config), 1, 12)
            """
        )
    )

    op.alter_column("strategy_sets", "name", nullable=False)
    op.alter_column("strategy_sets", "config", nullable=False)
    op.alter_column("strategy_sets", "nick", nullable=False)

    op.drop_constraint("uq_strategy_sets_name_config", "strategy_sets", type_="unique")
    op.create_unique_constraint("uq_strategy_sets_name_config", "strategy_sets", ["name", "config"])
    op.create_unique_constraint("uq_strategy_sets_nick", "strategy_sets", ["nick"])

    op.drop_column("strategy_sets", "strategy_name")
    op.drop_column("strategy_sets", "strategy_config")


def downgrade() -> None:
    op.add_column("strategy_sets", sa.Column("strategy_name", sa.Text(), nullable=True))
    op.add_column("strategy_sets", sa.Column("strategy_config", sa.Text(), nullable=True, server_default=""))

    op.execute(
        sa.text(
            """
            UPDATE strategy_sets
            SET
                strategy_name = name,
                strategy_config = config
            """
        )
    )

    op.alter_column("strategy_sets", "strategy_name", nullable=False)
    op.alter_column("strategy_sets", "strategy_config", nullable=False)

    op.drop_constraint("uq_strategy_sets_nick", "strategy_sets", type_="unique")
    op.drop_constraint("uq_strategy_sets_name_config", "strategy_sets", type_="unique")
    op.create_unique_constraint("uq_strategy_sets_name_config", "strategy_sets", ["strategy_name", "strategy_config"])

    op.drop_column("strategy_sets", "nick")
    op.drop_column("strategy_sets", "name")
    op.drop_column("strategy_sets", "config")