"""Add owner_raw column to cmdb_load_balancers and cmdb_cdn_domains.

Revision ID: 010
Revises: 009
Create Date: 2026-04-14
"""

from alembic import op
import sqlalchemy as sa

revision = "010"
down_revision = "009"
branch_labels = None
depends_on = None


def _column_exists(table: str, column: str) -> bool:
    conn = op.get_bind()
    result = conn.execute(
        sa.text(
            "SELECT COUNT(*) FROM information_schema.columns "
            "WHERE table_schema = DATABASE() AND table_name = :t AND column_name = :c"
        ),
        {"t": table, "c": column},
    )
    return result.scalar() > 0


def upgrade() -> None:
    if not _column_exists("cmdb_load_balancers", "owner_raw"):
        op.add_column(
            "cmdb_load_balancers",
            sa.Column("owner_raw", sa.String(255), nullable=True),
        )
    if not _column_exists("cmdb_cdn_domains", "owner_raw"):
        op.add_column(
            "cmdb_cdn_domains",
            sa.Column("owner_raw", sa.String(255), nullable=True),
        )


def downgrade() -> None:
    if _column_exists("cmdb_cdn_domains", "owner_raw"):
        op.drop_column("cmdb_cdn_domains", "owner_raw")
    if _column_exists("cmdb_load_balancers", "owner_raw"):
        op.drop_column("cmdb_load_balancers", "owner_raw")