"""initial schema

Revision ID: 20260326_0001
Revises:
Create Date: 2026-03-26 23:59:00.000000
"""

from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa


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


def upgrade() -> None:
    """Create the initial application tables."""

    op.create_table(
        "orders",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("customer_level", sa.Enum("low", "mid", name="customerlevel", native_enum=False), nullable=False),
        sa.Column("service_mode", sa.Enum("auto_basic", "semi_pro", name="servicemode", native_enum=False), nullable=False),
        sa.Column(
            "status",
            sa.Enum(
                "created",
                "running",
                "waiting_review",
                "succeeded",
                "failed",
                "cancelled",
                name="orderstatus",
                native_enum=False,
            ),
            nullable=False,
        ),
        sa.Column("model_id", sa.Integer(), nullable=False),
        sa.Column("pose_id", sa.Integer(), nullable=False),
        sa.Column("garment_asset_id", sa.Integer(), nullable=False),
        sa.Column("scene_ref_asset_id", sa.Integer(), nullable=False),
        sa.Column("final_asset_id", sa.Integer(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
    )

    op.create_table(
        "assets",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
        sa.Column(
            "asset_type",
            sa.Enum(
                "prepared_model",
                "tryon",
                "scene",
                "texture",
                "face",
                "fusion",
                "qc_candidate",
                "final",
                name="assettype",
                native_enum=False,
            ),
            nullable=False,
        ),
        sa.Column(
            "step_name",
            sa.Enum(
                "prepare_model",
                "tryon",
                "scene",
                "texture",
                "face",
                "fusion",
                "qc",
                "export",
                "review",
                name="workflowstepname",
                native_enum=False,
            ),
            nullable=True,
        ),
        sa.Column("uri", sa.String(length=500), nullable=False),
        sa.Column("metadata_json", sa.JSON(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
    )
    op.create_index("ix_assets_order_id", "assets", ["order_id"])

    op.create_table(
        "review_tasks",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
        sa.Column(
            "status",
            sa.Enum("pending", "submitted", name="reviewtaskstatus", native_enum=False),
            nullable=False,
        ),
        sa.Column(
            "decision",
            sa.Enum(
                "approve",
                "rerun_scene",
                "rerun_face",
                "rerun_fusion",
                "reject",
                name="reviewdecision",
                native_enum=False,
            ),
            nullable=True,
        ),
        sa.Column("reviewer_id", sa.Integer(), nullable=True),
        sa.Column("selected_asset_id", sa.Integer(), nullable=True),
        sa.Column("comment", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
    )
    op.create_index("ix_review_tasks_order_id", "review_tasks", ["order_id"])

    op.create_table(
        "workflow_runs",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("order_id", sa.Integer(), sa.ForeignKey("orders.id"), nullable=False),
        sa.Column("workflow_id", sa.String(length=255), nullable=False),
        sa.Column("workflow_type", sa.String(length=255), nullable=False),
        sa.Column(
            "status",
            sa.Enum(
                "created",
                "running",
                "waiting_review",
                "succeeded",
                "failed",
                "cancelled",
                name="orderstatus",
                native_enum=False,
            ),
            nullable=False,
        ),
        sa.Column(
            "current_step",
            sa.Enum(
                "prepare_model",
                "tryon",
                "scene",
                "texture",
                "face",
                "fusion",
                "qc",
                "export",
                "review",
                name="workflowstepname",
                native_enum=False,
            ),
            nullable=True,
        ),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
        sa.UniqueConstraint("workflow_id"),
    )
    op.create_index("ix_workflow_runs_order_id", "workflow_runs", ["order_id"])

    op.create_table(
        "workflow_steps",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("workflow_run_id", sa.Integer(), sa.ForeignKey("workflow_runs.id"), nullable=False),
        sa.Column(
            "step_name",
            sa.Enum(
                "prepare_model",
                "tryon",
                "scene",
                "texture",
                "face",
                "fusion",
                "qc",
                "export",
                "review",
                name="workflowstepname",
                native_enum=False,
            ),
            nullable=False,
        ),
        sa.Column(
            "step_status",
            sa.Enum("pending", "running", "waiting", "succeeded", "failed", name="stepstatus", native_enum=False),
            nullable=False,
        ),
        sa.Column("input_json", sa.JSON(), nullable=True),
        sa.Column("output_json", sa.JSON(), nullable=True),
        sa.Column("error_message", sa.Text(), nullable=True),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("ended_at", sa.DateTime(timezone=True), nullable=True),
    )
    op.create_index("ix_workflow_steps_workflow_run_id", "workflow_steps", ["workflow_run_id"])


def downgrade() -> None:
    """Drop the application tables."""

    op.drop_index("ix_workflow_steps_workflow_run_id", table_name="workflow_steps")
    op.drop_table("workflow_steps")
    op.drop_index("ix_workflow_runs_order_id", table_name="workflow_runs")
    op.drop_table("workflow_runs")
    op.drop_index("ix_review_tasks_order_id", table_name="review_tasks")
    op.drop_table("review_tasks")
    op.drop_index("ix_assets_order_id", table_name="assets")
    op.drop_table("assets")
    op.drop_table("orders")