src/applications/releeph/query/ReleephProductQuery.php

   1 <?php
   2
   3 final class ReleephProductQuery
   4   extends PhabricatorCursorPagedPolicyAwareQuery {
   5
   6   private $active;
   7   private $ids;
   8   private $phids;
   9   private $repositoryPHIDs;
  10
  11   const ORDER_ID    = 'order-id';
  12   const ORDER_NAME  = 'order-name';
  13
  14   public function withActive($active) {
  15     $this->active = $active;
  16     return $this;
  17   }
  18
  19   public function setOrder($order) {
  20     switch ($order) {
  21       case self::ORDER_ID:
  22         $this->setOrderVector(array('id'));
  23         break;
  24       case self::ORDER_NAME:
  25         $this->setOrderVector(array('name'));
  26         break;
  27       default:
  28         throw new Exception(pht('Order "%s" not supported.', $order));
  29     }
  30     return $this;
  31   }
  32
  33   public function withIDs(array $ids) {
  34     $this->ids = $ids;
  35     return $this;
  36   }
  37
  38   public function withPHIDs(array $phids) {
  39     $this->phids = $phids;
  40     return $this;
  41   }
  42
  43   public function withRepositoryPHIDs(array $repository_phids) {
  44     $this->repositoryPHIDs = $repository_phids;
  45     return $this;
  46   }
  47
  48   protected function loadPage() {
  49     $table = new ReleephProject();
  50     $conn_r = $table->establishConnection('r');
  51
  52     $rows = queryfx_all(
  53       $conn_r,
  54       'SELECT * FROM %T %Q %Q %Q',
  55       $table->getTableName(),
  56       $this->buildWhereClause($conn_r),
  57       $this->buildOrderClause($conn_r),
  58       $this->buildLimitClause($conn_r));
  59
  60     return $table->loadAllFromArray($rows);
  61   }
  62
  63   protected function willFilterPage(array $projects) {
  64     assert_instances_of($projects, 'ReleephProject');
  65
  66     $repository_phids = mpull($projects, 'getRepositoryPHID');
  67
  68     $repositories = id(new PhabricatorRepositoryQuery())
  69       ->setViewer($this->getViewer())
  70       ->withPHIDs($repository_phids)
  71       ->execute();
  72     $repositories = mpull($repositories, null, 'getPHID');
  73
  74     foreach ($projects as $key => $project) {
  75       $repo = idx($repositories, $project->getRepositoryPHID());
  76       if (!$repo) {
  77         unset($projects[$key]);
  78         continue;
  79       }
  80       $project->attachRepository($repo);
  81     }
  82
  83     return $projects;
  84   }
  85
  86   protected function buildWhereClause(AphrontDatabaseConnection $conn) {
  87     $where = array();
  88
  89     if ($this->active !== null) {
  90       $where[] = qsprintf(
  91         $conn,
  92         'isActive = %d',
  93         (int)$this->active);
  94     }
  95
  96     if ($this->ids !== null) {
  97       $where[] = qsprintf(
  98         $conn,
  99         'id IN (%Ls)',
 100         $this->ids);
 101     }
 102
 103     if ($this->phids !== null) {
 104       $where[] = qsprintf(
 105         $conn,
 106         'phid IN (%Ls)',
 107         $this->phids);
 108     }
 109
 110     if ($this->repositoryPHIDs !== null) {
 111       $where[] = qsprintf(
 112         $conn,
 113         'repositoryPHID IN (%Ls)',
 114         $this->repositoryPHIDs);
 115     }
 116
 117     $where[] = $this->buildPagingClause($conn);
 118
 119     return $this->formatWhereClause($conn, $where);
 120   }
 121
 122   public function getOrderableColumns() {
 123     return parent::getOrderableColumns() + array(
 124       'name' => array(
 125         'column' => 'name',
 126         'unique' => true,
 127         'reverse' => true,
 128         'type' => 'string',
 129       ),
 130     );
 131   }
 132
 133   protected function newPagingMapFromPartialObject($object) {
 134     return array(
 135       'id' => (int)$object->getID(),
 136       'name' => $object->getName(),
 137     );
 138   }
 139
 140   public function getQueryApplicationClass() {
 141     return 'PhabricatorReleephApplication';
 142   }
 143
 144 }