src/applications/drydock/query/DrydockCommandQuery.php

   1 <?php
   2
   3 final class DrydockCommandQuery extends DrydockQuery {
   4
   5   private $ids;
   6   private $targetPHIDs;
   7   private $consumed;
   8
   9   public function withIDs(array $ids) {
  10     $this->ids = $ids;
  11     return $this;
  12   }
  13
  14   public function withTargetPHIDs(array $phids) {
  15     $this->targetPHIDs = $phids;
  16     return $this;
  17   }
  18
  19   public function withConsumed($consumed) {
  20     $this->consumed = $consumed;
  21     return $this;
  22   }
  23
  24   public function newResultObject() {
  25     return new DrydockCommand();
  26   }
  27
  28   protected function loadPage() {
  29     return $this->loadStandardPage($this->newResultObject());
  30   }
  31
  32   protected function willFilterPage(array $commands) {
  33     $target_phids = mpull($commands, 'getTargetPHID');
  34
  35     $targets = id(new PhabricatorObjectQuery())
  36       ->setViewer($this->getViewer())
  37       ->setParentQuery($this)
  38       ->withPHIDs($target_phids)
  39       ->execute();
  40     $targets = mpull($targets, null, 'getPHID');
  41
  42     foreach ($commands as $key => $command) {
  43       $target = idx($targets, $command->getTargetPHID());
  44       if (!$target) {
  45         $this->didRejectResult($command);
  46         unset($commands[$key]);
  47         continue;
  48       }
  49       $command->attachCommandTarget($target);
  50     }
  51
  52     return $commands;
  53   }
  54
  55   protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
  56     $where = parent::buildWhereClauseParts($conn);
  57
  58     if ($this->ids !== null) {
  59       $where[] = qsprintf(
  60         $conn,
  61         'id IN (%Ld)',
  62         $this->ids);
  63     }
  64
  65     if ($this->targetPHIDs !== null) {
  66       $where[] = qsprintf(
  67         $conn,
  68         'targetPHID IN (%Ls)',
  69         $this->targetPHIDs);
  70     }
  71
  72     if ($this->consumed !== null) {
  73       $where[] = qsprintf(
  74         $conn,
  75         'isConsumed = %d',
  76         (int)$this->consumed);
  77     }
  78
  79     return $where;
  80   }
  81
  82 }