from __future__ import annotations

from dataclasses import dataclass, field
from typing import Any

from bson import ObjectId

from app.core.logging import bind_log_context, get_audit_logger


BASE_DOCUMENT_EXCLUDED_TYPES = {
    "epicrisis",
    "epicrisis_case_cache",
    "rips_case_payload",
    "rips_operational_template",
    "rda_case_artifact",
    "rda_case_job_status",
}
CASE_DERIVED_TYPES = {
    "epicrisis",
    "epicrisis_case_cache",
    "rips_case_payload",
    "rda_case_artifact",
    "rda_case_job_status",
}


class CaseDeletionError(Exception):
    def __init__(self, message: str, *, status_code: int) -> None:
        super().__init__(message)
        self.status_code = status_code


@dataclass(slots=True)
class CaseDeletionService:
    mongo_analyses: Any
    mongo_database: Any
    audit_logger: Any = field(init=False)
    case_runtime_collection: Any = field(init=False)

    def __post_init__(self) -> None:
        self.audit_logger = get_audit_logger()
        self.case_runtime_collection = self.mongo_database["processing_batch_cases"]

    def delete_document(self, *, username: str, document_id: str) -> dict[str, Any]:
        normalized_document_id = str(document_id or "").strip()
        if not ObjectId.is_valid(normalized_document_id):
            raise CaseDeletionError("ID de documento no válido.", status_code=404)

        document = self.mongo_analyses.collection.find_one({"_id": ObjectId(normalized_document_id)})
        if not document or str(document.get("usuario") or "").strip() != username:
            raise CaseDeletionError("Documento no encontrado.", status_code=404)

        document_type = str(document.get("tipo_documento") or "").strip()
        if document_type in BASE_DOCUMENT_EXCLUDED_TYPES:
            raise CaseDeletionError(
                "Este endpoint solo permite borrar documentos base del caso.",
                status_code=400,
            )

        case_key = str(document.get("case_key") or "").strip()
        bind_log_context(username=username, case_key=case_key, file_id=normalized_document_id)

        self.mongo_analyses.collection.delete_one({"_id": ObjectId(normalized_document_id), "usuario": username})
        remaining_documents = self._count_case_base_documents(username=username, case_key=case_key)
        derived_deleted = 0
        case_deleted = remaining_documents == 0 and bool(case_key)
        if case_deleted:
            derived_deleted = self._cleanup_case_derivatives(username=username, case_key=case_key)

        self.audit_logger.business_event(
            event_type="case.document_deleted",
            action="delete_case_document",
            outcome="success",
            service="case_deletion_service",
            resource={
                "case_key": case_key,
                "document_id": normalized_document_id,
                "document_type": document_type,
            },
            metrics={
                "remaining_documents": remaining_documents,
                "derived_artifacts_deleted": derived_deleted,
                "case_deleted": case_deleted,
            },
        )

        return {
            "message": "Documento eliminado correctamente.",
            "deleted_document_id": normalized_document_id,
            "case_key": case_key,
            "remaining_documents": remaining_documents,
            "case_deleted": case_deleted,
        }

    def delete_case(self, *, username: str, case_key: str) -> dict[str, Any]:
        normalized_case_key = str(case_key or "").strip()
        if not normalized_case_key:
            raise CaseDeletionError("El case_key es obligatorio.", status_code=400)

        base_documents = list(self._list_case_base_documents(username=username, case_key=normalized_case_key))
        if not base_documents:
            raise CaseDeletionError("Caso no encontrado.", status_code=404)

        bind_log_context(username=username, case_key=normalized_case_key)
        base_document_ids = [item["_id"] for item in base_documents if item.get("_id")]
        deleted_documents = 0
        if base_document_ids:
            result = self.mongo_analyses.collection.delete_many(
                {"_id": {"$in": base_document_ids}, "usuario": username}
            )
            deleted_documents = int(getattr(result, "deleted_count", 0) or 0)

        deleted_derived_artifacts = self._cleanup_case_derivatives(
            username=username,
            case_key=normalized_case_key,
        )

        self.audit_logger.business_event(
            event_type="case.deleted",
            action="delete_case",
            outcome="success",
            service="case_deletion_service",
            resource={"case_key": normalized_case_key},
            metrics={
                "deleted_documents": deleted_documents,
                "deleted_derived_artifacts": deleted_derived_artifacts,
            },
        )

        return {
            "message": "Caso eliminado correctamente.",
            "case_key": normalized_case_key,
            "deleted_documents": deleted_documents,
            "deleted_derived_artifacts": deleted_derived_artifacts,
        }

    def _list_case_base_documents(self, *, username: str, case_key: str) -> list[dict[str, Any]]:
        documents = self.mongo_analyses.collection.find(
            {
                "usuario": username,
                "case_key": case_key,
                "tipo_documento": {"$nin": sorted(BASE_DOCUMENT_EXCLUDED_TYPES)},
            }
        )
        return list(documents)

    def _count_case_base_documents(self, *, username: str, case_key: str) -> int:
        if not case_key:
            return 0
        count_documents = getattr(self.mongo_analyses.collection, "count_documents", None)
        query = {
            "usuario": username,
            "case_key": case_key,
            "tipo_documento": {"$nin": sorted(BASE_DOCUMENT_EXCLUDED_TYPES)},
        }
        if callable(count_documents):
            return int(count_documents(query) or 0)
        return len(self._list_case_base_documents(username=username, case_key=case_key))

    def _cleanup_case_derivatives(self, *, username: str, case_key: str) -> int:
        if not case_key:
            return 0
        analysis_result = self.mongo_analyses.collection.delete_many(
            {
                "usuario": username,
                "case_key": case_key,
                "tipo_documento": {"$in": sorted(CASE_DERIVED_TYPES)},
            }
        )
        runtime_result = self.case_runtime_collection.delete_many(
            {"usuario": username, "case_key": case_key}
        )
        return int(getattr(analysis_result, "deleted_count", 0) or 0) + int(
            getattr(runtime_result, "deleted_count", 0) or 0
        )
