"""add generation jobs

Revision ID: 0011_add_generation_jobs
Revises: 0010_add_story_audio_cache_path
Create Date: 2026-04-18

"""

import sqlalchemy as sa
from alembic import op


revision = "0011_add_generation_jobs"
down_revision = "0010_add_story_audio_cache_path"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "generation_jobs",
        sa.Column("id", sa.String(length=36), nullable=False),
        sa.Column("user_id", sa.String(length=255), nullable=False),
        sa.Column("story_id", sa.Integer(), nullable=True),
        sa.Column("output_mode", sa.String(length=32), nullable=False),
        sa.Column("input_type", sa.String(length=32), nullable=False),
        sa.Column("status", sa.String(length=32), nullable=False, server_default="running"),
        sa.Column(
            "current_step",
            sa.String(length=64),
            nullable=False,
            server_default="request_accepted",
        ),
        sa.Column("request_payload", sa.JSON(), nullable=False, server_default="{}"),
        sa.Column("result_snapshot", sa.JSON(), nullable=False, server_default="{}"),
        sa.Column("error_message", sa.Text(), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.ForeignKeyConstraint(["story_id"], ["stories.id"], ondelete="SET NULL"),
        sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_generation_jobs_user_id", "generation_jobs", ["user_id"])
    op.create_index("ix_generation_jobs_story_id", "generation_jobs", ["story_id"])
    op.create_index("ix_generation_jobs_status", "generation_jobs", ["status"])
    op.create_index("ix_generation_jobs_created_at", "generation_jobs", ["created_at"])

    op.create_table(
        "generation_job_events",
        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
        sa.Column("job_id", sa.String(length=36), nullable=False),
        sa.Column("story_id", sa.Integer(), nullable=True),
        sa.Column("event_type", sa.String(length=64), nullable=False),
        sa.Column("status", sa.String(length=32), nullable=False),
        sa.Column("message", sa.Text(), nullable=True),
        sa.Column("event_metadata", sa.JSON(), nullable=False, server_default="{}"),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
        sa.ForeignKeyConstraint(["job_id"], ["generation_jobs.id"], ondelete="CASCADE"),
        sa.ForeignKeyConstraint(["story_id"], ["stories.id"], ondelete="SET NULL"),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_generation_job_events_job_id", "generation_job_events", ["job_id"])
    op.create_index("ix_generation_job_events_story_id", "generation_job_events", ["story_id"])
    op.create_index("ix_generation_job_events_created_at", "generation_job_events", ["created_at"])


def downgrade() -> None:
    op.drop_index("ix_generation_job_events_created_at", table_name="generation_job_events")
    op.drop_index("ix_generation_job_events_story_id", table_name="generation_job_events")
    op.drop_index("ix_generation_job_events_job_id", table_name="generation_job_events")
    op.drop_table("generation_job_events")

    op.drop_index("ix_generation_jobs_created_at", table_name="generation_jobs")
    op.drop_index("ix_generation_jobs_status", table_name="generation_jobs")
    op.drop_index("ix_generation_jobs_story_id", table_name="generation_jobs")
    op.drop_index("ix_generation_jobs_user_id", table_name="generation_jobs")
    op.drop_table("generation_jobs")