"""Add raw source product snapshot tables.

Revision ID: 0024_add_source_product_snapshots
Revises: 0023_add_magento_sync_pending
Create Date: 2026-06-01 00:00:00.000000
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import JSONB

revision = "0024_add_source_product_snapshots"
down_revision = "0023_add_magento_sync_pending"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "magento_product_source_snapshot",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("connection_id", sa.Integer(), sa.ForeignKey("magento_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("magento_product_id", sa.Integer(), nullable=True),
        sa.Column("product_type", sa.String(64), nullable=True),
        sa.Column("attribute_set_id", sa.Integer(), nullable=True),
        sa.Column("status", sa.String(32), nullable=True),
        sa.Column("visibility", sa.String(32), nullable=True),
        sa.Column("price", sa.Numeric(12, 4), nullable=True),
        sa.Column("qty", sa.Numeric(14, 4), nullable=True),
        sa.Column("is_in_stock", sa.Boolean(), 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_magento_product_source_snapshot_sku", "magento_product_source_snapshot", ["sku"])
    op.create_index(
        "ix_magento_product_source_snapshot_conn_current",
        "magento_product_source_snapshot",
        ["connection_id", "valid_to"],
    )

    op.create_table(
        "plytix_product_source_snapshot",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=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("product_websites", sa.Text(), nullable=True),
        sa.Column("product_type", sa.String(64), 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_plytix_product_source_snapshot_sku", "plytix_product_source_snapshot", ["sku"])
    op.create_index(
        "ix_plytix_product_source_snapshot_current",
        "plytix_product_source_snapshot",
        ["valid_to"],
    )

    op.create_table(
        "product_supplemental_attribute_value",
        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
        sa.Column("load_id", sa.Integer(), sa.ForeignKey("ingest_run.id"), nullable=False),
        sa.Column("source_label", sa.String(128), nullable=False),
        sa.Column("file_name", sa.String(255), nullable=True),
        sa.Column("row_num", sa.Integer(), nullable=False),
        sa.Column("sku", sa.String(128), nullable=False),
        sa.Column("column_name", sa.Text(), nullable=False),
        sa.Column("column_key", sa.String(255), nullable=False),
        sa.Column("value", sa.Text(), nullable=True),
        sa.Column("destination_hint", sa.String(32), nullable=True),
        sa.Column("imported_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    op.create_index(
        "ix_product_supplemental_attribute_value_sku",
        "product_supplemental_attribute_value",
        ["sku"],
    )
    op.create_index(
        "ix_product_supplemental_attribute_value_column_key",
        "product_supplemental_attribute_value",
        ["column_key"],
    )
    op.create_index(
        "ix_product_supplemental_attribute_value_source_sku",
        "product_supplemental_attribute_value",
        ["source_label", "sku"],
    )


def downgrade() -> None:
    op.drop_index(
        "ix_product_supplemental_attribute_value_source_sku",
        table_name="product_supplemental_attribute_value",
    )
    op.drop_index(
        "ix_product_supplemental_attribute_value_column_key",
        table_name="product_supplemental_attribute_value",
    )
    op.drop_index(
        "ix_product_supplemental_attribute_value_sku",
        table_name="product_supplemental_attribute_value",
    )
    op.drop_table("product_supplemental_attribute_value")
    op.drop_index("ix_plytix_product_source_snapshot_current", table_name="plytix_product_source_snapshot")
    op.drop_index("ix_plytix_product_source_snapshot_sku", table_name="plytix_product_source_snapshot")
    op.drop_table("plytix_product_source_snapshot")
    op.drop_index("ix_magento_product_source_snapshot_conn_current", table_name="magento_product_source_snapshot")
    op.drop_index("ix_magento_product_source_snapshot_sku", table_name="magento_product_source_snapshot")
    op.drop_table("magento_product_source_snapshot")