Differential D20308 Diff 48505 src/applications/project/query/PhabricatorProjectTriggerQuery.php

Changeset View

Standalone View

src/applications/project/query/PhabricatorProjectTriggerQuery.php

	<?php			<?php

	final class PhabricatorProjectTriggerQuery			final class PhabricatorProjectTriggerQuery
	extends PhabricatorCursorPagedPolicyAwareQuery {			extends PhabricatorCursorPagedPolicyAwareQuery {

	private $ids;			private $ids;
	private $phids;			private $phids;
				private $activeColumnMin;
				private $activeColumnMax;

				private $needUsage;

	public function withIDs(array $ids) {			public function withIDs(array $ids) {
	$this->ids = $ids;			$this->ids = $ids;
	return $this;			return $this;
	}			}

	public function withPHIDs(array $phids) {			public function withPHIDs(array $phids) {
	$this->phids = $phids;			$this->phids = $phids;
	return $this;			return $this;
	}			}

				public function needUsage($need_usage) {
				$this->needUsage = $need_usage;
				return $this;
				}

				public function withActiveColumnCountBetween($min, $max) {
				$this->activeColumnMin = $min;
				$this->activeColumnMax = $max;
				return $this;
				}

	public function newResultObject() {			public function newResultObject() {
	return new PhabricatorProjectTrigger();			return new PhabricatorProjectTrigger();
	}			}

	protected function loadPage() {			protected function loadPage() {
	return $this->loadStandardPage($this->newResultObject());			return $this->loadStandardPage($this->newResultObject());
	}			}

	protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {			protected function buildWhereClauseParts(AphrontDatabaseConnection $conn) {
	$where = parent::buildWhereClauseParts($conn);			$where = parent::buildWhereClauseParts($conn);

	if ($this->ids !== null) {			if ($this->ids !== null) {
	$where[] = qsprintf(			$where[] = qsprintf(
	$conn,			$conn,
	'id IN (%Ld)',			'trigger.id IN (%Ld)',
	$this->ids);			$this->ids);
	}			}

	if ($this->phids !== null) {			if ($this->phids !== null) {
	$where[] = qsprintf(			$where[] = qsprintf(
	$conn,			$conn,
	'phid IN (%Ls)',			'trigger.phid IN (%Ls)',
	$this->phids);			$this->phids);
	}			}

				if ($this->activeColumnMin !== null) {
				$where[] = qsprintf(
				$conn,
				'trigger_usage.activeColumnCount >= %d',
				$this->activeColumnMin);
				}

				if ($this->activeColumnMax !== null) {
				$where[] = qsprintf(
				$conn,
				'trigger_usage.activeColumnCount <= %d',
				$this->activeColumnMax);
				}

	return $where;			return $where;
	}			}

				protected function buildJoinClauseParts(AphrontDatabaseConnection $conn) {
				$joins = parent::buildJoinClauseParts($conn);

				if ($this->shouldJoinUsageTable()) {
				$joins[] = qsprintf(
				$conn,
				'JOIN %R trigger_usage ON trigger.phid = trigger_usage.triggerPHID',
				new PhabricatorProjectTriggerUsage());
				}

				return $joins;
				}

				private function shouldJoinUsageTable() {
				if ($this->activeColumnMin !== null) {
				return true;
				}

				if ($this->activeColumnMax !== null) {
				return true;
				}

				return false;
				}

				protected function didFilterPage(array $triggers) {
				if ($this->needUsage) {
				$usage_map = id(new PhabricatorProjectTriggerUsage())->loadAllWhere(
				'triggerPHID IN (%Ls)',
				mpull($triggers, 'getPHID'));
				$usage_map = mpull($usage_map, null, 'getTriggerPHID');

				foreach ($triggers as $trigger) {
				$trigger_phid = $trigger->getPHID();

				$usage = idx($usage_map, $trigger_phid);
				if (!$usage) {
				$usage = id(new PhabricatorProjectTriggerUsage())
				->setTriggerPHID($trigger_phid)
				->setExamplePHID(null)
				->setColumnCount(0)
				->setActiveColumnCount(0);
				}

				$trigger->attachUsage($usage);
				}
				}

				return $triggers;
				}

	public function getQueryApplicationClass() {			public function getQueryApplicationClass() {
	return 'PhabricatorProjectApplication';			return 'PhabricatorProjectApplication';
	}			}

				protected function getPrimaryTableAlias() {
				return 'trigger';
				}

	}			}