From dc10a7e69ea33fd4b589f0d9e8c9921bc6f94765 Mon Sep 17 00:00:00 2001
From: Benjamin Kausch <benjamin.kausch@gmail.com>
Date: Sat, 2 Sep 2023 18:52:13 +0200
Subject: [PATCH] Implement ferret engine in typeahead datasource query for
 repos

Summary:
This broadens the typeahead datasource search for repos. Before this patch a repository named "Alligator Simulator" would not be found with the search string "simu...". This is patched with the ferret engine search and indexing features.

See T15583

Test Plan: Create repositories with titles with 2 or more words. Search for these repos with the global typeahead search. The search term should begin with the second/third/n-th word of the repo title.

Reviewers: O1 Blessed Committers, avivey

Reviewed By: O1 Blessed Committers, avivey

Subscribers: avivey, tobiaswiese, valerio.bozzolan, Matthew, Cigaryno

Maniphest Tasks: T15583

Differential Revision: https://we.phorge.it/D25430
---
 .../20230902.repository.01.rebuild-index.php  |  6 +++++
 .../DiffusionRepositoryDatasource.php         | 15 ++++++++-----
 .../query/PhabricatorRepositoryQuery.php      | 22 -------------------
 .../PhabricatorRepositoryFulltextEngine.php   | 15 +++++++++++--
 4 files changed, 28 insertions(+), 30 deletions(-)
 create mode 100644 resources/sql/autopatches/20230902.repository.01.rebuild-index.php

diff --git a/resources/sql/autopatches/20230902.repository.01.rebuild-index.php b/resources/sql/autopatches/20230902.repository.01.rebuild-index.php
new file mode 100644
index 0000000000..aba27ce78c
--- /dev/null
+++ b/resources/sql/autopatches/20230902.repository.01.rebuild-index.php
@@ -0,0 +1,6 @@
+<?php
+
+// @phase worker
+
+PhabricatorRebuildIndexesWorker::rebuildObjectsWithQuery(
+  'PhabricatorRepositoryQuery');
diff --git a/src/applications/diffusion/typeahead/DiffusionRepositoryDatasource.php b/src/applications/diffusion/typeahead/DiffusionRepositoryDatasource.php
index cc9da44723..5953dac8b3 100644
--- a/src/applications/diffusion/typeahead/DiffusionRepositoryDatasource.php
+++ b/src/applications/diffusion/typeahead/DiffusionRepositoryDatasource.php
@@ -16,13 +16,16 @@ final class DiffusionRepositoryDatasource
   }
 
   public function loadResults() {
-    $viewer = $this->getViewer();
-    $raw_query = $this->getRawQuery();
-
     $query = id(new PhabricatorRepositoryQuery())
-      ->setOrder('name')
-      ->withDatasourceQuery($raw_query);
-    $repos = $this->executeQuery($query);
+      ->setViewer($this->getViewer());
+
+    $this->applyFerretConstraints(
+      $query,
+      id(new PhabricatorRepository())->newFerretEngine(),
+      'title',
+      $this->getRawQuery());
+
+    $repos = $query->execute();
 
     $type_icon = id(new PhabricatorRepositoryRepositoryPHIDType())
       ->getTypeIcon();
diff --git a/src/applications/repository/query/PhabricatorRepositoryQuery.php b/src/applications/repository/query/PhabricatorRepositoryQuery.php
index 17a46f7b62..8dead39758 100644
--- a/src/applications/repository/query/PhabricatorRepositoryQuery.php
+++ b/src/applications/repository/query/PhabricatorRepositoryQuery.php
@@ -9,7 +9,6 @@ final class PhabricatorRepositoryQuery
   private $types;
   private $uuids;
   private $uris;
-  private $datasourceQuery;
   private $slugs;
   private $almanacServicePHIDs;
 
@@ -124,11 +123,6 @@ final class PhabricatorRepositoryQuery
     return $this;
   }
 
-  public function withDatasourceQuery($query) {
-    $this->datasourceQuery = $query;
-    return $this;
-  }
-
   public function withSlugs(array $slugs) {
     $this->slugs = $slugs;
     return $this;
@@ -637,22 +631,6 @@ final class PhabricatorRepositoryQuery
         $this->uuids);
     }
 
-    if (phutil_nonempty_string($this->datasourceQuery)) {
-      // This handles having "rP" match callsigns starting with "P...".
-      $query = trim($this->datasourceQuery);
-      if (preg_match('/^r/', $query)) {
-        $callsign = substr($query, 1);
-      } else {
-        $callsign = $query;
-      }
-      $where[] = qsprintf(
-        $conn,
-        'r.name LIKE %> OR r.callsign LIKE %> OR r.repositorySlug LIKE %>',
-        $query,
-        $callsign,
-        $query);
-    }
-
     if ($this->slugs !== null) {
       $where[] = qsprintf(
         $conn,
diff --git a/src/applications/repository/search/PhabricatorRepositoryFulltextEngine.php b/src/applications/repository/search/PhabricatorRepositoryFulltextEngine.php
index f666af552f..838692c351 100644
--- a/src/applications/repository/search/PhabricatorRepositoryFulltextEngine.php
+++ b/src/applications/repository/search/PhabricatorRepositoryFulltextEngine.php
@@ -7,10 +7,21 @@ final class PhabricatorRepositoryFulltextEngine
     PhabricatorSearchAbstractDocument $document,
     $object) {
     $repo = $object;
-    $document->setDocumentTitle($repo->getName());
+
+    $title_fields = array(
+      $repo->getName(),
+      $repo->getRepositorySlug(),
+    );
+    $callsign = $repo->getCallsign();
+    if ($callsign) {
+      $title_fields[] = $callsign;
+      $title_fields[] = 'r'.$callsign;
+    }
+
+    $document->setDocumentTitle(implode("\n", $title_fields));
     $document->addField(
       PhabricatorSearchDocumentFieldType::FIELD_BODY,
-      $repo->getRepositorySlug()."\n".$repo->getDetail('description'));
+      $repo->getDetail('description'));
 
     $document->setDocumentCreated($repo->getDateCreated());
     $document->setDocumentModified($repo->getDateModified());