src/applications/packages/query/PhabricatorPackagesPackageQuery.php

   1 <?php
   2
   3 final class PhabricatorPackagesPackageQuery
   4   extends PhabricatorPackagesQuery {
   5
   6   private $ids;
   7   private $phids;
   8   private $publisherPHIDs;
   9   private $packageKeys;
  10   private $fullKeys;
  11
  12   public function withIDs(array $ids) {
  13     $this->ids = $ids;
  14     return $this;
  15   }
  16
  17   public function withPHIDs(array $phids) {
  18     $this->phids = $phids;
  19     return $this;
  20   }
  21
  22   public function withPublisherPHIDs(array $phids) {
  23     $this->publisherPHIDs = $phids;
  24     return $this;
  25   }
  26
  27   public function withPackageKeys(array $keys) {
  28     $this->packageKeys = $keys;
  29     return $this;
  30   }
  31
  32   public function withFullKeys(array $keys) {
  33     $this->fullKeys = $keys;
  34     return $this;
  35   }
  36
  37   public function withNameNgrams($ngrams) {
  38     return $this->withNgramsConstraint(
  39       new PhabricatorPackagesPackageNameNgrams(),
  40       $ngrams);
  41   }
  42
  43   public function newResultObject() {
  44     return new PhabricatorPackagesPackage();
  45   }
  46
  47   protected function loadPage() {
  48     return $this->loadStandardPage($this->newResultObject());
  49   }
  50
  51   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  52     $where = parent::buildWhereClauseParts($conn);
  53
  54     if ($this->ids !== null) {
  55       $where[] = qsprintf(
  56         $conn,
  57         'p.id IN (%Ld)',
  58         $this->ids);
  59     }
  60
  61     if ($this->phids !== null) {
  62       $where[] = qsprintf(
  63         $conn,
  64         'p.phid IN (%Ls)',
  65         $this->phids);
  66     }
  67
  68     if ($this->publisherPHIDs !== null) {
  69       $where[] = qsprintf(
  70         $conn,
  71         'p.publisherPHID IN (%Ls)',
  72         $this->publisherPHIDs);
  73     }
  74
  75     if ($this->packageKeys !== null) {
  76       $where[] = qsprintf(
  77         $conn,
  78         'p.packageKey IN (%Ls)',
  79         $this->packageKeys);
  80     }
  81
  82     if ($this->fullKeys !== null) {
  83       $parts = $this->buildFullKeyClauseParts($conn, $this->fullKeys);
  84       $where[] = qsprintf($conn, '%Q', $parts);
  85     }
  86
  87     return $where;
  88   }
  89
  90   protected function buildJoinClauseParts(AphrontDatabaseConnection $conn) {
  91     $joins = parent::buildJoinClauseParts($conn);
  92
  93     $join_publisher = ($this->fullKeys !== null);
  94     if ($join_publisher) {
  95       $publisher_table = new PhabricatorPackagesPublisher();
  96
  97       $joins[] = qsprintf(
  98         $conn,
  99         'JOIN %T u ON u.phid = p.publisherPHID',
 100         $publisher_table->getTableName());
 101     }
 102
 103     return $joins;
 104   }
 105
 106   protected function willFilterPage(array $packages) {
 107     $publisher_phids = mpull($packages, 'getPublisherPHID');
 108
 109     $publishers = id(new PhabricatorPackagesPublisherQuery())
 110       ->setViewer($this->getViewer())
 111       ->setParentQuery($this)
 112       ->withPHIDs($publisher_phids)
 113       ->execute();
 114     $publishers = mpull($publishers, null, 'getPHID');
 115
 116     foreach ($packages as $key => $package) {
 117       $publisher = idx($publishers, $package->getPublisherPHID());
 118
 119       if (!$publisher) {
 120         unset($packages[$key]);
 121         $this->didRejectResult($package);
 122         continue;
 123       }
 124
 125       $package->attachPublisher($publisher);
 126     }
 127
 128     return $packages;
 129   }
 130
 131   protected function getPrimaryTableAlias() {
 132     return 'p';
 133   }
 134
 135 }