"""add ai call trace fields

Revision ID: 20260429_add_ai_call_trace_fields
Revises: 20260428_add_saved_idea_enrichment
Create Date: 2026-04-29
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa

try:
    from sqlalchemy.dialects.postgresql import JSONB
except Exception:  # pragma: no cover
    JSONB = sa.JSON


revision = "20260429_add_ai_call_trace_fields"
down_revision = "20260428_add_saved_idea_enrichment"
branch_labels = None
depends_on = None


def _json_type(bind):
    return JSONB() if bind.dialect.name == "postgresql" else sa.JSON()


def _table_exists(bind, table_name: str) -> bool:
    return sa.inspect(bind).has_table(table_name)


def _column_exists(bind, table_name: str, column_name: str) -> bool:
    inspector = sa.inspect(bind)
    if not inspector.has_table(table_name):
        return False
    return any(str(item.get("name") or "") == column_name for item in inspector.get_columns(table_name))


def upgrade() -> None:
    bind = op.get_bind()
    table_name = "ai_call_logs"
    if not _table_exists(bind, table_name):
        return

    columns = [
        ("trace_id", sa.Column("trace_id", sa.String(length=80), nullable=True)),
        ("provider", sa.Column("provider", sa.String(length=64), nullable=True)),
        ("model", sa.Column("model", sa.String(length=120), nullable=True)),
        ("latency_ms", sa.Column("latency_ms", sa.Integer(), nullable=True)),
        ("routing_meta", sa.Column("routing_meta", _json_type(bind), nullable=True)),
    ]
    for column_name, column in columns:
        if not _column_exists(bind, table_name, column_name):
            op.add_column(table_name, column)

    existing_indexes = {index["name"] for index in sa.inspect(bind).get_indexes(table_name)}
    for index_name, column_names in (
        ("ix_ai_call_logs_trace_id", ["trace_id"]),
        ("ix_ai_call_logs_provider", ["provider"]),
        ("ix_ai_call_logs_model", ["model"]),
    ):
        if index_name not in existing_indexes:
            op.create_index(index_name, table_name, column_names, unique=False)


def downgrade() -> None:
    bind = op.get_bind()
    table_name = "ai_call_logs"
    if not _table_exists(bind, table_name):
        return

    existing_indexes = {index["name"] for index in sa.inspect(bind).get_indexes(table_name)}
    for index_name in ("ix_ai_call_logs_model", "ix_ai_call_logs_provider", "ix_ai_call_logs_trace_id"):
        if index_name in existing_indexes:
            op.drop_index(index_name, table_name=table_name)
    for column_name in ("routing_meta", "latency_ms", "model", "provider", "trace_id"):
        if _column_exists(bind, table_name, column_name):
            op.drop_column(table_name, column_name)