"""Phase 19 Bank Analyzer Accounting + Reconciliation.

Revision ID: 20260824_bank_reconciliation_p19
Revises: 20260823_auto_stock_voucher_p18
"""
from alembic import op
import sqlalchemy as sa


revision = "20260824_bank_reconciliation_p19"
down_revision = "20260823_auto_stock_voucher_p18"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        "bank_statement_analysis_jobs",
        sa.Column("purpose", sa.String(30), nullable=False, server_default="analyze_only"),
    )
    op.add_column(
        "bank_statement_analysis_jobs",
        sa.Column("accounting_import_status", sa.String(30), nullable=False, server_default="not_requested"),
    )
    op.add_column(
        "bank_statement_analysis_jobs",
        sa.Column("accounting_import_json", sa.Text(), nullable=True),
    )
    op.create_index(
        "ix_bank_statement_analysis_jobs_purpose",
        "bank_statement_analysis_jobs",
        ["purpose"],
    )
    op.create_index(
        "ix_bank_statement_analysis_jobs_accounting_import_status",
        "bank_statement_analysis_jobs",
        ["accounting_import_status"],
    )

    op.create_table(
        "accounting_bank_reconciliation_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("source_job_id", sa.String(32), sa.ForeignKey("bank_statement_analysis_jobs.id", ondelete="CASCADE"), nullable=False),
        sa.Column("tally_guid", sa.String(120), nullable=False),
        sa.Column("company_name", sa.String(255), nullable=False, server_default=""),
        sa.Column("bank_ledger_name", sa.String(255), nullable=False),
        sa.Column("date_from", sa.String(10), nullable=False),
        sa.Column("date_to", sa.String(10), nullable=False),
        sa.Column("workstation_agent_id", sa.Integer(), sa.ForeignKey("erp_workstation_agents.id", ondelete="RESTRICT"), nullable=False),
        sa.Column("agent_job_id", sa.Integer(), sa.ForeignKey("erp_agent_jobs.id", ondelete="SET NULL"), nullable=True),
        sa.Column("status", sa.String(30), nullable=False, server_default="queued"),
        sa.Column("summary_json", sa.Text(), nullable=False, server_default="{}"),
        sa.Column("last_error", 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.Column("completed_at_utc", sa.DateTime(timezone=True), nullable=True),
    )
    for name in (
        "tenant_id",
        "client_id",
        "source_job_id",
        "tally_guid",
        "workstation_agent_id",
        "agent_job_id",
        "status",
        "created_at_utc",
    ):
        op.create_index(
            f"ix_accounting_bank_reconciliation_runs_{name}",
            "accounting_bank_reconciliation_runs",
            [name],
        )

    op.add_column(
        "accounting_bank_transactions",
        sa.Column("reconciliation_status", sa.String(30), nullable=False, server_default="not_tested"),
    )
    op.add_column(
        "accounting_bank_transactions",
        sa.Column(
            "last_reconciliation_run_id",
            sa.Integer(),
            sa.ForeignKey("accounting_bank_reconciliation_runs.id", ondelete="SET NULL"),
            nullable=True,
        ),
    )
    op.create_index(
        "ix_accounting_bank_transactions_reconciliation_status",
        "accounting_bank_transactions",
        ["reconciliation_status"],
    )
    op.create_index(
        "ix_accounting_bank_transactions_last_reconciliation_run_id",
        "accounting_bank_transactions",
        ["last_reconciliation_run_id"],
    )

    op.create_table(
        "accounting_bank_reconciliation_items",
        sa.Column("id", sa.Integer(), primary_key=True),
        sa.Column("run_id", sa.Integer(), sa.ForeignKey("accounting_bank_reconciliation_runs.id", ondelete="CASCADE"), nullable=False),
        sa.Column("bank_transaction_id", sa.Integer(), sa.ForeignKey("accounting_bank_transactions.id", ondelete="SET NULL"), nullable=True),
        sa.Column("match_status", sa.String(30), nullable=False),
        sa.Column("confidence", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("match_reason", sa.Text(), nullable=False, server_default=""),
        sa.Column("bank_date", sa.String(10), nullable=False, server_default=""),
        sa.Column("bank_direction", sa.String(10), nullable=False, server_default=""),
        sa.Column("bank_amount", sa.Float(), nullable=False, server_default="0"),
        sa.Column("bank_reference", sa.String(180), nullable=False, server_default=""),
        sa.Column("bank_narration", sa.Text(), nullable=False, server_default=""),
        sa.Column("tally_guid", sa.String(120), nullable=False, server_default=""),
        sa.Column("tally_voucher_number", sa.String(160), nullable=False, server_default=""),
        sa.Column("tally_voucher_type", sa.String(60), nullable=False, server_default=""),
        sa.Column("tally_date", sa.String(10), nullable=False, server_default=""),
        sa.Column("tally_reference", sa.String(180), nullable=False, server_default=""),
        sa.Column("tally_narration", sa.Text(), nullable=False, server_default=""),
        sa.Column("tally_amount", sa.Float(), nullable=False, server_default="0"),
        sa.Column("tally_direction", sa.String(10), nullable=False, server_default=""),
    )
    for name in ("run_id", "bank_transaction_id", "match_status"):
        op.create_index(
            f"ix_accounting_bank_reconciliation_items_{name}",
            "accounting_bank_reconciliation_items",
            [name],
        )


def downgrade():
    op.drop_table("accounting_bank_reconciliation_items")

    op.drop_index(
        "ix_accounting_bank_transactions_last_reconciliation_run_id",
        table_name="accounting_bank_transactions",
    )
    op.drop_index(
        "ix_accounting_bank_transactions_reconciliation_status",
        table_name="accounting_bank_transactions",
    )
    op.drop_column("accounting_bank_transactions", "last_reconciliation_run_id")
    op.drop_column("accounting_bank_transactions", "reconciliation_status")

    op.drop_table("accounting_bank_reconciliation_runs")

    op.drop_index(
        "ix_bank_statement_analysis_jobs_accounting_import_status",
        table_name="bank_statement_analysis_jobs",
    )
    op.drop_index(
        "ix_bank_statement_analysis_jobs_purpose",
        table_name="bank_statement_analysis_jobs",
    )
    op.drop_column("bank_statement_analysis_jobs", "accounting_import_json")
    op.drop_column("bank_statement_analysis_jobs", "accounting_import_status")
    op.drop_column("bank_statement_analysis_jobs", "purpose")