src/applications/phurl/query/PhabricatorPhurlURLQuery.php

   1 <?php
   2
   3 final class PhabricatorPhurlURLQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $ids;
   7   private $phids;
   8   private $names;
   9   private $longURLs;
  10   private $aliases;
  11   private $authorPHIDs;
  12
  13   public function newResultObject() {
  14     return new PhabricatorPhurlURL();
  15   }
  16
  17   public function withIDs(array $ids) {
  18     $this->ids = $ids;
  19     return $this;
  20   }
  21
  22   public function withPHIDs(array $phids) {
  23     $this->phids = $phids;
  24     return $this;
  25   }
  26
  27   public function withNames(array $names) {
  28     $this->names = $names;
  29     return $this;
  30   }
  31
  32   public function withNameNgrams($ngrams) {
  33     return $this->withNgramsConstraint(
  34       id(new PhabricatorPhurlURLNameNgrams()),
  35       $ngrams);
  36   }
  37
  38   public function withLongURLs(array $long_urls) {
  39     $this->longURLs = $long_urls;
  40     return $this;
  41   }
  42
  43   public function withAliases(array $aliases) {
  44     $this->aliases = $aliases;
  45     return $this;
  46   }
  47
  48   public function withAuthorPHIDs(array $author_phids) {
  49     $this->authorPHIDs = $author_phids;
  50     return $this;
  51   }
  52
  53   protected function loadPage() {
  54     return $this->loadStandardPage($this->newResultObject());
  55   }
  56
  57   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  58     $where = parent::buildWhereClauseParts($conn);
  59
  60     if ($this->ids !== null) {
  61       $where[] = qsprintf(
  62         $conn,
  63         'url.id IN (%Ld)',
  64         $this->ids);
  65     }
  66
  67     if ($this->phids !== null) {
  68       $where[] = qsprintf(
  69         $conn,
  70         'url.phid IN (%Ls)',
  71         $this->phids);
  72     }
  73
  74     if ($this->authorPHIDs !== null) {
  75       $where[] = qsprintf(
  76         $conn,
  77         'url.authorPHID IN (%Ls)',
  78         $this->authorPHIDs);
  79     }
  80
  81     if ($this->names !== null) {
  82       $where[] = qsprintf(
  83         $conn,
  84         'url.name IN (%Ls)',
  85         $this->names);
  86     }
  87
  88     if ($this->longURLs !== null) {
  89       $where[] = qsprintf(
  90         $conn,
  91         'url.longURL IN (%Ls)',
  92         $this->longURLs);
  93     }
  94
  95     if ($this->aliases !== null) {
  96       $where[] = qsprintf(
  97         $conn,
  98         'url.alias IN (%Ls)',
  99         $this->aliases);
 100     }
 101
 102     return $where;
 103   }
 104
 105   protected function getPrimaryTableAlias() {
 106     return 'url';
 107   }
 108
 109   public function getQueryApplicationClass() {
 110     return 'PhabricatorPhurlApplication';
 111   }
 112 }