"""Add persisted fixture sources and patch instances."""

from __future__ import annotations

import sqlalchemy as sa
from alembic import op

revision = "20260719_0002"
down_revision = "20260712_0001"
branch_labels = None
depends_on = None


def upgrade() -> None:
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    tables = set(inspector.get_table_names())

    if "fixture_sources" not in tables:
        op.create_table(
            "fixture_sources",
            sa.Column("id", sa.Integer(), primary_key=True),
            sa.Column("manufacturer_key", sa.String(length=100), nullable=False),
            sa.Column("fixture_key", sa.String(length=150), nullable=False),
            sa.Column("schema_ref", sa.String(length=255), nullable=False),
            sa.Column("source_url", sa.String(length=255), nullable=False),
            sa.Column("payload", sa.JSON(), nullable=False),
            sa.Column("payload_hash", sa.String(length=64), nullable=False),
            sa.Column("imported_at", sa.DateTime(timezone=True), nullable=False),
        )

    if "fixture_instances" not in tables:
        op.create_table(
            "fixture_instances",
            sa.Column("id", sa.Integer(), primary_key=True),
            sa.Column("universe", sa.Integer(), nullable=False, server_default="1"),
            sa.Column("name", sa.String(length=160), nullable=False),
            sa.Column("definition_id", sa.Integer(), nullable=False),
            sa.Column("mode_key", sa.String(length=120), nullable=False),
            sa.Column("start_address", sa.Integer(), nullable=False),
            sa.Column("channel_count", sa.Integer(), nullable=False),
            sa.Column("enabled", sa.Boolean(), nullable=False, server_default=sa.true()),
            sa.Column("group_names", sa.JSON(), nullable=False),
            sa.Column("position", sa.JSON(), nullable=False),
            sa.Column("tags", sa.JSON(), nullable=False),
            sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
        )
        op.create_index("ix_fixture_instances_definition_id", "fixture_instances", ["definition_id"], unique=False)
    else:
        columns = {column["name"] for column in inspector.get_columns("fixture_instances")}
        if "universe" not in columns:
            with op.batch_alter_table("fixture_instances") as batch_op:
                batch_op.add_column(sa.Column("universe", sa.Integer(), nullable=False, server_default="1"))


def downgrade() -> None:
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    tables = set(inspector.get_table_names())

    if "fixture_instances" in tables:
        columns = {column["name"] for column in inspector.get_columns("fixture_instances")}
        if "universe" in columns:
            with op.batch_alter_table("fixture_instances") as batch_op:
                batch_op.drop_column("universe")

    if "fixture_sources" in tables:
        op.drop_table("fixture_sources")