From a0f115fdc48acae789af930d6825cda3e23f9ab5 Mon Sep 17 00:00:00 2001
From: Jake Howard <git@theorangeone.net>
Date: Thu, 28 Jul 2022 22:40:23 +0100
Subject: [PATCH] Add the basics of a search view

---
 website/search/models.py                      | 35 +++++++++++++++++++
 .../search/templates/search/search_page.html  | 17 +++++++--
 2 files changed, 50 insertions(+), 2 deletions(-)

diff --git a/website/search/models.py b/website/search/models.py
index b17eb1d..9b8bb2a 100644
--- a/website/search/models.py
+++ b/website/search/models.py
@@ -1,4 +1,10 @@
+from django.core.paginator import EmptyPage, PageNotAnInteger, Paginator
+from django.http.request import HttpRequest
 from django.utils.functional import cached_property
+from wagtail.models import Page
+from wagtail.query import PageQuerySet
+from wagtail.search.models import Query
+from wagtail.search.utils import parse_query_string
 
 from website.common.models import BaseContentMixin, BasePage
 from website.common.utils import TocEntry
@@ -20,3 +26,32 @@ class SearchPage(BaseContentMixin, BasePage):  # type: ignore[misc]
     @cached_property
     def table_of_contents(self) -> list[TocEntry]:
         return []
+
+    def get_context(self, request: HttpRequest) -> dict:
+        context = super().get_context(request)
+        if query_string := request.GET.get("q", ""):
+            filters, query = parse_query_string(query_string)
+            Query.get(query_string).add_hit()
+            pages = Page.objects.live().search(query)
+        else:
+            pages = Page.objects.none()
+
+        paginator = Paginator(pages, 15)
+        page_num = request.GET.get("page", "1")
+        try:
+            results = paginator.page(page_num)
+
+            # HACK: Search results aren't a queryset, so we can't call `.specific` on it. This forces it to one as efficiently as possible
+            if not isinstance(results.object_list, PageQuerySet):
+                results.object_list = Page.objects.filter(
+                    id__in=list(
+                        results.object_list.get_queryset().values_list("id", flat=True)
+                    )
+                ).specific()
+        except (PageNotAnInteger, EmptyPage):
+            results = None
+            context["invalid_page"] = True
+
+        context["results"] = results
+
+        return context
diff --git a/website/search/templates/search/search_page.html b/website/search/templates/search/search_page.html
index 97aabb9..9d99d78 100644
--- a/website/search/templates/search/search_page.html
+++ b/website/search/templates/search/search_page.html
@@ -4,8 +4,21 @@
 
 {% include "common/hero.html" %}
 
-<section class="container content">
-  {{ page.body_html|safe }}
+{% if page.body_html %}
+  <section class="container content">
+    {{ page.body_html|safe }}
+  </section>
+{% endif %}
+
+<section class="container">
+  {% if invalid_page %}
+    <p>Invalid page</p>
+  {% else %}
+    {% for page in results %}
+      {% include "common/listing-item.html" %}
+    {% endfor %}
+  {% endif %}
 </section>
 
+
 {% endblock %}