"""add pmflow launch draft jobs

Revision ID: 20260501_add_pmflow_launch_draft_jobs
Revises: 20260430_extend_prompt_eval_replay
Create Date: 2026-05-01
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa

try:
    from sqlalchemy.dialects.postgresql import JSONB
except Exception:  # pragma: no cover
    JSONB = sa.JSON


revision = "20260501_add_pmflow_launch_draft_jobs"
down_revision = "20260430_extend_prompt_eval_replay"
branch_labels = None
depends_on = None


def _json_type(bind):
    return JSONB() if bind.dialect.name == "postgresql" else sa.JSON()


def _table_exists(bind, table_name: str) -> bool:
    return sa.inspect(bind).has_table(table_name)


def upgrade() -> None:
    bind = op.get_bind()
    if _table_exists(bind, "pmflow_launch_draft_jobs"):
        return

    json_type = _json_type(bind)
    op.create_table(
        "pmflow_launch_draft_jobs",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("tenant_id", sa.BigInteger(), nullable=False),
        sa.Column("created_at", sa.TIMESTAMP(), server_default=sa.func.now(), nullable=False),
        sa.Column("updated_at", sa.TIMESTAMP(), server_default=sa.func.now(), nullable=False),
        sa.Column("user_id", sa.BigInteger(), sa.ForeignKey("users.id"), nullable=False),
        sa.Column("job_id", sa.String(length=64), nullable=False),
        sa.Column("revision_id", sa.String(length=64), nullable=False),
        sa.Column("parent_draft_id", sa.String(length=64), nullable=True),
        sa.Column("superseded_by_job_id", sa.String(length=64), nullable=True),
        sa.Column("result_draft_id", sa.String(length=64), nullable=True),
        sa.Column("input_text", sa.Text(), nullable=False),
        sa.Column("input_type", sa.String(length=32), nullable=False, server_default="text"),
        sa.Column("source", sa.String(length=32), nullable=False, server_default="web"),
        sa.Column("attachments", json_type, nullable=False),
        sa.Column("mode", sa.String(length=32), nullable=False, server_default="create"),
        sa.Column("revision_note", sa.Text(), nullable=True),
        sa.Column("status", sa.String(length=32), nullable=False, server_default="queued"),
        sa.Column("result_payload", json_type, nullable=True),
        sa.Column("changed_fields", json_type, nullable=False),
        sa.Column("error_code", sa.String(length=64), nullable=True),
        sa.Column("error_message", sa.Text(), nullable=True),
        sa.Column("started_at", sa.TIMESTAMP(), nullable=True),
        sa.Column("completed_at", sa.TIMESTAMP(), nullable=True),
        sa.Column("expires_at", sa.TIMESTAMP(), nullable=False),
        sa.UniqueConstraint("tenant_id", "job_id", name="uq_pmflow_launch_draft_jobs_tenant_job_id"),
        sa.UniqueConstraint("tenant_id", "revision_id", name="uq_pmflow_launch_draft_jobs_tenant_revision_id"),
    )
    op.create_index("ix_pmflow_launch_draft_jobs_tenant_id", "pmflow_launch_draft_jobs", ["tenant_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_user_id", "pmflow_launch_draft_jobs", ["user_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_job_id", "pmflow_launch_draft_jobs", ["job_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_revision_id", "pmflow_launch_draft_jobs", ["revision_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_parent_draft_id", "pmflow_launch_draft_jobs", ["parent_draft_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_superseded_by_job_id", "pmflow_launch_draft_jobs", ["superseded_by_job_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_result_draft_id", "pmflow_launch_draft_jobs", ["result_draft_id"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_status", "pmflow_launch_draft_jobs", ["status"], unique=False)
    op.create_index("ix_pmflow_launch_draft_jobs_expires_at", "pmflow_launch_draft_jobs", ["expires_at"], unique=False)


def downgrade() -> None:
    bind = op.get_bind()
    if not _table_exists(bind, "pmflow_launch_draft_jobs"):
        return
    for index_name in (
        "ix_pmflow_launch_draft_jobs_expires_at",
        "ix_pmflow_launch_draft_jobs_status",
        "ix_pmflow_launch_draft_jobs_result_draft_id",
        "ix_pmflow_launch_draft_jobs_superseded_by_job_id",
        "ix_pmflow_launch_draft_jobs_parent_draft_id",
        "ix_pmflow_launch_draft_jobs_revision_id",
        "ix_pmflow_launch_draft_jobs_job_id",
        "ix_pmflow_launch_draft_jobs_user_id",
        "ix_pmflow_launch_draft_jobs_tenant_id",
    ):
        op.drop_index(index_name, table_name="pmflow_launch_draft_jobs")
    op.drop_table("pmflow_launch_draft_jobs")