2013-07-01 21:38:42 +02:00
|
|
|
<?php
|
|
|
|
|
|
|
|
final class DifferentialDiffQuery
|
|
|
|
extends PhabricatorCursorPagedPolicyAwareQuery {
|
|
|
|
|
|
|
|
private $ids;
|
|
|
|
private $revisionIDs;
|
2013-09-17 22:55:41 +02:00
|
|
|
private $needChangesets = false;
|
|
|
|
private $needArcanistProjects = false;
|
2013-07-01 21:38:42 +02:00
|
|
|
|
|
|
|
public function withIDs(array $ids) {
|
|
|
|
$this->ids = $ids;
|
|
|
|
return $this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public function withRevisionIDs(array $revision_ids) {
|
|
|
|
$this->revisionIDs = $revision_ids;
|
|
|
|
return $this;
|
|
|
|
}
|
|
|
|
|
2013-09-17 22:55:41 +02:00
|
|
|
public function needChangesets($bool) {
|
|
|
|
$this->needChangesets = $bool;
|
|
|
|
return $this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public function needArcanistProjects($bool) {
|
|
|
|
$this->needArcanistProjects = $bool;
|
|
|
|
return $this;
|
|
|
|
}
|
|
|
|
|
2013-07-01 21:38:42 +02:00
|
|
|
public function loadPage() {
|
|
|
|
$table = new DifferentialDiff();
|
|
|
|
$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);
|
|
|
|
}
|
|
|
|
|
2013-09-17 22:55:41 +02:00
|
|
|
public function willFilterPage(array $diffs) {
|
|
|
|
if ($this->needChangesets) {
|
|
|
|
$this->loadChangesets($diffs);
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($this->needArcanistProjects) {
|
|
|
|
$this->loadArcanistProjects($diffs);
|
|
|
|
}
|
|
|
|
|
|
|
|
return $diffs;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function loadChangesets(array $diffs) {
|
|
|
|
foreach ($diffs as $diff) {
|
|
|
|
$diff->attachChangesets(
|
|
|
|
$diff->loadRelatives(new DifferentialChangeset(), 'diffID'));
|
|
|
|
foreach ($diff->getChangesets() as $changeset) {
|
|
|
|
$changeset->attachHunks(
|
|
|
|
$changeset->loadRelatives(new DifferentialHunk(), 'changesetID'));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return $diffs;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function loadArcanistProjects(array $diffs) {
|
|
|
|
$phids = array_filter(mpull($diffs, 'getArcanistProjectPHID'));
|
|
|
|
$projects = array();
|
|
|
|
$project_map = array();
|
|
|
|
if ($phids) {
|
|
|
|
$projects = id(new PhabricatorRepositoryArcanistProject())
|
|
|
|
->loadAllWhere(
|
|
|
|
'phid IN (%Ls)',
|
|
|
|
$phids);
|
|
|
|
$project_map = mpull($projects, null, 'getPHID');
|
2013-09-24 19:48:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
foreach ($diffs as $diff) {
|
|
|
|
$project = null;
|
|
|
|
if ($diff->getArcanistProjectPHID()) {
|
|
|
|
$project = idx($project_map, $diff->getArcanistProjectPHID());
|
2013-09-17 22:55:41 +02:00
|
|
|
}
|
2013-09-24 19:48:40 +02:00
|
|
|
$diff->attachArcanistProject($project);
|
2013-09-17 22:55:41 +02:00
|
|
|
}
|
2013-09-24 19:48:40 +02:00
|
|
|
|
2013-09-17 22:55:41 +02:00
|
|
|
return $diffs;
|
|
|
|
}
|
|
|
|
|
2013-07-01 21:38:42 +02:00
|
|
|
private function buildWhereClause(AphrontDatabaseConnection $conn_r) {
|
|
|
|
$where = array();
|
|
|
|
|
|
|
|
if ($this->ids) {
|
|
|
|
$where[] = qsprintf(
|
|
|
|
$conn_r,
|
|
|
|
'id IN (%Ld)',
|
|
|
|
$this->ids);
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($this->revisionIDs) {
|
|
|
|
$where[] = qsprintf(
|
|
|
|
$conn_r,
|
|
|
|
'revisionID IN (%Ld)',
|
|
|
|
$this->revisionIDs);
|
|
|
|
}
|
|
|
|
|
|
|
|
$where[] = $this->buildPagingClause($conn_r);
|
|
|
|
return $this->formatWhereClause($where);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|