From fa17d9018f6f5ef90b8eafb328c082e178a7cf79 Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Thu, 1 Apr 2021 09:39:05 -0700
Subject: [PATCH] Adds user search to get started view

---
 bookwyrm/views/get_started.py | 37 ++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 9 deletions(-)

diff --git a/bookwyrm/views/get_started.py b/bookwyrm/views/get_started.py
index 184cfc3dd..f8e0cd03a 100644
--- a/bookwyrm/views/get_started.py
+++ b/bookwyrm/views/get_started.py
@@ -2,6 +2,8 @@
 import re
 
 from django.contrib.auth.decorators import login_required
+from django.contrib.postgres.search import TrigramSimilarity
+from django.db.models.functions import Greatest
 from django.db.models import Count, Q
 from django.http import HttpResponseNotFound
 from django.shortcuts import get_object_or_404, redirect
@@ -102,17 +104,34 @@ class GetStartedUsers(View):
 
     def get(self, request):
         """ basic profile info """
-        suggested_users = (
-            get_suggested_users(
-                request.user,
-                ~Q(id=request.user.id),
-                ~Q(followers=request.user),
-                bookwyrm_user=True,
+        query = request.GET.get("query")
+        user_results = (
+            models.User.viewer_aware_objects(request.user)
+            .annotate(
+                similarity=Greatest(
+                    TrigramSimilarity("username", query),
+                    TrigramSimilarity("localname", query),
+                )
             )
-            .order_by("shared_books", "-mutuals", "-last_active_date")
-            .all()[:5]
+            .filter(
+                similarity__gt=0.5,
+            )
+            .order_by("-similarity")[:5]
         )
+
+        if user_results.count() < 5:
+            suggested_users = (
+                get_suggested_users(
+                    request.user,
+                    ~Q(id=request.user.id),
+                    ~Q(followers=request.user),
+                    ~Q(id__in=user_results),
+                    bookwyrm_user=True,
+                )
+                .order_by("shared_books", "-mutuals", "-last_active_date")
+                .all()[:5 - user_results.count()]
+            )
         data = {
-            "suggested_users": suggested_users,
+            "suggested_users": list(user_results) + list(suggested_users),
         }
         return TemplateResponse(request, "get_started/users.html", data)