"""Add collection SKU prefix rules and manual include/exclude overrides.

Revision ID: 0044_add_collection_sku_mapping
Revises: 0043_add_collection_landing_page
Create Date: 2026-06-18 16:00:00.000000
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

revision = "0044_add_collection_sku_mapping"
down_revision = "0043_add_collection_landing_page"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "collection_landing_page",
        sa.Column("sku_prefixes", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
    )
    op.create_table(
        "collection_sku_override",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("path_slug", sa.String(512), nullable=False),
        sa.Column("master_sku", sa.String(128), nullable=False),
        sa.Column("action", sa.String(16), nullable=False),
        sa.Column("notes", sa.Text(), nullable=True),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.text("now()"),
        ),
        sa.UniqueConstraint("path_slug", "master_sku", name="uq_collection_sku_override_path_sku"),
    )
    op.create_index("ix_collection_sku_override_path_slug", "collection_sku_override", ["path_slug"])
    op.create_index("ix_collection_sku_override_master_sku", "collection_sku_override", ["master_sku"])


def downgrade() -> None:
    op.drop_index("ix_collection_sku_override_master_sku", table_name="collection_sku_override")
    op.drop_index("ix_collection_sku_override_path_slug", table_name="collection_sku_override")
    op.drop_table("collection_sku_override")
    op.drop_column("collection_landing_page", "sku_prefixes")