"""Add upload task metadata."""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "20260802_0002"
down_revision: str | Sequence[str] | None = "20260802_0001"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    with op.batch_alter_table("upload_tasks") as batch:
        batch.add_column(sa.Column("project_id", sa.String(36), nullable=True))
        batch.add_column(sa.Column("description", sa.Text(), nullable=True))
        batch.add_column(sa.Column("version", sa.String(100), nullable=True))
        batch.add_column(sa.Column("purpose", sa.String(200), nullable=True))
        batch.add_column(sa.Column("tags_json", sa.Text(), nullable=False, server_default="[]"))
        batch.add_column(
            sa.Column("overwrite_allowed", sa.Boolean(), nullable=False, server_default=sa.false())
        )
        batch.create_foreign_key(
            "fk_upload_tasks_project_id", "projects", ["project_id"], ["id"], ondelete="SET NULL"
        )
        batch.create_index("ix_upload_tasks_project_id", ["project_id"])


def downgrade() -> None:
    with op.batch_alter_table("upload_tasks") as batch:
        batch.drop_index("ix_upload_tasks_project_id")
        batch.drop_constraint("fk_upload_tasks_project_id", type_="foreignkey")
        for column in (
            "overwrite_allowed",
            "tags_json",
            "purpose",
            "version",
            "description",
            "project_id",
        ):
            batch.drop_column(column)