"""separate job orchestration from article content

Revision ID: 0006
Revises: 0005
Create Date: 2026-03-29

"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql

revision: str = "0006"
down_revision: Union[str, None] = "0005"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # Make article content fields nullable — they are now filled in step-by-step
    op.alter_column("translated_articles", "source_title", nullable=True)
    op.alter_column("translated_articles", "source_body", nullable=True)
    op.alter_column("translated_articles", "target_title", nullable=True)
    op.alter_column("translated_articles", "target_body", nullable=True)

    # Add source_body_pos to translated_articles
    op.add_column(
        "translated_articles",
        sa.Column("source_body_pos", postgresql.JSONB(), nullable=True),
    )

    # Add FK from jobs to the article they produce
    op.add_column(
        "jobs",
        sa.Column(
            "translated_article_id",
            postgresql.UUID(as_uuid=True),
            sa.ForeignKey("translated_articles.id"),
            nullable=True,
        ),
    )

    # Drop content columns that now live on translated_articles
    op.drop_column("jobs", "source_language")
    op.drop_column("jobs", "target_language")
    op.drop_column("jobs", "complexity_level")
    op.drop_column("jobs", "input_summary")
    op.drop_column("jobs", "generated_text")
    op.drop_column("jobs", "translated_text")
    op.drop_column("jobs", "audio_url")
    op.drop_column("jobs", "source_pos_data")
    op.drop_column("jobs", "target_pos_data")
    op.drop_column("jobs", "audio_transcript")


def downgrade() -> None:
    op.add_column("jobs", sa.Column("audio_transcript", postgresql.JSONB(), nullable=True))
    op.add_column("jobs", sa.Column("target_pos_data", postgresql.JSONB(), nullable=True))
    op.add_column("jobs", sa.Column("source_pos_data", postgresql.JSONB(), nullable=True))
    op.add_column("jobs", sa.Column("audio_url", sa.Text(), nullable=True))
    op.add_column("jobs", sa.Column("translated_text", sa.Text(), nullable=True))
    op.add_column("jobs", sa.Column("generated_text", sa.Text(), nullable=True))
    op.add_column("jobs", sa.Column("input_summary", sa.Text(), nullable=True))
    op.add_column("jobs", sa.Column("complexity_level", sa.String(5), nullable=False, server_default="B1"))
    op.add_column("jobs", sa.Column("target_language", sa.String(10), nullable=False, server_default="fr"))
    op.add_column("jobs", sa.Column("source_language", sa.String(10), nullable=False, server_default="en"))

    op.drop_column("jobs", "translated_article_id")

    op.alter_column("translated_articles", "target_body", nullable=False)
    op.alter_column("translated_articles", "target_title", nullable=False)
    op.alter_column("translated_articles", "source_body", nullable=False)
    op.alter_column("translated_articles", "source_title", nullable=False)