"""add trigger comment body to review jobs

Revision ID: 0002_trigger_comment_body
Revises: 0001_initial
Create Date: 2026-05-22 20:15:00
"""

from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op


revision: str = "0002_trigger_comment_body"
down_revision: Union[str, None] = "0001_initial"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column("review_jobs", sa.Column("trigger_comment_body", sa.Text(), nullable=True))


def downgrade() -> None:
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    table_names = inspector.get_table_names()
    if "review_jobs" not in table_names:
        return

    column_names = {column["name"] for column in inspector.get_columns("review_jobs")}
    if "trigger_comment_body" in column_names:
        op.drop_column("review_jobs", "trigger_comment_body")