phorge-phorge/src/applications/herald/query/HeraldTranscriptSearchEngine.php

<?php

final class HeraldTranscriptSearchEngine
  extends PhabricatorApplicationSearchEngine {

  public function getApplicationClassName() {
    return 'PhabricatorApplicationHerald';
  }

  public function buildSavedQueryFromRequest(AphrontRequest $request) {
    $saved = new PhabricatorSavedQuery();

    $object_monograms = $request->getStrList('objectMonograms');
    $saved->setParameter('objectMonograms', $object_monograms);

    $ids = $request->getStrList('ids');
    foreach ($ids as $key => $id) {
      if (!$id || !is_numeric($id)) {
        unset($ids[$key]);
      } else {
        $ids[$key] = $id;
      }
    }
    $saved->setParameter('ids', $ids);

    return $saved;
  }

  public function buildQueryFromSavedQuery(PhabricatorSavedQuery $saved) {
    $query = id(new HeraldTranscriptQuery());

    $object_monograms = $saved->getParameter('objectMonograms');
    if ($object_monograms) {
      $objects = id(new PhabricatorObjectQuery())
        ->setViewer($this->requireViewer())
        ->withNames($object_monograms)
        ->execute();
      $query->withObjectPHIDs(mpull($objects, 'getPHID'));
    }

    $ids = $saved->getParameter('ids');
    if ($ids) {
      $query->withIDs($ids);
    }

    return $query;
  }

  public function buildSearchForm(
    AphrontFormView $form,
    PhabricatorSavedQuery $saved) {

    $object_monograms = $saved->getParameter('objectMonograms', array());
    $ids = $saved->getParameter('ids', array());

    $form
      ->appendChild(
        id(new AphrontFormTextControl())
          ->setName('objectMonograms')
          ->setLabel(pht('Object Monograms'))
          ->setValue(implode(', ', $object_monograms)))
      ->appendChild(
        id(new AphrontFormTextControl())
          ->setName('ids')
          ->setLabel(pht('Transcript IDs'))
          ->setValue(implode(', ', $ids)));
  }

  protected function getURI($path) {
    return '/herald/transcript/'.$path;
  }

  public function getBuiltinQueryNames() {
    $names = array();

    $names['all'] = pht('All');

    return $names;
  }

  public function buildSavedQueryFromBuiltin($query_key) {

    $query = $this->newSavedQuery();
    $query->setQueryKey($query_key);

    $viewer_phid = $this->requireViewer()->getPHID();

    switch ($query_key) {
      case 'all':
        return $query;
    }

    return parent::buildSavedQueryFromBuiltin($query_key);
  }

  protected function getRequiredHandlePHIDsForResultList(
    array $transcripts,
    PhabricatorSavedQuery $query) {
    return mpull($transcripts, 'getObjectPHID');
  }

  protected function renderResultList(
    array $transcripts,
    PhabricatorSavedQuery $query,
    array $handles) {
    assert_instances_of($transcripts, 'HeraldTranscript');

    $viewer = $this->requireViewer();

    $list = new PHUIObjectItemListView();
    foreach ($transcripts as $xscript) {
      $view_href = phutil_tag(
        'a',
        array(
          'href' => '/herald/transcript/'.$xscript->getID().'/',
        ),
        pht('View Full Transcript'));

      $item = new PHUIObjectItemView();
      $item->setObjectName($xscript->getID());
      $item->setHeader($view_href);
      if ($xscript->getDryRun()) {
        $item->addAttribute(pht('Dry Run'));
      }
      $item->addAttribute($handles[$xscript->getObjectPHID()]->renderLink());
      $item->addAttribute(
        number_format((int)(1000 * $xscript->getDuration())).' ms');
      $item->addIcon(
        'none',
        phabricator_datetime($xscript->getTime(), $viewer));

      $list->addItem($item);
    }

    return $list;
  }

}