Misago/misago/threads/search.py

from django.contrib.postgres.search import SearchQuery, SearchRank, SearchVector
from django.utils.translation import gettext_lazy as _

from misago.conf import settings
from misago.core.shortcuts import paginate, pagination_dict
from misago.search import SearchProvider

from .filtersearch import filter_search
from .models import Post, Thread
from .permissions import exclude_invisible_threads
from .serializers import FeedSerializer
from .utils import add_categories_to_items
from .viewmodels import ThreadsRootCategory


HITS_CEILING = settings.MISAGO_POSTS_PER_PAGE * 5


class SearchThreads(SearchProvider):
    name = _("Threads")
    icon = "forum"
    url = "threads"

    def search(self, query, page=1):
        root_category = ThreadsRootCategory(self.request)
        threads_categories = [root_category.unwrap()] + root_category.subcategories

        if len(query) > 2:
            visible_threads = exclude_invisible_threads(
                self.request.user_acl, threads_categories, Thread.objects
            )
            results = search_threads(self.request, query, visible_threads)
        else:
            results = []

        list_page = paginate(
            results,
            page,
            settings.MISAGO_POSTS_PER_PAGE,
            settings.MISAGO_POSTS_TAIL,
            allow_explicit_first_page=True,
        )
        paginator = pagination_dict(list_page)

        posts = []
        threads = []
        if paginator["count"]:
            posts = list(
                list_page.object_list.select_related("thread", "poster", "poster__rank")
            )

            threads = []
            for post in posts:
                threads.append(post.thread)

            add_categories_to_items(
                root_category.unwrap(), threads_categories, posts + threads
            )

        results = {
            "results": FeedSerializer(
                posts, many=True, context={"user": self.request.user}
            ).data
        }
        results.update(paginator)

        return results


def search_threads(request, query, visible_threads):
    search_query = SearchQuery(
        filter_search(query), config=settings.MISAGO_SEARCH_CONFIG
    )
    search_vector = SearchVector(
        "search_document", config=settings.MISAGO_SEARCH_CONFIG
    )

    queryset = Post.objects.filter(
        is_event=False,
        is_hidden=False,
        is_unapproved=False,
        thread_id__in=visible_threads.values("id"),
        search_vector=search_query,
    )

    if queryset[: HITS_CEILING + 1].count() > HITS_CEILING:
        queryset = queryset.order_by("-id")[:HITS_CEILING]

    return (
        Post.objects.filter(id__in=queryset.values("id"))
        .annotate(rank=SearchRank(search_vector, search_query))
        .order_by("-rank", "-id")
    )