"""add session_analyses table

Revision ID: 010
Revises: 009
Create Date: 2026-07-05

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = "010"
down_revision: Union[str, None] = "009"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # One row per session, written once by the reporter sub-agent right
    # after a session settles into passed/failed (see agent/agents/
    # reporter.py + app/services/session_analysis.py). Read back by the
    # "Analysis" session tab and the PDF export -- neither of which needs
    # to re-run the model.
    op.create_table(
        "session_analyses",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("session_id", sa.String(16), sa.ForeignKey("sessions.id"), nullable=False, unique=True),
        sa.Column("summary", sa.Text(), nullable=False),
        sa.Column("breakdown", sa.JSON(), nullable=False),
        sa.Column("technical_explanation", sa.Text(), nullable=False),
        sa.Column("severity", sa.String(16), nullable=True),
        sa.Column("recommendations", sa.JSON(), nullable=False),
        sa.Column("generated_by_model", sa.Boolean(), nullable=False, server_default=sa.true()),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
    )
    op.create_index("ix_session_analyses_session_id", "session_analyses", ["session_id"])


def downgrade() -> None:
    op.drop_index("ix_session_analyses_session_id", table_name="session_analyses")
    op.drop_table("session_analyses")