src/applications/search/index/PhabricatorFulltextEngine.php

   1 <?php
   2
   3 abstract class PhabricatorFulltextEngine
   4   extends Phobject {
   5
   6   private $object;
   7
   8   public function setObject($object) {
   9     $this->object = $object;
  10     return $this;
  11   }
  12
  13   public function getObject() {
  14     return $this->object;
  15   }
  16
  17   protected function getViewer() {
  18     return PhabricatorUser::getOmnipotentUser();
  19   }
  20
  21   abstract protected function buildAbstractDocument(
  22     PhabricatorSearchAbstractDocument $document,
  23     $object);
  24
  25   final public function buildFulltextIndexes() {
  26     $object = $this->getObject();
  27
  28     $extensions = PhabricatorFulltextEngineExtension::getAllExtensions();
  29
  30     $enrich_extensions = array();
  31     $index_extensions = array();
  32     foreach ($extensions as $key => $extension) {
  33       if ($extension->shouldEnrichFulltextObject($object)) {
  34         $enrich_extensions[] = $extension;
  35       }
  36
  37       if ($extension->shouldIndexFulltextObject($object)) {
  38         $index_extensions[] = $extension;
  39       }
  40     }
  41
  42     $document = $this->newAbstractDocument($object);
  43
  44     $this->buildAbstractDocument($document, $object);
  45
  46     foreach ($enrich_extensions as $extension) {
  47       $extension->enrichFulltextObject($object, $document);
  48     }
  49
  50     foreach ($index_extensions as $extension) {
  51       $extension->indexFulltextObject($object, $document);
  52     }
  53
  54     PhabricatorSearchService::reindexAbstractDocument($document);
  55   }
  56
  57   protected function newAbstractDocument($object) {
  58     $phid = $object->getPHID();
  59     return id(new PhabricatorSearchAbstractDocument())
  60       ->setPHID($phid)
  61       ->setDocumentType(phid_get_type($phid));
  62   }
  63
  64 }