"""add ai generation runs and item variant lifecycle fields

Revision ID: 20260404_000003
Revises: 20260402_000002
Create Date: 2026-04-04 10:10:00
"""

from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = "20260404_000003"
down_revision: Union[str, None] = "20260402_000002"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "ai_generation_runs",
        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
        sa.Column("basis_item_id", sa.Integer(), nullable=False),
        sa.Column("source_snapshot_question_id", sa.Integer(), nullable=True),
        sa.Column("target_level", sa.String(length=50), nullable=False),
        sa.Column("requested_count", sa.Integer(), nullable=False, server_default="1"),
        sa.Column("model", sa.String(length=255), nullable=False),
        sa.Column("prompt_version", sa.String(length=50), nullable=False, server_default="v1"),
        sa.Column("operator_notes", sa.Text(), nullable=True),
        sa.Column("created_by", sa.String(length=255), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.ForeignKeyConstraint(["basis_item_id"], ["items.id"], ondelete="CASCADE", onupdate="CASCADE"),
        sa.ForeignKeyConstraint(
            ["source_snapshot_question_id"],
            ["tryout_snapshot_questions.id"],
            ondelete="SET NULL",
            onupdate="CASCADE",
        ),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_ai_generation_runs_basis_item_id", "ai_generation_runs", ["basis_item_id"], unique=False)
    op.create_index(
        "ix_ai_generation_runs_source_snapshot_question_id",
        "ai_generation_runs",
        ["source_snapshot_question_id"],
        unique=False,
    )

    op.add_column("items", sa.Column("generation_run_id", sa.Integer(), nullable=True))
    op.add_column("items", sa.Column("source_snapshot_question_id", sa.Integer(), nullable=True))
    op.add_column("items", sa.Column("variant_status", sa.String(length=50), nullable=False, server_default="active"))
    op.add_column("items", sa.Column("reviewed_by", sa.String(length=255), nullable=True))
    op.add_column("items", sa.Column("reviewed_at", sa.DateTime(timezone=True), nullable=True))
    op.add_column("items", sa.Column("review_notes", sa.Text(), nullable=True))

    op.create_foreign_key(
        "fk_items_generation_run_id",
        "items",
        "ai_generation_runs",
        ["generation_run_id"],
        ["id"],
        ondelete="SET NULL",
        onupdate="CASCADE",
    )
    op.create_foreign_key(
        "fk_items_source_snapshot_question_id",
        "items",
        "tryout_snapshot_questions",
        ["source_snapshot_question_id"],
        ["id"],
        ondelete="SET NULL",
        onupdate="CASCADE",
    )
    op.create_index("ix_items_generation_run_id", "items", ["generation_run_id"], unique=False)
    op.create_index(
        "ix_items_source_snapshot_question_id",
        "items",
        ["source_snapshot_question_id"],
        unique=False,
    )
    op.create_index("ix_items_variant_status", "items", ["variant_status"], unique=False)

    op.drop_index("ix_items_tryout_id_website_id_slot", table_name="items")
    op.create_index(
        "ix_items_tryout_id_website_id_slot",
        "items",
        ["tryout_id", "website_id", "slot", "level"],
        unique=False,
    )

    op.alter_column("items", "variant_status", server_default=None)


def downgrade() -> None:
    op.drop_index("ix_items_tryout_id_website_id_slot", table_name="items")
    op.create_index(
        "ix_items_tryout_id_website_id_slot",
        "items",
        ["tryout_id", "website_id", "slot", "level"],
        unique=True,
    )

    op.drop_index("ix_items_variant_status", table_name="items")
    op.drop_index("ix_items_source_snapshot_question_id", table_name="items")
    op.drop_index("ix_items_generation_run_id", table_name="items")
    op.drop_constraint("fk_items_source_snapshot_question_id", "items", type_="foreignkey")
    op.drop_constraint("fk_items_generation_run_id", "items", type_="foreignkey")
    op.drop_column("items", "review_notes")
    op.drop_column("items", "reviewed_at")
    op.drop_column("items", "reviewed_by")
    op.drop_column("items", "variant_status")
    op.drop_column("items", "source_snapshot_question_id")
    op.drop_column("items", "generation_run_id")

    op.drop_index("ix_ai_generation_runs_source_snapshot_question_id", table_name="ai_generation_runs")
    op.drop_index("ix_ai_generation_runs_basis_item_id", table_name="ai_generation_runs")
    op.drop_table("ai_generation_runs")