"""Phase 21A Opening Balance Comparison + Controlled Tally Correction.

Revision ID: 20260825_opening_balance_p21a
Revises: 20260824_stock_item_intel_p21
"""
from alembic import op
import sqlalchemy as sa

revision = "20260825_opening_balance_p21a"
down_revision = "20260824_stock_item_intel_p21"
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        "accounting_opening_balance_runs",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
        sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
        sa.Column("previous_company_name", sa.String(255), nullable=False),
        sa.Column("previous_company_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("current_company_name", sa.String(255), nullable=False),
        sa.Column("current_company_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("status", sa.String(30), nullable=False, server_default="completed"),
        sa.Column("ledger_count", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("stock_item_count", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("summary_json", sa.Text(), nullable=False, server_default="{}"),
        sa.Column("created_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
        sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
    )
    for name in ("tenant_id", "client_id", "status", "created_at_utc"):
        op.create_index(f"ix_accounting_opening_balance_runs_{name}", "accounting_opening_balance_runs", [name])

    op.create_table(
        "accounting_opening_ledger_items",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("run_id", sa.Integer(), sa.ForeignKey("accounting_opening_balance_runs.id", ondelete="CASCADE"), nullable=False),
        sa.Column("previous_name", sa.String(255), nullable=False, server_default=""),
        sa.Column("previous_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("previous_group", sa.String(255), nullable=False, server_default=""),
        sa.Column("previous_closing_balance", sa.Float(), nullable=False, server_default="0"),
        sa.Column("current_name", sa.String(255), nullable=False, server_default=""),
        sa.Column("current_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("current_group", sa.String(255), nullable=False, server_default=""),
        sa.Column("current_opening_balance", sa.Float(), nullable=False, server_default="0"),
        sa.Column("difference", sa.Float(), nullable=False, server_default="0"),
        sa.Column("match_status", sa.String(40), nullable=False),
        sa.Column("match_method", sa.String(50), nullable=False, server_default=""),
        sa.Column("confidence", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("correction_status", sa.String(40), nullable=False, server_default="not_requested"),
        sa.Column("correction_note", sa.Text(), nullable=False, server_default=""),
        sa.Column("verified_opening_balance", sa.Float(), nullable=True),
        sa.Column("corrected_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
        sa.Column("corrected_at_utc", sa.DateTime(timezone=True), nullable=True),
    )
    for name in ("run_id", "previous_name", "current_name", "match_status", "correction_status"):
        op.create_index(f"ix_accounting_opening_ledger_items_{name}", "accounting_opening_ledger_items", [name])

    op.create_table(
        "accounting_opening_stock_items",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("run_id", sa.Integer(), sa.ForeignKey("accounting_opening_balance_runs.id", ondelete="CASCADE"), nullable=False),
        sa.Column("previous_name", sa.String(255), nullable=False, server_default=""),
        sa.Column("previous_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("previous_group", sa.String(255), nullable=False, server_default=""),
        sa.Column("previous_hsn", sa.String(40), nullable=False, server_default=""),
        sa.Column("previous_unit", sa.String(80), nullable=False, server_default=""),
        sa.Column("previous_closing_qty", sa.Float(), nullable=False, server_default="0"),
        sa.Column("previous_closing_value", sa.Float(), nullable=False, server_default="0"),
        sa.Column("previous_closing_rate", sa.String(100), nullable=False, server_default=""),
        sa.Column("current_name", sa.String(255), nullable=False, server_default=""),
        sa.Column("current_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("current_group", sa.String(255), nullable=False, server_default=""),
        sa.Column("current_hsn", sa.String(40), nullable=False, server_default=""),
        sa.Column("current_unit", sa.String(80), nullable=False, server_default=""),
        sa.Column("current_opening_qty", sa.Float(), nullable=False, server_default="0"),
        sa.Column("current_opening_value", sa.Float(), nullable=False, server_default="0"),
        sa.Column("quantity_difference", sa.Float(), nullable=False, server_default="0"),
        sa.Column("value_difference", sa.Float(), nullable=False, server_default="0"),
        sa.Column("match_status", sa.String(50), nullable=False),
        sa.Column("match_method", sa.String(50), nullable=False, server_default=""),
        sa.Column("confidence", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("correction_status", sa.String(40), nullable=False, server_default="not_requested"),
        sa.Column("correction_note", sa.Text(), nullable=False, server_default=""),
        sa.Column("verified_opening_qty", sa.Float(), nullable=True),
        sa.Column("verified_opening_value", sa.Float(), nullable=True),
        sa.Column("corrected_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
        sa.Column("corrected_at_utc", sa.DateTime(timezone=True), nullable=True),
    )
    for name in ("run_id", "previous_name", "current_name", "match_status", "correction_status"):
        op.create_index(f"ix_accounting_opening_stock_items_{name}", "accounting_opening_stock_items", [name])

    op.create_table(
        "accounting_opening_master_mappings",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
        sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
        sa.Column("master_type", sa.String(20), nullable=False),
        sa.Column("previous_company_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("previous_name", sa.String(255), nullable=False),
        sa.Column("previous_name_norm", sa.String(255), nullable=False),
        sa.Column("current_company_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("current_name", sa.String(255), nullable=False),
        sa.Column("note", sa.Text(), nullable=False, server_default=""),
        sa.Column("created_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
        sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
        sa.UniqueConstraint("tenant_id", "client_id", "master_type", "previous_company_guid", "previous_name_norm", "current_company_guid", name="uq_accounting_opening_master_mapping"),
    )
    for name in ("tenant_id", "client_id", "master_type", "previous_name_norm"):
        op.create_index(f"ix_accounting_opening_master_mappings_{name}", "accounting_opening_master_mappings", [name])


def downgrade():
    op.drop_table("accounting_opening_master_mappings")
    op.drop_table("accounting_opening_stock_items")
    op.drop_table("accounting_opening_ledger_items")
    op.drop_table("accounting_opening_balance_runs")