"""add translated_articles table

Revision ID: 0005
Revises: 0004
Create Date: 2026-03-27

"""
from typing import Sequence, Union

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

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


def upgrade() -> None:
    op.create_table(
        "translated_articles",
        sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column("published_at", sa.DateTime(timezone=True), nullable=False),
        sa.Column("source_language", sa.String(10), nullable=False),
        sa.Column("source_title", sa.Text(), nullable=False),
        sa.Column("source_body", sa.Text(), nullable=False),
        sa.Column("target_language", sa.String(10), nullable=False),
        sa.Column("target_complexities", postgresql.ARRAY(sa.String(5)), nullable=False),
        sa.Column("target_title", sa.Text(), nullable=False),
        sa.Column("target_body", sa.Text(), nullable=False),
        sa.Column("audio_url", sa.Text(), nullable=True),
        sa.Column("target_body_pos", postgresql.JSONB(), nullable=True),
        sa.Column("target_body_transcript", postgresql.JSONB(), nullable=True),
    )
    op.create_index("ix_translated_articles_published_at", "translated_articles", ["published_at"])
    op.create_index("ix_translated_articles_target_language", "translated_articles", ["target_language"])


def downgrade() -> None:
    op.drop_index("ix_translated_articles_target_language", table_name="translated_articles")
    op.drop_index("ix_translated_articles_published_at", table_name="translated_articles")
    op.drop_table("translated_articles")