"""Assignment Engine global search integration."""

from __future__ import annotations

from typing import TYPE_CHECKING

from simorgh.core.search import SearchableSpec, SearchHit, register_searchable

if TYPE_CHECKING:
    from simorgh.core.context import RequestContext


def _search_assignments(ctx: RequestContext, query: str, limit: int) -> list[SearchHit]:
    from django.db.models import Q

    from simorgh.apps.assignments.models import Assignment

    if not ctx.tenant:
        return []

    qs = (
        Assignment.objects.filter(tenant_id=ctx.tenant.pk)
        .filter(
            Q(title__icontains=query)
            | Q(description__icontains=query)
        )
        .distinct()
        .order_by("title")[:limit]
    )

    return [
        SearchHit(
            entity="assignments.assignment",
            id=str(a.public_id),
            title=a.title,
            subtitle=a.get_status_display(),
            url=f"/assignments/{a.public_id}/",
            score=2.0 if query.lower() in a.title.lower() else 1.0,
        )
        for a in qs
    ]


register_searchable(
    SearchableSpec(
        entity="assignments.assignment",
        label_key="assignments.assignment",
        search=_search_assignments,
    )
)