"""scheme version pricing snapshot

Revision ID: 20260319_14
Revises: 20260319_13
Create Date: 2026-03-19 13:40:00
"""

from alembic import op
import sqlalchemy as sa


revision = "20260319_14"
down_revision = "20260319_13"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "scheme_version_pricing_categories",
        sa.Column("id", sa.BigInteger(), primary_key=True, autoincrement=True),
        sa.Column("snapshot_category_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_version_id", sa.String(length=32), nullable=False),
        sa.Column("source_pricing_category_id", sa.String(length=32), nullable=True),
        sa.Column("name", sa.String(length=255), nullable=False),
        sa.Column("code", sa.String(length=128), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
    )
    op.create_index(
        "ix_scheme_version_pricing_categories_snapshot_category_id",
        "scheme_version_pricing_categories",
        ["snapshot_category_id"],
        unique=True,
    )
    op.create_index(
        "ix_scheme_version_pricing_categories_scheme_version_id",
        "scheme_version_pricing_categories",
        ["scheme_version_id"],
        unique=False,
    )

    op.create_table(
        "scheme_version_price_rules",
        sa.Column("id", sa.BigInteger(), primary_key=True, autoincrement=True),
        sa.Column("snapshot_price_rule_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_version_id", sa.String(length=32), nullable=False),
        sa.Column("source_price_rule_id", sa.String(length=32), nullable=True),
        sa.Column("snapshot_category_id", sa.String(length=32), nullable=True),
        sa.Column("target_type", sa.String(length=32), nullable=False),
        sa.Column("target_ref", sa.String(length=128), nullable=False),
        sa.Column("amount", sa.Numeric(12, 2), nullable=False),
        sa.Column("currency", sa.String(length=8), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
    )
    op.create_index(
        "ix_scheme_version_price_rules_snapshot_price_rule_id",
        "scheme_version_price_rules",
        ["snapshot_price_rule_id"],
        unique=True,
    )
    op.create_index(
        "ix_scheme_version_price_rules_scheme_version_id",
        "scheme_version_price_rules",
        ["scheme_version_id"],
        unique=False,
    )
    op.create_index(
        "ix_scheme_version_price_rules_lookup",
        "scheme_version_price_rules",
        ["scheme_version_id", "target_type", "target_ref"],
        unique=False,
    )


def downgrade() -> None:
    op.drop_index("ix_scheme_version_price_rules_lookup", table_name="scheme_version_price_rules")
    op.drop_index("ix_scheme_version_price_rules_scheme_version_id", table_name="scheme_version_price_rules")
    op.drop_index("ix_scheme_version_price_rules_snapshot_price_rule_id", table_name="scheme_version_price_rules")
    op.drop_table("scheme_version_price_rules")

    op.drop_index("ix_scheme_version_pricing_categories_scheme_version_id", table_name="scheme_version_pricing_categories")
    op.drop_index("ix_scheme_version_pricing_categories_snapshot_category_id", table_name="scheme_version_pricing_categories")
    op.drop_table("scheme_version_pricing_categories")