from __future__ import annotations

import uuid

from app.core.logging_config import get_logger
from app.workers.celery_app import celery_app
from app.core.database import get_db_context
from app.services.document_service import DocumentProcessingService

logger = get_logger(__name__)


@celery_app.task(
    name="app.tasks.document_tasks.process_document_task",
    bind=True,
    max_retries=3,
    default_retry_delay=60,
    acks_late=True,
)
def process_document_task(self, document_id: str) -> dict:  # noqa: ANN001
    """Celery task to process a document asynchronously."""
    logger.info("task_started", task_id=self.request.id, document_id=document_id)

    try:
        doc_uuid = uuid.UUID(document_id)
        with get_db_context() as db:
            service = DocumentProcessingService(db)
            document = service.process_document(doc_uuid)
            logger.info(
                "task_completed",
                task_id=self.request.id,
                document_id=document_id,
                status=document.status,
            )
            return {
                "document_id": document_id,
                "status": document.status,
                "page_count": document.page_count,
            }
    except Exception as exc:
        logger.exception(
            "task_failed",
            task_id=self.request.id,
            document_id=document_id,
            error=str(exc),
            retry=self.request.retries,
        )
        raise self.retry(exc=exc)


@celery_app.task(
    name="app.tasks.document_tasks.match_document_task",
    bind=True,
    max_retries=2,
    default_retry_delay=30,
)
def match_document_task(
    self,  # noqa: ANN001
    document_id: str,
    min_confidence: float = 0.75,
    max_results: int = 5,
) -> dict:
    """Celery task to match a document against templates."""
    logger.info("match_task_started", task_id=self.request.id, document_id=document_id)

    try:
        doc_uuid = uuid.UUID(document_id)
        with get_db_context() as db:
            from app.services.matching_service import MatchingService
            service = MatchingService(db)
            matches = service.match_document(
                document_id=doc_uuid,
                min_confidence=min_confidence,
                max_results=max_results,
            )
            logger.info(
                "match_task_completed",
                task_id=self.request.id,
                document_id=document_id,
                matches=len(matches),
            )
            return {
                "document_id": document_id,
                "matches": len(matches),
                "best_score": matches[0].confidence_score if matches else 0.0,
            }
    except Exception as exc:
        logger.exception(
            "match_task_failed",
            task_id=self.request.id,
            document_id=document_id,
            error=str(exc),
        )
        raise self.retry(exc=exc)