"""add magento internal api transport fields

Revision ID: 0055_add_magento_internal_api_transport
Revises: 0054_consolidate_shopify_registry_shop_code
Create Date: 2026-07-15 00:00:00.000000
"""

from alembic import op
import sqlalchemy as sa


revision = "0055_add_magento_internal_api_transport"
down_revision = "0054_consolidate_shopify_registry_shop_code"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("magento_connections", sa.Column("internal_api_base_url", sa.Text(), nullable=True))
    op.add_column("magento_connections", sa.Column("internal_host_header", sa.String(length=255), nullable=True))
    op.add_column(
        "magento_connections",
        sa.Column("prefer_internal_api", sa.Boolean(), nullable=False, server_default=sa.false()),
    )
    op.alter_column("magento_connections", "prefer_internal_api", server_default=None)


def downgrade() -> None:
    op.drop_column("magento_connections", "prefer_internal_api")
    op.drop_column("magento_connections", "internal_host_header")
    op.drop_column("magento_connections", "internal_api_base_url")