Differential D7726 Diff 17470 src/applications/phragment/query/PhragmentFragmentQuery.php

Changeset View

Standalone View

src/applications/phragment/query/PhragmentFragmentQuery.php

This file was added.

				<?php

				final class PhragmentFragmentQuery
				extends PhabricatorCursorPagedPolicyAwareQuery {

				private $ids;
				private $phids;
				private $paths;
				private $leadingPath;
				private $depths;
				private $needsLatestVersion;

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

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

				public function withPaths(array $paths) {
				$this->paths = $paths;
				return $this;
				}

				public function withLeadingPath($path) {
				$this->leadingPath = $path;
				return $this;
				}

				public function withDepths($depths) {
				$this->depths = $depths;
				return $this;
				}

				public function needLatestVersion($need_latest_version) {
				$this->needsLatestVersion = $need_latest_version;
				return $this;
				}

				public function loadPage() {
				$table = new PhragmentFragment();
				$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);
				}

				protected function buildWhereClause($conn_r) {
				$where = array();

				if ($this->ids) {
				$where[] = qsprintf(
				$conn_r,
				'id IN (%Ld)',
				$this->ids);
				}

				if ($this->phids) {
				$where[] = qsprintf(
				$conn_r,
				'phid IN (%Ls)',
				$this->phids);
				}

				if ($this->paths) {
				$where[] = qsprintf(
				$conn_r,
				'path IN (%Ls)',
				$this->paths);
				}

				if ($this->leadingPath) {
				$where[] = qsprintf(
				$conn_r,
				'path LIKE %>',
				$this->leadingPath);
				}

				if ($this->depths) {
				$where[] = qsprintf(
				$conn_r,
				'depth IN (%Ld)',
				$this->depths);
				}

				$where[] = $this->buildPagingClause($conn_r);

				return $this->formatWhereClause($where);
				}

				protected function didFilterPage(array $page) {
				if ($this->needsLatestVersion) {
				$versions = array();

				$version_phids = array_filter(mpull($page, 'getLatestVersionPHID'));
				if ($version_phids) {
				$versions = id(new PhabricatorObjectQuery())
				->setViewer($this->getViewer())
				->withPHIDs($version_phids)
				->setParentQuery($this)
				->execute();
				$versions = mpull($versions, null, 'getPHID');
				}

				foreach ($page as $key => $fragment) {
				$version_phid = $fragment->getLatestVersionPHID();
				if (empty($versions[$version_phid])) {
				unset($page[$key]);
				continue;
				}
				$fragment->attachLatestVersion($versions[$version_phid]);
				}
				}

				return $page;
				}

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