"""Add Shopify product source snapshot table.

Revision ID: 0036_add_shopify_product_source_snapshot
Revises: 0035_add_channel_sku_prefix_mapping
Create Date: 2026-06-13 00:00:00.000000
"""

from __future__ import annotations

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

revision = "0036_add_shopify_product_source_snapshot"
down_revision = "0035_add_channel_sku_prefix_mapping"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "shopify_product_source_snapshot",
        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("load_id", sa.Integer(), sa.ForeignKey("ingest_run.id"), nullable=False),
        sa.Column("sku", sa.String(128), nullable=False),
        sa.Column("valid_from", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
        sa.Column("valid_to", sa.DateTime(timezone=True), nullable=True),
        sa.Column("shopify_product_id", sa.String(128), nullable=True),
        sa.Column("shopify_variant_id", sa.String(128), nullable=True),
        sa.Column("product_type", sa.String(128), nullable=True),
        sa.Column("status", sa.String(32), nullable=True),
        sa.Column("price", sa.Numeric(12, 4), nullable=True),
        sa.Column("payload", JSONB, nullable=False),
        sa.Column("column_names", JSONB, nullable=True),
        sa.Column("row_hash", sa.String(64), nullable=False),
    )
    op.create_index("ix_shopify_product_source_snapshot_sku", "shopify_product_source_snapshot", ["sku"])
    op.create_index(
        "ix_shopify_product_source_snapshot_conn_current",
        "shopify_product_source_snapshot",
        ["connection_id", "valid_to"],
    )


def downgrade() -> None:
    op.drop_index(
        "ix_shopify_product_source_snapshot_conn_current",
        table_name="shopify_product_source_snapshot",
    )
    op.drop_index("ix_shopify_product_source_snapshot_sku", table_name="shopify_product_source_snapshot")
    op.drop_table("shopify_product_source_snapshot")