Changeset View
Changeset View
Standalone View
Standalone View
src/applications/tokens/query/PhabricatorTokenGivenQuery.php
- This file was added.
| <?php | |||||
| final class PhabricatorTokenGivenQuery | |||||
| extends PhabricatorCursorPagedPolicyAwareQuery { | |||||
| private $authorPHIDs; | |||||
| private $objectPHIDs; | |||||
| private $tokenPHIDs; | |||||
| public function withTokenPHIDs(array $token_phids) { | |||||
| $this->tokenPHIDs = $token_phids; | |||||
| return $this; | |||||
| } | |||||
| public function withObjectPHIDs(array $object_phids) { | |||||
| $this->objectPHIDs = $object_phids; | |||||
| return $this; | |||||
| } | |||||
| public function withAuthorPHIDs(array $author_phids) { | |||||
| $this->authorPHIDs = $author_phids; | |||||
| return $this; | |||||
| } | |||||
| public function loadPage() { | |||||
| $table = new PhabricatorTokenGiven(); | |||||
| $conn_r = $table->establishConnection('r'); | |||||
| $rows = queryfx_all( | |||||
| $conn_r, | |||||
| 'SELECT * FROM %T %Q %Q %Q', | |||||
| $table->getTableName(), | |||||
| $this->buildWhereClause($conn_r), | |||||
| $this->buildOrderClause($conn_r), | |||||
| $this->buildLimitClause($conn_r)); | |||||
| return $table->loadAllFromArray($rows); | |||||
| } | |||||
| private function buildWhereClause(AphrontDatabaseConnection $conn_r) { | |||||
| $where = array(); | |||||
| if ($this->authorPHIDs) { | |||||
| $where[] = qsprintf( | |||||
| $conn_r, | |||||
| 'authorPHID IN (%Ls)', | |||||
| $this->authorPHIDs); | |||||
| } | |||||
| if ($this->objectPHIDs) { | |||||
| $where[] = qsprintf( | |||||
| $conn_r, | |||||
| 'objectPHID IN (%Ls)', | |||||
| $this->objectPHIDs); | |||||
| } | |||||
| if ($this->tokenPHIDs) { | |||||
| $where[] = qsprintf( | |||||
| $conn_r, | |||||
| 'tokenPHID IN (%Ls)', | |||||
| $this->tokenPHIDs); | |||||
| } | |||||
| return $this->formatWhereClause($where); | |||||
| } | |||||
| public function willFilterPage(array $results) { | |||||
| $object_phids = array_filter(mpull($results, 'getObjectPHID')); | |||||
| if (!$object_phids) { | |||||
| return array(); | |||||
| } | |||||
| $objects = id(new PhabricatorObjectHandleData($object_phids)) | |||||
| ->setViewer($this->getViewer()) | |||||
| ->loadObjects(); | |||||
| foreach ($results as $key => $result) { | |||||
| $phid = $result->getObjectPHID(); | |||||
| if (empty($objects[$phid])) { | |||||
| unset($results[$key]); | |||||
| } else { | |||||
| $result->attachObject($objects[$phid]); | |||||
| } | |||||
| } | |||||
| return $results; | |||||
| } | |||||
| } | |||||