import asyncio
from decimal import Decimal
from io import BytesIO
from types import SimpleNamespace
from typing import Any, cast

from fastapi import Request
from openpyxl import load_workbook

from app.batch_processing.infrastructure.batch_epicrisis_excel import (
    BatchEpicrisisExcelCaseRow,
    BatchEpicrisisExcelWorkbookBuilder,
)
from app.case_epicrisis.application.cost_ordering import (
    COST_ORDER_VERSION,
    apply_factura_cost_ordering,
    attach_and_sort_costs,
    extract_invoice_cost_lines,
    match_billed_cost,
    normalize_money,
)
from app.case_epicrisis.application.utils import (
    build_pdf_primary_ayudas_diagnosticas,
    build_pdf_primary_medicamentos,
    build_pdf_primary_procedimientos,
    build_procedimientos_clinicos,
)
from app.core.services import AppServices
from app.models import UserInDB
from app.routes.pages.epicrisis_pdf_cache import guardar_epicrisis_pdf_cache


def _factura(lines=None, services=None):
    return {
        "factura_json": {
            "lineas_canonicas": lines or [],
            "servicios_procedimientos": services or {},
        }
    }


def test_normalize_money_accepts_colombian_and_normalized_formats() -> None:
    assert normalize_money("$ 1.234.567,89") == Decimal("1234567.89")
    assert normalize_money("1,234,567.89") == Decimal("1234567.89")
    assert normalize_money("1234.50") == Decimal("1234.50")
    assert normalize_money("0") == Decimal("0")


def test_normalize_money_rejects_invalid_and_negative_values() -> None:
    assert normalize_money("") is None
    assert normalize_money("sin valor") is None
    assert normalize_money("1.2.3") is None
    assert normalize_money("-$ 10.000") is None


def test_cost_uses_quantity_times_unit_value_when_total_is_missing() -> None:
    lines = extract_invoice_cost_lines(
        _factura(
            [
                {
                    "categoria": "medicamentos",
                    "codigo_referencia": "CUM-1",
                    "descripcion": "Dipirona",
                    "cantidad": "2",
                    "valor_unitario": "$ 1.500,50",
                }
            ]
        )
    )

    cost = match_billed_cost(
        {"codigo_referencia": "CUM-1", "nombre": "Dipirona"},
        category="medicamento",
        lines=lines,
    )

    assert cost.valor_total == Decimal("3001.00")
    assert cost.origen == "cantidad_por_valor_unitario"
    assert cost.coincidencia == "codigo_exacto"


def test_repeated_invoice_lines_select_largest_individual_total() -> None:
    lines = extract_invoice_cost_lines(
        _factura(
            [
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "123456",
                    "descripcion": "Lavado quirúrgico",
                    "total": "80.000",
                },
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "123456",
                    "descripcion": "Lavado quirúrgico",
                    "total": "120.000",
                },
            ]
        )
    )

    cost = match_billed_cost(
        {"codigo_cups": "123456", "descripcion": "Lavado quirúrgico"},
        category="procedimiento",
        lines=lines,
    )

    assert cost.valor_total == Decimal("120000")
    assert cost.linea_factura_id == "lineas_canonicas:1"


def test_unique_text_matches_and_distinct_codes_make_same_text_ambiguous() -> None:
    unique_lines = extract_invoice_cost_lines(
        _factura(
            [
                {
                    "categoria": "imagenologia",
                    "codigo_facturacion": "RX01",
                    "descripcion": "Radiografía de tórax",
                    "total": "50.000",
                }
            ]
        )
    )
    assert match_billed_cost(
        {"tipo": "imagen", "nombre": "Radiografia de torax"},
        category="imagen",
        lines=unique_lines,
    ).coincidencia == "texto_normalizado_unico"

    ambiguous_lines = extract_invoice_cost_lines(
        _factura(
            [
                {
                    "categoria": "imagenologia",
                    "codigo_facturacion": "RX01",
                    "descripcion": "Radiografía de tórax",
                    "total": "50.000",
                },
                {
                    "categoria": "imagenologia",
                    "codigo_facturacion": "RX02",
                    "descripcion": "Radiografía de tórax",
                    "total": "70.000",
                },
            ]
        )
    )
    ambiguous = match_billed_cost(
        {"tipo": "imagen", "nombre": "Radiografía de tórax"},
        category="imagen",
        lines=ambiguous_lines,
    )
    assert ambiguous.valor_total is None
    assert ambiguous.coincidencia == "ambigua"


def test_stable_descending_sort_keeps_ties_and_unknown_items_last() -> None:
    lines = extract_invoice_cost_lines(
        _factura(
            [
                {"categoria": "medicamentos", "descripcion": "A", "total": "10"},
                {"categoria": "medicamentos", "descripcion": "B", "total": "20"},
                {"categoria": "medicamentos", "descripcion": "C", "total": "20"},
                {"categoria": "medicamentos", "descripcion": "D", "total": "0"},
            ]
        )
    )
    ordered = attach_and_sort_costs(
        [
            {"nombre": "A"},
            {"nombre": "B"},
            {"nombre": "C"},
            {"nombre": "D"},
            {"nombre": "Manual"},
        ],
        category="medicamento",
        lines=lines,
    )

    assert [item["nombre"] for item in ordered] == ["B", "C", "A", "D", "Manual"]


def test_context_orders_medications_and_aids_without_mixing_aid_categories() -> None:
    context = {
        "factura": _factura(
            [
                {"categoria": "medicamentos", "descripcion": "Acetaminofén", "total": "15.000"},
                {"categoria": "medicamentos", "descripcion": "Cefazolina", "total": "90.000"},
                {"categoria": "imagenologia", "descripcion": "RX tórax", "total": "40.000"},
                {"categoria": "examenes_laboratorio", "descripcion": "Hemograma", "total": "25.000"},
            ]
        ),
        "medicamentos_caso": [
            {"nombre": "Acetaminofén", "medicamento": "Acetaminofén", "estados": ["administrado"]},
            {"nombre": "Cefazolina", "medicamento": "Cefazolina", "estados": ["administrado"]},
            {"nombre": "Manual", "medicamento": "Manual", "estados": ["administrado"]},
        ],
        "ayudas_diagnosticas": [
            {"tipo": "laboratorio", "nombre": "Hemograma", "fuente": "factura", "exportable": True},
            {"tipo": "imagen", "nombre": "RX tórax", "fuente": "factura", "exportable": True},
            {"tipo": "imagen", "nombre": "Ayuda manual", "fuente": "manual_pdf", "exportable": True},
        ],
    }

    apply_factura_cost_ordering(context)

    assert [item["nombre"] for item in context["medicamentos_caso"]] == [
        "Cefazolina",
        "Acetaminofén",
        "Manual",
    ]
    assert [item["nombre"] for item in context["ayudas_diagnosticas"]] == [
        "RX tórax",
        "Hemograma",
        "Ayuda manual",
    ]
    assert [item["nombre"] for item in build_pdf_primary_medicamentos(context)] == [
        "Cefazolina",
        "Acetaminofén",
        "Manual",
    ]
    assert [item["nombre"] for item in build_pdf_primary_ayudas_diagnosticas(context)] == [
        "RX tórax",
        "Hemograma",
        "Ayuda manual",
    ]


def test_procedure_groups_and_pdf_projection_follow_cost_order() -> None:
    context = {
        "orden_costo_version": COST_ORDER_VERSION,
        "factura": _factura(
            [
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "010101",
                    "descripcion": "Procedimiento menor",
                    "total": "20.000",
                },
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "020202",
                    "descripcion": "Procedimiento mayor",
                    "total": "200.000",
                },
            ]
        ),
        "procedimientos_hc": [
            "010101 Procedimiento menor",
            "020202 Procedimiento mayor",
            "Procedimiento manual",
        ],
        "procedimientos_factura": [],
        "codigos_desde_soat": [],
        "pdf_draft": {},
    }

    procedures = build_procedimientos_clinicos(context)

    assert [item["codigo_cups"] for item in procedures[:2]] == ["020202", "010101"]
    assert procedures[-1]["descripcion"] == "Procedimiento manual"
    assert build_pdf_primary_procedimientos(context) == [
        "020202 - Procedimiento mayor",
        "010101 - Procedimiento menor",
        "Procedimiento manual",
    ]


def test_services_are_used_only_as_category_fallback() -> None:
    lines = extract_invoice_cost_lines(
        _factura(
            [
                {
                    "categoria": "medicamentos",
                    "descripcion": "Dipirona",
                    "total": "10.000",
                }
            ],
            {
                "medicamentos": [{"medicamento": "Dipirona", "total": "99.000"}],
                "imagenologia": [{"estudio": "TAC", "descripcion": "TAC", "total": "80.000"}],
            },
        )
    )

    medication = match_billed_cost(
        {"nombre": "Dipirona"}, category="medicamento", lines=lines
    )
    image = match_billed_cost({"nombre": "TAC"}, category="imagen", lines=lines)

    assert medication.valor_total == Decimal("10000")
    assert image.valor_total == Decimal("80000")


def test_batch_excel_preserves_cost_order_without_rendering_cost_metadata() -> None:
    context = {
        "factura": _factura(
            [
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "010101",
                    "descripcion": "Procedimiento menor",
                    "total": "20.000",
                },
                {
                    "categoria": "procedimientos_quirurgicos",
                    "codigo_referencia": "020202",
                    "descripcion": "Procedimiento mayor",
                    "total": "200.000",
                },
                {"categoria": "medicamentos", "descripcion": "Medicamento menor", "total": "10.000"},
                {"categoria": "medicamentos", "descripcion": "Medicamento mayor", "total": "90.000"},
                {"categoria": "imagenologia", "descripcion": "RX menor", "total": "30.000"},
                {"categoria": "imagenologia", "descripcion": "TAC mayor", "total": "120.000"},
            ]
        ),
        "procedimientos_hc": ["010101 Procedimiento menor", "020202 Procedimiento mayor"],
        "procedimientos_factura": [],
        "codigos_desde_soat": [],
        "pdf_draft": {},
        "medicamentos_caso": [
            {
                "nombre": "Medicamento menor",
                "medicamento": "Medicamento menor",
                "estados": ["administrado"],
            },
            {
                "nombre": "Medicamento mayor",
                "medicamento": "Medicamento mayor",
                "estados": ["administrado"],
            },
        ],
        "ayudas_diagnosticas": [
            {"tipo": "imagen", "nombre": "RX menor", "fuente": "factura", "exportable": True},
            {"tipo": "imagen", "nombre": "TAC mayor", "fuente": "factura", "exportable": True},
        ],
    }
    apply_factura_cost_ordering(context)
    context["procedimientos_clinicos"] = build_procedimientos_clinicos(context)

    content = BatchEpicrisisExcelWorkbookBuilder().build(
        batch_id="batch-cost-order",
        rows=[
            BatchEpicrisisExcelCaseRow(
                case_key="case-cost-order",
                patient_name="Paciente",
                patient_id="CC 1",
                case_number="1",
                procedure_description="",
                epicrisis_status="completado",
                included_in_excel=True,
                omission_reason="",
                context=context,
            )
        ],
    )
    sheet = load_workbook(BytesIO(content))["case-cost-order"]
    visible = [
        str(cell.value)
        for row in sheet.iter_rows()
        for cell in row
        if cell.value is not None
    ]

    assert visible.index("Procedimiento mayor") < visible.index("Procedimiento menor")
    assert visible.index("Medicamento mayor") < visible.index("Medicamento menor")
    assert visible.index("TAC mayor") < visible.index("RX menor")
    assert not any("costo_facturado" in value or value in {"200000", "120000", "90000"} for value in visible)


def test_pdf_cache_preserves_order_and_cost_version() -> None:
    class RequestStub:
        async def json(self):
            return {
                "html_pdf": "<html><body>Epicrisis</body></html>",
                "nombre_paciente": "Paciente",
                "procedimientos": ["020202 - Mayor", "010101 - Menor", "Manual"],
                "medicamentos": [
                    {"nombre": "Mayor", "medicamento": "Mayor"},
                    {"nombre": "Menor", "medicamento": "Menor"},
                ],
                "ayudas_diagnosticas": [
                    {"tipo": "imagen", "nombre": "TAC"},
                    {"tipo": "laboratorio", "nombre": "Hemograma"},
                ],
            }

    class CollectionStub:
        def __init__(self):
            self.inserted: Any = None

        def insert_one(self, document: dict[str, Any]):
            self.inserted = document
            return SimpleNamespace(inserted_id="cache-cost-order")

    collection = CollectionStub()
    services = SimpleNamespace(
        mongo_analyses=SimpleNamespace(
            collection=SimpleNamespace(database={"epicrisis_pdf_cache": collection})
        )
    )

    result = asyncio.run(
        guardar_epicrisis_pdf_cache(
            cast(Request, RequestStub()),
            cast(UserInDB, SimpleNamespace(username="tester")),
            cast(AppServices, services),
        )
    )

    assert result["cache_id"] == "cache-cost-order"
    assert collection.inserted["orden_costo_version"] == COST_ORDER_VERSION
    assert collection.inserted["procedimientos"] == [
        "020202 - Mayor",
        "010101 - Menor",
        "Manual",
    ]
    assert [item["medicamento"] for item in collection.inserted["medicamentos_detallados"]] == [
        "Mayor",
        "Menor",
    ]
    assert [item["nombre"] for item in collection.inserted["ayudas_diagnosticas"]] == [
        "TAC",
        "Hemograma",
    ]
