From df27a164eccc37546d9989692418fe1ec269205a Mon Sep 17 00:00:00 2001
From: Matt Westcott <matt@west.co.tt>
Date: Fri, 10 Jun 2022 21:57:50 +0100
Subject: [PATCH] Implement search form

---
 .../wagtailadmin/generic/chooser/chooser.html | 20 +++++++++-
 wagtail/admin/views/generic/chooser.py        | 40 +++++++++++++++++++
 2 files changed, 59 insertions(+), 1 deletion(-)

diff --git a/wagtail/admin/templates/wagtailadmin/generic/chooser/chooser.html b/wagtail/admin/templates/wagtailadmin/generic/chooser/chooser.html
index 5221904aa5..1a25e63b46 100644
--- a/wagtail/admin/templates/wagtailadmin/generic/chooser/chooser.html
+++ b/wagtail/admin/templates/wagtailadmin/generic/chooser/chooser.html
@@ -9,7 +9,25 @@
             role="tabpanel"
             aria-labelledby="tab-label-search"
         >
-            <div id="search-results" class="listing documents">
+
+            {% if filter_form.fields %}
+                <form data-chooser-modal-search class="search-bar" action="{% url results_url_name %}" method="GET" novalidate>
+                    {% if filter_form.hidden_fields %}
+                        {% for field in form.hidden_fields %}{{ field }}{% endfor %}
+                    {% endif %}
+
+                    {% if filter_form.visible_fields %}
+                        <ul class="fields">
+                            {% for field in filter_form.visible_fields %}
+                                {% include "wagtailadmin/shared/field_as_li.html" with field=field %}
+                            {% endfor %}
+                            <li class="submit"><input type="submit" value="{% trans 'Search' %}" class="button" /></li>
+                        </ul>
+                    {% endif %}
+                </form>
+            {% endif %}
+
+            <div id="search-results" class="listing">
                 {% include "wagtailadmin/generic/chooser/results.html" %}
             </div>
         </section>
diff --git a/wagtail/admin/views/generic/chooser.py b/wagtail/admin/views/generic/chooser.py
index f42250ed7f..06cdc65d8c 100644
--- a/wagtail/admin/views/generic/chooser.py
+++ b/wagtail/admin/views/generic/chooser.py
@@ -1,3 +1,4 @@
+from django import forms
 from django.contrib.admin.utils import unquote
 from django.core.exceptions import ObjectDoesNotExist
 from django.core.paginator import Paginator
@@ -9,6 +10,8 @@ from django.views.generic.base import ContextMixin, View
 from wagtail.admin.admin_url_finder import AdminURLFinder
 from wagtail.admin.modal_workflow import render_modal_workflow
 from wagtail.admin.ui.tables import Table, TitleColumn
+from wagtail.search.backends import get_search_backend
+from wagtail.search.index import class_is_indexed
 
 
 class ModalPageFurnitureMixin(ContextMixin):
@@ -37,12 +40,44 @@ class BaseChooseView(ModalPageFurnitureMixin, ContextMixin, View):
     results_url_name = None
     icon = "snippet"
     page_title = _("Choose")
+    filter_form_class = None
 
     def get_object_list(self):
         return self.model.objects.all()
 
+    def get_filter_form_class(self):
+        if self.filter_form_class:
+            return self.filter_form_class
+        else:
+            fields = {}
+            if class_is_indexed(self.model):
+                fields["q"] = forms.CharField(
+                    label=_("Search term"), widget=forms.TextInput(), required=False
+                )
+
+            return type(
+                "FilterForm",
+                (forms.Form,),
+                fields,
+            )
+
+    def get_filter_form(self):
+        FilterForm = self.get_filter_form_class()
+        return FilterForm(self.request.GET)
+
+    def filter_object_list(self, objects, form):
+        search_query = form.cleaned_data.get("q")
+        if search_query:
+            search_backend = get_search_backend()
+            objects = search_backend.search(search_query, objects)
+        return objects
+
     def get(self, request):
         objects = self.get_object_list()
+        self.filter_form = self.get_filter_form()
+        if self.filter_form.is_valid():
+            objects = self.filter_object_list(objects, self.filter_form)
+
         paginator = Paginator(objects, per_page=self.per_page)
         self.results = paginator.get_page(request.GET.get("p"))
 
@@ -76,6 +111,11 @@ class BaseChooseView(ModalPageFurnitureMixin, ContextMixin, View):
 
 
 class ChooseView(BaseChooseView):
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["filter_form"] = self.filter_form
+        return context
+
     def render_to_response(self):
         return render_modal_workflow(
             self.request,