Changeset View
Changeset View
Standalone View
Standalone View
src/applications/conduit/query/PhabricatorConduitTokenQuery.php
- This file was added.
<?php | |||||
final class PhabricatorConduitTokenQuery | |||||
extends PhabricatorCursorPagedPolicyAwareQuery { | |||||
private $ids; | |||||
private $objectPHIDs; | |||||
private $expired; | |||||
public function withExpired($expired) { | |||||
$this->expired = $expired; | |||||
return $this; | |||||
} | |||||
public function withIDs(array $ids) { | |||||
$this->ids = $ids; | |||||
return $this; | |||||
} | |||||
public function withObjectPHIDs(array $phids) { | |||||
$this->objectPHIDs = $phids; | |||||
return $this; | |||||
} | |||||
public function loadPage() { | |||||
$table = new PhabricatorConduitToken(); | |||||
$conn_r = $table->establishConnection('r'); | |||||
$data = 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($data);; | |||||
} | |||||
private function buildWhereClause(AphrontDatabaseConnection $conn_r) { | |||||
$where = array(); | |||||
if ($this->ids !== null) { | |||||
$where[] = qsprintf( | |||||
$conn_r, | |||||
'id IN (%Ld)', | |||||
$this->ids); | |||||
} | |||||
if ($this->objectPHIDs !== null) { | |||||
$where[] = qsprintf( | |||||
$conn_r, | |||||
'objectPHID IN (%Ls)', | |||||
$this->objectPHIDs); | |||||
} | |||||
if ($this->expired !== null) { | |||||
if ($this->expired) { | |||||
$where[] = qsprintf( | |||||
$conn_r, | |||||
'expires <= %d', | |||||
PhabricatorTime::getNow()); | |||||
} else { | |||||
$where[] = qsprintf( | |||||
$conn_r, | |||||
'expires IS NULL OR expires > %d', | |||||
PhabricatorTime::getNow()); | |||||
} | |||||
} | |||||
$where[] = $this->buildPagingClause($conn_r); | |||||
return $this->formatWhereClause($where); | |||||
} | |||||
protected function willFilterPage(array $tokens) { | |||||
$object_phids = mpull($tokens, 'getObjectPHID'); | |||||
$objects = id(new PhabricatorObjectQuery()) | |||||
->setViewer($this->getViewer()) | |||||
->setParentQuery($this) | |||||
->withPHIDs($object_phids) | |||||
->execute(); | |||||
$objects = mpull($objects, null, 'getPHID'); | |||||
foreach ($tokens as $key => $token) { | |||||
$object = idx($objects, $token->getObjectPHID(), null); | |||||
if (!$object) { | |||||
$this->didRejectResult($token); | |||||
unset($tokens[$key]); | |||||
continue; | |||||
} | |||||
$token->attachObject($object); | |||||
} | |||||
return $tokens; | |||||
} | |||||
public function getQueryApplicationClass() { | |||||
return 'PhabricatorConduitApplication'; | |||||
} | |||||
} |