"""create pricing tables

Revision ID: 20260316_09
Revises: 20260316_08
Create Date: 2026-03-16 16:40:00
"""

from alembic import op
import sqlalchemy as sa


revision = "20260316_09"
down_revision = "20260316_08"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "pricing_categories",
        sa.Column("id", sa.BigInteger(), primary_key=True, autoincrement=True),
        sa.Column("pricing_category_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_id", sa.String(length=32), nullable=False),
        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), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["scheme_id"], ["schemes.scheme_id"], ondelete="RESTRICT"),
    )
    op.create_index("ix_pricing_categories_pricing_category_id", "pricing_categories", ["pricing_category_id"], unique=True)
    op.create_index("ix_pricing_categories_scheme_id", "pricing_categories", ["scheme_id"], unique=False)

    op.create_table(
        "price_rules",
        sa.Column("id", sa.BigInteger(), primary_key=True, autoincrement=True),
        sa.Column("price_rule_id", sa.String(length=32), nullable=False),
        sa.Column("scheme_id", sa.String(length=32), nullable=False),
        sa.Column("pricing_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=255), nullable=False),
        sa.Column("amount", sa.Numeric(12, 2), nullable=False),
        sa.Column("currency", sa.String(length=3), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["scheme_id"], ["schemes.scheme_id"], ondelete="RESTRICT"),
        sa.ForeignKeyConstraint(["pricing_category_id"], ["pricing_categories.pricing_category_id"], ondelete="SET NULL"),
    )
    op.create_index("ix_price_rules_price_rule_id", "price_rules", ["price_rule_id"], unique=True)
    op.create_index("ix_price_rules_scheme_id", "price_rules", ["scheme_id"], unique=False)
    op.create_index("ix_price_rules_pricing_category_id", "price_rules", ["pricing_category_id"], unique=False)


def downgrade() -> None:
    op.drop_index("ix_price_rules_pricing_category_id", table_name="price_rules")
    op.drop_index("ix_price_rules_scheme_id", table_name="price_rules")
    op.drop_index("ix_price_rules_price_rule_id", table_name="price_rules")
    op.drop_table("price_rules")

    op.drop_index("ix_pricing_categories_scheme_id", table_name="pricing_categories")
    op.drop_index("ix_pricing_categories_pricing_category_id", table_name="pricing_categories")
    op.drop_table("pricing_categories")