src/applications/drydock/query/DrydockBlueprintQuery.php

   1 <?php
   2
   3 final class DrydockBlueprintQuery extends DrydockQuery {
   4
   5   private $ids;
   6   private $phids;
   7   private $blueprintClasses;
   8   private $datasourceQuery;
   9   private $disabled;
  10   private $authorizedPHIDs;
  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 withBlueprintClasses(array $classes) {
  23     $this->blueprintClasses = $classes;
  24     return $this;
  25   }
  26
  27   public function withDatasourceQuery($query) {
  28     $this->datasourceQuery = $query;
  29     return $this;
  30   }
  31
  32   public function withDisabled($disabled) {
  33     $this->disabled = $disabled;
  34     return $this;
  35   }
  36
  37   public function withAuthorizedPHIDs(array $phids) {
  38     $this->authorizedPHIDs = $phids;
  39     return $this;
  40   }
  41
  42   public function withNameNgrams($ngrams) {
  43     return $this->withNgramsConstraint(
  44       new DrydockBlueprintNameNgrams(),
  45       $ngrams);
  46   }
  47
  48   public function newResultObject() {
  49     return new DrydockBlueprint();
  50   }
  51
  52   protected function getPrimaryTableAlias() {
  53     return 'blueprint';
  54   }
  55
  56   protected function loadPage() {
  57     return $this->loadStandardPage($this->newResultObject());
  58   }
  59
  60   protected function willFilterPage(array $blueprints) {
  61     $impls = DrydockBlueprintImplementation::getAllBlueprintImplementations();
  62     foreach ($blueprints as $key => $blueprint) {
  63       $impl = idx($impls, $blueprint->getClassName());
  64       if (!$impl) {
  65         $this->didRejectResult($blueprint);
  66         unset($blueprints[$key]);
  67         continue;
  68       }
  69       $impl = clone $impl;
  70       $blueprint->attachImplementation($impl);
  71     }
  72
  73     return $blueprints;
  74   }
  75
  76   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  77     $where = parent::buildWhereClauseParts($conn);
  78
  79     if ($this->ids !== null) {
  80       $where[] = qsprintf(
  81         $conn,
  82         'blueprint.id IN (%Ld)',
  83         $this->ids);
  84     }
  85
  86     if ($this->phids !== null) {
  87       $where[] = qsprintf(
  88         $conn,
  89         'blueprint.phid IN (%Ls)',
  90         $this->phids);
  91     }
  92
  93     if ($this->datasourceQuery !== null) {
  94       $where[] = qsprintf(
  95         $conn,
  96         'blueprint.blueprintName LIKE %>',
  97         $this->datasourceQuery);
  98     }
  99
 100     if ($this->blueprintClasses !== null) {
 101       $where[] = qsprintf(
 102         $conn,
 103         'blueprint.className IN (%Ls)',
 104         $this->blueprintClasses);
 105     }
 106
 107     if ($this->disabled !== null) {
 108       $where[] = qsprintf(
 109         $conn,
 110         'blueprint.isDisabled = %d',
 111         (int)$this->disabled);
 112     }
 113
 114     return $where;
 115   }
 116
 117   protected function shouldGroupQueryResultRows() {
 118     if ($this->authorizedPHIDs !== null) {
 119       return true;
 120     }
 121     return parent::shouldGroupQueryResultRows();
 122   }
 123
 124   protected function buildJoinClauseParts(AphrontDatabaseConnection $conn) {
 125     $joins = parent::buildJoinClauseParts($conn);
 126
 127     if ($this->authorizedPHIDs !== null) {
 128       $joins[] = qsprintf(
 129         $conn,
 130         'JOIN %T authorization
 131           ON authorization.blueprintPHID = blueprint.phid
 132           AND authorization.objectPHID IN (%Ls)
 133           AND authorization.objectAuthorizationState = %s
 134           AND authorization.blueprintAuthorizationState = %s',
 135         id(new DrydockAuthorization())->getTableName(),
 136         $this->authorizedPHIDs,
 137         DrydockAuthorization::OBJECTAUTH_ACTIVE,
 138         DrydockAuthorization::BLUEPRINTAUTH_AUTHORIZED);
 139     }
 140
 141     return $joins;
 142   }
 143
 144 }