"""Add Shopify collection registry and channel taxonomy mapping.

Revision ID: 0037_add_taxonomy_registry_and_mapping
Revises: 0036_add_shopify_product_source_snapshot
Create Date: 2026-06-13 12:00:00.000000
"""

from __future__ import annotations

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

revision = "0037_add_taxonomy_registry_and_mapping"
down_revision = "0036_add_shopify_product_source_snapshot"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "shopify_collection_registry",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("connection_id", sa.Integer(), sa.ForeignKey("shopify_connections.id"), nullable=True),
        sa.Column("shop_code", sa.String(255), nullable=False),
        sa.Column("collection_id", sa.String(255), nullable=False),
        sa.Column("handle", sa.String(255), nullable=True),
        sa.Column("title", sa.Text(), nullable=False),
        sa.Column("collection_type", sa.String(32), nullable=False, server_default="manual"),
        sa.Column("rules_json", JSONB, nullable=True),
        sa.Column("product_count", sa.Integer(), nullable=True),
        sa.Column("raw_json", JSONB, nullable=True),
        sa.Column("fetched_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    op.create_index("ix_shopify_collection_registry_shop_code", "shopify_collection_registry", ["shop_code"])
    op.create_index(
        "ix_shopify_collection_registry_conn_handle",
        "shopify_collection_registry",
        ["connection_id", "handle"],
    )

    op.create_table(
        "channel_taxonomy_mapping",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("channel_code", sa.String(32), nullable=False),
        sa.Column("connection_id", sa.Integer(), nullable=True),
        sa.Column("master_category_l1", sa.Text(), nullable=True),
        sa.Column("master_category_l2", sa.Text(), nullable=True),
        sa.Column("master_category_l3", sa.Text(), nullable=True),
        sa.Column("master_collection", sa.Text(), nullable=True),
        sa.Column("master_path_key", sa.String(512), nullable=False),
        sa.Column("channel_taxonomy_kind", sa.String(32), nullable=False, server_default="collection"),
        sa.Column("channel_remote_id", sa.String(255), nullable=True),
        sa.Column("channel_handle_or_path", sa.Text(), nullable=True),
        sa.Column("transform_rule", JSONB, nullable=True),
        sa.Column("is_active", sa.Boolean(), nullable=False, server_default="true"),
        sa.Column("notes", sa.Text(), nullable=True),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            nullable=False,
            server_default=sa.func.now(),
        ),
    )
    op.create_index(
        "ix_channel_taxonomy_mapping_lookup",
        "channel_taxonomy_mapping",
        ["channel_code", "connection_id", "master_path_key"],
    )


def downgrade() -> None:
    op.drop_index("ix_channel_taxonomy_mapping_lookup", table_name="channel_taxonomy_mapping")
    op.drop_table("channel_taxonomy_mapping")
    op.drop_index("ix_shopify_collection_registry_conn_handle", table_name="shopify_collection_registry")
    op.drop_index("ix_shopify_collection_registry_shop_code", table_name="shopify_collection_registry")
    op.drop_table("shopify_collection_registry")