From d8132db75b83da39a64d974a6e5f7eca450a81f8 Mon Sep 17 00:00:00 2001
From: epriestley <git@epriestley.com>
Date: Thu, 7 Sep 2017 13:05:24 -0700
Subject: [PATCH] Support Ferret engine in Pholio

Summary: Ref T12819. Support for Pholio.

Test Plan: Indexed and searched mocks.

Reviewers: chad

Reviewed By: chad

Maniphest Tasks: T12819

Differential Revision: https://secure.phabricator.com/D18569
---
 .../20170907.ferret.28.mock.doc.sql            |  9 +++++++++
 .../20170907.ferret.29.mock.field.sql          |  8 ++++++++
 .../20170907.ferret.30.mock.ngrams.sql         |  5 +++++
 src/__phutil_library_map__.php                 |  3 +++
 .../pholio/search/PholioMockFerretEngine.php   | 18 ++++++++++++++++++
 src/applications/pholio/storage/PholioMock.php | 10 +++++++++-
 6 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 resources/sql/autopatches/20170907.ferret.28.mock.doc.sql
 create mode 100644 resources/sql/autopatches/20170907.ferret.29.mock.field.sql
 create mode 100644 resources/sql/autopatches/20170907.ferret.30.mock.ngrams.sql
 create mode 100644 src/applications/pholio/search/PholioMockFerretEngine.php

diff --git a/resources/sql/autopatches/20170907.ferret.28.mock.doc.sql b/resources/sql/autopatches/20170907.ferret.28.mock.doc.sql
new file mode 100644
index 0000000000..eb80ef3937
--- /dev/null
+++ b/resources/sql/autopatches/20170907.ferret.28.mock.doc.sql
@@ -0,0 +1,9 @@
+CREATE TABLE {$NAMESPACE}_pholio.pholio_mock_fdocument (
+  id INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,
+  objectPHID VARBINARY(64) NOT NULL,
+  isClosed BOOL NOT NULL,
+  authorPHID VARBINARY(64),
+  ownerPHID VARBINARY(64),
+  epochCreated INT UNSIGNED NOT NULL,
+  epochModified INT UNSIGNED NOT NULL
+) ENGINE=InnoDB, COLLATE {$COLLATE_TEXT};
diff --git a/resources/sql/autopatches/20170907.ferret.29.mock.field.sql b/resources/sql/autopatches/20170907.ferret.29.mock.field.sql
new file mode 100644
index 0000000000..0cb0e97d05
--- /dev/null
+++ b/resources/sql/autopatches/20170907.ferret.29.mock.field.sql
@@ -0,0 +1,8 @@
+CREATE TABLE {$NAMESPACE}_pholio.pholio_mock_ffield (
+  id INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,
+  documentID INT UNSIGNED NOT NULL,
+  fieldKey VARCHAR(4) NOT NULL COLLATE {$COLLATE_TEXT},
+  rawCorpus LONGTEXT NOT NULL COLLATE {$COLLATE_SORT},
+  termCorpus LONGTEXT NOT NULL COLLATE {$COLLATE_SORT},
+  normalCorpus LONGTEXT NOT NULL COLLATE {$COLLATE_SORT}
+) ENGINE=InnoDB, COLLATE {$COLLATE_TEXT};
diff --git a/resources/sql/autopatches/20170907.ferret.30.mock.ngrams.sql b/resources/sql/autopatches/20170907.ferret.30.mock.ngrams.sql
new file mode 100644
index 0000000000..e343ccf83b
--- /dev/null
+++ b/resources/sql/autopatches/20170907.ferret.30.mock.ngrams.sql
@@ -0,0 +1,5 @@
+CREATE TABLE {$NAMESPACE}_pholio.pholio_mock_fngrams (
+  id INT UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY,
+  documentID INT UNSIGNED NOT NULL,
+  ngram CHAR(3) NOT NULL COLLATE {$COLLATE_TEXT}
+) ENGINE=InnoDB, COLLATE {$COLLATE_TEXT};
diff --git a/src/__phutil_library_map__.php b/src/__phutil_library_map__.php
index 67bb9b73f3..f42f815f1f 100644
--- a/src/__phutil_library_map__.php
+++ b/src/__phutil_library_map__.php
@@ -4475,6 +4475,7 @@ phutil_register_library_map(array(
     'PholioMockEditController' => 'applications/pholio/controller/PholioMockEditController.php',
     'PholioMockEditor' => 'applications/pholio/editor/PholioMockEditor.php',
     'PholioMockEmbedView' => 'applications/pholio/view/PholioMockEmbedView.php',
+    'PholioMockFerretEngine' => 'applications/pholio/search/PholioMockFerretEngine.php',
     'PholioMockFulltextEngine' => 'applications/pholio/search/PholioMockFulltextEngine.php',
     'PholioMockHasTaskEdgeType' => 'applications/pholio/edge/PholioMockHasTaskEdgeType.php',
     'PholioMockHasTaskRelationship' => 'applications/pholio/relationships/PholioMockHasTaskRelationship.php',
@@ -10145,6 +10146,7 @@ phutil_register_library_map(array(
       'PhabricatorSpacesInterface',
       'PhabricatorMentionableInterface',
       'PhabricatorFulltextInterface',
+      'PhabricatorFerretInterface',
     ),
     'PholioMockArchiveController' => 'PholioController',
     'PholioMockAuthorHeraldField' => 'PholioMockHeraldField',
@@ -10154,6 +10156,7 @@ phutil_register_library_map(array(
     'PholioMockEditController' => 'PholioController',
     'PholioMockEditor' => 'PhabricatorApplicationTransactionEditor',
     'PholioMockEmbedView' => 'AphrontView',
+    'PholioMockFerretEngine' => 'PhabricatorFerretEngine',
     'PholioMockFulltextEngine' => 'PhabricatorFulltextEngine',
     'PholioMockHasTaskEdgeType' => 'PhabricatorEdgeType',
     'PholioMockHasTaskRelationship' => 'PholioMockRelationship',
diff --git a/src/applications/pholio/search/PholioMockFerretEngine.php b/src/applications/pholio/search/PholioMockFerretEngine.php
new file mode 100644
index 0000000000..f4f9b4610f
--- /dev/null
+++ b/src/applications/pholio/search/PholioMockFerretEngine.php
@@ -0,0 +1,18 @@
+<?php
+
+final class PholioMockFerretEngine
+  extends PhabricatorFerretEngine {
+
+  public function getApplicationName() {
+    return 'pholio';
+  }
+
+  public function getScopeName() {
+    return 'mock';
+  }
+
+  public function newSearchEngine() {
+    return new PholioMockSearchEngine();
+  }
+
+}
diff --git a/src/applications/pholio/storage/PholioMock.php b/src/applications/pholio/storage/PholioMock.php
index 7f58a95511..4aa9ef4055 100644
--- a/src/applications/pholio/storage/PholioMock.php
+++ b/src/applications/pholio/storage/PholioMock.php
@@ -12,7 +12,8 @@ final class PholioMock extends PholioDAO
     PhabricatorDestructibleInterface,
     PhabricatorSpacesInterface,
     PhabricatorMentionableInterface,
-    PhabricatorFulltextInterface {
+    PhabricatorFulltextInterface,
+    PhabricatorFerretInterface {
 
   const MARKUP_FIELD_DESCRIPTION  = 'markup:description';
 
@@ -320,4 +321,11 @@ final class PholioMock extends PholioDAO
   }
 
 
+/* -(  PhabricatorFerretInterface  )----------------------------------------- */
+
+  public function newFerretEngine() {
+    return new PholioMockFerretEngine();
+  }
+
+
 }
-- 
2.51.2