"""Хранение назначения строк ОСВ правилам по версии консолидации."""

import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

from alembic import op
from src.models.base import SCHEMA

revision = "0011_vgo_assignment"
down_revision = "0010_dbml_v5_23"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "vgo_assignment",
        sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True),
        sa.Column("consolidation_version_id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column(
            "statement_type",
            postgresql.ENUM("balance", "income", name="statement_type", schema=SCHEMA, create_type=False),
            nullable=False,
        ),
        sa.Column("osv_entry_id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("rule_id", postgresql.UUID(as_uuid=True)),
        sa.Column("action", sa.String(), nullable=False),
        sa.Column("is_exclude", sa.Boolean(), server_default=sa.text("false"), nullable=False),
        sa.Column("created_by", postgresql.UUID(as_uuid=True)),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.Column("updated_by", postgresql.UUID(as_uuid=True)),
        sa.Column("updated_at", sa.DateTime(timezone=True)),
        sa.ForeignKeyConstraint(
            ["consolidation_version_id"], [f"{SCHEMA}.consolidation_version.id"], name="fk_vgo_assignment_version"
        ),
        sa.ForeignKeyConstraint(["osv_entry_id"], [f"{SCHEMA}.osv_entry.id"], name="fk_vgo_assignment_entry"),
        sa.ForeignKeyConstraint(["rule_id"], [f"{SCHEMA}.elimination_rule.id"], name="fk_vgo_assignment_rule"),
        sa.ForeignKeyConstraint(["created_by"], [f"{SCHEMA}.app_user.id"], name="fk_vgo_assignment_created_by"),
        sa.ForeignKeyConstraint(["updated_by"], [f"{SCHEMA}.app_user.id"], name="fk_vgo_assignment_updated_by"),
        sa.CheckConstraint("action IN ('ASSIGN', 'UNASSIGNED')", name="ck_vgo_assignment_action_values"),
        sa.CheckConstraint(
            "(action = 'ASSIGN' AND rule_id IS NOT NULL) OR (action = 'UNASSIGNED' AND rule_id IS NULL)",
            name="ck_vgo_assignment_action_rule_consistency",
        ),
        schema=SCHEMA,
    )
    op.create_index(
        "uq_vgo_assignment_version_type_entry",
        "vgo_assignment",
        ["consolidation_version_id", "statement_type", "osv_entry_id"],
        unique=True,
        schema=SCHEMA,
    )
    op.create_index("ix_vgo_assignment_osv_entry_id", "vgo_assignment", ["osv_entry_id"], schema=SCHEMA)


def downgrade() -> None:
    op.drop_index("ix_vgo_assignment_osv_entry_id", table_name="vgo_assignment", schema=SCHEMA)
    op.drop_index("uq_vgo_assignment_version_type_entry", table_name="vgo_assignment", schema=SCHEMA)
    op.drop_table("vgo_assignment", schema=SCHEMA)